#include <TtSemiLepKinFitter.h>

Inheritance diagram for TtSemiLepKinFitter:

Public Types
enum	Constraint { kWHadMass = 1, kWLepMass, kTopHadMass, kTopLepMass, kNeutrinoMass, kEqualTopMasses, kSumPt }
	supported constraints More...

Public Types inherited from TopKinFitter
enum	Param { kEMom, kEtEtaPhi, kEtThetaPhi }
	supported parameterizations More...

Public Member Functions
TtSemiEvtSolution	addKinFitInfo (TtSemiEvtSolution *asol)
	add kin fit information to the old event solution (in for legacy reasons) More...

template<class LeptonType >
int	fit (const std::vector< pat::Jet > &jets, const pat::Lepton< LeptonType > &leps, const pat::MET &met)
	kinematic fit interface for PAT objects More...

int	fit (const TLorentzVector &p4HadP, const TLorentzVector &p4HadQ, const TLorentzVector &p4HadB, const TLorentzVector &p4LepB, const TLorentzVector &p4Lepton, const TLorentzVector &p4Neutrino, const int leptonCharge, const CovarianceMatrix::ObjectType leptonType)
	kinematic fit interface for plain 4-vecs More...

int	fit (const TLorentzVector &p4HadP, const TLorentzVector &p4HadQ, const TLorentzVector &p4HadB, const TLorentzVector &p4LepB, const TLorentzVector &p4Lepton, const TLorentzVector &p4Neutrino, const TMatrixD &covHadP, const TMatrixD &covHadQ, const TMatrixD &covHadB, const TMatrixD &covLepB, const TMatrixD &covLepton, const TMatrixD &covNeutrino, const int leptonCharge)
	common core of the fit interface More...

const pat::Particle	fittedHadB () const
	return hadronic b quark candidate More...

const pat::Particle	fittedHadP () const
	return hadronic light quark candidate More...

const pat::Particle	fittedHadQ () const
	return hadronic light quark candidate More...

const pat::Particle	fittedLepB () const
	return leptonic b quark candidate More...

const pat::Particle	fittedLepton () const
	return lepton candidate More...

const pat::Particle	fittedNeutrino () const
	return neutrino candidate More...

	TtSemiLepKinFitter ()
	default constructor More...

	TtSemiLepKinFitter (Param jetParam, Param lepParam, Param metParam, int maxNrIter, double maxDeltaS, double maxF, const std::vector< Constraint > &constraints, double mW=80.4, double mTop=173., const std::vector< edm::ParameterSet > udscResolutions=nullptr, const std::vector< edm::ParameterSet > bResolutions=nullptr, const std::vector< edm::ParameterSet > lepResolutions=nullptr, const std::vector< edm::ParameterSet > metResolutions=nullptr, const std::vector< double > jetEnergyResolutionScaleFactors=nullptr, const std::vector< double > jetEnergyResolutionEtaBinning=nullptr)
	constructor initialized with built-in types and class enum's custom parameters More...

	~TtSemiLepKinFitter ()
	default destructor More...

Public Member Functions inherited from TopKinFitter
int	fitNrIter () const
	return number of used iterations More...

double	fitProb () const
	return fit probability More...

double	fitS () const
	return chi2 of fit (not normalized to degrees of freedom) More...

void	setVerbosity (const int verbosityLevel)
	allows to change the verbosity of the TKinFitter More...

	TopKinFitter (const int maxNrIter=200, const double maxDeltaS=5e-5, const double maxF=1e-4, const double mW=80.4, const double mTop=173.)
	default constructor More...

	~TopKinFitter ()
	default destructor More...

Private Member Functions
void	printSetup () const
	print fitter setup More...

void	setupConstraints ()
	initialize constraints More...

void	setupFitter ()
	setup fitter More...

void	setupJets ()
	initialize jet inputs More...

void	setupLeptons ()
	initialize lepton inputs More...

Private Attributes
const std::vector< edm::ParameterSet > *	bResolutions_

bool	constrainSumPt_
	internally use simple boolean for this constraint to reduce the per-event computing time More...

std::vector< Constraint >	constrList_
	vector of constraints to be used More...

CovarianceMatrix *	covM_
	object used to construct the covariance matrices for the individual particles More...

pat::Particle	fittedHadB_
	output particles More...

pat::Particle	fittedHadP_

pat::Particle	fittedHadQ_

pat::Particle	fittedLepB_

pat::Particle	fittedLepton_

pat::Particle	fittedNeutrino_

TAbsFitParticle *	hadB_
	input particles More...

TAbsFitParticle *	hadP_

TAbsFitParticle *	hadQ_

const std::vector< double > *	jetEnergyResolutionEtaBinning_

const std::vector< double > *	jetEnergyResolutionScaleFactors_
	scale factors for the jet energy resolution More...

Param	jetParam_
	jet parametrization More...

TAbsFitParticle *	lepB_

Param	lepParam_
	lepton parametrization More...

const std::vector< edm::ParameterSet > *	lepResolutions_

TAbsFitParticle *	lepton_

std::map< Constraint, TFitConstraintM * >	massConstr_
	supported constraints More...

Param	metParam_
	met parametrization More...

const std::vector< edm::ParameterSet > *	metResolutions_

TAbsFitParticle *	neutrino_

TFitConstraintEp *	sumPxConstr_

TFitConstraintEp *	sumPyConstr_

const std::vector< edm::ParameterSet > *	udscResolutions_
	resolutions More...

Additional Inherited Members
Protected Member Functions inherited from TopKinFitter
std::string	param (const Param &param) const
	convert Param to human readable form More...

Protected Attributes inherited from TopKinFitter
TKinFitter *	fitter_
	kinematic fitter More...

double	maxDeltaS_
	maximal allowed chi2 (not normalized to degrees of freedom) More...

double	maxF_
	maximal allowed distance from constraints More...

int	maxNrIter_
	maximal allowed number of iterations to be used for the fit More...

double	mTop_
	top mass value used for constraints More...

double	mW_
	W mass value used for constraints. More...

Detailed Description

Definition at line 32 of file TtSemiLepKinFitter.h.

Member Enumeration Documentation

◆ Constraint

enum TtSemiLepKinFitter::Constraint

supported constraints

Enumerator
kWHadMass
kWLepMass
kTopHadMass
kTopLepMass
kNeutrinoMass
kEqualTopMasses
kSumPt

Definition at line 35 of file TtSemiLepKinFitter.h.

35 { kWHadMass = 1, kWLepMass, kTopHadMass, kTopLepMass, kNeutrinoMass, kEqualTopMasses, kSumPt };

Constructor & Destructor Documentation

◆ TtSemiLepKinFitter() [1/2]

TtSemiLepKinFitter::TtSemiLepKinFitter ( )

explicit

default constructor

default configuration is: Parametrization kEMom, Max iterations = 200, deltaS<= 5e-5, maxF<= 1e-4, no constraints

Definition at line 15 of file TtSemiLepKinFitter.cc.

     : TopKinFitter(),
       hadB_(nullptr),
       hadP_(nullptr),
       hadQ_(nullptr),
       lepB_(nullptr),
       lepton_(nullptr),
       neutrino_(nullptr),
       udscResolutions_(nullptr),
       bResolutions_(nullptr),
       lepResolutions_(nullptr),
       metResolutions_(nullptr),
       jetEnergyResolutionScaleFactors_(nullptr),
       jetEnergyResolutionEtaBinning_(nullptr),
       jetParam_(kEMom),
       lepParam_(kEMom),
       metParam_(kEMom) {
   setupFitter();
 }

References setupFitter().

◆ TtSemiLepKinFitter() [2/2]

TtSemiLepKinFitter::TtSemiLepKinFitter	(	Param	jetParam,
		Param	lepParam,
		Param	metParam,
		int	maxNrIter,
		double	maxDeltaS,
		double	maxF,
		const std::vector< Constraint > &	constraints,
		double	mW = `80.4`,
		double	mTop = `173.`,
		const std::vector< edm::ParameterSet > *	udscResolutions = `nullptr`,
		const std::vector< edm::ParameterSet > *	bResolutions = `nullptr`,
		const std::vector< edm::ParameterSet > *	lepResolutions = `nullptr`,
		const std::vector< edm::ParameterSet > *	metResolutions = `nullptr`,
		const std::vector< double > *	jetEnergyResolutionScaleFactors = `nullptr`,
		const std::vector< double > *	jetEnergyResolutionEtaBinning = `nullptr`
	)

explicit

constructor initialized with built-in types and class enum's custom parameters

Definition at line 35 of file TtSemiLepKinFitter.cc.

     : TopKinFitter(maxNrIter, maxDeltaS, maxF, mW, mTop),
       hadB_(nullptr),
       hadP_(nullptr),
       hadQ_(nullptr),
       lepB_(nullptr),
       lepton_(nullptr),
       neutrino_(nullptr),
       udscResolutions_(udscResolutions),
       bResolutions_(bResolutions),
       lepResolutions_(lepResolutions),
       metResolutions_(metResolutions),
       jetEnergyResolutionScaleFactors_(jetEnergyResolutionScaleFactors),
       jetEnergyResolutionEtaBinning_(jetEnergyResolutionEtaBinning),
       jetParam_(jetParam),
       lepParam_(lepParam),
       metParam_(metParam),
       constrList_(constraints) {
   setupFitter();
 }

References setupFitter().

◆ ~TtSemiLepKinFitter()

TtSemiLepKinFitter::~TtSemiLepKinFitter ( )

default destructor

Definition at line 70 of file TtSemiLepKinFitter.cc.

                                         {
   delete hadB_;
   delete hadP_;
   delete hadQ_;
   delete lepB_;
   delete lepton_;
   delete neutrino_;
   delete covM_;
   for (std::map<Constraint, TFitConstraintM*>::iterator it = massConstr_.begin(); it != massConstr_.end(); ++it)
     delete it->second;
   delete sumPxConstr_;
   delete sumPyConstr_;
 }

References covM_, hadB_, hadP_, hadQ_, lepB_, lepton_, massConstr_, neutrino_, sumPxConstr_, and sumPyConstr_.

Member Function Documentation

◆ addKinFitInfo()

TtSemiEvtSolution TtSemiLepKinFitter::addKinFitInfo ( TtSemiEvtSolution * asol )

add kin fit information to the old event solution (in for legacy reasons)

Definition at line 351 of file TtSemiLepKinFitter.cc.

                                                                            {
   TtSemiEvtSolution fitsol(*asol);
  
   std::vector<pat::Jet> jets;
   jets.resize(4);
   jets[TtSemiLepEvtPartons::LightQ] = fitsol.getCalHadp();
   jets[TtSemiLepEvtPartons::LightQBar] = fitsol.getCalHadq();
   jets[TtSemiLepEvtPartons::HadB] = fitsol.getCalHadb();
   jets[TtSemiLepEvtPartons::LepB] = fitsol.getCalLepb();
  
   // perform the fit, either using the electron or the muon
   if (fitsol.getDecay() == "electron")
     fit(jets, fitsol.getCalLepe(), fitsol.getCalLepn());
   if (fitsol.getDecay() == "muon")
     fit(jets, fitsol.getCalLepm(), fitsol.getCalLepn());
  
   // add fitted information to the solution
   if (fitter_->getStatus() == 0) {
     // fill the fitted particles
     fitsol.setFitHadb(fittedHadB());
     fitsol.setFitHadp(fittedHadP());
     fitsol.setFitHadq(fittedHadQ());
     fitsol.setFitLepb(fittedLepB());
     fitsol.setFitLepl(fittedLepton());
     fitsol.setFitLepn(fittedNeutrino());
     // store the fit's chi2 probability
     fitsol.setProbChi2(fitProb());
   }
   return fitsol;
 }

Referenced by TtSemiEvtSolutionMaker::produce().

◆ fit() [1/3]

template<class LeptonType >

int TtSemiLepKinFitter::fit	(	const std::vector< pat::Jet > &	jets,
		const pat::Lepton< LeptonType > &	leps,
		const pat::MET &	met
	)

kinematic fit interface for PAT objects

Definition at line 156 of file TtSemiLepKinFitter.h.

                                                     {
   if (jets.size() < 4)
     throw edm::Exception(edm::errors::Configuration, "Cannot run the TtSemiLepKinFitter with less than 4 jets");
  
   // get jets in right order
   const pat::Jet hadP = jets[TtSemiLepEvtPartons::LightQ];
   const pat::Jet hadQ = jets[TtSemiLepEvtPartons::LightQBar];
   const pat::Jet hadB = jets[TtSemiLepEvtPartons::HadB];
   const pat::Jet lepB = jets[TtSemiLepEvtPartons::LepB];
  
   // initialize particles
   const TLorentzVector p4HadP(hadP.px(), hadP.py(), hadP.pz(), hadP.energy());
   const TLorentzVector p4HadQ(hadQ.px(), hadQ.py(), hadQ.pz(), hadQ.energy());
   const TLorentzVector p4HadB(hadB.px(), hadB.py(), hadB.pz(), hadB.energy());
   const TLorentzVector p4LepB(lepB.px(), lepB.py(), lepB.pz(), lepB.energy());
   const TLorentzVector p4Lepton(lepton.px(), lepton.py(), lepton.pz(), lepton.energy());
   const TLorentzVector p4Neutrino(neutrino.px(), neutrino.py(), 0, neutrino.et());
  
   // initialize covariance matrices
   TMatrixD covHadP = covM_->setupMatrix(hadP, jetParam_);
   TMatrixD covHadQ = covM_->setupMatrix(hadQ, jetParam_);
   TMatrixD covHadB = covM_->setupMatrix(hadB, jetParam_, "bjets");
   TMatrixD covLepB = covM_->setupMatrix(lepB, jetParam_, "bjets");
   TMatrixD covLepton = covM_->setupMatrix(lepton, lepParam_);
   TMatrixD covNeutrino = covM_->setupMatrix(neutrino, metParam_);
  
   // now do the part that is fully independent of PAT features
   return fit(p4HadP,
              p4HadQ,
              p4HadB,
              p4LepB,
              p4Lepton,
              p4Neutrino,
              covHadP,
              covHadQ,
              covHadB,
              covLepB,
              covLepton,
              covNeutrino,
              lepton.charge());
 }

References edm::errors::Configuration, covM_, reco::LeafCandidate::energy(), reco::LeafCandidate::et(), TtSemiLepEvtPartons::HadB, jetParam_, singleTopDQM_cfi::jets, TtSemiLepEvtPartons::LepB, lepParam_, TtSemiLepEvtPartons::LightQ, TtSemiLepEvtPartons::LightQBar, metParam_, reco::LeafCandidate::px(), reco::LeafCandidate::py(), reco::LeafCandidate::pz(), and CovarianceMatrix::setupMatrix().

Referenced by addKinFitInfo(), fit(), and trackingPlots.Iteration::modules().

◆ fit() [2/3]

int TtSemiLepKinFitter::fit	(	const TLorentzVector &	p4HadP,
		const TLorentzVector &	p4HadQ,
		const TLorentzVector &	p4HadB,
		const TLorentzVector &	p4LepB,
		const TLorentzVector &	p4Lepton,
		const TLorentzVector &	p4Neutrino,
		const int	leptonCharge,
		const CovarianceMatrix::ObjectType	leptonType
	)

kinematic fit interface for plain 4-vecs

Definition at line 237 of file TtSemiLepKinFitter.cc.

                                                                          {
   // initialize covariance matrices
   TMatrixD covHadP = covM_->setupMatrix(p4HadP, CovarianceMatrix::kUdscJet, jetParam_);
   TMatrixD covHadQ = covM_->setupMatrix(p4HadQ, CovarianceMatrix::kUdscJet, jetParam_);
   TMatrixD covHadB = covM_->setupMatrix(p4HadB, CovarianceMatrix::kBJet, jetParam_);
   TMatrixD covLepB = covM_->setupMatrix(p4LepB, CovarianceMatrix::kBJet, jetParam_);
   TMatrixD covLepton = covM_->setupMatrix(p4Lepton, leptonType, lepParam_);
   TMatrixD covNeutrino = covM_->setupMatrix(p4Neutrino, CovarianceMatrix::kMet, metParam_);
  
   // now do the part that is fully independent of PAT features
   return fit(p4HadP,
              p4HadQ,
              p4HadB,
              p4LepB,
              p4Lepton,
              p4Neutrino,
              covHadP,
              covHadQ,
              covHadB,
              covLepB,
              covLepton,
              covNeutrino,
              leptonCharge);
 }

References covM_, fit(), jetParam_, CovarianceMatrix::kBJet, CovarianceMatrix::kMet, CovarianceMatrix::kUdscJet, lepParam_, objects.autophobj::leptonType, metParam_, and CovarianceMatrix::setupMatrix().

Referenced by trackingPlots.Iteration::modules().

◆ fit() [3/3]

int TtSemiLepKinFitter::fit	(	const TLorentzVector &	p4HadP,
		const TLorentzVector &	p4HadQ,
		const TLorentzVector &	p4HadB,
		const TLorentzVector &	p4LepB,
		const TLorentzVector &	p4Lepton,
		const TLorentzVector &	p4Neutrino,
		const TMatrixD &	covHadP,
		const TMatrixD &	covHadQ,
		const TMatrixD &	covHadB,
		const TMatrixD &	covLepB,
		const TMatrixD &	covLepton,
		const TMatrixD &	covNeutrino,
		const int	leptonCharge
	)

common core of the fit interface

Definition at line 269 of file TtSemiLepKinFitter.cc.

                                                     {
   // set the kinematics of the objects to be fitted
   hadP_->setIni4Vec(&p4HadP);
   hadQ_->setIni4Vec(&p4HadQ);
   hadB_->setIni4Vec(&p4HadB);
   lepB_->setIni4Vec(&p4LepB);
   lepton_->setIni4Vec(&p4Lepton);
   neutrino_->setIni4Vec(&p4Neutrino);
  
   hadP_->setCovMatrix(&covHadP);
   hadQ_->setCovMatrix(&covHadQ);
   hadB_->setCovMatrix(&covHadB);
   lepB_->setCovMatrix(&covLepB);
   lepton_->setCovMatrix(&covLepton);
   neutrino_->setCovMatrix(&covNeutrino);
  
   if (constrainSumPt_) {
     // setup Px and Py constraint for curent event configuration so that sum Pt will be conserved
     sumPxConstr_->setConstraint(p4HadP.Px() + p4HadQ.Px() + p4HadB.Px() + p4LepB.Px() + p4Lepton.Px() +
                                 p4Neutrino.Px());
     sumPyConstr_->setConstraint(p4HadP.Py() + p4HadQ.Py() + p4HadB.Py() + p4LepB.Py() + p4Lepton.Py() +
                                 p4Neutrino.Py());
   }
  
   // now do the fit
   fitter_->fit();
  
   // read back the resulting particles if the fit converged
   if (fitter_->getStatus() == 0) {
     // read back jet kinematics
     fittedHadP_ = pat::Particle(reco::LeafCandidate(
         0,
         math::XYZTLorentzVector(
             hadP_->getCurr4Vec()->X(), hadP_->getCurr4Vec()->Y(), hadP_->getCurr4Vec()->Z(), hadP_->getCurr4Vec()->E()),
         math::XYZPoint()));
     fittedHadQ_ = pat::Particle(reco::LeafCandidate(
         0,
         math::XYZTLorentzVector(
             hadQ_->getCurr4Vec()->X(), hadQ_->getCurr4Vec()->Y(), hadQ_->getCurr4Vec()->Z(), hadQ_->getCurr4Vec()->E()),
         math::XYZPoint()));
     fittedHadB_ = pat::Particle(reco::LeafCandidate(
         0,
         math::XYZTLorentzVector(
             hadB_->getCurr4Vec()->X(), hadB_->getCurr4Vec()->Y(), hadB_->getCurr4Vec()->Z(), hadB_->getCurr4Vec()->E()),
         math::XYZPoint()));
     fittedLepB_ = pat::Particle(reco::LeafCandidate(
         0,
         math::XYZTLorentzVector(
             lepB_->getCurr4Vec()->X(), lepB_->getCurr4Vec()->Y(), lepB_->getCurr4Vec()->Z(), lepB_->getCurr4Vec()->E()),
         math::XYZPoint()));
  
     // read back lepton kinematics
     fittedLepton_ = pat::Particle(reco::LeafCandidate(leptonCharge,
                                                       math::XYZTLorentzVector(lepton_->getCurr4Vec()->X(),
                                                                               lepton_->getCurr4Vec()->Y(),
                                                                               lepton_->getCurr4Vec()->Z(),
                                                                               lepton_->getCurr4Vec()->E()),
                                                       math::XYZPoint()));
  
     // read back the MET kinematics
     fittedNeutrino_ = pat::Particle(reco::LeafCandidate(0,
                                                         math::XYZTLorentzVector(neutrino_->getCurr4Vec()->X(),
                                                                                 neutrino_->getCurr4Vec()->Y(),
                                                                                 neutrino_->getCurr4Vec()->Z(),
                                                                                 neutrino_->getCurr4Vec()->E()),
                                                         math::XYZPoint()));
   }
   return fitter_->getStatus();
 }

References constrainSumPt_, TKinFitter::fit(), fittedHadB_, fittedHadP_, fittedHadQ_, fittedLepB_, fittedLepton_, fittedNeutrino_, TopKinFitter::fitter_, TAbsFitParticle::getCurr4Vec(), TKinFitter::getStatus(), hadB_, hadP_, hadQ_, lepB_, lepton_, neutrino_, TFitConstraintEp::setConstraint(), TAbsFitParticle::setCovMatrix(), TAbsFitParticle::setIni4Vec(), sumPxConstr_, and sumPyConstr_.

Referenced by trackingPlots.Iteration::modules().

◆ fittedHadB()

const pat::Particle TtSemiLepKinFitter::fittedHadB ( ) const

inline

return hadronic b quark candidate

Definition at line 86 of file TtSemiLepKinFitter.h.

86 { return (fitter_->getStatus() == 0 ? fittedHadB_ : pat::Particle()); };

References fittedHadB_, TopKinFitter::fitter_, and TKinFitter::getStatus().

Referenced by addKinFitInfo().

◆ fittedHadP()

const pat::Particle TtSemiLepKinFitter::fittedHadP ( ) const

inline

return hadronic light quark candidate

Definition at line 88 of file TtSemiLepKinFitter.h.

88 { return (fitter_->getStatus() == 0 ? fittedHadP_ : pat::Particle()); };

References fittedHadP_, TopKinFitter::fitter_, and TKinFitter::getStatus().

Referenced by addKinFitInfo().

◆ fittedHadQ()

const pat::Particle TtSemiLepKinFitter::fittedHadQ ( ) const

inline

return hadronic light quark candidate

Definition at line 90 of file TtSemiLepKinFitter.h.

90 { return (fitter_->getStatus() == 0 ? fittedHadQ_ : pat::Particle()); };

References fittedHadQ_, TopKinFitter::fitter_, and TKinFitter::getStatus().

Referenced by addKinFitInfo().

◆ fittedLepB()

const pat::Particle TtSemiLepKinFitter::fittedLepB ( ) const

inline

return leptonic b quark candidate

Definition at line 92 of file TtSemiLepKinFitter.h.

92 { return (fitter_->getStatus() == 0 ? fittedLepB_ : pat::Particle()); };

References fittedLepB_, TopKinFitter::fitter_, and TKinFitter::getStatus().

Referenced by addKinFitInfo().

◆ fittedLepton()

const pat::Particle TtSemiLepKinFitter::fittedLepton ( ) const

inline

return lepton candidate

Definition at line 94 of file TtSemiLepKinFitter.h.

94 { return (fitter_->getStatus() == 0 ? fittedLepton_ : pat::Particle()); };

References fittedLepton_, TopKinFitter::fitter_, and TKinFitter::getStatus().

Referenced by addKinFitInfo().

◆ fittedNeutrino()

const pat::Particle TtSemiLepKinFitter::fittedNeutrino ( ) const

inline

return neutrino candidate

Definition at line 96 of file TtSemiLepKinFitter.h.

                                            {
     return (fitter_->getStatus() == 0 ? fittedNeutrino_ : pat::Particle());
   };

References fittedNeutrino_, TopKinFitter::fitter_, and TKinFitter::getStatus().

Referenced by addKinFitInfo().

◆ printSetup()

void TtSemiLepKinFitter::printSetup ( ) const

private

print fitter setup

Definition at line 84 of file TtSemiLepKinFitter.cc.

                                           {
   std::stringstream constr;
   for (unsigned int i = 0; i < constrList_.size(); ++i) {
     switch (constrList_[i]) {
       case kWHadMass:
         constr << "    * hadronic W-mass (" << mW_ << " GeV) \n";
         break;
       case kWLepMass:
         constr << "    * leptonic W-mass (" << mW_ << " GeV) \n";
         break;
       case kTopHadMass:
         constr << "    * hadronic t-mass (" << mTop_ << " GeV) \n";
         break;
       case kTopLepMass:
         constr << "    * leptonic t-mass (" << mTop_ << " GeV) \n";
         break;
       case kNeutrinoMass:
         constr << "    * neutrino   mass (0 GeV) \n";
         break;
       case kEqualTopMasses:
         constr << "    * equal    t-masses \n";
         break;
       case kSumPt:
         constr << "    * summed transverse momentum \n";
         break;
     }
   }
   edm::LogVerbatim("TtSemiLepKinFitter") << "\n"
                                          << "+++++++++++ TtSemiLepKinFitter Setup ++++++++++++ \n"
                                          << "  Parametrization:                                \n"
                                          << "   * jet : " << param(jetParam_) << "\n"
                                          << "   * lep : " << param(lepParam_) << "\n"
                                          << "   * met : " << param(metParam_) << "\n"
                                          << "  Constraints:                                    \n"
                                          << constr.str() << "  Max(No iterations): " << maxNrIter_ << "\n"
                                          << "  Max(deltaS)       : " << maxDeltaS_ << "\n"
                                          << "  Max(F)            : " << maxF_ << "\n"
                                          << "+++++++++++++++++++++++++++++++++++++++++++++++++ \n";
 }

References constrList_, mps_fire::i, jetParam_, kEqualTopMasses, kNeutrinoMass, kSumPt, kTopHadMass, kTopLepMass, kWHadMass, kWLepMass, lepParam_, TopKinFitter::maxDeltaS_, TopKinFitter::maxF_, TopKinFitter::maxNrIter_, metParam_, TopKinFitter::mTop_, TopKinFitter::mW_, and TopKinFitter::param().

Referenced by setupFitter().

◆ setupConstraints()

void TtSemiLepKinFitter::setupConstraints ( )

private

initialize constraints

Definition at line 175 of file TtSemiLepKinFitter.cc.

                                           {
   massConstr_[kWHadMass] = new TFitConstraintM("WMassHad", "WMassHad", nullptr, nullptr, mW_);
   massConstr_[kWLepMass] = new TFitConstraintM("WMassLep", "WMassLep", nullptr, nullptr, mW_);
   massConstr_[kTopHadMass] = new TFitConstraintM("TopMassHad", "TopMassHad", nullptr, nullptr, mTop_);
   massConstr_[kTopLepMass] = new TFitConstraintM("TopMassLep", "TopMassLep", nullptr, nullptr, mTop_);
   massConstr_[kNeutrinoMass] = new TFitConstraintM("NeutrinoMass", "NeutrinoMass", nullptr, nullptr, 0.);
   massConstr_[kEqualTopMasses] = new TFitConstraintM("EqualTopMasses", "EqualTopMasses", nullptr, nullptr, 0.);
   sumPxConstr_ = new TFitConstraintEp("SumPx", "SumPx", nullptr, TFitConstraintEp::pX, 0.);
   sumPyConstr_ = new TFitConstraintEp("SumPy", "SumPy", nullptr, TFitConstraintEp::pY, 0.);
  
   massConstr_[kWHadMass]->addParticles1(hadP_, hadQ_);
   massConstr_[kWLepMass]->addParticles1(lepton_, neutrino_);
   massConstr_[kTopHadMass]->addParticles1(hadP_, hadQ_, hadB_);
   massConstr_[kTopLepMass]->addParticles1(lepton_, neutrino_, lepB_);
   massConstr_[kNeutrinoMass]->addParticle1(neutrino_);
   massConstr_[kEqualTopMasses]->addParticles1(hadP_, hadQ_, hadB_);
   massConstr_[kEqualTopMasses]->addParticles2(lepton_, neutrino_, lepB_);
   sumPxConstr_->addParticles(lepton_, neutrino_, hadP_, hadQ_, hadB_, lepB_);
   sumPyConstr_->addParticles(lepton_, neutrino_, hadP_, hadQ_, hadB_, lepB_);
  
   if (std::find(constrList_.begin(), constrList_.end(), kSumPt) != constrList_.end())
     constrainSumPt_ = true;
   constrainSumPt_ = false;
 }

References TFitConstraintEp::addParticles(), constrainSumPt_, constrList_, spr::find(), hadB_, hadP_, hadQ_, kEqualTopMasses, kNeutrinoMass, kSumPt, kTopHadMass, kTopLepMass, kWHadMass, kWLepMass, lepB_, lepton_, massConstr_, TopKinFitter::mTop_, TopKinFitter::mW_, neutrino_, TFitConstraintEp::pX, TFitConstraintEp::pY, sumPxConstr_, and sumPyConstr_.

Referenced by setupFitter().

◆ setupFitter()

void TtSemiLepKinFitter::setupFitter ( )

private

setup fitter

Definition at line 200 of file TtSemiLepKinFitter.cc.

                                      {
   printSetup();
  
   setupJets();
   setupLeptons();
   setupConstraints();
  
   // add measured particles
   fitter_->addMeasParticle(hadB_);
   fitter_->addMeasParticle(hadP_);
   fitter_->addMeasParticle(hadQ_);
   fitter_->addMeasParticle(lepB_);
   fitter_->addMeasParticle(lepton_);
   fitter_->addMeasParticle(neutrino_);
  
   // add constraints
   for (unsigned int i = 0; i < constrList_.size(); i++) {
     if (constrList_[i] != kSumPt)
       fitter_->addConstraint(massConstr_[constrList_[i]]);
   }
   if (constrainSumPt_) {
     fitter_->addConstraint(sumPxConstr_);
     fitter_->addConstraint(sumPyConstr_);
   }
  
   // initialize helper class used to bring the resolutions into covariance matrices
   if (!udscResolutions_->empty() && !bResolutions_->empty() && !lepResolutions_->empty() && !metResolutions_->empty())
     covM_ = new CovarianceMatrix(*udscResolutions_,
                                  *bResolutions_,
                                  *lepResolutions_,
                                  *metResolutions_,
                                  *jetEnergyResolutionScaleFactors_,
                                  *jetEnergyResolutionEtaBinning_);
   else
     covM_ = new CovarianceMatrix();
 }

References TKinFitter::addConstraint(), TKinFitter::addMeasParticle(), bResolutions_, constrainSumPt_, constrList_, covM_, TopKinFitter::fitter_, hadB_, hadP_, hadQ_, mps_fire::i, jetEnergyResolutionEtaBinning_, jetEnergyResolutionScaleFactors_, kSumPt, lepB_, lepResolutions_, lepton_, massConstr_, metResolutions_, neutrino_, printSetup(), setupConstraints(), setupJets(), setupLeptons(), sumPxConstr_, sumPyConstr_, and udscResolutions_.

Referenced by TtSemiLepKinFitter().

◆ setupJets()

void TtSemiLepKinFitter::setupJets ( )

private

initialize jet inputs

Definition at line 124 of file TtSemiLepKinFitter.cc.

                                    {
   TMatrixD empty3x3(3, 3);
   TMatrixD empty4x4(4, 4);
   switch (jetParam_) {  // setup jets according to parameterization
     case kEMom:
       hadB_ = new TFitParticleEMomDev("Jet1", "Jet1", nullptr, &empty4x4);
       hadP_ = new TFitParticleEMomDev("Jet2", "Jet2", nullptr, &empty4x4);
       hadQ_ = new TFitParticleEMomDev("Jet3", "Jet3", nullptr, &empty4x4);
       lepB_ = new TFitParticleEMomDev("Jet4", "Jet4", nullptr, &empty4x4);
       break;
     case kEtEtaPhi:
       hadB_ = new TFitParticleEtEtaPhi("Jet1", "Jet1", nullptr, &empty3x3);
       hadP_ = new TFitParticleEtEtaPhi("Jet2", "Jet2", nullptr, &empty3x3);
       hadQ_ = new TFitParticleEtEtaPhi("Jet3", "Jet3", nullptr, &empty3x3);
       lepB_ = new TFitParticleEtEtaPhi("Jet4", "Jet4", nullptr, &empty3x3);
       break;
     case kEtThetaPhi:
       hadB_ = new TFitParticleEtThetaPhi("Jet1", "Jet1", nullptr, &empty3x3);
       hadP_ = new TFitParticleEtThetaPhi("Jet2", "Jet2", nullptr, &empty3x3);
       hadQ_ = new TFitParticleEtThetaPhi("Jet3", "Jet3", nullptr, &empty3x3);
       lepB_ = new TFitParticleEtThetaPhi("Jet4", "Jet4", nullptr, &empty3x3);
       break;
   }
 }

References hadB_, hadP_, hadQ_, jetParam_, TopKinFitter::kEMom, TopKinFitter::kEtEtaPhi, TopKinFitter::kEtThetaPhi, and lepB_.

Referenced by setupFitter().

◆ setupLeptons()

void TtSemiLepKinFitter::setupLeptons ( )

private

initialize lepton inputs

Definition at line 149 of file TtSemiLepKinFitter.cc.

                                       {
   TMatrixD empty3x3(3, 3);
   switch (lepParam_) {  // setup lepton according to parameterization
     case kEMom:
       lepton_ = new TFitParticleEScaledMomDev("Lepton", "Lepton", nullptr, &empty3x3);
       break;
     case kEtEtaPhi:
       lepton_ = new TFitParticleEtEtaPhi("Lepton", "Lepton", nullptr, &empty3x3);
       break;
     case kEtThetaPhi:
       lepton_ = new TFitParticleEtThetaPhi("Lepton", "Lepton", nullptr, &empty3x3);
       break;
   }
   switch (metParam_) {  // setup neutrino according to parameterization
     case kEMom:
       neutrino_ = new TFitParticleEScaledMomDev("Neutrino", "Neutrino", nullptr, &empty3x3);
       break;
     case kEtEtaPhi:
       neutrino_ = new TFitParticleEtEtaPhi("Neutrino", "Neutrino", nullptr, &empty3x3);
       break;
     case kEtThetaPhi:
       neutrino_ = new TFitParticleEtThetaPhi("Neutrino", "Neutrino", nullptr, &empty3x3);
       break;
   }
 }

References TopKinFitter::kEMom, TopKinFitter::kEtEtaPhi, TopKinFitter::kEtThetaPhi, lepParam_, lepton_, metParam_, and neutrino_.

Referenced by setupFitter().

Member Data Documentation

◆ bResolutions_

const std::vector<edm::ParameterSet>* TtSemiLepKinFitter::bResolutions_

private

Definition at line 124 of file TtSemiLepKinFitter.h.

Referenced by setupFitter().

◆ constrainSumPt_

bool TtSemiLepKinFitter::constrainSumPt_

private

internally use simple boolean for this constraint to reduce the per-event computing time

Definition at line 152 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), and setupFitter().

◆ constrList_

std::vector<Constraint> TtSemiLepKinFitter::constrList_

private

vector of constraints to be used

Definition at line 150 of file TtSemiLepKinFitter.h.

Referenced by printSetup(), setupConstraints(), and setupFitter().

◆ covM_

CovarianceMatrix* TtSemiLepKinFitter::covM_

private

object used to construct the covariance matrices for the individual particles

Definition at line 131 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupFitter(), and ~TtSemiLepKinFitter().

◆ fittedHadB_

pat::Particle TtSemiLepKinFitter::fittedHadB_

private

output particles

Definition at line 137 of file TtSemiLepKinFitter.h.

Referenced by fit(), and fittedHadB().

◆ fittedHadP_

pat::Particle TtSemiLepKinFitter::fittedHadP_

private

Definition at line 138 of file TtSemiLepKinFitter.h.

Referenced by fit(), and fittedHadP().

◆ fittedHadQ_

pat::Particle TtSemiLepKinFitter::fittedHadQ_

private

Definition at line 139 of file TtSemiLepKinFitter.h.

Referenced by fit(), and fittedHadQ().

◆ fittedLepB_

pat::Particle TtSemiLepKinFitter::fittedLepB_

private

Definition at line 140 of file TtSemiLepKinFitter.h.

Referenced by fit(), and fittedLepB().

◆ fittedLepton_

pat::Particle TtSemiLepKinFitter::fittedLepton_

private

Definition at line 141 of file TtSemiLepKinFitter.h.

Referenced by fit(), and fittedLepton().

◆ fittedNeutrino_

pat::Particle TtSemiLepKinFitter::fittedNeutrino_

private

Definition at line 142 of file TtSemiLepKinFitter.h.

Referenced by fit(), and fittedNeutrino().

◆ hadB_

TAbsFitParticle* TtSemiLepKinFitter::hadB_

private

input particles

Definition at line 116 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), setupJets(), and ~TtSemiLepKinFitter().

◆ hadP_

TAbsFitParticle* TtSemiLepKinFitter::hadP_

private

Definition at line 117 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), setupJets(), and ~TtSemiLepKinFitter().

◆ hadQ_

TAbsFitParticle* TtSemiLepKinFitter::hadQ_

private

Definition at line 118 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), setupJets(), and ~TtSemiLepKinFitter().

◆ jetEnergyResolutionEtaBinning_

const std::vector<double>* TtSemiLepKinFitter::jetEnergyResolutionEtaBinning_

private

Definition at line 129 of file TtSemiLepKinFitter.h.

Referenced by setupFitter().

◆ jetEnergyResolutionScaleFactors_

const std::vector<double>* TtSemiLepKinFitter::jetEnergyResolutionScaleFactors_

private

scale factors for the jet energy resolution

Definition at line 128 of file TtSemiLepKinFitter.h.

Referenced by setupFitter().

◆ jetParam_

Param TtSemiLepKinFitter::jetParam_

private

jet parametrization

Definition at line 144 of file TtSemiLepKinFitter.h.

Referenced by fit(), printSetup(), and setupJets().

◆ lepB_

TAbsFitParticle* TtSemiLepKinFitter::lepB_

private

Definition at line 119 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), setupJets(), and ~TtSemiLepKinFitter().

◆ lepParam_

Param TtSemiLepKinFitter::lepParam_

private

lepton parametrization

Definition at line 146 of file TtSemiLepKinFitter.h.

Referenced by fit(), printSetup(), and setupLeptons().

◆ lepResolutions_

const std::vector<edm::ParameterSet>* TtSemiLepKinFitter::lepResolutions_

private

Definition at line 125 of file TtSemiLepKinFitter.h.

Referenced by setupFitter().

◆ lepton_

TAbsFitParticle* TtSemiLepKinFitter::lepton_

private

Definition at line 120 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), setupLeptons(), and ~TtSemiLepKinFitter().

◆ massConstr_

std::map<Constraint, TFitConstraintM*> TtSemiLepKinFitter::massConstr_

private

supported constraints

Definition at line 133 of file TtSemiLepKinFitter.h.

Referenced by setupConstraints(), setupFitter(), and ~TtSemiLepKinFitter().

◆ metParam_

Param TtSemiLepKinFitter::metParam_

private

met parametrization

Definition at line 148 of file TtSemiLepKinFitter.h.

Referenced by fit(), printSetup(), and setupLeptons().

◆ metResolutions_

const std::vector<edm::ParameterSet>* TtSemiLepKinFitter::metResolutions_

private

Definition at line 126 of file TtSemiLepKinFitter.h.

Referenced by setupFitter().

◆ neutrino_

TAbsFitParticle* TtSemiLepKinFitter::neutrino_

private

Definition at line 121 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), setupLeptons(), and ~TtSemiLepKinFitter().

◆ sumPxConstr_

TFitConstraintEp* TtSemiLepKinFitter::sumPxConstr_

private

Definition at line 134 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), and ~TtSemiLepKinFitter().

◆ sumPyConstr_

TFitConstraintEp* TtSemiLepKinFitter::sumPyConstr_

private

Definition at line 135 of file TtSemiLepKinFitter.h.

Referenced by fit(), setupConstraints(), setupFitter(), and ~TtSemiLepKinFitter().

◆ udscResolutions_

const std::vector<edm::ParameterSet>* TtSemiLepKinFitter::udscResolutions_

private

resolutions

Definition at line 123 of file TtSemiLepKinFitter.h.

Referenced by setupFitter().

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Enumeration Documentation

◆ Constraint

Constructor & Destructor Documentation

◆ TtSemiLepKinFitter() [1/2]

◆ TtSemiLepKinFitter() [2/2]

◆ ~TtSemiLepKinFitter()

Member Function Documentation

◆ addKinFitInfo()

◆ fit() [1/3]

◆ fit() [2/3]

◆ fit() [3/3]

◆ fittedHadB()

◆ fittedHadP()

◆ fittedHadQ()

◆ fittedLepB()

◆ fittedLepton()

◆ fittedNeutrino()

◆ printSetup()

◆ setupConstraints()

◆ setupFitter()

◆ setupJets()

◆ setupLeptons()

Member Data Documentation

◆ bResolutions_

◆ constrainSumPt_

◆ constrList_

◆ covM_

◆ fittedHadB_

◆ fittedHadP_

◆ fittedHadQ_

◆ fittedLepB_

◆ fittedLepton_

◆ fittedNeutrino_

◆ hadB_

◆ hadP_

◆ hadQ_

◆ jetEnergyResolutionEtaBinning_

◆ jetEnergyResolutionScaleFactors_

◆ jetParam_

◆ lepB_

◆ lepParam_

◆ lepResolutions_

◆ lepton_

◆ massConstr_

◆ metParam_

◆ metResolutions_

◆ neutrino_

◆ sumPxConstr_

◆ sumPyConstr_

◆ udscResolutions_