PndHoughTrackCorrection.h

Go to the documentation of this file.

//  PndHoughTrackCorrection
//  A class to correct a PndTrack: only hits which are close enough to the track are added to the track

#ifndef PndHoughTrackCorrection_H_
#define PndHoughTrackCorrection_H_

#include "PndHoughUtilities.h"
#include "PndHoughData.h"

#include <vector>

class PndHoughTrackCorrection {

 public:
  PndHoughTrackCorrection(PndHoughData *fData)
  {
    ioman = FairRootManager::Instance();
    fMapFairLinktoFairHit = fData->GetMapFairLinktoFairHit();
    fMapFairLinktoIsochrone = fData->GetMapFairLinktoIsochrone();
  };

  virtual ~PndHoughTrackCorrection() { delete ioman; };
  virtual void SetUtilities(PndHoughUtilities *utilities) { fPndHoughUtilities = utilities; };
  TVector3 FindMinimumHit(PndTrackCand &trackCand, std::vector<double> &circle);
  double calc_rho(TVector3 &Hit, TVector3 &Track, TVector3 &Hitmin);
  PndTrackCand calcCorrectTrackCand(PndTrackCand &trackCand, TVector3 &Track, PndTrackCand &trackCandTot);
  PndTrackCand GetApolloniusTrackCand() { return fApolloniusTrackCand; };

 private:
  PndHoughUtilities *fPndHoughUtilities = nullptr;
  FairRootManager *ioman = nullptr;
  std::map<FairLink, FairHit *> fMapFairLinktoFairHit;
  std::map<FairLink, Double_t> fMapFairLinktoIsochrone;
  PndTrackCand fApolloniusTrackCand;

  ClassDef(PndHoughTrackCorrection, 1);
};

#endif /*PndHoughTrackCorrection_H_*/