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