PndMdtTrkProducer.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 PNDMDTTRKPRODUCER_H
#define PNDMDTTRKPRODUCER_H 1

#include <PndPersistencyTask.h>
#include <map>
#include <vector>

#include "PndMdtTrk.h"
#include "PndMdtRecoPar.h"

#include "PndTrack.h"
#include "FairGeanePro.h"

class TClonesArray;

using std::map;
using std::vector;

class PndMdtTrkProducer : public PndPersistencyTask {

 public:
  PndMdtTrkProducer();

  ~PndMdtTrkProducer();

  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  virtual void SetParContainers();

  virtual void AlgorithmWithLheGenTrack();

  void SetRecMethod(Int_t rec_method) { fRec_method = rec_method; }

 private:
  Int_t fRec_method; // 0; previous method, use mdt info only; 1; use lhetrack as seed

  PndMdtTrk *AddTrk(PndMdtTrk *track);

  Bool_t MdtMapping(); // Creates maps of MDT hits
  void Reset();        // reset maps
  void SetGeometry();  // setting layer positions

  TClonesArray *fHitArray;

  TClonesArray *fLheGenTrack;

  TClonesArray *fTrkArray;

  // PndMdtRecoPar *fRecoPar;

  map<Int_t, vector<Int_t>> mapMdtBarrel;
  map<Int_t, vector<Int_t>> mapMdtEndcap;
  map<Int_t, vector<Int_t>> mapMdtForward;

  map<Int_t, TVector3> mapHitDirection;
  map<Int_t, Float_t> mapHitDistance;
  Float_t mdtLayerPos[3][20];
  Float_t mdtIronThickness[3][20];
  Float_t mdtModule1MaxZ;

  ClassDef(PndMdtTrkProducer, 1);
};

#endif