mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-12 16:21:13 +00:00
76 lines
1.9 KiB
C++
76 lines
1.9 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence version 3 (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
/**
|
|
* FairMQSplitter.cxx
|
|
*
|
|
* @since 2012-12-06
|
|
* @author D. Klein, A. Rybalchenko
|
|
*/
|
|
|
|
#include "FairMQSplitter.h"
|
|
|
|
#include "../FairMQLogger.h"
|
|
#include "../options/FairMQProgOptions.h"
|
|
|
|
using namespace std;
|
|
|
|
FairMQSplitter::FairMQSplitter()
|
|
: fMultipart(1)
|
|
, fNumOutputs(0)
|
|
, fDirection(0)
|
|
, fInChannelName()
|
|
, fOutChannelName()
|
|
{
|
|
}
|
|
|
|
FairMQSplitter::~FairMQSplitter()
|
|
{
|
|
}
|
|
|
|
void FairMQSplitter::InitTask()
|
|
{
|
|
fMultipart = fConfig->GetValue<int>("multipart");
|
|
fInChannelName = fConfig->GetValue<string>("in-channel");
|
|
fOutChannelName = fConfig->GetValue<string>("out-channel");
|
|
fNumOutputs = fChannels.at(fOutChannelName).size();
|
|
fDirection = 0;
|
|
|
|
if (fMultipart)
|
|
{
|
|
OnData(fInChannelName, &FairMQSplitter::HandleMultipartData);
|
|
}
|
|
else
|
|
{
|
|
OnData(fInChannelName, &FairMQSplitter::HandleSingleData);
|
|
}
|
|
}
|
|
|
|
bool FairMQSplitter::HandleSingleData(FairMQMessagePtr& payload, int /*index*/)
|
|
{
|
|
Send(payload, fOutChannelName, fDirection);
|
|
|
|
if (++fDirection >= fNumOutputs)
|
|
{
|
|
fDirection = 0;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool FairMQSplitter::HandleMultipartData(FairMQParts& payload, int /*index*/)
|
|
{
|
|
Send(payload, fOutChannelName, fDirection);
|
|
|
|
if (++fDirection >= fNumOutputs)
|
|
{
|
|
fDirection = 0;
|
|
}
|
|
|
|
return true;
|
|
}
|