Unity 机器学习代理工具包 (ML-Agents) 是一个开源项目,它使游戏和模拟能够作为训练智能代理的环境。
您最多选择25个主题 主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
 
 
 
 
 

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);
}
}
}