Refactor Node editors.

thebitcave · Dec 31, 2019 · 799eb9b · 799eb9b
1 parent 105c301
commit 799eb9b
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 5 deletions.
diff --git a/Scripts/Agents/AI/Graph/Actions/Editor/AIActionChangeAIBrainStateCommandNodeEditor.cs b/Scripts/Agents/AI/Graph/Actions/Editor/AIActionChangeAIBrainStateCommandNodeEditor.cs
@@ -9,12 +9,8 @@ public class AIActionChangeAIBrainStateCommandNodeEditor : AIActionNodeEditor
         private SerializedProperty _channel;
         private SerializedProperty _stateName;
 
-        public override void OnBodyGUI()
+        protected override void SerializeAdditionalProperties()
         {
-            base.OnBodyGUI();
-
-            if (CollapseNodeOn) return;
-
             EditorGUIUtility.labelWidth = 100;
             _channel = serializedObject.FindProperty("channel");
             _stateName = serializedObject.FindProperty("stateName");

diff --git a/Scripts/Agents/AI/Graph/Editor/AIActionNodeEditor.cs b/Scripts/Agents/AI/Graph/Editor/AIActionNodeEditor.cs
@@ -27,8 +27,16 @@ public override void OnBodyGUI()
             NodeEditorGUILayout.PropertyField(_label);
             NodeEditorGUILayout.PropertyField(_output);
             serializedObject.ApplyModifiedProperties();
+
+            if (CollapseNodeOn) return;
+            SerializeAdditionalProperties();
         }
 
+        protected virtual void SerializeAdditionalProperties()
+        {
+            // Add Action properties.
+        }
+
         protected bool CollapseNodeOn => ActionNode.BrainGraph.IsNodeCollapseModeOn && Selection.activeObject != target;
     }
 }
diff --git a/Scripts/Agents/AI/Graph/Editor/AIDecisionNodeEditor.cs b/Scripts/Agents/AI/Graph/Editor/AIDecisionNodeEditor.cs
@@ -27,6 +27,14 @@ public override void OnBodyGUI()
             NodeEditorGUILayout.PropertyField(_label);
             NodeEditorGUILayout.PropertyField(_output);
             serializedObject.ApplyModifiedProperties();
+
+            if (CollapseNodeOn) return;
+            SerializeAdditionalProperties();
+        }
+
+        protected virtual void SerializeAdditionalProperties()
+        {
+            // Add Decision properties.
         }
 
         protected bool CollapseNodeOn => DecisionNode.BrainGraph.IsNodeCollapseModeOn && Selection.activeObject != target;