CbmDeviceMcbmEventSink.h

Go to the documentation of this file.

/* Copyright (C) 2020-2021 Facility for Antiproton and Ion Research in Europe, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Pierre-Alain Loizeau [committer] */
 
#ifndef CBMDEVICEMCBMEVTSINK_H_
#define CBMDEVICEMCBMEVTSINK_H_
 
#include "CbmEvent.h"
#include "CbmMqTMessage.h"
#include "CbmMuchBeamTimeDigi.h"
#include "CbmPsdDigi.h"
#include "CbmRichDigi.h"
#include "CbmStsDigi.h"
#include "CbmTofDigi.h"
#include "CbmTrdDigi.h"
 
#include "TimesliceMetaData.h"
 
#include "FairMQDevice.h"
 
#include "Rtypes.h"
#include "TClonesArray.h"
#include "TObjArray.h"
 
#include <chrono>
#include <map>
#include <vector>
 
class TFile;
class TList;
class TClonesArray;
//class TimesliceMetaData;
class FairRunOnline;
class FairRootManager;
 
class CbmUnpackedTimeslice {
public:
  CbmUnpackedTimeslice(FairMQParts& parts);
  ~CbmUnpackedTimeslice();
 
  TimesliceMetaData fTsMetaData;
  std::vector<CbmTofDigi> fvDigiBmon;
  std::vector<CbmStsDigi> fvDigiSts;
  std::vector<CbmMuchBeamTimeDigi> fvDigiMuch;
  std::vector<CbmTrdDigi> fvDigiTrd;
  std::vector<CbmTofDigi> fvDigiTof;
  std::vector<CbmRichDigi> fvDigiRich;
  std::vector<CbmPsdDigi> fvDigiPsd;
  TClonesArray fEventsArray;
};
 
class CbmDeviceMcbmEventSink : public FairMQDevice {
public:
  CbmDeviceMcbmEventSink();
  virtual ~CbmDeviceMcbmEventSink();
 
protected:
  virtual void InitTask();
  bool HandleMissTsData(FairMQMessagePtr&, int);
  bool HandleData(FairMQParts&, int);
  bool HandleCommand(FairMQMessagePtr&, int);
 
private:
 
  Bool_t fbFillHistos = kFALSE;  
  Bool_t fbFinishDone = kFALSE;  
 
  std::string fsOutputFileName = "mcbm_digis_events.root";
  std::string fsChannelNameMissedTs     = "missedts";
  std::string fsChannelNameDataInput    = "events";
  std::string fsChannelNameCommands     = "commands";
  std::string fsChannelNameHistosInput  = "histogram-in";
  std::string fsChannelNameHistosConfig = "histo-conf";
  std::string fsChannelNameCanvasConfig = "canvas-conf";
  uint32_t fuPublishFreqTs  = 100;
  double_t fdMinPublishTime = 0.5;
  double_t fdMaxPublishTime = 5.0;
 
  std::vector<std::string> fsAllowedChannels = {fsChannelNameDataInput};
 
  //      TList* fParCList = nullptr;
  //      Bool_t InitParameters( TList* fParCList );
 
  uint64_t fuPrevTsIndex                                 = 0;
  uint64_t fulNumMessages                                = 0;
  uint64_t fulTsCounter                                  = 0;
  uint64_t fulMissedTsCounter                            = 0;
  std::chrono::system_clock::time_point fLastPublishTime = std::chrono::system_clock::now();
 
  bool fbReceivedEof      = false;
  uint64_t fuLastTsIndex  = 0;
  uint64_t fuTotalTsCount = 0;
 
  //      FairEventHeader*        fEvtHeader;
  TClonesArray* fTimeSliceMetaDataArray = nullptr;  
  TimesliceMetaData* fTsMetaData        = nullptr;
  std::vector<CbmTofDigi>* fvDigiBmon          = nullptr;
  std::vector<CbmStsDigi>* fvDigiSts           = nullptr;
  std::vector<CbmMuchBeamTimeDigi>* fvDigiMuch = nullptr;
  std::vector<CbmTrdDigi>* fvDigiTrd           = nullptr;
  std::vector<CbmTofDigi>* fvDigiTof           = nullptr;
  std::vector<CbmRichDigi>* fvDigiRich         = nullptr;
  std::vector<CbmPsdDigi>* fvDigiPsd           = nullptr;
  TClonesArray* fEventsArray = nullptr;  
  //      std::vector< CbmEvent * > &        fEventVector;    //! vector with all created events
 
  std::vector<uint64_t> fvulMissedTsIndices = {};
  std::map<uint64_t, CbmUnpackedTimeslice> fmFullTsStorage = {};
 
  FairRunOnline* fpRun           = nullptr;
  FairRootManager* fpFairRootMgr = nullptr;
 
  TObjArray fArrayHisto = {};
  std::vector<std::pair<std::string, std::string>> fvpsHistosFolder = {};
  std::vector<std::pair<std::string, std::string>> fvpsCanvasConfig = {};
 
  bool IsChannelNameAllowed(std::string channelName);
  //      Bool_t InitContainers();
  void CheckTsQueues();
  void PrepareTreeEntry(CbmUnpackedTimeslice unpTs);
  void DumpTreeEntry();
  bool SendHistograms();
  void Finish();
};
 
#endif /* CBMDEVICEMCBMEVTSINK_H_ */