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