CbmDigitization.h

Go to the documentation of this file.

 
#ifndef CBMDIGITIZATION_H
#define CBMDIGITIZATION_H 1
 
#include "CbmDaq.h"
#include "CbmDefs.h"
#include "CbmDigitizationSource.h"
#include "CbmDigitizeInfo.h"
#include "FairLogger.h"
#include "TList.h"
#include "TNamed.h"
#include "TString.h"
#include "TSystem.h"
#include <map>
#include <vector>
 
class TGeoManager;
class CbmDigitizeBase;
 
 
class CbmDigitization : public TNamed {
 
public:
  CbmDigitization();
 
 
  virtual ~CbmDigitization();
 
 
  void AddInput(UInt_t inputId,
                TString fileName,
                Double_t eventRate  = -1.,
                ECbmTreeAccess mode = ECbmTreeAccess::kRegular);
 
 
  void AddInput(TString fileName,
                Double_t eventRate  = -1.,
                ECbmTreeAccess mode = ECbmTreeAccess::kRegular) {
    AddInput(0, fileName, eventRate, mode);
  }
 
  Bool_t AddParameterAsciiFile(TString fileName);
 
 
  void Deactivate(ECbmModuleId system);
 
 
  void DeactivateAllBut(ECbmModuleId system);
 
 
  void EmbedInput(UInt_t inputId,
                  TString fileName,
                  UInt_t targetInputId,
                  ECbmTreeAccess mode = ECbmTreeAccess::kRegular);
 
 
  void GenerateRunInfo(Bool_t choice = kTRUE) { fGenerateRunInfo = choice; }
 
  void DefaultInit();
 
  void Run() { Run(-1, -1); }
 
 
  void Run(Int_t nEvents) { Run(0, nEvents); }
 
 
  void Run(Int_t event1, Int_t event2);
 
 
  void SetCreateMatches(Bool_t choice = kTRUE) { fCreateMatches = choice; }
 
 
  void SetDigitizer(ECbmModuleId system,
                    CbmDigitizeBase* digitizer,
                    TString branch    = "",
                    Bool_t persistent = kTRUE);
 
 
  void SetEventMode(Bool_t choice = kTRUE) { fEventMode = choice; }
 
 
  void SetMonitorFile(const char* fileName) { fMoniFile = fileName; }
 
 
  void SetOutputFile(TString fileName, Bool_t overwrite = kFALSE);
 
 
  void SetParameterRootFile(TString fileName);
 
 
  void SetProduceNoise(Bool_t choice = kTRUE) { fProduceNoise = choice; }
 
 
  void SetStartTime(Double_t time) { fSource->SetTimeStart(time); }
 
 
  void SetTimeSliceLength(Double_t length) { fTimeSliceLength = length; }
 
 
  void StoreAllTimeSlices(Bool_t choice = kTRUE) {
    fDaq->StoreAllTimeSlices(choice);
  }
 
  CbmDigitizeBase* GetDigitizer(ECbmModuleId system);
 
 
private:
  Bool_t fIsInit;
  Bool_t fEventMode;
  Double_t fTimeSliceLength;
  Bool_t fProduceNoise;
  Bool_t fCreateMatches;
  std::map<ECbmModuleId, CbmDigitizeInfo*> fDigitizers;  
  CbmDaq* fDaq;
  CbmDigitizationSource* fSource;  
  TString fOutFile;                
  TString fParRootFile;            
  TList fParAsciiFiles;            
  TString fMoniFile;               
  Bool_t fOverwriteOutput;
  Bool_t fGenerateRunInfo;
  Int_t fRun;
 
 
  CbmDigitization(const CbmDigitization&) = delete;
 
 
  CbmDigitization operator=(const CbmDigitization&) = delete;
 
 
  Int_t CheckInput();
 
 
  Int_t CreateDefaultDigitizers();
 
 
  TString GetGeoTag(ECbmModuleId system, TGeoManager* geo);
 
 
  void SetDefaultBranches();
 
 
  ClassDef(CbmDigitization, 3);
};
 
#endif /* CBMDIGITIZATION_H */