// Copyright CERN and copyright holders of ALICE O2. This software is

// distributed under the terms of the GNU General Public License v3 (GPL

// Version 3), copied verbatim in the file "COPYING".

//

// See http://alice-o2.web.cern.ch/license for full licensing information.

//

// In applying this license CERN does not waive the privileges and immunities

// granted to it by virtue of its status as an Intergovernmental Organization

// or submit itself to any jurisdiction.

#include <cstring>

#include "DataFlow/FakeTimeframeGeneratorDevice.h"

#include "DataFlow/FakeTimeframeBuilder.h"

#include "DataFlow/TimeframeParser.h"

#include "Headers/SubframeMetadata.h"

#include "Headers/DataHeader.h"

#include <options/FairMQProgOptions.h>

#include <vector>

using DataHeader = o2::header::DataHeader;

namespace

{

struct OneShotReadBuf : public std::streambuf {

OneShotReadBuf(char* s, std::size_t n)

{

setg(s, s, s + n);

}

};

} // namespace

namespace o2

{

namespace data_flow

{

FakeTimeframeGeneratorDevice::FakeTimeframeGeneratorDevice()

: O2Device{}, mOutChannelName{}, mMaxTimeframes{}, mTimeframeCount{0}

{

}

void FakeTimeframeGeneratorDevice::InitTask()

{

mOutChannelName = GetConfig()->GetValue<std::string>(OptionKeyOutputChannelName);

mMaxTimeframes = GetConfig()->GetValue<size_t>(OptionKeyMaxTimeframes);

}

bool FakeTimeframeGeneratorDevice::ConditionalRun()

{

auto addPartFn = [this](FairMQParts& parts, char* buffer, size_t size) {

parts.AddPart(this->NewMessage(

buffer,

size,

[](void* data, void* hint) { delete[](char*) data; },

nullptr));

};

auto sendFn = [this](FairMQParts& parts) { this->Send(parts, this->mOutChannelName); };

auto zeroFiller = [](char* b, size_t s) { memset(b, 0, s); };

std::vector<o2::data_flow::FakeTimeframeSpec> specs = {

{.origin = "TPC",

.dataDescription = "CLUSTERS",

.bufferFiller = zeroFiller,

.bufferSize = 1000},

{.origin = "ITS",

.dataDescription = "CLUSTERS",

.bufferFiller = zeroFiller,

.bufferSize = 500}};

try {

size_t totalSize;

auto buffer = fakeTimeframeGenerator(specs, totalSize);

OneShotReadBuf osrb(buffer.get(), totalSize);

std::istream s(&osrb);

streamTimeframe(s,

addPartFn,

sendFn);

} catch (std::runtime_error& e) {

LOG(ERROR) << e.what() << "\n";

}

mTimeframeCount++;

if (mTimeframeCount < mMaxTimeframes) {

return true;

}

return false;

}

} // namespace data_flow

} // namespace o2