Belle II Software light-2406-ragdoll
RecoilMassKFit.cc
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * *
5 * See git log for contributors and copyright holders. *
6 * This file is licensed under LGPL-3.0, see LICENSE.md. *
7 **************************************************************************/
8
9#include <cstdio>
10
11#include <analysis/VertexFitting/KFit/RecoilMassKFit.h>
12#include <analysis/VertexFitting/KFit/MakeMotherKFit.h>
13#include <analysis/utility/CLHEPToROOT.h>
14#include <framework/gearbox/Const.h>
15
16#include <TMath.h>
17#include <TMatrixFSym.h>
18
19using namespace std;
20using namespace Belle2;
21using namespace Belle2::analysis;
22using namespace CLHEP;
23using namespace ROOT::Math;
24
26{
27 m_FlagFitted = false;
30 m_FlagAtDecayPoint = true;
32 m_d = HepMatrix(1, 1, 0);
33 m_V_D = HepMatrix(1, 1, 0);
34 m_lam = HepMatrix(1, 1, 0);
35 m_AfterVertexError = HepSymMatrix(3, 0);
36 m_recoilMass = -1.0;
37 m_FourMomentum = PxPyPzEVector();
38}
39
40
42
43
45RecoilMassKFit::setVertex(const HepPoint3D& v) {
47
49}
50
51
53RecoilMassKFit::setVertexError(const HepSymMatrix& e) {
54 if (e.num_row() != 3)
55 {
57 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
58 return m_ErrorCode;
59 }
60
63
65}
66
69 m_recoilMass = m;
70
72}
73
74
76RecoilMassKFit::setFourMomentum(const PxPyPzEVector& m) {
78
80}
81
82
85 m_FlagAtDecayPoint = flag;
86
88}
89
90
93 m_IsFixMass.push_back(true);
94
96}
97
98
101 m_IsFixMass.push_back(false);
102
104}
105
106
109 if (e.num_row() != 3 || e.num_col() != KFitConst::kNumber7)
110 {
112 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
113 return m_ErrorCode;
114 }
115
116 m_BeforeTrackVertexError.push_back(e);
119
121}
122
123
126 HepMatrix zero(3, KFitConst::kNumber7, 0);
127
128 return this->setTrackVertexError(zero);
129}
130
131
134 return KFitBase::setCorrelation(m);
135}
136
137
141}
142
143
144const HepPoint3D
145RecoilMassKFit::getVertex(const int flag) const
146{
147 if (flag == KFitConst::kAfterFit && !isFitted()) return HepPoint3D();
148
149 switch (flag) {
151 return m_BeforeVertex;
152
154 return m_AfterVertex;
155
156 default:
157 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
158 return HepPoint3D();
159 }
160}
161
162
163const HepSymMatrix
165{
166 if (flag == KFitConst::kAfterFit && !isFitted()) return HepSymMatrix(3, 0);
167
168 if (flag == KFitConst::kBeforeFit)
169 return m_BeforeVertexError;
171 return m_AfterVertexError;
172 else {
173 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
174 return HepSymMatrix(3, 0);
175 }
176}
177
178
179bool
181{
182 return m_FlagAtDecayPoint;
183}
184
185
186bool
188{
190}
191
192
193double
195{
196 return m_CHIsq;
197}
198
199
200const HepMatrix
201RecoilMassKFit::getTrackVertexError(const int id, const int flag) const
202{
203 if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(3, KFitConst::kNumber7, 0);
204 if (!isTrackIDInRange(id)) return HepMatrix(3, KFitConst::kNumber7, 0);
205
206 if (flag == KFitConst::kBeforeFit)
207 return m_BeforeTrackVertexError[id];
209 return m_AfterTrackVertexError[id];
210 else {
211 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
212 return HepMatrix(3, KFitConst::kNumber7, 0);
213 }
214}
215
216
217double
219{
220 if (!isFitted()) return -1;
221 if (!isTrackIDInRange(id)) return -1;
222
223 if (m_IsFixMass[id]) {
224
225 HepMatrix da(m_Tracks[id].getFitParameter(KFitConst::kBeforeFit) - m_Tracks[id].getFitParameter(KFitConst::kAfterFit));
226 int err_inverse = 0;
227 const double chisq = (da.T() * (m_Tracks[id].getFitError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
228
229 if (err_inverse) {
230 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
231 return -1;
232 }
233
234 return chisq;
235
236 } else {
237
238 HepMatrix da(m_Tracks[id].getMomPos(KFitConst::kBeforeFit) - m_Tracks[id].getMomPos(KFitConst::kAfterFit));
239 int err_inverse = 0;
240 const double chisq = (da.T() * (m_Tracks[id].getError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
241
242 if (err_inverse) {
243 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
244 return -1;
245 }
246
247 return chisq;
248
249 }
250}
251
252
253const HepMatrix
254RecoilMassKFit::getCorrelation(const int id1, const int id2, const int flag) const
255{
256 if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
257 if (!isTrackIDInRange(id1)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
258 if (!isTrackIDInRange(id2)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
259
260 switch (flag) {
262 return KFitBase::getCorrelation(id1, id2, flag);
263
265 return makeError3(
266 this->getTrackMomentum(id1),
267 this->getTrackMomentum(id2),
268 m_V_al_1.sub(KFitConst::kNumber7 * id1 + 1, KFitConst::kNumber7 * (id1 + 1), KFitConst::kNumber7 * id2 + 1,
269 KFitConst::kNumber7 * (id2 + 1)),
270 m_IsFixMass[id1],
271 m_IsFixMass[id2]);
272
273 default:
274 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
275 return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
276 }
277}
278
279
282 return KFitBase::doFit1();
283}
284
285
289 {
291 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
292 return m_ErrorCode;
293 }
294
295
296 if (m_IsFixMass.size() == 0)
297 {
298 // If no fix_mass flag at all,
299 // all tracks are considered to be fixed at mass.
300 for (int i = 0; i < m_TrackCount; i++) this->fixMass();
301 } else if (m_IsFixMass.size() != (unsigned int)m_TrackCount)
302 {
304 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
305 return m_ErrorCode;
306 }
307
308
310 {
311 int index = 0;
312 m_al_0 = HepMatrix(KFitConst::kNumber7 * m_TrackCount, 1, 0);
313 m_property = HepMatrix(m_TrackCount, 3, 0);
314 m_V_al_0 = HepSymMatrix(KFitConst::kNumber7 * m_TrackCount, 0);
315
316 for (auto& track : m_Tracks) {
317 // momentum x,y,z and position x,y,z
318 m_al_0[index * KFitConst::kNumber7 + 0][0] = track.getMomentum(KFitConst::kBeforeFit).x();
319 m_al_0[index * KFitConst::kNumber7 + 1][0] = track.getMomentum(KFitConst::kBeforeFit).y();
320 m_al_0[index * KFitConst::kNumber7 + 2][0] = track.getMomentum(KFitConst::kBeforeFit).z();
321 m_al_0[index * KFitConst::kNumber7 + 3][0] = track.getMomentum(KFitConst::kBeforeFit).t();
322 m_al_0[index * KFitConst::kNumber7 + 4][0] = track.getPosition(KFitConst::kBeforeFit).x();
323 m_al_0[index * KFitConst::kNumber7 + 5][0] = track.getPosition(KFitConst::kBeforeFit).y();
324 m_al_0[index * KFitConst::kNumber7 + 6][0] = track.getPosition(KFitConst::kBeforeFit).z();
325 // these error
326 m_V_al_0.sub(index * KFitConst::kNumber7 + 1, track.getError(KFitConst::kBeforeFit));
327 // charge, mass, a
328 m_property[index][0] = track.getCharge();
329 m_property[index][1] = track.getMass();
330 const double c = Belle2::Const::speedOfLight * 1e-4;
331 m_property[index][2] = -c * m_MagneticField * track.getCharge();
332 index++;
333 }
334
335 // error between track and track
336 if (m_FlagCorrelation) {
337 this->prepareCorrelation();
339 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
340 return m_ErrorCode;
341 }
342 }
343
344 // set member matrix
345 m_al_1 = m_al_0;
346
347 // define size of matrix
350
351 } else
352 {
353 //TODO: Not Implemented
355 }
356
358}
359
360
363 char buf[1024];
364 sprintf(buf, "%s:%s(): internal error; this function should never be called", __FILE__, __func__);
365 B2FATAL(buf);
366
367 /* NEVER REACHEd HERE */
369}
370
371
374 if (m_BeforeCorrelation.size() != static_cast<unsigned int>(m_TrackCount * (m_TrackCount - 1) / 2))
375 {
377 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
378 return m_ErrorCode;
379 }
380
381 int row = 0, col = 0;
382
383 for (auto& hm : m_BeforeCorrelation)
384 {
385 // counter
386 row++;
387 if (row == m_TrackCount) {
388 col++;
389 row = col + 1;
390 }
391
392 int ii = 0, jj = 0;
393 for (int i = KFitConst::kNumber7 * row; i < KFitConst::kNumber7 * (row + 1); i++) {
394 for (int j = KFitConst::kNumber7 * col; j < KFitConst::kNumber7 * (col + 1); j++) {
395 m_V_al_0[i][j] = hm[ii][jj];
396 jj++;
397 }
398 jj = 0;
399 ii++;
400 }
401 }
402
404 {
405 //TODO: Not Implemented
407
408 // ...error of vertex
410
411 // ...error matrix between vertex and tracks
413 if (m_BeforeTrackVertexError.size() != (unsigned int)m_TrackCount) {
415 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
416 return m_ErrorCode;
417 }
418
419 int i = 0;
420 for (auto& hm : m_BeforeTrackVertexError) {
421 for (int j = 0; j < 3; j++) for (int k = 0; k < KFitConst::kNumber7; k++) {
423 }
424 i++;
425 }
426 }
427 }
428
430}
431
432
435 Hep3Vector h3v;
436 int index = 0;
437 for (auto& pdata : m_Tracks)
438 {
439 // tracks
440 // momentum
441 h3v.setX(m_al_1[index * KFitConst::kNumber7 + 0][0]);
442 h3v.setY(m_al_1[index * KFitConst::kNumber7 + 1][0]);
443 h3v.setZ(m_al_1[index * KFitConst::kNumber7 + 2][0]);
444 if (m_IsFixMass[index])
445 pdata.setMomentum(HepLorentzVector(h3v, sqrt(h3v.mag2() + pdata.getMass()*pdata.getMass())), KFitConst::kAfterFit);
446 else
447 pdata.setMomentum(HepLorentzVector(h3v, m_al_1[index * KFitConst::kNumber7 + 3][0]), KFitConst::kAfterFit);
448 // position
449 pdata.setPosition(HepPoint3D(
450 m_al_1[index * KFitConst::kNumber7 + 4][0],
451 m_al_1[index * KFitConst::kNumber7 + 5][0],
453 // error of the tracks
454 pdata.setError(this->makeError3(pdata.getMomentum(),
455 m_V_al_1.sub(
456 index * KFitConst::kNumber7 + 1,
457 (index + 1)*KFitConst::kNumber7,
458 index * KFitConst::kNumber7 + 1,
459 (index + 1)*KFitConst::kNumber7), m_IsFixMass[index]),
461 if (m_ErrorCode != KFitError::kNoError) break;
462 index++;
463 }
464
466 {
467 //TODO: Not Implemented
469 } else
470 {
471 // not fit
473 }
474
476}
477
478
482 {
483
484 HepMatrix al_1_prime(m_al_1);
485 HepMatrix Sum_al_1(4, 1, 0);
486 std::vector<double> energy(m_TrackCount);
487 double a;
488
489 for (int i = 0; i < m_TrackCount; i++) {
490 a = m_property[i][2];
491 if (!m_FlagAtDecayPoint) a = 0.;
492 al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (m_BeforeVertex.y() - al_1_prime[i * KFitConst::kNumber7 + 5][0]);
493 al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (m_BeforeVertex.x() - al_1_prime[i * KFitConst::kNumber7 + 4][0]);
494 energy[i] = sqrt(al_1_prime[i * KFitConst::kNumber7 + 0][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] +
495 al_1_prime[i * KFitConst::kNumber7 + 1][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] +
496 al_1_prime[i * KFitConst::kNumber7 + 2][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] +
497 m_property[i][1] * m_property[i][1]);
498 if (m_IsFixMass[i])
499 Sum_al_1[3][0] += energy[i];
500 else
501 Sum_al_1[3][0] += al_1_prime[i * KFitConst::kNumber7 + 3][0];
502 }
503
504 for (int i = 0; i < m_TrackCount; i++) {
505 for (int j = 0; j < 3; j++) Sum_al_1[j][0] += al_1_prime[i * KFitConst::kNumber7 + j][0];
506 }
507
508 Sum_al_1[0][0] -= m_FourMomentum.Px();
509 Sum_al_1[1][0] -= m_FourMomentum.Py();
510 Sum_al_1[2][0] -= m_FourMomentum.Pz();
511 Sum_al_1[3][0] -= m_FourMomentum.E();
512
513 m_d[0][0] =
514 + Sum_al_1[3][0] * Sum_al_1[3][0] - Sum_al_1[0][0] * Sum_al_1[0][0]
515 - Sum_al_1[1][0] * Sum_al_1[1][0] - Sum_al_1[2][0] * Sum_al_1[2][0]
517
518 for (int i = 0; i < m_TrackCount; i++) {
519 if (energy[i] == 0) {
521 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
522 break;
523 }
524
525 a = m_property[i][2];
526 if (!m_FlagAtDecayPoint) a = 0.;
527
528 if (m_IsFixMass[i]) {
529 double invE = 1. / energy[i];
530 m_D[0][i * KFitConst::kNumber7 + 0] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE - Sum_al_1[0][0]);
531 m_D[0][i * KFitConst::kNumber7 + 1] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE - Sum_al_1[1][0]);
532 m_D[0][i * KFitConst::kNumber7 + 2] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE - Sum_al_1[2][0]);
533 m_D[0][i * KFitConst::kNumber7 + 3] = 0.;
534 m_D[0][i * KFitConst::kNumber7 + 4] = -2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE - Sum_al_1[1][0]) * a;
535 m_D[0][i * KFitConst::kNumber7 + 5] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE - Sum_al_1[0][0]) * a;
536 m_D[0][i * KFitConst::kNumber7 + 6] = 0.;
537 } else {
538 m_D[0][i * KFitConst::kNumber7 + 0] = -2.*Sum_al_1[0][0];
539 m_D[0][i * KFitConst::kNumber7 + 1] = -2.*Sum_al_1[1][0];
540 m_D[0][i * KFitConst::kNumber7 + 2] = -2.*Sum_al_1[2][0];
541 m_D[0][i * KFitConst::kNumber7 + 3] = 2.*Sum_al_1[3][0];
542 m_D[0][i * KFitConst::kNumber7 + 4] = 2.*Sum_al_1[1][0] * a;
543 m_D[0][i * KFitConst::kNumber7 + 5] = -2.*Sum_al_1[0][0] * a;
544 m_D[0][i * KFitConst::kNumber7 + 6] = 0.;
545 }
546 }
547
548 } else
549 {
550 //TODO: Not Implemented
552 }
553
555}
556
557
560 m_NDF = 1;
561
563}
564
566{
567 MakeMotherKFit kmm;
569 unsigned n = getTrackCount();
570 for (unsigned i = 0; i < n; ++i) {
572 getTrack(i).getCharge());
575 for (unsigned j = i + 1; j < n; ++j) {
577 }
578 }
579 kmm.setVertex(getVertex());
582 m_ErrorCode = kmm.doMake();
584 return m_ErrorCode;
585 double chi2 = getCHIsq();
586 int ndf = getNDF();
587 double prob = TMath::Prob(chi2, ndf);
588 //
589 mother->writeExtraInfo("chiSquared", chi2);
590 mother->writeExtraInfo("ndf", ndf);
591
592 mother->updateMomentum(
593 CLHEPToROOT::getLorentzVector(kmm.getMotherMomentum()),
594 CLHEPToROOT::getXYZVector(kmm.getMotherPosition()),
595 CLHEPToROOT::getTMatrixFSym(kmm.getMotherError()),
596 prob);
598 return m_ErrorCode;
599}
static const double speedOfLight
[cm/ns]
Definition: Const.h:695
Class to store reconstructed particles.
Definition: Particle.h:75
void writeExtraInfo(const std::string &name, const double value)
Sets the user defined extraInfo.
Definition: Particle.cc:1308
void updateMomentum(const ROOT::Math::PxPyPzEVector &p4, const ROOT::Math::XYZVector &vertex, const TMatrixFSym &errMatrix, double pValue)
Sets Lorentz vector, position, 7x7 error matrix and p-value.
Definition: Particle.h:386
int m_NecessaryTrackCount
Number needed tracks to perform fit.
Definition: KFitBase.h:303
double m_MagneticField
Magnetic field.
Definition: KFitBase.h:311
CLHEP::HepMatrix m_al_1
See J.Tanaka Ph.D (2001) p136 for definition.
Definition: KFitBase.h:259
virtual enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &c)
Set a correlation matrix.
Definition: KFitBase.cc:70
const CLHEP::HepSymMatrix makeError3(const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e, const bool is_fix_mass) const
Rebuild an error matrix from a Lorentz vector and an error matrix.
Definition: KFitBase.cc:320
const CLHEP::HepSymMatrix getTrackError(const int id) const
Get an error matrix of the track.
Definition: KFitBase.cc:168
const CLHEP::HepLorentzVector getTrackMomentum(const int id) const
Get a Lorentz vector of the track.
Definition: KFitBase.cc:154
CLHEP::HepMatrix m_lam
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:276
const HepPoint3D getTrackPosition(const int id) const
Get a position of the track.
Definition: KFitBase.cc:161
CLHEP::HepMatrix m_property
Container of charges and masses.
Definition: KFitBase.h:263
enum KFitError::ECode m_ErrorCode
Error code.
Definition: KFitBase.h:243
virtual enum KFitError::ECode setZeroCorrelation(void)
Indicate no correlation between tracks.
Definition: KFitBase.cc:85
CLHEP::HepMatrix m_V_al_1
See J.Tanaka Ph.D (2001) p138 for definition.
Definition: KFitBase.h:274
virtual int getNDF(void) const
Get an NDF of the fit.
Definition: KFitBase.cc:114
CLHEP::HepMatrix m_d
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:268
bool isFitted(void) const
Return false if fit is not performed yet or performed fit is failed; otherwise true.
Definition: KFitBase.cc:728
CLHEP::HepMatrix m_D
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:266
CLHEP::HepMatrix m_V_D
See J.Tanaka Ph.D (2001) p138 for definition.
Definition: KFitBase.h:271
bool isTrackIDInRange(const int id) const
Check if the id is in the range.
Definition: KFitBase.cc:739
virtual const CLHEP::HepMatrix getCorrelation(const int id1, const int id2, const int flag=KFitConst::kAfterFit) const
Get a correlation matrix between two tracks.
Definition: KFitBase.cc:183
bool m_FlagCorrelation
Flag whether a correlation among tracks exists.
Definition: KFitBase.h:306
CLHEP::HepSymMatrix m_V_al_0
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:255
const KFitTrack getTrack(const int id) const
Get a specified track object.
Definition: KFitBase.cc:175
std::vector< CLHEP::HepMatrix > m_BeforeCorrelation
Container of input correlation matrices.
Definition: KFitBase.h:251
bool m_FlagFitted
Flag to indicate if the fit is performed and succeeded.
Definition: KFitBase.h:245
double m_CHIsq
chi-square of the fit.
Definition: KFitBase.h:297
int getTrackCount(void) const
Get the number of added tracks.
Definition: KFitBase.cc:107
int m_NDF
NDF of the fit.
Definition: KFitBase.h:295
std::vector< KFitTrack > m_Tracks
Container of input tracks.
Definition: KFitBase.h:249
int m_TrackCount
Number of tracks.
Definition: KFitBase.h:301
CLHEP::HepMatrix m_al_0
See J.Tanaka Ph.D (2001) p136 for definition.
Definition: KFitBase.h:257
enum KFitError::ECode doFit1(void)
Perform a fit (used in MassFitKFit::doFit()).
Definition: KFitBase.cc:502
static void displayError(const char *file, const int line, const char *func, const enum ECode code)
Display a description of error and its location.
Definition: KFitError.h:72
ECode
ECode is a error code enumerate.
Definition: KFitError.h:34
@ kUnimplemented
Unprepared.
Definition: KFitError.h:44
@ kCannotGetMatrixInverse
Cannot calculate matrix inverse (bad track property or internal error)
Definition: KFitError.h:58
@ kOutOfRange
Specified track-id out of range.
Definition: KFitError.h:42
@ kDivisionByZero
Division by zero (bad track property or internal error)
Definition: KFitError.h:56
@ kBadTrackSize
Track count too small to perform fit.
Definition: KFitError.h:47
@ kBadMatrixSize
Wrong correlation matrix size.
Definition: KFitError.h:49
@ kBadCorrelationSize
Wrong correlation matrix size (internal error)
Definition: KFitError.h:51
MakeMotherKFit is a class to build mother particle from kinematically fitted daughters.
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set a vertex position of the mother particle.
enum KFitError::ECode addTrack(const KFitTrack &kp)
Add a track to the make-mother object.
enum KFitError::ECode doMake(void)
Perform a reconstruction of mother particle.
const CLHEP::HepSymMatrix getMotherError(void) const
Get an error matrix of the mother particle.
enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &e)
Set a correlation matrix.
const HepPoint3D getMotherPosition(void) const
Get a position of the mother particle.
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set a vertex error matrix of the mother particle.
enum KFitError::ECode setTrackVertexError(const CLHEP::HepMatrix &e)
Set a vertex error matrix of the child particle in the addTrack'ed order.
const CLHEP::HepLorentzVector getMotherMomentum(void) const
Get a Lorentz vector of the mother particle.
enum KFitError::ECode setMagneticField(const double mf)
Change a magnetic field from the default value KFitConst::kDefaultMagneticField.
enum KFitError::ECode setZeroCorrelation(void) override
Indicate no correlation between tracks.
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set an initial vertex position for the four momentum-constraint fit.
bool m_FlagTrackVertexError
Flag to indicate if the vertex error matrix of the child particle is preset.
enum KFitError::ECode setFourMomentum(const ROOT::Math::PxPyPzEVector &m)
Set a recoil mass .
bool m_FlagAtDecayPoint
Flag controlled by setFlagAtDecayPoint().
bool getFlagAtDecayPoint(void) const
Get a flag if to constraint at the decay point in the four momentum-constraint fit.
enum KFitError::ECode prepareInputMatrix(void) override
Build grand matrices for minimum search from input-track properties.
bool getFlagFitWithVertex(void) const
Get a flag if the fit is allowed with moving the vertex position.
enum KFitError::ECode calculateNDF(void) override
Calculate an NDF of the fit.
double getCHIsq(void) const override
Get a chi-square of the fit.
enum KFitError::ECode setFlagAtDecayPoint(const bool flag)
Set a flag if to constraint at the decay point in the four momentum-constraint fit.
enum KFitError::ECode setTrackZeroVertexError(void)
Indicate no vertex uncertainty in the child particle in the addTrack'ed order.
std::vector< int > m_IsFixMass
Array of flags whether the track property is fixed at the mass.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
enum KFitError::ECode unfixMass(void)
Tell the object to unfix the last added track property at the invariant mass.
enum KFitError::ECode setRecoilMass(const double m)
Set an invariant mass for the four momentum-constraint fit.
enum KFitError::ECode prepareInputSubMatrix(void) override
Build sub-matrices for minimum search from input-track properties.
bool m_FlagFitIncludingVertex
Flag to indicate if the fit is allowed with moving the vertex position.
enum KFitError::ECode doFit(void)
Perform a recoil-mass constraint fit.
std::vector< CLHEP::HepMatrix > m_BeforeTrackVertexError
array of vertex error matrices before the fit.
enum KFitError::ECode makeCoreMatrix(void) override
Build matrices using the kinematical constraint.
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set an initial vertex error matrix for the four momentum-constraint fit.
enum KFitError::ECode setTrackVertexError(const CLHEP::HepMatrix &e)
Set a vertex error matrix of the child particle in the addTrack'ed order.
enum KFitError::ECode prepareCorrelation(void) override
Build a grand correlation matrix from input-track properties.
HepPoint3D m_AfterVertex
Vertex position after the fit.
enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &m) override
Set a correlation matrix.
std::vector< CLHEP::HepMatrix > m_AfterTrackVertexError
array of vertex error matrices after the fit.
const CLHEP::HepMatrix getCorrelation(const int id1, const int id2, const int flag=KFitConst::kAfterFit) const override
Get a correlation matrix between two tracks.
double getTrackCHIsq(const int id) const override
Get a chi-square of the track.
const CLHEP::HepSymMatrix getVertexError(const int flag=KFitConst::kAfterFit) const
Get a vertex error matrix.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
const CLHEP::HepMatrix getTrackVertexError(const int id, const int flag=KFitConst::kAfterFit) const
Get a vertex error matrix of the track.
CLHEP::HepSymMatrix m_AfterVertexError
Vertex error matrix after the fit.
enum KFitError::ECode prepareOutputMatrix(void) override
Build an output error matrix.
~RecoilMassKFit(void)
Destruct the object.
CLHEP::HepSymMatrix m_BeforeVertexError
Vertex error matrix before the fit.
HepPoint3D m_BeforeVertex
Vertex position before the fit.
enum KFitError::ECode fixMass(void)
Tell the object to fix the last added track property at the invariant mass.
ROOT::Math::PxPyPzEVector m_FourMomentum
Four Momentum.
RecoilMassKFit(void)
Construct an object with no argument.
Abstract base class for different kinds of events.
Definition: ClusterUtils.h:24
STL namespace.
static const int kMaxTrackCount
Maximum track size.
Definition: KFitConst.h:40
static const int kAfterFit
Input parameter to specify after-fit when setting/getting a track attribute.
Definition: KFitConst.h:37
static const int kBeforeFit
Input parameter to specify before-fit when setting/getting a track attribute.
Definition: KFitConst.h:35
static const int kNumber7
Constant 7 to check matrix size (internal use)
Definition: KFitConst.h:32