Adding a custom editor for BufferSensorComponent

Author: vincentpierre

com.unity.ml-agents/Editor/BufferSensorComponentEditor.cs

@@ -0,0 +1,31 @@
+using UnityEditor;
+using Unity.MLAgents.Sensors;
+
+namespace Unity.MLAgents.Editor
+{
+ [CustomEditor(typeof(BufferSensorComponent))]
+ [CanEditMultipleObjects]
+ internal class BufferSensorComponentEditor : UnityEditor.Editor
+ {
+ public override void OnInspectorGUI()
+ {
+ var so = serializedObject;
+ so.Update();
+
+ // Drawing the BufferSensorComponent
+
+ EditorGUI.BeginDisabledGroup(!EditorUtilities.CanUpdateModelProperties());
+ {
+ // These fields affect the sensor order or observation size,
+ // So can't be changed at runtime.
+ EditorGUILayout.PropertyField(so.FindProperty("m_SensorName"), true);
+ EditorGUILayout.PropertyField(so.FindProperty("m_ObservableSize"), true);
+ EditorGUILayout.PropertyField(so.FindProperty("m_MaxNumObservables"), true);
+ }
+ EditorGUI.EndDisabledGroup();
+
+ so.ApplyModifiedProperties();
+ }
+
+ }
+}

com.unity.ml-agents/Editor/BufferSensorComponentEditor.cs.meta

@@ -19,20 +19,32 @@ public string SensorName
  get { return m_SensorName; }
  set { m_SensorName = value; }
  }
-
- public string m_SensorName = "BufferSensor";
+ [HideInInspector, SerializeField]
+ private string m_SensorName = "BufferSensor";
 
  /// <summary>
  /// This is how many floats each entities will be represented with. This number
  /// is fixed and all entities must have the same representation.
  /// </summary>
- public int ObservableSize;
+ public int ObservableSize
+ {
+ get { return m_ObservableSize; }
+ set { m_ObservableSize = value; }
+ }
+ [HideInInspector, SerializeField]
+ private int m_ObservableSize;
 
  /// <summary>
  /// This is the maximum number of entities the `BufferSensor` will be able to
  /// collect.
  /// </summary>
- public int MaxNumObservables;
+ public int MaxNumObservables
+ {
+ get { return m_MaxNumObservables; }
+ set { m_MaxNumObservables = value; }
+ }
+ [HideInInspector, SerializeField]
+ private int m_MaxNumObservables;
 
  private BufferSensor m_Sensor;