CbmRecoSts.h

Go to the documentation of this file.

 
#ifndef CBMRECOSTS_H
#define CBMRECOSTS_H 1
 
 
#include <FairTask.h>
#include <TStopwatch.h>
 
 
class CbmDigiManager;
class CbmEvent;
class CbmStsElement;
class CbmStsParAsic;
class CbmStsParModule;
class CbmStsParSensor;
class CbmStsParSensorCond;
class CbmStsParSetModule;
class CbmStsParSetSensor;
class CbmStsParSetSensorCond;
class CbmStsParSim;
class CbmStsRecoModule;
class CbmStsSensor;
class CbmStsSetup;
 
 
enum ECbmRecoMode { kCbmRecoTimeslice, kCbmRecoEvent };
 
 
class CbmRecoSts : public FairTask {
 
public:
  CbmRecoSts(ECbmRecoMode mode    = kCbmRecoTimeslice,
             Bool_t writeClusters = kFALSE,
             Bool_t runParallel   = kFALSE);
 
 
  CbmRecoSts(const CbmRecoSts&) = delete;
 
 
  CbmRecoSts operator=(const CbmRecoSts&) = delete;
 
 
  virtual ~CbmRecoSts();
 
 
  virtual void Exec(Option_t* opt);
 
 
  virtual void Finish();
 
 
  TClonesArray* GetClusters() { return fClusters; }
 
 
  virtual InitStatus Init();
 
 
  void SetEventMode(Bool_t choice = kTRUE) {
    fMode = (choice ? kCbmRecoEvent : kCbmRecoTimeslice);
  }
 
 
  void SetMode(ECbmRecoMode mode) { fMode = mode; }
 
 
  virtual void SetParContainers();
 
 
  void SetTimeCutClustersAbs(Double_t value) { fTimeCutClustersAbs = value; }
 
 
  void SetTimeCutClustersSig(Double_t value) { fTimeCutClustersSig = value; }
 
 
  void SetTimeCutDigisAbs(Double_t value) { fTimeCutDigisAbs = value; }
 
 
  void SetTimeCutDigisSig(Double_t value) { fTimeCutDigisSig = value; }
 
 
  void UseModulePar(CbmStsParModule* modulePar) { fUserParModule = modulePar; }
 
 
  void UseSensorCond(CbmStsParSensorCond* sensorCond) {
    fUserParSensorCond = sensorCond;
  }
 
 
  void UseSensorPar(CbmStsParSensor* sensorPar) { fUserParSensor = sensorPar; }
 
 
private:
  std::pair<Double_t, Double_t>
  LorentzShift(const CbmStsParSensorCond& conditions, Double_t dZ, Double_t bY);
 
 
  UInt_t CreateModules();
 
 
  void GetSensorParameters(CbmStsElement* geoSensor);
 
 
  void ProcessData(CbmEvent* event = nullptr);
 
 
private:
  // --- I/O
  TClonesArray* fEvents        = nullptr;  
  CbmDigiManager* fDigiManager = nullptr;  
  TClonesArray* fClusters      = nullptr;  
  TClonesArray* fHits          = nullptr;  
 
  // --- Setup and parameters
  CbmStsSetup* fSetup                 = nullptr;  
  CbmStsParSim* fParSim               = nullptr;  
  CbmStsParSetModule* fParSetModule   = nullptr;  
  CbmStsParSetSensor* fParSetSensor   = nullptr;  
  CbmStsParSetSensorCond* fParSetCond = nullptr;  
 
  // --- User-defined parameters, not from database
  CbmStsParAsic* fUserParAsic             = nullptr;
  CbmStsParModule* fUserParModule         = nullptr;
  CbmStsParSensor* fUserParSensor         = nullptr;
  CbmStsParSensorCond* fUserParSensorCond = nullptr;
 
  // --- Settings
  ECbmRecoMode fMode           = kCbmRecoEvent;  
  Double_t fTimeCutDigisSig    = 3.;      
  Double_t fTimeCutDigisAbs    = -1.;     
  Double_t fTimeCutClustersSig = 4.;      
  Double_t fTimeCutClustersAbs = -1.;     
  Bool_t fWriteClusters        = kFALSE;  
  Bool_t fRunParallel          = kFALSE;  
 
  // --- Counters
  TStopwatch fTimer {};            
  Int_t fNofTimeslices      = 0;   
  Int_t fNofEvents          = 0;   
  Double_t fNofDigis        = 0;   
  Double_t fNofDigisUsed    = 0;   
  Double_t fNofDigisIgnored = 0;   
  Double_t fNofClusters     = 0;   
  Double_t fNofHits         = 0;   
  Double_t fTimeTot         = 0.;  
  Double_t fTime1           = 0.;  
  Double_t fTime2           = 0.;  
  Double_t fTime3           = 0.;  
  Double_t fTime4           = 0.;  
 
  // --- Reconstruction modules
  std::map<UInt_t, CbmStsRecoModule*> fModules {};  
  std::vector<CbmStsRecoModule*> fModuleIndex {};   
 
 
  ClassDef(CbmRecoSts, 1);
};
 
#endif