mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +00:00
59 lines
1.7 KiB
C++
59 lines
1.7 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014-2021 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" *
|
|
********************************************************************************/
|
|
|
|
#ifndef FAIR_MQ_SPLITTER_H
|
|
#define FAIR_MQ_SPLITTER_H
|
|
|
|
#include <fairmq/Device.h>
|
|
|
|
#include <string>
|
|
|
|
namespace fair::mq
|
|
{
|
|
|
|
class Splitter : public Device
|
|
{
|
|
protected:
|
|
bool fMultipart = true;
|
|
int fNumOutputs = 0;
|
|
int fDirection = 0;
|
|
std::string fInChannelName;
|
|
std::string fOutChannelName;
|
|
|
|
void InitTask() override
|
|
{
|
|
fMultipart = fConfig->GetProperty<bool>("multipart");
|
|
fInChannelName = fConfig->GetProperty<std::string>("in-channel");
|
|
fOutChannelName = fConfig->GetProperty<std::string>("out-channel");
|
|
fNumOutputs = GetNumSubChannels(fOutChannelName);
|
|
fDirection = 0;
|
|
|
|
if (fMultipart) {
|
|
OnData(fInChannelName, &Splitter::HandleData<FairMQParts>);
|
|
} else {
|
|
OnData(fInChannelName, &Splitter::HandleData<FairMQMessagePtr>);
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
bool HandleData(T& payload, int)
|
|
{
|
|
Send(payload, fOutChannelName, fDirection);
|
|
|
|
if (++fDirection >= fNumOutputs) {
|
|
fDirection = 0;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
};
|
|
|
|
} // namespace fair::mq
|
|
|
|
#endif /* FAIR_MQ_SPLITTER_H */
|