PndMdt.h

Go to the documentation of this file.

#ifndef PNDMDT_H
#define PNDMDT_H 1

#include "TROOT.h"
#include "TClonesArray.h"
#include "TLorentzVector.h"
#include "TString.h"

#include "TGeoManager.h"
#include "TGeoVolume.h"

#include "FairRootManager.h"
#include "FairDetector.h"
#include "FairVolume.h"

#include "PndGeoMdtPar.h"
//#include "PndMdtPoint.h"

class PndMdt : public FairDetector {
 public:
  PndMdt();
  PndMdt(const char *name, Bool_t active);
  ~PndMdt();

  inline void SetMdtMagnet(bool opt = false)
  {
    mdtMagnet = opt;
    return;
  };
  inline void SetMdtCoil(bool opt = false)
  {
    mdtCoil = opt;
    return;
  };
  inline void SetMdtMFIron(bool opt = false)
  {
    mdtMFI = opt;
    return;
  };

  void SetBarrel(TString name) { fBarrel = name; };
  void SetEndcap(TString name) { fEndcap = name; };
  void SetMuonFilter(TString name) { fMuonFilter = name; };
  void SetForward(TString name) { fForward = name; };
  // hit
  inline void Register() { FairRootManager::Instance()->Register("MdtPoint", "Mdt", fMdtCollection, kTRUE); };

  void ResetParameters();
  void Print() const;
  void Reset();

  TClonesArray *GetCollection(Int_t iColl) const;

  // detector
  inline void SetVerbosity(Bool_t verbosity) { fVerboseLevel = verbosity; };

  void ConstructGeometry();
  void Initialize();
  void BeginEvent();
  Bool_t ProcessHits(FairVolume *vol);
  Bool_t ProcessHitsFast(FairVolume *vol);
  Bool_t ProcessHitsRoot(FairVolume *vol);
  void EndOfEvent();

 private:
  Bool_t CheckIfSensitive(std::string name);
  Int_t PdgToIndex(Int_t pdg);

  void PndMdtMagnet();          
  void PndMdtMuonFilter();      
  void PndMdtMFIron();          
  void PndMdtForward();         
  void PndMdtCoil();            
  void ConstructGeometryTo();   
  void ConstructGeometryFast(); 
  void ConstructGeometryRoot(); 
  void ConstructGeometryDu();   

  // hit
  TClonesArray *fMdtCollection; 
  Int_t fTrkIn;
  Float_t fELoss;
  Float_t fTime;
  TLorentzVector fPos_In;
  TLorentzVector fMom_In;

  // detector
  Bool_t fVerboseLevel;

  bool mdtMagnet; 
  bool mdtCoil;   
  bool mdtMFI;    

  TString fBarrel;
  TString fEndcap;
  TString fMuonFilter;
  TString fForward;

  ClassDef(PndMdt, 1)
};

#endif /* !PNDMDT_H */