#include <ChiSquaredFitBase.h>

Inheritance diagram for tmtt::ChiSquaredFitBase:

Public Types
enum	PAR_IDS { INVR, PHI0, T, Z0, D0 }

Public Member Functions
	ChiSquaredFitBase (const Settings *settings, const uint nPar)

L1fittedTrack	fit (const L1track3D &l1track3D) override

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

virtual	~TrackFitGeneric ()=default

Protected Member Functions
virtual TMatrixD	D (const TVectorD &x)=0

virtual TVectorD	residuals (const TVectorD &x)=0

virtual TVectorD	seed (const L1track3D &l1track3D)=0

virtual TMatrixD	Vinv ()=0

Protected Attributes
double	chiSq_

int	ilargestresid_

float	largestresid_

uint	nPar_

double	qOverPt_seed_

std::vector< Stub * >	stubs_

TVectorD	trackParams_

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

const Settings *	settings_

Private Member Functions
void	calculateChiSq (const TVectorD &resids)

void	calculateDeltaChiSq (const TVectorD &deltaX, const TVectorD &covX)

Private Attributes
double	generalResidualCut_

double	killingResidualCut_

int	killTrackFitWorstHit_

unsigned int	minStubLayers_

unsigned int	minStubLayersRed_

int	numFittingIterations_

Detailed Description

Definition at line 21 of file ChiSquaredFitBase.h.

Member Enumeration Documentation

◆ PAR_IDS

enum tmtt::ChiSquaredFitBase::PAR_IDS

Enumerator
INVR
PHI0
T
Z0
D0

Definition at line 23 of file ChiSquaredFitBase.h.

23 { INVR, PHI0, T, Z0, D0 };

Constructor & Destructor Documentation

◆ ChiSquaredFitBase()

tmtt::ChiSquaredFitBase::ChiSquaredFitBase	(	const Settings *	settings,
		const uint	nPar
	)

Definition at line 17 of file ChiSquaredFitBase.cc.

       : TrackFitGeneric(settings), chiSq_(0.0) {
     // Bad stub killing settings
     numFittingIterations_ = settings_->numTrackFitIterations();
     killTrackFitWorstHit_ = settings_->killTrackFitWorstHit();
     generalResidualCut_ = settings_->generalResidualCut();  // The cut used to remove bad stubs (if nStubs > minLayers)
     killingResidualCut_ = settings_->killingResidualCut();  // The cut used to kill off tracks entirely
  
     //--- These two parameters are used to check if after the fit, there are still enough stubs on the track
     minStubLayers_ = settings_->minStubLayers();
     nPar_ = nPar;
   }

References tmtt::Settings::generalResidualCut(), generalResidualCut_, tmtt::Settings::killingResidualCut(), killingResidualCut_, tmtt::Settings::killTrackFitWorstHit(), killTrackFitWorstHit_, tmtt::Settings::minStubLayers(), minStubLayers_, HcalResponse_cfi::nPar, nPar_, numFittingIterations_, tmtt::Settings::numTrackFitIterations(), and tmtt::TrackFitGeneric::settings_.

Member Function Documentation

◆ calculateChiSq()

void tmtt::ChiSquaredFitBase::calculateChiSq ( const TVectorD & resids )

private

Definition at line 30 of file ChiSquaredFitBase.cc.

                                                                {
     chiSq_ = 0.0;
     uint j = 0;
     for (uint i = 0; i < stubs_.size(); i++) {
       chiSq_ += resids[j] * resids[j] + resids[j + 1] * resids[j + 1];
       j = j + 2;
     }
   }

References chiSq_, mps_fire::i, dqmiolumiharvest::j, stubs_, and parallelization::uint.

Referenced by fit().

◆ calculateDeltaChiSq()

void tmtt::ChiSquaredFitBase::calculateDeltaChiSq	(	const TVectorD &	deltaX,
		const TVectorD &	covX
	)

private

Definition at line 39 of file ChiSquaredFitBase.cc.

                                                                                         {
     for (int i = 0; i < covX.GetNrows(); i++) {
       chiSq_ -= (delX[i]) * covX[i];
     }
   }

References chiSq_, and mps_fire::i.

Referenced by fit().

◆ D()

virtual TMatrixD tmtt::ChiSquaredFitBase::D ( const TVectorD & x )

protectedpure virtual

Implemented in tmtt::ChiSquaredFit4.

Referenced by fit().

◆ fit()

L1fittedTrack tmtt::ChiSquaredFitBase::fit ( const L1track3D & l1track3D )

overridevirtual

Reimplemented from tmtt::TrackFitGeneric.

Definition at line 45 of file ChiSquaredFitBase.cc.

                                                                  {
     qOverPt_seed_ = l1track3D.qOverPt();
     stubs_ = l1track3D.stubs();
  
     // Get cut on number of layers including variation due to dead sectors, pt dependence etc.
     minStubLayersRed_ = Utility::numLayerCut(Utility::AlgoStep::FIT,
                                              settings_,
                                              l1track3D.iPhiSec(),
                                              l1track3D.iEtaReg(),
                                              std::abs(l1track3D.qOverPt()),
                                              l1track3D.eta());
  
     TVectorD x = seed(l1track3D);
  
     for (int i = 0; i < numFittingIterations_; i++) {
       TMatrixD d = D(x);
       TMatrixD dTrans(TMatrixD::kTransposed, d);
       TMatrixD dtVinv = dTrans * Vinv();
       TMatrixD dtVinvTrans(TMatrixD::kTransposed, dtVinv);
       //TMatrixD M = dtVinv * d; // Must insert extra factor Vinv, due to unconventional Vinv() definition.
       TMatrixD M = dtVinv * dtVinvTrans;
       TMatrixD Minv(TMatrixD::kInverted, M);
       TVectorD resids = residuals(x);
       TVectorD deltaX = Minv * dtVinv * resids;
       x = x - deltaX;
       TVectorD covX = dTrans * Vinv() * resids;
       calculateChiSq(resids);
       calculateDeltaChiSq(deltaX, covX);
  
       if (i < numFittingIterations_ - 1) {  // Don't kill stub if will not refit.
  
         resids = residuals(x);  // update resids & largestresid_
  
         bool killWorstStub = false;
         if (killTrackFitWorstHit_) {
           if (largestresid_ > killingResidualCut_) {
             killWorstStub = true;
           } else if (largestresid_ > generalResidualCut_) {
             std::vector<Stub*> stubsTmp = stubs_;
             stubsTmp.erase(stubsTmp.begin() + ilargestresid_);
             if (Utility::countLayers(settings_, stubsTmp) >= minStubLayersRed_)
               killWorstStub = true;
           } else {
             // Get better apparent tracking performance by always killing worst stub until only 4 layers left.
             if (Utility::countLayers(settings_, stubs_) > minStubLayersRed_)
               killWorstStub = true;
           }
         }
  
         if (killWorstStub) {
           stubs_.erase(stubs_.begin() + ilargestresid_);
  
           // Reject tracks with too many killed stubs & stop iterating.
           unsigned int nLayers = Utility::countLayers(settings_, stubs_);  // Count tracker layers with stubs
           bool valid = nLayers >= minStubLayersRed_;
  
           if (not valid) {
             L1fittedTrack rejectedTrk;
             return rejectedTrk;
           }
         } else {
           break;
         }
       }
     }
  
     // Reject tracks with too many killed stubs
     unsigned int nLayers = Utility::countLayers(settings_, stubs_);  // Count tracker layers with stubs
     bool valid = nLayers >= minStubLayersRed_;
  
     if (valid) {
       const unsigned int hitPattern = 0;  // FIX: Needs setting
       const float chi2rz = 0;             // FIX: Needs setting
       return L1fittedTrack(settings_,
                            &l1track3D,
                            stubs_,
                            hitPattern,
                            x[INVR] / (settings_->invPtToInvR()),
                            0,
                            x[PHI0],
                            x[Z0],
                            x[T],
                            chiSq_,
                            chi2rz,
                            nPar_);
     } else {
       L1fittedTrack rejectedTrk;
       return rejectedTrk;
     }
   }

References funct::abs(), calculateChiSq(), calculateDeltaChiSq(), chiSq_, tmtt::Utility::countLayers(), D(), ztail::d, tmtt::L1track3D::eta(), tmtt::Utility::FIT, generalResidualCut_, mps_fire::i, tmtt::L1track3D::iEtaReg(), ilargestresid_, tmtt::Settings::invPtToInvR(), INVR, tmtt::L1track3D::iPhiSec(), killingResidualCut_, killTrackFitWorstHit_, largestresid_, minStubLayersRed_, MuonTCMETValueMapProducer_cff::nLayers, nPar_, numFittingIterations_, tmtt::Utility::numLayerCut(), PHI0, tmtt::L1track3D::qOverPt(), qOverPt_seed_, residuals(), seed(), tmtt::TrackFitGeneric::settings_, tmtt::L1track3D::stubs(), stubs_, RunInfoPI::valid, Vinv(), and Z0.

Referenced by trackingPlots.Iteration::modules().

◆ residuals()

virtual TVectorD tmtt::ChiSquaredFitBase::residuals ( const TVectorD & x )

protectedpure virtual

Implemented in tmtt::ChiSquaredFit4.

Referenced by fit().

◆ seed()

virtual TVectorD tmtt::ChiSquaredFitBase::seed ( const L1track3D & l1track3D )

protectedpure virtual

Implemented in tmtt::ChiSquaredFit4.

Referenced by fit().

◆ Vinv()

virtual TMatrixD tmtt::ChiSquaredFitBase::Vinv ( )

protectedpure virtual

Implemented in tmtt::ChiSquaredFit4.

Referenced by fit().

Member Data Documentation

◆ chiSq_

double tmtt::ChiSquaredFitBase::chiSq_

protected

Definition at line 44 of file ChiSquaredFitBase.h.

Referenced by calculateChiSq(), calculateDeltaChiSq(), and fit().

◆ generalResidualCut_

double tmtt::ChiSquaredFitBase::generalResidualCut_

private

Definition at line 52 of file ChiSquaredFitBase.h.

Referenced by ChiSquaredFitBase(), and fit().

◆ ilargestresid_

int tmtt::ChiSquaredFitBase::ilargestresid_

protected

Definition at line 43 of file ChiSquaredFitBase.h.

Referenced by tmtt::ChiSquaredFit4::ChiSquaredFit4(), fit(), and tmtt::ChiSquaredFit4::residuals().

◆ killingResidualCut_

double tmtt::ChiSquaredFitBase::killingResidualCut_

private

Definition at line 53 of file ChiSquaredFitBase.h.

Referenced by ChiSquaredFitBase(), and fit().

◆ killTrackFitWorstHit_

int tmtt::ChiSquaredFitBase::killTrackFitWorstHit_

private

Definition at line 51 of file ChiSquaredFitBase.h.

Referenced by ChiSquaredFitBase(), and fit().

◆ largestresid_

float tmtt::ChiSquaredFitBase::largestresid_

protected

Definition at line 42 of file ChiSquaredFitBase.h.

Referenced by tmtt::ChiSquaredFit4::ChiSquaredFit4(), fit(), and tmtt::ChiSquaredFit4::residuals().

◆ minStubLayers_

unsigned int tmtt::ChiSquaredFitBase::minStubLayers_

private

Definition at line 55 of file ChiSquaredFitBase.h.

Referenced by ChiSquaredFitBase().

◆ minStubLayersRed_

unsigned int tmtt::ChiSquaredFitBase::minStubLayersRed_

private

Definition at line 56 of file ChiSquaredFitBase.h.

Referenced by fit().

◆ nPar_

uint tmtt::ChiSquaredFitBase::nPar_

protected

Definition at line 41 of file ChiSquaredFitBase.h.

Referenced by ChiSquaredFitBase(), tmtt::ChiSquaredFit4::D(), and fit().

◆ numFittingIterations_

int tmtt::ChiSquaredFitBase::numFittingIterations_

private

Definition at line 50 of file ChiSquaredFitBase.h.

Referenced by ChiSquaredFitBase(), and fit().

◆ qOverPt_seed_

double tmtt::ChiSquaredFitBase::qOverPt_seed_

protected

Definition at line 38 of file ChiSquaredFitBase.h.

Referenced by fit(), tmtt::ChiSquaredFit4::residuals(), and tmtt::ChiSquaredFit4::Vinv().

◆ stubs_

std::vector<Stub*> tmtt::ChiSquaredFitBase::stubs_

protected

Definition at line 39 of file ChiSquaredFitBase.h.

Referenced by calculateChiSq(), tmtt::ChiSquaredFit4::D(), fit(), tmtt::ChiSquaredFit4::residuals(), and tmtt::ChiSquaredFit4::Vinv().

◆ trackParams_

TVectorD tmtt::ChiSquaredFitBase::trackParams_

protected

Definition at line 40 of file ChiSquaredFitBase.h.

Public Types

Public Member Functions

Protected Member Functions

Protected Attributes

Private Member Functions

Private Attributes

Detailed Description

Member Enumeration Documentation

◆ PAR_IDS

Constructor & Destructor Documentation

◆ ChiSquaredFitBase()

Member Function Documentation

◆ calculateChiSq()

◆ calculateDeltaChiSq()

◆ D()

◆ fit()

◆ residuals()

◆ seed()

◆ Vinv()

Member Data Documentation

◆ chiSq_

◆ generalResidualCut_

◆ ilargestresid_

◆ killingResidualCut_

◆ killTrackFitWorstHit_

◆ largestresid_

◆ minStubLayers_

◆ minStubLayersRed_

◆ nPar_

◆ numFittingIterations_

◆ qOverPt_seed_

◆ stubs_

◆ trackParams_