9#include <analysis/modules/TagVertex/TagVertexModule.h>
15#include <framework/gearbox/Unit.h>
16#include <framework/gearbox/Const.h>
17#include <framework/logging/Logger.h>
20#include <analysis/dataobjects/RestOfEvent.h>
21#include <analysis/dataobjects/FlavorTaggerInfo.h>
24#include <analysis/utility/PCmsLabTransform.h>
25#include <analysis/variables/TrackVariables.h>
26#include <analysis/utility/ParticleCopy.h>
27#include <analysis/utility/CLHEPToROOT.h>
28#include <analysis/utility/ROOTToCLHEP.h>
29#include <analysis/utility/DistanceTools.h>
30#include <analysis/utility/RotationTools.h>
33#include <analysis/VertexFitting/RaveInterface/RaveSetup.h>
34#include <analysis/VertexFitting/RaveInterface/RaveVertexFitter.h>
36#include <CLHEP/Geometry/Point3D.h>
37#include <CLHEP/Matrix/SymMatrix.h>
38#include <CLHEP/Vector/LorentzVector.h>
41#include <mdst/dataobjects/HitPatternVXD.h>
44#include <framework/geometry/BFieldManager.h>
48#include <Math/Vector4D.h>
54static const double realNaN = std::numeric_limits<double>::quiet_NaN();
57static const ROOT::Math::XYZVector vecNaN(realNaN, realNaN, realNaN);
60static const double arrayNaN[] = {
61 realNaN, realNaN, realNaN,
62 realNaN, realNaN, realNaN,
63 realNaN, realNaN, realNaN,
67static const TMatrixDSym matNaN(3, arrayNaN);
70using RotationTools::rotateTensor;
71using RotationTools::rotateTensorInv;
72using RotationTools::toSymMatrix;
73using RotationTools::toVec;
74using RotationTools::getUnitOrthogonal;
100 "required confidence level of fit to keep particles in the list. Note that even with confidenceLevel == 0.0, errors during the fit might discard Particles in the list. confidenceLevel = -1 if an error occurs during the fit",
103 "'': no MC association. breco: use standard Breco MC association. internal: use internal MC association",
string(
"breco"));
105 "Choose the type of the constraint: noConstraint, IP (tag tracks constrained to be within the beam spot), tube (long tube along the BTag line of flight, only for fully reconstruced B rec), boost (long tube along the Upsilon(4S) boost direction), (breco)",
108 "Choose how to reconstruct the tracks on the tag side: standard, standard_PXD",
109 string(
"standard_PXD"));
113 "TRUE when requesting MC Information from the tracks performing the vertex fit",
false);
115 "Minimum number of PXD hits for a track to be used in the vertex fit", 0);
117 "Fitter used for the tag vertex fit: Rave or KFit",
string(
"KFit"));
119 "The required chi2/ndf to accept the kFit result, if it is higher, iteration procedure is applied", 5.0);
121 "Use the true track parameters in the vertex fit",
false);
123 "Use rolled back non-primary tracks",
false);
132 B2INFO(
"TagVertexModule : magnetic field = " <<
m_Bfield);
146 B2FATAL(
"TagVertexModule: invalid fitting algorithm (must be set to either Rave or KFit).");
148 B2FATAL(
"TagVertexModule: invalid fitting option (useRollBack and useTruthInFit cannot be simultaneously set to true).");
151 B2FATAL(
"TagVertexModule : the singleTrack option is temporarily broken.");
163 B2ERROR(
"TagVertexModule: ParticleList " <<
m_listName <<
" not found");
170 std::vector<unsigned int> toRemove;
172 for (
unsigned i = 0; i <
m_plist->getListSize(); ++i) {
182 toRemove.push_back(particle->getArrayIndex());
187 particle->addRelationTo(ver);
249 m_plist->removeParticles(toRemove);
276 B2ERROR(
"TagVertex: No magnetic field");
289 double bg = beta /
sqrt(1 - beta * beta);
294 double lB0 = tauB * bg * c;
305 B2ERROR(
"TagVertex: Invalid constraintType selected");
310 B2ERROR(
"TagVertex: No correct fit constraint");
319 double minPVal = (
m_fitAlgo !=
"KFit") ? 0.001 : 0.;
357 if (Breco->
getPValue() < 0.)
return make_pair(vecNaN, matNaN);
359 TMatrixDSym beamSpotCov(3);
369 TMatrixDSym PerrMatrix(7);
371 for (
int i = 0; i < 3; ++i) {
372 for (
int j = 0; j < 3; ++j) {
374 PerrMatrix(i, j) = (beamSpotCov(i, j) + TerrMatrix(i, j)) * pmag / xmag;
376 PerrMatrix(i, j) = TerrMatrix(i, j);
378 PerrMatrix(i + 4, j + 4) = TerrMatrix(i + 4, j + 4);
382 PerrMatrix(3, 3) = 0.;
390 if (BRecoRes->
getPValue() < 0)
return make_pair(vecNaN, matNaN);
398 ROOT::Math::XYZVector BvertDiff = pmag * (Breco->
getVertex() - BRecoRes->
getVertex()).Unit();
399 ROOT::Math::PxPyPzMVector pBrecEstimate(BvertDiff.X(), BvertDiff.Y(), BvertDiff.Z(), Breco->
getPDGMass());
401 pBtagEstimate.SetPxPyPzE(-pBtagEstimate.px(), -pBtagEstimate.py(), -pBtagEstimate.pz(), pBtagEstimate.E());
405 TMatrixD TubeZ = rotateTensorInv(pBrecEstimate.Vect(), errFinal);
407 TubeZ(2, 2) = cut * cut;
408 TubeZ(2, 0) = 0; TubeZ(0, 2) = 0;
409 TubeZ(2, 1) = 0; TubeZ(1, 2) = 0;
413 TMatrixD Tube = rotateTensor(pBtagEstimate.Vect(), TubeZ);
423 B2WARNING(
"In TagVertexModule::findConstraintBTube: cannot get a proper vertex for BReco. BTube constraint replaced by Boost.");
431 if (tubecreatorBCopy->
getPValue() < 0)
return make_pair(vecNaN, matNaN);
435 ROOT::Math::PxPyPzEVector pBrec = tubecreatorBCopy->
get4Vector();
446 pBtag.SetPxPyPzE(-pBtag.px(), -pBtag.py(), -pBtag.pz(), pBtag.E());
458 B2DEBUG(10,
"Brec direction after fit: " <<
printVector(
float(1. / tubecreatorBCopy->
getP()) * tubecreatorBCopy->
getMomentum()));
462 B2DEBUG(10,
"Brec PV covariance: " <<
printMatrix(pv));
463 B2DEBUG(10,
"BTag direction: " <<
printVector((1. / pBtag.P())*pBtag.Vect()));
467 TMatrixD longerror(3, 3); longerror(2, 2) = cut * cut;
471 TMatrixD longerrorRotated = rotateTensor(pBtag.Vect(), longerror);
474 TMatrixD pvNew = TMatrixD(pv) + longerrorRotated;
477 ROOT::Math::XYZVector constraintCenter = tubecreatorBCopy->
getVertex();
488 B2DEBUG(10,
"IPTube covariance: " <<
printMatrix(pvNew));
499 return make_pair(constraintCenter, toSymMatrix(pvNew));
507 TMatrixD longerror(3, 3); longerror(2, 2) = cut * cut;
508 longerror(0, 0) = longerror(1, 1) = d * d;
512 TMatrixD longerrorRotated = rotateTensor(boostDir, longerror);
516 TMatrixD Tube = TMatrixD(beamSpotCov) + longerrorRotated;
521 return make_pair(constraintCenter, toSymMatrix(Tube));
525static double getProperLifeTime(
const MCParticle* mc)
527 double beta = mc->getMomentum().R() / mc->getEnergy();
528 return 1e3 * mc->getLifetime() *
sqrt(1 - pow(beta, 2));
534 vector<const MCParticle*> mcBs;
536 if (abs(mc.getPDG()) == abs(Breco->
getPDGCode()))
540 if (mcBs.size() < 2)
return;
542 if (mcBs.size() > 2) {
543 B2WARNING(
"TagVertexModule:: Too many Bs found in MC");
552 if (!isReco(mcBs[0]) && !isReco(mcBs[1])) {
557 if (!isReco(mcBs[0]) && isReco(mcBs[1]))
558 swap(mcBs[0], mcBs[1]);
561 if (isReco(mcBs[0]) && isReco(mcBs[1])) {
562 double dist0 = (mcBs[0]->getDecayVertex() - Breco->
getVertex()).Mag2();
563 double dist1 = (mcBs[1]->getDecayVertex() - Breco->
getVertex()).Mag2();
565 swap(mcBs[0], mcBs[1]);
570 m_mcTagV = mcBs[1]->getDecayVertex();
579 bool isDecMode =
true;
581 const std::vector<Particle*> recDau = Breco->
getDaughters();
582 const std::vector<MCParticle*> genDau = Bgen->
getDaughters();
584 if (recDau.size() > 0 && genDau.size() > 0) {
585 for (
auto dauRec : recDau) {
587 for (
auto dauGen : genDau) {
588 if (dauGen->getPDG() == dauRec->getPDGCode())
591 if (!isDau) isDecMode =
false;
594 if (recDau.size() == 0) {
596 }
else {isDecMode =
false;}
608 std::vector<const Particle*> fitParticles;
610 if (!roe)
return fitParticles;
613 if (ROEParticles.size() == 0)
return fitParticles;
615 for (
auto& ROEParticle : ROEParticles) {
616 HitPatternVXD roeTrackPattern = ROEParticle->getTrackFitResult()->getHitPatternVXD();
619 fitParticles.push_back(ROEParticle);
628 vector<ParticleAndWeight> particleAndWeights;
630 for (
const Particle* particle : tagParticles) {
631 ROOT::Math::PxPyPzEVector mom = particle->get4Vector();
632 if (!isfinite(mom.mag2()))
continue;
636 particleAndWeight.
weight = -1111.;
637 particleAndWeight.
particle = particle;
642 particleAndWeights.push_back(particleAndWeight);
645 return particleAndWeights;
657 unsigned n = particleAndWeights.size();
658 sort(particleAndWeights.begin(), particleAndWeights.end(),
665 for (
unsigned i = 0; i < n; ++i) {
698 for (
const auto& pw : particleAndWeights) {
713 rFit.
addTrack(pw.particle->getTrackFitResult());
715 }
catch (
const rave::CheckedFloatException&) {
716 B2ERROR(
"Exception caught in TagVertexModule::makeGeneralFitRave(): Invalid inputs (nan/inf)?");
725 isGoodFit = rFit.
fit(
"avf");
727 if (isGoodFit < 1)
return false;
728 }
catch (
const rave::CheckedFloatException&) {
729 B2ERROR(
"Exception caught in TagVertexModule::makeGeneralFitRave(): Invalid inputs (nan/inf)?");
735 for (
unsigned int i(0); i < particleAndWeights.size() && isGoodFit >= 1; ++i)
736 particleAndWeights.at(i).weight = rFit.
getWeight(i);
763 CLHEP::HepSymMatrix err(7, 0);
765 err.sub(5, ROOTToCLHEP::getHepSymMatrix(
m_pvCov));
766 kFit.setIpTubeProfile(
778 for (
auto& pawi : particleAndWeights) {
781 if (pawi.mcParticle) {
782 addedOK = kFit.addTrack(
783 ROOTToCLHEP::getHepLorentzVector(pawi.mcParticle->get4Vector()),
785 ROOTToCLHEP::getHepSymMatrix(pawi.particle->getMomentumVertexErrorMatrix()),
786 pawi.particle->getCharge());
791 if (pawi.mcParticle) {
792 addedOK = kFit.addTrack(
793 ROOTToCLHEP::getHepLorentzVector(pawi.mcParticle->get4Vector()),
795 ROOTToCLHEP::getHepSymMatrix(pawi.particle->getMomentumVertexErrorMatrix()),
796 pawi.particle->getCharge());
801 addedOK = kFit.addParticle(pawi.particle);
807 B2WARNING(
"TagVertexModule::makeGeneralFitKFit: failed to add a track");
813 int nTracksAdded = kFit.getTrackCount();
816 if ((nTracksAdded < 2 &&
m_constraintType ==
"noConstraint") || nTracksAdded < 1)
819 int isGoodFit = kFit.doFit();
828 int largest_chi2_trackid = -1;
829 double largest_track_chi2 = -1;
830 for (
int i = 0; i < kFit.getTrackCount(); ++i) {
831 double track_chi2 = kFit.getTrackCHIsq(i);
832 if (track_chi2 > largest_track_chi2) {
833 largest_track_chi2 = track_chi2;
834 largest_chi2_trackid = i;
837 return largest_chi2_trackid;
852 for (
int iteration_counter = 0; iteration_counter < 100; ++iteration_counter) {
860 if (nTracks !=
int(particleAndWeights.size()))
861 B2ERROR(
"TagVertexModule: Different number of tracks in kFit and particles");
867 int badTrackID = getLargestChi2ID(kFitTemp);
868 if (0 <= badTrackID && badTrackID <
int(particleAndWeights.size()))
869 particleAndWeights.erase(particleAndWeights.begin() + badTrackID);
871 B2ERROR(
"TagVertexModule: Obtained badTrackID is not within limits");
881 fillTagVinfo(CLHEPToROOT::getXYZVector(kFit.getVertex()),
882 CLHEPToROOT::getTMatrixDSym(kFit.getVertexError()));
895 ROOT::Math::XYZVector boostDir = boost.Unit();
896 double bg = boost.R() /
sqrt(1 - boost.Mag2());
901 double dl = dVert.Dot(boostDir);
906 double MCdl = MCdVert.Dot(boostDir);
914 TVectorD bVec = toVec(boostDir);
929 ROOT::Math::XYZVector oboost = getUnitOrthogonal(boostDir);
930 TVectorD oVec = toVec(oboost);
950 int nvert = rsg.
fit(fitType);
965 B2ERROR(
"In TagVertexModule::getTrackWithTrueCoordinate: no MC particle set");
984 B2ERROR(
"In TagVertexModule::getTruePoca: no MC particle set");
985 return ROOT::Math::XYZVector(0., 0., 0.);
1009 B2ERROR(
"In TagVertexModule::getTruePoca: no MC particle set");
1010 return ROOT::Math::XYZVector(0., 0., 0.);
1054 std::ostringstream oss;
1056 oss <<
"(" << std::setw(w) << vec.X() <<
", " << std::setw(w) << vec.Y() <<
", " << std::setw(w) << vec.Z() <<
")" << std::endl;
1063 std::ostringstream oss;
1065 for (
int i = 0; i < mat.GetNrows(); ++i) {
1066 for (
int j = 0; j < mat.GetNcols(); ++j) {
1067 oss << std::setw(w) << mat(i, j) <<
" ";
1077 std::ostringstream oss;
1079 for (
int i = 0; i < mat.GetNrows(); ++i) {
1080 for (
int j = 0; j < mat.GetNcols(); ++j) {
1081 oss << std::setw(w) << mat(i, j) <<
" ";
static ROOT::Math::XYZVector getFieldInTesla(const ROOT::Math::XYZVector &pos)
return the magnetic field at a given position in Tesla.
static const double speedOfLight
[cm/ns]
Hit pattern of the VXD within a track.
unsigned short getNPXDHits() const
Get total number of hits in the PXD.
A Class to store the Monte Carlo particle information.
std::vector< Belle2::MCParticle * > getDaughters() const
Get vector of all daughter particles, empty vector if none.
ROOT::Math::XYZVector getProductionVertex() const
Return production vertex position.
ROOT::Math::PxPyPzEVector get4Vector() const
Return 4Vector of particle.
int getPDG() const
Return PDG code of particle.
ROOT::Math::XYZVector getMomentum() const
Return momentum.
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...
Class to store reconstructed particles.
TMatrixFSym getVertexErrorMatrix() const
Returns the 3x3 position error sub-matrix.
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)
int getPDGCode(void) const
Returns PDG code.
double getPDGMass(void) const
Returns uncertainty on the invariant mass (requires valid momentum error matrix)
ROOT::Math::PxPyPzEVector get4Vector() const
Returns Lorentz vector.
std::vector< Particle * > getDaughters() const
Returns a vector of pointers to daughter particles.
void setMomentumVertexErrorMatrix(const TMatrixFSym &errMatrix)
Sets 7x7 error matrix.
ROOT::Math::XYZVector getMomentum() const
Returns momentum vector.
void setPValue(double pValue)
Sets chi^2 probability of fit.
TMatrixFSym getMomentumVertexErrorMatrix() const
Returns 7x7 error matrix.
const TrackFitResult * getTrackFitResult() const
Returns the pointer to the TrackFitResult that was used to create this Particle (ParticleType == c_Tr...
double getMomentumMagnitude() const
Returns momentum magnitude.
double getP() const
Returns momentum magnitude (same as getMomentumMagnitude but with shorter name)
TO * getRelatedTo(const std::string &name="", const std::string &namedRelation="") const
Get the object to which this object has a relation.
T * getRelated(const std::string &name="", const std::string &namedRelation="") const
Get the object to or from which this object has a relation.
This is a general purpose class for collecting reconstructed MDST data objects that are not used in r...
std::vector< const Particle * > getChargedParticles(const std::string &maskName=c_defaultMaskName, unsigned int pdg=0, bool unpackComposite=true) const
Get charged particles from ROE mask.
static constexpr const char * c_defaultMaskName
Default mask name.
bool isRequired(const std::string &name="")
Ensure this array/object has been registered previously.
Accessor to arrays stored in the data store.
bool registerRelationTo(const StoreArray< TO > &toArray, DataStore::EDurability durability=DataStore::c_Event, DataStore::EStoreFlags storeFlags=DataStore::c_WriteOut, const std::string &namedRelation="") const
Register a relation to the given StoreArray.
int m_fitTruthStatus
Store info about whether the fit was performed with the truth info 0 fit performed with measured para...
TMatrixDSym m_constraintCov
constraint to be used in the tag vertex fit
double m_truthTagVol
MC tagV component in the direction orthogonal to the boost.
std::vector< const Particle * > m_tagParticles
tracks of the rest of the event
double m_tagVl
tagV component in the boost direction
bool doVertexFit(const Particle *Breco)
central method for the tag side vertex fit
std::vector< const Particle * > getTagTracks_standardAlgorithm(const Particle *Breco, int nPXDHits) const
performs the fit using the standard algorithm - using all tracks in RoE The user can specify a reques...
double m_truthTagVl
MC tagV component in the boost direction.
std::pair< ROOT::Math::XYZVector, TMatrixDSym > findConstraintBoost(double cut) const
calculate the standard constraint for the vertex fit on the tag side
bool m_useTruthInFit
Set to true if the tag fit is to be made with the TRUE tag track momentum and position.
std::vector< double > m_raveWeights
Store the weights used by Rave in the vtx fit so that they can be accessed later.
TrackFitResult getTrackWithRollBackCoordinates(ParticleAndWeight const &paw)
If the fit has to be done with the rolled back tracks, Rave or KFit is fed with a track where the pos...
virtual void initialize() override
Initialize the Module.
void fillTagVinfo(const ROOT::Math::XYZVector &tagVpos, const TMatrixDSym &tagVposErr)
Fill tagV vertex info.
ROOT::Math::XYZVector m_mcVertReco
generated Breco decay vertex
TMatrixDSym m_pvCov
covariance matrix of the PV (useful with tube and KFit)
double m_mcDeltaT
generated DeltaT with boost-direction approximation
static std::string printVector(const ROOT::Math::XYZVector &vec)
Print a XYZVector (useful for debugging)
ROOT::Math::XYZVector m_tagV
tag side fit result
virtual void event() override
Event processor.
std::pair< ROOT::Math::XYZVector, TMatrixDSym > findConstraintBTube(const Particle *Breco, double cut)
calculate constraint for the vertex fit on the tag side using the B tube (cylinder along the expected...
Particle * doVertexFitForBTube(const Particle *mother, std::string fitType) const
it returns an intersection between B rec and beam spot (= origin of BTube)
std::string m_listName
Breco particle list name.
std::vector< ParticleAndWeight > getParticlesAndWeights(const std::vector< const Particle * > &tagParticles) const
Get a list of particles with attached weight and associated MC particle.
bool m_useRollBack
Set to true if the tag fit is to be made with the tag track position rolled back to mother B.
double m_tagVlErr
Error of the tagV component in the boost direction.
std::string m_roeMaskName
ROE particles from this mask will be used for vertex fitting.
double m_tagVChi2
chi^2 value of the tag vertex fit result
static ROOT::Math::XYZVector getTruePoca(ParticleAndWeight const &paw)
This finds the point on the true particle trajectory closest to the measured track position.
void BtagMCVertex(const Particle *Breco)
get the vertex of the MC B particle associated to Btag.
std::pair< ROOT::Math::XYZVector, TMatrixDSym > findConstraint(const Particle *Breco, double cut) const
calculate the constraint for the vertex fit on the tag side using Breco information
bool m_mcInfo
true if user wants to retrieve MC information out from the tracks used in the fit
double m_kFitReqReducedChi2
The required chi2/ndf to accept the kFit result, if it is higher, iteration procedure is applied.
std::string m_useMCassociation
No MC association or standard Breco particle or internal MCparticle association.
ROOT::Math::XYZVector m_constraintCenter
centre position of the constraint for the tag Vertex fit
double m_tagVolErr
Error of the tagV component in the direction orthogonal to the boost.
double m_mcTagLifeTime
generated tag side life time of B-decay
ROOT::Math::XYZVector m_BeamSpotCenter
Beam spot position.
double m_tagVNDF
Number of degrees of freedom in the tag vertex fit.
double m_deltaTErr
reconstructed DeltaT error
std::string m_fitAlgo
Algorithm used for the tag fit (Rave or KFit)
bool makeGeneralFit()
TO DO: tag side vertex fit in the case of semileptonic tag side decay.
ROOT::Math::XYZVector getRollBackPoca(ParticleAndWeight const &paw)
This shifts the position of tracks by the vector difference of mother B and production point of track...
virtual void beginRun() override
Called when entering a new run.
double m_mcDeltaTau
generated DeltaT
double m_fitPval
P value of the tag side fit result.
StoreArray< TagVertex > m_verArray
StoreArray of TagVertexes.
DBObjPtr< BeamSpot > m_beamSpotDB
Beam spot database object.
void deltaT(const Particle *Breco)
calculate DeltaT and MC-DeltaT (rec - tag) in ps from Breco and Btag vertices DT = Dl / gamma beta c ...
std::vector< const Particle * > m_raveParticles
tracks given to rave for the track fit (after removing Kshorts
void fillParticles(std::vector< ParticleAndWeight > &particleAndWeights)
Fill sorted list of particles into external variable.
TagVertexModule()
Constructor.
int m_reqPXDHits
N of PXD hits for a track to be used.
double m_confidenceLevel
required fit confidence level
double m_tagVChi2IP
IP component of the chi^2 of the tag vertex fit result.
double m_tagVol
tagV component in the direction orthogonal to the boost
analysis::VertexFitKFit doSingleKfit(std::vector< ParticleAndWeight > &particleAndWeights)
performs single KFit on particles stored in particleAndWeights this function can be iterated several ...
std::string m_constraintType
Choose constraint: noConstraint, IP, tube, boost, (breco)
void resetReturnParams()
Reset all parameters that are computed in each event and then used to compute tuple variables.
ROOT::Math::PxPyPzEVector m_tagMomentum
B tag momentum computed from fully reconstructed B sig.
bool makeGeneralFitRave()
make the vertex fit on the tag side: RAVE AVF tracks coming from Ks removed all other tracks used
TrackFitResult getTrackWithTrueCoordinates(ParticleAndWeight const &paw) const
If the fit has to be done with the truth info, Rave is fed with a track where the momentum is replace...
StoreArray< MCParticle > m_mcParticles
StoreArray of MCParticles.
double m_mcLifeTimeReco
generated Breco life time
static bool compBrecoBgen(const Particle *Breco, const MCParticle *Bgen)
compare Breco with the two MC B particles
double m_Bfield
magnetic field from data base
TMatrixDSym m_tagVErrMatrix
Error matrix of the tag side fit result.
int m_mcPDG
generated tag side B flavor
StoreObjPtr< ParticleList > m_plist
input particle list
static std::string printMatrix(const TMatrixD &mat)
Print a TMatrix (useful for debugging)
ROOT::Math::XYZVector m_mcTagV
generated tag side vertex
int m_rollbackStatus
Store info about whether the fit was performed with the rolled back tracks 0 fit performed with measu...
std::string m_trackFindingType
Choose how to find the tag tracks: standard, standard_PXD.
bool m_verbose
choose if you want to print extra infos
int m_FitType
fit algo used
bool makeGeneralFitKFit()
make the vertex fit on the tag side: KFit tracks coming from Ks removed all other tracks used
std::vector< const MCParticle * > m_raveMCParticles
Store the MC particles corresponding to each track used by Rave in the vtx fit.
TMatrixDSym m_BeamSpotCov
size of the beam spot == covariance matrix on the beam spot position
double m_deltaT
reconstructed DeltaT
TagVertex data object: contains Btag Vertex and DeltaT.
void setConstraintType(const std::string &constraintType)
Set the type of the constraint for the tag fit.
void setTagVlErr(float TagVlErr)
Set the error of the tagV component in the boost direction.
void setTruthTagVl(float TruthTagVl)
Set the MC tagV component in the boost direction.
void setTruthTagVol(float TruthTagVol)
Set the tagV component in the direction orthogonal to the boost.
void setMCTagBFlavor(int mcTagBFlavor)
Set generated Btag PDG code.
void setTagVolErr(float TagVolErr)
Set the error of the tagV component in the direction orthogonal to the boost.
void setTagVNDF(float TagVNDF)
Set the number of degrees of freedom in the tag vertex fit.
void setDeltaTErr(float DeltaTErr)
Set DeltaTErr.
void setConstraintCenter(const ROOT::Math::XYZVector &constraintCenter)
Set the centre of the constraint for the tag fit.
void setNTracks(int nTracks)
Set number of tracks used in the fit.
void setTagVChi2(float TagVChi2)
Set the chi^2 value of the tag vertex fit result.
void setMCDeltaT(float mcDeltaT)
Set generated DeltaT (in kin.
void setRollBackStatus(int backStatus)
Set the status of the fit performed with the rolled back tracks.
void setVertexFitMCParticles(const std::vector< const MCParticle * > &vtxFitMCParticles)
Set a vector of pointers to the MC p'cles corresponding to the tracks in the tag vtx fit.
void setTagVol(float TagVol)
Set the tagV component in the direction orthogonal to the boost.
void setDeltaT(float DeltaT)
Set DeltaT.
void setRaveWeights(const std::vector< double > &raveWeights)
Set the weights used by Rave in the tag vtx fit.
void setTagVertexPval(float TagVertexPval)
Set BTag Vertex P value.
void setTagVertex(const ROOT::Math::XYZVector &TagVertex)
Set BTag Vertex.
void setMCDeltaTau(float mcDeltaTau)
Set generated DeltaT.
void setTagVl(float TagVl)
Set the tagV component in the boost direction.
void setTagVertexErrMatrix(const TMatrixDSym &TagVertexErrMatrix)
Set BTag Vertex (3x3) error matrix.
void setConstraintCov(const TMatrixDSym &constraintCov)
Set the covariance matrix of the constraint for the tag fit.
void setFitType(float FitType)
Set fit algo type.
void setVertexFitParticles(const std::vector< const Particle * > &vtxFitParticles)
Set a vector of pointers to the tracks used in the tag vtx fit.
void setMCTagVertex(const ROOT::Math::XYZVector &mcTagVertex)
Set generated BTag Vertex.
void setTagVChi2IP(float TagVChi2IP)
Set the IP component of the chi^2 value of the tag vertex fit result.
void setFitTruthStatus(int truthStatus)
Set the status of the fit performed with the truth info of the tracks.
Values of the result of a track fit with a given particle hypothesis.
float getNDF() const
Getter for number of degrees of freedom of the track fit.
short getChargeSign() const
Return track charge (1 or -1).
double getPValue() const
Getter for Chi2 Probability of the track fit.
TMatrixDSym getCovariance6() const
Position and Momentum Covariance Matrix.
Const::ParticleType getParticleType() const
Getter for ParticleType of the mass hypothesis of the track fit.
ROOT::Math::XYZVector getMomentum() const
Getter for vector of momentum at closest approach of track in r/phi projection.
ROOT::Math::XYZVector getPosition() const
Getter for vector of position at closest approach of track in r/phi projection.
virtual int getNDF(void) const
Get an NDF of the fit.
bool isFitted(void) const
Return false if fit is not performed yet or performed fit is failed; otherwise true.
int getTrackCount(void) const
Get the number of added tracks.
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 getNdf(VecSize vertexId=0) const
get the number of degrees of freedom (NDF) of the fitted vertex.
double getChi2(VecSize vertexId=0) const
get the χ² of the fitted vertex.
void updateDaughters()
update the Daughters particles
double getWeight(int trackId, VecSize vertexId=0) const
get the weight Rave assigned to a specific input track.
double getPValue(VecSize vertexId=0) const
get the p value of the fitted vertex.
VertexFitKFit is a derived class from KFitBase to perform vertex-constraint kinematical fit.
double getCHIsq(void) const override
Get a chi-square of the fit.
static const double realNaN
This collects the B-meson properties in the hadronic B-decays It is used for the Ecms calibration in ...
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.
double sqrt(double a)
sqrt for double
Particle * copyParticle(const Particle *original)
Function takes argument Particle and creates a copy of it and copies of all its (grand-)^n-daughters.
Abstract base class for different kinds of events.
this struct is used to store and sort the tag tracks
const Particle * particle
tag track fit result with pion mass hypo, for sorting purposes
const MCParticle * mcParticle
mc particle matched to the tag track, for sorting purposes
double weight
rave weight associated to the track, for sorting purposes