PndPidTrackingSystemInfo.h

Go to the documentation of this file.

#ifndef PID_TRACKING_INFO_H
#define PID_TRACKING_INFO_H

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

#include "PndPidCandidate.h"
#include "PndPidTrackInfo.h"
#include "PndPidTrackingDetectorInfo.h"

class TGeant3;

class PndPidTrackingSystemInfo : public PndPidTrackInfo {
 public:
  PndPidTrackingSystemInfo(Bool_t use_default_setup = kTRUE);

  virtual ~PndPidTrackingSystemInfo();

  void SetDetectors(std::vector<PndPidTrackingDetectorInfo*> detectors) {
    fDetectors = detectors;
  };
  void AddDetector(PndPidTrackingDetectorInfo* detector_info) {
    fDetectors.push_back(detector_info);
  }
  std::vector<PndPidTrackingDetectorInfo*> GetDetectors() { return fDetectors; };

  void Reset();

  void SetPidHypo(Int_t pid_hyp);

  void SetPropagator(PndPropagator *propagator);

  void SetParContainers(FairRuntimeDb *db);

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

  Bool_t GetCompleteTrackingSystemInfo(Int_t track_index, PndPidCandidate *pidCand);

 protected:
  Bool_t fUseDefaultSetup;
  std::vector<PndPidTrackingDetectorInfo*> fDetectors = {};

  ClassDef(PndPidTrackingSystemInfo, 1);
};

#endif // PID_TRACKING_INFO_H