10 #include <analysis/modules/ParticleVertexFitter/ParticleVertexFitterModule.h>
13 #include <framework/gearbox/Unit.h>
14 #include <framework/gearbox/Const.h>
15 #include <framework/geometry/B2Vector3.h>
16 #include <framework/logging/Logger.h>
17 #include <framework/particledb/EvtGenDatabasePDG.h>
20 #include <analysis/dataobjects/Particle.h>
21 #include <analysis/dataobjects/Btube.h>
24 #include <analysis/utility/CLHEPToROOT.h>
25 #include <analysis/utility/PCmsLabTransform.h>
26 #include <analysis/utility/ParticleCopy.h>
27 #include <analysis/utility/ROOTToCLHEP.h>
30 #include <framework/geometry/BFieldManager.h>
33 #include <analysis/VertexFitting/KFit/KFitConst.h>
36 #include <TRotation.h>
53 ParticleVertexFitterModule::ParticleVertexFitterModule() :
Module(),
63 "Confidence level to accept the fit. Particle candidates with "
64 "p-value less than confidenceLevel are removed from the particle "
65 "list. If set to -1, all candidates are kept; if set to 0, "
66 "the candidates failing the fit are removed.",
69 addParam(
"fitType",
m_fitType,
"type of the kinematic fit (vertex, massvertex, mass)",
string(
"vertex"));
71 "additional constraint on vertex: ipprofile, iptube, mother, iptubecut, pointing, btube",
73 addParam(
"decayString",
m_decayString,
"specifies which daughter particles are included in the kinematic fit",
string(
""));
78 "Type::[int]. List of daughter particles to mass constrain with int = pdg code. (only for MassFourCKFit)", {});
80 "Type::[string]. List of daughter particles to mass constrain with string = particle name. (only for MassFourCKFit)", {});
95 B2DEBUG(1,
"ParticleVertexFitterModule : magnetic field = " <<
m_Bfield);
103 TParticlePDG* particletemp = TDatabasePDG::Instance()->GetParticle((containedParticle).c_str());
110 B2INFO(
"ParticleVertexFitter: Using specified decay string: " <<
m_decayString);
112 B2INFO(
"ParticleVertexFitter: Additional " <<
m_withConstraint <<
" will be applied");
130 TMatrixDSym beamSpotCov(3);
147 std::vector<unsigned int> toRemove;
148 unsigned int nParticles =
m_plist->getListSize();
150 for (
unsigned iPart = 0; iPart < nParticles; iPart++) {
156 else B2ERROR(
"Daughters update works only when all daughters are selected. Daughters will not be updated");
164 TMatrixFSym mother_errMatrix(7);
166 for (
int k = 0; k < 7; k++) {
167 for (
int j = 0; j < 7; j++) {
168 if (mother_errMatrix[k][j] > 0) {
195 m_plist->removeParticles(toRemove);
207 B2FATAL(
"ParticleVertexFitter: No magnetic field");
217 B2FATAL(
"ParticleVertexFitter: " <<
m_withConstraint <<
" ***invalid Constraint ");
224 B2FATAL(
"ParticleVertexFitter: KFit does not support yet selection of daughters via decay string except for vertex fit!");
240 B2FATAL(
"ParticleVertexFitter: Invalid options - mass-constrained fit using KFit does not work with iptube or ipprofile constraint.");
251 B2FATAL(
"ParticleVertexFitter: Invalid options - mass fit using KFit does not work with iptube or ipprofile constraint.");
260 B2FATAL(
"ParticleVertexFitter: Invalid options - four C fit using KFit does not work with iptube or ipprofile constraint.");
269 B2FATAL(
"ParticleVertexFitter: Invalid options - four C fit using KFit does not work with iptube or ipprofile constraint.");
278 B2FATAL(
"ParticleVertexFitter: Invalid options - recoil mass fit using KFit does not work with iptube or ipprofile constraint.");
291 B2FATAL(
"ParticleVertexFitter: " <<
m_fitType <<
" ***invalid fit type for the vertex fitter ");
298 }
catch (
const rave::CheckedFloatException&) {
299 B2ERROR(
"Invalid inputs (nan/inf)?");
306 B2FATAL(
"ParticleVertexFitter: " <<
m_vertexFitter <<
" ***invalid vertex fitter ");
308 if (!ok)
return false;
318 std::vector<const Particle*>& twoPhotonChildren)
322 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
326 if (mother->
getProperty() == Particle::PropertyFlags::c_IsUnspecified and child->
getPValue() < 0) {
329 fitChildren.push_back(child);
335 auto itr = fitChildren.begin();
336 while (itr != fitChildren.end()) {
340 B2WARNING(
"Daughter with PDG code " << child->
getPDGCode() <<
" does not have a valid error matrix.");
343 bool isTwoPhotonParticle =
false;
349 isTwoPhotonParticle =
true;
354 if (isTwoPhotonParticle) {
356 twoPhotonChildren.push_back(child);
357 itr = fitChildren.erase(itr);
367 const std::vector<const Particle*>& fitChildren)
369 if (fitChildren.empty())
370 B2WARNING(
"[ParticleVertexFitterModule::fillNotFitParticles] fitChildren is empty! Please call fillFitParticles firstly");
371 if (!notFitChildren.empty())
372 B2WARNING(
"[ParticleVertexFitterModule::fillNotFitParticles] notFitChildren is NOT empty!"
373 <<
" The function should be called only once");
379 std::function<bool(
const Particle*)> funcCheckInFit =
380 [&funcCheckInFit, ¬FitChildren, fitChildren](
const Particle * part) {
384 if (std::find(fitChildren.begin(), fitChildren.end(), part) != fitChildren.end())
388 if (part->getNDaughters() == 0)
393 bool isAnyChildrenInFit =
false;
394 vector<const Particle*> notFitChildren_tmp;
395 for (
unsigned ichild = 0; ichild < part->getNDaughters(); ichild++) {
398 bool isChildrenInFit = funcCheckInFit(child);
399 isAnyChildrenInFit = isChildrenInFit or isAnyChildrenInFit;
402 if (!isChildrenInFit)
403 notFitChildren_tmp.push_back(child);
407 if (isAnyChildrenInFit)
408 notFitChildren.insert(notFitChildren.end(), notFitChildren_tmp.begin(), notFitChildren_tmp.end());
412 return isAnyChildrenInFit;
417 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
419 bool isGivenParticleOrAnyChildrenInFit = funcCheckInFit(child);
420 if (!isGivenParticleOrAnyChildrenInFit)
421 notFitChildren.push_back(child);
444 TMatrixFSym errMatrix(3);
445 for (
int i = 0; i < 3; i++)
446 for (
int j = 0; j < 3; j++)
447 errMatrix(i, j) = posErrorMatrix[i][j];
449 g1ErrMatrix.SetSub(4, errMatrix);
450 g2ErrMatrix.SetSub(4, errMatrix);
466 int err = km.
doFit();
484 std::vector<const Particle*> fitChildren;
485 std::vector<const Particle*> twoPhotonChildren;
486 bool validChildren =
fillFitParticles(mother, fitChildren, twoPhotonChildren);
491 std::vector<const Particle*> notFitChildren;
495 if (twoPhotonChildren.size() > 1) {
496 B2FATAL(
"[ParticleVertexFitterModule::doKVertexFit] Vertex fit using KFit does not support fit with multiple particles decaying to two photons like pi0 (yet).");
499 if ((fitChildren.size() < 2 && !ipTubeConstraint) || fitChildren.size() < 1) {
500 B2WARNING(
"[ParticleVertexFitterModule::doKVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is too low.");
508 for (
auto& child : fitChildren)
511 if (ipProfileConstraint)
514 if (ipTubeConstraint)
518 int err = kv.
doFit();
528 if (twoPhotonChildren.size() == 0)
531 else if (twoPhotonChildren.size() == 1) {
537 const Particle* twoPhotonDaughter = twoPhotonChildren[0];
547 for (
auto& child : fitChildren)
552 if (ipProfileConstraint)
564 ROOT::Math::PxPyPzEVector total4Vector(mother->
get4Vector());
565 for (
auto& child : notFitChildren)
566 total4Vector += child->get4Vector();
576 std::vector<const Particle*> fitChildren;
577 std::vector<const Particle*> twoPhotonChildren;
578 bool validChildren =
fillFitParticles(mother, fitChildren, twoPhotonChildren);
583 if (twoPhotonChildren.size() > 1) {
584 B2FATAL(
"[ParticleVertexFitterModule::doKVertexFit] MassVertex fit using KFit does not support fit with multiple particles decaying to two photons like pi0 (yet).");
587 if (fitChildren.size() < 2) {
588 B2WARNING(
"[ParticleVertexFitterModule::doKVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is less than 2.");
593 if (twoPhotonChildren.size() == 0) {
598 for (
auto child : fitChildren)
602 int err = kmv.
doFit();
608 }
else if (twoPhotonChildren.size() == 1) {
617 for (
auto child : fitChildren)
621 int err = kv.
doFit();
625 const Particle* twoPhotonDaughter = twoPhotonChildren[0];
635 for (
auto child : fitChildren)
660 std::vector<const Particle*> fitChildren;
661 std::vector<const Particle*> twoPhotonChildren;
662 bool validChildren =
fillFitParticles(mother, fitChildren, twoPhotonChildren);
667 if (twoPhotonChildren.size() > 0) {
668 B2FATAL(
"[ParticleVertexFitterModule::doKMassPointingVertexFit] MassPointingVertex fit using KFit does not support fit with two-photon daughters (yet).");
671 if (fitChildren.size() < 2) {
672 B2WARNING(
"[ParticleVertexFitterModule::doKMassPointingVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is less than 2.");
681 for (
auto child : fitChildren)
685 HepPoint3D productionVertex(mother->
getExtraInfo(
"prodVertX"),
689 int err = kmpv.
doFit();
690 if (err != 0)
return false;
704 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
707 if (child->
getPValue() < 0)
return false;
715 int err = km.
doFit();
717 if (err != 0)
return false;
731 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
736 if (!err)
return false;
738 if (child->
getPValue() < 0)
return false;
748 int err = kf.
doFit();
750 if (err != 0)
return false;
764 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
770 std::vector<unsigned> childId;
773 if (!err)
return false;
775 if (child->
getPValue() < 0)
return false;
784 int err = kf.
doFit();
786 if (err != 0)
return false;
798 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
801 if (child->
getPValue() < 0)
return false;
813 int err = kf.
doFit();
815 if (err != 0)
return false;
833 std::vector<Particle*> daughters = mother->
getDaughters();
836 if (daughters.size() != track_count)
839 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
844 ROOT::Math::PxPyPzEVector i4Vector(kv.
getTrackMomentum(iChild).x() - a * dy,
848 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
850 daughters[iChild]->setVertex(
852 daughters[iChild]->setMomentumVertexErrorMatrix(
861 if (fitChildren.size() != track_count)
864 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
865 auto daughter =
const_cast<Particle*
>(fitChildren[iChild]);
871 ROOT::Math::PxPyPzEVector i4Vector(kv.
getTrackMomentum(iChild).x() - a * dy,
875 daughter->set4VectorDividingByMomentumScaling(i4Vector);
877 daughter->setVertex(CLHEPToROOT::getXYZVector(kv.
getTrackPosition(iChild)));
878 daughter->setMomentumVertexErrorMatrix(CLHEPToROOT::getTMatrixFSym(kv.
getTrackError(iChild)));
882 std::function<bool(
Particle*)> funcUpdateMomentum =
883 [&funcUpdateMomentum, fitChildren](
Particle * part) {
885 if (part->getNDaughters() == 0) {
887 if (std::find(fitChildren.begin(), fitChildren.end(), part) != fitChildren.end())
893 bool includeFitChildren =
false;
896 for (
auto daughter : part->getDaughters())
897 includeFitChildren = funcUpdateMomentum(daughter) || includeFitChildren;
899 if (includeFitChildren) {
901 ROOT::Math::PxPyPzEVector sum4Vector;
902 for (
auto daughter : part->getDaughters())
903 sum4Vector += daughter->get4Vector();
905 part->set4VectorDividingByMomentumScaling(sum4Vector);
908 return includeFitChildren;
913 funcUpdateMomentum(daughter);
932 std::vector<Particle*> daughters = mother->
getDaughters();
935 if (daughters.size() != track_count)
938 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
943 ROOT::Math::PxPyPzEVector i4Vector(kmv.
getTrackMomentum(iChild).x() - a * dy,
947 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
949 daughters[iChild]->setVertex(
951 daughters[iChild]->setMomentumVertexErrorMatrix(
972 std::vector<Particle*> daughters = mother->
getDaughters();
975 if (daughters.size() != track_count)
978 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
983 ROOT::Math::PxPyPzEVector i4Vector(kmpv.
getTrackMomentum(iChild).x() - a * dy,
987 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
989 daughters[iChild]->setVertex(
991 daughters[iChild]->setMomentumVertexErrorMatrix(
1011 std::vector<Particle*> daughters = mother->
getDaughters();
1014 if (daughters.size() != track_count)
1017 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
1022 ROOT::Math::PxPyPzEVector i4Vector(km.
getTrackMomentum(iChild).x() - a * dy,
1026 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
1028 daughters[iChild]->setVertex(
1030 daughters[iChild]->setMomentumVertexErrorMatrix(
1052 std::vector<Particle*> daughters = mother->
getDaughters();
1054 const unsigned nd = daughters.size();
1056 std::vector<std::vector<unsigned>> pars;
1057 std::vector<Particle*> allparticles;
1058 for (
unsigned ichild = 0; ichild < nd; ichild++) {
1060 std::vector<unsigned> pard;
1061 if (daughter->getNDaughters() > 0) {
1063 pars.push_back(pard);
1064 allparticles.push_back(daughters[ichild]);
1067 pars.push_back(pard);
1068 allparticles.push_back(daughters[ichild]);
1074 if (l != track_count)
1077 for (
unsigned iDaug = 0; iDaug < allparticles.size(); iDaug++) {
1078 ROOT::Math::PxPyPzEVector childMoms;
1079 ROOT::Math::XYZVector childPoss;
1080 TMatrixFSym childErrMatrixs(7);
1081 for (
unsigned int iChild : pars[iDaug]) {
1082 childMoms = childMoms +
1083 CLHEPToROOT::getLorentzVector(
1085 childPoss = childPoss +
1086 CLHEPToROOT::getXYZVector(
1088 TMatrixFSym childErrMatrix =
1090 childErrMatrixs = childErrMatrixs + childErrMatrix;
1092 allparticles[iDaug]->set4Vector(childMoms);
1093 allparticles[iDaug]->setVertex(childPoss);
1094 allparticles[iDaug]->setMomentumVertexErrorMatrix(childErrMatrixs);
1114 std::vector<Particle*> daughters = mother->
getDaughters();
1116 const unsigned nd = daughters.size();
1118 std::vector<std::vector<unsigned>> pars;
1119 std::vector<Particle*> allparticles;
1120 for (
unsigned ichild = 0; ichild < nd; ichild++) {
1122 std::vector<unsigned> pard;
1123 if (daughter->getNDaughters() > 0) {
1125 pars.push_back(pard);
1126 allparticles.push_back(daughters[ichild]);
1129 pars.push_back(pard);
1130 allparticles.push_back(daughters[ichild]);
1136 if (l != track_count)
1139 for (
unsigned iDaug = 0; iDaug < allparticles.size(); iDaug++) {
1140 ROOT::Math::PxPyPzEVector childMoms;
1141 ROOT::Math::XYZVector childPoss;
1142 TMatrixFSym childErrMatrixs(7);
1143 for (
unsigned int iChild : pars[iDaug]) {
1144 childMoms = childMoms +
1145 CLHEPToROOT::getLorentzVector(
1147 childPoss = childPoss +
1148 CLHEPToROOT::getXYZVector(
1150 TMatrixFSym childErrMatrix =
1152 childErrMatrixs = childErrMatrixs + childErrMatrix;
1154 allparticles[iDaug]->set4Vector(childMoms);
1155 allparticles[iDaug]->setVertex(childPoss);
1156 allparticles[iDaug]->setMomentumVertexErrorMatrix(childErrMatrixs);
1176 std::vector<Particle*> daughters = mother->
getDaughters();
1179 if (daughters.size() != track_count)
1182 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
1187 ROOT::Math::PxPyPzEVector i4Vector(kf.
getTrackMomentum(iChild).x() - a * dy,
1191 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
1193 daughters[iChild]->setVertex(
1195 daughters[iChild]->setMomentumVertexErrorMatrix(
1205 std::vector<unsigned>& parm, std::vector<Particle*>& allparticles,
const Particle* daughter)
1207 std::vector <Belle2::Particle*> childs = daughter->getDaughters();
1208 for (
unsigned ichild = 0; ichild < daughter->getNDaughters(); ichild++) {
1209 const Particle* child = daughter->getDaughter(ichild);
1210 std::vector<unsigned> pard;
1213 parm.insert(parm.end(), pard.begin(), pard.end());
1214 pars.push_back(pard);
1215 allparticles.push_back(childs[ichild]);
1219 pars.push_back(pard);
1220 allparticles.push_back(childs[ichild]);
1246 for (
auto& itrack : tracksVertex) {
1247 if (itrack != mother) rf.
addTrack(itrack);
1253 bool mothSel =
false;
1255 for (
unsigned itrack = 0; itrack < tracksVertex.size(); itrack++) {
1256 if (tracksVertex[itrack] != mother) {
1257 rsf.
addTrack(tracksVertex[itrack]);
1258 B2DEBUG(1,
"ParticleVertexFitterModule: Adding particle " << tracksName[itrack] <<
" to vertex fit ");
1261 if (tracksVertex[itrack] == mother) mothSel =
true;
1266 bool mothIPfit =
false;
1267 if (tracksVertex.size() == 1 && mothSel ==
true &&
m_withConstraint !=
"" && nTrk == 0) {
1269 if (tracksVertex[0] != mother)
1270 B2FATAL(
"ParticleVertexFitterModule: FATAL Error in IP constrained mother fit");
1276 ROOT::Math::XYZVector pos;
1277 TMatrixDSym RerrMatrix(3);
1283 for (
auto& itrack : tracksVertex) {
1285 nvert = rsg.
fit(
"kalman");
1288 RerrMatrix = rsg.
getCov(0);
1290 ROOT::Math::PxPyPzEVector mom(mother->
get4Vector());
1291 TMatrixDSym errMatrix(7);
1292 for (
int i = 0; i < 7; i++) {
1293 for (
int j = 0; j < 7; j++) {
1294 if (i > 3 && j > 3) {errMatrix[i][j] = RerrMatrix[i - 4][j - 4];}
1295 else {errMatrix[i][j] = 0;}
1315 }
else {
return false;}
1325 ROOT::Math::PxPyPzEVector mom(mother->
get4Vector());
1326 TMatrixDSym errMatrix(7);
1327 for (
int i = 0; i < 7; i++) {
1328 for (
int j = 0; j < 7; j++) {
1329 if (i > 3 && j > 3) {errMatrix[i][j] = RerrMatrix[i - 4][j - 4];}
1330 else {errMatrix[i][j] = 0;}
1334 }
else {
return false;}
1337 if (mothSel && nTrk > 1) {
1340 int nKfit = rf.
fit();
1344 if (nKfit > 0) {
return true;}
1353 int nVert = rf.
fit();
1356 if (nVert != 1)
return false;
1363 int nVert = rf.
fit();
1365 if (nVert != 1)
return false;
1370 int nVert = rf.
fit();
1373 if (nVert != 1)
return false;
1376 B2FATAL(
"fitType : " <<
m_fitType <<
" ***invalid fit type ");
1383 const std::vector<const Particle*>& tracksVertex)
1392 for (
auto& vi : tracksVertex) {
1394 nNotIncluded = nNotIncluded - 1;
1402 if (nNotIncluded == 0) isAll =
true;
1408 for (
unsigned ichild = 0; ichild < particle->
getNDaughters(); ichild++) {
1412 if (child->
getPValue() < 0)
return false;
1421 std::vector<unsigned>& particleId)
1423 for (
unsigned ichild = 0; ichild < particle->
getNDaughters(); ichild++) {
1428 std::vector<unsigned> childId;
1431 particleId.insert(particleId.end(), childId.begin(), childId.end());
1433 if (child->
getPValue() < 0)
return false;
1443 HepPoint3D pos(0.0, 0.0, 0.0);
1444 CLHEP::HepSymMatrix covMatrix(3, 0);
1446 for (
int i = 0; i < 3; i++) {
1448 for (
int j = 0; j < 3; j++) {
1458 CLHEP::HepSymMatrix err(7, 0);
1460 for (
int i = 0; i < 3; i++) {
1461 for (
int j = 0; j < 3; j++) {
1470 ROOTToCLHEP::getHepLorentzVector(iptube_mom),
1483 TMatrixDSym beamSpotCov =
m_beamSpotDB->getCovVertex();
1484 beamSpotCov(2, 2) = cut * cut;
1485 double thetab = boostDir.
Theta();
1486 double phib = boostDir.
Phi();
1488 double stb = TMath::Sin(thetab);
1489 double ctb = TMath::Cos(thetab);
1490 double spb = TMath::Sin(phib);
1491 double cpb = TMath::Cos(phib);
1494 TMatrix rz(3, 3); rz(2, 2) = 1;
1495 rz(0, 0) = cpb; rz(0, 1) = spb;
1496 rz(1, 0) = -1 * spb; rz(1, 1) = cpb;
1498 TMatrix ry(3, 3); ry(1, 1) = 1;
1499 ry(0, 0) = ctb; ry(0, 2) = -1 * stb;
1500 ry(2, 0) = stb; ry(2, 2) = ctb;
1502 TMatrix r(3, 3); r.Mult(rz, ry);
1503 TMatrix rt(3, 3); rt.Transpose(r);
1505 TMatrix TubePart(3, 3); TubePart.Mult(rt, beamSpotCov);
1506 TMatrix Tube(3, 3); Tube.Mult(TubePart, r);
1515 TMatrixDSym beamSpotCov =
m_beamSpotDB->getCovVertex();
1516 for (
int i = 0; i < 3; i++)
1517 beamSpotCov(i, i) += width * width;
1524 double chi2TrackL = 0;
1526 for (
int iTrack = 0; iTrack < kv.
getTrackCount(); iTrack++) {
1538 TVectorD boostD(0, 6, 0., 0., 0., 0., boost3.
X(), boost3.
Y(), boost3.
Z(),
"END");
1540 double dLBoost = dPos.
Dot(boost3);
1542 chi2TrackL += TMath::Power(dLBoost, 2) / err.Similarity(boostD);
DataType Phi() const
The azimuth angle.
DataType Z() const
access variable Z (= .at(2) without boundary check)
DataType Theta() const
The polar angle.
DataType X() const
access variable X (= .at(0) without boundary check)
DataType Y() const
access variable Y (= .at(1) without boundary check)
B2Vector3< DataType > Unit() const
Unit vector parallel to this.
DataType Dot(const B2Vector3< DataType > &p) const
Scalar product.
void SetXYZ(DataType x, DataType y, DataType z)
set all coordinates using data type
static ROOT::Math::XYZVector getFieldInTesla(const ROOT::Math::XYZVector &pos)
return the magnetic field at a given position in Tesla.
For each MCParticle with hits in the CDC, this class stores some summarising information on those hit...
TMatrixFSym getTubeMatrix() const
Returns Btube matrix.
Eigen::Matrix< double, 3, 1 > getTubeCenter() const
Returns Btube center.
int getPDGCode() const
PDG code.
static const ParticleType pi0
neutral pion particle
static const double speedOfLight
[cm/ns]
static const ParticleType photon
photon particle
bool init(const std::string &str)
Initialise the DecayDescriptor from given string.
std::vector< std::string > getSelectionNames()
Return list of human readable names of selected particles.
std::vector< const Particle * > getSelectionParticles(const Particle *particle)
Get a vector of pointers with selected daughters in the decay tree.
void setDescription(const std::string &description)
Sets the description of the module.
void setPropertyFlags(unsigned int propertyFlags)
Sets the flags for the module properties.
@ c_ParallelProcessingCertified
This module can be run in parallel processing mode safely (All I/O must be done through the data stor...
TMatrixDSym m_beamSpotCov
Beam spot covariance matrix.
bool makeKMassMother(analysis::MassFitKFit &kv, Particle *p)
Update mother particle after mass fit using KFit.
std::string m_withConstraint
additional constraint on vertex
bool doKMassPointingVertexFit(Particle *p)
Mass-constrained vertex fit with additional pointing constraint using KFit.
void smearBeamSpot(double width)
smear beam spot covariance
bool makeMassKFourCMother(analysis::MassFourCFitKFit &kv, Particle *p)
Update mother particle after MassFourC fit using KFit.
bool doKVertexFit(Particle *p, bool ipProfileConstraint, bool ipTubeConstraint)
Unconstrained vertex fit using KFit.
bool doKMassVertexFit(Particle *p)
Mass-constrained vertex fit using KFit.
virtual void initialize() override
Initialize the Module.
void addIPTubeToKFit(analysis::VertexFitKFit &kv)
Adds IPTube constraint to the vertex fit using KFit.
bool addChildofParticletoMassKFit(analysis::MassFourCFitKFit &kf, const Particle *particle, std::vector< unsigned > &particleId)
Adds given particle's child to the MassFourCFitKFit.
virtual void event() override
Event processor.
bool makeKMassPointingVertexMother(analysis::MassPointingVertexFitKFit &kv, Particle *p)
Update mother particle after mass-constrained vertex fit with additional pointing constraint using KF...
bool m_updateDaughters
flag for daughters update
std::string m_decayString
daughter particles selection
std::vector< std::string > m_massConstraintListParticlename
Name of the particles to be mass constraint (massfourC)
std::string m_listName
particle list name
std::vector< int > m_massConstraintList
PDG codes of the particles to be mass constraint (massfourC)
bool m_hasCovMatrix
flag for mother covariance matrix (PseudoFitter)
bool makeKVertexMother(analysis::VertexFitKFit &kv, Particle *p)
Update mother particle after unconstrained vertex fit using KFit.
bool redoTwoPhotonDaughterMassFit(Particle *postFit, const Particle *preFit, const analysis::VertexFitKFit &kv)
Combines preFit particle and vertex information from vertex fit kv to create new postFit particle.
bool makeKMassVertexMother(analysis::MassVertexFitKFit &kv, Particle *p)
Update mother particle after mass-constrained vertex fit using KFit.
bool makeKFourCMother(analysis::FourCFitKFit &kv, Particle *p)
Update mother particle after FourC fit using KFit.
bool fillFitParticles(const Particle *mother, std::vector< const Particle * > &fitChildren, std::vector< const Particle * > &twoPhotonChildren)
Fills valid particle's children (with valid error matrix) in the vector of Particles that will enter ...
bool doKFourCFit(Particle *p)
FourC fit using KFit.
B2Vector3D m_BeamSpotCenter
Beam spot position.
virtual void beginRun() override
Called when entering a new run.
bool doKRecoilMassFit(Particle *p)
RecoilMass fit using KFit.
void updateMapOfTrackAndDaughter(unsigned &l, std::vector< std::vector< unsigned >> &pars, std::vector< unsigned > &pard, std::vector< Particle * > &allparticles, const Particle *daughter)
update the map of daughter and tracks, find out which tracks belong to each daughter.
DBObjPtr< BeamSpot > m_beamSpotDB
Beam spot database object.
bool doVertexFit(Particle *p)
Main steering routine.
std::string m_vertexFitter
Vertex Fitter name.
bool doRaveFit(Particle *mother)
Fit using Rave.
bool makeKRecoilMassMother(analysis::RecoilMassKFit &kf, Particle *p)
Update mother particle after RecoilMass fit using KFit.
bool doKMassFit(Particle *p)
Mass fit using KFit.
double m_recoilMass
recoil mass for constraint
double m_confidenceLevel
required fit confidence level
bool doKMassFourCFit(Particle *p)
MassFourC fit using KFit.
DecayDescriptor m_decaydescriptor
Decay descriptor of decays to look for.
bool fillNotFitParticles(const Particle *mother, std::vector< const Particle * > ¬FitChildren, const std::vector< const Particle * > &fitChildren)
Fills valid particle's children (with valid error matrix) in the vector of Particles that will not en...
void findConstraintBoost(double cut)
calculate iptube constraint (quasi cylinder along boost direction) for RAVE fit
double m_Bfield
magnetic field from data base
bool addChildofParticletoKFit(analysis::FourCFitKFit &kv, const Particle *particle)
Adds given particle's child to the FourCFitKFit.
double getChi2TracksLBoost(const analysis::VertexFitKFit &kv)
calculate the chi2 using only lboost information of tracks
double m_smearing
smearing width applied to IP tube
std::string m_fitType
type of the kinematic fit
void addIPProfileToKFit(analysis::VertexFitKFit &kv)
Adds IPProfile constraint to the vertex fit using KFit.
StoreObjPtr< ParticleList > m_plist
particle list
bool allSelectedDaughters(const Particle *mother, const std::vector< const Particle * > &tracksVertex)
check if all the Daughters (o grand-daughters) are selected for the vertex fit
Class to store reconstructed particles.
TMatrixFSym getVertexErrorMatrix() const
Returns the 3x3 position error sub-matrix.
void writeExtraInfo(const std::string &name, const double value)
Sets the user defined extraInfo.
double getPValue() const
Returns chi^2 probability of fit if done or -1.
ROOT::Math::XYZVector getVertex() const
Returns vertex position (POCA for charged, IP for neutral FS particles)
bool hasExtraInfo(const std::string &name) const
Return whether the extra info with the given name is set.
int getPDGCode(void) const
Returns PDG code.
int getProperty() const
Returns particle property as a bit pattern The values are defined in the PropertyFlags enum and descr...
unsigned getNDaughters(void) const
Returns number of daughter particles.
std::vector< Belle2::Particle * > getDaughters() const
Returns a vector of pointers to daughter particles.
double getPDGMass(void) const
Returns uncertainty on the invariant mass (requires valid momentum error matrix)
ROOT::Math::PxPyPzEVector get4Vector() const
Returns Lorentz vector.
void addExtraInfo(const std::string &name, double value)
Sets the user-defined data of given name to the given value.
void set4Vector(const ROOT::Math::PxPyPzEVector &p4)
Sets Lorentz vector.
void setPValue(double pValue)
Sets chi^2 probability of fit.
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.
TMatrixFSym getMomentumVertexErrorMatrix() const
Returns 7x7 error matrix.
const Particle * getDaughter(unsigned i) const
Returns a pointer to the i-th daughter particle.
double getExtraInfo(const std::string &name) const
Return given value if set.
int getArrayIndex() const
Returns this object's array index (in StoreArray), or -1 if not found.
TO * getRelatedTo(const std::string &name="", const std::string &namedRelation="") const
Get the object to which this object has a relation.
FourCFitKFit is a derived class from KFitBase to perform 4 momentum-constraint kinematical fit.
enum KFitError::ECode setFourMomentum(const ROOT::Math::PxPyPzEVector &m)
Set an 4 Momentum for the FourC-constraint fit.
double getCHIsq(void) const override
Get a chi-square of the fit.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
enum KFitError::ECode doFit(void)
Perform a four momentum-constraint fit.
const CLHEP::HepSymMatrix getTrackError(const int id) const
Get an error matrix of the track.
const CLHEP::HepLorentzVector getTrackMomentum(const int id) const
Get a Lorentz vector of the track.
const HepPoint3D getTrackPosition(const int id) const
Get a position of the track.
virtual int getNDF(void) const
Get an NDF of the fit.
enum KFitError::ECode setMagneticField(const double mf)
Change a magnetic field from the default value KFitConst::kDefaultMagneticField.
const KFitTrack getTrack(const int id) const
Get a specified track object.
enum KFitError::ECode addParticle(const Particle *particle)
Add a particle to the fitter.
int getTrackCount(void) const
Get the number of added tracks.
ECode
ECode is a error code enumerate.
KFitTrack is a container of the track information (Lorentz vector, position, and error matrix),...
const CLHEP::HepSymMatrix getError(const int flag=KFitConst::kAfterFit) const
Get an error matrix of the track.
const HepPoint3D getPosition(const int flag=KFitConst::kAfterFit) const
Get a position of the track.
MassFitKFit is a derived class from KFitBase to perform mass-constraint kinematical fit.
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set an initial vertex position for the mass-constraint fit.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
enum KFitError::ECode doFit(void)
Perform a mass-constraint fit.
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-constraint fit.
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set an initial vertex error matrix for the mass-constraint fit.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
MassFourCFitKFit is a derived class from KFitBase to perform mass and 4 momentum-constraint kinematic...
enum KFitError::ECode setFourMomentum(const ROOT::Math::PxPyPzEVector &m)
Set an 4 Momentum for the mass-four-constraint fit.
enum KFitError::ECode addMassConstraint(const double m, std::vector< unsigned > &childTrackId)
Set an invariant mass of daughter particle for the mass-four-momentum-constraint fit.
double getCHIsq(void) const override
Get a chi-square of the fit.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
enum KFitError::ECode doFit(void)
Perform a mass-four-momentum-constraint fit.
MassPointingVertexFitKFit is a derived class from KFitBase It performs a kinematical fit with three c...
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
enum KFitError::ECode doFit(void)
Perform a mass-vertex-pointing constraint fit.
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-vertex-pointing constraint fit.
enum KFitError::ECode setProductionVertex(const HepPoint3D &v)
Set the production vertex of the particle.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
MassVertexFitKFit is a derived class from KFitBase to perform mass-vertex-constraint kinematical fit.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
enum KFitError::ECode doFit(void)
Perform a mass-vertex-constraint fit.
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-vertex constraint fit.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
The RaveKinematicVertexFitter class is part of the RaveInterface together with RaveSetup.
void updateMother()
update the mother particle
void addMother(const Particle *aMotherParticlePtr)
All daughters of the argument of this function will be used as input for the vertex fit.
ROOT::Math::XYZVector getPos()
get the position of the fitted vertex.
void addTrack(const Particle *aParticlePtr)
add a track (in the format of a Belle2::Particle) to set of tracks that should be fitted to a vertex
void setMother(const Particle *aMotherParticlePtr)
Set Mother particle for Vertex/momentum update.
int fit()
do the kinematic vertex fit with all tracks previously added with the addTrack or addMother function.
void setVertFit(bool isVertFit=true)
Set vertex fit: set false in case of mass fit only.
void setMassConstFit(bool isConstFit=true)
Set mass constrained fit
double getPValue()
get the p value of the fitted vertex.
TMatrixDSym getVertexErrorMatrix()
get the covariance matrix (3x3) of the of the fitted vertex position.
void updateDaughters()
update the Daughters particles
void unsetBeamSpot()
unset beam spot constraint
static void initialize(int verbosity=1, double MagneticField=1.5)
Set everything up so everything needed for vertex fitting is there.
static RaveSetup * getInstance()
get the pointer to the instance to get/set any of options stored in RaveSetup
void setBeamSpot(const B2Vector3D &beamSpot, const TMatrixDSym &beamSpotCov)
The beam spot position and covariance is known you can set it here so that and a vertex in the beam s...
void reset()
frees memory allocated by initialize().
The RaveVertexFitter class is part of the RaveInterface together with RaveSetup.
TMatrixDSym getCov(VecSize vertexId=0) const
get the covariance matrix (3x3) of the of the fitted vertex position.
int fit(std::string options="default")
do the vertex fit with all tracks previously added with the addTrack or addMother function.
void addTrack(const Particle *const aParticlePtr)
add a track (in the format of a Belle2::Particle) to set of tracks that should be fitted to a vertex
B2Vector3D getPos(VecSize vertexId=0) const
get the position of the fitted vertex.
double getPValue(VecSize vertexId=0) const
get the p value of the fitted vertex.
RecoilMassKFit is a derived class from KFitBase to perform a kinematical fit with a recoil mass const...
enum KFitError::ECode setFourMomentum(const ROOT::Math::PxPyPzEVector &m)
Set a recoil mass .
double getCHIsq(void) const override
Get a chi-square of the fit.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
enum KFitError::ECode setRecoilMass(const double m)
Set an invariant mass for the four momentum-constraint fit.
enum KFitError::ECode doFit(void)
Perform a recoil-mass constraint fit.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
VertexFitKFit is a derived class from KFitBase to perform vertex-constraint kinematical fit.
enum KFitError::ECode setIpTubeProfile(const CLHEP::HepLorentzVector &p, const HepPoint3D &x, const CLHEP::HepSymMatrix &e, const double q)
Set a virtual IP-tube track for the vertex constraint fit.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
const CLHEP::HepSymMatrix getVertexError(void) const
Get a fitted vertex error matrix.
enum KFitError::ECode doFit(void)
Perform a vertex-constraint fit.
enum KFitError::ECode setIpProfile(const HepPoint3D &ip, const CLHEP::HepSymMatrix &ipe)
Set an IP-ellipsoid shape for the vertex constraint fit.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
void addParam(const std::string &name, T ¶mVariable, const std::string &description, const T &defaultValue)
Adds a new parameter to the module.
#define REG_MODULE(moduleName)
Register the given module (without 'Module' suffix) with the framework.
B2Vector3< double > B2Vector3D
typedef for common usage with double
void copyDaughters(Particle *mother)
Function copies all (grand-)^n-daughter particles of the argument mother Particle.
Abstract base class for different kinds of events.
static const int kBeforeFit
Input parameter to specify before-fit when setting/getting a track attribute.