com.unity.perception/com.unity.perception/Editor/Randomization/Editors/ScenarioBaseEditor.cs

using System.Collections;
using System.Linq;
using UnityEditor;
using UnityEditor.UIElements;
using UnityEngine.Experimental.Perception.Randomization.Scenarios;
using UnityEngine.Experimental.Perception.Randomization.VisualElements;
using UnityEngine.UIElements;

namespace UnityEngine.Experimental.Perception.Randomization.Editor
{
    [CustomEditor(typeof(ScenarioBase), true)]
    class ScenarioBaseEditor : UnityEditor.Editor
    {
        bool m_HasConstantsField;
        ScenarioBase m_Scenario;
        SerializedObject m_SerializedObject;
        VisualElement m_Root;
        VisualElement m_InspectorPropertiesContainer;
        VisualElement m_ConstantsListVisualContainer;
        VisualElement m_RandomizerListPlaceholder;

        public override VisualElement CreateInspectorGUI()
        {
            m_Scenario = (ScenarioBase)target;
            m_SerializedObject = new SerializedObject(m_Scenario);
            m_Root = AssetDatabase.LoadAssetAtPath<VisualTreeAsset>(
                $"{StaticData.uxmlDir}/ScenarioBaseElement.uxml").CloneTree();

            m_RandomizerListPlaceholder = m_Root.Q<VisualElement>("randomizer-list-placeholder");

            CreatePropertyFields();
            CheckIfConstantsExist();

            var serializeConstantsButton = m_Root.Q<Button>("serialize");
            serializeConstantsButton.clicked += () =>
            {
                m_Scenario.SerializeToFile();
                AssetDatabase.Refresh();
                var newConfigFileAsset = AssetDatabase.LoadAssetAtPath<Object>(m_Scenario.defaultConfigFileAssetPath);
                EditorGUIUtility.PingObject(newConfigFileAsset);
            };

            var deserializeConstantsButton = m_Root.Q<Button>("deserialize");
            deserializeConstantsButton.clicked += () =>
            {
                Undo.RecordObject(m_Scenario, "Deserialized scenario configuration");
                m_Scenario.DeserializeFromFile(m_Scenario.defaultConfigFilePath);
            };

            return m_Root;
        }

        void CreatePropertyFields()
        {
            m_InspectorPropertiesContainer = m_Root.Q<VisualElement>("inspector-properties");
            m_InspectorPropertiesContainer.Clear();

            m_ConstantsListVisualContainer = m_Root.Q<VisualElement>("constants-list");
            m_ConstantsListVisualContainer.Clear();

            var iterator = m_SerializedObject.GetIterator();
            var foundProperties = false;
            m_HasConstantsField = false;
            if (iterator.NextVisible(true))
            {
                do
                {
                    switch (iterator.name)
                    {
                        case "m_Script":
                            break;
                        case "m_Randomizers":
                            m_RandomizerListPlaceholder.Add(new RandomizerList(iterator.Copy()));
                            break;
                        case "constants":
                            m_HasConstantsField = true;
                            CreateConstantsFieldsWithToolTips(iterator.Copy());
                            break;
                        default:
                        {
                            foundProperties = true;
                            var propertyField = new PropertyField(iterator.Copy());
                            propertyField.Bind(m_SerializedObject);
                            m_InspectorPropertiesContainer.Add(propertyField);
                            break;
                        }
                    }
                } while (iterator.NextVisible(false));
            }

            if (!foundProperties)
                m_InspectorPropertiesContainer.style.display = new StyleEnum<DisplayStyle>(DisplayStyle.None);
        }

        void CheckIfConstantsExist()
        {
            m_ConstantsListVisualContainer = m_Root.Q<VisualElement>("constants-container");
            if (!m_HasConstantsField)
            {
                m_InspectorPropertiesContainer.style.marginBottom = 0;
                m_ConstantsListVisualContainer.style.display = new StyleEnum<DisplayStyle>(DisplayStyle.None);
            }
        }

        void CreateConstantsFieldsWithToolTips(SerializedProperty constantsProperty)
        {
            constantsProperty.NextVisible(true);
            do
            {
                var constantsPropertyField = new PropertyField(constantsProperty.Copy());
                constantsPropertyField.Bind(m_SerializedObject);

                var originalField = target.GetType().GetField("constants").FieldType.GetField(constantsProperty.name);
                var tooltipAttribute = originalField.GetCustomAttributes(true).ToList().Find(att => att.GetType() == typeof(TooltipAttribute));
                if (tooltipAttribute != null)
                    constantsPropertyField.tooltip = (tooltipAttribute as TooltipAttribute)?.tooltip;

                m_ConstantsListVisualContainer.Add(constantsPropertyField);

            } while (constantsProperty.NextVisible(true));
        }
    }
}