10#include <analysis/modules/ParticleVertexFitter/ParticleVertexFitterModule.h>
13#include <framework/gearbox/Unit.h>
14#include <framework/gearbox/Const.h>
15#include <framework/logging/Logger.h>
16#include <framework/particledb/EvtGenDatabasePDG.h>
19#include <analysis/dataobjects/Particle.h>
20#include <analysis/dataobjects/Btube.h>
21#include <mdst/dataobjects/V0.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>
34#include <analysis/VertexFitting/KFit/MassFitKFit.h>
35#include <analysis/VertexFitting/KFit/FourCFitKFit.h>
36#include <analysis/VertexFitting/KFit/MassFourCFitKFit.h>
37#include <analysis/VertexFitting/KFit/MassPointingVertexFitKFit.h>
38#include <analysis/VertexFitting/KFit/MassVertexFitKFit.h>
39#include <analysis/VertexFitting/KFit/VertexFitKFit.h>
40#include <analysis/VertexFitting/KFit/MakeMotherKFit.h>
41#include <analysis/VertexFitting/KFit/RecoilMassKFit.h>
44#include <analysis/VertexFitting/RaveInterface/RaveSetup.h>
45#include <analysis/VertexFitting/RaveInterface/RaveVertexFitter.h>
46#include <analysis/VertexFitting/RaveInterface/RaveKinematicVertexFitter.h>
76 "Confidence level to accept the fit. Particle candidates with "
77 "p-value less than confidenceLevel are removed from the particle "
78 "list. If set to -1, all candidates are kept; if set to 0, "
79 "the candidates failing the fit are removed.",
82 addParam(
"fitType",
m_fitType,
"type of the kinematic fit (vertex, massvertex, mass)",
string(
"vertex"));
84 "additional constraint on vertex: ipprofile, iptube, mother, iptubecut, pointing, btube",
86 addParam(
"decayString",
m_decayString,
"specifies which daughter particles are included in the kinematic fit",
string(
""));
91 "Type::[int]. List of daughter particles to mass constrain with int = pdg code. (only for MassFourCKFit)", {});
106 B2DEBUG(1,
"ParticleVertexFitterModule : magnetic field = " <<
m_Bfield);
114 B2INFO(
"ParticleVertexFitter: Using specified decay string: " <<
m_decayString);
116 B2INFO(
"ParticleVertexFitter: Additional " <<
m_withConstraint <<
" will be applied");
150 std::vector<unsigned int> toRemove;
151 unsigned int nParticles =
m_plist->getListSize();
153 for (
unsigned iPart = 0; iPart < nParticles; iPart++) {
159 else B2ERROR(
"Daughters update works only when all daughters are selected. Daughters will not be updated");
167 TMatrixFSym mother_errMatrix(7);
168 mother_errMatrix = particle->getMomentumVertexErrorMatrix();
169 for (
int k = 0; k < 7; k++) {
170 for (
int j = 0; j < 7; j++) {
171 if (mother_errMatrix[k][j] > 0) {
182 toRemove.push_back(particle->getArrayIndex());
193 particle->setPValue(-1);
195 toRemove.push_back(particle->getArrayIndex());
198 m_plist->removeParticles(toRemove);
210 B2FATAL(
"ParticleVertexFitter: No magnetic field");
220 B2FATAL(
"ParticleVertexFitter: " <<
m_withConstraint <<
" ***invalid Constraint ");
227 B2FATAL(
"ParticleVertexFitter: KFit does not support yet selection of daughters via decay string except for vertex fit!");
243 B2FATAL(
"ParticleVertexFitter: Invalid options - mass-constrained fit using KFit does not work with iptube or ipprofile constraint.");
254 B2FATAL(
"ParticleVertexFitter: Invalid options - mass fit using KFit does not work with iptube or ipprofile constraint.");
263 B2FATAL(
"ParticleVertexFitter: Invalid options - four C fit using KFit does not work with iptube or ipprofile constraint.");
272 B2FATAL(
"ParticleVertexFitter: Invalid options - four C fit using KFit does not work with iptube or ipprofile constraint.");
281 B2FATAL(
"ParticleVertexFitter: Invalid options - recoil mass fit using KFit does not work with iptube or ipprofile constraint.");
294 B2FATAL(
"ParticleVertexFitter: " <<
m_fitType <<
" ***invalid fit type for the vertex fitter ");
301 }
catch (
const rave::CheckedFloatException&) {
302 B2ERROR(
"Invalid inputs (nan/inf)?");
309 B2FATAL(
"ParticleVertexFitter: " <<
m_vertexFitter <<
" ***invalid vertex fitter ");
311 if (!ok)
return false;
321 std::vector<const Particle*>& twoPhotonChildren)
325 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
332 fitChildren.push_back(child);
338 auto itr = fitChildren.begin();
339 while (itr != fitChildren.end()) {
342 if (child->getPValue() < 0) {
343 B2WARNING(
"Daughter with PDG code " << child->getPDGCode() <<
" does not have a valid error matrix.");
346 bool isTwoPhotonParticle =
false;
348 if (child->getPDGCode() ==
Const::pi0.getPDGCode() or child->getPDGCode() == 221) {
349 if (child->getNDaughters() == 2) {
350 if (child->getDaughter(0)->getPDGCode() ==
Const::photon.getPDGCode()
351 && child->getDaughter(1)->getPDGCode() ==
Const::photon.getPDGCode()) {
352 isTwoPhotonParticle =
true;
357 if (isTwoPhotonParticle) {
359 twoPhotonChildren.push_back(child);
360 itr = fitChildren.erase(itr);
370 const std::vector<const Particle*>& fitChildren)
372 if (fitChildren.empty())
373 B2WARNING(
"[ParticleVertexFitterModule::fillNotFitParticles] fitChildren is empty! Please call fillFitParticles firstly");
374 if (!notFitChildren.empty())
375 B2WARNING(
"[ParticleVertexFitterModule::fillNotFitParticles] notFitChildren is NOT empty!"
376 <<
" The function should be called only once");
382 std::function<bool(
const Particle*)> funcCheckInFit =
383 [&funcCheckInFit, ¬FitChildren, fitChildren](
const Particle * part) {
387 if (std::find(fitChildren.begin(), fitChildren.end(), part) != fitChildren.end())
391 if (part->getNDaughters() == 0)
396 bool isAnyChildrenInFit =
false;
397 vector<const Particle*> notFitChildren_tmp;
398 for (
unsigned ichild = 0; ichild < part->getNDaughters(); ichild++) {
400 const Particle* child = part->getDaughter(ichild);
401 bool isChildrenInFit = funcCheckInFit(child);
402 isAnyChildrenInFit = isChildrenInFit or isAnyChildrenInFit;
405 if (!isChildrenInFit)
406 notFitChildren_tmp.push_back(child);
410 if (isAnyChildrenInFit)
411 notFitChildren.insert(notFitChildren.end(), notFitChildren_tmp.begin(), notFitChildren_tmp.end());
415 return isAnyChildrenInFit;
420 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
422 bool isGivenParticleOrAnyChildrenInFit = funcCheckInFit(child);
423 if (!isGivenParticleOrAnyChildrenInFit)
424 notFitChildren.push_back(child);
447 TMatrixFSym errMatrix(3);
448 for (
int i = 0; i < 3; i++)
449 for (
int j = 0; j < 3; j++)
450 errMatrix(i, j) = posErrorMatrix[i][j];
452 g1ErrMatrix.SetSub(4, errMatrix);
453 g2ErrMatrix.SetSub(4, errMatrix);
469 int err = km.
doFit();
487 std::vector<const Particle*> fitChildren;
488 std::vector<const Particle*> twoPhotonChildren;
489 bool validChildren =
fillFitParticles(mother, fitChildren, twoPhotonChildren);
494 std::vector<const Particle*> notFitChildren;
498 if (twoPhotonChildren.size() > 1) {
499 B2FATAL(
"[ParticleVertexFitterModule::doKVertexFit] Vertex fit using KFit does not support fit with multiple particles decaying to two photons like pi0 (yet).");
502 if ((fitChildren.size() < 2 && !ipTubeConstraint) || fitChildren.size() < 1) {
503 B2WARNING(
"[ParticleVertexFitterModule::doKVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is too low.");
511 if (mother->
getV0()) {
515 for (
auto& child : fitChildren)
518 if (ipProfileConstraint)
521 if (ipTubeConstraint)
525 int err = kv.
doFit();
535 if (twoPhotonChildren.size() == 0)
538 else if (twoPhotonChildren.size() == 1) {
544 const Particle* twoPhotonDaughter = twoPhotonChildren[0];
554 for (
auto& child : fitChildren)
559 if (ipProfileConstraint)
571 ROOT::Math::PxPyPzEVector total4Vector(mother->
get4Vector());
572 for (
auto& child : notFitChildren)
573 total4Vector += child->get4Vector();
583 std::vector<const Particle*> fitChildren;
584 std::vector<const Particle*> twoPhotonChildren;
585 bool validChildren =
fillFitParticles(mother, fitChildren, twoPhotonChildren);
590 if (twoPhotonChildren.size() > 1) {
591 B2FATAL(
"[ParticleVertexFitterModule::doKVertexFit] MassVertex fit using KFit does not support fit with multiple particles decaying to two photons like pi0 (yet).");
594 if (fitChildren.size() < 2) {
595 B2WARNING(
"[ParticleVertexFitterModule::doKVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is less than 2.");
600 if (twoPhotonChildren.size() == 0) {
605 if (mother->
getV0()) {
612 for (
auto child : fitChildren)
616 int err = kmv.
doFit();
622 }
else if (twoPhotonChildren.size() == 1) {
631 for (
auto child : fitChildren)
635 int err = kv.
doFit();
639 const Particle* twoPhotonDaughter = twoPhotonChildren[0];
649 for (
auto child : fitChildren)
674 std::vector<const Particle*> fitChildren;
675 std::vector<const Particle*> twoPhotonChildren;
676 bool validChildren =
fillFitParticles(mother, fitChildren, twoPhotonChildren);
681 if (twoPhotonChildren.size() > 0) {
682 B2FATAL(
"[ParticleVertexFitterModule::doKMassPointingVertexFit] MassPointingVertex fit using KFit does not support fit with two-photon daughters (yet).");
685 if (fitChildren.size() < 2) {
686 B2WARNING(
"[ParticleVertexFitterModule::doKMassPointingVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is less than 2.");
695 for (
auto child : fitChildren)
699 HepPoint3D productionVertex(mother->
getExtraInfo(
"prodVertX"),
703 int err = kmpv.
doFit();
704 if (err != 0)
return false;
718 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
721 if (child->getPValue() < 0)
return false;
729 int err = km.
doFit();
731 if (err != 0)
return false;
745 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
748 if (child->getNDaughters() > 0) {
750 if (!err)
return false;
752 if (child->getPValue() < 0)
return false;
762 int err = kf.
doFit();
764 if (err != 0)
return false;
778 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
781 if (child->getNDaughters() > 0) {
784 std::vector<unsigned> childId;
787 if (!err)
return false;
789 if (child->getPValue() < 0)
return false;
798 int err = kf.
doFit();
800 if (err != 0)
return false;
812 for (
unsigned ichild = 0; ichild < mother->
getNDaughters(); ichild++) {
815 if (child->getPValue() < 0)
return false;
827 int err = kf.
doFit();
829 if (err != 0)
return false;
847 std::vector<Particle*> daughters = mother->
getDaughters();
850 if (daughters.size() != track_count)
853 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
858 ROOT::Math::PxPyPzEVector i4Vector(kv.
getTrackMomentum(iChild).x() - a * dy,
862 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
864 daughters[iChild]->setVertex(
866 daughters[iChild]->setMomentumVertexErrorMatrix(
872 std::vector<const Particle*> fitChildren =
m_decaydescriptor.getSelectionParticles(mother);
875 if (fitChildren.size() != track_count)
878 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
879 auto daughter =
const_cast<Particle*
>(fitChildren[iChild]);
885 ROOT::Math::PxPyPzEVector i4Vector(kv.
getTrackMomentum(iChild).x() - a * dy,
889 daughter->set4VectorDividingByMomentumScaling(i4Vector);
891 daughter->setVertex(CLHEPToROOT::getXYZVector(kv.
getTrackPosition(iChild)));
892 daughter->setMomentumVertexErrorMatrix(CLHEPToROOT::getTMatrixFSym(kv.
getTrackError(iChild)));
896 std::function<bool(
Particle*)> funcUpdateMomentum =
897 [&funcUpdateMomentum, fitChildren](
Particle * part) {
899 if (part->getNDaughters() == 0) {
901 if (std::find(fitChildren.begin(), fitChildren.end(), part) != fitChildren.end())
907 bool includeFitChildren =
false;
910 for (
auto daughter : part->getDaughters())
911 includeFitChildren = funcUpdateMomentum(daughter) || includeFitChildren;
913 if (includeFitChildren) {
915 ROOT::Math::PxPyPzEVector sum4Vector;
916 for (
auto daughter : part->getDaughters())
917 sum4Vector += daughter->get4Vector();
919 part->set4VectorDividingByMomentumScaling(sum4Vector);
922 return includeFitChildren;
927 funcUpdateMomentum(daughter);
946 std::vector<Particle*> daughters = mother->
getDaughters();
949 if (daughters.size() != track_count)
952 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
957 ROOT::Math::PxPyPzEVector i4Vector(kmv.
getTrackMomentum(iChild).x() - a * dy,
961 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
963 daughters[iChild]->setVertex(
965 daughters[iChild]->setMomentumVertexErrorMatrix(
986 std::vector<Particle*> daughters = mother->
getDaughters();
989 if (daughters.size() != track_count)
992 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
997 ROOT::Math::PxPyPzEVector i4Vector(kmpv.
getTrackMomentum(iChild).x() - a * dy,
1001 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
1003 daughters[iChild]->setVertex(
1005 daughters[iChild]->setMomentumVertexErrorMatrix(
1025 std::vector<Particle*> daughters = mother->
getDaughters();
1028 if (daughters.size() != track_count)
1031 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
1036 ROOT::Math::PxPyPzEVector i4Vector(km.
getTrackMomentum(iChild).x() - a * dy,
1040 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
1042 daughters[iChild]->setVertex(
1044 daughters[iChild]->setMomentumVertexErrorMatrix(
1066 std::vector<Particle*> daughters = mother->
getDaughters();
1068 const unsigned nd = daughters.size();
1070 std::vector<std::vector<unsigned>> pars;
1071 std::vector<Particle*> allparticles;
1072 for (
unsigned ichild = 0; ichild < nd; ichild++) {
1074 std::vector<unsigned> pard;
1075 if (daughter->getNDaughters() > 0) {
1077 pars.push_back(pard);
1078 allparticles.push_back(daughters[ichild]);
1081 pars.push_back(pard);
1082 allparticles.push_back(daughters[ichild]);
1088 if (l != track_count)
1091 for (
unsigned iDaug = 0; iDaug < allparticles.size(); iDaug++) {
1092 ROOT::Math::PxPyPzEVector childMoms;
1093 ROOT::Math::XYZVector childPoss;
1094 TMatrixFSym childErrMatrixs(7);
1095 for (
unsigned int iChild : pars[iDaug]) {
1096 childMoms = childMoms +
1097 CLHEPToROOT::getLorentzVector(
1099 childPoss = childPoss +
1100 CLHEPToROOT::getXYZVector(
1102 TMatrixFSym childErrMatrix =
1104 childErrMatrixs = childErrMatrixs + childErrMatrix;
1106 allparticles[iDaug]->set4Vector(childMoms);
1107 allparticles[iDaug]->setVertex(childPoss);
1108 allparticles[iDaug]->setMomentumVertexErrorMatrix(childErrMatrixs);
1128 std::vector<Particle*> daughters = mother->
getDaughters();
1130 const unsigned nd = daughters.size();
1132 std::vector<std::vector<unsigned>> pars;
1133 std::vector<Particle*> allparticles;
1134 for (
unsigned ichild = 0; ichild < nd; ichild++) {
1136 std::vector<unsigned> pard;
1137 if (daughter->getNDaughters() > 0) {
1139 pars.push_back(pard);
1140 allparticles.push_back(daughters[ichild]);
1143 pars.push_back(pard);
1144 allparticles.push_back(daughters[ichild]);
1150 if (l != track_count)
1153 for (
unsigned iDaug = 0; iDaug < allparticles.size(); iDaug++) {
1154 ROOT::Math::PxPyPzEVector childMoms;
1155 ROOT::Math::XYZVector childPoss;
1156 TMatrixFSym childErrMatrixs(7);
1157 for (
unsigned int iChild : pars[iDaug]) {
1158 childMoms = childMoms +
1159 CLHEPToROOT::getLorentzVector(
1161 childPoss = childPoss +
1162 CLHEPToROOT::getXYZVector(
1164 TMatrixFSym childErrMatrix =
1166 childErrMatrixs = childErrMatrixs + childErrMatrix;
1168 allparticles[iDaug]->set4Vector(childMoms);
1169 allparticles[iDaug]->setVertex(childPoss);
1170 allparticles[iDaug]->setMomentumVertexErrorMatrix(childErrMatrixs);
1190 std::vector<Particle*> daughters = mother->
getDaughters();
1193 if (daughters.size() != track_count)
1196 for (
unsigned iChild = 0; iChild < track_count; iChild++) {
1201 ROOT::Math::PxPyPzEVector i4Vector(kf.
getTrackMomentum(iChild).x() - a * dy,
1205 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
1207 daughters[iChild]->setVertex(
1209 daughters[iChild]->setMomentumVertexErrorMatrix(
1219 std::vector<unsigned>& parm, std::vector<Particle*>& allparticles,
const Particle* daughter)
1221 std::vector <Particle*> daughters = daughter->getDaughters();
1222 for (
unsigned ichild = 0; ichild < daughter->getNDaughters(); ichild++) {
1223 const Particle* child = daughter->getDaughter(ichild);
1224 std::vector<unsigned> pard;
1225 if (child->getNDaughters() > 0) {
1227 parm.insert(parm.end(), pard.begin(), pard.end());
1228 pars.push_back(pard);
1229 allparticles.push_back(daughters[ichild]);
1233 pars.push_back(pard);
1234 allparticles.push_back(daughters[ichild]);
1256 std::vector<const Particle*> tracksVertex =
m_decaydescriptor.getSelectionParticles(mother);
1260 for (
auto& itrack : tracksVertex) {
1261 if (itrack != mother) rf.
addTrack(itrack);
1267 bool mothSel =
false;
1269 for (
unsigned itrack = 0; itrack < tracksVertex.size(); itrack++) {
1270 if (tracksVertex[itrack] != mother) {
1271 rsf.
addTrack(tracksVertex[itrack]);
1272 B2DEBUG(1,
"ParticleVertexFitterModule: Adding particle " << tracksName[itrack] <<
" to vertex fit ");
1275 if (tracksVertex[itrack] == mother) mothSel =
true;
1280 bool mothIPfit =
false;
1281 if (tracksVertex.size() == 1 && mothSel ==
true &&
m_withConstraint !=
"" && nTrk == 0) {
1283 if (tracksVertex[0] != mother)
1284 B2FATAL(
"ParticleVertexFitterModule: FATAL Error in IP constrained mother fit");
1290 ROOT::Math::XYZVector pos;
1291 TMatrixDSym RerrMatrix(3);
1297 for (
auto& itrack : tracksVertex) {
1299 nvert = rsg.
fit(
"kalman");
1302 RerrMatrix = rsg.
getCov(0);
1304 ROOT::Math::PxPyPzEVector mom(mother->
get4Vector());
1305 TMatrixDSym errMatrix(7);
1306 for (
int i = 0; i < 7; i++) {
1307 for (
int j = 0; j < 7; j++) {
1308 if (i > 3 && j > 3) {errMatrix[i][j] = RerrMatrix[i - 4][j - 4];}
1309 else {errMatrix[i][j] = 0;}
1329 }
else {
return false;}
1339 ROOT::Math::PxPyPzEVector mom(mother->
get4Vector());
1340 TMatrixDSym errMatrix(7);
1341 for (
int i = 0; i < 7; i++) {
1342 for (
int j = 0; j < 7; j++) {
1343 if (i > 3 && j > 3) {errMatrix[i][j] = RerrMatrix[i - 4][j - 4];}
1344 else {errMatrix[i][j] = 0;}
1348 }
else {
return false;}
1351 if (mothSel && nTrk > 1) {
1354 int nKfit = rf.
fit();
1358 if (nKfit > 0) {
return true;}
1367 int nVert = rf.
fit();
1370 if (nVert != 1)
return false;
1377 int nVert = rf.
fit();
1379 if (nVert != 1)
return false;
1384 int nVert = rf.
fit();
1387 if (nVert != 1)
return false;
1390 B2FATAL(
"fitType : " <<
m_fitType <<
" ***invalid fit type ");
1397 const std::vector<const Particle*>& tracksVertex)
1406 for (
auto& vi : tracksVertex) {
1408 nNotIncluded = nNotIncluded - 1;
1416 if (nNotIncluded == 0) isAll =
true;
1422 for (
unsigned ichild = 0; ichild < particle->getNDaughters(); ichild++) {
1423 const Particle* child = particle->getDaughter(ichild);
1426 if (child->getPValue() < 0)
return false;
1435 std::vector<unsigned>& particleId)
1437 for (
unsigned ichild = 0; ichild < particle->getNDaughters(); ichild++) {
1438 const Particle* child = particle->getDaughter(ichild);
1439 if (child->getNDaughters() > 0) {
1442 std::vector<unsigned> childId;
1445 particleId.insert(particleId.end(), childId.begin(), childId.end());
1447 if (child->getPValue() < 0)
return false;
1458 CLHEP::HepSymMatrix covMatrix = ROOTToCLHEP::getHepSymMatrix(
m_beamSpotCov);
1465 CLHEP::HepSymMatrix err(7, 0);
1467 for (
int i = 0; i < 3; i++) {
1468 for (
int j = 0; j < 3; j++) {
1477 ROOTToCLHEP::getHepLorentzVector(iptube_mom),
1489 TMatrixDSym beamSpotCov =
m_beamSpotDB->getCovVertex();
1490 beamSpotCov(2, 2) = cut * cut;
1491 double thetab = boostDir.Theta();
1492 double phib = boostDir.Phi();
1494 double stb = TMath::Sin(thetab);
1495 double ctb = TMath::Cos(thetab);
1496 double spb = TMath::Sin(phib);
1497 double cpb = TMath::Cos(phib);
1500 TMatrix rz(3, 3); rz(2, 2) = 1;
1501 rz(0, 0) = cpb; rz(0, 1) = spb;
1502 rz(1, 0) = -1 * spb; rz(1, 1) = cpb;
1504 TMatrix ry(3, 3); ry(1, 1) = 1;
1505 ry(0, 0) = ctb; ry(0, 2) = -1 * stb;
1506 ry(2, 0) = stb; ry(2, 2) = ctb;
1508 TMatrix r(3, 3); r.Mult(rz, ry);
1509 TMatrix rt(3, 3); rt.Transpose(r);
1511 TMatrix TubePart(3, 3); TubePart.Mult(rt, beamSpotCov);
1512 TMatrix Tube(3, 3); Tube.Mult(TubePart, r);
1521 TMatrixDSym beamSpotCov =
m_beamSpotDB->getCovVertex();
1522 for (
int i = 0; i < 3; i++)
1523 beamSpotCov(i, i) += width * width;
1530 double chi2TrackL = 0;
1532 for (
int iTrack = 0; iTrack < kv.
getTrackCount(); iTrack++) {
1539 ROOT::Math::XYZVector x_after = CLHEPToROOT::getXYZVector(trk_i.
getPosition());
1540 ROOT::Math::XYZVector dPos = x_after - x_before;
1544 TVectorD boostD(0, 6, 0., 0., 0., 0., boost3.X(), boost3.Y(), boost3.Z(),
"END");
1546 double dLBoost = dPos.Dot(boost3);
1548 chi2TrackL += TMath::Power(dLBoost, 2) / err.Similarity(boostD);
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.
static const ParticleType pi0
neutral pion particle
static const double speedOfLight
[cm/ns]
static const ParticleType photon
photon particle
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.
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.
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
ParticleVertexFitterModule()
Constructor.
std::string m_decayString
daughter particles selection
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.
ROOT::Math::XYZVector m_BeamSpotCenter
Beam spot position.
virtual void beginRun() override
Called when entering a new run.
bool doKRecoilMassFit(Particle *p)
RecoilMass fit using KFit.
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.
void writeExtraInfo(const std::string &name, const double value)
Sets the user defined extraInfo.
bool hasExtraInfo(const std::string &name) const
Return whether the extra info with the given name is set.
const V0 * getV0() const
Returns the pointer to the V0 object that was used to create this Particle (if ParticleType == c_V0).
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.
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.
std::vector< Particle * > getDaughters() const
Returns a vector of pointers to daughter particles.
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.
@ c_IsUnspecified
Ordinary particles.
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.
double getFittedVertexX() const
Get the X coordinate of the fitted vertex.
double getFittedVertexY() const
Get the Y coordinate of the fitted vertex.
double getFittedVertexZ() const
Get the Z coordinate of the fitted vertex.
ROOT::Math::XYZVector getFittedVertexPosition() const
Get the fitted vertex position.
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 setInitialVertex(const HepPoint3D &v)
Set an initial vertex point for the mass-vertex constraint 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 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 ROOT::Math::XYZVector &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.
ROOT::Math::XYZVector getPos(VecSize vertexId=0) const
get the position of the fitted vertex.
void addTrack(const Particle *const aParticlePtr)
add a track (in the format of a Particle) to set of tracks that should be fitted to a 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 setInitialVertex(const HepPoint3D &v)
Set an initial vertex point for the vertex-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.
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.