PndPidTrackingSystemInfo.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#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