/********************************************************************************
 *    Copyright (C) 2020 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

#include "Header.h"

#include <fairmq/Device.h>
#include <fairmq/runDevice.h>

#include <string>

using namespace std;
using namespace example_n_m;
namespace bpo = boost::program_options;

class Sender : public FairMQDevice
{
  public:
    Sender() = default;

    void InitTask() override
    {
        fIndex = GetConfig()->GetProperty<int>("sender-index");
        fSubtimeframeSize = GetConfig()->GetProperty<int>("subtimeframe-size");
        fNumReceivers = GetConfig()->GetProperty<int>("num-receivers");
    }

    void Run() override
    {
        FairMQChannel& dataInChannel = fChannels.at("sync").at(0);

        while (!NewStatePending()) {
            Header h;
            FairMQMessagePtr id(NewMessage());
            if (dataInChannel.Receive(id) > 0) {
                h.id = *(static_cast<uint16_t*>(id->GetData()));
                h.senderIndex = fIndex;
            } else {
                continue;
            }

            FairMQParts parts;
            parts.AddPart(NewSimpleMessage(h));
            parts.AddPart(NewMessage(fSubtimeframeSize));

            uint64_t currentDataId = h.id;
            int direction = currentDataId % fNumReceivers;

            if (Send(parts, "data", direction, 0) < 0) {
                LOG(debug) << "Failed to queue Subtimeframe #" << currentDataId << " to Receiver[" << direction << "]";
            }
        }
    }

  private:
    int fNumReceivers = 0;
    unsigned int fIndex = 0;
    int fSubtimeframeSize = 10000;
};

void addCustomOptions(bpo::options_description& options)
{
    options.add_options()
        ("sender-index", bpo::value<int>()->default_value(0), "Sender Index")
        ("subtimeframe-size", bpo::value<int>()->default_value(1000), "Subtimeframe size in bytes")
        ("num-receivers", bpo::value<int>()->required(), "Number of EPNs");
}
std::unique_ptr<fair::mq::Device> getDevice(FairMQProgOptions& /* config */)
{
    return std::make_unique<Sender>();
}