PndHoughTrackCorrection.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

//  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; };
  PndTrackCand changeHitOrder(PndTrackCand &trackCand, TVector3 &Track, TVector3 &Hitmin);

 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_*/