PndDisc.h

Go to the documentation of this file.

//-------------------------------------------------------------------------
// Author:      Oliver Merle (Oliver.Merle@exp2.physik.uni-giessen.de)
// Changes:     Mustafa Schmidt (Mustafa.A.Schmidt@physik.uni-giessen.de)
// Date:        30.11.2015
// Description: Disc DIRC Implementation for PandaRoot (Giessen)
//-------------------------------------------------------------------------

#ifndef PNDDISC_HH
#define PNDDISC_HH

// Subdetector specific
#include <PndDiscSensorMCPoint.h>
#include <PndDiscParticleMCPoint.h>

// ROOT
#include <Rtypes.h>
#include <TLorentzVector.h>

// FairRoot / PROOT
#include <FairDetector.h>

// cpp
#include <set>
#include <string>
#include <map>

class FairMCEventHeader;

class PndDisc : public FairDetector {
 public:
  // ctor / dtor
  PndDisc();
  PndDisc(const char *name, Bool_t active, Int_t det_id = 0);
  ~PndDisc();

  // interface - initialization
  virtual void Initialize(); //* detector initialization

  // interface - geo (FairModule)
  virtual void ConstructGeometry();
  virtual void ConstructOpGeometry();

  // interface - transport (FairDetector)
  virtual Bool_t ProcessHits(FairVolume *v = 0); //* called for each step during simulation (see FairMCApplication::Stepping())
  // virtual void   BeginPrimary();                      // define actions at the beginning of the primary track (-> TVirtualMCApplication)
  // virtual void   FinishPrimary();                     // define actions at the end of the primary track       (-> TVirtualMCApplication)
  virtual void BeginEvent(); // define actions at the beginning of the event         (-> TVirtualMCApplication)
  // virtual void   FinishEvent();                       // define actions at the end of the event               (-> TVirtualMCApplication)
  virtual void EndOfEvent(); //* called after FinishEvent() and after fRootManager->Fill() in FairMCApplication::FinishEvent (time to free the data in mem?)
  virtual void PreTrack();   // define actions at the beginning of each track        (-> TVirtualMCApplication)
  virtual void PostTrack();  // define actions at the end of each track              (-> TVirtualMCApplication)
  // virtual void   FinishRun();                         // called after sim all events (gMC->ProcessRun(int))   (-> FairMCApplication::FinishRun)
  // virtual void   SetSpecialPhysicsCuts();             //

  virtual bool CheckIfSensitive(std::string name); // Fairroot uses this function to query sensitive volumes

  // interface - collections
  virtual void Register();                                //* register the produced collections in FAIRRootManager
  virtual TClonesArray *GetCollection(Int_t iColl) const; //* return pointer to the produced collections **/
  virtual void Reset();                                   //* reset containers (abstract in FairDetector)
  // virtual void   CopyClones( TClonesArray* cl1,  TClonesArray* cl2 , Int_t offset);

  // --------------------------------
  // Detector specific functionality:
  // --------------------------------
  void StorePhotonTracks(Bool_t bval);
  void SetFilterInterval(Double_t const &wl_min_nm_, Double_t const &wl_max_nm_);

 private:
  Bool_t store_photon_tracks;
  Bool_t track_is_photon;
  TLorentzVector old_momentum; // cache the photon momentum.

  TClonesArray *clarr_sensor_hits;     
  TClonesArray *clarr_photon_tracks;   
  TClonesArray *clarr_particle_tracks; 

  // keep track of the number of entries. Looking in Roots GetEntriesFast
  // makes you pulling your hair out.
  int nextid_clarr_sensor_hits;
  int nextid_clarr_photon_tracks;
  int nextid_clarr_particle_tracks;

  FairMCEventHeader *ev_header;                                                     
  std::map<int, std::pair<int, double>> last_track_occurence;                       
  std::set<std::string> names_of_sensitive_volumes;                                 
  std::map<int, double> internal_reflection_angle_of_photons;                       
  std::map<int, std::pair<TLorentzVector, TLorentzVector>> photons_entering_optics; 

  int design_id;
  Double_t wl_min_nm;
  Double_t wl_max_nm;

  ClassDef(PndDisc, 1)
};

#endif // PNDDISC_HH