//
// Copyright © 2019 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include <common/include/CommandHandlerFunctor.hpp>
#include <common/include/Packet.hpp>

#include <vector>

namespace armnn
{

namespace gatordmock
{

struct PacketVersion
{
    uint16_t m_PacketFamily;
    uint16_t m_PacketId;
    uint32_t m_PacketVersion;
};

class StreamMetadataCommandHandler : public arm::pipe::CommandHandlerFunctor
{

public:
    /**
     * @param familyId The family of the packets this handler will service
     * @param packetId The id of packets this handler will process
     * @param version The version of that id
     * @param quietOperation Optional parameter to turn off printouts. This is useful for unit tests
     */
    StreamMetadataCommandHandler(uint32_t familyId,
                                 uint32_t packetId,
                                 uint32_t version,
                                 bool quietOperation = false)
        : CommandHandlerFunctor(familyId, packetId, version)
        , m_QuietOperation(quietOperation)
    {}

    void operator()(const arm::pipe::Packet& packet) override;

private:
    void ParseData(const arm::pipe::Packet& packet);

    uint32_t m_PipeMagic;
    uint32_t m_StreamMetadataVersion;
    uint32_t m_MaxDataLen;
    uint32_t m_Pid;
    uint32_t m_OffsetInfo;
    uint32_t m_OffsetHwVersion;
    uint32_t m_OffsetSwVersion;
    uint32_t m_OffsetProcessName;
    uint32_t m_OffsetPacketVersionTable;

    std::string m_SoftwareInfo;
    std::string m_HardwareVersion;
    std::string m_SoftwareVersion;
    std::string m_ProcessName;

    std::vector<PacketVersion> m_PacketVersionTable;

    bool m_QuietOperation;
};

} // namespace gatordmock

} // namespace armnn