TFitConstraintM Class Reference

#include <TFitConstraintM.h>

Inheritance diagram for TFitConstraintM:

Public Member Functions
void	addParticle1 (TAbsFitParticle *particle)
void	addParticle2 (TAbsFitParticle *particle)
void	addParticles1 (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)
void	addParticles2 (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	CalcMass (std::vector< TAbsFitParticle * > *List, Bool_t IniVal)
Double_t	getCurrentValue () override
TMatrixD *	getDerivative (TAbsFitParticle *particle) override
TString	getInfoString () override
Double_t	getInitValue () override
Bool_t	OnList (std::vector< TAbsFitParticle * > *List, TAbsFitParticle *particle)
void	print () override
void	setMassConstraint (Double_t Mass)
	TFitConstraintM ()
	TFitConstraintM (std::vector< TAbsFitParticle * > *ParList1, std::vector< TAbsFitParticle * > *ParList2, Double_t Mass=0)
	TFitConstraintM (const TString &name, const TString &title, std::vector< TAbsFitParticle * > *ParList1, std::vector< TAbsFitParticle * > *ParList2, Double_t Mass=0)
	~TFitConstraintM () 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

Protected Attributes
std::vector< TAbsFitParticle * >	_ParList1
std::vector< TAbsFitParticle * >	_ParList2
Double_t	_TheMassConstraint
Protected Attributes inherited from TAbsFitConstraint
TMatrixD	_covMatrix
TMatrixD	_covMatrixDeltaAlpha
TMatrixD	_covMatrixFit
TMatrixD	_iniparameters
Int_t	_nPar
TMatrixD	_parameters

Additional Inherited Members
Protected Member Functions inherited from TAbsFitConstraint
void	calcCovMatrixDeltaAlpha ()

Detailed Description

Definition at line 11 of file TFitConstraintM.h.

Constructor & Destructor Documentation

TFitConstraintM::TFitConstraintM

(

)

Definition at line 22 of file TFitConstraintM.cc.

: TAbsFitConstraint(), _ParList1(0), _ParList2(0), _TheMassConstraint(0) {}

TFitConstraintM::TFitConstraintM	(	std::vector< TAbsFitParticle * > *	ParList1,
		std::vector< TAbsFitParticle * > *	ParList2,
		Double_t	Mass = 0
	)

Definition at line 24 of file TFitConstraintM.cc.

References _ParList1, _ParList2, _TheMassConstraint, and HLT_FULL_cff::Mass.

    : TAbsFitConstraint(), _ParList1(0), _ParList2(0) {
  // ParList1: Vector containing first list of constrained particles
  //           ( sum[ m_i ] - sum[ m_j ] - MassConstraint == 0 )
  // ParList2: Vector containing second list of constrained particles
  //           ( sum[ m_i ] - sum[ m_j ]  - MassConstraint == 0 )

  if (ParList1) {
    _ParList1 = (*ParList1);
  }
  if (ParList2) {
    _ParList2 = (*ParList2);
  }
  if (Mass >= 0) {
    _TheMassConstraint = Mass;
  } else if (Mass < 0) {
    edm::LogWarning("NegativeMassConstr")
        << "Mass constraint in TFitConstraintM cannot be set to a negative value, will be set to 0.";
    _TheMassConstraint = 0.;
  }
}

TFitConstraintM::TFitConstraintM	(	const TString &	name,
		const TString &	title,
		std::vector< TAbsFitParticle * > *	ParList1,
		std::vector< TAbsFitParticle * > *	ParList2,
		Double_t	Mass = 0
	)

Definition at line 48 of file TFitConstraintM.cc.

References _ParList1, _ParList2, _TheMassConstraint, and HLT_FULL_cff::Mass.

    : TAbsFitConstraint(name, title), _ParList1(0), _ParList2(0) {
  // ParList1: Vector containing first list of constrained particles
  //           ( sum[ m_i ] - sum[ m_j ] - MassConstraint == 0 )
  // ParList2: Vector containing second list of constrained particles
  //           ( sum[ m_i ] - sum[ m_j ] - MassConstraint == 0 )

  if (ParList1) {
    _ParList1 = (*ParList1);
  }
  if (ParList2) {
    _ParList2 = (*ParList2);
  }
  if (Mass >= 0) {
    _TheMassConstraint = Mass;
  } else if (Mass < 0) {
    edm::LogWarning("NegativeMassConstr")
        << "Mass constraint in TFitConstraintM cannot be set to a negative value, will be set to 0.";
    _TheMassConstraint = 0.;
  }
}

TFitConstraintM::~TFitConstraintM ( )

override

Definition at line 160 of file TFitConstraintM.cc.

{}

Member Function Documentation

void TFitConstraintM::addParticle1

(

TAbsFitParticle *

particle

)

Definition at line 73 of file TFitConstraintM.cc.

References _ParList1.

Referenced by addParticles1(), TSLToyGen::doToyExperiments(), and CandMassKinFitter::set().

                                                            {
  // Add one constrained particle to first list of particles
  // ( sum[ m_i ] - sum[ m_j ] - MassConstraint == 0 )

  _ParList1.push_back(particle);
}

void TFitConstraintM::addParticle2

(

TAbsFitParticle *

particle

)

Definition at line 80 of file TFitConstraintM.cc.

References _ParList2.

Referenced by addParticles2().

                                                            {
  // Add one constrained particle to second list of particles
  // ( sum[ m_i ] - sum[ m_j ] - MassConstraint == 0 )

  _ParList2.push_back(particle);
}

void TFitConstraintM::addParticles1	(	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 87 of file TFitConstraintM.cc.

References addParticle1().

Referenced by TSLToyGen::doToyExperiments().

                                                          {
  // Add many constrained particle to first list of particles
  // ( sum[ m_i ] - sum[ m_j ] - MassConstraint == 0 )

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

void TFitConstraintM::addParticles2	(	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 122 of file TFitConstraintM.cc.

References addParticle2().

Referenced by TSLToyGen::doToyExperiments().

                                                          {
  // Add many constrained particle to second list of particles
  // ( sum[ m_i ] - sum[ m_j ] - MassConstraint == 0 )

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

Double_t TFitConstraintM::CalcMass	(	std::vector< TAbsFitParticle * > *	List,
		Bool_t	IniVal
	)

Definition at line 244 of file TFitConstraintM.cc.

References mps_fire::i.

Referenced by TFitConstraintMGaus::getCurrentValue(), getCurrentValue(), TFitConstraintMGaus::getInitValue(), and getInitValue().

                                                                                   {
  // Calculates initial/current invariant mass of provided list of particles

  TLorentzVector P(0., 0., 0., 0.);
  UInt_t Npart = List->size();
  for (unsigned int i = 0; i < Npart; i++) {
    const TLorentzVector* FourVec = nullptr;
    if (IniVal)
      FourVec = ((*List)[i])->getIni4Vec();
    else
      FourVec = ((*List)[i])->getCurr4Vec();
    P += (*FourVec);
  }
  return P.M();
}

Double_t TFitConstraintM::getCurrentValue ( )

overridevirtual

Implements TAbsFitConstraint.

Reimplemented in TFitConstraintMGaus.

Definition at line 223 of file TFitConstraintM.cc.

References _ParList1, _ParList2, _TheMassConstraint, and CalcMass().

Referenced by TSLToyGen::doToyExperiments(), and getInfoString().

                                          {
  // Get value of constraint after the fit

  Double_t CurrentValue(0);
  CurrentValue = CalcMass(&_ParList1, false) - CalcMass(&_ParList2, false) - _TheMassConstraint;
  return CurrentValue;
}

TMatrixD * TFitConstraintM::getDerivative ( TAbsFitParticle *  particle )

overridevirtual

Implements TAbsFitConstraint.

Definition at line 165 of file TFitConstraintM.cc.

References _ParList1, _ParList2, mps_fire::i, and OnList().

                                                                  {
  // 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.;

  // Pf[4] is the 4-Mom (p,E) of the sum of particles on
  // the list particle is part of

  Double_t Factor = 0.;
  TLorentzVector Pf(0., 0., 0., 0.);

  if (OnList(&_ParList1, particle)) {
    UInt_t Npart = _ParList1.size();
    for (unsigned int i = 0; i < Npart; i++) {
      const TLorentzVector* FourVec = (_ParList1[i])->getCurr4Vec();
      Pf += (*FourVec);
    }
    if (Pf.M() == 0.) {
      edm::LogInfo("KinFitter") << "Division by zero in " << IsA()->GetName() << " (named " << GetName() << ", titled "
                                << GetTitle() << ") will lead to Inf in derivative matrix for particle "
                                << particle->GetName() << ".";
    }
    Factor = 1. / Pf.M();
  } else if (OnList(&_ParList2, particle)) {
    UInt_t Npart = _ParList2.size();
    for (unsigned int i = 0; i < Npart; i++) {
      const TLorentzVector* FourVec = (_ParList2[i])->getCurr4Vec();
      Pf += (*FourVec);
    }
    if (Pf.M() == 0.) {
      edm::LogInfo("KinFitter") << "Division by zero in " << IsA()->GetName() << " (named " << GetName() << ", titled "
                                << GetTitle() << ") will lead to Inf in derivative matrix for particle "
                                << particle->GetName() << ".";
    }
    Factor = -1. / Pf.M();
  } else {
    Factor = 0.;
  }

  (*DerivativeMatrix)(0, 0) = -Pf[0];
  (*DerivativeMatrix)(0, 1) = -Pf[1];
  (*DerivativeMatrix)(0, 2) = -Pf[2];
  (*DerivativeMatrix)(0, 3) = +Pf[3];
  (*DerivativeMatrix) *= Factor;

  return DerivativeMatrix;
}

TString TFitConstraintM::getInfoString ( )

overridevirtual

Reimplemented from TAbsFitConstraint.

Reimplemented in TFitConstraintMGaus.

Definition at line 260 of file TFitConstraintM.cc.

References _TheMassConstraint, 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 << "mass: " << _TheMassConstraint << std::endl;

  return info.str();
}

Double_t TFitConstraintM::getInitValue ( )

overridevirtual

Implements TAbsFitConstraint.

Reimplemented in TFitConstraintMGaus.

Definition at line 215 of file TFitConstraintM.cc.

References _ParList1, _ParList2, _TheMassConstraint, and CalcMass().

Referenced by getInfoString().

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

  Double_t InitValue(0);
  InitValue = CalcMass(&_ParList1, true) - CalcMass(&_ParList2, true) - _TheMassConstraint;
  return InitValue;
}

Bool_t TFitConstraintM::OnList	(	std::vector< TAbsFitParticle * > *	List,
		TAbsFitParticle *	particle
	)

Definition at line 231 of file TFitConstraintM.cc.

References mps_fire::i, and convertSQLiteXML::ok.

Referenced by getDerivative().

                                                                                           {
  // Checks whether list contains given particle

  Bool_t ok(false);
  UInt_t Npart = List->size();
  for (unsigned int i = 0; i < Npart; i++) {
    ok = (particle == (*List)[i]);
    if (ok)
      break;
  }
  return ok;
}

void TFitConstraintM::print ( void  )

overridevirtual

Reimplemented from TAbsFitConstraint.

Reimplemented in TFitConstraintMGaus.

Definition at line 276 of file TFitConstraintM.cc.

References getInfoString().

                            {
  // Print constraint contents

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

void TFitConstraintM::setMassConstraint ( Double_t  Mass )

inline

Definition at line 45 of file TFitConstraintM.h.

References _TheMassConstraint, and HLT_FULL_cff::Mass.

{ _TheMassConstraint = Mass; }

Member Data Documentation

std::vector<TAbsFitParticle*> TFitConstraintM::_ParList1

protected

Definition at line 61 of file TFitConstraintM.h.

Referenced by addParticle1(), TFitConstraintMGaus::getCurrentValue(), getCurrentValue(), getDerivative(), TFitConstraintMGaus::getInitValue(), getInitValue(), and TFitConstraintM().

std::vector<TAbsFitParticle*> TFitConstraintM::_ParList2

protected

Definition at line 63 of file TFitConstraintM.h.

Referenced by addParticle2(), TFitConstraintMGaus::getCurrentValue(), getCurrentValue(), getDerivative(), TFitConstraintMGaus::getInitValue(), getInitValue(), and TFitConstraintM().

Double_t TFitConstraintM::_TheMassConstraint

protected

Definition at line 64 of file TFitConstraintM.h.

Referenced by TFitConstraintMGaus::getCurrentValue(), getCurrentValue(), TFitConstraintMGaus::getDerivativeAlpha(), TFitConstraintMGaus::getInfoString(), getInfoString(), TFitConstraintMGaus::getInitValue(), getInitValue(), TFitConstraintMGaus::setMassConstraint(), setMassConstraint(), and TFitConstraintM().