#include <Make3Dtracks.h>

Public Member Functions
std::vector< const L1track3D * >	assocTrackCands3D (const TP &tp, bool rzFiltered) const

	Make3Dtracks (const Settings *settings, unsigned int iPhiSec, unsigned int iEtaReg, float etaMinSector, float etaMaxSector, float phiCentreSector)

bool	ranRZfilter () const

void	run (const std::list< L1track2D > &vecTracksRphi)

const TrkRZfilter *	rzFilter () const

const std::list< L1track3D > &	trackCands3D (bool rzFiltered) const

Private Member Functions
void	makeRZfilteredTrks (const std::list< L1track2D > &vecTracksRphi)

void	makeUnfilteredTrks (const std::list< L1track2D > &vecTracksRphi)

Private Attributes
float	etaMaxSector_

float	etaMinSector_

unsigned int	iEtaReg_

unsigned int	iPhiSec_

float	phiCentreSector_

bool	runRZfilter_

std::unique_ptr< TrkRZfilter >	rzFilter_

const Settings *	settings_

std::list< L1track3D >	vecTracks3D_rzFiltered_

std::list< L1track3D >	vecTracks3D_unfiltered_

Detailed Description

Definition at line 25 of file Make3Dtracks.h.

Constructor & Destructor Documentation

tmtt::Make3Dtracks::Make3Dtracks	(	const Settings *	settings,
		unsigned int	iPhiSec,
		unsigned int	iEtaReg,
		float	etaMinSector,
		float	etaMaxSector,
		float	phiCentreSector
	)

Definition at line 16 of file Make3Dtracks.cc.

References etaMaxSector_, etaMinSector_, iEtaReg_, iPhiSec_, phiCentreSector_, runRZfilter_, rzFilter_, and settings_.

       :  // Store config params & arguments.
         settings_(settings),
         iPhiSec_(iPhiSec),  // Sector number
         iEtaReg_(iEtaReg),
         etaMinSector_(etaMinSector),        // Range of eta sector
         etaMaxSector_(etaMaxSector),        // Range of eta sector
         phiCentreSector_(phiCentreSector),  // Centre of phi sector
 
         // Note if any fitters require an r-z track filter to be run.
         runRZfilter_(not settings->useRZfilter().empty()) {
     // Initialize any track filters (e.g. r-z) run after the r-phi Hough transform.
     if (runRZfilter_)
       rzFilter_ =
           std::make_unique<TrkRZfilter>(settings_, iPhiSec_, iEtaReg_, etaMinSector_, etaMaxSector_, phiCentreSector_);
   }

Member Function Documentation

vector< const L1track3D * > tmtt::Make3Dtracks::assocTrackCands3D	(	const TP &	tp,
		bool	rzFiltered
	)		const

Definition at line 113 of file Make3Dtracks.cc.

References tmtt::TP::index(), vecTracks3D_rzFiltered_, and vecTracks3D_unfiltered_.

                                                                                               {
     const list<L1track3D>& allTracks3D = (rzFiltered) ? vecTracks3D_rzFiltered_ : vecTracks3D_unfiltered_;
 
     vector<const L1track3D*> assocRecoTrk;
 
     // Loop over track candidates, looking for those associated to given TP.
     for (const L1track3D& trk : allTracks3D) {
       if (trk.matchedTP() != nullptr) {
         if (trk.matchedTP()->index() == tp.index())
           assocRecoTrk.push_back(&trk);
       }
     }
 
     return assocRecoTrk;
   }

void tmtt::Make3Dtracks::makeRZfilteredTrks ( const std::list< L1track2D > & vecTracksRphi )

private

Definition at line 92 of file Make3Dtracks.cc.

References tmtt::L1track3D::cheat(), rzFilter_, settings_, tmtt::Settings::trackFitCheat(), and vecTracks3D_rzFiltered_.

Referenced by run().

                                                                             {
     vecTracks3D_rzFiltered_ = rzFilter_->filterTracks(vecTracksRphi);
 
     // Optionally use MC truth to eliminate all fake tracks & all incorrect stubs assigned to tracks
     // before doing fit (for debugging).
     if (settings_->trackFitCheat()) {
       list<L1track3D> vecTracks3D_tmp;
       for (const L1track3D& trk : vecTracks3D_rzFiltered_) {
         L1track3D trk_tmp = trk;
         bool cheat_keep = trk_tmp.cheat();
         if (cheat_keep)
           vecTracks3D_tmp.push_back(trk_tmp);
       }
       vecTracks3D_rzFiltered_ = vecTracks3D_tmp;
     }
   }

void tmtt::Make3Dtracks::makeUnfilteredTrks ( const std::list< L1track2D > & vecTracksRphi )

private

Definition at line 41 of file Make3Dtracks.cc.

References tmtt::L1track3D::cheat(), tmtt::Settings::enableDigitize(), etaMaxSector_, etaMinSector_, funct::exp(), iEtaReg_, tmtt::Settings::invPtToInvR(), iPhiSec_, tmtt::Settings::kf_oneOver2rRange(), tmtt::Settings::kf_phi0Range(), funct::pow(), settings_, funct::tan(), tmtt::Settings::trackFitCheat(), and vecTracks3D_unfiltered_.

Referenced by run().

                                                                             {
     vecTracks3D_unfiltered_.clear();
 
     for (const L1track2D& trkRphi : vecTracksRphi) {
       const vector<Stub*>& stubsOnTrkRphi = trkRphi.stubs();  // stubs assigned to track
 
       float qOverPt = trkRphi.helix2D().first;
       float phi0 = trkRphi.helix2D().second;
 
       if (settings_->enableDigitize()) {
         // Centre of HT bin lies on boundary of two fitted track digi bins, so nudge slightly +ve (like FW)
         // to remove ambiguity.
         const float small = 0.1;
         const unsigned int nHelixBits = 18;  // Bits used internally in KF HLS to represent helix params.
         qOverPt += (2. / settings_->invPtToInvR()) * small * settings_->kf_oneOver2rRange() / pow(2., nHelixBits);
         phi0 += small * settings_->kf_phi0Range() / pow(2., nHelixBits);
       }
       pair<float, float> helixRphi(qOverPt, phi0);
 
       // Estimate r-z track helix parameters from centre of eta sector.
       float z0 = 0.;
       float tan_lambda = 0.5 * (1 / tan(2 * atan(exp(-etaMinSector_))) + 1 / tan(2 * atan(exp(-etaMaxSector_))));
 
       pair<float, float> helixRz(z0, tan_lambda);
 
       // Create 3D track, by adding r-z helix params to 2D track
       L1track3D trk3D(settings_,
                       stubsOnTrkRphi,
                       trkRphi.cellLocationHT(),
                       helixRphi,
                       helixRz,
                       iPhiSec_,
                       iEtaReg_,
                       trkRphi.optoLinkID(),
                       trkRphi.mergedHTcell());
 
       // Optionally use MC truth to eliminate all fake tracks & all incorrect stubs assigned to tracks
       // before doing fit (for debugging).
       bool cheat_keep = true;
       if (settings_->trackFitCheat())
         cheat_keep = trk3D.cheat();
 
       // Add to list of stored 3D tracks.
       if (cheat_keep)
         vecTracks3D_unfiltered_.push_back(trk3D);
     }
   }

bool tmtt::Make3Dtracks::ranRZfilter ( ) const

inline

Definition at line 64 of file Make3Dtracks.h.

References runRZfilter_.

64 { return runRZfilter_; } // Was r-z filter required/run?

tmtt::Make3Dtracks::runRZfilter_

bool runRZfilter_

Definition: Make3Dtracks.h:87

void tmtt::Make3Dtracks::run ( const std::list< L1track2D > & vecTracksRphi )

inline

Definition at line 38 of file Make3Dtracks.h.

References makeRZfilteredTrks(), makeUnfilteredTrks(), and runRZfilter_.

Referenced by tmtt::TMTrackProducer::produce().

                                                       {
       this->makeUnfilteredTrks(vecTracksRphi);
       if (runRZfilter_)
         this->makeRZfilteredTrks(vecTracksRphi);
     }

const TrkRZfilter* tmtt::Make3Dtracks::rzFilter ( ) const

inline

Definition at line 66 of file Make3Dtracks.h.

References rzFilter_.

66 { return rzFilter_.get(); }

tmtt::Make3Dtracks::rzFilter_

std::unique_ptr< TrkRZfilter > rzFilter_

Definition: Make3Dtracks.h:90

const std::list<L1track3D>& tmtt::Make3Dtracks::trackCands3D ( bool rzFiltered ) const

inline

Definition at line 49 of file Make3Dtracks.h.

References vecTracks3D_rzFiltered_, and vecTracks3D_unfiltered_.

Referenced by tmtt::TMTrackProducer::produce().

                                                                   {
       if (rzFiltered) {
         return vecTracks3D_rzFiltered_;
       } else {
         return vecTracks3D_unfiltered_;
       }
     }