mirror of
https://github.com/EsotericSoftware/spine-runtimes.git
synced 2026-02-04 22:34:53 +08:00
348 lines
12 KiB
C#
348 lines
12 KiB
C#
|
|
/*****************************************************************************
|
|
* Spine Attribute Drawers created by Mitch Thompson
|
|
* Full irrevocable rights and permissions granted to Esoteric Software
|
|
*****************************************************************************/
|
|
using UnityEngine;
|
|
using UnityEditor;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Reflection;
|
|
using Spine;
|
|
|
|
|
|
public struct SpineDrawerValuePair {
|
|
public string str;
|
|
public SerializedProperty property;
|
|
|
|
public SpineDrawerValuePair(string val, SerializedProperty property) {
|
|
this.str = val;
|
|
this.property = property;
|
|
}
|
|
}
|
|
|
|
public abstract class SpineTreeItemDrawerBase<T> : PropertyDrawer where T:SpineAttributeBase {
|
|
protected SkeletonDataAsset skeletonDataAsset;
|
|
protected T TargetAttribute { get { return (T)attribute; } }
|
|
|
|
public override void OnGUI (Rect position, SerializedProperty property, GUIContent label) {
|
|
if (property.propertyType != SerializedPropertyType.String) {
|
|
EditorGUI.LabelField(position, "ERROR:", "May only apply to type string");
|
|
return;
|
|
}
|
|
|
|
var dataProperty = property.serializedObject.FindProperty(TargetAttribute.dataField);
|
|
|
|
if (dataProperty != null) {
|
|
if (dataProperty.objectReferenceValue is SkeletonDataAsset) {
|
|
skeletonDataAsset = (SkeletonDataAsset)dataProperty.objectReferenceValue;
|
|
} else if (dataProperty.objectReferenceValue is SkeletonRenderer) {
|
|
var renderer = (SkeletonRenderer)dataProperty.objectReferenceValue;
|
|
if (renderer != null)
|
|
skeletonDataAsset = renderer.skeletonDataAsset;
|
|
} else {
|
|
EditorGUI.LabelField(position, "ERROR:", "Invalid reference type");
|
|
return;
|
|
}
|
|
|
|
} else if (property.serializedObject.targetObject is Component) {
|
|
var component = (Component)property.serializedObject.targetObject;
|
|
if (component.GetComponentInChildren<SkeletonRenderer>() != null) {
|
|
var skeletonRenderer = component.GetComponentInChildren<SkeletonRenderer>();
|
|
skeletonDataAsset = skeletonRenderer.skeletonDataAsset;
|
|
}
|
|
}
|
|
|
|
if (skeletonDataAsset == null) {
|
|
EditorGUI.LabelField(position, "ERROR:", "Must have reference to a SkeletonDataAsset");
|
|
return;
|
|
}
|
|
|
|
position = EditorGUI.PrefixLabel(position, label);
|
|
|
|
if (GUI.Button(position, property.stringValue, EditorStyles.popup)) {
|
|
Selector(property);
|
|
}
|
|
|
|
}
|
|
|
|
protected virtual void Selector (SerializedProperty property) {
|
|
SkeletonData data = skeletonDataAsset.GetSkeletonData(true);
|
|
if (data == null)
|
|
return;
|
|
|
|
GenericMenu menu = new GenericMenu();
|
|
PopulateMenu (menu, property, this.TargetAttribute, data);
|
|
menu.ShowAsContext();
|
|
}
|
|
|
|
protected abstract void PopulateMenu (GenericMenu menu, SerializedProperty property, T targetAttribute, SkeletonData data);
|
|
|
|
protected virtual void HandleSelect (object val) {
|
|
var pair = (SpineDrawerValuePair)val;
|
|
pair.property.stringValue = pair.str;
|
|
pair.property.serializedObject.ApplyModifiedProperties();
|
|
}
|
|
|
|
public override float GetPropertyHeight(SerializedProperty property, GUIContent label) {
|
|
return 18;
|
|
}
|
|
|
|
}
|
|
|
|
[CustomPropertyDrawer(typeof(SpineSlot))]
|
|
public class SpineSlotDrawer : SpineTreeItemDrawerBase<SpineSlot> {
|
|
|
|
protected override void PopulateMenu (GenericMenu menu, SerializedProperty property, SpineSlot targetAttribute, SkeletonData data) {
|
|
for (int i = 0; i < data.Slots.Count; i++) {
|
|
string name = data.Slots.Items[i].Name;
|
|
if (name.StartsWith(targetAttribute.startsWith)) {
|
|
if (targetAttribute.containsBoundingBoxes) {
|
|
|
|
int slotIndex = i;
|
|
|
|
List<Attachment> attachments = new List<Attachment>();
|
|
foreach (var skin in data.Skins) {
|
|
skin.FindAttachmentsForSlot(slotIndex, attachments);
|
|
}
|
|
|
|
bool hasBoundingBox = false;
|
|
foreach (var attachment in attachments) {
|
|
if (attachment is BoundingBoxAttachment) {
|
|
menu.AddItem(new GUIContent(name), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
hasBoundingBox = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!hasBoundingBox)
|
|
menu.AddDisabledItem(new GUIContent(name));
|
|
|
|
|
|
} else {
|
|
menu.AddItem(new GUIContent(name), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
[CustomPropertyDrawer(typeof(SpineSkin))]
|
|
public class SpineSkinDrawer : SpineTreeItemDrawerBase<SpineSkin> {
|
|
|
|
protected override void PopulateMenu (GenericMenu menu, SerializedProperty property, SpineSkin targetAttribute, SkeletonData data) {
|
|
menu.AddDisabledItem(new GUIContent(skeletonDataAsset.name));
|
|
menu.AddSeparator("");
|
|
|
|
for (int i = 0; i < data.Skins.Count; i++) {
|
|
string name = data.Skins.Items[i].Name;
|
|
if (name.StartsWith(targetAttribute.startsWith))
|
|
menu.AddItem(new GUIContent(name), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
[CustomPropertyDrawer(typeof(SpineAnimation))]
|
|
public class SpineAnimationDrawer : SpineTreeItemDrawerBase<SpineAnimation> {
|
|
protected override void PopulateMenu (GenericMenu menu, SerializedProperty property, SpineAnimation targetAttribute, SkeletonData data) {
|
|
var animations = skeletonDataAsset.GetAnimationStateData().SkeletonData.Animations;
|
|
for (int i = 0; i < animations.Count; i++) {
|
|
string name = animations.Items[i].Name;
|
|
if (name.StartsWith(targetAttribute.startsWith))
|
|
menu.AddItem(new GUIContent(name), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
[CustomPropertyDrawer(typeof(SpineEvent))]
|
|
public class SpineEventNameDrawer : SpineTreeItemDrawerBase<SpineEvent> {
|
|
protected override void PopulateMenu (GenericMenu menu, SerializedProperty property, SpineEvent targetAttribute, SkeletonData data) {
|
|
var events = skeletonDataAsset.GetSkeletonData(false).Events;
|
|
for (int i = 0; i < events.Count; i++) {
|
|
string name = events.Items[i].Name;
|
|
if (name.StartsWith(targetAttribute.startsWith))
|
|
menu.AddItem(new GUIContent(name), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
[CustomPropertyDrawer(typeof(SpineAttachment))]
|
|
public class SpineAttachmentDrawer : SpineTreeItemDrawerBase<SpineAttachment> {
|
|
protected override void PopulateMenu (GenericMenu menu, SerializedProperty property, SpineAttachment targetAttribute, SkeletonData data) {
|
|
List<Skin> validSkins = new List<Skin>();
|
|
SkeletonRenderer skeletonRenderer = null;
|
|
|
|
var component = property.serializedObject.targetObject as Component;
|
|
if (component != null) {
|
|
if (component.GetComponentInChildren<SkeletonRenderer>() != null) {
|
|
skeletonRenderer = component.GetComponentInChildren<SkeletonRenderer>();
|
|
//if (skeletonDataAsset != skeletonRenderer.skeletonDataAsset) Debug.LogWarning("DataField SkeletonDataAsset and SkeletonRenderer/SkeletonAnimation's SkeletonDataAsset do not match. Remove the explicit dataField parameter of your [SpineAttachment] field.");
|
|
skeletonDataAsset = skeletonRenderer.skeletonDataAsset;
|
|
}
|
|
}
|
|
|
|
if (skeletonRenderer != null && targetAttribute.currentSkinOnly) {
|
|
if (skeletonRenderer.skeleton.Skin != null) {
|
|
validSkins.Add(skeletonRenderer.skeleton.Skin);
|
|
} else {
|
|
validSkins.Add(data.Skins.Items[0]);
|
|
}
|
|
} else {
|
|
foreach (Skin skin in data.Skins) {
|
|
if (skin != null)
|
|
validSkins.Add(skin);
|
|
}
|
|
}
|
|
|
|
List<string> attachmentNames = new List<string>();
|
|
List<string> placeholderNames = new List<string>();
|
|
|
|
string prefix = "";
|
|
|
|
if (skeletonRenderer != null && targetAttribute.currentSkinOnly)
|
|
menu.AddDisabledItem(new GUIContent(skeletonRenderer.gameObject.name + " (SkeletonRenderer)"));
|
|
else
|
|
menu.AddDisabledItem(new GUIContent(skeletonDataAsset.name));
|
|
|
|
menu.AddSeparator("");
|
|
|
|
menu.AddItem(new GUIContent("Null"), property.stringValue == "", HandleSelect, new SpineDrawerValuePair("", property));
|
|
|
|
menu.AddSeparator("");
|
|
|
|
Skin defaultSkin = data.Skins.Items[0];
|
|
|
|
SerializedProperty slotProperty = property.serializedObject.FindProperty(targetAttribute.slotField);
|
|
string slotMatch = "";
|
|
if (slotProperty != null) {
|
|
if (slotProperty.propertyType == SerializedPropertyType.String) {
|
|
slotMatch = slotProperty.stringValue.ToLower();
|
|
}
|
|
}
|
|
|
|
foreach (Skin skin in validSkins) {
|
|
string skinPrefix = skin.Name + "/";
|
|
|
|
if (validSkins.Count > 1)
|
|
prefix = skinPrefix;
|
|
|
|
for (int i = 0; i < data.Slots.Count; i++) {
|
|
if (slotMatch.Length > 0 && data.Slots.Items[i].Name.ToLower().Contains(slotMatch) == false)
|
|
continue;
|
|
|
|
attachmentNames.Clear();
|
|
placeholderNames.Clear();
|
|
|
|
skin.FindNamesForSlot(i, attachmentNames);
|
|
if (skin != defaultSkin) {
|
|
defaultSkin.FindNamesForSlot(i, attachmentNames);
|
|
skin.FindNamesForSlot(i, placeholderNames);
|
|
}
|
|
|
|
|
|
for (int a = 0; a < attachmentNames.Count; a++) {
|
|
|
|
string attachmentPath = attachmentNames[a];
|
|
string menuPath = prefix + data.Slots.Items[i].Name + "/" + attachmentPath;
|
|
string name = attachmentNames[a];
|
|
|
|
if (targetAttribute.returnAttachmentPath)
|
|
name = skin.Name + "/" + data.Slots.Items[i].Name + "/" + attachmentPath;
|
|
|
|
if (targetAttribute.placeholdersOnly && placeholderNames.Contains(attachmentPath) == false) {
|
|
menu.AddDisabledItem(new GUIContent(menuPath));
|
|
} else {
|
|
menu.AddItem(new GUIContent(menuPath), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
}
|
|
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
[CustomPropertyDrawer(typeof(SpineBone))]
|
|
public class SpineBoneDrawer : SpineTreeItemDrawerBase<SpineBone> {
|
|
|
|
protected override void PopulateMenu (GenericMenu menu, SerializedProperty property, SpineBone targetAttribute, SkeletonData data) {
|
|
menu.AddDisabledItem(new GUIContent(skeletonDataAsset.name));
|
|
menu.AddSeparator("");
|
|
|
|
for (int i = 0; i < data.Bones.Count; i++) {
|
|
string name = data.Bones.Items[i].Name;
|
|
if (name.StartsWith(targetAttribute.startsWith))
|
|
menu.AddItem(new GUIContent(name), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
[CustomPropertyDrawer(typeof(SpineAtlasRegion))]
|
|
public class SpineAtlasRegionDrawer : PropertyDrawer {
|
|
Component component;
|
|
SerializedProperty atlasProp;
|
|
|
|
public override void OnGUI(Rect position, SerializedProperty property, GUIContent label) {
|
|
if (property.propertyType != SerializedPropertyType.String) {
|
|
EditorGUI.LabelField(position, "ERROR:", "May only apply to type string");
|
|
return;
|
|
}
|
|
|
|
component = (Component)property.serializedObject.targetObject;
|
|
|
|
if (component != null)
|
|
atlasProp = property.serializedObject.FindProperty("atlasAsset");
|
|
else
|
|
atlasProp = null;
|
|
|
|
|
|
if (atlasProp == null) {
|
|
EditorGUI.LabelField(position, "ERROR:", "Must have AtlasAsset variable!");
|
|
return;
|
|
} else if (atlasProp.objectReferenceValue == null) {
|
|
EditorGUI.LabelField(position, "ERROR:", "Atlas variable must not be null!");
|
|
return;
|
|
} else if (atlasProp.objectReferenceValue.GetType() != typeof(AtlasAsset)) {
|
|
EditorGUI.LabelField(position, "ERROR:", "Atlas variable must be of type AtlasAsset!");
|
|
}
|
|
|
|
position = EditorGUI.PrefixLabel(position, label);
|
|
|
|
if (GUI.Button(position, property.stringValue, EditorStyles.popup)) {
|
|
Selector(property);
|
|
}
|
|
|
|
}
|
|
|
|
void Selector (SerializedProperty property) {
|
|
GenericMenu menu = new GenericMenu();
|
|
AtlasAsset atlasAsset = (AtlasAsset)atlasProp.objectReferenceValue;
|
|
Atlas atlas = atlasAsset.GetAtlas();
|
|
FieldInfo field = typeof(Atlas).GetField("regions", BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.NonPublic);
|
|
List<AtlasRegion> regions = (List<AtlasRegion>)field.GetValue(atlas);
|
|
|
|
for (int i = 0; i < regions.Count; i++) {
|
|
string name = regions[i].name;
|
|
menu.AddItem(new GUIContent(name), name == property.stringValue, HandleSelect, new SpineDrawerValuePair(name, property));
|
|
}
|
|
|
|
|
|
menu.ShowAsContext();
|
|
}
|
|
|
|
static void HandleSelect (object val) {
|
|
var pair = (SpineDrawerValuePair)val;
|
|
pair.property.stringValue = pair.str;
|
|
pair.property.serializedObject.ApplyModifiedProperties();
|
|
}
|
|
|
|
}
|