PndGemTrackFinderOnHitsTB.h

Go to the documentation of this file.

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

#ifndef PNDGEMTRACKFINDERONHITSTB_H
#define PNDGEMTRACKFINDERONHITSTB_H

#include "TStopwatch.h"

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

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

#include <vector>

struct TrackSegmentTB {
  Int_t stationIndex[2];
  Int_t sensorNumber[2];
  Int_t hitIndex[2];
  Double_t trackMom;
  Double_t trackPhi;
  Double_t trackTheta;
  Int_t recoTrackIndex;
};

class PndGemTrackFinderOnHitsTB : public PndGemTrackFinder
{

 public:

  PndGemTrackFinderOnHitsTB();

  virtual ~PndGemTrackFinderOnHitsTB();

  virtual Int_t DoFind(TClonesArray* hitArray,
                       TClonesArray* trackArray,
                       TClonesArray* trackCandArray);

  void SetVerbose(const Int_t& verbose) { fVerbose = verbose; };
  void SetPrimary(const Int_t& primary) { fPrimary = primary; };
  void SetSigmaMult(Double_t tempd)     { fSigmaMult = tempd; };

  
 private:
 
  PndGemDigiPar* fDigiPar;

  TClonesArray* fMCTrackArray;
  TClonesArray* fMCPointArray;

  //  PndGemStructure *fGemStructure;

  Int_t fVerbose;

  Int_t fPrimary;
  
  Double_t fSigmaMult;

  // 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];

  TStopwatch fTimer;

  Double_t fPrepTime;
  Double_t fSegmTime;
  Double_t fMatchTime;
  Double_t fRemoveTime;
  Double_t fWriteTime;
  Double_t fAllTime;

  Int_t fNofEvents;  // event counter
  Int_t fMCAvailable;
  Int_t fNofClHits;

  Int_t fNofExpectedTrackSegments;
  Int_t fNofFoundTrackSegments;
  std::vector<TrackSegmentTB> fTrackSegments;
  Int_t FindTrackSegments(TClonesArray* hitArray, Int_t stat1Id, Int_t stat2Id);
  Int_t MatchTrackSegments();
  void RemoveCloneTracks(Int_t nofRecoTracks);
  Int_t CreateTracks(TClonesArray* hitArray, 
                     TClonesArray* trackArray, 
                     TClonesArray* trackCandArray, 
                     Int_t nofRecoTracks);

  void PrintTrackSegments(TClonesArray* hitArray);
  void PrintTracks(TClonesArray* hitArray, Int_t nofRecoTracks);
  void PrintMCTrackSegments(TClonesArray* hitArray);
  void PrintMCTracks(TClonesArray* hitArray, Int_t nofRecoTracks);

  virtual void SetParContainers();

  virtual void Init();

  virtual void Finish();

  ClassDef(PndGemTrackFinderOnHitsTB,1);

};

#endif