BaseGraph.cs

/******************************************/
/*                                        */
/*     Copyright (c) 2021 monitor1394     */
/*     https://github.com/monitor1394     */
/*                                        */
/******************************************/
 
using UnityEngine;
using UnityEngine.UI;
using System;
using UnityEngine.EventSystems;
 
namespace XCharts
{
    [RequireComponent(typeof(CanvasRenderer))]
    public partial class BaseGraph : MaskableGraphic, IPointerDownHandler, IPointerUpHandler,
        IPointerEnterHandler, IPointerExitHandler, IBeginDragHandler, IPointerClickHandler,
        IDragHandler, IEndDragHandler, IScrollHandler
    {
        protected static readonly string s_BackgroundObjectName = "background";
        [SerializeField] protected bool m_MultiComponentMode = false;
        [SerializeField] protected bool m_DebugMode = false;
        [SerializeField] protected bool m_EnableTextMeshPro = false;
        [SerializeField] protected Background m_Background = Background.defaultBackground;
        protected Painter m_Painter;
        protected int m_SiblingIndex;
 
        protected float m_GraphWidth;
        protected float m_GraphHeight;
        protected float m_GraphX;
        protected float m_GraphY;
        protected Vector3 m_GraphPosition = Vector3.zero;
        protected Vector2 m_GraphMinAnchor;
        protected Vector2 m_GraphMaxAnchor;
        protected Vector2 m_GraphPivot;
        protected Vector2 m_GraphSizeDelta;
        protected Vector2 m_GraphAnchoredPosition;
        protected Rect m_GraphRect = new Rect(0, 0, 0, 0);
        protected bool m_RefreshChart = false;
        protected bool m_ForceOpenRaycastTarget;
        protected bool m_IsControlledByLayout = false;
        protected bool m_PainerDirty = false;
        protected bool m_IsOnValidate = false;
        protected Vector3 m_LastLocalPosition;
 
        protected Action<PointerEventData, BaseGraph> m_OnPointerClick;
        protected Action<PointerEventData, BaseGraph> m_OnPointerDown;
        protected Action<PointerEventData, BaseGraph> m_OnPointerUp;
        protected Action<PointerEventData, BaseGraph> m_OnPointerEnter;
        protected Action<PointerEventData, BaseGraph> m_OnPointerExit;
        protected Action<PointerEventData, BaseGraph> m_OnBeginDrag;
        protected Action<PointerEventData, BaseGraph> m_OnDrag;
        protected Action<PointerEventData, BaseGraph> m_OnEndDrag;
        protected Action<PointerEventData, BaseGraph> m_OnScroll;
 
        protected Vector2 graphAnchorMax { get { return m_GraphMinAnchor; } }
        protected Vector2 graphAnchorMin { get { return m_GraphMaxAnchor; } }
        protected Vector2 graphPivot { get { return m_GraphPivot; } }
        public HideFlags chartHideFlags { get { return m_DebugMode ? HideFlags.None : HideFlags.HideInHierarchy; } }
 
        private ScrollRect m_ScrollRect;
 
 
        protected virtual void InitComponent()
        {
            InitPainter();
            InitBackground();
        }
 
        protected override void Awake()
        {
            CheckTextMeshPro();
            m_SiblingIndex = 0;
            if (transform.parent != null)
            {
                m_IsControlledByLayout = transform.parent.GetComponent<LayoutGroup>() != null;
            }
            raycastTarget = false;
            m_LastLocalPosition = transform.localPosition;
            UpdateSize();
            InitComponent();
            CheckIsInScrollRect();
        }
 
        protected override void Start()
        {
            m_RefreshChart = true;
        }
 
        protected virtual void Update()
        {
            CheckSize();
            if (m_IsOnValidate)
            {
                m_IsOnValidate = false;
                m_RefreshChart = true;
                CheckTextMeshPro();
                InitComponent();
            }
            else
            {
                CheckComponent();
            }
            CheckPointerPos();
            CheckRefreshChart();
            CheckRefreshPainter();
        }
 
        protected virtual void SetAllComponentDirty()
        {
#if UNITY_EDITOR
            if (!Application.isPlaying)
            {
                m_IsOnValidate = true;
                Update();
            }
#endif
            m_PainerDirty = true;
            m_Background.SetAllDirty();
        }
 
        protected virtual void CheckComponent()
        {
            CheckComponentDirty(m_Background);
            if (m_PainerDirty)
            {
                InitPainter();
                m_PainerDirty = false;
            }
        }
 
       
 
        private void CheckTextMeshPro()
        {
#if dUI_TextMeshPro
            var enableTextMeshPro = true;
#else
            var enableTextMeshPro = false;
#endif
            if (m_EnableTextMeshPro != enableTextMeshPro)
            {
                m_EnableTextMeshPro = enableTextMeshPro;
                RemoveChartObject();
            }
        }
 
        protected void CheckComponentDirty(ChartComponent component)
        {
            if (component.anyDirty)
            {
                if (component.componentDirty && component.refreshComponent != null)
                {
                    component.refreshComponent.Invoke();
                }
                if (component.vertsDirty)
                {
                    if (component.painter != null)
                    {
                        RefreshPainter(component.painter);
                    }
                }
                component.ClearDirty();
            }
        }
 
#if UNITY_EDITOR
        protected override void Reset()
        {
        }
 
        protected override void OnValidate()
        {
            m_IsOnValidate = true;
        }
#endif
 
        protected override void OnDestroy()
        {
            for (int i = transform.childCount - 1; i >= 0; i--)
            {
                DestroyImmediate(transform.GetChild(i).gameObject);
            }
        }
 
        protected override void OnPopulateMesh(VertexHelper vh)
        {
            vh.Clear();
        }
 
        private void InitBackground()
        {
            m_Background.painter = m_Painter;
            m_Background.refreshComponent = delegate ()
            {
                var backgroundObj = ChartHelper.AddObject(s_BackgroundObjectName, transform, m_GraphMinAnchor,
                m_GraphMaxAnchor, m_GraphPivot, m_GraphSizeDelta);
                m_Background.gameObject = backgroundObj;
                backgroundObj.hideFlags = chartHideFlags;
                var backgroundImage = ChartHelper.GetOrAddComponent<Image>(backgroundObj);
                ChartHelper.UpdateRectTransform(backgroundObj, m_GraphMinAnchor,
                    m_GraphMaxAnchor, m_GraphPivot, m_GraphSizeDelta);
                backgroundImage.sprite = m_Background.image;
                backgroundImage.type = m_Background.imageType;
                backgroundImage.color = m_Background.imageColor;
                backgroundObj.transform.SetSiblingIndex(0);
                backgroundObj.SetActive(m_Background.show);
            };
            m_Background.refreshComponent();
        }
 
        protected virtual void InitPainter()
        {
            m_Painter = ChartHelper.AddPainterObject("painter_b", transform, m_GraphMinAnchor,
                    m_GraphMaxAnchor, m_GraphPivot, new Vector2(m_GraphWidth, m_GraphHeight), chartHideFlags, 1);
            m_Painter.type = Painter.Type.Base;
            m_Painter.onPopulateMesh = OnDrawPainterBase;
        }
 
        private void CheckSize()
        {
            var currWidth = rectTransform.rect.width;
            var currHeight = rectTransform.rect.height;
 
            if (m_GraphWidth == 0 && m_GraphHeight == 0 && (currWidth != 0 || currHeight != 0))
            {
                Awake();
            }
 
            if (m_GraphWidth != currWidth
                || m_GraphHeight != currHeight
                || m_GraphMinAnchor != rectTransform.anchorMin
                || m_GraphMaxAnchor != rectTransform.anchorMax
                || m_GraphAnchoredPosition != rectTransform.anchoredPosition)
            {
                UpdateSize();
            }
            if (!ChartHelper.IsValueEqualsVector3(m_LastLocalPosition, transform.localPosition))
            {
                m_LastLocalPosition = transform.localPosition;
                OnLocalPositionChanged();
            }
        }
 
        protected void UpdateSize()
        {
            m_GraphWidth = rectTransform.rect.width;
            m_GraphHeight = rectTransform.rect.height;
 
            m_GraphMaxAnchor = rectTransform.anchorMax;
            m_GraphMinAnchor = rectTransform.anchorMin;
            m_GraphSizeDelta = rectTransform.sizeDelta;
            m_GraphAnchoredPosition = rectTransform.anchoredPosition;
 
            rectTransform.pivot = LayerHelper.ResetChartPositionAndPivot(m_GraphMinAnchor, m_GraphMaxAnchor,
               m_GraphWidth, m_GraphHeight, ref m_GraphX, ref m_GraphY);
            m_GraphPivot = rectTransform.pivot;
 
            m_GraphRect.x = m_GraphX;
            m_GraphRect.y = m_GraphY;
            m_GraphRect.width = m_GraphWidth;
            m_GraphRect.height = m_GraphHeight;
            m_GraphPosition.x = m_GraphX;
            m_GraphPosition.y = m_GraphY;
 
            OnSizeChanged();
        }
 
        private void CheckPointerPos()
        {
            if (m_ForceOpenRaycastTarget) raycastTarget = true;
            if (IsNeedCheckPointerPos())
            {
                raycastTarget = true;
                if (canvas == null) return;
                Vector2 local;
                if (!ScreenPointToChartPoint(Input.mousePosition, out local))
                {
                    pointerPos = Vector2.zero;
                }
                else
                {
                    pointerPos = local;
                }
            }
            else
            {
                raycastTarget = false;
            }
        }
 
        protected virtual void CheckIsInScrollRect()
        {
            m_ScrollRect = GetComponentInParent<ScrollRect>();
        }
 
        protected virtual bool IsNeedCheckPointerPos()
        {
            return raycastTarget;
        }
 
        protected virtual void CheckRefreshChart()
        {
            if (m_RefreshChart)
            {
                m_Painter.Refresh();
                m_RefreshChart = false;
            }
        }
 
        protected virtual void CheckRefreshPainter()
        {
            m_Painter.CheckRefresh();
        }
 
        internal virtual void RefreshPainter(Painter painter)
        {
            if (painter == null) return;
            painter.Refresh();
        }
 
        protected virtual void OnSizeChanged()
        {
            m_RefreshChart = true;
        }
 
        protected virtual void OnLocalPositionChanged()
        {
        }
 
        protected virtual void OnDrawPainterBase(VertexHelper vh, Painter painter)
        {
            DrawBackground(vh);
            DrawPainterBase(vh);
        }
 
        protected virtual void DrawPainterBase(VertexHelper vh)
        {
        }
 
        protected virtual void DrawBackground(VertexHelper vh)
        {
        }
 
        public virtual void OnPointerClick(PointerEventData eventData)
        {
            if (m_OnPointerClick != null) m_OnPointerClick(eventData, this);
        }
 
        public virtual void OnPointerDown(PointerEventData eventData)
        {
            if (m_OnPointerDown != null) m_OnPointerDown(eventData, this);
        }
 
        public virtual void OnPointerUp(PointerEventData eventData)
        {
            if (m_OnPointerUp != null) m_OnPointerUp(eventData, this);
        }
 
        public virtual void OnPointerEnter(PointerEventData eventData)
        {
            isPointerInChart = true;
            if (m_OnPointerEnter != null) m_OnPointerEnter(eventData, this);
        }
 
        public virtual void OnPointerExit(PointerEventData eventData)
        {
            isPointerInChart = false;
            if (m_OnPointerExit != null) m_OnPointerExit(eventData, this);
        }
 
        public virtual void OnBeginDrag(PointerEventData eventData)
        {
            if (m_ScrollRect != null) m_ScrollRect.OnBeginDrag(eventData);
            if (m_OnBeginDrag != null) m_OnBeginDrag(eventData, this);
        }
 
        public virtual void OnEndDrag(PointerEventData eventData)
        {
            if (m_ScrollRect != null) m_ScrollRect.OnEndDrag(eventData);
            if (m_OnEndDrag != null) m_OnEndDrag(eventData, this);
        }
 
        public virtual void OnDrag(PointerEventData eventData)
        {
            if (m_ScrollRect != null) m_ScrollRect.OnDrag(eventData);
            if (m_OnDrag != null) m_OnDrag(eventData, this);
        }
 
        public virtual void OnScroll(PointerEventData eventData)
        {
            if (m_ScrollRect != null) m_ScrollRect.OnScroll(eventData);
            if (m_OnScroll != null) m_OnScroll(eventData, this);
        }
    }
}