TtFullHadKinFitter Class Reference

#include <TtFullHadKinFitter.h>

Inheritance diagram for TtFullHadKinFitter:

Classes
class	KinFit
	class that does the fitting More...
struct	KinFitResult
	struct for fit results More...

Public Types
enum	Constraint {   kWPlusMass = 1, kWMinusMass, kTopMass, kTopBarMass,   kEqualTopMasses }
	supported constraints More...
Public Types inherited from TopKinFitter
enum	Param { kEMom, kEtEtaPhi, kEtThetaPhi }
	supported parameterizations More...

Public Member Functions
TtHadEvtSolution	addKinFitInfo (TtHadEvtSolution *asol)
	add kin fit information to the old event solution (in for legacy reasons) More...
int	fit (const std::vector< pat::Jet > &jets)
	kinematic fit interface More...
const pat::Particle	fittedB () const
	return fitted b quark candidate More...
const pat::Particle	fittedBBar () const
	return fitted b quark candidate More...
const pat::Particle	fittedLightP () const
	return fitted light quark candidate More...
const pat::Particle	fittedLightPBar () const
	return fitted light quark candidate More...
const pat::Particle	fittedLightQ () const
	return fitted light quark candidate More...
const pat::Particle	fittedLightQBar () const
	return fitted light quark candidate More...
std::vector< TtFullHadKinFitter::Constraint >	intToConstraint (const std::vector< unsigned int > &constraints)
	used to convert vector of int's to vector of constraints (just used in TtFullHadKinFitter(int, int, double, double, std::vector<unsigned int>)) More...
	TtFullHadKinFitter ()
	default constructor More...
	TtFullHadKinFitter (int jetParam, int maxNrIter, double maxDeltaS, double maxF, const std::vector< unsigned int > &constraints, double mW=80.4, double mTop=173., const std::vector< edm::ParameterSet > *udscResolutions=0, const std::vector< edm::ParameterSet > *bResolutions=0, const std::vector< double > *jetEnergyResolutionScaleFactors=0, const std::vector< double > *jetEnergyResolutionEtaBinning=0)
	constructor initialized with build-in types as custom parameters (only included to keep TtHadEvtSolutionMaker.cc running) More...
	TtFullHadKinFitter (Param jetParam, int maxNrIter, double maxDeltaS, double maxF, const std::vector< Constraint > &constraints, double mW=80.4, double mTop=173., const std::vector< edm::ParameterSet > *udscResolutions=0, const std::vector< edm::ParameterSet > *bResolutions=0, const std::vector< double > *jetEnergyResolutionScaleFactors=0, const std::vector< double > *jetEnergyResolutionEtaBinning=0)
	constructor initialized with built-in types and class enum's custom parameters More...
	~TtFullHadKinFitter ()
	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...

Private Attributes
TAbsFitParticle *	b_
	input particles More...
TAbsFitParticle *	bBar_
const std::vector< edm::ParameterSet > *	bResolutions_
std::vector< Constraint >	constraints_
	vector of constraints to be used More...
CovarianceMatrix *	covM_
	get object resolutions and put them into a matrix More...
pat::Particle	fittedB_
	output particles More...
pat::Particle	fittedBBar_
pat::Particle	fittedLightP_
pat::Particle	fittedLightPBar_
pat::Particle	fittedLightQ_
pat::Particle	fittedLightQBar_
const std::vector< double > *	jetEnergyResolutionEtaBinning_
const std::vector< double > *	jetEnergyResolutionScaleFactors_
	scale factors for the jet energy resolution More...
Param	jetParam_
	jet parametrization More...
TAbsFitParticle *	lightP_
TAbsFitParticle *	lightPBar_
TAbsFitParticle *	lightQ_
TAbsFitParticle *	lightQBar_
std::map< Constraint, TFitConstraintM * >	massConstr_
	supported constraints More...
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 29 of file TtFullHadKinFitter.h.

Member Enumeration Documentation

enum TtFullHadKinFitter::Constraint

supported constraints

Enumerator
kWPlusMass
kWMinusMass
kTopMass
kTopBarMass
kEqualTopMasses

Definition at line 32 of file TtFullHadKinFitter.h.

{ kWPlusMass = 1, kWMinusMass, kTopMass, kTopBarMass, kEqualTopMasses };

Constructor & Destructor Documentation

TtFullHadKinFitter::TtFullHadKinFitter

(

)

default constructor

Definition at line 16 of file TtFullHadKinFitter.cc.

References setupFitter().

Referenced by TtFullHadKinFitter::KinFit::KinFit().

    : TopKinFitter(),
      b_(nullptr),
      bBar_(nullptr),
      lightQ_(nullptr),
      lightQBar_(nullptr),
      lightP_(nullptr),
      lightPBar_(nullptr),
      udscResolutions_(nullptr),
      bResolutions_(nullptr),
      jetEnergyResolutionScaleFactors_(nullptr),
      jetEnergyResolutionEtaBinning_(nullptr),
      jetParam_(kEMom) {
  setupFitter();
}

TtFullHadKinFitter::TtFullHadKinFitter	(	int	jetParam,
		int	maxNrIter,
		double	maxDeltaS,
		double	maxF,
		const std::vector< unsigned int > &	constraints,
		double	mW = 80.4,
		double	mTop = 173.,
		const std::vector< edm::ParameterSet > *	udscResolutions = 0,
		const std::vector< edm::ParameterSet > *	bResolutions = 0,
		const std::vector< double > *	jetEnergyResolutionScaleFactors = 0,
		const std::vector< double > *	jetEnergyResolutionEtaBinning = 0
	)

constructor initialized with build-in types as custom parameters (only included to keep TtHadEvtSolutionMaker.cc running)

Definition at line 44 of file TtFullHadKinFitter.cc.

References setupFitter().

    : TopKinFitter(maxNrIter, maxDeltaS, maxF, mW, mTop),
      b_(nullptr),
      bBar_(nullptr),
      lightQ_(nullptr),
      lightQBar_(nullptr),
      lightP_(nullptr),
      lightPBar_(nullptr),
      udscResolutions_(udscResolutions),
      bResolutions_(bResolutions),
      jetEnergyResolutionScaleFactors_(jetEnergyResolutionScaleFactors),
      jetEnergyResolutionEtaBinning_(jetEnergyResolutionEtaBinning),
      jetParam_((Param)jetParam),
      constraints_(intToConstraint(constraints)) {
  setupFitter();
}

TtFullHadKinFitter::TtFullHadKinFitter	(	Param	jetParam,
		int	maxNrIter,
		double	maxDeltaS,
		double	maxF,
		const std::vector< Constraint > &	constraints,
		double	mW = 80.4,
		double	mTop = 173.,
		const std::vector< edm::ParameterSet > *	udscResolutions = 0,
		const std::vector< edm::ParameterSet > *	bResolutions = 0,
		const std::vector< double > *	jetEnergyResolutionScaleFactors = 0,
		const std::vector< double > *	jetEnergyResolutionEtaBinning = 0
	)

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

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

Definition at line 72 of file TtFullHadKinFitter.cc.

References setupFitter().

    : TopKinFitter(maxNrIter, maxDeltaS, maxF, mW, mTop),
      b_(nullptr),
      bBar_(nullptr),
      lightQ_(nullptr),
      lightQBar_(nullptr),
      lightP_(nullptr),
      lightPBar_(nullptr),
      udscResolutions_(udscResolutions),
      bResolutions_(bResolutions),
      jetEnergyResolutionScaleFactors_(jetEnergyResolutionScaleFactors),
      jetEnergyResolutionEtaBinning_(jetEnergyResolutionEtaBinning),
      jetParam_(jetParam),
      constraints_(constraints) {
  setupFitter();
}

TtFullHadKinFitter::~TtFullHadKinFitter

(

)

default destructor

Definition at line 100 of file TtFullHadKinFitter.cc.

References b_, bBar_, covM_, lightP_, lightPBar_, lightQ_, lightQBar_, and massConstr_.

                                        {
  delete b_;
  delete bBar_;
  delete lightQ_;
  delete lightQBar_;
  delete lightP_;
  delete lightPBar_;
  delete covM_;
  for (std::map<Constraint, TFitConstraintM*>::iterator it = massConstr_.begin(); it != massConstr_.end(); ++it)
    delete it->second;
}

Member Function Documentation

TtHadEvtSolution TtFullHadKinFitter::addKinFitInfo

(

TtHadEvtSolution *

asol

)

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

Definition at line 312 of file TtFullHadKinFitter.cc.

References TtFullHadEvtPartons::B, TtFullHadEvtPartons::BBar, fit(), TopKinFitter::fitProb(), fittedB_, fittedBBar_, fittedLightP_, fittedLightPBar_, fittedLightQ_, fittedLightQBar_, TopKinFitter::fitter_, TtHadEvtSolution::getCalHadb(), TtHadEvtSolution::getCalHadbbar(), TtHadEvtSolution::getCalHadj(), TtHadEvtSolution::getCalHadk(), TtHadEvtSolution::getCalHadp(), TtHadEvtSolution::getCalHadq(), TKinFitter::getStatus(), singleTopDQM_cfi::jets, TtFullHadEvtPartons::LightP, TtFullHadEvtPartons::LightPBar, TtFullHadEvtPartons::LightQ, TtFullHadEvtPartons::LightQBar, TtHadEvtSolution::setFitHadb(), TtHadEvtSolution::setFitHadbbar(), TtHadEvtSolution::setFitHadj(), TtHadEvtSolution::setFitHadk(), TtHadEvtSolution::setFitHadp(), TtHadEvtSolution::setFitHadq(), and TtHadEvtSolution::setProbChi2().

Referenced by fittedLightPBar(), and TtHadEvtSolutionMaker::produce().

                                                                         {
  TtHadEvtSolution fitsol(*asol);

  std::vector<pat::Jet> jets;
  jets.resize(6);
  jets[TtFullHadEvtPartons::LightQ] = fitsol.getCalHadp();
  jets[TtFullHadEvtPartons::LightQBar] = fitsol.getCalHadq();
  jets[TtFullHadEvtPartons::B] = fitsol.getCalHadb();
  jets[TtFullHadEvtPartons::LightP] = fitsol.getCalHadj();
  jets[TtFullHadEvtPartons::LightPBar] = fitsol.getCalHadk();
  jets[TtFullHadEvtPartons::BBar] = fitsol.getCalHadbbar();

  fit(jets);

  // add fitted information to the solution
  if (fitter_->getStatus() == 0) {
    // finally fill the fitted particles
    fitsol.setFitHadb(fittedB_);
    fitsol.setFitHadp(fittedLightQ_);
    fitsol.setFitHadq(fittedLightQBar_);
    fitsol.setFitHadk(fittedLightP_);
    fitsol.setFitHadj(fittedLightPBar_);
    fitsol.setFitHadbbar(fittedBBar_);

    // store the fit's chi2 probability
    fitsol.setProbChi2(fitProb());
  }
  return fitsol;
}

int TtFullHadKinFitter::fit

(

const std::vector< pat::Jet > &

jets

)

kinematic fit interface

Definition at line 221 of file TtFullHadKinFitter.cc.

References TtFullHadEvtPartons::B, b, b_, TtFullHadEvtPartons::BBar, bBar_, edm::errors::Configuration, covM_, reco::LeafCandidate::energy(), Exception, TKinFitter::fit(), fittedB_, fittedBBar_, fittedLightP_, fittedLightPBar_, fittedLightQ_, fittedLightQBar_, TopKinFitter::fitter_, TAbsFitParticle::getCurr4Vec(), TKinFitter::getStatus(), jetParam_, TtFullHadEvtPartons::LightP, lightP_, TtFullHadEvtPartons::LightPBar, lightPBar_, TtFullHadEvtPartons::LightQ, lightQ_, TtFullHadEvtPartons::LightQBar, lightQBar_, reco::LeafCandidate::px(), reco::LeafCandidate::py(), reco::LeafCandidate::pz(), TAbsFitParticle::setCovMatrix(), TAbsFitParticle::setIni4Vec(), and CovarianceMatrix::setupMatrix().

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

                                                         {
  if (jets.size() < 6) {
    throw edm::Exception(edm::errors::Configuration, "Cannot run the TtFullHadKinFitter with less than 6 jets");
  }

  // get jets in right order
  const pat::Jet& b = jets[TtFullHadEvtPartons::B];
  const pat::Jet& bBar = jets[TtFullHadEvtPartons::BBar];
  const pat::Jet& lightQ = jets[TtFullHadEvtPartons::LightQ];
  const pat::Jet& lightQBar = jets[TtFullHadEvtPartons::LightQBar];
  const pat::Jet& lightP = jets[TtFullHadEvtPartons::LightP];
  const pat::Jet& lightPBar = jets[TtFullHadEvtPartons::LightPBar];

  // initialize particles
  const TLorentzVector p4B(b.px(), b.py(), b.pz(), b.energy());
  const TLorentzVector p4BBar(bBar.px(), bBar.py(), bBar.pz(), bBar.energy());
  const TLorentzVector p4LightQ(lightQ.px(), lightQ.py(), lightQ.pz(), lightQ.energy());
  const TLorentzVector p4LightQBar(lightQBar.px(), lightQBar.py(), lightQBar.pz(), lightQBar.energy());
  const TLorentzVector p4LightP(lightP.px(), lightP.py(), lightP.pz(), lightP.energy());
  const TLorentzVector p4LightPBar(lightPBar.px(), lightPBar.py(), lightPBar.pz(), lightPBar.energy());

  // initialize covariance matrices
  TMatrixD m1 = covM_->setupMatrix(lightQ, jetParam_);
  TMatrixD m2 = covM_->setupMatrix(lightQBar, jetParam_);
  TMatrixD m3 = covM_->setupMatrix(b, jetParam_, "bjets");
  TMatrixD m4 = covM_->setupMatrix(lightP, jetParam_);
  TMatrixD m5 = covM_->setupMatrix(lightPBar, jetParam_);
  TMatrixD m6 = covM_->setupMatrix(bBar, jetParam_, "bjets");

  // set the kinematics of the objects to be fitted
  b_->setIni4Vec(&p4B);
  bBar_->setIni4Vec(&p4BBar);
  lightQ_->setIni4Vec(&p4LightQ);
  lightQBar_->setIni4Vec(&p4LightQBar);
  lightP_->setIni4Vec(&p4LightP);
  lightPBar_->setIni4Vec(&p4LightPBar);

  // initialize covariance matrices
  lightQ_->setCovMatrix(&m1);
  lightQBar_->setCovMatrix(&m2);
  b_->setCovMatrix(&m3);
  lightP_->setCovMatrix(&m4);
  lightPBar_->setCovMatrix(&m5);
  bBar_->setCovMatrix(&m6);

  // perform the fit!
  fitter_->fit();

  // add fitted information to the solution
  if (fitter_->getStatus() == 0) {
    // read back jet kinematics
    fittedB_ = pat::Particle(reco::LeafCandidate(
        0,
        math::XYZTLorentzVector(
            b_->getCurr4Vec()->X(), b_->getCurr4Vec()->Y(), b_->getCurr4Vec()->Z(), b_->getCurr4Vec()->E()),
        math::XYZPoint()));
    fittedLightQ_ = pat::Particle(reco::LeafCandidate(0,
                                                      math::XYZTLorentzVector(lightQ_->getCurr4Vec()->X(),
                                                                              lightQ_->getCurr4Vec()->Y(),
                                                                              lightQ_->getCurr4Vec()->Z(),
                                                                              lightQ_->getCurr4Vec()->E()),
                                                      math::XYZPoint()));
    fittedLightQBar_ = pat::Particle(reco::LeafCandidate(0,
                                                         math::XYZTLorentzVector(lightQBar_->getCurr4Vec()->X(),
                                                                                 lightQBar_->getCurr4Vec()->Y(),
                                                                                 lightQBar_->getCurr4Vec()->Z(),
                                                                                 lightQBar_->getCurr4Vec()->E()),
                                                         math::XYZPoint()));

    fittedBBar_ = pat::Particle(reco::LeafCandidate(
        0,
        math::XYZTLorentzVector(
            bBar_->getCurr4Vec()->X(), bBar_->getCurr4Vec()->Y(), bBar_->getCurr4Vec()->Z(), bBar_->getCurr4Vec()->E()),
        math::XYZPoint()));
    fittedLightP_ = pat::Particle(reco::LeafCandidate(0,
                                                      math::XYZTLorentzVector(lightP_->getCurr4Vec()->X(),
                                                                              lightP_->getCurr4Vec()->Y(),
                                                                              lightP_->getCurr4Vec()->Z(),
                                                                              lightP_->getCurr4Vec()->E()),
                                                      math::XYZPoint()));
    fittedLightPBar_ = pat::Particle(reco::LeafCandidate(0,
                                                         math::XYZTLorentzVector(lightPBar_->getCurr4Vec()->X(),
                                                                                 lightPBar_->getCurr4Vec()->Y(),
                                                                                 lightPBar_->getCurr4Vec()->Z(),
                                                                                 lightPBar_->getCurr4Vec()->E()),
                                                         math::XYZPoint()));
  }
  return fitter_->getStatus();
}

const pat::Particle TtFullHadKinFitter::fittedB ( ) const

inline

return fitted b quark candidate

Definition at line 69 of file TtFullHadKinFitter.h.

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

Referenced by TtFullHadKinFitter::KinFit::fit().

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

const pat::Particle TtFullHadKinFitter::fittedBBar ( ) const

inline

return fitted b quark candidate

Definition at line 71 of file TtFullHadKinFitter.h.

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

Referenced by TtFullHadKinFitter::KinFit::fit().

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

const pat::Particle TtFullHadKinFitter::fittedLightP ( ) const

inline

return fitted light quark candidate

Definition at line 79 of file TtFullHadKinFitter.h.

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

Referenced by TtFullHadKinFitter::KinFit::fit().

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

const pat::Particle TtFullHadKinFitter::fittedLightPBar ( ) const

inline

return fitted light quark candidate

Definition at line 81 of file TtFullHadKinFitter.h.

References addKinFitInfo(), fittedLightPBar_, TopKinFitter::fitter_, TKinFitter::getStatus(), printSetup(), setupConstraints(), setupFitter(), and setupJets().

Referenced by TtFullHadKinFitter::KinFit::fit().

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

const pat::Particle TtFullHadKinFitter::fittedLightQ ( ) const

inline

return fitted light quark candidate

Definition at line 73 of file TtFullHadKinFitter.h.

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

Referenced by TtFullHadKinFitter::KinFit::fit().

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

const pat::Particle TtFullHadKinFitter::fittedLightQBar ( ) const

inline

return fitted light quark candidate

Definition at line 75 of file TtFullHadKinFitter.h.

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

Referenced by TtFullHadKinFitter::KinFit::fit().

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

std::vector< TtFullHadKinFitter::Constraint > TtFullHadKinFitter::intToConstraint

(

const std::vector< unsigned int > &

constraints

)

used to convert vector of int's to vector of constraints (just used in TtFullHadKinFitter(int, int, double, double, std::vector<unsigned int>))

Definition at line 33 of file TtFullHadKinFitter.cc.

References mps_fire::i.

                                                {
  std::vector<TtFullHadKinFitter::Constraint> cConstraints;
  cConstraints.resize(constraints.size());
  for (unsigned int i = 0; i < constraints.size(); ++i) {
    cConstraints[i] = (Constraint)constraints[i];
  }
  return cConstraints;
}

void TtFullHadKinFitter::printSetup ( ) const

private

print fitter setup

Definition at line 113 of file TtFullHadKinFitter.cc.

References constraints_, mps_fire::i, jetParam_, kEqualTopMasses, kTopBarMass, kTopMass, kWMinusMass, kWPlusMass, TopKinFitter::maxDeltaS_, TopKinFitter::maxF_, TopKinFitter::maxNrIter_, TopKinFitter::mTop_, TopKinFitter::mW_, and TopKinFitter::param().

Referenced by fittedLightPBar(), and setupFitter().

                                          {
  std::stringstream constr;
  for (unsigned int i = 0; i < constraints_.size(); ++i) {
    switch (constraints_[i]) {
      case kWPlusMass:
        constr << "    * W+-mass   (" << mW_ << " GeV) \n";
        break;
      case kWMinusMass:
        constr << "    * W--mass   (" << mW_ << " GeV) \n";
        break;
      case kTopMass:
        constr << "    * t-mass    (" << mTop_ << " GeV) \n";
        break;
      case kTopBarMass:
        constr << "    * tBar-mass (" << mTop_ << " GeV) \n";
        break;
      case kEqualTopMasses:
        constr << "    * equal t-masses \n";
        break;
    }
  }
  edm::LogVerbatim("TtFullHadKinFitter") << "\n"
                                         << "+++++++++++ TtFullHadKinFitter Setup ++++++++++++ \n"
                                         << "  Parametrization:                                \n"
                                         << "   * jet : " << param(jetParam_) << "\n"
                                         << "  Constraints:                                    \n"
                                         << constr.str() << "  Max(No iterations): " << maxNrIter_ << "\n"
                                         << "  Max(deltaS)       : " << maxDeltaS_ << "\n"
                                         << "  Max(F)            : " << maxF_ << "\n"
                                         << "+++++++++++++++++++++++++++++++++++++++++++++++++ \n";
}

void TtFullHadKinFitter::setupConstraints ( )

private

initialize constraints

Definition at line 178 of file TtFullHadKinFitter.cc.

References b_, bBar_, kEqualTopMasses, kTopBarMass, kTopMass, kWMinusMass, kWPlusMass, lightP_, lightPBar_, lightQ_, lightQBar_, massConstr_, TopKinFitter::mTop_, and TopKinFitter::mW_.

Referenced by fittedLightPBar(), and setupFitter().

                                          {
  massConstr_[kWPlusMass] = new TFitConstraintM("WPlusMass", "WPlusMass", nullptr, nullptr, mW_);
  massConstr_[kWMinusMass] = new TFitConstraintM("WMinusMass", "WMinusMass", nullptr, nullptr, mW_);
  massConstr_[kTopMass] = new TFitConstraintM("TopMass", "TopMass", nullptr, nullptr, mTop_);
  massConstr_[kTopBarMass] = new TFitConstraintM("TopBarMass", "TopBarMass", nullptr, nullptr, mTop_);
  massConstr_[kEqualTopMasses] = new TFitConstraintM("EqualTopMasses", "EqualTopMasses", nullptr, nullptr, 0);

  massConstr_[kWPlusMass]->addParticles1(lightQ_, lightQBar_);
  massConstr_[kWMinusMass]->addParticles1(lightP_, lightPBar_);
  massConstr_[kTopMass]->addParticles1(b_, lightQ_, lightQBar_);
  massConstr_[kTopBarMass]->addParticles1(bBar_, lightP_, lightPBar_);
  massConstr_[kEqualTopMasses]->addParticles1(b_, lightQ_, lightQBar_);
  massConstr_[kEqualTopMasses]->addParticles2(bBar_, lightP_, lightPBar_);
}

void TtFullHadKinFitter::setupFitter ( )

private

setup fitter

Definition at line 194 of file TtFullHadKinFitter.cc.

References TKinFitter::addConstraint(), TKinFitter::addMeasParticle(), b_, bBar_, bResolutions_, constraints_, covM_, TopKinFitter::fitter_, mps_fire::i, jetEnergyResolutionEtaBinning_, jetEnergyResolutionScaleFactors_, lightP_, lightPBar_, lightQ_, lightQBar_, massConstr_, printSetup(), setupConstraints(), setupJets(), and udscResolutions_.

Referenced by fittedLightPBar(), and TtFullHadKinFitter().

                                     {
  printSetup();
  setupJets();
  setupConstraints();

  // add measured particles
  fitter_->addMeasParticle(b_);
  fitter_->addMeasParticle(bBar_);
  fitter_->addMeasParticle(lightQ_);
  fitter_->addMeasParticle(lightQBar_);
  fitter_->addMeasParticle(lightP_);
  fitter_->addMeasParticle(lightPBar_);

  // add constraints
  for (unsigned int i = 0; i < constraints_.size(); i++) {
    fitter_->addConstraint(massConstr_[constraints_[i]]);
  }

  // initialize helper class used to bring the resolutions into covariance matrices
  if (!udscResolutions_->empty() && !bResolutions_->empty())
    covM_ = new CovarianceMatrix(
        *udscResolutions_, *bResolutions_, *jetEnergyResolutionScaleFactors_, *jetEnergyResolutionEtaBinning_);
  else
    covM_ = new CovarianceMatrix();
}

void TtFullHadKinFitter::setupJets ( )

private

initialize jet inputs

Definition at line 146 of file TtFullHadKinFitter.cc.

References b_, bBar_, jetParam_, TopKinFitter::kEMom, TopKinFitter::kEtEtaPhi, TopKinFitter::kEtThetaPhi, lightP_, lightPBar_, lightQ_, and lightQBar_.

Referenced by fittedLightPBar(), and setupFitter().

                                   {
  TMatrixD empty3x3(3, 3);
  TMatrixD empty4x4(4, 4);
  switch (jetParam_) {  // setup jets according to parameterization
    case kEMom:
      b_ = new TFitParticleEMomDev("Jet1", "Jet1", nullptr, &empty4x4);
      bBar_ = new TFitParticleEMomDev("Jet2", "Jet2", nullptr, &empty4x4);
      lightQ_ = new TFitParticleEMomDev("Jet3", "Jet3", nullptr, &empty4x4);
      lightQBar_ = new TFitParticleEMomDev("Jet4", "Jet4", nullptr, &empty4x4);
      lightP_ = new TFitParticleEMomDev("Jet5", "Jet5", nullptr, &empty4x4);
      lightPBar_ = new TFitParticleEMomDev("Jet6", "Jet6", nullptr, &empty4x4);
      break;
    case kEtEtaPhi:
      b_ = new TFitParticleEtEtaPhi("Jet1", "Jet1", nullptr, &empty3x3);
      bBar_ = new TFitParticleEtEtaPhi("Jet2", "Jet2", nullptr, &empty3x3);
      lightQ_ = new TFitParticleEtEtaPhi("Jet3", "Jet3", nullptr, &empty3x3);
      lightQBar_ = new TFitParticleEtEtaPhi("Jet4", "Jet4", nullptr, &empty3x3);
      lightP_ = new TFitParticleEtEtaPhi("Jet5", "Jet5", nullptr, &empty3x3);
      lightPBar_ = new TFitParticleEtEtaPhi("Jet6", "Jet6", nullptr, &empty3x3);
      break;
    case kEtThetaPhi:
      b_ = new TFitParticleEtThetaPhi("Jet1", "Jet1", nullptr, &empty3x3);
      bBar_ = new TFitParticleEtThetaPhi("Jet2", "Jet2", nullptr, &empty3x3);
      lightQ_ = new TFitParticleEtThetaPhi("Jet3", "Jet3", nullptr, &empty3x3);
      lightQBar_ = new TFitParticleEtThetaPhi("Jet4", "Jet4", nullptr, &empty3x3);
      lightP_ = new TFitParticleEtThetaPhi("Jet5", "Jet5", nullptr, &empty3x3);
      lightPBar_ = new TFitParticleEtThetaPhi("Jet6", "Jet6", nullptr, &empty3x3);
      break;
  }
}

Member Data Documentation

TAbsFitParticle* TtFullHadKinFitter::b_

private

input particles

Definition at line 99 of file TtFullHadKinFitter.h.

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

TAbsFitParticle* TtFullHadKinFitter::bBar_

private

Definition at line 100 of file TtFullHadKinFitter.h.

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

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

private

Definition at line 107 of file TtFullHadKinFitter.h.

Referenced by setupFitter().

std::vector<Constraint> TtFullHadKinFitter::constraints_

private

vector of constraints to be used

Definition at line 123 of file TtFullHadKinFitter.h.

Referenced by printSetup(), and setupFitter().

CovarianceMatrix* TtFullHadKinFitter::covM_

private

get object resolutions and put them into a matrix

Definition at line 126 of file TtFullHadKinFitter.h.

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

pat::Particle TtFullHadKinFitter::fittedB_

private

output particles

Definition at line 114 of file TtFullHadKinFitter.h.

Referenced by addKinFitInfo(), fit(), and fittedB().

pat::Particle TtFullHadKinFitter::fittedBBar_

private

Definition at line 115 of file TtFullHadKinFitter.h.

Referenced by addKinFitInfo(), fit(), and fittedBBar().

pat::Particle TtFullHadKinFitter::fittedLightP_

private

Definition at line 118 of file TtFullHadKinFitter.h.

Referenced by addKinFitInfo(), fit(), and fittedLightP().

pat::Particle TtFullHadKinFitter::fittedLightPBar_

private

Definition at line 119 of file TtFullHadKinFitter.h.

Referenced by addKinFitInfo(), fit(), and fittedLightPBar().

pat::Particle TtFullHadKinFitter::fittedLightQ_

private

Definition at line 116 of file TtFullHadKinFitter.h.

Referenced by addKinFitInfo(), fit(), and fittedLightQ().

pat::Particle TtFullHadKinFitter::fittedLightQBar_

private

Definition at line 117 of file TtFullHadKinFitter.h.

Referenced by addKinFitInfo(), fit(), and fittedLightQBar().

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

private

Definition at line 110 of file TtFullHadKinFitter.h.

Referenced by setupFitter().

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

private

scale factors for the jet energy resolution

Definition at line 109 of file TtFullHadKinFitter.h.

Referenced by setupFitter().

Param TtFullHadKinFitter::jetParam_

private

jet parametrization

Definition at line 121 of file TtFullHadKinFitter.h.

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

TAbsFitParticle* TtFullHadKinFitter::lightP_

private

Definition at line 103 of file TtFullHadKinFitter.h.

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

TAbsFitParticle* TtFullHadKinFitter::lightPBar_

private

Definition at line 104 of file TtFullHadKinFitter.h.

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

TAbsFitParticle* TtFullHadKinFitter::lightQ_

private

Definition at line 101 of file TtFullHadKinFitter.h.

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

TAbsFitParticle* TtFullHadKinFitter::lightQBar_

private

Definition at line 102 of file TtFullHadKinFitter.h.

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

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

private

supported constraints

Definition at line 112 of file TtFullHadKinFitter.h.

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

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

private

resolutions

Definition at line 106 of file TtFullHadKinFitter.h.

Referenced by setupFitter().