Add default value handling when adding item from inspector

Unity-Technologies · May 24, 2024 · fd7e911 · fd7e911
1 parent 5f07858
commit fd7e911
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/com.unity.cinemachine/Editor/Utility/SplineDataInspectorUtility.cs b/com.unity.cinemachine/Editor/Utility/SplineDataInspectorUtility.cs
@@ -64,6 +64,7 @@ public static PropertyField CreateDataListField<T>(
             GetSplineDelegate getSpline)
         {
             var sortMethod = splineData.GetType().GetMethod("ForceSort", BindingFlags.Instance | BindingFlags.NonPublic);
+            var setDataPointMethod = splineData.GetType().GetMethod("SetDataPoint", BindingFlags.Instance | BindingFlags.Public);
 
             var pathUnitProp = splineDataProp.FindPropertyRelative("m_IndexUnit");
             var arrayProp = splineDataProp.FindPropertyRelative("m_DataPoints");
@@ -77,6 +78,16 @@ public static PropertyField CreateDataListField<T>(
                 listView.showBoundCollectionSize = false;
                 listView.showAddRemoveFooter = true;
                 listView.showAlternatingRowBackgrounds = AlternatingRowBackground.None;
+
+                var button = list.Q<Button>("unity-list-view__add-button");
+                button.clicked += () => 
+                {
+                    if (arrayProp.arraySize == 1)
+                    {
+                        setDataPointMethod.Invoke(splineData, new object[] { 0, new DataPoint<T> () { Value = splineData.DefaultValue } });
+                        arrayProp.serializedObject.Update();
+                    }
+                };
             });
 
             list.TrackPropertyValue(arrayProp, (p) =>