您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
94 行
3.3 KiB
94 行
3.3 KiB
using UnityEngine;
|
|
using UnityEditor;
|
|
using Barracuda;
|
|
|
|
namespace MLAgents
|
|
{
|
|
internal class RayPerceptionSensorComponentBaseEditor : Editor
|
|
{
|
|
bool m_RequireSensorUpdate;
|
|
|
|
protected void OnRayPerceptionInspectorGUI(bool is3d)
|
|
{
|
|
var so = serializedObject;
|
|
so.Update();
|
|
|
|
// Drawing the RayPerceptionSensorComponent
|
|
EditorGUI.BeginChangeCheck();
|
|
EditorGUI.indentLevel++;
|
|
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_SensorName"), true);
|
|
|
|
// Because the number of rays and the tags affect the observation shape,
|
|
// they are not editable during play mode.
|
|
EditorGUI.BeginDisabledGroup(Application.isPlaying);
|
|
{
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_DetectableTags"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_RaysPerDirection"), true);
|
|
}
|
|
EditorGUI.EndDisabledGroup();
|
|
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_MaxRayDegrees"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_SphereCastRadius"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_RayLength"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_RayLayerMask"), true);
|
|
|
|
// Because the number of observation stacks affects the observation shape,
|
|
// it is not editable during play mode.
|
|
EditorGUI.BeginDisabledGroup(Application.isPlaying);
|
|
{
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_ObservationStacks"), true);
|
|
}
|
|
EditorGUI.EndDisabledGroup();
|
|
|
|
if (is3d)
|
|
{
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_StartVerticalOffset"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_EndVerticalOffset"), true);
|
|
}
|
|
|
|
EditorGUILayout.PropertyField(so.FindProperty("rayHitColor"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("rayMissColor"), true);
|
|
|
|
EditorGUI.indentLevel--;
|
|
if (EditorGUI.EndChangeCheck())
|
|
{
|
|
m_RequireSensorUpdate = true;
|
|
}
|
|
|
|
UpdateSensorIfDirty();
|
|
so.ApplyModifiedProperties();
|
|
}
|
|
|
|
|
|
void UpdateSensorIfDirty()
|
|
{
|
|
if (m_RequireSensorUpdate)
|
|
{
|
|
var sensorComponent = serializedObject.targetObject as RayPerceptionSensorComponentBase;
|
|
sensorComponent?.UpdateSensor();
|
|
m_RequireSensorUpdate = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
[CustomEditor(typeof(RayPerceptionSensorComponent2D))]
|
|
[CanEditMultipleObjects]
|
|
internal class RayPerceptionSensorComponent2DEditor : RayPerceptionSensorComponentBaseEditor
|
|
{
|
|
public override void OnInspectorGUI()
|
|
{
|
|
OnRayPerceptionInspectorGUI(false);
|
|
}
|
|
}
|
|
|
|
[CustomEditor(typeof(RayPerceptionSensorComponent3D))]
|
|
[CanEditMultipleObjects]
|
|
internal class RayPerceptionSensorComponent3DEditor : RayPerceptionSensorComponentBaseEditor
|
|
{
|
|
public override void OnInspectorGUI()
|
|
{
|
|
OnRayPerceptionInspectorGUI(true);
|
|
}
|
|
}
|
|
}
|