#ifndef ALICEO2_DEVICES_EPNRECEIVER_H_

#define ALICEO2_DEVICES_EPNRECEIVER_H_

#include <string>

#include <unordered_map>

#include <unordered_set>

#include <chrono>

#include <FairMQDevice.h>

namespace AliceO2 {

namespace Devices {

/// Container for (sub-)timeframes

struct TFBuffer

{

FairMQParts parts;

std::chrono::steady_clock::time_point start;

std::chrono::steady_clock::time_point end;

};

/// Receives sub-timeframes from the flpSenders and merges these into full timeframes.

class EPNReceiverDevice : public FairMQDevice

{

public:

EPNReceiverDevice() = default;

virtual ~EPNReceiverDevice() final = default;

virtual void InitTask() final;

/// Prints the contents of the timeframe container

void PrintBuffer(const std::unordered_map<uint16_t, TFBuffer> &buffer) const;

/// Discared incomplete timeframes after \p fBufferTimeoutInMs.

void DiscardIncompleteTimeframes();

protected:

/// Overloads the Run() method of FairMQDevice

virtual void Run();

std::unordered_map<uint16_t, TFBuffer> mTimeframeBuffer; ///< Stores (sub-)timeframes

std::unordered_set<uint16_t> mDiscardedSet; ///< Set containing IDs of dropped timeframes

int mNumFLPs = 0; ///< Number of flpSenders

int mBufferTimeoutInMs = 5000; ///< Time after which incomplete timeframes are dropped

int mTestMode = 0; ///< Run the device in test mode (only syncSampler+flpSender+epnReceiver)

std::string mInChannelName = "";

std::string mOutChannelName = "";

std::string mAckChannelName = "";

};

} // namespace Devices

} // namespace AliceO2

#endif