Initial Commit

Assets/Imstk/Scripts/Editor/DynamicalModelEditor.cs

@@ -0,0 +1,146 @@
+using ImstkUnity;
+using System.Linq;
+using System.Collections.Generic;
+using UnityEditor;
+using UnityEngine;
+
+namespace ImstkEditor
+{
+    public class DynamicalModelEditor : Editor
+    {
+		bool _collisionsFolded = false;
+        List<DynamicalModel> _allDeformables;
+
+        public DynamicalModelEditor()
+        {
+            EditorApplication.hierarchyChanged += OnHierarchyChanged;
+        }
+
+        ~DynamicalModelEditor() 
+        {
+            EditorApplication.hierarchyChanged -= OnHierarchyChanged;
+        }
+
+        public void OnEnable()
+        {
+        }
+
+        private void OnHierarchyChanged()
+        {
+            _allDeformables = (Resources.FindObjectsOfTypeAll(typeof(DynamicalModel)) as DynamicalModel[]).ToList<DynamicalModel>();
+            SimulationManager.Instance().collisions.RemoveAllNull();
+        }
+
+        public void HandleColliders(DynamicalModel script) 
+		{
+            if (_allDeformables == null) 
+            {
+                OnHierarchyChanged();
+            }
+
+			EditorGUI.BeginChangeCheck();
+            var editorData = DrawColliders(script);
+            if (EditorGUI.EndChangeCheck())
+            {
+                var simulationManager = SimulationManager.Instance();
+                Undo.RegisterCompleteObjectUndo(simulationManager, "Update Collisions");
+                simulationManager.collisions = editorData;
+            }
+		}
+		
+        protected CollisionInteractions DrawColliders(DynamicalModel script)
+        {
+            var allCollisions = SimulationManager.Instance().collisions;
+            var editorData = new CollisionInteractions(allCollisions);
+
+            _collisionsFolded = EditorGUILayout.Foldout(_collisionsFolded, "Colliding Objects");
+            if (_collisionsFolded)
+            {
+                EditorGUILayout.BeginVertical();
+                for (int i = 0; i < _allDeformables.Count; ++i)
+                {
+                    var item = _allDeformables[i];
+                    if (item == script) continue;
+                    EditorGUILayout.BeginVertical();
+
+                    // Title line with Toggle
+                    EditorGUILayout.BeginHorizontal();
+                    var enabled = allCollisions.IsEnabled(script, item);
+
+                    var autoType = ImstkUnity.CollisionInteraction.GetCDType(script, item);
+
+                    var newEnabled = false;
+
+                    EditorGUI.BeginDisabledGroup(autoType == "");
+                    newEnabled = EditorGUILayout.Toggle(enabled, GUILayout.Width(20));
+                    EditorGUI.EndDisabledGroup();
+                    if (enabled != newEnabled)
+                    {
+                        if (newEnabled)
+                        {
+                            editorData.Add(script, item);
+                        }
+                        else
+                        {
+                            editorData.Remove(script, item);
+                        }
+                    }
+                    EditorGUI.BeginDisabledGroup(!newEnabled && autoType == "");
+                    EditorGUILayout.LabelField(item.name);
+                    if (autoType == "")
+                    {
+                        EditorGUILayout.LabelField("No collision type available");
+                    }
+                    EditorGUI.EndDisabledGroup();
+                    EditorGUILayout.EndHorizontal();
+                    if (newEnabled)
+                    {
+                        EditorGUILayout.BeginVertical();
+                        var d = editorData.GetData(script, item);
+
+                        var _message = "Auto Type: ";
+                        if (d.model1 != null && d.model1.GetCollidingGeometry() != null &&
+                            d.model2 != null && d.model2.GetCollidingGeometry() != null)
+                        {
+                            _message += ImstkUnity.CollisionInteraction.GetCDType(d.model1, d.model2);
+                        }
+                        else
+                        {
+                            _message += " None ";
+                        }
+                        EditorGUILayout.LabelField(_message);
+
+                        var selected = System.Array.IndexOf(CollisionInteractionEditor.CDOptions, d.collisionTypeName);
+                        
+                        if (selected < 0) selected = 0;
+
+                        selected = EditorGUILayout.Popup("Detection Type", selected, CollisionInteractionEditor.CDOptions);
+                        d.collisionTypeName = CollisionInteractionEditor.CDOptions[selected];
+
+                        d.friction = EditorGUILayout.DoubleField("Friction", d.friction);
+                        d.restitution = EditorGUILayout.DoubleField("Restitution", d.restitution);
+
+                        var guiContent = new GUIContent("Deform. Stiffness 1", "A good default value is 1/number of iterations from the simulation manager");
+                        d.deformableStiffness1 = EditorGUILayout.DoubleField(guiContent, d.deformableStiffness1);
+
+                        guiContent = new GUIContent("Deform. Stiffness 2", "A good default value is 1/number of iterations from the simulation manager");
+                        d.deformableStiffness2 = EditorGUILayout.DoubleField(guiContent, d.deformableStiffness2);
+
+                        d.rigidBodyCompliance = EditorGUILayout.DoubleField("Rigid Compliance", d.rigidBodyCompliance);
+
+                        EditorGUILayout.EndVertical();
+                    }
+
+                    EditorGUILayout.EndVertical();
+                    // Needs to move into the undo section
+
+
+                }
+                EditorGUILayout.EndVertical();
+            }
+
+            return editorData;
+        }
+    }
+}
+