XCharts/Assets/XChartsDemo/Runtime/Demo.cs

/************************************************/
/*                                              */
/*     Copyright (c) 2018 - 2021 monitor1394    */
/*     https://github.com/monitor1394           */
/*                                              */
/************************************************/

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

namespace XChartsDemo
{
    [System.Serializable]
    public class ChartModule
    {
        [SerializeField] private string m_Name;
        [SerializeField] private string m_SubName;
        [SerializeField] private int m_Column = 3;

        [SerializeField] private string m_Title;
        [SerializeField] private bool m_Selected;
        [SerializeField] private GameObject m_Panel;

        public int column { get { return m_Column; } set { m_Column = value; } }
        public string name { get { return m_Name; } set { m_Name = value; } }
        public string subName { get { return m_SubName; } set { m_SubName = value; } }
        public string title { get { return m_Title; } set { m_Title = value; } }
        public bool select { get { return m_Selected; } set { m_Selected = value; } }
        public GameObject panel { get { return m_Panel; } set { m_Panel = value; } }
        public Button button { get; set; }
    }

    [DisallowMultipleComponent]
    [ExecuteInEditMode]
    public class Demo : MonoBehaviour
    {
        [SerializeField] private float m_LeftWidth = 150;
        [SerializeField] private float m_LeftButtonHeight = 60;
        [SerializeField] private Vector2 m_ChartSpacing = new Vector2(5, 5);
        [SerializeField] private Vector2 m_ChartSizeRatio = new Vector2(5, 3);
        [SerializeField] private Color m_ButtonNormalColor;
        [SerializeField] private Color m_ButtonSelectedColor;
        [SerializeField] private Color m_ButtonHighlightColor;
        [SerializeField] private List<ChartModule> m_ChartModule = new List<ChartModule>();

        private GameObject m_BtnClone;
        private Theme m_SelectedTheme;
        private int m_LastSelectedModuleIndex;
        private float m_LastCheckLeftWidth;

        private Button m_DefaultThemeButton;
        private Button m_LightThemeButton;
        private Button m_DarkThemeButton;

        private Text m_Title;

        private ScrollRect m_ScrollRect;
        private Mask m_Mark;

        void Awake()
        {
            m_SelectedTheme = Theme.Default;

            m_ButtonNormalColor = ChartHelper.GetColor("#293C55FF");
            m_ButtonSelectedColor = ChartHelper.GetColor("#e43c59ff");
            m_ButtonHighlightColor = ChartHelper.GetColor("#0E151FFF");
            Init();
        }


        void Init()
        {
            m_ScrollRect = transform.Find("chart_detail").GetComponent<ScrollRect>();
            m_Mark = transform.Find("chart_detail/Viewport").GetComponent<Mask>();
            m_Mark.enabled = true;
            m_Title = transform.Find("chart_title/Text").GetComponent<Text>();
            InitThemeButton();
            InitModuleButton();

            InitSize();
        }

        void InitSize()
        {
            UIUtil.SetRectTransformWidth(transform, m_LeftWidth, "chart_list");
            UIUtil.SetRectTransformLeft(transform, m_LeftWidth, "chart_detail");
            UIUtil.SetRectTransformLeft(transform, m_LeftWidth, "chart_title");
            UIUtil.SetGridLayoutGroup(transform, new Vector2(m_LeftWidth, m_LeftButtonHeight), new Vector2(0, 2), "chart_list");
            foreach (var module in m_ChartModule)
            {
                SetChartRootInfo(module);
            }
        }

        private void SetChartGridLayoutGroup(GridLayoutGroup grid, int column)
        {
            if (grid == null) return;
            var screenWidth = Screen.width;
            var screenHeight = Screen.height;
#if UNITY_EDITOR
            screenWidth = Camera.main.pixelWidth;
#endif
            var chartWidth = (screenWidth - m_LeftWidth - m_ChartSpacing.x * (column + 1)) / column;

            var rect = grid.gameObject.GetComponent<RectTransform>();
            rect.anchorMin = new Vector2(0, 1);
            rect.anchorMax = new Vector2(1, 1);
            rect.pivot = new Vector2(0, 1);
            rect.anchoredPosition = Vector2.zero;
            rect.sizeDelta = new Vector2(0, rect.sizeDelta.y);
            grid.spacing = m_ChartSpacing;
            grid.cellSize = new Vector2(chartWidth, chartWidth * m_ChartSizeRatio.y / m_ChartSizeRatio.x);
        }

        private void SetChartRootInfo(ChartModule module)
        {
            var chartRoot = module.panel;
            var grid = chartRoot.GetComponent<GridLayoutGroup>();
            var hig = Mathf.CeilToInt(chartRoot.transform.childCount * 1f / module.column) * (grid.cellSize.y + grid.spacing.y);
            SetChartGridLayoutGroup(grid, module.column);
            UIUtil.SetRectTransformHeight(chartRoot.transform, hig);
        }

        void ResetParam()
        {
            var charts = transform.GetComponentsInChildren<BaseChart>();
            foreach (var chart in charts)
            {
                chart.RemoveChartObject();
            }
        }

        void Update()
        {
#if UNITY_EDITOR
            if (m_ChartModule.Count <= 0) return;
            int selectedModuleIndex = -1;
            for (int i = 0; i < m_ChartModule.Count; i++)
            {
                if (selectedModuleIndex >= 0 && i > selectedModuleIndex)
                {
                    m_ChartModule[i].select = false;
                }
                else if (m_ChartModule[i].select)
                {
                    selectedModuleIndex = i;
                }
            }
            if (selectedModuleIndex < 0) selectedModuleIndex = 0;
            if (selectedModuleIndex != m_LastSelectedModuleIndex)
            {
                InitModuleButton();
            }
            if (!Application.isPlaying) m_Mark.enabled = false;

            if (m_LastCheckLeftWidth != m_LeftWidth)
            {
                m_LastCheckLeftWidth = m_LeftWidth;
                InitSize();
            }
#endif
        }

        void InitModuleButton()
        {
            var btnPanel = transform.Find("chart_list");
            m_BtnClone = transform.Find("btn_clone").gameObject;
            m_BtnClone.SetActive(false);
            ChartHelper.HideAllObject(btnPanel);
            foreach (var module in m_ChartModule)
            {
                var btnName = "btn_" + module.name;
                GameObject btn;
                if (btnPanel.Find(btnName))
                {
                    btn = btnPanel.Find(btnName).gameObject;
                    btn.name = btnName;
                    ChartHelper.SetActive(btn, true);
                }
                else
                {
                    btn = GameObject.Instantiate(m_BtnClone);
                    btn.SetActive(true);
                    btn.name = btnName;
                    btn.transform.SetParent(btnPanel);
                    btn.transform.localPosition = Vector3.zero;
                }
                btn.transform.localScale = Vector3.one;
                module.button = btn.GetComponent<Button>();
                module.button.transform.Find("Text").GetComponent<Text>().text = module.name.Replace("\\n", "\n");
                module.button.transform.Find("SubText").GetComponent<Text>().text = module.subName.Replace("\\n", "\n");

                ChartHelper.AddEventListener(btn.gameObject, EventTriggerType.PointerDown, (data) =>
                {
                    ClickModule(module);
                });
            }

            for (int i = 0; i < m_ChartModule.Count; i++)
            {
                var module = m_ChartModule[i];
                if (module.select)
                {
                    ClickModule(module);
                    m_LastSelectedModuleIndex = i;
                    break;
                }
            }
        }

        void ClickModule(ChartModule selectedModule)
        {
            foreach (var module in m_ChartModule)
            {
                if (selectedModule != module)
                {
                    var block = module.button.colors;
                    block.highlightedColor = m_ButtonHighlightColor;
                    block.normalColor = m_ButtonNormalColor;
                    module.button.colors = block;
                    module.panel.SetActive(false);
                    module.select = false;
                }
                else
                {
                    var block = module.button.colors;
                    block.highlightedColor = m_ButtonSelectedColor;
                    block.normalColor = m_ButtonSelectedColor;
                    module.button.colors = block;
                    module.panel.SetActive(true);
                    module.select = true;
                }
            }
            m_ScrollRect.content = selectedModule.panel.GetComponent<RectTransform>();
            SetChartRootInfo(selectedModule);
            m_Title.text = string.IsNullOrEmpty(selectedModule.title) ?
                selectedModule.name : selectedModule.title;
        }

        void InitThemeButton()
        {
            m_DefaultThemeButton = transform.Find("chart_theme/btn_default").GetComponent<Button>();
            m_LightThemeButton = transform.Find("chart_theme/btn_light").GetComponent<Button>();
            m_DarkThemeButton = transform.Find("chart_theme/btn_dark").GetComponent<Button>();

            m_DefaultThemeButton.transform.Find("selected").gameObject.SetActive(m_SelectedTheme == Theme.Default);
            m_LightThemeButton.transform.Find("selected").gameObject.SetActive(m_SelectedTheme == Theme.Light);
            m_DarkThemeButton.transform.Find("selected").gameObject.SetActive(m_SelectedTheme == Theme.Dark);

            m_DefaultThemeButton.onClick.AddListener(delegate () { SelecteTheme(Theme.Default); });
            m_LightThemeButton.onClick.AddListener(delegate () { SelecteTheme(Theme.Light); });
            m_DarkThemeButton.onClick.AddListener(delegate () { SelecteTheme(Theme.Dark); });

            //SelecteTheme(Theme.Default);
        }

        void SelecteTheme(Theme theme)
        {
            m_SelectedTheme = theme;
            m_DefaultThemeButton.transform.Find("selected").gameObject.SetActive(m_SelectedTheme == Theme.Default);
            m_LightThemeButton.transform.Find("selected").gameObject.SetActive(m_SelectedTheme == Theme.Light);
            m_DarkThemeButton.transform.Find("selected").gameObject.SetActive(m_SelectedTheme == Theme.Dark);
            var charts = transform.GetComponentsInChildren<BaseChart>();
            foreach (var chart in charts)
            {
                chart.UpdateTheme(theme);
            }
        }
    }
}