355 lines
11 KiB
C#
355 lines
11 KiB
C#
|
using System.Collections;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using UnityEngine;
|
|||
|
|
|
|||
|
|
|
|||
|
|
namespace SuperTreeView
|
|||
|
|
{
|
|||
|
|
public class TreeList : MonoBehaviour
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
float mTreeListMaxWidth = 9999f;
|
|||
|
|
protected List<TreeViewItem> mTreeItemList = new List<TreeViewItem>();
|
|||
|
|
protected TreeView mRootTreeView;
|
|||
|
|
protected TreeViewItem mParentTreeItem;
|
|||
|
|
|
|||
|
|
protected RectTransform mCachedRectTransform;
|
|||
|
|
protected bool mNeedReposition = true;
|
|||
|
|
|
|||
|
|
protected float mContentTotalWidth;
|
|||
|
|
protected float mContentTotalHeight;
|
|||
|
|
|
|||
|
|
string mDefaultItemPrefabName;
|
|||
|
|
|
|||
|
|
public string DefaultItemPrefabName
|
|||
|
|
{
|
|||
|
|
get { return mDefaultItemPrefabName; }
|
|||
|
|
set { mDefaultItemPrefabName = value; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public float TreeListMaxWidth
|
|||
|
|
{
|
|||
|
|
get { return mTreeListMaxWidth; }
|
|||
|
|
set { mTreeListMaxWidth = value; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public float ContentTotalHeight
|
|||
|
|
{
|
|||
|
|
get { return mContentTotalHeight; }
|
|||
|
|
}
|
|||
|
|
public float ContentTotalWidth
|
|||
|
|
{
|
|||
|
|
get { return mContentTotalWidth; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public bool NeedReposition
|
|||
|
|
{
|
|||
|
|
get { return mNeedReposition; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public RectTransform CachedRectTransform
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
if (mCachedRectTransform == null)
|
|||
|
|
{
|
|||
|
|
mCachedRectTransform = gameObject.GetComponent<RectTransform>();
|
|||
|
|
}
|
|||
|
|
return mCachedRectTransform;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public TreeView RootTreeView
|
|||
|
|
{
|
|||
|
|
get { return mRootTreeView; }
|
|||
|
|
set { mRootTreeView = value; }
|
|||
|
|
}
|
|||
|
|
public TreeViewItem ParentTreeItem
|
|||
|
|
{
|
|||
|
|
get { return mParentTreeItem; }
|
|||
|
|
set { mParentTreeItem = value; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public int ItemCount
|
|||
|
|
{
|
|||
|
|
get { return mTreeItemList.Count; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public bool IsEmpty
|
|||
|
|
{
|
|||
|
|
get { return (ItemCount == 0); }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public bool IsRootTree
|
|||
|
|
{
|
|||
|
|
get { return System.Object.ReferenceEquals(RootTreeView, this); }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public void Init()
|
|||
|
|
{
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
mContentTotalHeight = 0;
|
|||
|
|
mContentTotalWidth = 0;
|
|||
|
|
mTreeItemList.Clear();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
//This method is used to expand all the child TreeViewIItem’s childtree.
|
|||
|
|
//If immediate is set to true, then all the child TreeViewIItem’s childtree will expand immediately.
|
|||
|
|
public void ExpandAllItem(bool immediate = false)
|
|||
|
|
{
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
int count = ItemCount;
|
|||
|
|
for(int i =0;i< count;++i)
|
|||
|
|
{
|
|||
|
|
mTreeItemList[i].ExpandAll(immediate);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//This method is used to collapse all the child TreeViewIItem’s childtree.
|
|||
|
|
//If immediate is set to true, then all the child TreeViewIItem’s childtree will collapse immediately.
|
|||
|
|
public void CollapseAllItem(bool immediate = false)
|
|||
|
|
{
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
int count = ItemCount;
|
|||
|
|
for (int i = 0; i < count; ++i)
|
|||
|
|
{
|
|||
|
|
mTreeItemList[i].CollapseAll(immediate);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
void InitTreeViewItem(TreeViewItem tViewItem)
|
|||
|
|
{
|
|||
|
|
tViewItem.CachedRectTransform.SetParent(CachedRectTransform);
|
|||
|
|
tViewItem.CachedRectTransform.localEulerAngles = Vector3.zero;
|
|||
|
|
tViewItem.CachedRectTransform.localScale = Vector3.one;
|
|||
|
|
tViewItem.RootTreeView = RootTreeView;
|
|||
|
|
tViewItem.ParentTreeList = this;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
//This method is used to append a new item to a TreeList (TreeView is also a TreeList).
|
|||
|
|
//prefabName is the name of the item prefab gameobject, and should be one element of the ItemPrefabList in the TreeView.
|
|||
|
|
//userData is the custom data of the new created TreeViewItem, and you can get the useData by call TreeViewItem.UserData.
|
|||
|
|
public TreeViewItem AppendItem(string prefabName = "", System.Object userData = null)
|
|||
|
|
{
|
|||
|
|
if (string.IsNullOrEmpty(prefabName))
|
|||
|
|
{
|
|||
|
|
prefabName = DefaultItemPrefabName;
|
|||
|
|
}
|
|||
|
|
TreeViewItem tViewItem = mRootTreeView.NewTreeItem(prefabName);
|
|||
|
|
if(tViewItem == null)
|
|||
|
|
{
|
|||
|
|
Debug.LogError("AppendItem return null ");
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
InitTreeViewItem(tViewItem);
|
|||
|
|
tViewItem.Init();
|
|||
|
|
tViewItem.UserData = userData;
|
|||
|
|
tViewItem.ItemIndex = mTreeItemList.Count;
|
|||
|
|
mTreeItemList.Add(tViewItem);
|
|||
|
|
tViewItem.OnActived();
|
|||
|
|
UpdateAllItemSiblingIndex();
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
if (IsRootTree)
|
|||
|
|
{
|
|||
|
|
RootTreeView.NeedRepositionView = true;
|
|||
|
|
}
|
|||
|
|
if (RootTreeView.OnTreeListAddOneItem != null)
|
|||
|
|
{
|
|||
|
|
RootTreeView.OnTreeListAddOneItem(this);
|
|||
|
|
}
|
|||
|
|
return tViewItem;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
// This method is used to insert a new item to a TreeList at itemIndex Pos.
|
|||
|
|
//All the child items of a TreeList are indexed from top to bottom starting at 0.
|
|||
|
|
//That is to say the top most item of a TreeList is indexed 0 and the right below item is indexed 1.
|
|||
|
|
public TreeViewItem InsertItem(int itemIndex, string prefabName = "", System.Object userData = null)
|
|||
|
|
{
|
|||
|
|
if (itemIndex < 0 || itemIndex > mTreeItemList.Count)
|
|||
|
|
{
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
if (string.IsNullOrEmpty(prefabName))
|
|||
|
|
{
|
|||
|
|
prefabName = DefaultItemPrefabName;
|
|||
|
|
}
|
|||
|
|
TreeViewItem tViewItem = mRootTreeView.NewTreeItem(prefabName);
|
|||
|
|
InitTreeViewItem(tViewItem);
|
|||
|
|
tViewItem.Init();
|
|||
|
|
tViewItem.UserData = userData;
|
|||
|
|
mTreeItemList.Insert(itemIndex, tViewItem);
|
|||
|
|
ResetAllItemIndex();
|
|||
|
|
tViewItem.OnActived();
|
|||
|
|
UpdateAllItemSiblingIndex();
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
if (IsRootTree)
|
|||
|
|
{
|
|||
|
|
RootTreeView.NeedRepositionView = true;
|
|||
|
|
}
|
|||
|
|
if (RootTreeView.OnTreeListAddOneItem != null)
|
|||
|
|
{
|
|||
|
|
RootTreeView.OnTreeListAddOneItem(this);
|
|||
|
|
}
|
|||
|
|
return tViewItem;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void UpdateAllItemSiblingIndex()
|
|||
|
|
{
|
|||
|
|
int count = mTreeItemList.Count;
|
|||
|
|
for (int i = 0; i < count; ++i)
|
|||
|
|
{
|
|||
|
|
mTreeItemList[i].CachedRectTransform.SetSiblingIndex(i);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Clear()
|
|||
|
|
{
|
|||
|
|
for (int i = 0; i < mTreeItemList.Count; ++i)
|
|||
|
|
{
|
|||
|
|
mTreeItemList[i].Clear();
|
|||
|
|
mRootTreeView.RecycleTreeItem(mTreeItemList[i]);
|
|||
|
|
}
|
|||
|
|
mTreeItemList.Clear();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void ResetAllItemIndex()
|
|||
|
|
{
|
|||
|
|
for (int i = 0; i < mTreeItemList.Count; ++i)
|
|||
|
|
{
|
|||
|
|
mTreeItemList[i].ItemIndex = i;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void DeleteItem(TreeViewItem item)
|
|||
|
|
{
|
|||
|
|
DeleteItemByIndex(item.ItemIndex);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void DeleteItemByIndex(int itemIndex)
|
|||
|
|
{
|
|||
|
|
TreeViewItem tItem = GetItemByIndex(itemIndex);
|
|||
|
|
if (tItem == null)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
mTreeItemList.RemoveAt(itemIndex);
|
|||
|
|
ResetAllItemIndex();
|
|||
|
|
tItem.Clear();
|
|||
|
|
mRootTreeView.RecycleTreeItem(tItem);
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
UpdateAllItemSiblingIndex();
|
|||
|
|
if (IsRootTree)
|
|||
|
|
{
|
|||
|
|
RootTreeView.NeedRepositionView = true;
|
|||
|
|
}
|
|||
|
|
if (RootTreeView.OnTreeListDeleteOneItem != null)
|
|||
|
|
{
|
|||
|
|
RootTreeView.OnTreeListDeleteOneItem(this);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void OnUpdate()
|
|||
|
|
{
|
|||
|
|
int count = mTreeItemList.Count;
|
|||
|
|
for (int i = 0; i < count; ++i)
|
|||
|
|
{
|
|||
|
|
TreeViewItem tItem = mTreeItemList[i];
|
|||
|
|
tItem.OnUpdate();
|
|||
|
|
if (tItem.NeedReposition)
|
|||
|
|
{
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public TreeViewItem GetItemByIndex(int itemIndex)
|
|||
|
|
{
|
|||
|
|
if (itemIndex < 0 || itemIndex >= mTreeItemList.Count)
|
|||
|
|
{
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
return mTreeItemList[itemIndex];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Sort(System.Comparison<TreeViewItem> sortComp)
|
|||
|
|
{
|
|||
|
|
mTreeItemList.Sort(sortComp);
|
|||
|
|
ResetAllItemIndex();
|
|||
|
|
mNeedReposition = true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Reposition()
|
|||
|
|
{
|
|||
|
|
if (NeedReposition == false)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
DoReposition();
|
|||
|
|
if(mRootTreeView.OnTreeListRepositionFinish != null)
|
|||
|
|
{
|
|||
|
|
mRootTreeView.OnTreeListRepositionFinish(this);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void DoReposition()
|
|||
|
|
{
|
|||
|
|
mNeedReposition = false;
|
|||
|
|
int count = mTreeItemList.Count;
|
|||
|
|
mContentTotalHeight = 0;
|
|||
|
|
mContentTotalWidth = 0;
|
|||
|
|
CachedRectTransform.SetSizeWithCurrentAnchors(RectTransform.Axis.Vertical, 0);
|
|||
|
|
if(IsRootTree == false)
|
|||
|
|
{
|
|||
|
|
CachedRectTransform.SetSizeWithCurrentAnchors(RectTransform.Axis.Horizontal, TreeListMaxWidth);
|
|||
|
|
}
|
|||
|
|
if (count == 0)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
float xOffset = CachedRectTransform.pivot.x* CachedRectTransform.sizeDelta.x;
|
|||
|
|
float itemPadding = 0;
|
|||
|
|
if (ParentTreeItem != null)
|
|||
|
|
{
|
|||
|
|
itemPadding = ParentTreeItem.ChildTreeItemPadding;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
itemPadding = RootTreeView.ItemPadding;
|
|||
|
|
}
|
|||
|
|
float curY = 0;
|
|||
|
|
if (ParentTreeItem != null)
|
|||
|
|
{
|
|||
|
|
curY = -ParentTreeItem.ChildTreeListPadding;
|
|||
|
|
}
|
|||
|
|
for (int i = 0; i < count; ++i)
|
|||
|
|
{
|
|||
|
|
TreeViewItem tItem = mTreeItemList[i];
|
|||
|
|
tItem.CachedRectTransform.anchoredPosition3D = new Vector3(xOffset, curY, 0);
|
|||
|
|
tItem.Reposition();
|
|||
|
|
|
|||
|
|
curY = curY - tItem.TotalHeight - itemPadding;
|
|||
|
|
if (tItem.MaxWidth > mContentTotalWidth)
|
|||
|
|
{
|
|||
|
|
mContentTotalWidth = tItem.MaxWidth;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
mContentTotalHeight = -curY - itemPadding;
|
|||
|
|
CachedRectTransform.SetSizeWithCurrentAnchors(RectTransform.Axis.Vertical, mContentTotalHeight);
|
|||
|
|
if (IsRootTree)
|
|||
|
|
{
|
|||
|
|
CachedRectTransform.SetSizeWithCurrentAnchors(RectTransform.Axis.Horizontal, mContentTotalWidth);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|