BABA_YAGA/Assets/Scripts/Player/FootStep/Editor/vFootStepEditor.cs

using UnityEngine;
using UnityEditor;

namespace Invector
{
    [CanEditMultipleObjects]
    [CustomEditor(typeof(vFootStep), true)]
    public class vFootStepEditor : UnityEditor.Editor
    {
        GUISkin skin;
        bool openWindow;
        private Texture2D m_Logo = null;

        void OnEnable()
        {
            m_Logo = (Texture2D)Resources.Load("footStepIcon", typeof(Texture2D));
            CheckColliders();
        }

        public override void OnInspectorGUI()
        {
            if (!skin) skin = Resources.Load("vSkin") as GUISkin;
            GUI.skin = skin;

            if (serializedObject == null) return;

            GUILayout.BeginVertical("FOOTSTEP", "window");
            GUILayout.Label(m_Logo, GUILayout.MaxHeight(25));

            openWindow = GUILayout.Toggle(openWindow, openWindow ? "Close" : "Open", EditorStyles.toolbarButton);
            if (openWindow)
            {
                EditorGUILayout.PropertyField(serializedObject.FindProperty("m_Script"));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("animationType"));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("_volume"));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("_spawnParticle"));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("_spawnStepMark"));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("_useTriggerEnter"));             
                serializedObject.FindProperty("debugTextureName").boolValue = EditorGUILayout.Toggle("Debug Texture Name", serializedObject.FindProperty("debugTextureName").boolValue);

                GUILayout.BeginVertical("box");
                GUILayout.Box("AI Noise Settings", GUILayout.ExpandWidth(true));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("emitAINoise"));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("aiNoiseRange"));
                EditorGUILayout.PropertyField(serializedObject.FindProperty("npcLayer"));
                GUILayout.EndVertical();

                if (serializedObject.FindProperty("animationType").enumValueIndex == (int)AnimationType.Humanoid)
                {
                    GUILayout.BeginHorizontal("box");
                    if (CheckColliders())
                    {
                        EditorGUILayout.PropertyField(serializedObject.FindProperty("leftFootTrigger"), new GUIContent("", null, "leftFootTrigger"));
                        EditorGUILayout.Separator();
                        EditorGUILayout.PropertyField(serializedObject.FindProperty("rightFootTrigger"), new GUIContent("", null, "rightFootTrigger"));
                    }
                    else
                    {
                        EditorGUILayout.HelpBox("Can't Create FootStepTriggers", MessageType.Warning);
                        CheckColliders();
                    }
                    GUILayout.EndHorizontal();
                }
                else
                {
                    DrawFootStepList();
                }

                GUILayout.BeginVertical("box");
                EditorGUILayout.PropertyField(serializedObject.FindProperty("defaultSurface"));
                EditorGUILayout.HelpBox("This audio will play on any terrain or texture as the primary footstep.", MessageType.Info);
                GUILayout.EndVertical();

                GUILayout.BeginVertical("box");
                DrawMultipleSurface(serializedObject.FindProperty("customSurfaces"));
                EditorGUILayout.HelpBox("Create new CustomSurfaces on the 3rd Person Controller menu > Resources > New AudioSurface", MessageType.Info);
                GUILayout.EndVertical();
            }
            GUILayout.EndVertical();
            if (GUI.changed)
            {
                serializedObject.ApplyModifiedProperties();
            }
        }
        
        bool CheckColliders()
        {
            if (AssetDatabase.Contains(target))
                return true;

            var transform = (serializedObject.targetObject as vFootStep).transform;
            if (transform == null) return false;
            var animator = transform.GetComponent<Animator>();
            if (animator == null) return false;
            var leftFoot = animator.GetBoneTransform(HumanBodyBones.LeftFoot);
            vFootStepTrigger leftFoot_trigger = null;
            if (leftFoot != null)
                leftFoot_trigger = leftFoot.GetComponentInChildren<vFootStepTrigger>();

            if (leftFoot_trigger == null && leftFoot != null)
            {
                var lFoot = new GameObject("leftFoot_trigger");
                lFoot.tag = "Ignore Ragdoll";
                var collider = lFoot.AddComponent<SphereCollider>();
                collider.radius = 0.1f;
                leftFoot_trigger = lFoot.AddComponent<vFootStepTrigger>();
                leftFoot_trigger.transform.position = new Vector3(leftFoot.position.x, transform.position.y, leftFoot.position.z);
                leftFoot_trigger.transform.rotation = transform.rotation;
                leftFoot_trigger.gameObject.layer = LayerMask.NameToLayer("Ignore Raycast");
                leftFoot_trigger.transform.parent = leftFoot;
                serializedObject.FindProperty("leftFootTrigger").objectReferenceValue = leftFoot_trigger;
                serializedObject.ApplyModifiedProperties();
            }
            serializedObject.FindProperty("leftFootTrigger").objectReferenceValue = leftFoot_trigger;

            if (leftFoot_trigger != null && leftFoot_trigger.GetComponent<Collider>() == null)
            {
                var collider = leftFoot_trigger.gameObject.AddComponent<SphereCollider>();
                collider.radius = 0.1f;
            }

            var rightFoot = animator.GetBoneTransform(HumanBodyBones.RightFoot);
            vFootStepTrigger rightFoot_trigger = null;
            if (rightFoot != null)
                rightFoot_trigger = rightFoot.GetComponentInChildren<vFootStepTrigger>();

            if (rightFoot_trigger == null && rightFoot != null)
            {
                var rFoot = new GameObject("rightFoot_trigger");
                rFoot.tag = "Ignore Ragdoll";
                var collider = rFoot.AddComponent<SphereCollider>();
                collider.radius = 0.1f;
                rightFoot_trigger = rFoot.gameObject.AddComponent<vFootStepTrigger>();
                rightFoot_trigger.transform.position = new Vector3(rightFoot.position.x, transform.position.y, rightFoot.position.z);
                rightFoot_trigger.transform.rotation = transform.rotation;
                rightFoot_trigger.gameObject.layer = LayerMask.NameToLayer("Ignore Raycast");
                rightFoot_trigger.transform.parent = rightFoot;
                serializedObject.FindProperty("rightFootTrigger").objectReferenceValue = rightFoot_trigger;
                serializedObject.ApplyModifiedProperties();
            }
            serializedObject.FindProperty("rightFootTrigger").objectReferenceValue = rightFoot_trigger;
            if (rightFoot_trigger != null && rightFoot_trigger.GetComponent<Collider>() == null)
            {
                var collider = rightFoot_trigger.gameObject.AddComponent<SphereCollider>();
                collider.radius = 0.1f;
            }

            if (serializedObject.FindProperty("rightFootTrigger").objectReferenceValue != null && serializedObject.FindProperty("leftFootTrigger").objectReferenceValue != null) return true;
            return false;
        }

        void DrawFootStepList()
        {
            var footStepList = serializedObject.FindProperty("footStepTriggers");
            if (footStepList != null)
            {
                GUILayout.BeginVertical("TRIGGERS", "window");
                GUILayout.Space(30);
                GUILayout.BeginHorizontal();
                if (GUILayout.Button("New", EditorStyles.miniButton))
                {
                    footStepList.arraySize++;
                    var go = new GameObject("Trigger-" + footStepList.arraySize.ToString("00"), typeof(vFootStepTrigger), typeof(SphereCollider));
                    go.GetComponent<SphereCollider>().radius = 0.05f;
                    go.transform.position = (target as vFootStep).transform.position;
                    go.layer = LayerMask.NameToLayer("Ignore Raycast");
                    go.transform.parent = (target as vFootStep).transform;

                    footStepList.GetArrayElementAtIndex(footStepList.arraySize - 1).objectReferenceValue = go.GetComponent<vFootStepTrigger>();
                }
                GUILayout.EndHorizontal();
                GUILayout.BeginVertical();
                for (int i = 0; i < footStepList.arraySize; i++)
                {
                    if (!DrawFootStepElement(footStepList, footStepList.GetArrayElementAtIndex(i), i)) break;
                }
                GUILayout.EndVertical();
                GUILayout.EndVertical();
            }
        }

        bool DrawFootStepElement(SerializedProperty list, SerializedProperty footStepElement, int index)
        {
            GUILayout.BeginHorizontal("box");
            EditorGUILayout.PropertyField(footStepElement, new GUIContent(""));
            if (GUILayout.Button("-", EditorStyles.miniButtonMid, GUILayout.MaxWidth(15)))
            {
                if ((footStepElement.objectReferenceValue as vFootStepTrigger) != null)
                {
                    DestroyImmediate((footStepElement.objectReferenceValue as vFootStepTrigger).gameObject);
                    list.DeleteArrayElementAtIndex(index);
                }

                list.DeleteArrayElementAtIndex(index);
                GUILayout.EndHorizontal();
                return false;
            }
            GUILayout.EndHorizontal();
            return true;
        }

        void DrawSingleSurface(SerializedProperty surface, bool showListNames)
        {
            //GUILayout.BeginVertical("window");
            EditorGUILayout.PropertyField(surface.FindPropertyRelative("source"), false);
            EditorGUILayout.PropertyField(surface.FindPropertyRelative("name"), new GUIContent("Surface Name"), false);

            if (showListNames)
                DrawSimpleList(surface.FindPropertyRelative("TextureOrMaterialNames"), false);

            DrawSimpleList(surface.FindPropertyRelative("audioClips"), true);
            //GUILayout.EndVertical();
        }

        void DrawMultipleSurface(SerializedProperty surfaceList)
        {
            //GUILayout.BeginVertical();
            EditorGUILayout.PropertyField(surfaceList, new GUIContent("Custom Surfaces"));
            if (surfaceList.isExpanded)
            {
                GUILayout.BeginHorizontal();
                if (GUILayout.Button("Add"))
                {
                    surfaceList.arraySize++;
                }
                if (GUILayout.Button("Clear"))
                {
                    surfaceList.arraySize = 0;
                }
                GUILayout.EndHorizontal();
                EditorGUILayout.Space();
                for (int i = 0; i < surfaceList.arraySize; i++)
                {
                    GUILayout.BeginHorizontal();
                    GUILayout.BeginHorizontal("box");
                    EditorGUILayout.Space();
                    if (i < surfaceList.arraySize && i >= 0)
                    {
                        GUILayout.BeginVertical();
                        EditorGUILayout.PropertyField(surfaceList.GetArrayElementAtIndex(i),
                            new GUIContent(surfaceList.GetArrayElementAtIndex(i).objectReferenceValue != null ? surfaceList.GetArrayElementAtIndex(i).objectReferenceValue.name : "Surface " + (i + 1).ToString("00")));
                        EditorGUILayout.Space();
                        GUILayout.EndVertical();
                    }
                    GUILayout.EndHorizontal();

                    if (GUILayout.Button("-"))
                    {
                        surfaceList.DeleteArrayElementAtIndex(i);
                    }
                    GUILayout.EndHorizontal();
                }
                //GUILayout.EndVertical();
            }
        }

        void DrawTextureNames(SerializedProperty textureNames)
        {
            for (int i = 0; i < textureNames.arraySize; i++)
                EditorGUILayout.PropertyField(textureNames.GetArrayElementAtIndex(i), true);
        }

        void DrawSimpleList(SerializedProperty list, bool useDraBox)
        {
            EditorGUILayout.PropertyField(list);

            if (list.isExpanded)
            {
                if (useDraBox)
                    DrawDragBox(list);
                GUILayout.BeginHorizontal();
                if (GUILayout.Button("Add"))
                {
                    list.arraySize++;
                }
                if (GUILayout.Button("Clear"))
                {
                    list.arraySize = 0;
                }
                GUILayout.EndHorizontal();
                EditorGUILayout.Space();

                for (int i = 0; i < list.arraySize; i++)
                {
                    GUILayout.BeginHorizontal();
                    if (GUILayout.Button("-"))
                    {
                        RemoveElementAtIndex(list, i);
                    }

                    if (i < list.arraySize && i >= 0)
                        EditorGUILayout.PropertyField(list.GetArrayElementAtIndex(i), new GUIContent("", null, ""));

                    GUILayout.EndHorizontal();
                }
            }
        }

        private void RemoveElementAtIndex(SerializedProperty array, int index)
        {
            if (index != array.arraySize - 1)
            {
                array.GetArrayElementAtIndex(index).objectReferenceValue = array.GetArrayElementAtIndex(array.arraySize - 1).objectReferenceValue;
            }
            array.arraySize--;
        }

        void DrawDragBox(SerializedProperty list)
        {
            //var dragAreaGroup = GUILayoutUtility.GetRect(0f, 35f, GUILayout.ExpandWidth(true));
            GUI.skin.box.alignment = TextAnchor.MiddleCenter;
            GUI.skin.box.normal.textColor = Color.white;
            GUILayout.Box("Drag your audio clips here!", "box", GUILayout.MinHeight(50), GUILayout.ExpandWidth(true));
            var dragAreaGroup = GUILayoutUtility.GetLastRect();

            switch (Event.current.type)
            {
                case EventType.DragUpdated:
                case EventType.DragPerform:
                    if (!dragAreaGroup.Contains(Event.current.mousePosition))
                        break;

                    DragAndDrop.visualMode = DragAndDropVisualMode.Copy;

                    if (Event.current.type == EventType.DragPerform)
                    {
                        DragAndDrop.AcceptDrag();

                        foreach (var dragged in DragAndDrop.objectReferences)
                        {
                            var clip = dragged as AudioClip;
                            if (clip == null)
                                continue;
                            list.arraySize++;
                            list.GetArrayElementAtIndex(list.arraySize - 1).objectReferenceValue = clip;
                        }
                    }
                    serializedObject.ApplyModifiedProperties();
                    Event.current.Use();
                    break;
            }
        }

        public override bool UseDefaultMargins()
        {
            return false;
        }
    }
}