FairMQPrimaryGeneratorDevice.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2017 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"                       *
 ********************************************************************************/

#ifndef FAIRMQPRIMARYGENERATORDEVICE_H_
#define FAIRMQPRIMARYGENERATORDEVICE_H_

#include <FairMQDevice.h>
#include <cstdint>
#include <string>
#include <thread>

class FairMCEventHeader;
class FairPrimaryGenerator;
class TObject;
class FairStack;

class FairMQPrimaryGeneratorDevice : public FairMQDevice
{
  public:
    FairMQPrimaryGeneratorDevice();
    virtual ~FairMQPrimaryGeneratorDevice();

    virtual void SendPrimaries();

    void SetNofEvents(int64_t nofev) { fNofEvents = nofev; };
    void SetGenerator(FairPrimaryGenerator* primGen) { fPrimaryGenerator = primGen; };

    void SetChunkSize(int64_t ti) { fChunkSize = ti; };

    void RunInPushMode(bool tb = true) { fRunConditional = tb; };
    void RunInRepMode(bool tb = true) { fRunConditional = !tb; };

    void ListenForAcks();

    void SetGeneratorChannelName(std::string tstr) { fGeneratorChannelName = tstr; }
    void SetAckChannelName(std::string tstr) { fAckChannelName = tstr; }

  protected:
    bool Reply(FairMQMessagePtr&, int);
    virtual void InitTask();
    virtual void PreRun();
    virtual void PostRun();
    virtual bool ConditionalRun();

  private:
    std::string fGeneratorChannelName;
    std::string fAckChannelName;

    bool fRunConditional;   // if true run ConditionalRun, if false run Reply

    FairPrimaryGenerator* fPrimaryGenerator;
    FairMCEventHeader* fMCEventHeader;
    FairStack* fStack;
    int64_t fNofEvents;
    int64_t fEventCounter;

    int64_t fChunkSize;      // to set the maximal number of primaries sent in one bunch
    int64_t fChunkPointer;   // to set the first primary to be sent

    bool GenerateAndSendData();
    void SendObject(TObject* obj, std::string chan);

    FairMQPrimaryGeneratorDevice(const FairMQPrimaryGeneratorDevice&);
    FairMQPrimaryGeneratorDevice& operator=(const FairMQPrimaryGeneratorDevice&);

    std::thread fAckListener;
};

#endif /* FAIRMQPRIMARYGENERATORDEVICE_H_ */