TFitParticleMCMomDev Class Reference

#include <TFitParticleMCMomDev.h>

Inheritance diagram for TFitParticleMCMomDev:

Public Member Functions
TLorentzVector *	calc4Vec (const TMatrixD *params) override
TAbsFitParticle *	clone (const TString &newname=TString("")) const override
TMatrixD *	getDerivative () override
void	setIni4Vec (const TLorentzVector *pini) override
void	setIni4Vec (const TVector3 *p, Double_t M)
	TFitParticleMCMomDev ()
	TFitParticleMCMomDev (const TFitParticleMCMomDev &fitParticle)
	TFitParticleMCMomDev (TVector3 *p, Double_t M, const TMatrixD *theCovMatrix)
	TFitParticleMCMomDev (const TString &name, const TString &title, TVector3 *p, Double_t M, const TMatrixD *theCovMatrix)
TMatrixD *	transform (const TLorentzVector &vec) override
	~TFitParticleMCMomDev () override
Public Member Functions inherited from TAbsFitParticle
virtual void	applycorr (TMatrixD *corrMatrix)
virtual const TMatrixD *	getCovMatrix () const
virtual const TMatrixD *	getCovMatrixDeltaY ()
virtual const TMatrixD *	getCovMatrixFit () const
const TLorentzVector *	getCurr4Vec ()
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)
void	setParIni (const TMatrixD *parini)
	TAbsFitParticle ()
	TAbsFitParticle (const TString &name, const TString &title)
	~TAbsFitParticle () override

Protected Member Functions
void	init (TVector3 *p, Double_t M, const TMatrixD *theCovMatrix)
Protected Member Functions inherited from TAbsFitParticle
void	calcCovMatrixDeltaY ()

Additional Inherited Members
Protected Attributes inherited from TAbsFitParticle
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 10 of file TFitParticleMCMomDev.h.

Constructor & Destructor Documentation

TFitParticleMCMomDev::TFitParticleMCMomDev

(

)

Definition at line 34 of file TFitParticleMCMomDev.cc.

References init().

Referenced by clone().

  :TAbsFitParticle()
{ 
  init( nullptr, 0., nullptr);
}

TFitParticleMCMomDev::TFitParticleMCMomDev

(

const TFitParticleMCMomDev &

fitParticle

)

Definition at line 40 of file TFitParticleMCMomDev.cc.

References TAbsFitParticle::_covMatrix, TAbsFitParticle::_iniparameters, TAbsFitParticle::_nPar, TAbsFitParticle::_parameters, TAbsFitParticle::_pcurr, TAbsFitParticle::_pini, TAbsFitParticle::_u1, TAbsFitParticle::_u2, and TAbsFitParticle::_u3.

  :TAbsFitParticle( fitParticle.GetName(), fitParticle.GetTitle() )
{

  _nPar = fitParticle._nPar;
  _u1 = fitParticle._u1;
  _u2 = fitParticle._u2;
  _u3 = fitParticle._u3;
  _covMatrix.ResizeTo(  fitParticle._covMatrix );
  _covMatrix = fitParticle._covMatrix;
  _iniparameters.ResizeTo( fitParticle._iniparameters );
  _iniparameters = fitParticle._iniparameters;
  _parameters.ResizeTo( fitParticle._parameters );
  _parameters = fitParticle._parameters;
  _pini = fitParticle._pini;
  _pcurr = fitParticle._pcurr;

}

TFitParticleMCMomDev::TFitParticleMCMomDev	(	TVector3 *	p,
		Double_t	M,
		const TMatrixD *	theCovMatrix
	)

Definition at line 59 of file TFitParticleMCMomDev.cc.

References init().

  :TAbsFitParticle()
{ 
  init(p, M, theCovMatrix);
}

TFitParticleMCMomDev::TFitParticleMCMomDev	(	const TString &	name,
		const TString &	title,
		TVector3 *	p,
		Double_t	M,
		const TMatrixD *	theCovMatrix
	)

Definition at line 65 of file TFitParticleMCMomDev.cc.

References init().

  :TAbsFitParticle(name, title)
{ 
  init(p, M, theCovMatrix);
}

TFitParticleMCMomDev::~TFitParticleMCMomDev ( )

override

Definition at line 84 of file TFitParticleMCMomDev.cc.

                                            {

}

Member Function Documentation

TLorentzVector * TFitParticleMCMomDev::calc4Vec ( const TMatrixD *  params )

overridevirtual

Implements TAbsFitParticle.

Definition at line 106 of file TFitParticleMCMomDev.cc.

References TAbsFitParticle::_nPar, TAbsFitParticle::_pini, TAbsFitParticle::_u1, TAbsFitParticle::_u2, TAbsFitParticle::_u3, X, DOFs::Y, and DOFs::Z.

                                                                       {
  // Calculates a 4vector corresponding to the given
  // parameter values

  if (params == nullptr) {
    return nullptr;
  }

  if ( params->GetNcols() != 1 || params->GetNrows() !=_nPar ) {
    edm::LogError ("WrongMatrixSize")
      << GetName() << "::calc4Vec - Parameter matrix has wrong size.";
    return nullptr;
  }

  Double_t X =  _pini.P() * (*params)(0,0) *_u1.X()+
    (*params)(1,0) * _u2.X()+
    (*params)(2,0) * _u3.X() ;
  Double_t Y = _pini.P() * (*params)(0,0) *_u1.Y()+
    (*params)(1,0) * _u2.Y()+
    (*params)(2,0) * _u3.Y() ;
  Double_t Z = _pini.P() * (*params)(0,0) *_u1.Z()+
    (*params)(1,0) * _u2.Z()+
    (*params)(2,0) * _u3.Z() ;
  Double_t E =  TMath::Sqrt(  X*X + Y*Y + Z*Z + _pini.M2() );

  TLorentzVector* vec = new TLorentzVector( X, Y, Z, E );
  return vec;

}

TAbsFitParticle * TFitParticleMCMomDev::clone ( const TString &  newname = TString("") ) const

overridevirtual

Implements TAbsFitParticle.

Definition at line 72 of file TFitParticleMCMomDev.cc.

References TFitParticleMCMomDev().

                                                                          {
  // Returns a copy of itself
  
  TAbsFitParticle* myclone = new TFitParticleMCMomDev( *this );
  if ( newname.Length() > 0 ) myclone->SetName(newname);
  return myclone;

}

TMatrixD * TFitParticleMCMomDev::getDerivative ( )

overridevirtual

Implements TAbsFitParticle.

Definition at line 175 of file TFitParticleMCMomDev.cc.

References TAbsFitParticle::_parameters, TAbsFitParticle::_pcurr, TAbsFitParticle::_pini, TAbsFitParticle::_u1, TAbsFitParticle::_u2, and TAbsFitParticle::_u3.

                                              {
  // returns derivative dP/dy with P=(p,E) and y=(r, theta, phi) 
  // the free parameters of the fit. The columns of the matrix contain 
  // (dP/dr, dP/dtheta, dP/dphi).

  TMatrixD* DerivativeMatrix = new TMatrixD(4,3);
  (*DerivativeMatrix) *= 0.;
  //1st column: dP/dr
  (*DerivativeMatrix)(0,0)=_pini.P()*_u1.X();
  (*DerivativeMatrix)(1,0)=_pini.P()*_u1.Y();
  (*DerivativeMatrix)(2,0)=_pini.P()*_u1.Z();
  (*DerivativeMatrix)(3,0)=_pini.P()*_pini.P()*_parameters(0,0)/_pcurr.E();

//  (*DerivativeMatrix)(3,0)=0.;

  //2nd column: dP/dtheta
  (*DerivativeMatrix)(0,1)=_u2.X();
  (*DerivativeMatrix)(1,1)=_u2.Y();
  (*DerivativeMatrix)(2,1)=_u2.Z();
  (*DerivativeMatrix)(3,1)=_parameters(1,0)/_pcurr.E();

  //(*DerivativeMatrix)(3,1)=0.;

  //3rd column: dP/dphi
  (*DerivativeMatrix)(0,2)=_u3.X();
  (*DerivativeMatrix)(1,2)=_u3.Y();
  (*DerivativeMatrix)(2,2)=_u3.Z();
  (*DerivativeMatrix)(3,2)=_parameters(2,0)/_pcurr.E();

  //(*DerivativeMatrix)(3,2)=0.;

  return DerivativeMatrix;  

}

void TFitParticleMCMomDev::init ( TVector3 *  p, Double_t  M, const TMatrixD *  theCovMatrix  )

protected

Definition at line 92 of file TFitParticleMCMomDev.cc.

References TAbsFitParticle::_iniparameters, TAbsFitParticle::_nPar, TAbsFitParticle::_parameters, TAbsFitParticle::setCovMatrix(), and setIni4Vec().

Referenced by TFitParticleMCMomDev().

                                                                                     {

  _nPar = 3;
  setIni4Vec(p, M);
  _iniparameters.ResizeTo(_nPar,1);
  _iniparameters(0,0)=1.;
  _iniparameters(1,0)=0.;
  _iniparameters(2,0)=0.;
  _parameters.ResizeTo(_nPar, 1);
  _parameters = _iniparameters;
  setCovMatrix(theCovMatrix);
  
}

void TFitParticleMCMomDev::setIni4Vec ( const TLorentzVector *  pini )

overridevirtual

Implements TAbsFitParticle.

Definition at line 136 of file TFitParticleMCMomDev.cc.

Referenced by init().

                                                                {
  // Set the initial 4vector. Will also set the 
  // inital parameter values

  TVector3 vec( pini->Vect() );
  setIni4Vec( &vec, pini->M() );

}

void TFitParticleMCMomDev::setIni4Vec	(	const TVector3 *	p,
		Double_t	M
	)

Definition at line 145 of file TFitParticleMCMomDev.cc.

References TAbsFitParticle::_iniparameters, TAbsFitParticle::_parameters, TAbsFitParticle::_pcurr, TAbsFitParticle::_pini, TAbsFitParticle::_u1, TAbsFitParticle::_u2, and TAbsFitParticle::_u3.

                                                                   {
  // Set the initial 4vector. Will also set the 
  // inital parameter values

  if ( p == nullptr ) {

    _pini.SetXYZM( 0., 0., 0., M);
    _pcurr = _pini;

  } else {

    _pini.SetXYZM( p->x(), p->y(), p->z(), M);
    _pcurr = _pini;

    _u1 = (*p);
    _u1 *= 1./_u1.Mag();

    TVector3 uz(0., 0., 1.);
    _u3 = uz.Cross(_u1);
    _u3 *= 1./_u3.Mag();
  
    _u2 = _u3.Cross(_u1);
    _u2 *= 1./_u2.Mag();
  
  }

  _parameters = _iniparameters;

}

TMatrixD * TFitParticleMCMomDev::transform ( const TLorentzVector &  vec )

overridevirtual

Implements TAbsFitParticle.

Definition at line 210 of file TFitParticleMCMomDev.cc.

References TAbsFitParticle::_nPar, TAbsFitParticle::_pini, TAbsFitParticle::_u1, TAbsFitParticle::_u2, TAbsFitParticle::_u3, and makeMuonMisalignmentScenario::rot.

                                                                   {
  // Returns the parameters corresponding to the given 
  // 4vector wrt. to the current base vectors u_r, u_theta, and u_phi

  // construct rotation matrix
  TRotation rot;
  rot.RotateAxes( _u1, _u2, _u3 );
  rot.Invert();

  // rotate vector
  TVector3 vec3( vec.Vect() );
  vec3.Transform( rot );

  // retrieve parameters
  TMatrixD* tparams = new TMatrixD( _nPar, 1 );
  (*tparams)(0,0) = vec3(0)/_pini.P();
  (*tparams)(1,0) = vec3(1);
  (*tparams)(2,0) = vec3(2);

  return tparams;

}