PndHoughPreprocessing.h

Go to the documentation of this file.

#ifndef PndHoughPreprocessing_H_
#define PndHoughPreprocessing_H_

#include "FairTask.h"
#include "FairRootManager.h"
#include "PndTrackCand.h"
#include "PndHoughData.h"

class PndHoughPreprocessing {

 public:
  PndHoughPreprocessing(PndHoughData *data)
    : fHits(data->GetHits()), fMapTubetoHit(data->GetMapTubetoHit()), fMapFairLinktoFairHit(data->GetMapFairLinktoFairHit()),
      fMapFairLinktoIsochrone(data->GetMapFairLinktoIsochrone()), fMapFairLinktoIsochroneError(data->GetMapFairLinktoIsochroneError()),
      fMapFairLinktoTubeId(data->GetMapFairLinktoTubeId()), fStrawMap(data->GetStrawMap())
  {
    ioman = FairRootManager::Instance();
  };

  virtual ~PndHoughPreprocessing(){};
  PndTrackCand CleanData(PndTrackCand &trackCand);
  PndTrackCand RejectSkewed(PndTrackCand &trackCand);
  PndTrackCand CleanAndRejectSkewed(PndTrackCand &trackCand);

 private:
  FairRootManager *ioman = nullptr;
  PndSttStrawMap *fStrawMap = nullptr;
  std::map<FairLink, FairHit *> fMapFairLinktoFairHit;
  std::map<FairLink, Double_t> fMapFairLinktoIsochrone;
  std::map<FairLink, Double_t> fMapFairLinktoIsochroneError;
  std::map<FairLink, Int_t> fMapFairLinktoTubeId;
  std::map<int, FairHit *> fMapTubetoHit; // maps hitted tube to FairHit
  std::vector<FairHit *> fHits;           // vector with selected hits of an event

  ClassDef(PndHoughPreprocessing, 1);
};

#endif /*PndHoughPreprocessing_H_*/