tmtt::ChiSquaredFitBase Class Reference

#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.

{ 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_, validateGeometry_cfg::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.