using System;
|
using System.Collections.Generic;
|
using UnityEngine;
|
using UnityEditor;
|
using qtools.qhierarchy.pcomponent.pbase;
|
using qtools.qhierarchy.pdata;
|
using System.Reflection;
|
|
namespace qtools.qhierarchy.pcomponent
|
{
|
public class QTagLayerComponent: QBaseComponent
|
{
|
// PRIVATE
|
private GUIStyle labelStyle;
|
private Color greyLightColor;
|
private Color greyColor;
|
private Color backgroundColor;
|
private bool showAlways;
|
private bool sizeIsPixel;
|
private int pixelSize;
|
private float percentSize;
|
private GameObject[] gameObjects;
|
private QHierarchyTagAndLayerLabelSize labelSize;
|
|
// CONSTRUCTOR
|
public QTagLayerComponent()
|
{
|
backgroundColor = QResources.getInstance().getColor(QColor.Background);
|
greyColor = QResources.getInstance().getColor(QColor.GrayDark);
|
greyLightColor = QResources.getInstance().getColor(QColor.GrayLight);
|
|
labelStyle = new GUIStyle();
|
labelStyle.normal.textColor = greyColor;
|
labelStyle.fontSize = 8;
|
labelStyle.clipping = TextClipping.Clip;
|
|
QSettings.getInstance().addEventListener(QSetting.TagAndLayerType , settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.TagAndLayerSizeValueType , settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.TagAndLayerSizeValuePixel , settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.TagAndLayerSizeValuePercent, settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.TagAndLayerLabelSize , settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.ShowTagLayerComponent , settingsChanged);
|
QSettings.getInstance().addEventListener(QSetting.ShowTagLayerComponentDuringPlayMode, settingsChanged);
|
settingsChanged();
|
}
|
|
// PRIVATE
|
private void settingsChanged()
|
{
|
showAlways = QSettings.getInstance().get<int>(QSetting.TagAndLayerType) == (int)QHierarchyTagAndLayerType.Always;
|
sizeIsPixel = QSettings.getInstance().get<int>(QSetting.TagAndLayerSizeValueType) == (int)QHierarchyTagAndLayerSizeType.Pixel;
|
pixelSize = QSettings.getInstance().get<int>(QSetting.TagAndLayerSizeValuePixel);
|
percentSize = QSettings.getInstance().get<float>(QSetting.TagAndLayerSizeValuePercent);
|
labelSize = (QHierarchyTagAndLayerLabelSize)QSettings.getInstance().get<int>(QSetting.TagAndLayerLabelSize);
|
enabled = QSettings.getInstance().get<bool>(QSetting.ShowTagLayerComponent);
|
showComponentDuringPlayMode = QSettings.getInstance().get<bool>(QSetting.ShowTagLayerComponentDuringPlayMode);
|
}
|
|
// DRAW
|
public override void layout(GameObject gameObject, QObjectList objectList, ref Rect rect)
|
{
|
float textWidth = sizeIsPixel ? pixelSize : percentSize * rect.x;
|
rect.width = textWidth + 4;
|
rect.x -= rect.width;
|
}
|
|
public override void draw(GameObject gameObject, QObjectList objectList, Rect selectionRect, Rect curRect)
|
{
|
int layer = gameObject.layer;
|
string tag = getTagName(gameObject);
|
|
EditorGUI.DrawRect(curRect, backgroundColor);
|
|
if (showAlways || (tag != "Untagged" || layer != 0))
|
{
|
curRect.height = 17;
|
curRect.y -= 2;
|
curRect.width -= 4;
|
curRect.x += 2;
|
|
int aligment = QSettings.getInstance().get<int>(QSetting.TagAndLayerAligment);
|
if (aligment == (int)QHierarchyTagAndLayerAligment.Left)
|
labelStyle.alignment = TextAnchor.UpperLeft;
|
else if (aligment == (int)QHierarchyTagAndLayerAligment.Center)
|
labelStyle.alignment = TextAnchor.UpperCenter;
|
else if (aligment == (int)QHierarchyTagAndLayerAligment.Right)
|
labelStyle.alignment = TextAnchor.UpperRight;
|
|
if (layer == 0 && tag != "Untagged" && !showAlways)
|
{
|
if (labelSize == QHierarchyTagAndLayerLabelSize.Small) labelStyle.fontSize = 8;
|
else labelStyle.fontSize = 9;
|
|
curRect.y += 5;
|
labelStyle.normal.textColor = greyLightColor;
|
EditorGUI.LabelField(curRect, tag, labelStyle);
|
}
|
else if (layer != 0 && tag == "Untagged" && !showAlways)
|
{
|
if (labelSize == QHierarchyTagAndLayerLabelSize.Small) labelStyle.fontSize = 8;
|
else labelStyle.fontSize = 9;
|
|
curRect.y += 5;
|
labelStyle.normal.textColor = greyLightColor;
|
EditorGUI.LabelField(curRect, getLayerName(layer), labelStyle);
|
}
|
else
|
{
|
if (labelSize == QHierarchyTagAndLayerLabelSize.Normal) labelStyle.fontSize = 9;
|
else labelStyle.fontSize = 8;
|
|
labelStyle.normal.textColor = tag == "Untagged" ? greyColor : greyLightColor;
|
EditorGUI.LabelField(curRect, tag, labelStyle);
|
curRect.y += 8;
|
labelStyle.normal.textColor = layer == 0 ? greyColor : greyLightColor;
|
EditorGUI.LabelField(curRect, getLayerName(layer), labelStyle);
|
}
|
}
|
}
|
|
public override void eventHandler(GameObject gameObject, QObjectList objectList, Event currentEvent, Rect rect)
|
{
|
if (Event.current.isMouse && currentEvent.type == EventType.MouseDown && Event.current.button == 0 && rect.Contains(Event.current.mousePosition))
|
{
|
int layer = gameObject.layer;
|
string tag = getTagName(gameObject);
|
|
if (showAlways || (tag != "Untagged" || layer != 0))
|
{
|
Event.current.Use();
|
gameObjects = Selection.Contains(gameObject) ? Selection.gameObjects : new GameObject[] { gameObject };
|
|
if (layer == 0 && tag != "Untagged" && !showAlways)
|
{
|
showTagsContextMenu(tag);
|
}
|
else if (layer != 0 && tag == "Untagged" && !showAlways)
|
{
|
showLayersContextMenu(LayerMask.LayerToName(layer));
|
}
|
else
|
{
|
if (Event.current.mousePosition.y < rect.y + rect.height / 2)
|
{
|
showTagsContextMenu(tag);
|
}
|
else
|
{
|
showLayersContextMenu(LayerMask.LayerToName(layer));
|
}
|
}
|
}
|
}
|
}
|
|
private string getTagName(GameObject gameObject)
|
{
|
string tag = "Undefined";
|
try { tag = gameObject.tag; }
|
catch {}
|
return tag;
|
}
|
|
public string getLayerName(int layer)
|
{
|
string layerName = LayerMask.LayerToName(layer);
|
if (layerName.Equals("")) layerName = "Undefined";
|
return layerName;
|
}
|
|
// PRIVATE
|
private void showTagsContextMenu(string tag)
|
{
|
List<string> tags = new List<string>(UnityEditorInternal.InternalEditorUtility.tags);
|
|
GenericMenu menu = new GenericMenu();
|
menu.AddItem(new GUIContent("Untagged" ), false, tagChangedHandler, "Untagged");
|
|
for (int i = 0, n = tags.Count; i < n; i++)
|
{
|
string curTag = tags[i];
|
menu.AddItem(new GUIContent(curTag), tag == curTag, tagChangedHandler, curTag);
|
}
|
|
menu.AddSeparator("");
|
menu.AddItem(new GUIContent("Add Tag..." ), false, addTagOrLayerHandler, "Tags");
|
menu.ShowAsContext();
|
}
|
|
private void showLayersContextMenu(string layer)
|
{
|
List<string> layers = new List<string>(UnityEditorInternal.InternalEditorUtility.layers);
|
|
GenericMenu menu = new GenericMenu();
|
menu.AddItem(new GUIContent("Default" ), false, layerChangedHandler, "Default");
|
|
for (int i = 0, n = layers.Count; i < n; i++)
|
{
|
string curLayer = layers[i];
|
menu.AddItem(new GUIContent(curLayer), layer == curLayer, layerChangedHandler, curLayer);
|
}
|
|
menu.AddSeparator("");
|
menu.AddItem(new GUIContent("Add Layer..." ), false, addTagOrLayerHandler, "Layers");
|
menu.ShowAsContext();
|
}
|
|
private void tagChangedHandler(object newTag)
|
{
|
for (int i = gameObjects.Length - 1; i >= 0; i--)
|
{
|
GameObject gameObject = gameObjects[i];
|
Undo.RecordObject(gameObject, "Change Tag");
|
gameObject.tag = (string)newTag;
|
EditorUtility.SetDirty(gameObject);
|
}
|
}
|
|
private void layerChangedHandler(object newLayer)
|
{
|
int newLayerId = LayerMask.NameToLayer((string)newLayer);
|
for (int i = gameObjects.Length - 1; i >= 0; i--)
|
{
|
GameObject gameObject = gameObjects[i];
|
Undo.RecordObject(gameObject, "Change Layer");
|
gameObject.layer = newLayerId;
|
EditorUtility.SetDirty(gameObject);
|
}
|
}
|
|
private void addTagOrLayerHandler(object value)
|
{
|
PropertyInfo propertyInfo = typeof(EditorApplication).GetProperty("tagManager", BindingFlags.Static | BindingFlags.NonPublic | BindingFlags.GetProperty);
|
UnityEngine.Object obj = (UnityEngine.Object)(propertyInfo.GetValue(null, null));
|
obj.GetType().GetField("m_DefaultExpandedFoldout").SetValue(obj, value);
|
Selection.activeObject = obj;
|
}
|
}
|
}
|
