PndPidMuonSystemInfo.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_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