PndFtsHoughTrackFinder.h

Go to the documentation of this file.

#ifndef PndFtsHoughTrackFinder_H
#define PndFtsHoughTrackFinder_H

#include "PndFtsHoughTrackFinderParams.h"

//#include "TClonesArray.h"
#include "Rtypes.h"     // for Double_t, Int_t, etc
#include "FairLogger.h" // for FairLogger, MESSAGE_ORIGIN

#include <cmath>
#include "TMath.h"
#include <math.h>
#include <algorithm>
#include <set>
#include <vector>
#include <map>
#include <fstream>
#include <iostream>

#include "PndFtsHit.h"
#include "PndFtsHoughTrackerTask.h"
#include "PndFtsHoughSpace.h"
#include "PndFtsHoughTracklet.h"
#include "PndFtsHoughTrackCand.h"
#include "PndTrackCand.h"
#include "PndTrack.h"

// For error throwing
#include "TString.h"
#include <stdexcept>

class PndFtsHoughTrackFinder {
 public:
  PndFtsHoughTrackFinder(PndFtsHoughTrackerTask *trackerTask); 
  virtual ~PndFtsHoughTrackFinder();                           

  // use this for parameter optimization
  void OverwriteTrackFinderParams(PndFtsHoughTrackFinderParams newParams) { fParams = newParams; };

  virtual void FindTracks(); 

  // Output
  Int_t NTracks() const { return fHoughTrackCandsComplete.size(); }; 

  PndTrack GetPndTrack(int i) { return fHoughTrackCandsComplete[i].getPndTrack(); };
  PndTrackCand GetPndTrackCand(int i) { return fHoughTrackCandsComplete[i].getPndTrackCand(); };
  PndFtsHoughTrackCand GetHoughTrack(int i) const { return fHoughTrackCandsComplete[i]; };

  // Output event statistics
  Int_t getNLinesBeforeDipoleFound() const { return fNLinesBeforeDipoleFound; };
  Int_t getNLinesBehindDipoleFound() const { return fNLinesBehindDipoleFound; };
  Int_t getNParabolasFound() const { return fNParabolasFound; };
  Int_t getNTracksFound() const { return fNTracksFound; };

 protected:
  PndFtsHoughTrackerTask *fTrackerTask;

  void throwError(const TString s) const { throw std::runtime_error(s.Data()); };

  //    Int_t   fFtsBranchId; // needed for saving and accessing hits
  //    TClonesArray *fFtsHitArray; ///< @brief Input array of all FTS hits.

  // for Hough
  PndFtsHoughTrackFinderParams fParams;

  //-----------
  std::vector<PndFtsHoughTrackCand> fHoughTrackCandsComplete;    
  std::vector<PndFtsHoughTrackCand> fHoughTrackCandsZxPlaneOnly; 
  //    std::vector<PndTrackCand> fTrackCand; // resulting track candidates, also used for returning PndTracks

  static const Double_t fThetaRadLineBehindDipoleMatchesToParabolaIfBelow;

  const UInt_t fMinPeakHeightZxLineBeforeDipole; 
  const UInt_t fMinPeakHeightZxParabola; 
  const UInt_t fMinPeakHeightZxLineBehindDipole; 
  const UInt_t fMinPeakHeightZyLine; 

  // event statistics
  Int_t fNLinesBeforeDipoleFound;
  Int_t fNLinesBehindDipoleFound;
  Int_t fNParabolasFound;
  Int_t fNTracksFound;

  // takes the heighest peak (according to peak finder)
  // of all peaks that share > maxSameHits
  Bool_t FilterTrackletsBasedOnSharedHits(UInt_t maxAcceptableSharedHits, std::vector<PndFtsHoughTracklet> &tracklets);
  // helper functions for tracking algorithm
  std::vector<PndFtsHoughTracklet> FindLinesBehindDipoleZx();
  std::vector<PndFtsHoughTracklet> FindLinesBeforeDipoleZx();
  void FindMatchingParabolaToLineBeforeDipoleZxAndAddLineBehindDipole(const std::vector<PndFtsHoughTracklet> &trackletsLineBeforeDipole,
                                                                      const std::vector<PndFtsHoughTracklet> &trackletsLineBehindDipole);
  // kTRUE iif angles of parabola and of line behind dipole match at z coordinate where I switch from parabola to line (in zx plane)
  inline Bool_t LineBehindDipoleMatchesToLinePlusParabola(const PndFtsHoughTrackCand &lineParabola, const PndFtsHoughTracklet &lineBehindDipole) const;
  void FindZyLineMatchingToLineParabolaLineInZx();

 private:
  ClassDef(PndFtsHoughTrackFinder, 1);
};

// inline
Bool_t PndFtsHoughTrackFinder::LineBehindDipoleMatchesToLinePlusParabola(const PndFtsHoughTrackCand &lineParabola, const PndFtsHoughTracklet &lineBehindDipole) const
{

  // the angles should be compared at the z coordinate where I switch from parabola to line behind dipole
  Double_t zParabolaLine = lineBehindDipole.getZRefLabSys();
  return fabs(lineParabola.getThetaZyRad(zParabolaLine) - lineBehindDipole.getThetaRadVal()) < fThetaRadLineBehindDipoleMatchesToParabolaIfBelow;
}

#endif /*PndFtsHoughTrackFinder_H*/