Mirror/FairMQ
1
0
Fork 0
mirror of https://github.com/FairRootGroup/FairMQ.git synced 2025-10-15 09:31:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

add serialization API

Browse Source
This commit is contained in:
winckler
2016-03-24 15:59:33 +01:00
parent 9a340fb7f6
commit 4c50409af5
5 changed files with 218 additions and 281 deletions
Download Patch File Download Diff File Expand all files Collapse all files

144
fairmq/devices/GenericProcessor.h
View File

@@ -17,107 +17,55 @@
#include "FairMQDevice.h"
/*********************************************************************
* -------------- NOTES -----------------------
* All policies must have a default constructor
* Function to define in (parent) policy classes :
*
* -------- INPUT POLICY --------
* deserialization_type::InitContainer(...)
* CONTAINER_TYPE deserialization_type::DeserializeMsg(FairMQMessage* msg)
* deserialization_type::InitContainer(...) // if GenericProcessor::InitInputContainer(...) is used
*
*
* -------- OUTPUT POLICY --------
* serialization_type::SerializeMsg(CONTAINER_TYPE)
* serialization_type::SetMessage(FairMQMessage* msg)
* serialization_type::InitContainer(...) // if GenericProcessor::InitOutputContainer(...) is used
*
* -------- TASK POLICY --------
* CONTAINER_TYPE proc_task_type::GetOutputData()
* proc_task_type::ExecuteTask(CONTAINER_TYPE container)
* proc_task_type::InitTask(...) // if GenericProcessor::InitTask(...) is used
*
**********************************************************************/
template <typename T, typename U, typename V>
class GenericProcessor : public FairMQDevice, public T, public U, public V
template < typename T/*=deserialization type*/,
typename U/*=serialization type*/,
typename V/*=task type*///,
//typename W/*=input creator type*/,
//typename X/*=output creator type*/
>
class GenericProcessor : public FairMQDevice, public T, public U,
public V//,
//public W,
//public X
{
protected:
typedef T deserialization_type;
typedef U serialization_type;
typedef V proc_task_type;
public:
public:
typedef T input_policy;
typedef U output_policy;
typedef V task_type;
//typedef W input_creator_type;
//typedef X output_creator_type;
GenericProcessor()
: deserialization_type()
, serialization_type()
, proc_task_type()
: FairMQDevice(), T(), U()
, task_type()
//, input_creator_type()
//, output_creator_type()
{}
virtual ~GenericProcessor()
{}
// the four following methods ensure
// that the correct policy method is called
template <typename... Args>
void SetTransport(Args... args)
template<typename... Args>
void InitInputData(Args&&... args)
{
FairMQDevice::SetTransport(std::forward<Args>(args)...);
input_policy::Create(std::forward<Args>(args)...);
}
template <typename... Args>
void InitTask(Args... args)
template<typename... Args>
void InitOutputData(Args&&... args)
{
proc_task_type::InitTask(std::forward<Args>(args)...);
output_policy::Create(std::forward<Args>(args)...);
}
template <typename... Args>
void InitInputContainer(Args... args)
{
deserialization_type::InitContainer(std::forward<Args>(args)...);
}
template <typename... Args>
void InitOutputContainer(Args... args)
{
serialization_type::InitContainer(std::forward<Args>(args)...);
}
/*
*
// *********************** TODO: implement multipart features
void SendPart()
{
fChannels["data-out"].at(0).Send(serialization_type::SerializeMsg(proc_task_type::GetData()), "snd-more");
serialization_type::CloseMessage();
}
// void SendPart();
// bool ReceivePart();
bool ReceivePart()
{
if (fChannels["data-in"].at(0).ExpectsAnotherPart())
{
deserialization_type::CloseMessage();
// fProcessorTask->GetPayload()->CloseMessage();
fProcessorTask->SetPayload(fTransportFactory->CreateMessage());
return fChannels["data-in"].at(0).Receive(fProcessorTask->GetPayload());
}
else
{
return false;
}
}
*/
protected:
using input_policy::fInput;
using output_policy::fOutput;
virtual void InitTask()
{
// TODO: implement multipart features
// fProcessorTask->InitTask();
// fProcessorTask->SetSendPart(boost::bind(&FairMQProcessor::SendPart, this));
// fProcessorTask->SetReceivePart(boost::bind(&FairMQProcessor::ReceivePart, this));
}
virtual void Run()
@@ -125,34 +73,22 @@ class GenericProcessor : public FairMQDevice, public T, public U, public V
int receivedMsgs = 0;
int sentMsgs = 0;
const FairMQChannel& inputChannel = fChannels["data-in"].at(0);
const FairMQChannel& outputChannel = fChannels["data-out"].at(0);
while (CheckCurrentState(RUNNING))
{
std::unique_ptr<FairMQMessage> msg(fTransportFactory->CreateMessage());
if (inputChannel.Receive(msg) > 0)
if (Receive<T>(fInput, "data-in") > 0)
{
receivedMsgs++;
// deserialization_type::DeserializeMsg(msg) --> deserialize data of msg and fill output container
// proc_task_type::ExecuteTask( ... ) --> process output container
proc_task_type::ExecuteTask(deserialization_type::DeserializeMsg(msg.get()));
// serialization_type::fMessage point to msg
serialization_type::SetMessage(msg.get());
// proc_task_type::GetOutputData() --> Get processed output container
// serialization_type::message(...) --> Serialize output container and fill fMessage
outputChannel.Send(serialization_type::SerializeMsg(proc_task_type::GetOutputData()));
task_type::Exec(fInput,fOutput);
Send<U>(fOutput, "data-out");
sentMsgs++;
}
}
MQLOG(INFO) << "Received " << receivedMsgs << " and sent " << sentMsgs << " messages!";
LOG(INFO) << "Received " << receivedMsgs << " and sent " << sentMsgs << " messages!";
}
};
#endif /* GENERICPROCESSOR_H */