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