smart-bow-infrared/Assets/SmartBow/Scripts/Views/InfraredViewParts/Line.cs

using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UI;

namespace LineUI
{
    [RequireComponent(typeof(CanvasRenderer))]
    [RequireComponent(typeof(RectTransform))]
    public class Line : Graphic
    {
        [SerializeField] private bool loop = false;
        [SerializeField] private float thickness = 1f;
        [SerializeField] private int roundCount = 0;
        [SerializeField] private Vector2 quadrilateralSize = new Vector2(100, 100);
        [SerializeField] private Color quadColor = Color.red;
        [SerializeField] private List<Vector2> screenPositions = new List<Vector2>();
        //[SerializeField] private Color quadTextColor = Color.black;
        //[SerializeField] private int quadFontSize = 14;

        //private List<Vector2> quadsToDraw = new List<Vector2>();
        //private List<string> quadTextToDraw = new List<string>();
        //private List<GameObject> quadObjects = new List<GameObject>();
        //private Stack<GameObject> objectPool = new Stack<GameObject>();


        private const int ObjectPoolLimit = 4;

        public RectTransform RectTr => rectTransform;

        public float MyThickness
        {
            get => thickness;
            set
            {
                if (value >= 10)
                {
                    thickness = value;
                }
            }
        }

        public void SetLine(List<Vector2> screenPositions)
        {
            this.screenPositions = screenPositions;
            SetVerticesDirty();
        }

        protected override void OnPopulateMesh(VertexHelper vh)
        {
            vh.Clear();
            //quadsToDraw.Clear();
            //quadTextToDraw.Clear();

            if (screenPositions.Count < 2)
                return;

            SetLineVertices(vh);
            SetLineTriangles(vh);
            SetQuadrilateralVertices(vh);
        }

        private void SetLineVertices(VertexHelper vh)
        {
            UIVertex vert = UIVertex.simpleVert;
            vert.color = color;

            List<Vector2> _screenPositions = new List<Vector2>(screenPositions);
            if (loop)
            {
                _screenPositions.Add(screenPositions[0]);
                _screenPositions.Add(screenPositions[1]);
            }

            float lastAngle = 0;
            for (int i = 1; i < _screenPositions.Count; i++)
            {
                Vector2 previousPos = _screenPositions[i - 1];
                Vector2 currentPos = _screenPositions[i];
                Vector2 dif = currentPos - previousPos;
                float angle = Vector2.SignedAngle(Vector2.right, dif);

                Vector2 offset = Quaternion.Euler(0, 0, angle) * Vector3.up * thickness;

                if (i > 1)
                {
                    float anglePerRound = (angle - lastAngle) / roundCount;
                    for (int j = 0; j <= roundCount; j++)
                    {
                        float ang = lastAngle + anglePerRound * j;
                        Vector2 roundOffset = Quaternion.Euler(0, 0, ang) * Vector2.up * thickness;
                        vert.position = previousPos - roundOffset;
                        vh.AddVert(vert);
                        vert.position = previousPos + roundOffset;
                        vh.AddVert(vert);
                    }
                }
                lastAngle = angle;

                vert.position = previousPos - offset;
                vh.AddVert(vert);

                vert.position = previousPos + offset;
                vh.AddVert(vert);

                vert.position = currentPos - offset;
                vh.AddVert(vert);

                vert.position = currentPos + offset;
                vh.AddVert(vert);
            }
        }

        private void SetQuadrilateralVertices(VertexHelper vh)
        {
            UIVertex vert = UIVertex.simpleVert;
            vert.color = quadColor;

            List<Vector2> _screenPositions = new List<Vector2>(screenPositions);
            if (loop)
            {
                _screenPositions.Add(screenPositions[0]);
                _screenPositions.Add(screenPositions[1]);
            }

            for (int i = 1; i < _screenPositions.Count; i++)
            {
                Vector2 previousPos = _screenPositions[i - 1];
                Vector2 currentPos = _screenPositions[i];
                if (i > 1)
                {
                    Vector2 prevDir = (_screenPositions[i - 1] - _screenPositions[i - 2]).normalized;
                    Vector2 currentDir = (currentPos - previousPos).normalized;
                    int index = i == 4 ? 4 : i % 4;
                    if (index == 4 || index == 2)
                    {

                        DrawSquareAtCorner(vh, previousPos, prevDir, currentDir, quadrilateralSize.x, quadrilateralSize.y, index);
                    }
                    else {

                        DrawSquareAtCorner(vh, previousPos, prevDir, currentDir, quadrilateralSize.y, quadrilateralSize.x, index);
                    }
                }
            }
        }

        private void DrawSquareAtCorner(VertexHelper vh, Vector2 corner, Vector2 prevDir, Vector2 currentDir, float width, float height, int index)
        {
            UIVertex vert = UIVertex.simpleVert;
            vert.color = quadColor;

            Vector2[] corners = new Vector2[4];
            corners[0] = corner;
            corners[1] = corner + prevDir * -width;
            corners[2] = corner + prevDir * -width + currentDir * height;
            corners[3] = corner + currentDir * height;

            foreach (Vector2 pos in corners)
            {
                vert.position = pos;
                vh.AddVert(vert);
            }

            int startIndex = vh.currentVertCount - 4;
            vh.AddTriangle(startIndex, startIndex + 1, startIndex + 2);
            vh.AddTriangle(startIndex + 2, startIndex + 3, startIndex);

            // Store the quad's center position and text for later use
            //Vector2 quadCenter = corner + (prevDir * -width + currentDir * height) / 2;
            //quadsToDraw.Add(quadCenter);
            //quadTextToDraw.Add(index.ToString());
        }

        private void SetLineTriangles(VertexHelper vh)
        {
            for (int i = 0; i < vh.currentVertCount - 2; i += 2)
            {
                int index = i;
                vh.AddTriangle(index, index + 1, index + 3);
                vh.AddTriangle(index + 3, index + 2, index);
            }
        }

        //private void LateUpdate()
        //{
        //    foreach (var quadObj in quadObjects)
        //    {
        //        if (quadObj != null)
        //        {
        //            objectPool.Push(quadObj);
        //        }
        //    }
        //    quadObjects.Clear();

        //    for (int i = 0; i < quadsToDraw.Count; i++)
        //    {
        //        GameObject quadGO = GetQuadObject();
        //        if (quadGO != null)
        //        {
        //            quadGO.SetActive(true);
        //            quadGO.GetComponent<RectTransform>().anchoredPosition = quadsToDraw[i];

        //            Text quadTextComponent = quadGO.GetComponent<Text>();
        //            quadTextComponent.text = quadTextToDraw[i];
        //            quadTextComponent.fontSize = quadFontSize;
        //            quadTextComponent.color = quadTextColor;

        //            quadTextComponent.alignment = TextAnchor.MiddleCenter;
        //        }
        //    }
        //    quadsToDraw.Clear();
        //    quadTextToDraw.Clear();
        //}

        //private GameObject GetQuadObject()
        //{
        //    if (objectPool.Count > 0)
        //    {
        //        GameObject quadGO = objectPool.Pop();
        //        if (quadGO != null)
        //        {
        //            quadObjects.Add(quadGO);
        //            return quadGO;
        //        }
        //    }

        //    if (quadObjects.Count < ObjectPoolLimit)
        //    {
        //        GameObject newQuadGO = new GameObject("QuadText", typeof(RectTransform), typeof(Text));
        //        newQuadGO.transform.SetParent(this.transform);
        //        Text quadTextComponent = newQuadGO.GetComponent<Text>();
        //        quadTextComponent.alignment = TextAnchor.MiddleCenter;
        //        quadTextComponent.rectTransform.sizeDelta = new Vector2(quadrilateralSize.x, quadrilateralSize.y);

        //        quadObjects.Add(newQuadGO);
        //        return newQuadGO;
        //    }

        //    return null;
        //}
    }
}