UniVRM/Assets/VRM10_Samples/ClothSample/ClothWarp/Editor/RotateParticleRuntimeProviderEditor.cs

using UnityEngine.UIElements;
using UnityEditor;
using UnityEditor.UIElements;


namespace UniVRM10.ClothWarp.Components
{
    [CustomEditor(typeof(ClothWarpRuntimeProvider))]
    public class RotateParticleRuntimeProviderEditor : Editor
    {
        ClothWarpRuntimeProvider _target;
        Vrm10Instance _vrm;

        void OnEnable()
        {
            _target = (ClothWarpRuntimeProvider)target;
            if (_target != null)
            {
                _vrm = _target.GetComponent<Vrm10Instance>();
            }
        }

        public override VisualElement CreateInspectorGUI()
        {
            var root = new VisualElement();
            root.Bind(serializedObject);
            {
                var s = new PropertyField { bindingPath = "m_Script" };
                s.SetEnabled(false);
                root.Add(s);
            }
            root.Add(new PropertyField { bindingPath = nameof(_target.UseJob) });
            root.Add(new PropertyField { bindingPath = nameof(_target.Warps) });
            root.Add(new PropertyField { bindingPath = nameof(_target.Cloths) });

            {
                var setup = new Foldout { text = "Setup" };

                var from_vrm10 = new Button { text = "Load VRM10 Springs to ClothWarp Warps" };
                setup.Add(from_vrm10);
                from_vrm10.RegisterCallback<ClickEvent>(e =>
                {
                    Undo.IncrementCurrentGroup();
                    Undo.SetCurrentGroupName("Load Vrm-1.0 Springs to ClothWarp Warps");
                    var undo = Undo.GetCurrentGroup();

                    // attach ClothWarp from VRM10Instance.Springs
                    ClothWarpRuntimeProvider.FromVrm10(_vrm, Undo.AddComponent<ClothWarpRoot>);
                    Undo.RegisterFullObjectHierarchyUndo(_vrm.gameObject, "RegisterFullObjectHierarchyUndo");

                    // update ClothWarpRuntimeProvider
                    Undo.RegisterCompleteObjectUndo(_target, "RegisterCompleteObjectUndo");
                    _target.Reset();

                    Undo.CollapseUndoOperations(undo);
                });

                var clear_vrm10_springs = new Button { text = "Clear Vrm-1.0 springs" };
                setup.Add(clear_vrm10_springs);
                clear_vrm10_springs.RegisterCallback<ClickEvent>(e =>
                {
                    Undo.IncrementCurrentGroup();
                    Undo.SetCurrentGroupName("Clear VRM10 Srpings");
                    var undo = Undo.GetCurrentGroup();

                    Undo.RegisterCompleteObjectUndo(_vrm, "RegisterCompleteObjectUndo");
                    foreach (var spring in _vrm.SpringBone.Springs)
                    {
                        if (spring != null)
                        {
                            foreach (var joint in spring.Joints)
                            {
                                if (joint != null)
                                {
                                    Undo.DestroyObjectImmediate(joint);
                                }
                            }
                        }
                        spring.Joints.Clear();
                    }
                    _vrm.SpringBone.Springs.Clear();
                    Undo.RegisterFullObjectHierarchyUndo(_vrm.gameObject, "RegisterFullObjectHierarchyUndo");

                    Undo.CollapseUndoOperations(undo);
                });

                var reload = new Button { text = "Reload" };
                setup.Add(reload);
                reload.RegisterCallback<ClickEvent>(e =>
                {
                    _target.Reset();
                });

                root.Add(setup);
            }

            {
                // runtime: reset button
                var runtime = new Foldout { text = "Runtime" };
                root.Add(runtime);

                var button = new Button
                {
                    text = "RestoreInitialTransform",
                };
                runtime.Add(button);
                button.RegisterCallback<ClickEvent>((e) =>
                {
                    _vrm.Runtime.SpringBone.RestoreInitialTransform();
                });
            }

            return root;
        }
    }
}