PndPidTrackInfo.h

Go to the documentation of this file.

#ifndef PID_TRACK_INFO_H
#define PID_TRACK_INFO_H

#include "TString.h"
#include "TClonesArray.h"

#include "FairPropagator.h"
#include "PndPropagator.h"
#include "FairGeane.h"
#include "FairRootManager.h"

#include "PndTrack.h"
#include "PndMCTrack.h"
#include "PndPidCandidate.h"

class TGeant3;

class PndPidTrackInfo {
 public:
  PndPidTrackInfo();
  virtual ~PndPidTrackInfo(); 

  void SetUsePropagator(Bool_t use_propagator) { fUsePropagator = use_propagator; };
  void SetPropagator(PndPropagator *propagator) { fPropagator = propagator; };
  void SetPidHypo(Int_t pid_hyp) { fPidHyp = pid_hyp; };
  void SetVerbose(Int_t verbose) { fVerbose = verbose; };
  void SetBackPropagate(Bool_t back_propagate) { fBackPropagate = back_propagate; };
  void SetTrackBranchBarrel(TString track_branch) { fTrackBranchBarrel = track_branch; };
  void SetTrackBranchForward(TString track_branch) { fTrackBranchForward = track_branch; };
  void SetIsBarrel(Bool_t is_barrel) { fIsBarrel = is_barrel; };

  Int_t GetNTracksTotal() { return ((fTrackBarrel != nullptr) ? fTrackBarrel->GetEntriesFast() : 0) + ((fTrackForward != nullptr) ? fTrackForward->GetEntriesFast() : 0); };

  virtual void Register() {};

  Bool_t SetMcTrack(FairRootManager *manager);

  Bool_t GetInfo(Int_t track_index, PndPidCandidate *pidCand);

  Int_t GetNTracks();

  PndTrack *GetTrack(Int_t index);

  PndMCTrack *GetMcTrack(Int_t index);

  Bool_t SetTrackArrays(FairRootManager *manager);

 protected:
  Bool_t fUsePropagator = kFALSE;       
  PndPropagator *fPropagator = nullptr;  
  Int_t fPidHyp = 0;                    
  Int_t fVerbose = 0;                   

  TString fTrackBranchBarrel = "";
  TString fTrackBranchForward = "";
  TClonesArray *fTrackBarrel = nullptr;
  TClonesArray *fTrackForward = nullptr;
  TClonesArray *fMcTrack = nullptr;
  Bool_t fBackPropagate = kFALSE;
  Bool_t fIsBarrel = kTRUE;

  ClassDef(PndPidTrackInfo, 1);
};

#endif // PID_TRACK_INFO_H