XCharts/Scripts/BaseChart.cs

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

namespace xcharts
{
    [System.Serializable]
    public enum ChartType
    {
        line,
        bar
    }

    [System.Serializable]
    public class Title
    {
        public bool show = true;
        public string text = "Chart Title";
        public Align align = Align.center;
        public float left;
        public float right;
        public float top = 5;
        public float bottom;
    }

    [System.Serializable]
    public enum Align
    {
        left,                       //左对齐
        right,                      //右对齐
        center                      //居中对齐
    }

    [System.Serializable]
    public enum Location
    {
        left,
        right,
        top,
        bottom,
        start,
        middle,
        center,
        end,
    }

    [System.Serializable]
    public class Legend
    {
        public bool show = true;
        public Location location = Location.right;
        public float itemWidth = 50.0f;
        public float itemHeight = 20.0f;
        public float itemGap = 5;
        public float left;
        public float right = 5;
        public float top;
        public float bottom;
        public List<string> dataList = new List<string>();
        public int checkDataListCount { get; set; }

        private List<bool> dataShowList = new List<bool>();
        private List<Button> dataBtnList = new List<Button>();

        public bool IsShowSeries(int seriesIndex)
        {
            if (seriesIndex < 0 || seriesIndex > dataList.Count - 1) seriesIndex = 0;
            if (seriesIndex >= dataList.Count) return false;
            if (seriesIndex < 0 || seriesIndex > dataShowList.Count - 1)
            {
                return true;
            }
            else
            {
                return dataShowList[seriesIndex];
            }
        }

        public void SetShowData(int index, bool flag)
        {
            dataShowList[index] = flag;
        }

        public void SetDataButton(int index, Button btn)
        {
            if (index < 0 || index > dataBtnList.Count - 1)
            {
                dataBtnList.Add(btn);
            }
            else
            {
                dataBtnList[index] = btn;
            }
        }

        public void SetShowData(string name, bool flag)
        {
            for (int i = 0; i < dataList.Count; i++)
            {
                if (dataList[i].Equals(name))
                {
                    dataShowList[i] = flag;
                    break;
                }
            }
        }

        public Button GetButton(int index)
        {
            return dataBtnList[index];
        }
    }

    [System.Serializable]
    public class Tooltip
    {
        public bool show;

        public int DataIndex { get; set; }
        public int LastDataIndex { get; set; }
        public float Width { get { return bgRect.sizeDelta.x; } }
        public float Height { get { return bgRect.sizeDelta.y; } }


        private GameObject gameObject;
        private Text text;
        private RectTransform bgRect;

        public void SetObj(GameObject obj)
        {
            gameObject = obj;
            bgRect = gameObject.GetComponent<RectTransform>();
            text = gameObject.GetComponentInChildren<Text>();
        }

        public void SetBackgroundColor(Color color)
        {
            gameObject.GetComponent<Image>().color = color;
        }

        public void SetTextColor(Color color)
        {
            text.color = color;
        }

        public void UpdateTooltipText(string txt)
        {
            text.text = txt;
            bgRect.sizeDelta = new Vector2(text.preferredWidth + 8, text.preferredHeight + 8);
        }

        public void SetActive(bool flag)
        {
            gameObject.SetActive(flag);
        }

        public void UpdatePos(Vector2 pos)
        {
            gameObject.transform.localPosition = pos;
        }

        public Vector3 GetPos()
        {
            return gameObject.transform.localPosition;
        }
    }

    [System.Serializable]
    public class Series
    {
        public string name;
        public int showDataNumber = 0;
        public List<float> dataList = new List<float>();

        public float Max
        {
            get
            {
                float max = 0;
                foreach (var data in dataList)
                {
                    if (data > max)
                    {
                        max = data;
                    }
                }
                return max;
            }
        }

        public float Total
        {
            get
            {
                float total = 0;
                foreach (var data in dataList)
                {
                    total += data;
                }
                return total;
            }
        }

        public void AddData(float value)
        {
            if (dataList.Count >= showDataNumber && showDataNumber != 0)
            {
                dataList.RemoveAt(0);
            }
            dataList.Add(value);
        }

        public float GetData(int index)
        {
            if (index >= 0 && index <= dataList.Count - 1)
            {
                return dataList[index];
            }
            return 0;
        }

        public void UpdataData(int index, float value)
        {
            if (index >= 0 && index <= dataList.Count - 1)
            {
                dataList[index] = value;
            }
        }
    }

    public class BaseChart : MaskableGraphic
    {
        private const string TILTE_TEXT = "title";
        private const string LEGEND_TEXT = "legend";
        [SerializeField]
        protected Theme theme = Theme.Dark;
        [SerializeField]
        protected ThemeInfo themeInfo = new ThemeInfo();
        [SerializeField]
        protected Title title = new Title();
        [SerializeField]
        protected Legend legend = new Legend();
        [SerializeField]
        protected Tooltip tooltip = new Tooltip();
        [SerializeField]
        protected List<Series> seriesList = new List<Series>();

        private Theme checkTheme = 0;
        private Title checkTitle = new Title();
        private Legend checkLegend = new Legend();

        protected Text titleText;
        protected List<Text> legendTextList = new List<Text>();
        protected float chartWid { get { return rectTransform.sizeDelta.x; } }
        protected float chartHig { get { return rectTransform.sizeDelta.y; } }

        protected override void Awake()
        {
            themeInfo = ThemeInfo.Dark;
            rectTransform.anchorMax = Vector2.zero;
            rectTransform.anchorMin = Vector2.zero;
            rectTransform.pivot = Vector2.zero;
            InitTitle();
            InitLegend();
            InitTooltip();
        }

        protected virtual void Update()
        {
            CheckTheme();
            CheckTile();
            CheckLegend();
            CheckTooltip();
        }

        protected override void OnDestroy()
        {
            for (int i = transform.childCount - 1; i >= 0; i--)
            {
                DestroyImmediate(transform.GetChild(i).gameObject);
            }
        }

        public void AddData(string legend, float value)
        {
            for (int i = 0; i < seriesList.Count; i++)
            {
                if (seriesList[i].name.Equals(legend))
                {
                    seriesList[i].AddData(value);
                    break;
                }
            }
            RefreshChart();
        }

        public void UpdateData(string legend, float value, int dataIndex = 0)
        {
            for (int i = 0; i < seriesList.Count; i++)
            {
                if (seriesList[i].name.Equals(legend))
                {
                    seriesList[i].UpdataData(dataIndex, value);
                    break;
                }
            }
            RefreshChart();
        }

        public void UpdateData(int legendIndex, float value, int dataIndex = 0)
        {
            for (int i = 0; i < seriesList.Count; i++)
            {
                if (i == legendIndex)
                {
                    seriesList[i].UpdataData(dataIndex, value);
                    break;
                }
            }
            RefreshChart();
        }

        public void UpdateTheme(Theme theme)
        {
            this.theme = theme;
            OnThemeChanged();
            SetAllDirty();
        }

        protected void HideChild(string match = null)
        {
            for (int i = 0; i < transform.childCount; i++)
            {
                if (match == null)
                    transform.GetChild(i).gameObject.SetActive(false);
                else
                {
                    var go = transform.GetChild(i);
                    if (go.name.StartsWith(match))
                    {
                        go.gameObject.SetActive(false);
                    }
                }
            }
        }

        private void InitTitle()
        {
            TextAnchor anchor = TextAnchor.MiddleCenter;
            Vector2 anchorMin = new Vector2(0, 0);
            Vector2 anchorMax = new Vector2(0, 0);
            Vector3 titlePosition;
            float titleWid = 200;
            float titleHig = 20;
            switch (title.align)
            {
                case Align.left:
                    anchor = TextAnchor.MiddleLeft;
                    titlePosition = new Vector3(title.left, chartHig - title.top, 0);
                    break;
                case Align.right:
                    anchor = TextAnchor.MiddleRight;
                    titlePosition = new Vector3(chartWid - title.right - titleWid,
                        chartHig - title.top, 0);
                    break;
                case Align.center:
                    anchor = TextAnchor.MiddleCenter;
                    titlePosition = new Vector3(chartWid / 2 - titleWid / 2, chartHig - title.top, 0);
                    break;
                default:
                    anchor = TextAnchor.MiddleCenter;
                    titlePosition = new Vector3(0, -title.top, 0);
                    break;
            }
            titleText = ChartUtils.AddTextObject(TILTE_TEXT, transform, themeInfo.font,
                        themeInfo.textColor, anchor, anchorMin, anchorMax, new Vector2(0, 1),
                        new Vector2(titleWid, titleHig), 16);
            titleText.alignment = anchor;
            titleText.gameObject.SetActive(title.show);
            titleText.transform.localPosition = titlePosition;
            titleText.text = title.text;
        }

        private void InitLegend()
        {
            for (int i = 0; i < legend.dataList.Count; i++)
            {
                //LegendData data = legend.dataList[i];
                Button btn = ChartUtils.AddButtonObject(LEGEND_TEXT + i, transform, themeInfo.font,
                    themeInfo.textColor, Vector2.zero, Vector2.zero, Vector2.zero,
                    new Vector2(legend.itemWidth, legend.itemHeight));
                legend.SetDataButton(i, btn);
                Color bcolor = themeInfo.GetColor(i);
                btn.gameObject.SetActive(legend.show);
                btn.transform.localPosition = GetLegendPosition(i);
                btn.GetComponent<Image>().color = bcolor;
                btn.GetComponentInChildren<Text>().text = legend.dataList[i];
                btn.onClick.AddListener(delegate ()
                {
                    legend.SetShowData(i, !legend.IsShowSeries(i));
                    btn.GetComponent<Image>().color = legend.IsShowSeries(i) ?
                        themeInfo.GetColor(i) : themeInfo.unableColor;
                    OnYMaxValueChanged();
                    OnLegendButtonClicked();
                    RefreshChart();
                });
            }
        }

        private void InitTooltip()
        {
            GameObject obj = ChartUtils.AddTooltipObject("tooltip", transform, themeInfo.font);
            tooltip.SetObj(obj);
            tooltip.SetBackgroundColor(themeInfo.tooltipBackgroundColor);
            tooltip.SetTextColor(themeInfo.tooltipTextColor);
            tooltip.SetActive(false);
        }

        private Vector3 GetLegendPosition(int i)
        {
            int legendCount = legend.dataList.Count;
            switch (legend.location)
            {
                case Location.bottom:
                case Location.top:
                    float startX = legend.left;
                    if (startX <= 0)
                    {
                        startX = (chartWid - (legendCount * legend.itemWidth -
                            (legendCount - 1) * legend.itemGap)) / 2;
                    }
                    float posY = legend.location == Location.bottom ?
                        legend.bottom : chartHig - legend.top - legend.itemHeight;
                    return new Vector3(startX + i * (legend.itemWidth + legend.itemGap), posY, 0);
                case Location.left:
                case Location.right:
                    float startY = 0;
                    if (legend.top > 0)
                    {
                        startY = chartHig - legend.top - legend.itemHeight;
                    }
                    else if (startY <= 0)
                    {
                        float legendHig = legendCount * legend.itemHeight - (legendCount - 1) * legend.itemGap;
                        float offset = (chartHig - legendHig) / 2;
                        startY = chartHig - offset - legend.itemHeight;
                    }
                    float posX = legend.location == Location.left ?
                        legend.left :
                        chartWid - legend.right - legend.itemWidth;
                    return new Vector3(posX, startY - i * (legend.itemHeight + legend.itemGap), 0);
                default: break;
            }
            return Vector3.zero;
        }

        protected float GetMaxValue()
        {
            float max = 0;
            for (int i = 0; i < seriesList.Count; i++)
            {
                if (legend.IsShowSeries(i) && seriesList[i].Max > max) max = seriesList[i].Max;
            }
            int bigger = (int)(max * 1.3f);
            return bigger < 10 ? bigger : bigger - bigger % 10;
        }

        private void CheckTheme()
        {
            if (checkTheme != theme)
            {
                checkTheme = theme;
                OnThemeChanged();
            }
        }

        private void CheckTile()
        {
            if (checkTitle == null || title == null) return;
            if (checkTitle.align != title.align ||
                checkTitle.left != title.left ||
                checkTitle.right != title.right ||
                checkTitle.top != title.top)
            {
                checkTitle.align = title.align;
                checkTitle.left = title.left;
                checkTitle.right = title.right;
                checkTitle.top = title.top;
                OnTitleChanged();
            }
        }

        private void CheckLegend()
        {
            if (checkLegend.checkDataListCount != legend.dataList.Count)
            {
                checkLegend.checkDataListCount = legend.dataList.Count;
                OnLegendDataListChanged();
            }

            if (checkLegend.itemWidth != legend.itemWidth ||
                checkLegend.itemHeight != legend.itemHeight ||
                checkLegend.itemGap != legend.itemGap ||
                checkLegend.left != legend.left ||
                checkLegend.right != legend.right ||
                checkLegend.bottom != legend.bottom ||
                checkLegend.top != legend.top ||
                checkLegend.location != legend.location ||
                checkLegend.show != legend.show)
            {
                checkLegend.itemWidth = legend.itemWidth;
                checkLegend.itemHeight = legend.itemHeight;
                checkLegend.itemGap = legend.itemGap;
                checkLegend.left = legend.left;
                checkLegend.right = legend.right;
                checkLegend.bottom = legend.bottom;
                checkLegend.top = legend.top;
                checkLegend.location = legend.location;
                checkLegend.show = legend.show;
                OnLegendChanged();
            }
        }

        private void CheckTooltip()
        {
            if (!tooltip.show) return;
            tooltip.DataIndex = 0;
            Vector2 local;

            if (!RectTransformUtility.ScreenPointToLocalPointInRectangle(rectTransform,
                Input.mousePosition, null, out local))
                return;

            if (local.x < 0 || local.x > chartWid ||
                local.y < 0 || local.y > chartHig)
                return;

            CheckTootipArea(local);
        }

        protected virtual void CheckTootipArea(Vector2 localPostion)
        {
        }

        protected virtual void OnThemeChanged()
        {
            switch (theme)
            {
                case Theme.Dark:
                    themeInfo.Copy(ThemeInfo.Dark);
                    break;
                case Theme.Default:
                    themeInfo.Copy(ThemeInfo.Default);
                    break;
                case Theme.Light:
                    themeInfo.Copy(ThemeInfo.Light);
                    break;
            }
            InitTitle();
            InitLegend();
        }

        protected virtual void OnTitleChanged()
        {
            InitTitle();
        }

        protected virtual void OnLegendChanged()
        {
            for (int i = 0; i < legend.dataList.Count; i++)
            {
                Button btn = legend.GetButton(i);
                btn.GetComponent<RectTransform>().sizeDelta =
                    new Vector2(legend.itemWidth, legend.itemHeight);
                Text txt = btn.GetComponentInChildren<Text>();
                txt.transform.GetComponent<RectTransform>().sizeDelta =
                    new Vector2(legend.itemWidth, legend.itemHeight);
                txt.transform.localPosition = Vector3.zero;
                btn.transform.localPosition = GetLegendPosition(i);
                btn.gameObject.SetActive(legend.show);
            }
        }

        protected virtual void OnLegendDataListChanged()
        {
            InitLegend();
        }

        protected virtual void OnYMaxValueChanged()
        {
        }

        protected virtual void OnLegendButtonClicked()
        {
        }

        protected void RefreshChart()
        {
            int tempWid = (int)chartWid;
            rectTransform.SetSizeWithCurrentAnchors(RectTransform.Axis.Horizontal, tempWid - 1);
            rectTransform.SetSizeWithCurrentAnchors(RectTransform.Axis.Horizontal, tempWid);
        }

        protected virtual void RefreshTooltip()
        {
        }

        protected override void OnPopulateMesh(VertexHelper vh)
        {
            vh.Clear();
            DrawBackground(vh);
            DrawChart(vh);
            DrawTooltip(vh);
        }

        protected virtual void DrawChart(VertexHelper vh)
        {
        }

        protected virtual void DrawTooltip(VertexHelper vh)
        {
        }

        private void DrawBackground(VertexHelper vh)
        {
            // draw bg
            Vector3 p1 = new Vector3(0, chartHig);
            Vector3 p2 = new Vector3(chartWid, chartHig);
            Vector3 p3 = new Vector3(chartWid, 0);
            Vector3 p4 = new Vector3(0, 0);
            ChartUtils.DrawPolygon(vh, p1, p2, p3, p4, themeInfo.backgroundColor);
        }
    }
}