CbmDigitizationSource.h

Go to the documentation of this file.

/* Copyright (C) 2018-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer] */
 
#ifndef CBMDIGITIZATIONSOURCE_H
#define CBMDIGITIZATIONSOURCE_H 1
 
#include "CbmDefs.h"
#include "CbmMCInputSet.h"
 
#include "FairEventHeader.h"
#include "FairMCEventHeader.h"
#include "FairSource.h"
#include <Logger.h>
 
#include "TObject.h"
#include "TString.h"
 
#include <map>
#include <set>
 
class FairEventHeader;
 
class CbmDigitizationSource : public FairSource {
 
public:
  CbmDigitizationSource();
 
 
  virtual ~CbmDigitizationSource();
 
 
  virtual Bool_t ActivateObject(TObject** object, const char* branchName);
 
 
  void AddInput(UInt_t inputId, TChain* chain, cbm::sim::TimeDist dist, Double_t rate,
                ECbmTreeAccess mode = ECbmTreeAccess::kRegular);
 
 
  virtual Int_t CheckMaxEventNo(Int_t lastEntry = 0);
 
 
  virtual void Close() {}
 
 
  void EmbedInput(UInt_t inputId, TChain* chain, UInt_t targetInputId, ECbmTreeAccess mode = ECbmTreeAccess::kRegular);
 
 
  virtual void FillEventHeader(FairEventHeader* event);
 
 
  const std::set<TString>& GetBranchList() const { return fBranches; }
 
 
  CbmMCInput* GetFirstInput();
 
 
  virtual Source_Type GetSourceType()
  {
    LOG(fatal) << "GetSourceTpye";
 
    return kFILE;
  }
 
  Double_t GetStartTime() { return fTimeStart; }
 
 
  virtual Bool_t Init();
 
 
  virtual Bool_t InitUnpackers() { return kTRUE; }
 
 
  virtual Int_t ReadEvent(UInt_t event = 0);
 
 
  virtual Bool_t ReInitUnpackers() { return kTRUE; }
 
 
  virtual void Reset() {}
 
 
  virtual Bool_t SpecifyRunId()
  {
    if (ReadEvent(0)) return true;
 
    return false;
  }
 
  void SetMode(cbm::sim::Mode mode) { fMode = mode; }
 
 
  virtual void SetParUnpackers() {}
 
 
  void SetTimeStart(Double_t time) { fTimeStart = time; }
 
 
private:
  std::vector<CbmMCInputSet*> fInputSets;
  std::map<UInt_t, CbmMCInputSet*> fInputMap;          
  std::multimap<Double_t, CbmMCInputSet*> fNextEvent;  
  FairMCEventHeader* fMCEventHeader;
  TObjArray* fListOfFolders;
  std::set<TString> fBranches;  // List of branches names
  Double_t fTimeStart;          // Time offset first first event
  Double_t fCurrentTime;
  Int_t fCurrentEntryId;
  Int_t fCurrentInputId;
  Int_t fCurrentRunId;
  Bool_t fFirstCall;
  cbm::sim::Mode fMode;
  CbmMCInputSet* fCurrentInputSet;
  Bool_t fSwitchInputSet;  // Flag to switch the input set at next ReadEvent
 
 
  Bool_t CheckBranchList(CbmMCInputSet* input);
 
 
  Int_t ReadEventByEvent(UInt_t event);
 
 
  void ReadRunId();
 
 
  ClassDef(CbmDigitizationSource, 1);
};
 
#endif /* CBMDIGITIZATIONSOURCE_H */