PndDsk.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                     PndDsk header file                        -----
// -----                Created 23/10/07  by P. Koch                   -----
// -------------------------------------------------------------------------

#ifndef PNDDSK_H
#define PNDDSK_H

#include "TLorentzVector.h"
#include "TVector3.h"
#include "TH2F.h"
#include "PndGeoDskFLG.h"

#include "FairDetector.h"
#include "PndDskFLGHit.h"

class TClonesArray;
class PndDskCerenkov;
class PndDskParticle;
class PndDskTrackPoint;

class PndDsk : public FairDetector {

 public:
  PndDsk();

  PndDsk(const char *name, Bool_t active);

  virtual ~PndDsk();

  virtual void Initialize();

  virtual Bool_t ProcessHits(FairVolume *vol = 0);

  virtual void EndOfEvent();

  virtual void Register();

  virtual TClonesArray *GetCollection(Int_t iColl) const;

  virtual void Print() const;

  virtual void Reset();

  virtual void CopyClones(TClonesArray *cl1, TClonesArray *cl2, Int_t offset);

  virtual void ConstructGeometry();

  virtual Bool_t CheckIfSensitive(std::string name);

  Bool_t DichroicMirrorTransmitted(Double_t wavelength, Int_t detector_type);

  PndDskCerenkov *AddCerenkov(Int_t trackID, Int_t detectorID, TVector3 position, TVector3 momentum, Double_t time, Double_t energy, Double_t wavelength, Int_t motherTrackID,
                              Int_t motherPdgCode, TString motherPdgName);

  PndDskFLGHit *AddHit(Int_t trackID, Int_t detectorID, TVector3 position_store, TVector3 momentum_store, Double_t time, Double_t angIn, Double_t thetaC_store,
                       TVector3 Cherenkov_photon, Int_t light_guide, Int_t pixel);

  PndDskParticle *AddParticle(Int_t trackID, Int_t detectorID, TVector3 position, TVector3 momentum, Double_t time, Int_t pdgCode, TString pdgName, Double_t energy,
                              Int_t motherTrackID, Int_t motherPdgCode, TString motherPdgName, Double_t mass, Double_t angIn, Double_t thetaC, Int_t nPhot);

  PndDskTrackPoint *AddTrackPoint(Int_t trackID, Int_t detectorID, TVector3 position, TVector3 momentum, Double_t time, Double_t length, Double_t eLoss);

  void SetDebugLevel(Int_t debugLevel) { fDebugLevel = debugLevel; }
  // void SetDetectors(Int_t detectorTypes, Int_t detectorsPerArray, Bool_t usingMirrors);  // not implemented
  void SetPDE(Double_t pde) { fPDE = pde; }
  void SetStoreCerenkovs(Bool_t storeCerenkovs) { fStoreCerenkovs = storeCerenkovs; }
  void SetStoreParticles(Bool_t storeParticles) { fStoreParticles = storeParticles; }
  void SetStoreTrackPoints(Bool_t storeTrackPoints) { fStoreTrackPoints = storeTrackPoints; }

  void SetTrapFraction(std::string name); // use trap fraction from fast sim, temp.
 private:
  Bool_t ProcessHitsCerenkov(FairVolume *vol = 0);
  Bool_t ProcessHitsCerenkov_FLG(FairVolume *vol = 0);
  Bool_t ProcessHitsParticle(FairVolume *vol = 0);

  Bool_t DoNotTrackCerenkov();

  TH2F *trapfrac[5]; 

  PndGeoDskFLG *fGeo; 

  Int_t fDebugLevel; 

  TClonesArray *fDskCerenkovCollection;   
  TClonesArray *fDskParticleCollection;   
  TClonesArray *fDskTrackPointCollection; 

  TClonesArray *fDskFLGHitArray; 

  Bool_t fStoreCerenkovs;       
  Bool_t fStoreParticles;       
  Bool_t fStoreTrackPoints;     
  Bool_t fStoreFLGHits;         
  Bool_t fCalcPWay;             
  Bool_t fMeasureTotalRefAngle; 

  Int_t fTrackID;     
  Int_t fDetectorID;  
  TVector3 fPosition; 
  TVector3 fMomentum; 
  Double_t fTime;     
  Double_t fLength;   
  Double_t fELoss;    

  Double_t fPDE;        
  Int_t fPdgCode;       
  Double_t fEnergy;     
  Double_t fWavelength; 

  Int_t fMotherTrackID;   
  Int_t fMotherPdgCode;   
  TString fMotherPdgName; 

  Short_t fDetType;      
  Int_t fDetNumber;      
  Double_t fDetTime;     
  TVector3 fDetMomentum; 

  Double_t fPrimaryAngleToCerenkov; 
  Double_t fPrimaryHitAngle;        
  TVector3 fPrimaryHitMomentum;     

  TString fPdgName;      
  TVector3 fEndPosition; 
  Double_t fEndTime;     
  TVector3 fEndMomentum; 
  Double_t fEndEnergy;   

  Double_t fAngIn;  
  Double_t fThetaC; 

  TLorentzVector tmpLVec; 

  ClassDef(PndDsk, 1)
};

#endif // PNDDSK_H