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 24 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 33 of file TFitConstraintM.cc.

References _ParList1, _ParList2, _TheMassConstraint, and bwdGsfElectronPropagator_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 60 of file TFitConstraintM.cc.

References _ParList1, _ParList2, _TheMassConstraint, and bwdGsfElectronPropagator_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 145 of file TFitConstraintM.cc.

                                  {

}

Member Function Documentation

void TFitConstraintM::addParticle1

(

TAbsFitParticle *

particle

)

Definition at line 87 of file TFitConstraintM.cc.

References _ParList1.

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

                                                              {
  // 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 95 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 103 of file TFitConstraintM.cc.

References addParticle1().

Referenced by TSLToyGen::doToyExperiments(), and StKinFitter::setupFitter().

                                                                {
  // 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 238 of file TFitConstraintM.cc.

References mps_fire::i.

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

                                                                                   {
  // 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 216 of file TFitConstraintM.cc.

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

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

                                          {
  // 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 152 of file TFitConstraintM.cc.

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

Referenced by setMassConstraint().

                                                                    {
  // 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 254 of file TFitConstraintM.cc.

References _TheMassConstraint, getCurrentValue(), getInitValue(), and info().

Referenced by print(), and setMassConstraint().

                                       {
  // 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 208 of file TFitConstraintM.cc.

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

Referenced by getInfoString(), and setMassConstraint().

                                       {
  // 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 225 of file TFitConstraintM.cc.

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

Referenced by getDerivative(), and setMassConstraint().

                                              {
  // 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 272 of file TFitConstraintM.cc.

References getInfoString().

Referenced by setMassConstraint().

                            {
  // Print constraint contents

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

}

void TFitConstraintM::setMassConstraint ( Double_t  Mass )

inline

Definition at line 34 of file TFitConstraintM.h.

References _TheMassConstraint, CalcMass(), getCurrentValue(), getDerivative(), getInfoString(), getInitValue(), bwdGsfElectronPropagator_cff::Mass, OnList(), and print().

{ _TheMassConstraint = Mass; }

Member Data Documentation

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

protected

Definition at line 50 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 51 of file TFitConstraintM.h.

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

Double_t TFitConstraintM::_TheMassConstraint

protected

Definition at line 52 of file TFitConstraintM.h.

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