tmtt::Make3Dtracks Class Reference

#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

Make3Dtracks()

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

assocTrackCands3D()

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

Definition at line 113 of file Make3Dtracks.cc.

References cmsswSequenceInfo::tp, 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;
  }

makeRZfilteredTrks()

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;
    }
  }

makeUnfilteredTrks()

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_, JetChargeProducer_cfi::exp, iEtaReg_, tmtt::Settings::invPtToInvR(), iPhiSec_, tmtt::Settings::kf_oneOver2rRange(), tmtt::Settings::kf_phi0Range(), funct::pow(), settings_, funct::tan(), tmtt::Settings::trackFitCheat(), vecTracks3D_unfiltered_, and HLTMuonOfflineAnalyzer_cfi::z0.

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);
    }
  }

ranRZfilter()

bool tmtt::Make3Dtracks::ranRZfilter ( ) const

inline

Definition at line 64 of file Make3Dtracks.h.

References runRZfilter_.

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

run()

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);
    }

rzFilter()

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

inline

Definition at line 66 of file Make3Dtracks.h.

References rzFilter_.

{ return rzFilter_.get(); }

trackCands3D()

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_;
      }
    }

Member Data Documentation

etaMaxSector_

float tmtt::Make3Dtracks::etaMaxSector_

private

Definition at line 84 of file Make3Dtracks.h.

Referenced by Make3Dtracks(), and makeUnfilteredTrks().

etaMinSector_

float tmtt::Make3Dtracks::etaMinSector_

private

Definition at line 83 of file Make3Dtracks.h.

Referenced by Make3Dtracks(), and makeUnfilteredTrks().

iEtaReg_

unsigned int tmtt::Make3Dtracks::iEtaReg_

private

Definition at line 82 of file Make3Dtracks.h.

Referenced by Make3Dtracks(), and makeUnfilteredTrks().

iPhiSec_

unsigned int tmtt::Make3Dtracks::iPhiSec_

private

Definition at line 81 of file Make3Dtracks.h.

Referenced by Make3Dtracks(), and makeUnfilteredTrks().

phiCentreSector_

float tmtt::Make3Dtracks::phiCentreSector_

private

Definition at line 85 of file Make3Dtracks.h.

Referenced by Make3Dtracks().

runRZfilter_

bool tmtt::Make3Dtracks::runRZfilter_

private

Definition at line 87 of file Make3Dtracks.h.

Referenced by Make3Dtracks(), ranRZfilter(), and run().

rzFilter_

std::unique_ptr<TrkRZfilter> tmtt::Make3Dtracks::rzFilter_

private

Definition at line 90 of file Make3Dtracks.h.

Referenced by Make3Dtracks(), makeRZfilteredTrks(), and rzFilter().

settings_

const Settings* tmtt::Make3Dtracks::settings_

private

Definition at line 80 of file Make3Dtracks.h.

Referenced by Make3Dtracks(), makeRZfilteredTrks(), and makeUnfilteredTrks().

vecTracks3D_rzFiltered_

std::list<L1track3D> tmtt::Make3Dtracks::vecTracks3D_rzFiltered_

private

Definition at line 93 of file Make3Dtracks.h.

Referenced by assocTrackCands3D(), makeRZfilteredTrks(), and trackCands3D().

vecTracks3D_unfiltered_

std::list<L1track3D> tmtt::Make3Dtracks::vecTracks3D_unfiltered_

private

Definition at line 94 of file Make3Dtracks.h.

Referenced by assocTrackCands3D(), makeUnfilteredTrks(), and trackCands3D().