CbmTof.h

Go to the documentation of this file.

/* Copyright (C) 2006-2019 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese, Denis Bertini [committer], Christian Simon */
 
// -------------------------------------------------------------------------
// -----                         CbmTof header file                    -----
// -----                  Created 28/07/04  by V. Friese               -----
// -------------------------------------------------------------------------
 
#ifndef CBMTOF_H
#define CBMTOF_H
 
 
#include "FairDetector.h"
 
#include "TLorentzVector.h"
#include <tuple>
 
#include <map>
#include <set>
 
 
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);
 
  Bool_t CheckIfSensitive(std::string name);
  virtual Bool_t IsSensitive(const std::string& name);
 
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();
 
 
  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