diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Constants.cs b/UnitySDK/Assets/ML-Agents/Scripts/Constants.cs new file mode 100644 index 0000000000..720e2863f2 --- /dev/null +++ b/UnitySDK/Assets/ML-Agents/Scripts/Constants.cs @@ -0,0 +1,11 @@ +namespace MLAgents +{ + /// + /// Grouping for use in AddComponentMenu (instead of nesting the menus). + /// + public enum MenuGroup + { + Default=0, + Sensors=50 + } +} diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Constants.cs.meta b/UnitySDK/Assets/ML-Agents/Scripts/Constants.cs.meta new file mode 100644 index 0000000000..f963ba55aa --- /dev/null +++ b/UnitySDK/Assets/ML-Agents/Scripts/Constants.cs.meta @@ -0,0 +1,3 @@ +fileFormatVersion: 2 +guid: 0622d88401ec464d9d2cf2fb03ce17b5 +timeCreated: 1579215785 \ No newline at end of file diff --git a/UnitySDK/Assets/ML-Agents/Scripts/DemonstrationRecorder.cs b/UnitySDK/Assets/ML-Agents/Scripts/DemonstrationRecorder.cs index 33881f7102..3adf6cc990 100644 --- a/UnitySDK/Assets/ML-Agents/Scripts/DemonstrationRecorder.cs +++ b/UnitySDK/Assets/ML-Agents/Scripts/DemonstrationRecorder.cs @@ -10,6 +10,7 @@ namespace MLAgents /// Demonstration Recorder Component. /// [RequireComponent(typeof(Agent))] + [AddComponentMenu("ML Agents/Demonstration Recorder", (int) MenuGroup.Default)] public class DemonstrationRecorder : MonoBehaviour { public bool record; diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Policy/BehaviorParameters.cs b/UnitySDK/Assets/ML-Agents/Scripts/Policy/BehaviorParameters.cs index dca26b0d6a..3e07a33993 100644 --- a/UnitySDK/Assets/ML-Agents/Scripts/Policy/BehaviorParameters.cs +++ b/UnitySDK/Assets/ML-Agents/Scripts/Policy/BehaviorParameters.cs @@ -9,6 +9,8 @@ namespace MLAgents /// /// The Factory to generate policies. /// + /// + [AddComponentMenu("ML Agents/Behavior Parameters", (int) MenuGroup.Default)] public class BehaviorParameters : MonoBehaviour { diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/CameraSensorComponent.cs b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/CameraSensorComponent.cs index 607706c194..90a8cd92a9 100644 --- a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/CameraSensorComponent.cs +++ b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/CameraSensorComponent.cs @@ -3,6 +3,7 @@ namespace MLAgents.Sensor { + [AddComponentMenu("ML Agents/Camera Sensor", (int) MenuGroup.Sensors)] public class CameraSensorComponent : SensorComponent { public new Camera camera; diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/ISensor.cs b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/ISensor.cs index 9fee45a0d6..5efa86a23e 100644 --- a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/ISensor.cs +++ b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/ISensor.cs @@ -75,5 +75,4 @@ public static int ObservationSize(this ISensor sensor) return count; } } - } diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent2D.cs b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent2D.cs index abb750128e..36211cadc5 100644 --- a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent2D.cs +++ b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent2D.cs @@ -1,5 +1,8 @@ +using UnityEngine; + namespace MLAgents.Sensor { + [AddComponentMenu("ML Agents/Ray Perception Sensor 2D", (int) MenuGroup.Sensors)] public class RayPerceptionSensorComponent2D : RayPerceptionSensorComponentBase { public override RayPerceptionSensor.CastType GetCastType() diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent3D.cs b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent3D.cs index 6d7718971d..560fc82793 100644 --- a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent3D.cs +++ b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RayPerceptionSensorComponent3D.cs @@ -3,6 +3,7 @@ namespace MLAgents.Sensor { + [AddComponentMenu("ML Agents/Ray Perception Sensor 3D", (int) MenuGroup.Sensors)] public class RayPerceptionSensorComponent3D : RayPerceptionSensorComponentBase { [Header("3D Properties", order = 100)] diff --git a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RenderTextureSensorComponent.cs b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RenderTextureSensorComponent.cs index 8c22bcd80f..7ad185517a 100644 --- a/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RenderTextureSensorComponent.cs +++ b/UnitySDK/Assets/ML-Agents/Scripts/Sensor/RenderTextureSensorComponent.cs @@ -3,6 +3,7 @@ namespace MLAgents.Sensor { + [AddComponentMenu("ML Agents/Render Texture Sensor", (int) MenuGroup.Sensors)] public class RenderTextureSensorComponent : SensorComponent { public RenderTexture renderTexture;