PndMdtTrkProducer.h

Go to the documentation of this file.

#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