CbmDigitizationSource.h

Go to the documentation of this file.

 
#ifndef CBMDIGITIZATIONSOURCE_H
#define CBMDIGITIZATIONSOURCE_H 1
 
#include "CbmDefs.h"
#include "CbmMCInputSet.h"
#include "FairEventHeader.h"
#include "FairLogger.h"
#include "FairMCEventHeader.h"
#include "FairSource.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,
                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;
  }
 
 
  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() {}
 
 
  void SetEventMode(Bool_t choice = kTRUE) { fEventMode = choice; }
 
 
  virtual void SetParUnpackers() {}
 
 
  void SetTimeStart(Double_t time) { fTimeStart = time; }
 
 
private:
  std::vector<CbmMCInputSet*> fInputSets;
  std::map<UInt_t, CbmMCInputSet*> fInputMap;     
  std::map<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;
  Bool_t fEventMode;
  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 */