您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
43 行
1.3 KiB
43 行
1.3 KiB
using UnityEngine;
|
|
using UnityEditor;
|
|
using MLAgents.Sensors;
|
|
namespace MLAgents.Editor
|
|
{
|
|
[CustomEditor(typeof(RenderTextureSensorComponent))]
|
|
[CanEditMultipleObjects]
|
|
internal class RenderTextureSensorComponentEditor : UnityEditor.Editor
|
|
{
|
|
public override void OnInspectorGUI()
|
|
{
|
|
var so = serializedObject;
|
|
so.Update();
|
|
|
|
// Drawing the RenderTextureComponent
|
|
EditorGUI.BeginChangeCheck();
|
|
|
|
EditorGUI.BeginDisabledGroup(Application.isPlaying);
|
|
{
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_RenderTexture"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_SensorName"), true);
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_Grayscale"), true);
|
|
}
|
|
EditorGUI.EndDisabledGroup();
|
|
|
|
EditorGUILayout.PropertyField(so.FindProperty("m_Compression"), true);
|
|
|
|
var requireSensorUpdate = EditorGUI.EndChangeCheck();
|
|
so.ApplyModifiedProperties();
|
|
|
|
if (requireSensorUpdate)
|
|
{
|
|
UpdateSensor();
|
|
}
|
|
}
|
|
|
|
void UpdateSensor()
|
|
{
|
|
var sensorComponent = serializedObject.targetObject as RenderTextureSensorComponent;
|
|
sensorComponent?.UpdateSensor();
|
|
}
|
|
}
|
|
}
|