PndMvdGemTrackFinderOnHits.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                   PndMvdGemTrackFinderOnHits header file          -----
// -----                  Created 02.06.2009 by R. Karabowicz          -----
// -------------------------------------------------------------------------

#ifndef PNDMVDGEMTRACKFINDERONHITS_H
#define PNDMVDGEMTRACKFINDERONHITS_H

#include "PndPersistencyTask.h"
#include "FairMCPoint.h"
#include "PndMCTrack.h"

#include "PndSdsHit.h"
#include "PndGemHit.h"
#include "PndGemTrackFinder.h"
#include "PndGemDigiPar.h"

#include <vector>

struct TrackSegment {
  Int_t detId[2];
  Int_t hitIndex[2];
  Double_t trackMom;
  Double_t trackPhi;
  Double_t trackTheta;
  Int_t recoTrackIndex;
};

class PndMvdGemTrackFinderOnHits : public PndPersistencyTask {

 public:
  PndMvdGemTrackFinderOnHits();

  PndMvdGemTrackFinderOnHits(Int_t iVerbose);

  virtual ~PndMvdGemTrackFinderOnHits();

  virtual void Exec(Option_t *opt);

  void SetVerbose(const Int_t &verbose) { fVerbose = verbose; };

 private:
  PndGemDigiPar *fDigiPar;

  TClonesArray *fMvdPixelHitArray;
  TClonesArray *fMvdStripHitArray;
  TClonesArray *fGemHitArray;

  TClonesArray *fTrackArray;
  TClonesArray *fTrackCandArray;

  TClonesArray *fMCTrackArray;
  TClonesArray *fMCGemPointArray;
  TClonesArray *fMCMvdPointArray;

  //  PndGemStructure *fGemStructure;

  Int_t fVerbose;

  Int_t fPrimary;

  // Parameters, taken from the digiPar file
  Double_t fParThetaA;
  Double_t fParThetaB;

  Double_t fParTheta0;
  Double_t fParTheta1;
  Double_t fParTheta2;
  Double_t fParTheta3;

  Double_t fParRadPhi0;
  Double_t fParRadPhi2;

  Double_t fParMat0[3];
  Double_t fParMat1[3];

  Int_t fNofEvents; // event counter
  Int_t fMCAvailable;

  Int_t fNofExpectedTrackSegments;
  Int_t fNofFoundTrackSegments;
  std::vector<TrackSegment> fTrackSegments;
  Int_t FindTrackSegments(Int_t stat1Id, Int_t stat2Id);
  Int_t FindTrackSegments(Int_t stat2Id);
  Int_t FindTrackSegments();
  Int_t MatchTrackSegments();
  void RemoveCloneTracks(Int_t nofRecoTracks);
  Int_t CreateTracks(Int_t nofRecoTracks);

  void PrintTrackSegments();
  void PrintTracks(Int_t nofRecoTracks);
  void PrintMCTrackSegments();
  void PrintMCTracks(Int_t nofRecoTracks);

  virtual void SetParContainers();

  virtual void Finish();

  virtual InitStatus Init();

  virtual InitStatus ReInit();

  ClassDef(PndMvdGemTrackFinderOnHits, 1);
};

#endif