File size: 615 Bytes
05c9ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
namespace Unity.MLAgents.Sensors.Reflection
{
/// <summary>
/// Sensor that wraps a boolean field or property of an object, and returns
/// that as an observation.
/// </summary>
internal class BoolReflectionSensor : ReflectionSensorBase
{
public BoolReflectionSensor(ReflectionSensorInfo reflectionSensorInfo)
: base(reflectionSensorInfo, 1)
{ }
internal override void WriteReflectedField(ObservationWriter writer)
{
var boolVal = (System.Boolean)GetReflectedValue();
writer[0] = boolVal ? 1.0f : 0.0f;
}
}
}
|