gongyejiqiren/Assets/Third/PDFRenderer/Internal/Scripts/Viewer/PDFBookmarkListItem.cs

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

namespace Paroxe.PdfRenderer.Internal.Viewer
{
	public class PDFBookmarkListItem : UIBehaviour
    {
        [SerializeField]
        private Sprite m_CollapseSprite;
        [SerializeField]
        private Image m_ExpandImage;
        [SerializeField]
        private Sprite m_ExpandSprite;
        [SerializeField]
        private Image m_Highlighted;
        [SerializeField]
        private RectTransform m_HorizontalLine;
        [SerializeField]
        private RectTransform m_Internal;
        [SerializeField]
        private RectTransform m_NextSibling;
        [SerializeField]
        private Text m_Title;
        [SerializeField]
        private RectTransform m_VerticalLine;
        [SerializeField]
        private RectTransform m_VerticalLine2;

#if !UNITY_WEBGL
        private CanvasGroup m_CanvasGroup;
        private List<PDFBookmarkListItem> m_ChildrenItems;
        private bool m_Expanded;
        private bool m_Initialized;
        private bool m_IsLastSibling;
        private float m_LastClickTimestamp;
        private LayoutElement m_LayoutElement;
        private PDFBookmark m_PDFBookmark;
        private int m_SizeAdjusted;
        private PDFBookmarksViewer m_BookmarksViewer;

        private RectTransform RectTransform
        {
	        get { return (RectTransform)transform; }
        }

        public void Initialize(PDFBookmark bookmark, int indent, bool lastSibling)
        {
            m_ChildrenItems = new List<PDFBookmarkListItem>();

            m_IsLastSibling = lastSibling;
            m_PDFBookmark = bookmark;
            m_HorizontalLine.gameObject.SetActive(true);
            m_VerticalLine.gameObject.SetActive(false);
            m_VerticalLine2.gameObject.SetActive(true);
            m_ExpandImage.gameObject.SetActive(true);

            if (m_PDFBookmark.IsTopLevelBookmark && m_PDFBookmark.ChildCount == 0)
            {
                m_HorizontalLine.gameObject.SetActive(false);
                m_ExpandImage.gameObject.SetActive(false);
            }
            else if (m_PDFBookmark.ChildCount > 0)
            {
                m_HorizontalLine.gameObject.SetActive(true);
                m_ExpandImage.gameObject.SetActive(true);
            }
            else if (!m_PDFBookmark.IsTopLevelBookmark)
            {
                m_HorizontalLine.gameObject.SetActive(true);
                m_HorizontalLine.offsetMin = new Vector2(m_HorizontalLine.offsetMin.x + 6, m_HorizontalLine.offsetMin.y);

                if (!m_IsLastSibling)
	                m_VerticalLine2.gameObject.SetActive(true);

                m_ExpandImage.gameObject.SetActive(false);
            }

            m_Title.text = m_PDFBookmark.GetTitle();

            name = m_Title.text.Substring(0, Mathf.Min(24, m_Title.text.Length));

            for (int i = 0; i < m_PDFBookmark.ChildCount; ++i)
            {
                PDFBookmark child = m_PDFBookmark.GetChild(i);

                PDFBookmarkListItem item = Instantiate(gameObject).GetComponent<PDFBookmarkListItem>();
                m_ChildrenItems.Add(item);

                RectTransform itemTransform = (RectTransform)item.transform;
                itemTransform.SetParent(transform.parent, false);
                itemTransform.localScale = Vector3.one;
                itemTransform.anchorMin = new Vector2(0.0f, 1.0f);
                itemTransform.anchorMax = new Vector2(0.0f, 1.0f);
                itemTransform.offsetMin = Vector2.zero;
                itemTransform.offsetMax = Vector2.zero;

                item.Initialize(child, indent + 1, i == m_PDFBookmark.ChildCount - 1);

                if (indent == 0)
                {
                    StartCoroutine(SetVisible());

                    item.m_CanvasGroup.alpha = 0.0f;
                }
            }

            for (int i = 0; i < m_ChildrenItems.Count - 1; ++i)
            {
                m_ChildrenItems[i].m_NextSibling = (RectTransform)m_ChildrenItems[i + 1].gameObject.transform;
            }

            m_Internal.offsetMin = new Vector2(20.0f * indent, m_Internal.offsetMin.y);
            m_Initialized = true;
            m_SizeAdjusted = 1;
        }

        public void OnExpandButton()
        {
            if (m_PDFBookmark.ChildCount > 0)
            {
                m_Expanded = !m_Expanded;

                m_VerticalLine.gameObject.SetActive(m_Expanded);

                m_ExpandImage.sprite = m_Expanded ? m_CollapseSprite : m_ExpandSprite;

                foreach (PDFBookmarkListItem child in m_ChildrenItems)
	                child.SetState(m_Expanded);
            }
        }

        public void OnItemClicked()
        {
            if (Time.fixedTime - m_LastClickTimestamp < 0.5f)
	            OnExpandButton();

            m_LastClickTimestamp = Time.fixedTime;

            if (m_BookmarksViewer == null)
                m_BookmarksViewer = GetComponentInParent<PDFBookmarksViewer>();

            if (m_BookmarksViewer.LastHighlightedImage != null)
                m_BookmarksViewer.LastHighlightedImage.gameObject.SetActive(false);

            m_Highlighted.gameObject.SetActive(true);
            m_BookmarksViewer.LastHighlightedImage = m_Highlighted;

            m_PDFBookmark.ExecuteBookmarkAction(m_BookmarksViewer.Viewer);
        }

        public void SetState(bool active)
        {
            gameObject.SetActive(active);

            if (active)
	            StartCoroutine(SetVisible());

            if (m_ChildrenItems.Count > 0)
            {
                m_VerticalLine.gameObject.SetActive(!active);
                m_ExpandImage.sprite = !active ? m_CollapseSprite : m_ExpandSprite;
            }

            if (!active)
            {
                foreach (PDFBookmarkListItem child in m_ChildrenItems)
	                child.SetState(false);
            }
        }

        protected override void OnEnable()
        {
            if (m_LayoutElement == null)
				m_LayoutElement = GetComponent<LayoutElement>();
            if (m_CanvasGroup == null)
				m_CanvasGroup = GetComponent<CanvasGroup>();

            if (m_CanvasGroup == null)
            {
                m_CanvasGroup = gameObject.AddComponent<CanvasGroup>();
                m_CanvasGroup.interactable = true;
                m_CanvasGroup.blocksRaycasts = true;
            }

            m_CanvasGroup.alpha = 1.0f;
        }

        protected override void OnDestroy()
        {
            base.OnDestroy();

            m_PDFBookmark = null;
            m_ChildrenItems = null;

            if (m_BookmarksViewer != null)
            {
                if (m_BookmarksViewer.LastHighlightedImage == this)
                    m_BookmarksViewer.LastHighlightedImage = null;

                m_BookmarksViewer = null;
            }
        }

        private IEnumerator SetVisible()
        {
            m_CanvasGroup.alpha = 0.0f;

            yield return new WaitForSeconds(1.0f / 60.0f);

            m_CanvasGroup.alpha = 1.0f;
        }

        private void Update()
        {
            if (m_Initialized && m_SizeAdjusted == 0)
            {
                m_LayoutElement.preferredHeight = m_Title.preferredHeight + 20.0f;

                if (m_VerticalLine.gameObject.activeInHierarchy)
	                m_VerticalLine.sizeDelta = new Vector2(1.0f, m_LayoutElement.preferredHeight);

                if (!m_PDFBookmark.IsTopLevelBookmark)
	                gameObject.SetActive(false);

                m_SizeAdjusted = -1;
            }
            else if (m_SizeAdjusted > 0)
            {
                m_SizeAdjusted--;
            }

            if (m_Initialized && m_SizeAdjusted <= -1)
            {
                m_LayoutElement.preferredHeight = m_Title.preferredHeight + 20.0f;

                if (m_VerticalLine.gameObject.activeInHierarchy)
	                m_VerticalLine.sizeDelta = new Vector2(1.0f, m_LayoutElement.preferredHeight);

                if (m_NextSibling != null)
                {
                    if (m_VerticalLine2.gameObject.activeInHierarchy)
                    {
                        float newHeight = Mathf.Abs((m_NextSibling.anchoredPosition - RectTransform.anchoredPosition).y);

                        m_VerticalLine2.sizeDelta = new Vector2(1.0f, newHeight);
                    }
                }
                else if (m_VerticalLine2.gameObject.activeInHierarchy)
                {
                    m_VerticalLine2.gameObject.SetActive(false);
                }

                if (m_SizeAdjusted == -1)
	                m_SizeAdjusted = -2;
            }
        }
#endif
    }
}