Belle II Software  release-05-02-19
MassVertexFitKFit.cc
1 /**************************************************************************
2  * Copyright(C) 1997 - J. Tanaka *
3  * *
4  * Author: J. Tanaka *
5  * Contributor: J. Tanaka and *
6  * conversion to Belle II structure by T. Higuchi *
7  * *
8  * This software is provided "as is" without any warranty. *
9  **************************************************************************/
10 
11 #include <TMatrixFSym.h>
12 
13 #include <analysis/VertexFitting/KFit/MakeMotherKFit.h>
14 #include <analysis/VertexFitting/KFit/MassVertexFitKFit.h>
15 #include <analysis/utility/CLHEPToROOT.h>
16 
17 
18 using namespace std;
19 using namespace Belle2;
20 using namespace Belle2::analysis;
21 using namespace CLHEP;
22 
23 MassVertexFitKFit::MassVertexFitKFit():
24  m_BeforeVertex(HepPoint3D(0., 0., 0.)),
25  m_AfterVertexError(HepSymMatrix(3, 0))
26 {
27  m_FlagFitted = false;
29  m_V_E = HepMatrix(3, 3, 0);
30  m_v = HepMatrix(3, 1, 0);
31  m_v_a = HepMatrix(3, 1, 0);
32  m_InvariantMass = -1.0;
33 }
34 
35 
37 
38 
41  m_BeforeVertex = v;
42 
44 }
45 
47 {
48  m_BeforeVertex = HepPoint3D(v.X(), v.Y(), v.Z());
50  return m_ErrorCode;
51 }
52 
55  m_InvariantMass = m;
56 
58 }
59 
60 
63  m_IsFixMass.push_back(true);
64 
66 }
67 
68 
71  m_IsFixMass.push_back(false);
72 
74 }
75 
76 
78 MassVertexFitKFit::setCorrelation(const HepMatrix& m) {
79  return KFitBase::setCorrelation(m);
80 }
81 
82 
86 }
87 
88 
89 const HepPoint3D
90 MassVertexFitKFit::getVertex(const int flag) const
91 {
92  if (flag == KFitConst::kAfterFit && !isFitted()) return HepPoint3D();
93 
94  switch (flag) {
96  return m_BeforeVertex;
97 
99  return m_AfterVertex;
100 
101  default:
102  KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
103  return HepPoint3D();
104  }
105 }
106 
107 
108 const HepSymMatrix
110 {
111  return m_AfterVertexError;
112 }
113 
114 
115 double
117 {
118  return m_InvariantMass;
119 }
120 
121 
122 double
124 {
125  return m_CHIsq;
126 }
127 
128 
129 const HepMatrix
131 {
132  if (!isTrackIDInRange(id)) return HepMatrix(3, KFitConst::kNumber7, 0);
133 
134  return m_AfterTrackVertexError[id];
135 }
136 
137 
138 double
140 {
141  if (!isFitted()) return -1;
142  if (!isTrackIDInRange(id)) return -1;
143 
144  if (m_IsFixMass[id]) {
145 
146  HepMatrix da(m_Tracks[id].getFitParameter(KFitConst::kBeforeFit) - m_Tracks[id].getFitParameter(KFitConst::kAfterFit));
147  int err_inverse = 0;
148  const double chisq = (da.T() * (m_Tracks[id].getFitError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
149 
150  if (err_inverse) {
151  KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
152  return -1;
153  }
154 
155  return chisq;
156 
157  } else {
158 
159  HepMatrix da(m_Tracks[id].getMomPos(KFitConst::kBeforeFit) - m_Tracks[id].getMomPos(KFitConst::kAfterFit));
160  int err_inverse = 0;
161  const double chisq = (da.T() * (m_Tracks[id].getError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
162 
163  if (err_inverse) {
164  KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
165  return -1;
166  }
167 
168  return chisq;
169  }
170 }
171 
172 
173 const HepMatrix
174 MassVertexFitKFit::getCorrelation(const int id1, const int id2, const int flag) const
175 {
176  if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
177  if (!isTrackIDInRange(id1)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
178  if (!isTrackIDInRange(id2)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
179 
180  switch (flag) {
182  return KFitBase::getCorrelation(id1, id2, flag);
183 
185  return makeError3(
186  this->getTrackMomentum(id1),
187  this->getTrackMomentum(id2),
188  m_V_al_1.sub(KFitConst::kNumber7 * id1 + 1, KFitConst::kNumber7 * (id1 + 1), KFitConst::kNumber7 * id2 + 1,
189  KFitConst::kNumber7 * (id2 + 1)),
190  m_IsFixMass[id1],
191  m_IsFixMass[id2]);
192 
193  default:
194  KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
195  return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
196  }
197 }
198 
199 
200 enum KFitError::ECode
202  return KFitBase::doFit2();
203 }
204 
205 
206 enum KFitError::ECode
209  {
211  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
212  return m_ErrorCode;
213  }
214 
215 
216  if (m_IsFixMass.size() == 0)
217  {
218  // If no fix_mass flag at all,
219  // all tracks are considered to be fixed at mass.
220  for (int i = 0; i < m_TrackCount; i++) this->fixMass();
221  } else if (m_IsFixMass.size() != (unsigned int)m_TrackCount)
222  {
224  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
225  return m_ErrorCode;
226  }
227 
228 
229  int index = 0;
230  m_al_0 = HepMatrix(KFitConst::kNumber7 * m_TrackCount, 1, 0);
231  m_property = HepMatrix(m_TrackCount, 3, 0);
232  m_V_al_0 = HepSymMatrix(KFitConst::kNumber7 * m_TrackCount, 0);
233 
234  for (auto& track : m_Tracks)
235  {
236  // momentum x,y,z and position x,y,z
237  m_al_0[index * KFitConst::kNumber7 + 0][0] = track.getMomentum(KFitConst::kBeforeFit).x();
238  m_al_0[index * KFitConst::kNumber7 + 1][0] = track.getMomentum(KFitConst::kBeforeFit).y();
239  m_al_0[index * KFitConst::kNumber7 + 2][0] = track.getMomentum(KFitConst::kBeforeFit).z();
240  m_al_0[index * KFitConst::kNumber7 + 3][0] = track.getMomentum(KFitConst::kBeforeFit).t();
241  m_al_0[index * KFitConst::kNumber7 + 4][0] = track.getPosition(KFitConst::kBeforeFit).x();
242  m_al_0[index * KFitConst::kNumber7 + 5][0] = track.getPosition(KFitConst::kBeforeFit).y();
243  m_al_0[index * KFitConst::kNumber7 + 6][0] = track.getPosition(KFitConst::kBeforeFit).z();
244  // these error
245  m_V_al_0.sub(index * KFitConst::kNumber7 + 1, track.getError(KFitConst::kBeforeFit));
246  // charge, mass, a
247  m_property[index][0] = track.getCharge();
248  m_property[index][1] = track.getMass();
249  const double c = KFitConst::kLightSpeed; // C++ bug?
250  // m_property[index][2] = -KFitConst::kLightSpeed * m_MagneticField * it->getCharge();
251  m_property[index][2] = -c * m_MagneticField * track.getCharge();
252  index++;
253  }
254 
255  // error between track and track
256  if (m_FlagCorrelation)
257  {
258  this->prepareCorrelation();
260  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
261  return m_ErrorCode;
262  }
263  }
264 
265  // vertex
266  m_v_a[0][0] = m_BeforeVertex.x();
267  m_v_a[1][0] = m_BeforeVertex.y();
268  m_v_a[2][0] = m_BeforeVertex.z();
269 
270  // set member matrix
271  m_al_1 = m_al_0;
272 
274  m_D = m_V_al_1.sub(1, m_TrackCount * 2 + 1, 1, KFitConst::kNumber7 * m_TrackCount);
275  m_E = m_V_al_1.sub(1, m_TrackCount * 2 + 1, 1, 3);
276  m_d = m_V_al_1.sub(1, m_TrackCount * 2 + 1, 1, 1);
277  m_V_D = m_V_al_1.sub(1, m_TrackCount * 2 + 1, 1, m_TrackCount * 2 + 1);
278  m_lam = m_V_al_1.sub(1, m_TrackCount * 2 + 1, 1, 1);
279  m_lam0 = m_V_al_1.sub(1, m_TrackCount * 2 + 1, 1, 1);
280  m_V_Dt = m_V_al_1.sub(1, m_TrackCount * 2 + 1, 1, m_TrackCount * 2 + 1);
282 
284 }
285 
286 
287 enum KFitError::ECode
289  // vertex
290  for (int i = 0; i < 3; i++) m_v[i][0] = m_v_a[i][0];
291 
293 }
294 
295 
296 enum KFitError::ECode
298  if (m_BeforeCorrelation.size() != static_cast<unsigned int>(m_TrackCount * (m_TrackCount - 1) / 2))
299  {
301  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
302  return m_ErrorCode;
303  }
304 
305  int row = 0, col = 0;
306 
307  for (auto& hm : m_BeforeCorrelation)
308  {
309  // counter
310  row++;
311  if (row == m_TrackCount) {
312  col++;
313  row = col + 1;
314  }
315 
316  int ii = 0, jj = 0;
317  for (int i = KFitConst::kNumber7 * row; i < KFitConst::kNumber7 * (row + 1); i++) {
318  for (int j = KFitConst::kNumber7 * col; j < KFitConst::kNumber7 * (col + 1); j++) {
319  m_V_al_0[i][j] = hm[ii][jj];
320  jj++;
321  }
322  jj = 0;
323  ii++;
324  }
325  }
326 
328 }
329 
330 
331 enum KFitError::ECode
333  Hep3Vector h3v;
334  int index = 0;
335  for (auto& pdata : m_Tracks)
336  {
337  // tracks
338  // momentum
339  h3v.setX(m_al_1[index * KFitConst::kNumber7 + 0][0]);
340  h3v.setY(m_al_1[index * KFitConst::kNumber7 + 1][0]);
341  h3v.setZ(m_al_1[index * KFitConst::kNumber7 + 2][0]);
342  if (m_IsFixMass[index])
343  pdata.setMomentum(HepLorentzVector(h3v, sqrt(h3v.mag2() + pdata.getMass()*pdata.getMass())), KFitConst::kAfterFit);
344  else
345  pdata.setMomentum(HepLorentzVector(h3v, m_al_1[index * KFitConst::kNumber7 + 3][0]), KFitConst::kAfterFit);
346  // position
347  pdata.setPosition(HepPoint3D(
348  m_al_1[index * KFitConst::kNumber7 + 4][0],
349  m_al_1[index * KFitConst::kNumber7 + 5][0],
351  // error of the tracks
352  pdata.setError(this->makeError3(pdata.getMomentum(),
353  m_V_al_1.sub(
354  index * KFitConst::kNumber7 + 1,
355  (index + 1)*KFitConst::kNumber7,
356  index * KFitConst::kNumber7 + 1,
357  (index + 1)*KFitConst::kNumber7), m_IsFixMass[index]),
359  if (m_ErrorCode != KFitError::kNoError) break;
360  index++;
361  }
362 
363  // vertex
364  m_AfterVertex.setX(m_v_a[0][0]);
365  m_AfterVertex.setY(m_v_a[1][0]);
366  m_AfterVertex.setZ(m_v_a[2][0]);
367  // error of the vertex
368  for (int i = 0; i < 3; i++) for (int j = i; j < 3; j++)
369  {
370  m_AfterVertexError[i][j] = m_V_E[i][j];
371  }
372  // error between vertex and tracks
373  for (int i = 0; i < m_TrackCount; i++)
374  {
375  HepMatrix hm(3, KFitConst::kNumber7, 0);
376  for (int j = 0; j < 3; j++) for (int k = 0; k < KFitConst::kNumber7; k++) {
377  hm[j][k] = m_Cov_v_al_1[j][KFitConst::kNumber7 * i + k];
378  }
379  if (m_IsFixMass[i])
380  m_AfterTrackVertexError.push_back(this->makeError4(m_Tracks[i].getMomentum(), hm));
381  else
382  m_AfterTrackVertexError.push_back(hm);
383  }
384 
386 }
387 
388 
389 enum KFitError::ECode
391  // Mass Constraint
392  HepMatrix al_1_prime(m_al_1);
393  HepMatrix Sum_al_1(4, 1, 0);
394  double energy[KFitConst::kMaxTrackCount2];
395  double a;
396 
397  for (int i = 0; i < m_TrackCount; i++)
398  {
399  a = m_property[i][2];
400  al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (m_v_a[1][0] - al_1_prime[i * KFitConst::kNumber7 + 5][0]);
401  al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (m_v_a[0][0] - al_1_prime[i * KFitConst::kNumber7 + 4][0]);
402  energy[i] = sqrt(al_1_prime[i * KFitConst::kNumber7 + 0][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] +
403  al_1_prime[i * KFitConst::kNumber7 + 1][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] +
404  al_1_prime[i * KFitConst::kNumber7 + 2][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] +
405  m_property[i][1] * m_property[i][1]);
406  }
407 
408  for (int i = 0; i < m_TrackCount; i++)
409  {
410  if (m_IsFixMass[i])
411  Sum_al_1[3][0] += energy[i];
412  else
413  Sum_al_1[3][0] += al_1_prime[i * KFitConst::kNumber7 + 3][0];
414 
415  for (int j = 0; j < 3; j++) Sum_al_1[j][0] += al_1_prime[i * KFitConst::kNumber7 + j][0];
416  }
417 
418  m_d[2 * m_TrackCount][0] =
419  + Sum_al_1[3][0] * Sum_al_1[3][0] - Sum_al_1[0][0] * Sum_al_1[0][0]
420  - Sum_al_1[1][0] * Sum_al_1[1][0] - Sum_al_1[2][0] * Sum_al_1[2][0]
422 
423  double Sum_a = 0., Sum_tmpx = 0., Sum_tmpy = 0.;
424  for (int i = 0; i < m_TrackCount; i++)
425  {
426  if (energy[i] == 0) {
428  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
429  return m_ErrorCode;
430  }
431 
432  a = m_property[i][2];
433 
434  if (m_IsFixMass[i]) {
435  double invE = 1. / energy[i];
436  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 0] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
437  Sum_al_1[0][0]);
438  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 1] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
439  Sum_al_1[1][0]);
440  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 2] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE -
441  Sum_al_1[2][0]);
442  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 3] = 0.;
443  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 4] = -2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
444  Sum_al_1[1][0]) * a;
445  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 5] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
446  Sum_al_1[0][0]) * a;
447  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 6] = 0.;
448  Sum_tmpx += al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE * a;
449  Sum_tmpy += al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE * a;
450  } else {
451  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 0] = -2.*Sum_al_1[0][0];
452  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 1] = -2.*Sum_al_1[1][0];
453  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 2] = -2.*Sum_al_1[2][0];
454  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 3] = 2.*Sum_al_1[3][0];
455  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 4] = 2.*Sum_al_1[1][0] * a;
456  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 5] = -2.*Sum_al_1[0][0] * a;
457  m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 6] = 0.;
458  }
459  Sum_a += a;
460  }
461 
462  // m_E
463  m_E[2 * m_TrackCount][0] = -2.*Sum_al_1[1][0] * Sum_a + 2.*Sum_al_1[3][0] * Sum_tmpy;
464  m_E[2 * m_TrackCount][1] = 2.*Sum_al_1[0][0] * Sum_a - 2.*Sum_al_1[3][0] * Sum_tmpx;
465  m_E[2 * m_TrackCount][2] = 0.;
466 
467  for (int i = 0; i < m_TrackCount; i++)
468  {
469  double B, S, U;
470  double sininv, sqrtag;
471 
472  double px = m_al_1[i * KFitConst::kNumber7 + 0][0];
473  double py = m_al_1[i * KFitConst::kNumber7 + 1][0];
474  double pz = m_al_1[i * KFitConst::kNumber7 + 2][0];
475  double x = m_al_1[i * KFitConst::kNumber7 + 4][0];
476  double y = m_al_1[i * KFitConst::kNumber7 + 5][0];
477  double z = m_al_1[i * KFitConst::kNumber7 + 6][0];
478  a = m_property[i][2];
479 
480  double pt = sqrt(px * px + py * py);
481 
482  if (pt == 0) {
484  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
485  return m_ErrorCode;
486  }
487 
488  double invPt = 1. / pt;
489  double invPt2 = invPt * invPt;
490  double dlx = m_v_a[0][0] - x;
491  double dly = m_v_a[1][0] - y;
492  double dlz = m_v_a[2][0] - z;
493  double a1 = -dlx * py + dly * px;
494  double a2 = dlx * px + dly * py;
495  double r2d2 = dlx * dlx + dly * dly;
496  double Rx = dlx - 2.*px * a2 * invPt2;
497  double Ry = dly - 2.*py * a2 * invPt2;
498 
499  if (a != 0.) { // charged
500 
501  B = a * a2 * invPt2;
502  if (fabs(B) > 1.) {
504  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
505  return m_ErrorCode;
506  }
507  // sin^(-1)(B)
508  sininv = asin(B);
509  double tmp0 = 1.0 - B * B;
510  if (tmp0 == 0) {
512  KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
513  return m_ErrorCode;
514  }
515  // 1/sqrt(1-B^2)
516  sqrtag = 1.0 / sqrt(tmp0);
517  S = sqrtag * invPt2;
518  U = dlz - pz * sininv / a;
519 
520  } else { // neutral
521 
522  B = 0.0;
523  sininv = 0.0;
524  sqrtag = 1.0;
525  S = invPt2;
526  U = dlz - pz * a2 * invPt2;
527 
528  }
529 
530  // d
531  m_d[i * 2 + 0][0] = a1 - 0.5 * a * r2d2;
532  m_d[i * 2 + 1][0] = U * pt;
533 
534  // D
535  m_D[i * 2 + 0][i * KFitConst::kNumber7 + 0] = dly;
536  m_D[i * 2 + 0][i * KFitConst::kNumber7 + 1] = -dlx;
537  m_D[i * 2 + 0][i * KFitConst::kNumber7 + 2] = 0.0;
538  m_D[i * 2 + 0][i * KFitConst::kNumber7 + 4] = py + a * dlx;
539  m_D[i * 2 + 0][i * KFitConst::kNumber7 + 5] = -px + a * dly;
540  m_D[i * 2 + 0][i * KFitConst::kNumber7 + 6] = 0.0;
541  m_D[i * 2 + 1][i * KFitConst::kNumber7 + 0] = -pz * pt * S * Rx + U * px * invPt;
542  m_D[i * 2 + 1][i * KFitConst::kNumber7 + 1] = -pz * pt * S * Ry + U * py * invPt;
543  if (a != 0.)
544  m_D[i * 2 + 1][i * KFitConst::kNumber7 + 2] = -sininv * pt / a;
545  else
546  m_D[i * 2 + 1][i * KFitConst::kNumber7 + 2] = -a2 * invPt;
547  m_D[i * 2 + 1][i * KFitConst::kNumber7 + 4] = px * pz * pt * S;
548  m_D[i * 2 + 1][i * KFitConst::kNumber7 + 5] = py * pz * pt * S;
549  m_D[i * 2 + 1][i * KFitConst::kNumber7 + 6] = -pt;
550 
551  // E
552  m_E[i * 2 + 0][0] = -py - a * dlx;
553  m_E[i * 2 + 0][1] = px - a * dly;
554  m_E[i * 2 + 0][2] = 0.0;
555  m_E[i * 2 + 1][0] = -px * pz * pt * S;
556  m_E[i * 2 + 1][1] = -py * pz * pt * S;
557  m_E[i * 2 + 1][2] = pt;
558  }
559 
561 }
562 
563 
564 enum KFitError::ECode
566  m_NDF = 2 * m_TrackCount - 3 + 1;
567 
569 }
570 
572 {
573  MakeMotherKFit kmm;
575  unsigned n = getTrackCount();
576  for (unsigned i = 0; i < n; ++i) {
578  getTrack(i).getCharge());
580  for (unsigned j = i + 1; j < n; ++j) {
581  kmm.setCorrelation(getCorrelation(i, j));
582  }
583  }
584  kmm.setVertex(getVertex());
586  m_ErrorCode = kmm.doMake();
588  return m_ErrorCode;
589  double chi2 = getCHIsq();
590  int ndf = getNDF();
591  double prob = TMath::Prob(chi2, ndf);
592  //
593  bool haschi2 = mother->hasExtraInfo("chiSquared");
594  if (haschi2) {
595  mother->setExtraInfo("chiSquared", chi2);
596  mother->setExtraInfo("ndf", ndf);
597  } else {
598  mother->addExtraInfo("chiSquared", chi2);
599  mother->addExtraInfo("ndf", ndf);
600  }
601 
602  mother->updateMomentum(
603  CLHEPToROOT::getTLorentzVector(kmm.getMotherMomentum()),
604  CLHEPToROOT::getTVector3(kmm.getMotherPosition()),
605  CLHEPToROOT::getTMatrixFSym(kmm.getMotherError()),
606  prob);
608  return m_ErrorCode;
609 }
Belle2::analysis::KFitBase::m_MagneticField
double m_MagneticField
Magnetic field.
Definition: KFitBase.h:311
Belle2::analysis::KFitBase::m_v
CLHEP::HepMatrix m_v
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:285
Belle2::analysis::KFitBase::makeError3
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.
Belle2::analysis::KFitBase::m_CHIsq
double m_CHIsq
chi-square of the fit.
Definition: KFitBase.h:297
Belle2::analysis::KFitBase::m_NDF
int m_NDF
NDF of the fit.
Definition: KFitBase.h:295
Belle2::analysis::KFitBase::m_ErrorCode
enum KFitError::ECode m_ErrorCode
Error code.
Definition: KFitBase.h:243
Belle2::analysis::KFitBase::m_FlagCorrelation
bool m_FlagCorrelation
Flag whether a correlation among tracks exists.
Definition: KFitBase.h:306
Belle2::analysis::MassVertexFitKFit::getTrackCHIsq
double getTrackCHIsq(const int id) const override
Get a chi-square of the track.
Definition: MassVertexFitKFit.cc:139
Belle2::analysis::KFitBase::m_D
CLHEP::HepMatrix m_D
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:266
Belle2::analysis::KFitConst::kBeforeFit
static const int kBeforeFit
Input parameter to specify before-fit when setting/getting a track attribute.
Definition: KFitConst.h:43
Belle2::analysis::MakeMotherKFit::setTrackVertexError
enum KFitError::ECode setTrackVertexError(const CLHEP::HepMatrix &e)
Set a vertex error matrix of the child particle in the addTrack'ed order.
Definition: MakeMotherKFit.cc:95
Belle2::analysis::KFitBase::getCorrelation
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:184
Belle2::analysis::MakeMotherKFit::setVertex
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set a vertex position of the mother particle.
Definition: MakeMotherKFit.cc:71
Belle2::analysis::MassVertexFitKFit::getTrackVertexError
const CLHEP::HepMatrix getTrackVertexError(const int id) const
Get a vertex error matrix of the track.
Definition: MassVertexFitKFit.cc:130
Belle2::analysis::KFitBase::isFitted
bool isFitted(void) const
Return false if fit is not performed yet or performed fit is failed; otherwise true.
Definition: KFitBase.cc:728
Belle2::analysis::MassVertexFitKFit::setInitialVertex
enum KFitError::ECode setInitialVertex(const HepPoint3D &v)
Set an initial vertex point for the mass-vertex constraint fit.
Definition: MassVertexFitKFit.cc:40
Belle2::analysis::KFitBase::doFit2
enum KFitError::ECode doFit2(void)
Perform a fit (used in VertexFitKFit::doFit() and MassVertexFitKFit::doFit()).
Definition: KFitBase.cc:578
Belle2::analysis::MassVertexFitKFit::prepareCorrelation
enum KFitError::ECode prepareCorrelation(void) override
Build a grand correlation matrix from input-track properties.
Definition: MassVertexFitKFit.cc:297
Belle2::analysis::KFitBase::m_NecessaryTrackCount
int m_NecessaryTrackCount
Number needed tracks to perform fit.
Definition: KFitBase.h:303
Belle2::analysis::MassVertexFitKFit::prepareInputSubMatrix
enum KFitError::ECode prepareInputSubMatrix(void) override
Build sub-matrices for minimum search from input-track properties.
Definition: MassVertexFitKFit.cc:288
Belle2::analysis::KFitBase::getTrackError
const CLHEP::HepSymMatrix getTrackError(const int id) const
Get an error matrix of the track.
Definition: KFitBase.cc:169
Belle2::analysis::MassVertexFitKFit::prepareInputMatrix
enum KFitError::ECode prepareInputMatrix(void) override
Build grand matrices for minimum search from input-track properties.
Definition: MassVertexFitKFit.cc:207
Belle2::analysis::KFitBase::m_lam0
CLHEP::HepMatrix m_lam0
See J.Tanaka Ph.D (2001) p138 for definition.
Definition: KFitBase.h:283
Belle2::analysis::KFitBase::m_Cov_v_al_1
CLHEP::HepMatrix m_Cov_v_al_1
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:291
Belle2::analysis::KFitBase::m_Tracks
std::vector< KFitTrack > m_Tracks
Container of input tracks.
Definition: KFitBase.h:249
Belle2::analysis::MassVertexFitKFit::setCorrelation
enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &m) override
Set a correlation matrix.
Definition: MassVertexFitKFit.cc:78
Belle2::analysis::KFitBase::m_property
CLHEP::HepMatrix m_property
Container of charges and masses.
Definition: KFitBase.h:263
Belle2::analysis::KFitBase::m_lam
CLHEP::HepMatrix m_lam
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:276
Belle2::analysis::MassVertexFitKFit::updateMother
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
Definition: MassVertexFitKFit.cc:571
Belle2::analysis::KFitBase::m_V_D
CLHEP::HepMatrix m_V_D
See J.Tanaka Ph.D (2001) p138 for definition.
Definition: KFitBase.h:271
Belle2::analysis::MassVertexFitKFit::m_AfterVertexError
CLHEP::HepSymMatrix m_AfterVertexError
Vertex error matrix after the fit.
Definition: MassVertexFitKFit.h:125
Belle2::analysis::KFitBase::m_al_1
CLHEP::HepMatrix m_al_1
See J.Tanaka Ph.D (2001) p136 for definition.
Definition: KFitBase.h:259
Belle2::analysis::MassVertexFitKFit::getInvariantMass
double getInvariantMass(void) const
Get an invariant mass.
Definition: MassVertexFitKFit.cc:116
Belle2::analysis::KFitError::displayError
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:79
Belle2::analysis::MassVertexFitKFit::m_IsFixMass
std::vector< int > m_IsFixMass
Array of flags whether the track property is fixed at the mass.
Definition: MassVertexFitKFit.h:132
Belle2::analysis::KFitError::kBadCorrelationSize
@ kBadCorrelationSize
Wrong correlation matrix size (internal error)
Definition: KFitError.h:58
Belle2::analysis::KFitBase::m_E
CLHEP::HepMatrix m_E
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:279
Belle2::analysis::MakeMotherKFit::getMotherError
const CLHEP::HepSymMatrix getMotherError(void) const
Get an error matrix of the mother particle.
Definition: MakeMotherKFit.cc:185
Belle2::analysis::KFitBase::setZeroCorrelation
virtual enum KFitError::ECode setZeroCorrelation(void)
Indicate no correlation between tracks.
Definition: KFitBase.cc:86
Belle2::analysis::MassVertexFitKFit::prepareOutputMatrix
enum KFitError::ECode prepareOutputMatrix(void) override
Build an output error matrix.
Definition: MassVertexFitKFit.cc:332
Belle2::analysis::MassVertexFitKFit::unfixMass
enum KFitError::ECode unfixMass(void)
Tell the object to unfix the last added track property at the invariant mass.
Definition: MassVertexFitKFit.cc:70
Belle2::analysis::KFitBase::m_FlagFitted
bool m_FlagFitted
Flag to indicate if the fit is performed and succeeded.
Definition: KFitBase.h:245
Belle2::analysis::MakeMotherKFit
MakeMotherKFit is a class to build mother particle from kinematically fitted daughters.
Definition: MakeMotherKFit.h:41
Belle2::analysis::MassVertexFitKFit::getVertexError
const CLHEP::HepSymMatrix getVertexError(void) const
Get a fitted vertex error matrix.
Definition: MassVertexFitKFit.cc:109
Belle2::analysis::MakeMotherKFit::getMotherMomentum
const CLHEP::HepLorentzVector getMotherMomentum(void) const
Get a Lorentz vector of the mother particle.
Definition: MakeMotherKFit.cc:171
Belle2::analysis::KFitError::kNoError
@ kNoError
No error.
Definition: KFitError.h:45
Belle2::analysis::MassVertexFitKFit::makeCoreMatrix
enum KFitError::ECode makeCoreMatrix(void) override
Build matrices using the kinematical constraint.
Definition: MassVertexFitKFit.cc:390
Belle2::analysis::MassVertexFitKFit::calculateNDF
enum KFitError::ECode calculateNDF(void) override
Calculate an NDF of the fit.
Definition: MassVertexFitKFit.cc:565
Belle2::analysis::KFitBase::getTrackCount
int getTrackCount(void) const
Get the number of added tracks.
Definition: KFitBase.cc:108
Belle2::analysis::KFitConst::kAfterFit
static const int kAfterFit
Input parameter to specify after-fit when setting/getting a track attribute.
Definition: KFitConst.h:45
Belle2::analysis::KFitBase::m_v_a
CLHEP::HepMatrix m_v_a
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:287
Belle2::analysis::MassVertexFitKFit::fixMass
enum KFitError::ECode fixMass(void)
Tell the object to fix the last added track property at the invariant mass.
Definition: MassVertexFitKFit.cc:62
Belle2::analysis::KFitConst::kMaxTrackCount
static const int kMaxTrackCount
Maximum track size.
Definition: KFitConst.h:48
Belle2::analysis::KFitConst::kMaxTrackCount2
static const int kMaxTrackCount2
Maximum track size (internal use)
Definition: KFitConst.h:50
Belle2::analysis::MassVertexFitKFit::m_InvariantMass
double m_InvariantMass
Invariant mass.
Definition: MassVertexFitKFit.h:130
Belle2::analysis::KFitBase::m_V_al_1
CLHEP::HepMatrix m_V_al_1
See J.Tanaka Ph.D (2001) p138 for definition.
Definition: KFitBase.h:274
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:19
Belle2::analysis::KFitBase::getNDF
virtual int getNDF(void) const
Get an NDF of the fit.
Definition: KFitBase.cc:115
Belle2::analysis::KFitError::kDivisionByZero
@ kDivisionByZero
Division by zero (bad track property or internal error)
Definition: KFitError.h:63
Belle2::analysis::MassVertexFitKFit::getVertex
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
Definition: MassVertexFitKFit.cc:90
Belle2::analysis::KFitBase::setCorrelation
virtual enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &c)
Set a correlation matrix.
Definition: KFitBase.cc:71
Belle2::analysis::MakeMotherKFit::setMagneticField
enum KFitError::ECode setMagneticField(const double mf)
Change a magnetic field from the default value KFitConst::kDefaultMagneticField.
Definition: MakeMotherKFit.cc:63
Belle2::analysis::MassVertexFitKFit::~MassVertexFitKFit
~MassVertexFitKFit(void)
Destruct the object.
Belle2::analysis::KFitBase::m_TrackCount
int m_TrackCount
Number of tracks.
Definition: KFitBase.h:301
Belle2::analysis::KFitError::kCannotGetMatrixInverse
@ kCannotGetMatrixInverse
Cannot calculate matrix inverse (bad track property or internal error)
Definition: KFitError.h:65
Belle2::analysis::MassVertexFitKFit::setZeroCorrelation
enum KFitError::ECode setZeroCorrelation(void) override
Indicate no correlation between tracks.
Definition: MassVertexFitKFit.cc:84
Belle2::analysis::MassVertexFitKFit::setInvariantMass
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-vertex constraint fit.
Definition: MassVertexFitKFit.cc:54
Belle2::analysis::KFitBase::m_al_0
CLHEP::HepMatrix m_al_0
See J.Tanaka Ph.D (2001) p136 for definition.
Definition: KFitBase.h:257
Belle2::analysis::KFitConst::kLightSpeed
static constexpr double kLightSpeed
Speed of light.
Definition: KFitConst.h:62
Belle2::analysis::MakeMotherKFit::setCorrelation
enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &e)
Set a correlation matrix.
Definition: MakeMotherKFit.cc:119
Belle2::analysis::MakeMotherKFit::setVertexError
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set a vertex error matrix of the mother particle.
Definition: MakeMotherKFit.cc:79
Belle2::analysis::KFitBase::getTrack
const KFitTrack getTrack(const int id) const
Get a specified track object.
Definition: KFitBase.cc:176
Belle2::analysis::MassVertexFitKFit::m_AfterTrackVertexError
std::vector< CLHEP::HepMatrix > m_AfterTrackVertexError
array of vertex error matrices after the fit.
Definition: MassVertexFitKFit.h:127
Belle2::analysis::KFitBase::m_d
CLHEP::HepMatrix m_d
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:268
Belle2::analysis::MakeMotherKFit::getMotherPosition
const HepPoint3D getMotherPosition(void) const
Get a position of the mother particle.
Definition: MakeMotherKFit.cc:178
Belle2::Particle
Class to store reconstructed particles.
Definition: Particle.h:77
Belle2::analysis::KFitBase::m_V_Dt
CLHEP::HepMatrix m_V_Dt
See J.Tanaka Ph.D (2001) p138 for definition.
Definition: KFitBase.h:289
Belle2::analysis::MakeMotherKFit::addTrack
enum KFitError::ECode addTrack(const KFitTrack &kp)
Add a track to the make-mother object.
Definition: MakeMotherKFit.cc:40
Belle2::analysis::MassVertexFitKFit::getCorrelation
const CLHEP::HepMatrix getCorrelation(const int id1, const int id2, const int flag=KFitConst::kAfterFit) const override
Get a correlation matrix between two tracks.
Definition: MassVertexFitKFit.cc:174
Belle2::analysis::KFitBase::isTrackIDInRange
bool isTrackIDInRange(const int id) const
Check if the id is in the range.
Definition: KFitBase.cc:739
Belle2::analysis::MassVertexFitKFit::m_BeforeVertex
HepPoint3D m_BeforeVertex
Vertex position before the fit.
Definition: MassVertexFitKFit.h:120
Belle2::analysis::KFitBase::getTrackPosition
const HepPoint3D getTrackPosition(const int id) const
Get a position of the track.
Definition: KFitBase.cc:162
Belle2::analysis::KFitError::ECode
ECode
ECode is a error code enumerate.
Definition: KFitError.h:43
Belle2::analysis::MakeMotherKFit::doMake
enum KFitError::ECode doMake(void)
Perform a reconstruction of mother particle.
Definition: MakeMotherKFit.cc:192
Belle2::analysis::KFitError::kOutOfRange
@ kOutOfRange
Specified track-id out of range.
Definition: KFitError.h:51
HepGeom::Point3D< double >
Belle2::analysis::KFitBase::makeError4
const CLHEP::HepMatrix makeError4(const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
Rebuild an error matrix from a Lorentz vector and an error matrix.
Definition: KFitBase.cc:439
Belle2::analysis::KFitConst::kNumber7
static const int kNumber7
Constant 7 to check matrix size (internal use)
Definition: KFitConst.h:40
Belle2::analysis::KFitBase::m_V_E
CLHEP::HepMatrix m_V_E
See J.Tanaka Ph.D (2001) p138 for definition.
Definition: KFitBase.h:281
Belle2::analysis::MassVertexFitKFit::doFit
enum KFitError::ECode doFit(void)
Perform a mass-vertex-constraint fit.
Definition: MassVertexFitKFit.cc:201
Belle2::analysis::KFitError::kBadTrackSize
@ kBadTrackSize
Track count too small to perform fit.
Definition: KFitError.h:54
Belle2::analysis::KFitBase::m_V_al_0
CLHEP::HepSymMatrix m_V_al_0
See J.Tanaka Ph.D (2001) p137 for definition.
Definition: KFitBase.h:255
Belle2::analysis::KFitBase::m_BeforeCorrelation
std::vector< CLHEP::HepMatrix > m_BeforeCorrelation
Container of input correlation matrices.
Definition: KFitBase.h:251
Belle2::analysis::MassVertexFitKFit::m_AfterVertex
HepPoint3D m_AfterVertex
Vertex position after the fit.
Definition: MassVertexFitKFit.h:123
Belle2::analysis::MassVertexFitKFit::getCHIsq
double getCHIsq(void) const override
Get a chi-square of the fit.
Definition: MassVertexFitKFit.cc:123
Belle2::analysis::KFitBase::getTrackMomentum
const CLHEP::HepLorentzVector getTrackMomentum(const int id) const
Get a Lorentz vector of the track.
Definition: KFitBase.cc:155