CbmDigitization.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 CBMDIGITIZATION_H
#define CBMDIGITIZATION_H 1
 
#include "CbmDaq.h"
#include "CbmDefs.h"
#include "CbmDigitizationSource.h"
#include "CbmDigitizeInfo.h"
#include "Defs.h"
#include "TList.h"
#include "TNamed.h"
#include "TString.h"
#include "TSystem.h"
 
#include <Logger.h>
 
#include <map>
#include <vector>
 
class TGeoManager;
class CbmDigitizeBase;
 
 
class CbmDigitization : public TNamed {
 
 public:
  CbmDigitization();
 

  virtual ~CbmDigitization();
 

  void AddInput(UInt_t inputId, TString fileName, cbm::sim::TimeDist dist = cbm::sim::TimeDist::Poisson,
                Double_t eventRate = 0., ECbmTreeAccess mode = ECbmTreeAccess::kRegular);
 

  void AddInput(TString fileName, cbm::sim::TimeDist dist = cbm::sim::TimeDist::Poisson, Double_t eventRate = 0.,
                ECbmTreeAccess mode = ECbmTreeAccess::kRegular)
  {
    AddInput(0, fileName, dist, 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 SetMode(cbm::sim::Mode mode) { fMode = mode; }
 

  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); }

  // TODO: This does not work, since the defaulot digitizers are not instantiated
  // at instation of this class, but in the Exec() method.
  CbmDigitizeBase* GetDigitizer(ECbmModuleId system);
 
 
 private:
  Bool_t fIsInit            = kFALSE;
  cbm::sim::Mode fMode      = cbm::sim::Mode::Timebased;
  Double_t fTimeSliceLength = -1.;
  Bool_t fProduceNoise      = kTRUE;
  ;
  Bool_t fCreateMatches                                = kTRUE;
  std::map<ECbmModuleId, CbmDigitizeInfo*> fDigitizers = {};  
  CbmDaq* fDaq                                         = new CbmDaq();
  CbmDigitizationSource* fSource                       = new CbmDigitizationSource();  
  TString fOutFile                                     = {};                           
  TString fParRootFile                                 = {};                           
  TList fParAsciiFiles                                 = {};                           
  TString fMoniFile                                    = {};  
  Bool_t fOverwriteOutput                              = kFALSE;
  Bool_t fGenerateRunInfo                              = kTRUE;
  Int_t fRun                                           = 0;
 

  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 */