using UnityEngine;
|
using UnityEditor;
|
using System;
|
using System.Collections.Generic;
|
using qtools.qhierarchy.pcomponent;
|
using qtools.qhierarchy.pcomponent.pbase;
|
using qtools.qhierarchy.pdata;
|
using qtools.qhierarchy.phelper;
|
using System.Reflection;
|
|
namespace qtools.qhierarchy.phierarchy
|
{
|
public class QHierarchy
|
{
|
// PRIVATE
|
private HashSet<int> errorHandled = new HashSet<int>();
|
private Dictionary<int, QBaseComponent> componentDictionary;
|
private List<QBaseComponent> preComponents;
|
private List<QBaseComponent> orderedComponents;
|
private List<QBaseComponent> postComponents;
|
private bool hideIconsIfThereIsNoFreeSpace;
|
private int indentation;
|
private Texture2D trimIcon;
|
|
// CONSTRUCTOR
|
public QHierarchy ()
|
{
|
componentDictionary = new Dictionary<int, QBaseComponent>();
|
componentDictionary.Add((int)QHierarchyComponentEnum.LockComponent , new QLockComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.VisibilityComponent , new QVisibilityComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.StaticComponent , new QStaticComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.RendererComponent , new QRendererComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.TagLayerComponent , new QTagLayerComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.GameObjectIconComponent , new QGameObjectIconComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.ErrorComponent , new QErrorComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.TagIconComponent , new QTagIconComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.ColorComponent , new QColorComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.ComponentsComponent , new QComponentsComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.ChildrenCountComponent , new QChildrenCountComponent());
|
componentDictionary.Add((int)QHierarchyComponentEnum.PrefabComponent , new QPrefabComponent());
|
|
preComponents = new List<QBaseComponent>();
|
preComponents.Add(new QTreeMapComponent());
|
preComponents.Add(new QMonoBehaviorIconComponent());
|
|
orderedComponents = new List<QBaseComponent>();
|
|
postComponents = new List<QBaseComponent>();
|
postComponents.Add(new QSeparatorComponent());
|
|
trimIcon = QResources.getInstance().getTexture(QTexture.QTrimIcon);
|
|
QSettings.getInstance().addEventListener(QSetting.Identation , settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.ComponentOrder , settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.HideIconsIfNotFit , settingsChanged);
|
settingsChanged();
|
}
|
|
// PRIVATE
|
private void settingsChanged()
|
{
|
orderedComponents.Clear();
|
string componentOrder = QSettings.getInstance().get<string>(QSetting.ComponentOrder);
|
string[] componentIds = componentOrder.Split(';');
|
if (componentIds.Length != QSettings.DEFAULT_ORDER_COUNT)
|
{
|
QSettings.getInstance().set(QSetting.ComponentOrder, QSettings.DEFAULT_ORDER);
|
componentIds = QSettings.DEFAULT_ORDER.Split(';');
|
}
|
|
for (int i = 0; i < componentIds.Length; i++)
|
orderedComponents.Add(componentDictionary[int.Parse(componentIds[i])]);
|
orderedComponents.Add(componentDictionary[(int)QHierarchyComponentEnum.ComponentsComponent]);
|
|
indentation = QSettings.getInstance().get<int>(QSetting.Identation);
|
hideIconsIfThereIsNoFreeSpace = QSettings.getInstance().get<bool>(QSetting.HideIconsIfNotFit);
|
}
|
|
public void hierarchyWindowItemOnGUIHandler(int instanceId, Rect selectionRect)
|
{
|
try
|
{
|
GameObject gameObject = (GameObject)EditorUtility.InstanceIDToObject(instanceId);
|
if (gameObject == null) return;
|
|
Rect curRect = new Rect(selectionRect);
|
curRect.width = 16;
|
curRect.x += selectionRect.width - indentation;
|
|
float gameObjectNameWidth = hideIconsIfThereIsNoFreeSpace ? GUI.skin.label.CalcSize(new GUIContent(gameObject.name)).x : 0;
|
|
QObjectList objectList = QObjectListManager.getInstance().getObjectList(gameObject, false);
|
|
drawComponents(preComponents , selectionRect, ref curRect, gameObject, objectList);
|
drawComponents(orderedComponents, selectionRect, ref curRect, gameObject, objectList, hideIconsIfThereIsNoFreeSpace, selectionRect.x + gameObjectNameWidth + 7);
|
drawComponents(postComponents , selectionRect, ref curRect, gameObject, objectList);
|
|
errorHandled.Remove(instanceId);
|
}
|
catch (Exception exception)
|
{
|
if (errorHandled.Add(instanceId))
|
{
|
Debug.LogError(exception.ToString());
|
}
|
}
|
}
|
|
private void drawComponents(List<QBaseComponent> components, Rect selectionRect, ref Rect curRect, GameObject gameObject, QObjectList objectList, bool trim = false, float minX = 50)
|
{
|
Rect rect = new Rect(curRect);
|
if (Event.current.type == EventType.Repaint)
|
{
|
for (int i = 0, n = components.Count; i < n; i++)
|
{
|
QBaseComponent component = components[i];
|
if (component.isEnabled())
|
{
|
component.layout(gameObject, objectList, ref rect);
|
if (trim && minX > rect.x)
|
{
|
rect.Set(curRect.x - 7, curRect.y, 7, 16);
|
GUI.DrawTexture(rect, trimIcon);
|
return;
|
}
|
component.draw(gameObject, objectList, selectionRect, rect);
|
curRect.Set(rect.x, rect.y, rect.width, rect.height);
|
}
|
else
|
{
|
component.disabledHandler(gameObject, objectList);
|
}
|
}
|
}
|
else if (Event.current.isMouse)
|
{
|
for (int i = 0, n = components.Count; i < n; i++)
|
{
|
QBaseComponent component = components[i];
|
if (component.isEnabled())
|
{
|
component.layout(gameObject, objectList, ref rect);
|
if (trim && minX > rect.x)
|
{
|
rect.Set(curRect.x - 7, curRect.y, 7, 16);
|
GUI.DrawTexture(rect, trimIcon);
|
return;
|
}
|
component.eventHandler(gameObject, objectList, Event.current, rect);
|
curRect.Set(rect.x, rect.y, rect.width, rect.height);
|
}
|
}
|
}
|
}
|
}
|
}
|
