PndPidMuonSystemInfo.h

Go to the documentation of this file.

#ifndef PID_MUONSYSTEM_INFO_H
#define PID_MUONSYSTEM_INFO_H

#include <map>
#include <vector>
#include <memory>

#include "TClonesArray.h"

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

#include "PndRecoKalmanFit.h"
#include "PndTrack.h"
#include "PndPidDetectorInfo.h"

class PndPidMuonSystemInfo : public PndPidDetectorInfo {
 public:
  PndPidMuonSystemInfo();  
  ~PndPidMuonSystemInfo(); 

  void Register()
  {
    if (fMdtRefit)
      RegisterMdtTrack();
  };
  void ResetMdtTrack()
  {
    if (fMdtRefit)
      fMdtTrack->Delete();
  };
  void RegisterMdtTrack()
  {
    FairRootManager::Instance()->Register("MdtTrack", "Pid", fMdtTrack, kTRUE);
    LOG(info) << "PndPidCorrelator::Register(): "
              << "registered branch "
              << "MdtTrack";
  };

  Int_t SetDetectorHits(FairRootManager *manager);

  Bool_t SetMdtRefit(Bool_t mdt_refit);

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

  void Reset();

  PndTrack *AddMdtTrack(PndTrack *track);

  Bool_t MdtMapping();

  Bool_t MdtGeometry();

 protected:
  TClonesArray *fMdtHit = nullptr;   
  TClonesArray *fMdtPoint = nullptr; 
  TClonesArray *fMdtTrk = nullptr;   
  TClonesArray *fMdtTrack = nullptr; 
  Bool_t fMdtRefit;                  

  std::unique_ptr<PndRecoKalmanFit> fFitter{nullptr}; 
  Float_t fMdtLayerPos[3][20];                        
  Float_t fMdtIronThickness[3][20];                   
  std::map<Int_t, std::vector<Int_t>> fMapMdtBarrel;  
  std::map<Int_t, std::vector<Int_t>> fMapMdtEndcap;  
  std::map<Int_t, std::vector<Int_t>> fMapMdtForward; 

  ClassDef(PndPidMuonSystemInfo, 1);
};

#endif // PID_MUONSYSTEM_INFO_H