CbmTrdDigitizer.h

Go to the documentation of this file.

/* Copyright (C) 2008-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Florian Uhlig [committer], Andrey Lebedev */
 
#ifndef CBMTRDDIGITIZER_H_
#define CBMTRDDIGITIZER_H_
 
#include "CbmDefs.h"  // for ECbmModuleId
#include "CbmDigitize.h"
#include "CbmMatch.h"
#include "CbmTrdCheckUtil.h"
#include "CbmTrdDigi.h"
#include "CbmTrdRawToDigiR.h"
 
#include <map>
#include <memory>
 
class TClonesArray;
 
class CbmTrdParSetAsic;
class CbmTrdParSetGas;
class CbmTrdParSetDigi;
class CbmTrdParSetGain;
class CbmTrdParSetGeo;
class CbmTrdRadiator;
 
//class CbmTrdDigi;
//class CbmMatch;
 
//class CbmTrdGeoHandler;
class CbmTrdModuleSim;
 
class CbmTrdDigitizer : public CbmDigitize<CbmTrdDigi> {
public:
  enum CbmTrdSimDef
  {
    kTime = 0,        
    kNoise,           
    kWeightDistance,  
    kFASP             
  };
 
  CbmTrdDigitizer(std::shared_ptr<CbmTrdRadiator> radiator = nullptr);
 
  CbmTrdDigitizer(CbmTrdRadiator* radiator);
 
  virtual ~CbmTrdDigitizer();
  static Bool_t AddNoise() { return TESTBIT(fConfig, kNoise); }
  static Bool_t IsTimeBased() { return TESTBIT(fConfig, kTime); }
  static Bool_t UseWeightedDist() { return TESTBIT(fConfig, kWeightDistance); }
  static Bool_t UseFASP() { return TESTBIT(fConfig, kFASP); }
 
 
  ECbmModuleId GetSystemId() const { return ECbmModuleId::kTrd; }
 
  virtual InitStatus Init();
 
  virtual void SetParContainers();
 
  virtual void Exec(Option_t* option);
 
  virtual void Finish();
 
  static void SetAddNoise(Bool_t set = kTRUE) { set ? SETBIT(fConfig, kNoise) : CLRBIT(fConfig, kNoise); }
  static void SetUseFASP(Bool_t set = kTRUE) { set ? SETBIT(fConfig, kFASP) : CLRBIT(fConfig, kFASP); }
  static void SetTimeBased(Bool_t set = kTRUE) { set ? SETBIT(fConfig, kTime) : CLRBIT(fConfig, kTime); }
  static void SetWeightedDist(Bool_t set = kTRUE)
  {
    set ? SETBIT(fConfig, kWeightDistance) : CLRBIT(fConfig, kWeightDistance);
  }
 
private:
  CbmTrdDigitizer(const CbmTrdDigitizer&);
  CbmTrdDigitizer& operator=(const CbmTrdDigitizer&);
 
  void FlushBuffers();
  CbmTrdModuleSim* AddModule(Int_t detId);
  void ResetCounters();
 
  virtual void ResetArrays();
 
  static Int_t fConfig;  
 
  // event info
  Double_t fLastEventTime;  
 
  // counters
  Int_t fpoints;
  Int_t nofBackwardTracks;
 
  Float_t fEfficiency;  
 
  TClonesArray* fPoints;                
  TClonesArray* fTracks;                
  std::vector<CbmTrdDigi>* fDigis;      
  std::vector<CbmMatch>* fDigiMatches;  
 
  CbmTrdParSetAsic* fAsicPar;  
  CbmTrdParSetGas* fGasPar;    
  CbmTrdParSetDigi* fDigiPar;  
  CbmTrdParSetGain* fGainPar;  
  CbmTrdParSetGeo* fGeoPar;    
  std::shared_ptr<CbmTrdRadiator> fRadiator = nullptr;
  std::shared_ptr<CbmTrdRadiator> fRadiator2D = nullptr;
 
  CbmTrdRawToDigiR* fConverter;
  CbmTrdCheckUtil* fQA;
 
  //CbmTrdGeoHandler *fGeoHandler;///< helper to decript geoManager path into proper module type
  std::map<Int_t, CbmTrdModuleSim*> fModuleMap;  
 
  // Temporary storage for digis.
  std::map<Int_t, std::pair<CbmTrdDigi*, CbmMatch*>> fDigiMap;
 
  ClassDef(CbmTrdDigitizer, 1);
};
#endif  // CBMTRDDIGITIZER_H_