PndPidDetectorInfo.h

Go to the documentation of this file.

#ifndef PID_DETECTOR_INFO_H
#define PID_DETECTOR_INFO_H

#include "TNtuple.h"

#include "FairGeane.h"
#include "FairRootManager.h"
#include "FairRuntimeDb.h"
#include "FairTrackParH.h"

#include "PndPidCorrPar.h"
#include "PndTrack.h"
#include "PndPidCandidate.h"
#include "PndPropagator.h"
#include "PndGeanePro.h"

namespace PndPidDetectorParams
{
  struct DetectorParams {
    Bool_t debug_mode;
    Bool_t use_propagator;
    Bool_t ideal;
    Bool_t corr_error_prop;
    Int_t pid_hyp;
    Bool_t use_ideal_hyp;
    Int_t verbosity_level;
    PndPidCorrPar *corr_par;
    Bool_t mdt_refit;
    Bool_t fast;
    Bool_t back_propagate;
    TString track_branch;
  };
}

class PndPidDetectorInfo {
 public:
  PndPidDetectorInfo();

  virtual ~PndPidDetectorInfo(){}; 

  virtual Int_t SetDetectorHits(FairRootManager *manager) = 0;

  virtual Bool_t PreprocessTracks(const Int_t /*trackid*/, PndTrack */*track*/){return kTRUE;};

  virtual Bool_t GetInfo(PndTrack *track, PndPidCandidate *pidCand) = 0;

  virtual void GetNeutralInfo(Int_t, TClonesArray *){};

  virtual void ActivateDebugNtuple(Bool_t debug_mode) { fDebugMode = debug_mode; };
  void SetDetectorMode(Short_t detector_mode) { fDetMode = detector_mode; };
  void SetUsePropagator(Bool_t use_propagator) { fUsePropagator = use_propagator; };
  void SetPropagator(PndPropagator *propagator) { fPropagator = propagator; };
  void SetIdeal(Bool_t is_ideal) { fIdeal = is_ideal; };
  void SetPidHypo(Int_t pid_hyp) { fPidHyp = pid_hyp; };
  void SetVerbose(Int_t verbose) { fVerbose = verbose; };
  void SetCorrPar(PndPidCorrPar *pid_corr_par) { fCorrPar = pid_corr_par; };
  virtual Bool_t SetParContainers(FairRuntimeDb *) { return kTRUE; };

  TNtuple *GetDetCorr() { return fDetCorr; };
  Int_t GetDetectorMode() { return fDetMode; };

  void Finish();

  virtual void Reset(){};
  virtual void Register(){};

  virtual Bool_t Setup(FairRootManager *, PndPropagator *propagator, PndPidDetectorParams::DetectorParams *params);

  virtual Bool_t Propagate(TVector3 pos, FairTrackParH *helix, FairTrackParH *res,
                           Int_t pdg_code, Int_t direction = 1);

  virtual Bool_t Propagate(std::string volume, FairTrackParH *helix, FairTrackParH *res, Int_t pdg_code);

 protected:
  Short_t fDetMode = -1;
  Bool_t fUsePropagator = kFALSE;                 
  PndPropagator *fPropagator = nullptr;
  Bool_t fIdeal = kFALSE;                         
  Bool_t fDebugMode = kTRUE;                      
  Int_t fPidHyp = 0;                              
  Int_t fVerbose = 0;                             
  Float_t fGLength = -1000;
  PndPidDetectorParams::DetectorParams *fDetectorParams = nullptr;

  TNtuple *fDetCorr = nullptr;                    
  PndPidCorrPar *fCorrPar = new PndPidCorrPar();  

  ClassDef(PndPidDetectorInfo, 1);
};

#endif // PID_DETECTOR_INFO_H