#include <TAbsFitParticle.h>

Inheritance diagram for TAbsFitParticle:

Public Member Functions
virtual void	applycorr (TMatrixD *corrMatrix)
virtual TLorentzVector *	calc4Vec (const TMatrixD *params)=0
virtual TAbsFitParticle *	clone (TString newname="") const =0
virtual const TMatrixD *	getCovMatrix () const
virtual const TMatrixD *	getCovMatrixDeltaY ()
virtual const TMatrixD *	getCovMatrixFit () const
const TLorentzVector *	getCurr4Vec ()
virtual TMatrixD *	getDerivative ()=0
TString	getInfoString ()
const TLorentzVector *	getIni4Vec ()
Int_t	getNPar () const
const TMatrixD *	getParCurr ()
const TMatrixD *	getParIni ()
virtual const TMatrixD *	getPull ()
const TVector3 *	getu1 ()
const TVector3 *	getu2 ()
const TVector3 *	getu3 ()
virtual void	print ()
virtual void	reset ()
virtual void	setCovMatrix (const TMatrixD *theCovMatrix)
virtual void	setCovMatrixFit (const TMatrixD *theCovMatrixFit)
virtual void	setIni4Vec (const TLorentzVector *pini)=0
void	setParIni (const TMatrixD *parini)
	TAbsFitParticle (const TString &name, const TString &title)
	TAbsFitParticle ()
virtual TMatrixD *	transform (const TLorentzVector &vec)=0
virtual	~TAbsFitParticle ()
Protected Member Functions
void	calcCovMatrixDeltaY ()
Protected Attributes
TMatrixD	_covMatrix
TMatrixD	_covMatrixDeltaY
TMatrixD	_covMatrixFit
TMatrixD	_iniparameters
Int_t	_nPar
TMatrixD	_parameters
TLorentzVector	_pcurr
TLorentzVector	_pini
TMatrixD	_pull
TVector3	_u1
TVector3	_u2
TVector3	_u3

Detailed Description

Definition at line 9 of file TAbsFitParticle.h.

Constructor & Destructor Documentation

TAbsFitParticle::TAbsFitParticle ( )

Definition at line 20 of file TAbsFitParticle.cc.

                                :
  TNamed("NoName","NoTitle")
  ,_nPar(0)     
  ,_u1()   
  ,_u2()
  ,_u3()
  ,_covMatrix()
  ,_covMatrixFit()
  ,_covMatrixDeltaY()
  ,_pull()
  ,_iniparameters(1,1)
  ,_parameters(1,1)
  ,_pini()
  ,_pcurr()
{
 
}

TAbsFitParticle::TAbsFitParticle	(	const TString &	name,
		const TString &	title
	)

Definition at line 38 of file TAbsFitParticle.cc.

                                                                          :
  TNamed(name,title)
  ,_nPar(0)     
  ,_u1()   
  ,_u2()
  ,_u3()
  ,_covMatrix()
  ,_covMatrixFit()
  ,_covMatrixDeltaY()
  ,_pull()
  ,_iniparameters(1,1)
  ,_parameters(1,1)
  ,_pini()
  ,_pcurr()
{
 
}

TAbsFitParticle::~TAbsFitParticle ( ) [virtual]

Definition at line 57 of file TAbsFitParticle.cc.

                                  {

}

Member Function Documentation

void TAbsFitParticle::applycorr ( TMatrixD * corrMatrix ) [virtual]

Definition at line 186 of file TAbsFitParticle.cc.

References _iniparameters, _parameters, _pcurr, and calc4Vec().

Referenced by TKinFitter::applyDeltaA(), and TKinFitter::applyDeltaY().

                                                    {
  // Apply corrections to the parameters wrt. to the
  // initial parameters y* = y + delta(y)
  // This method will also calculate the fitted 
  // 4vector of the particle

  // update _parameters-Matrix
  _parameters = _iniparameters;
  _parameters += (*corrMatrix);

  // calculates new 4vec
  TLorentzVector* vec = calc4Vec( &_parameters );
  _pcurr = (*vec);
  delete vec;

}

virtual TLorentzVector* TAbsFitParticle::calc4Vec ( const TMatrixD * params ) [pure virtual]

Implemented in TFitParticleCart, TFitParticleECart, TFitParticleEMomDev, TFitParticleEScaledMomDev, TFitParticleESpher, TFitParticleEtEtaPhi, TFitParticleEtThetaPhi, TFitParticleMCCart, TFitParticleMCMomDev, TFitParticleMCPInvSpher, TFitParticleMCSpher, TFitParticleMomDev, and TFitParticleSpher.

Referenced by applycorr(), and TSLToyGen::smearParticles().

void TAbsFitParticle::calcCovMatrixDeltaY ( ) [protected]

Definition at line 150 of file TAbsFitParticle.cc.

References _covMatrix, _covMatrixDeltaY, _covMatrixFit, and _nPar.

Referenced by getCovMatrixDeltaY(), and getPull().

                                          {
  // Calculates V(deltaY) ==  V(y_meas) - V(y_fit)

  _covMatrixDeltaY.ResizeTo( _nPar, _nPar );
  _covMatrixDeltaY = _covMatrix;
  if(_covMatrixFit.GetNrows() == _nPar && _covMatrixFit.GetNcols() == _nPar)
    _covMatrixDeltaY -= _covMatrixFit;
  else 
    edm::LogError ("WrongMatrixSize")
      << GetName() << "::calcCovMatrixDeltaY - _covMatrixFit probably not set.";
}

virtual TAbsFitParticle* TAbsFitParticle::clone ( TString newname = "" ) const [pure virtual]

Implemented in TFitParticleCart, TFitParticleECart, TFitParticleEMomDev, TFitParticleEScaledMomDev, TFitParticleESpher, TFitParticleEtEtaPhi, TFitParticleEtThetaPhi, TFitParticleMCCart, TFitParticleMCMomDev, TFitParticleMCPInvSpher, TFitParticleMCSpher, TFitParticleMomDev, and TFitParticleSpher.

Referenced by TSLToyGen::TSLToyGen().

virtual const TMatrixD* TAbsFitParticle::getCovMatrix ( ) const [inline, virtual]

Definition at line 26 of file TAbsFitParticle.h.

References _covMatrix.

Referenced by TKinFitter::calcV(), TKinFitter::print(), and TSLToyGen::smearParticles().

{ return &_covMatrix; }

const TMatrixD * TAbsFitParticle::getCovMatrixDeltaY ( ) [virtual]

Definition at line 215 of file TAbsFitParticle.cc.

References _covMatrixDeltaY, and calcCovMatrixDeltaY().

                                                    {
  //
  calcCovMatrixDeltaY(); 
  return &_covMatrixDeltaY; 
}

virtual const TMatrixD* TAbsFitParticle::getCovMatrixFit ( ) const [inline, virtual]

Definition at line 28 of file TAbsFitParticle.h.

References _covMatrixFit.

{ return &_covMatrixFit; }

const TLorentzVector* TAbsFitParticle::getCurr4Vec ( ) [inline]

Definition at line 33 of file TAbsFitParticle.h.

References _pcurr.

Referenced by StKinFitter::addKinFitInfo(), TSLToyGen::fillM(), TtFullHadKinFitter::fit(), TtSemiLepKinFitter::fit(), getInfoString(), and CandMassKinFitter::set().

{ return &_pcurr; }

virtual TMatrixD* TAbsFitParticle::getDerivative ( ) [pure virtual]

Implemented in TFitParticleCart, TFitParticleECart, TFitParticleEMomDev, TFitParticleEScaledMomDev, TFitParticleESpher, TFitParticleEtEtaPhi, TFitParticleEtThetaPhi, TFitParticleMCCart, TFitParticleMCMomDev, TFitParticleMCPInvSpher, TFitParticleMCSpher, TFitParticleMomDev, and TFitParticleSpher.

Referenced by TKinFitter::calcA(), and TKinFitter::calcB().

TString TAbsFitParticle::getInfoString ( )

Definition at line 62 of file TAbsFitParticle.cc.

References _nPar, _u1, _u2, _u3, getCurr4Vec(), getIni4Vec(), getParCurr(), getParIni(), i, and info.

Referenced by print(), and TKinFitter::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 << std::setw(22) << "initial parameters:"  << std::setw(5) << " " << std::setw(20) << "current parameters:" << std::endl;
  for (int i = 0; i< _nPar ;i++){
    info << "par[" << i << "] = "
         << std::setw(18) << (*getParIni())(i,0) 
         << std::setw(20) << (*getParCurr())(i,0) << std::endl;
  }

  info << std::setw(22) << "initial 4vector:" << std::setw(5) << " " << std::setw(20) << "current 4vector:" << std::endl;
  for (int i = 0; i< 4 ;i++){
    info << "p[" << i << "] = "
         << std::setw(20) << (*getIni4Vec())[i] 
         << std::setw(20) << (*getCurr4Vec())[i] << std::endl;
   }
  info << "mass = " 
       << std::setw(20) << (*getIni4Vec()).M() 
       << std::setw(20) << (*getCurr4Vec()).M() << std::endl;

   info << "u1  = " << _u1.X() << ", " << _u1.Y() << ", " << _u1.Z() << std::endl;
   info << "u2  = " << _u2.X() << ", " << _u2.Y() << ", " << _u2.Z() << std::endl;
   info << "u3  = " << _u3.X() << ", " << _u3.Y() << ", " << _u3.Z() << std::endl;

   return info.str();

}

const TLorentzVector* TAbsFitParticle::getIni4Vec ( ) [inline]

Definition at line 34 of file TAbsFitParticle.h.

References _pini.

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

{ return &_pini; }

Int_t TAbsFitParticle::getNPar ( ) const [inline]

Definition at line 32 of file TAbsFitParticle.h.

References _nPar.

Referenced by TKinFitter::applyDeltaA(), TKinFitter::applyDeltaY(), TKinFitter::applyVFit(), TKinFitter::calcV(), TSLToyGen::createHists(), TKinFitter::print(), and TSLToyGen::smearParticles().

{ return _nPar; }

const TMatrixD* TAbsFitParticle::getParCurr ( ) [inline]

Definition at line 39 of file TAbsFitParticle.h.

References _parameters.

Referenced by TKinFitter::calcC(), getInfoString(), and TKinFitter::print().

{ return &_parameters; }

const TMatrixD* TAbsFitParticle::getParIni ( ) [inline]

Definition at line 37 of file TAbsFitParticle.h.

References _iniparameters.

Referenced by TKinFitter::calcC(), getInfoString(), and TSLToyGen::smearParticles().

{ return &_iniparameters; }

const TMatrixD * TAbsFitParticle::getPull ( ) [virtual]

Definition at line 162 of file TAbsFitParticle.cc.

References _covMatrixDeltaY, _iniparameters, _nPar, _parameters, _pull, calcCovMatrixDeltaY(), and i.

                                         {
  // Calculates the pull (y_fit - y_meas) / sigma
  // with sigma = Sqrt( sigma[y_meas]^2 - V[y_fit]^2 )
  // for all parameters

  _pull.ResizeTo( _nPar, 1 );
  _pull = _parameters;
  _pull -= _iniparameters;
  calcCovMatrixDeltaY(); 
  for (int i = 0; i<_nPar; i++) {
    Double_t sigmaDeltaY = _covMatrixDeltaY(i, i);
    if (sigmaDeltaY < 0) {
      edm::LogWarning ("NegativeDiagonalElem") << "V[deltaY] has a negative diagonal element.";
      _pull.Zero();
      return &_pull;
    } else {
      _pull(i,0) /= TMath::Sqrt( sigmaDeltaY );
    }
  }

  return &_pull;

}

const TVector3* TAbsFitParticle::getu1 ( ) [inline]

Definition at line 40 of file TAbsFitParticle.h.

References _u1.

{ return &_u1; }

const TVector3* TAbsFitParticle::getu2 ( ) [inline]

Definition at line 41 of file TAbsFitParticle.h.

References _u2.

{ return &_u2; }

const TVector3* TAbsFitParticle::getu3 ( ) [inline]

Definition at line 42 of file TAbsFitParticle.h.

References _u3.

{ return &_u3; }

void TAbsFitParticle::print ( void ) [virtual]

Definition at line 99 of file TAbsFitParticle.cc.

References getInfoString().

Referenced by TSLToyGen::doToyExperiments().

                       {
  // Print particle contents

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

}

void TAbsFitParticle::reset ( void ) [virtual]

Definition at line 106 of file TAbsFitParticle.cc.

References _iniparameters, _nPar, _parameters, _pcurr, _pini, _pull, and setCovMatrixFit().

Referenced by TKinFitter::resetParams().

                            {
  // Reset particle to initial values

  _parameters = _iniparameters;  
  _pcurr = _pini;
  setCovMatrixFit( 0 );
  _pull.ResizeTo(_nPar, 1);
  _pull.Zero();

}

void TAbsFitParticle::setCovMatrix ( const TMatrixD * theCovMatrix ) [virtual]

Definition at line 117 of file TAbsFitParticle.cc.

References _covMatrix, and _nPar.

Referenced by StKinFitter::addKinFitInfo(), TtFullHadKinFitter::fit(), TtSemiLepKinFitter::fit(), TFitParticleECart::init(), TFitParticleEScaledMomDev::init(), TFitParticleEtThetaPhi::init(), TFitParticleEMomDev::init(), TFitParticleEtEtaPhi::init(), TFitParticleMomDev::init(), TFitParticleESpher::init(), TFitParticleMCSpher::init(), TFitParticleMCPInvSpher::init(), TFitParticleMCMomDev::init(), TFitParticleCart::init(), TFitParticleMCCart::init(), and TFitParticleSpher::init().

                                                               {
  // Set the measured covariance matrix

  _covMatrix.ResizeTo(_nPar, _nPar);
  if(theCovMatrix==0) {
    _covMatrix.Zero();
  } else if (theCovMatrix->GetNcols() ==_nPar && theCovMatrix->GetNrows() ==_nPar) {
    _covMatrix = (*theCovMatrix);
  } else {
    edm::LogError ("WrongMatrixSize")
      << GetName() << "::setCovMatrix - Covariance matrix needs to be a "
      << _nPar << "x" << _nPar << " matrix.";
  }

}

void TAbsFitParticle::setCovMatrixFit ( const TMatrixD * theCovMatrixFit ) [virtual]

Definition at line 134 of file TAbsFitParticle.cc.

References _covMatrixFit, and _nPar.

Referenced by TKinFitter::applyVFit(), and reset().

                                                                     {
  // Set the fitted covariance matrix

  _covMatrixFit.ResizeTo(_nPar, _nPar);
  if(theCovMatrixFit==0) {
    _covMatrixFit.Zero();
  } else if (theCovMatrixFit->GetNcols() ==_nPar && theCovMatrixFit->GetNrows() ==_nPar) {
    _covMatrixFit = (*theCovMatrixFit);
  } else {
    edm::LogError ("WrongMatrixSize")
      << GetName() << "::setCovMatrixFit - Fitted covariance matrix needs to be a "
      << _nPar << "x" << _nPar << " matrix.";
  }

}

virtual void TAbsFitParticle::setIni4Vec ( const TLorentzVector * pini ) [pure virtual]

Implemented in TFitParticleCart, TFitParticleECart, TFitParticleEMomDev, TFitParticleEScaledMomDev, TFitParticleESpher, TFitParticleEtEtaPhi, TFitParticleEtThetaPhi, TFitParticleMCCart, TFitParticleMCMomDev, TFitParticleMCPInvSpher, TFitParticleMCSpher, TFitParticleMomDev, and TFitParticleSpher.

Referenced by StKinFitter::addKinFitInfo(), TtFullHadKinFitter::fit(), TtSemiLepKinFitter::fit(), and TSLToyGen::smearParticles().

void TAbsFitParticle::setParIni ( const TMatrixD * parini )

Definition at line 203 of file TAbsFitParticle.cc.

References _iniparameters.

                                                      {
  if (parini == 0) return;
  else if( parini->GetNrows() == _iniparameters.GetNrows()
           && parini->GetNcols() == _iniparameters.GetNcols() )
    _iniparameters = (*parini) ;
  else {
    edm::LogError ("WrongMatrixSize")
      << GetName() << "::setParIni - Matrices don't fit.";
    return;
      }
}