TFitConstraintEp Class Reference

#include <TFitConstraintEp.h>

Inheritance diagram for TFitConstraintEp:

Public Types
enum	component { pX, pY, pZ, E }

Public Member Functions
void	addParticle (TAbsFitParticle *particle)
void	addParticles (TAbsFitParticle *p1, TAbsFitParticle *p2=nullptr, TAbsFitParticle *p3=nullptr, TAbsFitParticle *p4=nullptr, TAbsFitParticle *p5=nullptr, TAbsFitParticle *p6=nullptr, TAbsFitParticle *p7=nullptr, TAbsFitParticle *p8=nullptr, TAbsFitParticle *p9=nullptr, TAbsFitParticle *p10=nullptr)
Double_t	getCurrentValue () override
TMatrixD *	getDerivative (TAbsFitParticle *particle) override
TString	getInfoString () override
Double_t	getInitValue () override
void	print () override
void	setConstraint (Double_t constraint)
	TFitConstraintEp ()
	TFitConstraintEp (const TString &name, const TString &title, TFitConstraintEp::component thecomponent, Double_t constraint=0.)
	TFitConstraintEp (std::vector< TAbsFitParticle * > *particles, TFitConstraintEp::component thecomponent, Double_t constraint=0.)
	TFitConstraintEp (const TString &name, const TString &title, std::vector< TAbsFitParticle * > *particles, TFitConstraintEp::component thecomponent, Double_t constraint=0.)
	~TFitConstraintEp () override
Public Member Functions inherited from TAbsFitConstraint
virtual void	applyDeltaAlpha (TMatrixD *corrMatrix)
virtual const TMatrixD *	getCovMatrix () const
virtual const TMatrixD *	getCovMatrixDeltaAlpha ()
virtual const TMatrixD *	getCovMatrixFit () const
virtual TMatrixD *	getDerivativeAlpha ()
Int_t	getNPar ()
const TMatrixD *	getParCurr ()
const TMatrixD *	getParIni ()
virtual void	reset ()
virtual void	setCovMatrix (const TMatrixD *theCovMatrix)
virtual void	setCovMatrixFit (const TMatrixD *theCovMatrixFit)
void	setParIni (const TMatrixD *parini)
	TAbsFitConstraint ()
	TAbsFitConstraint (const TString &name, const TString &title)
	~TAbsFitConstraint () override

Private Attributes
TFitConstraintEp::component	_component
Double_t	_constraint
std::vector< TAbsFitParticle * >	_particles

Additional Inherited Members
Protected Member Functions inherited from TAbsFitConstraint
void	calcCovMatrixDeltaAlpha ()
Protected Attributes inherited from TAbsFitConstraint
TMatrixD	_covMatrix
TMatrixD	_covMatrixDeltaAlpha
TMatrixD	_covMatrixFit
TMatrixD	_iniparameters
Int_t	_nPar
TMatrixD	_parameters

Detailed Description

Definition at line 9 of file TFitConstraintEp.h.

Member Enumeration Documentation

enum TFitConstraintEp::component

Enumerator
pX
pY
pZ
E

Definition at line 11 of file TFitConstraintEp.h.

{ pX, pY, pZ, E };

Constructor & Destructor Documentation

TFitConstraintEp::TFitConstraintEp

(

)

Definition at line 22 of file TFitConstraintEp.cc.

    : TAbsFitConstraint(), _particles(0), _constraint(0.), _component(TFitConstraintEp::pX) {}

TFitConstraintEp::TFitConstraintEp	(	const TString &	name,
		const TString &	title,
		TFitConstraintEp::component	thecomponent,
		Double_t	constraint = 0.
	)

Definition at line 25 of file TFitConstraintEp.cc.

    : TAbsFitConstraint(name, title), _constraint(constraint), _component(thecomponent) {}

TFitConstraintEp::TFitConstraintEp	(	std::vector< TAbsFitParticle * > *	particles,
		TFitConstraintEp::component	thecomponent,
		Double_t	constraint = 0.
	)

Definition at line 31 of file TFitConstraintEp.cc.

References _particles.

    : TAbsFitConstraint(), _particles(0), _constraint(constraint), _component(thecomponent) {
  // particles: vector containing pointer to TAbsFitParticle objects.
  //            Energy or momentum conservation will be calculated for
  //            those particles.
  // thecomponent: conserved 4vector component ( pX, pY, pZ, E ). For
  //            full 4vector conservation four objects of type TFitConstraintEp
  //            are needed (four constraints)
  // constraint:value of energy or momentum constraint ( e.g. sum[ pX_i ] = constraint )

  if (particles) {
    _particles = (*particles);
  }
}

TFitConstraintEp::TFitConstraintEp	(	const TString &	name,
		const TString &	title,
		std::vector< TAbsFitParticle * > *	particles,
		TFitConstraintEp::component	thecomponent,
		Double_t	constraint = 0.
	)

Definition at line 48 of file TFitConstraintEp.cc.

References _particles.

    : TAbsFitConstraint(name, title), _particles(0), _constraint(constraint), _component(thecomponent) {
  // particles: vector containing pointer to TAbsFitParticle objects.
  //            Energy or momentum conservation will be calculated for
  //            those particles.
  // thecomponent: conserved 4vector component ( pX, pY, pZ, E ). For
  //            full 4vector conservation four objects of type TFitConstraintEp
  //            are needed (four constraints)
  // constraint:value of energy or momentum constraint ( e.g. sum[ pX_i ] = constraint )

  if (particles) {
    _particles = (*particles);
  }
}

TFitConstraintEp::~TFitConstraintEp ( )

override

Definition at line 70 of file TFitConstraintEp.cc.

{}

Member Function Documentation

void TFitConstraintEp::addParticle

(

TAbsFitParticle *

particle

)

Definition at line 72 of file TFitConstraintEp.cc.

References _particles.

Referenced by addParticles().

                                                            {
  // Add one particles to list of constrained particles

  _particles.push_back(particle);
}

void TFitConstraintEp::addParticles	(	TAbsFitParticle *	p1,
		TAbsFitParticle *	p2 = nullptr,
		TAbsFitParticle *	p3 = nullptr,
		TAbsFitParticle *	p4 = nullptr,
		TAbsFitParticle *	p5 = nullptr,
		TAbsFitParticle *	p6 = nullptr,
		TAbsFitParticle *	p7 = nullptr,
		TAbsFitParticle *	p8 = nullptr,
		TAbsFitParticle *	p9 = nullptr,
		TAbsFitParticle *	p10 = nullptr
	)

Definition at line 78 of file TFitConstraintEp.cc.

References addParticle().

Referenced by TtSemiLepKinFitter::setupConstraints().

                                                          {
  // Add many particles to list of constrained particles

  if (p1)
    addParticle(p1);
  if (p2)
    addParticle(p2);
  if (p3)
    addParticle(p3);
  if (p4)
    addParticle(p4);
  if (p5)
    addParticle(p5);
  if (p6)
    addParticle(p6);
  if (p7)
    addParticle(p7);
  if (p8)
    addParticle(p8);
  if (p9)
    addParticle(p9);
  if (p10)
    addParticle(p10);
}

Double_t TFitConstraintEp::getCurrentValue ( )

overridevirtual

Implements TAbsFitConstraint.

Definition at line 138 of file TFitConstraintEp.cc.

References _component, _constraint, _particles, and mps_fire::i.

Referenced by getInfoString().

                                           {
  // Get value of constraint after the fit

  Double_t CurrentValue(0);
  UInt_t Npart = _particles.size();
  for (unsigned int i = 0; i < Npart; i++) {
    const TLorentzVector* FourVec = _particles[i]->getCurr4Vec();
    CurrentValue += (*FourVec)[(int)_component];
  }
  CurrentValue -= _constraint;
  return CurrentValue;
}

TMatrixD * TFitConstraintEp::getDerivative ( TAbsFitParticle *  particle )

overridevirtual

Implements TAbsFitConstraint.

Definition at line 115 of file TFitConstraintEp.cc.

References _component.

                                                                   {
  // returns derivative df/dP with P=(p,E) and f the constraint (f=0).
  // The matrix contains one row (df/dp, df/dE).

  TMatrixD* DerivativeMatrix = new TMatrixD(1, 4);
  (*DerivativeMatrix) *= 0.;
  (*DerivativeMatrix)(0, (int)_component) = 1.;
  return DerivativeMatrix;
}

TString TFitConstraintEp::getInfoString ( )

overridevirtual

Reimplemented from TAbsFitConstraint.

Definition at line 151 of file TFitConstraintEp.cc.

References _component, _constraint, getCurrentValue(), getInitValue(), and info().

Referenced by print().

                                        {
  // Collect information to be used for printout

  std::stringstream info;
  info << std::scientific << std::setprecision(6);

  info << "__________________________" << std::endl << std::endl;
  info << "OBJ: " << IsA()->GetName() << "\t" << GetName() << "\t" << GetTitle() << std::endl;

  info << "initial value: " << getInitValue() << std::endl;
  info << "current value: " << getCurrentValue() << std::endl;
  info << "component: " << _component << std::endl;
  info << "constraint: " << _constraint << std::endl;

  return info.str();
}

Double_t TFitConstraintEp::getInitValue ( )

overridevirtual

Implements TAbsFitConstraint.

Definition at line 125 of file TFitConstraintEp.cc.

References _component, _constraint, _particles, and mps_fire::i.

Referenced by getInfoString().

                                        {
  // Get initial value of constraint (before the fit)

  Double_t InitValue(0);
  UInt_t Npart = _particles.size();
  for (unsigned int i = 0; i < Npart; i++) {
    const TLorentzVector* FourVec = _particles[i]->getIni4Vec();
    InitValue += (*FourVec)[(int)_component];
  }
  InitValue -= _constraint;
  return InitValue;
}

void TFitConstraintEp::print ( void  )

overridevirtual

Reimplemented from TAbsFitConstraint.

Definition at line 168 of file TFitConstraintEp.cc.

References getInfoString().

                             {
  // Print constraint contents

  edm::LogVerbatim("KinFitter") << this->getInfoString();
}

void TFitConstraintEp::setConstraint ( Double_t  constraint )

inline

Definition at line 42 of file TFitConstraintEp.h.

References _constraint, and HLT_FULL_cff::constraint.

Referenced by TtSemiLepKinFitter::fit().

{ _constraint = constraint; };

Member Data Documentation

TFitConstraintEp::component TFitConstraintEp::_component

private

Definition at line 57 of file TFitConstraintEp.h.

Referenced by getCurrentValue(), getDerivative(), getInfoString(), and getInitValue().

Double_t TFitConstraintEp::_constraint

private

Definition at line 56 of file TFitConstraintEp.h.

Referenced by getCurrentValue(), getInfoString(), getInitValue(), and setConstraint().

std::vector<TAbsFitParticle*> TFitConstraintEp::_particles

private

Definition at line 55 of file TFitConstraintEp.h.

Referenced by addParticle(), getCurrentValue(), getInitValue(), and TFitConstraintEp().