BmnOnlineSource.h

Go to the documentation of this file.

#ifndef BMN_ONLINE_SOURCE_H
#define BMN_ONLINE_SOURCE_H
 
#include <FairOnlineSource.h>
#include <TClonesArray.h>
#include <TMessage.h>
#include <TString.h>
#include <TTree.h>
#include <zmq.h>
 
class _CustomTMessage : public TMessage {
   public:
    _CustomTMessage(void* buf, Int_t len) : TMessage(buf, len) {
        ResetBit(kIsOwner);
    }
};
 
class BmnOnlineSource : public FairOnlineSource {
   public:
    BmnOnlineSource(TString endpoint = "tcp://localhost:5555");
    virtual ~BmnOnlineSource();
 
    virtual Bool_t Init();
    virtual Int_t ReadEvent(UInt_t = 0);
    virtual void Close();
 
    BmnOnlineSource(const BmnOnlineSource&) = delete;
    BmnOnlineSource& operator=(const BmnOnlineSource&) = delete;
 
    template <typename T>
    static void DeserializeZmqMessage(zmq_msg_t* zmq_message,
                                      std::unique_ptr<T>& output) {
        _CustomTMessage _message(zmq_msg_data(zmq_message),
                                 zmq_msg_size(zmq_message));
 
        output.reset(
            static_cast<T*>(_message.ReadObjectAny(_message.GetClass())));
    };
 
   protected:
    TString fEndpoint;
 
    std::shared_ptr<void> fZmqContext;
    std::shared_ptr<void> fZmqSubSocket;
 
    std::unordered_map<std::string, std::unique_ptr<TClonesArray>>
        fRegisteredBranches;
 
    void RegisterBranches(std::unique_ptr<TTree>& tree);
    void UploadDataToRegisteredBranches(std::unique_ptr<TTree>& tree);
 
   private:
    Bool_t fAreBranchesRegistered;
 
    ClassDef(BmnOnlineSource, 0)
};
 
#endif  // BMN_ONLINE_SOURCE_H