#include <KFParamsComb.h>

Inheritance diagram for tmtt::KFParamsComb:

Public Member Functions
	KFParamsComb (const Settings *settings, const uint nHelixPar, const std::string &fitterName)

	~KFParamsComb () override=default

Public Member Functions inherited from tmtt::KFbase
L1fittedTrack	fit (const L1track3D &l1track3D) override

	KFbase (const Settings *settings, const uint nHelixPar, const std::string &fitterName="", const uint nMeas=2)

	KFbase (const KFbase &kf)=delete

	KFbase (KFbase &&kf)=delete

KFbase &	operator= (const KFbase &kf)=delete

KFbase &	operator= (KFbase &&kf)=delete

	~KFbase () override

Public Member Functions inherited from tmtt::TrackFitGeneric
	TrackFitGeneric (const Settings *settings, const std::string &fitterName="")

virtual	~TrackFitGeneric ()=default

Protected Member Functions
bool	isGoodState (const KalmanState &state) const override

TMatrixD	matrixF (const Stub stub, const KalmanState state) const override

TMatrixD	matrixH (const Stub *stub) const override

TMatrixD	matrixV (const Stub stub, const KalmanState state) const override

TMatrixD	seedC (const L1track3D &l1track3D) const override

TVectorD	seedX (const L1track3D &l1track3D) const override

TVectorD	trackParams (const KalmanState *state) const override

TVectorD	trackParams_BeamConstr (const KalmanState *state, double &chi2rphi) const override

TVectorD	vectorM (const Stub *stub) const override

Protected Member Functions inherited from tmtt::KFbase
virtual void	adjustChi2 (const KalmanState *state, const TMatrixD &covRinv, const TVectorD &delta, double &chi2rphi, double &chi2rz) const

void	adjustState (const TMatrixD &K, const TMatrixD &pxcov, const TVectorD &x, const TMatrixD &h, const TVectorD &delta, TVectorD &new_x, TMatrixD &new_xcov) const

float	approxB (float z, float r) const

const KalmanState *	doKF (const L1track3D &l1track3D, const std::vector< Stub > &stubs, const TP tpa)

TMatrixD	getKalmanGainMatrix (const TMatrixD &h, const TMatrixD &pxcov, const TMatrixD &covRinv) const

virtual bool	isHLS ()

virtual bool	kalmanAmbiguousLayer (unsigned int iEtaReg, unsigned int kfLayer)

std::set< unsigned >	kalmanDeadLayers (bool &remove2PSCut) const

virtual unsigned int	kalmanLayer (unsigned int iEtaReg, unsigned int layerIDreduced, bool barrel, float r, float z) const

virtual const KalmanState *	kalmanUpdate (unsigned nSkipped, unsigned layer, Stub stub, const KalmanState state, const TP *tp)

TMatrixD	matrixHCHt (const TMatrixD &h, const TMatrixD &c) const

TMatrixD	matrixRinv (const TMatrixD &matH, const TMatrixD &matCref, const TMatrixD &matV) const

const KalmanState *	mkState (const L1track3D &candidate, unsigned nSkipped, unsigned layer, const KalmanState last_state, const TVectorD &x, const TMatrixD &pxx, const TMatrixD &K, const TMatrixD &dcov, Stub stub, double chi2rphi, double chi2rz)

void	printStub (const Stub *stub) const

void	printStubLayers (const std::vector< Stub *> &stubs, unsigned int iEtaReg) const

void	printStubs (const std::vector< Stub *> &stubs) const

void	printTP (const TP *tp) const

void	resetStates ()

virtual TVectorD	residual (const Stub *stub, const TVectorD &x, double candQoverPt) const

double	sectorPhi () const

Protected Attributes
std::vector< double >	kfLayerVsChiSq4_

std::vector< double >	kfLayerVsChiSq5_

std::vector< double >	kfLayerVsD0Cut5_

std::vector< double >	kfLayerVsPtToler_

std::vector< double >	kfLayerVsZ0Cut4_

std::vector< double >	kfLayerVsZ0Cut5_

Protected Attributes inherited from tmtt::KFbase
unsigned int	iEtaReg_

unsigned int	iPhiSec_

std::vector< std::unique_ptr< const KalmanState > >	listAllStates_

unsigned	nHelixPar_

unsigned	nMeas_

unsigned	numEtaRegions_

unsigned int	numUpdateCalls_

const TP *	tpa_

Protected Attributes inherited from tmtt::TrackFitGeneric
const std::string	fitterName_

const Settings *	settings_

Additional Inherited Members
Public Types inherited from tmtt::KFbase
enum	MEAS_IDS { PHI, Z }

enum	PAR_IDS { INV2R, PHI0, T, Z0, D0 }

enum	PAR_IDS_VAR { QOVERPT }

Static Public Attributes inherited from tmtt::KFbase
static constexpr unsigned int	invalidKFlayer_ = nKFlayer_

static constexpr std::pair< unsigned, unsigned >	layerMap_ [nEta_/2][nGPlayer_+1]

static const unsigned int	nEta_ = 16

static const unsigned int	nGPlayer_ = 7

static const unsigned int	nKFlayer_ = 7

Detailed Description

Definition at line 35 of file KFParamsComb.h.

Constructor & Destructor Documentation

◆ KFParamsComb()

tmtt::KFParamsComb::KFParamsComb	(	const Settings *	settings,
		const uint	nHelixPar,
		const std::string &	fitterName
	)

Definition at line 17 of file KFParamsComb.cc.

       : KFbase(settings, nHelixPar, fitterName),
         // Initialize cuts applied to helix states vs KF layer number of last added stub.
         kfLayerVsPtToler_(settings->kfLayerVsPtToler()),
         kfLayerVsD0Cut5_(settings->kfLayerVsD0Cut5()),
         kfLayerVsZ0Cut5_(settings->kfLayerVsZ0Cut5()),
         kfLayerVsZ0Cut4_(settings->kfLayerVsZ0Cut4()),
         kfLayerVsChiSq5_(settings->kfLayerVsChiSq5()),
         kfLayerVsChiSq4_(settings->kfLayerVsChiSq4()) {}

◆ ~KFParamsComb()

tmtt::KFParamsComb::~KFParamsComb ( )

overridedefault

Member Function Documentation

◆ isGoodState()

bool tmtt::KFParamsComb::isGoodState ( const KalmanState & state ) const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 227 of file KFParamsComb.cc.

References funct::abs(), tmtt::KFbase::D0, d0, alignBH_cfg::fixed, tmtt::Settings::hybrid(), tmtt::Settings::kalmanDebugLevel(), kfLayerVsChiSq4_, kfLayerVsChiSq5_, kfLayerVsD0Cut5_, kfLayerVsPtToler_, kfLayerVsZ0Cut4_, kfLayerVsZ0Cut5_, tmtt::KFbase::nHelixPar_, tmtt::KFbase::PHI0, DiDispStaMuonMonitor_cfi::pt, tmtt::TP::pt(), tmtt::KFbase::QOVERPT, tmtt::TrackFitGeneric::settings_, tmtt::KFbase::T, submitPVValidationJobs::text, tmtt::KFbase::tpa_, trackParams(), and tmtt::KFbase::Z0.

                                                                {
     // Set cut values that are different for 4 & 5 param helix fits.
     vector<double> kfLayerVsZ0Cut = (nHelixPar_ == 5) ? kfLayerVsZ0Cut5_ : kfLayerVsZ0Cut4_;
     vector<double> kfLayerVsChiSqCut = (nHelixPar_ == 5) ? kfLayerVsChiSq5_ : kfLayerVsChiSq4_;
 
     unsigned nStubLayers = state.nStubLayers();
     bool goodState(true);
 
     TVectorD vecY = trackParams(&state);
     double qOverPt = vecY[QOVERPT];
     double pt = std::abs(1 / qOverPt);
     double z0 = std::abs(vecY[Z0]);
 
     // state parameter selections
 
     if (z0 > kfLayerVsZ0Cut[nStubLayers])
       goodState = false;
     if (pt < settings_->houghMinPt() - kfLayerVsPtToler_[nStubLayers])
       goodState = false;
     if (nHelixPar_ == 5) {  // fit without d0 constraint
       double d0 = std::abs(state.vectorX()[D0]);
       if (d0 > kfLayerVsD0Cut5_[nStubLayers])
         goodState = false;
     }
 
     // chi2 selection
 
     double chi2scaled = state.chi2scaled();  // chi2(r-phi) scaled down to improve electron performance.
 
     if (chi2scaled > kfLayerVsChiSqCut[nStubLayers])
       goodState = false;  // No separate pT selection needed
 
     const bool countUpdateCalls = false;  // Print statement to count calls to Updator.
 
     if (countUpdateCalls || (settings_->kalmanDebugLevel() >= 2 && tpa_ != nullptr) ||
         (settings_->kalmanDebugLevel() >= 2 && settings_->hybrid())) {
       std::stringstream text;
       text << std::fixed << std::setprecision(4);
       if (not goodState)
         text << "State veto:";
       if (goodState)
         text << "State kept:";
       text << " nlay=" << nStubLayers << " nskip=" << state.nSkippedLayers() << " chi2_scaled=" << chi2scaled;
       if (tpa_ != nullptr)
         text << " pt(mc)=" << tpa_->pt();
       text << " pt=" << pt << " q/pt=" << qOverPt << " tanL=" << vecY[T] << " z0=" << vecY[Z0]
            << " phi0=" << vecY[PHI0];
       if (nHelixPar_ == 5)  // fit without d0 constraint
         text << " d0=" << vecY[D0];
       text << " fake" << (tpa_ == nullptr);
       if (tpa_ != nullptr)
         text << " pt(mc)=" << tpa_->pt();
       PrintL1trk() << text.str();
     }
 
     return goodState;
   }

◆ matrixF()

TMatrixD tmtt::KFParamsComb::matrixF	(	const Stub *	stub,
		const KalmanState *	state
	)		const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 180 of file KFParamsComb.cc.

References tmtt::KFbase::nHelixPar_.

                                                                                  {
     const TMatrixD unitMatrix(TMatrixD::kUnit, TMatrixD(nHelixPar_, nHelixPar_));
     return unitMatrix;
   }

◆ matrixH()

TMatrixD tmtt::KFParamsComb::matrixH ( const Stub * stub ) const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 165 of file KFParamsComb.cc.

References tmtt::KFbase::D0, tmtt::KFbase::INV2R, tmtt::KFbase::nHelixPar_, tmtt::KFbase::PHI, tmtt::KFbase::PHI0, tmtt::Stub::r(), tmtt::KFbase::Z, and tmtt::KFbase::Z0.

                                                        {
     TMatrixD matH(2, nHelixPar_);
     double r = stub->r();
     matH(PHI, INV2R) = -r;
     matH(PHI, PHI0) = 1;
     if (nHelixPar_ == 5) {  // fit without d0 constraint
       matH(PHI, D0) = -1. / r;
     }
     matH(Z, Z0) = 1;
     matH(Z, T) = r;
     return matH;
   }

◆ matrixV()

TMatrixD tmtt::KFParamsComb::matrixV	(	const Stub *	stub,
		const KalmanState *	state
	)		const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 80 of file KFParamsComb.cc.

References a, tmtt::Stub::alpha(), tmtt::KFbase::approxB(), b, tmtt::Stub::barrel(), HLT_2023v12_cff::beta, funct::cos(), tmtt::Settings::invPtToInvR(), tmtt::Settings::kalmanHOalpha(), tmtt::Settings::kalmanHOfw(), tmtt::Settings::kalmanHOprojZcorr(), tmtt::Settings::kalmanHOtilted(), tmtt::Settings::kalmanMultiScattTerm(), tmtt::KFbase::PHI, tmtt::Stub::psModule(), tmtt::Stub::r(), diffTwoXMLs::r2, tmtt::TrackFitGeneric::settings_, tmtt::Stub::sigmaPar(), tmtt::Stub::sigmaPerp(), funct::sin(), tmtt::KFbase::T, tmtt::Stub::tiltAngle(), tmtt::Stub::tiltedBarrel(), tmtt::Settings::useApproxB(), run3scouting_cff::vz, tmtt::KFbase::Z, and tmtt::Stub::z().

                                                                                  {
     // Take Pt from input track candidate as more stable.
     double inv2R = (settings_->invPtToInvR()) * 0.5 * state->candidate().qOverPt();
     double inv2R2 = inv2R * inv2R;
 
     double tanl = state->vectorX()(T);  // factor of 0.9 improves rejection
     double tanl2 = tanl * tanl;
 
     double vphi(0);
     double vz(0);
     double vcorr(0);
 
     double a = stub->sigmaPerp() * stub->sigmaPerp();
     double b = stub->sigmaPar() * stub->sigmaPar();
     double r2 = stub->r() * stub->r();
     double invr2 = 1. / r2;
 
     // Scattering term scaling as 1/Pt.
     double sigmaScat = settings_->kalmanMultiScattTerm() / (state->candidate().pt());
     double sigmaScat2 = sigmaScat * sigmaScat;
 
     if (stub->barrel()) {
       vphi = (a * invr2) + sigmaScat2;
 
       if (stub->tiltedBarrel()) {
         // Convert uncertainty in (r,phi) to (z,phi).
         float scaleTilted = 1.;
         if (settings_->kalmanHOtilted()) {
           if (settings_->useApproxB()) {  // Simple firmware approximation
             scaleTilted = approxB(stub->z(), stub->r());
           } else {  // Exact C++ implementation.
             float tilt = stub->tiltAngle();
             scaleTilted = sin(tilt) + cos(tilt) * tanl;
           }
         }
         float scaleTilted2 = scaleTilted * scaleTilted;
         // This neglects the non-radial strip effect, assumed negligeable for PS.
         vz = b * scaleTilted2;
       } else {
         vz = b;
       }
 
       if (settings_->kalmanHOfw()) {
         // Use approximation corresponding to current firmware.
         vz = b;
       }
 
     } else {
       vphi = a * invr2 + sigmaScat2;
       vz = (b * tanl2);
 
       if (not stub->psModule()) {  // Neglect these terms in PS
         double beta = 0.;
         // Add correlation term related to conversion of stub residuals from (r,phi) to (z,phi).
         if (settings_->kalmanHOprojZcorr() == 2)
           beta += -inv2R;
         // Add alpha correction for non-radial 2S endcap strips..
         if (settings_->kalmanHOalpha() == 2)
           beta += -stub->alpha();  // alpha is 0 except in endcap 2S disks
 
         double beta2 = beta * beta;
         vphi += b * beta2;
         vcorr = b * (beta * tanl);
 
         if (settings_->kalmanHOfw()) {
           // Use approximation corresponding to current firmware.
           vphi = (a * invr2) + (b * inv2R2) + sigmaScat2;
           vcorr = 0.;
           vz = (b * tanl2);
         }
       }
     }
 
     TMatrixD matV(2, 2);
     matV(PHI, PHI) = vphi;
     matV(Z, Z) = vz;
     matV(PHI, Z) = vcorr;
     matV(Z, PHI) = vcorr;
 
     return matV;
   }

◆ seedC()

TMatrixD tmtt::KFParamsComb::seedC ( const L1track3D & l1track3D ) const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 44 of file KFParamsComb.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), tmtt::KFbase::D0, tmtt::KFbase::INV2R, tmtt::Settings::invPtToInvR(), tmtt::KFbase::nHelixPar_, tmtt::KFbase::PHI0, funct::pow(), tmtt::TrackFitGeneric::settings_, tmtt::KFbase::T, and tmtt::KFbase::Z0.

                                                                {
     TMatrixD matC(nHelixPar_, nHelixPar_);
 
     double invPtToInv2R = settings_->invPtToInvR() / 2;
 
     // Assumed track seed (from HT) uncertainty in transverse impact parameter.
 
     // Constants optimised by hand for TMTT algo.
     const float inv2Rsigma = 0.0314 * invPtToInv2R;
     constexpr float phi0sigma = 0.0102;
     constexpr float z0sigma = 5.0;
     constexpr float tanLsigma = 0.5;
     constexpr float d0Sigma = 1.0;
     // (z0, tanL, d0) uncertainties could be smaller for Hybrid, if seeded in PS? -- To check!
     // if (L1track3D.seedPS() > 0) z0sigma /= 4; ???
     matC[INV2R][INV2R] = pow(inv2Rsigma, 2);
     matC[PHI0][PHI0] = pow(phi0sigma, 2);
     matC[Z0][Z0] = pow(z0sigma, 2);
     matC[T][T] = pow(tanLsigma, 2);
     if (nHelixPar_ == 5) {  // fit without d0 constraint
       matC[D0][D0] = pow(d0Sigma, 2);
     }
     return matC;
   }

◆ seedX()

TVectorD tmtt::KFParamsComb::seedX ( const L1track3D & l1track3D ) const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 29 of file KFParamsComb.cc.

References tmtt::KFbase::D0, tmtt::L1track3D::d0(), reco::deltaPhi(), tmtt::KFbase::INV2R, tmtt::Settings::invPtToInvR(), tmtt::KFbase::nHelixPar_, tmtt::KFbase::PHI0, tmtt::L1track3D::phi0(), tmtt::L1track3D::qOverPt(), tmtt::KFbase::sectorPhi(), tmtt::TrackFitGeneric::settings_, tmtt::KFbase::T, tmtt::L1track3D::tanLambda(), tmtt::KFbase::Z0, and tmtt::L1track3D::z0().

                                                                {
     TVectorD vecX(nHelixPar_);
     vecX[INV2R] = settings_->invPtToInvR() * l1track3D.qOverPt() / 2;
     vecX[PHI0] = reco::deltaPhi(l1track3D.phi0() - sectorPhi(), 0.);
     vecX[Z0] = l1track3D.z0();
     vecX[T] = l1track3D.tanLambda();
     if (nHelixPar_ == 5) {  // fit without d0 constraint
       vecX[D0] = l1track3D.d0();
     }
 
     return vecX;
   }

◆ trackParams()

TVectorD tmtt::KFParamsComb::trackParams ( const KalmanState * state ) const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 187 of file KFParamsComb.cc.

References tmtt::KFbase::D0, reco::deltaPhi(), tmtt::KFbase::INV2R, tmtt::Settings::invPtToInvR(), tmtt::KFbase::nHelixPar_, tmtt::KFbase::PHI0, tmtt::KFbase::QOVERPT, tmtt::KFbase::sectorPhi(), tmtt::TrackFitGeneric::settings_, tmtt::KFbase::T, and tmtt::KFbase::Z0.

Referenced by isGoodState(), and trackParams_BeamConstr().

                                                                    {
     TVectorD vecX = state->vectorX();
     TVectorD vecY(nHelixPar_);
     vecY[QOVERPT] = 2. * vecX[INV2R] / settings_->invPtToInvR();
     vecY[PHI0] = reco::deltaPhi(vecX[PHI0] + sectorPhi(), 0.);
     vecY[Z0] = vecX[Z0];
     vecY[T] = vecX[T];
     if (nHelixPar_ == 5) {  // fit without d0 constraint
       vecY[D0] = vecX[D0];
     }
     return vecY;
   }

◆ trackParams_BeamConstr()

TVectorD tmtt::KFParamsComb::trackParams_BeamConstr	(	const KalmanState *	state,
		double &	chi2rphi
	)		const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 203 of file KFParamsComb.cc.

References tmtt::KFbase::D0, reco::deltaPhi(), tmtt::KFbase::INV2R, tmtt::Settings::invPtToInvR(), tmtt::KFbase::nHelixPar_, tmtt::KFbase::PHI0, tmtt::KFbase::QOVERPT, tmtt::KFbase::sectorPhi(), tmtt::TrackFitGeneric::settings_, tmtt::KFbase::T, trackParams(), and tmtt::KFbase::Z0.

                                                                                                 {
     if (nHelixPar_ == 5) {  // fit without d0 constraint
       TVectorD vecX = state->vectorX();
       TMatrixD matC = state->matrixC();
       TVectorD vecY(nHelixPar_);
       double delChi2rphi = (vecX[D0] * vecX[D0]) / matC[D0][D0];
       chi2rphi = state->chi2rphi() + delChi2rphi;
       // Apply beam-spot constraint to helix params in transverse plane only, as most sensitive to it.
       vecX[INV2R] -= vecX[D0] * (matC[INV2R][D0] / matC[D0][D0]);
       vecX[PHI0] -= vecX[D0] * (matC[PHI0][D0] / matC[D0][D0]);
       vecX[D0] = 0.0;
       vecY[QOVERPT] = 2. * vecX[INV2R] / settings_->invPtToInvR();
       vecY[PHI0] = reco::deltaPhi(vecX[PHI0] + sectorPhi(), 0.);
       vecY[Z0] = vecX[Z0];
       vecY[T] = vecX[T];
       vecY[D0] = vecX[D0];
       return vecY;
     } else {
       return (this->trackParams(state));
     }
   }

◆ vectorM()

TVectorD tmtt::KFParamsComb::vectorM ( const Stub * stub ) const

overrideprotectedvirtual

Implements tmtt::KFbase.

Definition at line 71 of file KFParamsComb.cc.

References reco::deltaPhi(), tmtt::KFbase::PHI, tmtt::Stub::phi(), tmtt::KFbase::sectorPhi(), tmtt::KFbase::Z, and tmtt::Stub::z().

                                                        {
     TVectorD meas(2);
     meas[PHI] = reco::deltaPhi(stub->phi(), sectorPhi());
     meas[Z] = stub->z();
     return meas;
   }

Member Data Documentation

◆ kfLayerVsChiSq4_

std::vector<double> tmtt::KFParamsComb::kfLayerVsChiSq4_

protected

Definition at line 72 of file KFParamsComb.h.

Referenced by isGoodState().

◆ kfLayerVsChiSq5_

std::vector<double> tmtt::KFParamsComb::kfLayerVsChiSq5_

protected

Definition at line 71 of file KFParamsComb.h.

Referenced by isGoodState().

◆ kfLayerVsD0Cut5_

std::vector<double> tmtt::KFParamsComb::kfLayerVsD0Cut5_

protected

Definition at line 68 of file KFParamsComb.h.

Referenced by isGoodState().

◆ kfLayerVsPtToler_

std::vector<double> tmtt::KFParamsComb::kfLayerVsPtToler_

protected

Definition at line 67 of file KFParamsComb.h.

Referenced by isGoodState().

◆ kfLayerVsZ0Cut4_

std::vector<double> tmtt::KFParamsComb::kfLayerVsZ0Cut4_

protected

Definition at line 70 of file KFParamsComb.h.

Referenced by isGoodState().

◆ kfLayerVsZ0Cut5_

std::vector<double> tmtt::KFParamsComb::kfLayerVsZ0Cut5_

protected

Definition at line 69 of file KFParamsComb.h.

Referenced by isGoodState().

Public Member Functions

Protected Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ KFParamsComb()

◆ ~KFParamsComb()

Member Function Documentation

◆ isGoodState()

◆ matrixF()

◆ matrixH()

◆ matrixV()

◆ seedC()

◆ seedX()

◆ trackParams()

◆ trackParams_BeamConstr()

◆ vectorM()

Member Data Documentation

◆ kfLayerVsChiSq4_

◆ kfLayerVsChiSq5_

◆ kfLayerVsD0Cut5_

◆ kfLayerVsPtToler_

◆ kfLayerVsZ0Cut4_

◆ kfLayerVsZ0Cut5_