ppo-SnowballTarget / com.unity.ml-agents /Runtime /Sensors /Reflection /QuaternionReflectionSensor.cs
ashish2244's picture
Initial commit (part 5)
d353048 verified
Raw
History Blame Contribute Delete
624 Bytes
namespace Unity.MLAgents.Sensors.Reflection
{
/// <summary>
/// Sensor that wraps a quaternion field or property of an object, and returns
/// that as an observation.
/// </summary>
internal class QuaternionReflectionSensor : ReflectionSensorBase
{
public QuaternionReflectionSensor(ReflectionSensorInfo reflectionSensorInfo)
: base(reflectionSensorInfo, 4)
{ }
internal override void WriteReflectedField(ObservationWriter writer)
{
var quatVal = (UnityEngine.Quaternion)GetReflectedValue();
writer.Add(quatVal);
}
}
}