mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 00:31:14 +00:00
445ef356d1
- Add a generic merger. add SYSTEM_INCLUDE_DIRECTORIES in project_template CMakeLists.txt so that boost include dirs are found.
110 lines
3.5 KiB
C++
110 lines
3.5 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" *
|
|
********************************************************************************/
|
|
/*
|
|
* File: GenericSampler.h
|
|
* Author: winckler
|
|
*
|
|
* Created on November 24, 2014, 3:30 PM
|
|
*/
|
|
|
|
|
|
#ifndef GENERICSAMPLER_H
|
|
#define GENERICSAMPLER_H
|
|
|
|
#include <vector>
|
|
#include <iostream>
|
|
#include <stdint.h>
|
|
|
|
#include <boost/thread.hpp>
|
|
#include <boost/bind.hpp>
|
|
#include <boost/timer/timer.hpp>
|
|
|
|
#include "FairMQDevice.h"
|
|
#include "FairMQLogger.h"
|
|
|
|
/* GENERIC SAMPLER (data source) MQ-DEVICE */
|
|
/*********************************************************************
|
|
* -------------- NOTES -----------------------
|
|
* All policies must have a default constructor
|
|
* Function to define in (parent) policy classes :
|
|
*
|
|
* -------- INPUT POLICY (SAMPLER POLICY) --------
|
|
* SamplerPolicy::InitSampler()
|
|
* int64_t SamplerPolicy::GetNumberOfEvent()
|
|
* CONTAINER_TYPE SamplerPolicy::GetDataBranch(int64_t eventNr)
|
|
* SamplerPolicy::SetFileProperties(Args&... args)
|
|
*
|
|
* -------- OUTPUT POLICY --------
|
|
* OutputPolicy::SerializeMsg(CONTAINER_TYPE)
|
|
* OutputPolicy::SetMessage(FairMQMessage* msg)
|
|
*
|
|
**********************************************************************/
|
|
|
|
template <typename SamplerPolicy,
|
|
typename OutputPolicy>
|
|
class GenericSampler: public FairMQDevice,
|
|
public SamplerPolicy,
|
|
public OutputPolicy
|
|
{
|
|
public:
|
|
|
|
enum
|
|
{
|
|
InputFile = FairMQDevice::Last,
|
|
Branch,
|
|
ParFile,
|
|
EventRate
|
|
};
|
|
|
|
GenericSampler();
|
|
virtual ~GenericSampler();
|
|
virtual void SetTransport(FairMQTransportFactory* factory);
|
|
void ResetEventCounter();
|
|
virtual void ListenToCommands();
|
|
|
|
template <typename... Args>
|
|
void SetFileProperties(Args&... args)
|
|
{
|
|
SamplerPolicy::SetFileProperties(args...);
|
|
}
|
|
|
|
virtual void SetProperty(const int key, const std::string& value, const int slot = 0);
|
|
virtual std::string GetProperty(const int key, const std::string& default_ = "", const int slot = 0);
|
|
virtual void SetProperty(const int key, const int value, const int slot = 0);
|
|
virtual int GetProperty(const int key, const int default_ = 0, const int slot = 0);
|
|
|
|
/**
|
|
* Sends the currently available output of the Sampler Task as part of a multipart message
|
|
* and reinitializes the message to be filled with the next part.
|
|
* This method can be given as a callback to the SamplerTask.
|
|
* The final message part must be sent with normal Send method.
|
|
*/
|
|
// temporary disabled
|
|
//void SendPart();
|
|
|
|
void SetContinuous(bool flag) { fContinuous = flag; }
|
|
|
|
protected:
|
|
virtual void Init();
|
|
virtual void Run();
|
|
|
|
protected:
|
|
std::string fInputFile; // Filename of a root file containing the simulated digis.
|
|
std::string fParFile;
|
|
std::string fBranch; // The name of the sub-detector branch to stream the digis from.
|
|
int64_t fNumEvents;
|
|
int fEventRate;
|
|
int fEventCounter;
|
|
bool fContinuous;
|
|
};
|
|
|
|
#include "GenericSampler.tpl"
|
|
|
|
#endif /* GENERICSAMPLER_H */
|
|
|