CMS 3D CMS Logo

List of all members | Public Member Functions | Protected Member Functions
TFitParticleMomDev Class Reference

#include <TFitParticleMomDev.h>

Inheritance diagram for TFitParticleMomDev:
TAbsFitParticle

Public Member Functions

TLorentzVector * calc4Vec (const TMatrixD *params) override
 
TAbsFitParticleclone (const TString &newname=TString("")) const override
 
TMatrixD * getDerivative () override
 
void setIni4Vec (const TLorentzVector *pini) override
 
 TFitParticleMomDev ()
 
 TFitParticleMomDev (const TFitParticleMomDev &fitParticle)
 
 TFitParticleMomDev (TLorentzVector *pini, const TMatrixD *theCovMatrix)
 
 TFitParticleMomDev (const TString &name, const TString &title, TLorentzVector *pini, const TMatrixD *theCovMatrix)
 
TMatrixD * transform (const TLorentzVector &vec) override
 
 ~TFitParticleMomDev () 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 (TLorentzVector *pini, 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 TFitParticleMomDev.h.

Constructor & Destructor Documentation

◆ TFitParticleMomDev() [1/4]

TFitParticleMomDev::TFitParticleMomDev ( )

Definition at line 32 of file TFitParticleMomDev.cc.

References init().

Referenced by clone().

32 : TAbsFitParticle() { init(nullptr, nullptr); }
void init(TLorentzVector *pini, const TMatrixD *theCovMatrix)

◆ TFitParticleMomDev() [2/4]

TFitParticleMomDev::TFitParticleMomDev ( const TFitParticleMomDev fitParticle)

Definition at line 34 of file TFitParticleMomDev.cc.

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

35  : TAbsFitParticle(fitParticle.GetName(), fitParticle.GetTitle()) {
36  _nPar = fitParticle._nPar;
37  _u1 = fitParticle._u1;
38  _u2 = fitParticle._u2;
39  _u3 = fitParticle._u3;
40  _covMatrix.ResizeTo(fitParticle._covMatrix);
41  _covMatrix = fitParticle._covMatrix;
42  _iniparameters.ResizeTo(fitParticle._iniparameters);
43  _iniparameters = fitParticle._iniparameters;
44  _parameters.ResizeTo(fitParticle._parameters);
45  _parameters = fitParticle._parameters;
46  _pini = fitParticle._pini;
47  _pcurr = fitParticle._pcurr;
48 }
TLorentzVector _pini
TMatrixD _parameters
TLorentzVector _pcurr
TMatrixD _iniparameters

◆ TFitParticleMomDev() [3/4]

TFitParticleMomDev::TFitParticleMomDev ( TLorentzVector *  pini,
const TMatrixD *  theCovMatrix 
)

Definition at line 50 of file TFitParticleMomDev.cc.

References init().

50  : TAbsFitParticle() {
51  init(pini, theCovMatrix);
52 }
void init(TLorentzVector *pini, const TMatrixD *theCovMatrix)

◆ TFitParticleMomDev() [4/4]

TFitParticleMomDev::TFitParticleMomDev ( const TString &  name,
const TString &  title,
TLorentzVector *  pini,
const TMatrixD *  theCovMatrix 
)

Definition at line 54 of file TFitParticleMomDev.cc.

References init().

59  init(pini, theCovMatrix);
60 }
void init(TLorentzVector *pini, const TMatrixD *theCovMatrix)

◆ ~TFitParticleMomDev()

TFitParticleMomDev::~TFitParticleMomDev ( )
override

Definition at line 74 of file TFitParticleMomDev.cc.

74 {}

Member Function Documentation

◆ calc4Vec()

TLorentzVector * TFitParticleMomDev::calc4Vec ( const TMatrixD *  params)
overridevirtual

Implements TAbsFitParticle.

Definition at line 122 of file TFitParticleMomDev.cc.

References TAbsFitParticle::_nPar, TAbsFitParticle::_pini, TAbsFitParticle::_u1, TAbsFitParticle::_u2, TAbsFitParticle::_u3, submitPVValidationJobs::params, X, BeamSpotPI::Y, and BeamSpotPI::Z.

122  {
123  // Calculates a 4vector corresponding to the given
124  // parameter values
125 
126  if (params == nullptr) {
127  return nullptr;
128  }
129 
130  if (params->GetNcols() != 1 || params->GetNrows() != _nPar) {
131  edm::LogError("WrongMatrixSize") << GetName() << "::calc4Vec - Parameter matrix has wrong size.";
132  return nullptr;
133  }
134 
135  Double_t X = _pini.P() * (*params)(0, 0) * _u1.X() + (*params)(1, 0) * _u2.X() + (*params)(2, 0) * _u3.X();
136  Double_t Y = _pini.P() * (*params)(0, 0) * _u1.Y() + (*params)(1, 0) * _u2.Y() + (*params)(2, 0) * _u3.Y();
137  Double_t Z = _pini.P() * (*params)(0, 0) * _u1.Z() + (*params)(1, 0) * _u2.Z() + (*params)(2, 0) * _u3.Z();
138  Double_t E = TMath::Sqrt(X * X + Y * Y + Z * Z + (*params)(3, 0) * (*params)(3, 0) * _pini.M2());
139 
140  TLorentzVector* vec = new TLorentzVector(X, Y, Z, E);
141  return vec;
142 }
#define X(str)
Definition: MuonsGrabber.cc:38
Log< level::Error, false > LogError
TLorentzVector _pini

◆ clone()

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

Implements TAbsFitParticle.

Definition at line 62 of file TFitParticleMomDev.cc.

References TFitParticleMomDev().

62  {
63  // Returns a copy of itself
64 
65  TAbsFitParticle* myclone = new TFitParticleMomDev(*this);
66  if (newname.Length() > 0)
67  myclone->SetName(newname);
68  return myclone;
69 }

◆ getDerivative()

TMatrixD * TFitParticleMomDev::getDerivative ( )
overridevirtual

Implements TAbsFitParticle.

Definition at line 144 of file TFitParticleMomDev.cc.

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

144  {
145  // returns derivative dP/dy with P=(p,E) and y=(r, theta, phi, d)
146  // the free parameters of the fit. The columns of the matrix contain
147  // (dP/dr, dP/dtheta, dP/dphi, dP/dd).
148 
149  TMatrixD* DerivativeMatrix = new TMatrixD(4, 4);
150  (*DerivativeMatrix) *= 0.;
151 
152  //1st column: dP/dr
153  (*DerivativeMatrix)(0, 0) = _pini.P() * _u1.X();
154  (*DerivativeMatrix)(1, 0) = _pini.P() * _u1.Y();
155  (*DerivativeMatrix)(2, 0) = _pini.P() * _u1.Z();
156  (*DerivativeMatrix)(3, 0) = 0.;
157 
158  //2nd column: dP/dtheta
159  (*DerivativeMatrix)(0, 1) = _u2.X();
160  (*DerivativeMatrix)(1, 1) = _u2.Y();
161  (*DerivativeMatrix)(2, 1) = _u2.Z();
162  (*DerivativeMatrix)(3, 1) = 0.;
163 
164  //3rd column: dP/dphi
165  (*DerivativeMatrix)(0, 2) = _u3.X();
166  (*DerivativeMatrix)(1, 2) = _u3.Y();
167  (*DerivativeMatrix)(2, 2) = _u3.Z();
168  (*DerivativeMatrix)(3, 2) = 0.;
169 
170  //4th column: dP/dm
171  (*DerivativeMatrix)(0, 3) = 0.;
172  (*DerivativeMatrix)(1, 3) = 0.;
173  (*DerivativeMatrix)(2, 3) = 0.;
174  (*DerivativeMatrix)(3, 3) = _pini.M() * _pini.M() * _parameters(3, 0) / _pcurr.E();
175 
176  return DerivativeMatrix;
177 }
TLorentzVector _pini
TMatrixD _parameters
TLorentzVector _pcurr

◆ init()

void TFitParticleMomDev::init ( TLorentzVector *  pini,
const TMatrixD *  theCovMatrix 
)
protected

Definition at line 79 of file TFitParticleMomDev.cc.

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

Referenced by TFitParticleMomDev().

79  {
80  _nPar = 4;
81  _iniparameters.ResizeTo(_nPar, 1);
82  _iniparameters(0, 0) = 1.;
83  _iniparameters(1, 0) = 0.;
84  _iniparameters(2, 0) = 0.;
85  _iniparameters(3, 0) = 1.;
86  _parameters.ResizeTo(_nPar, 1);
88  setIni4Vec(pini);
89  setCovMatrix(theCovMatrix);
90 }
void setIni4Vec(const TLorentzVector *pini) override
TMatrixD _parameters
virtual void setCovMatrix(const TMatrixD *theCovMatrix)
TMatrixD _iniparameters

◆ setIni4Vec()

void TFitParticleMomDev::setIni4Vec ( const TLorentzVector *  pini)
overridevirtual

Implements TAbsFitParticle.

Definition at line 92 of file TFitParticleMomDev.cc.

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

Referenced by init().

92  {
93  // Set the initial 4vector. Will also set the
94  // inital parameter values
95 
96  if (pini == nullptr) {
97  _u1.SetXYZ(0., 0., 0.);
98  _u3.SetXYZ(0., 0., 0.);
99  _u2.SetXYZ(0., 0., 0.);
100  _pini.SetXYZT(0., 0., 0., 0.);
101  _pcurr = _pini;
102 
103  } else {
104  _pini = (*pini);
105  _pcurr = _pini;
106 
107  _u1 = pini->Vect();
108  _u1 *= 1. / _u1.Mag();
109 
110  TVector3 uz(0., 0., 1.);
111  _u3 = uz.Cross(_u1);
112  _u3 *= 1. / _u3.Mag();
113 
114  _u2 = _u3.Cross(_u1);
115  _u2 *= 1. / _u2.Mag();
116  }
117 
118  // reset parameters
120 }
TLorentzVector _pini
TMatrixD _parameters
TLorentzVector _pcurr
TMatrixD _iniparameters

◆ transform()

TMatrixD * TFitParticleMomDev::transform ( const TLorentzVector &  vec)
overridevirtual

Implements TAbsFitParticle.

Definition at line 179 of file TFitParticleMomDev.cc.

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

179  {
180  // Returns the parameters corresponding to the given
181  // 4vector wrt. to the current base vectors u_r, u_theta, and u_phi
182 
183  // construct rotation matrix
184  TRotation rot;
185  rot.RotateAxes(_u1, _u2, _u3);
186  rot.Invert();
187 
188  // rotate vector
189  TVector3 vec3(vec.Vect());
190  vec3.Transform(rot);
191 
192  // retrieve parameters
193  TMatrixD* tparams = new TMatrixD(_nPar, 1);
194  (*tparams)(0, 0) = vec3(0) / _pini.P();
195  (*tparams)(1, 0) = vec3(1);
196  (*tparams)(2, 0) = vec3(2);
197  (*tparams)(3, 0) = vec.M() / _pini.M();
198 
199  return tparams;
200 }
TLorentzVector _pini
std::vector< vec2 > vec3
Definition: HCALResponse.h:17