CbmTof.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                         CbmTof header file                    -----
// -----                  Created 28/07/04  by V. Friese               -----
// -------------------------------------------------------------------------
 
#ifndef CBMTOF_H
#define CBMTOF_H
 
 
#include "FairDetector.h"
 
#include "TLorentzVector.h"
 
#include <map>
#include <set>
#include <tuple>
 
 
class CbmTofPoint;
class CbmTofGeoHandler;
class FairVolume;
class TClonesArray;
class TVector3;
class TGeoCombiTrans;
class TGeoPhysicalNode;
 
class CbmTof : public FairDetector {
 
public:
  CbmTof();
 
 
  CbmTof(const char* name, Bool_t active);
 
 
  virtual ~CbmTof();
 
  virtual void PreTrack();
  virtual void FinishEvent();
 
  virtual Bool_t ProcessHits(FairVolume* vol = 0);
 
 
  virtual void EndOfEvent();
 
 
  virtual void Register();
 
 
  virtual TClonesArray* GetCollection(Int_t iColl) const;
 
 
  virtual void Print(Option_t* = "") const;
 
 
  virtual void Reset();
 
 
  virtual void CopyClones(TClonesArray* cl1, TClonesArray* cl2, Int_t offset);
 
 
  virtual void ConstructGeometry();
 
  virtual void Initialize();
 
  void GenerateOnePointPerTrack(Bool_t bOnePointPerTrack = kTRUE) {
    fbOnePointPerTrack = bOnePointPerTrack;
  }
  void SetProcessAnyTrack(Bool_t bProcess = kTRUE) {
    fbProcessAnyTrack = bProcess;
  }
  void SetAllCountersInactive(Bool_t bInactive = kTRUE) {
    fbAllCountersInactive = bInactive;
  }
 
  void
  SetCounterActive(Int_t iModuleType, Int_t iModuleIndex, Int_t iCounterIndex);
 
  void SetCounterInactive(Int_t iModuleType,
                          Int_t iModuleIndex,
                          Int_t iCounterIndex);
 
  void
  SetCounterInBeam(Int_t iModuleType, Int_t iModuleIndex, Int_t iCounterIndex);
 
private:
  Int_t fTrackID;       
  Int_t fVolumeID;      
  TLorentzVector fPos;  
  TLorentzVector fMom;  
  Double32_t fTime;     
  Double32_t fLength;   
  Double32_t fELoss;    
 
  Int_t fPosIndex;                
  TClonesArray* fTofCollection;   
  CbmTofGeoHandler* fGeoHandler;  
 
  TGeoCombiTrans*
    fCombiTrans;  
 
  std::string fVolumeName;  
 
  Bool_t fbOnePointPerTrack;
  Bool_t fbIsNewTrack;
 
  TString fTofNodePath;        
  TString fCurrentNodePath;    
  Int_t fCurrentModuleType;    
  Int_t fCurrentModuleIndex;   
  Int_t fCurrentCounterIndex;  
 
  std::set<std::tuple<Int_t, Int_t, Int_t>>
    fActiveCounters;  
  std::set<std::tuple<Int_t, Int_t, Int_t>>
    fInactiveCounters;                  
  std::set<Int_t> fInactiveCounterIDs;  
 
  std::map<std::tuple<Int_t, Int_t, Int_t>,
           std::pair<TString, TGeoPhysicalNode*>>
    fCountersInBeam;  
 
  Int_t fOutputTreeEntry;
 
  Bool_t fbProcessAnyTrack;
 
  Bool_t fbAllCountersInactive;
 
  CbmTofPoint* AddHit(Int_t trackID,
                      Int_t detID,
                      TVector3 pos,
                      TVector3 mom,
                      Double_t time,
                      Double_t length,
                      Double_t eLoss);
 
  void ResetParameters();
 
  Bool_t CheckIfSensitive(std::string name);
 
  virtual void ConstructRootGeometry(TGeoMatrix* shift = NULL);
 
  void CreateInBeamNodes();
 
  CbmTof(const CbmTof&);
  CbmTof& operator=(const CbmTof&);
 
 
  ClassDef(CbmTof, 6)
};
 
 
inline void CbmTof::ResetParameters() {
  fTrackID = fVolumeID = 0;
  fPos.SetXYZM(0.0, 0.0, 0.0, 0.0);
  fMom.SetXYZM(0.0, 0.0, 0.0, 0.0);
  fTime = fLength = fELoss = 0;
  fPosIndex                = 0;
};
 
 
#endif