File size: 2,131 Bytes
05c9ac2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | using System;
using Unity.MLAgents.Analytics;
using Unity.MLAgents.CommunicatorObjects;
namespace Unity.MLAgents.SideChannels
{
/// <summary>
/// Side Channel implementation for recording which training features are being used.
/// </summary>
internal class TrainingAnalyticsSideChannel : SideChannel
{
const string k_TrainingAnalyticsConfigId = "b664a4a9-d86f-5a5f-95cb-e8353a7e8356";
/// <summary>
/// Initializes the side channel. The constructor is internal because only one instance is
/// supported at a time, and is created by the Academy.
/// </summary>
internal TrainingAnalyticsSideChannel()
{
ChannelId = new Guid(k_TrainingAnalyticsConfigId);
}
/// <inheritdoc/>
protected override void OnMessageReceived(IncomingMessage msg)
{
Google.Protobuf.WellKnownTypes.Any anyMessage = null;
try
{
anyMessage = Google.Protobuf.WellKnownTypes.Any.Parser.ParseFrom(msg.GetRawBytes());
}
catch (Google.Protobuf.InvalidProtocolBufferException)
{
// Bad message, nothing we can do about it, so just ignore.
return;
}
if (anyMessage.Is(TrainingEnvironmentInitialized.Descriptor))
{
var envInitProto = anyMessage.Unpack<TrainingEnvironmentInitialized>();
var envInitEvent = envInitProto.ToTrainingEnvironmentInitializedEvent();
TrainingAnalytics.TrainingEnvironmentInitialized(envInitEvent);
}
else if (anyMessage.Is(TrainingBehaviorInitialized.Descriptor))
{
var behaviorInitProto = anyMessage.Unpack<TrainingBehaviorInitialized>();
var behaviorTrainingEvent = behaviorInitProto.ToTrainingBehaviorInitializedEvent();
TrainingAnalytics.TrainingBehaviorInitialized(behaviorTrainingEvent);
}
// Don't do anything for unknown types, since the user probably can't do anything about it.
}
}
}
|