LiveDebuggerServerBinary.h

#ifndef GRAIL_LIVE_DEBUGGER_SERVER_BINARY_H
#define GRAIL_LIVE_DEBUGGER_SERVER_BINARY_H
 
#include "LiveDebuggerServer.h"
#include "DataType.h"
#include "flatbuffers/flatbuffer_builder.h"
 
namespace grail
{
namespace live
{
    class LiveDebuggerServerBinary : public LiveDebuggerServer
    {
        enum class DataDecorator : uint8_t
        {
            STATE = 1,
            UTILITY = 2,
            PLANNER = 3,
            SIMGAMES = 4,
        };
 
    public:
        LiveDebuggerServerBinary(const std::string& serverName);
 
        virtual void ProcessSnapshot(data::GrailStateSnapshot& stateSnapshot) override;
        virtual void ProcessSnapshot(data::UtilityReasonerSnapshot& utilitySnapshot) override;
        virtual void ProcessSnapshot(data::PlannerReasonerSnapshot& plannerSnapshot) override;
        virtual void ProcessSnapshot(data::SimulatedGameReasonerSnapshot& simulatedGamesSnapshot) override;
 
    private:
        template <typename SnapshotType>
        void SendSnapshots(SnapshotType snapshot, DataDecorator dataDecorator)
        {
            if(tcpServer.IsAnyConnected())
            {
                builder.Clear();
                auto serializedSnapshot = grail::data::Pack(builder, snapshot);
                builder.Finish(serializedSnapshot);
                auto decoratedBufferSize = builder.GetSize() + 1;
                uint8_t* decoratedBuffer = new uint8_t[decoratedBufferSize];
                decoratedBuffer[0] = static_cast<uint8_t>(dataDecorator);
                memcpy(decoratedBuffer + 1, builder.GetBufferPointer(), builder.GetSize() * sizeof(uint8_t));
                tcpServer.SendToAll(decoratedBuffer, decoratedBufferSize, DataType::BINARY);
                delete[] decoratedBuffer;
            }
        }
 
        flatbuffers::FlatBufferBuilder builder{};
    };
}
}
 
#endif // GRAIL_LIVE_DEBUGGER_SERVER_BINARY_H