RaveKinematicVertexFitter Class Reference

The RaveKinematicVertexFitter class is part of the RaveInterface together with RaveSetup. More...

#include <RaveKinematicVertexFitter.h>

Collaboration diagram for RaveKinematicVertexFitter:

Public Member Functions
	RaveKinematicVertexFitter ()
	The default constructor checks if RaveSetup was initialized and will set the attributes of RaveKinematicVertexFitter.
	~RaveKinematicVertexFitter ()
	Destructor.
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	addMother (const Particle *aMotherParticlePtr)
	All daughters of the argument of this function will be used as input for the vertex fit.
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.
ROOT::Math::XYZVector	getPos ()
	get the position of the fitted vertex.
double	getPValue ()
	get the p value of the fitted vertex.
double	getNdf ()
	get the number of degrees of freedom (NDF) of the fitted vertex.
double	getChi2 ()
	get the χ² of the fitted vertex.
TMatrixDSym	getCov ()
	get the covariance matrix (7x7).
TMatrixDSym	getVertexErrorMatrix ()
	get the covariance matrix (3x3) of the of the fitted vertex position.
void	useBeamSpot (bool beamSpot=true)
	Overwrite the global option in ReveSetup that tells the fitter if beam spot info should be used or not.
Particle *	getMother ()
	returns a pointer to the mother particle
void	updateMother ()
	update the mother particle
void	updateDaughters ()
	update the Daughters particles
void	clearTracks ()
	Delete all information of previously added tracks and fitted results.
void	setMassConstFit (bool isConstFit=true)
	Set mass constrained fit
void	setVertFit (bool isVertFit=true)
	Set vertex fit: set false in case of mass fit only.
void	Print () const
	Print all attributes of this object to terminal.

Protected Attributes
bool	m_useBeamSpot
	flag determines if the beam spot will be used or not.
Particle *	m_motherParticlePtr
	pointer to the mother particle who's daughters will be used in the fit.
std::string	m_raveAlgorithm
	Algorithm used by rave (kalman, avr, ...)
rave::KinematicTree	m_fittedResult
	the output of the kinematic fit
std::vector< rave::Track >	m_raveTracks
	holds the tracks that were added to a RaveVertexFitter object in the format used by Rave
std::vector< rave::Vertex >	m_raveVertices
	holds the fitted vertices after fit() was called in the format used by Rave
std::vector< rave::KinematicParticle >	m_inputParticles
	input particles for vertex fit in rave format
rave::KinematicParticle	m_fittedParticle
	Particle fit output.
std::vector< Particle * >	m_belleDaughters
	Belle Particle pointers input.
bool	m_massConstFit
	flag determines if the mass fit is performed
bool	m_vertFit
	flag determines if the vertex fit is performed
double	m_fittedNdf
	Ndf of the vertex fit.
double	m_fittedPValue
	Pvalue of the fit result.
double	m_fittedChi2
	chi^2 of the vertex fit
ROOT::Math::XYZVector	m_fittedPos
	Fitted vertex position.
ROOT::Math::PxPyPzEVector	m_fitted4Vector
	4 momentum of the mother particle after the fit
TMatrixFSym	m_fitted7Cov
	7x7 error matrix of the mother particle after the fit

Private Member Functions
TMatrixDSym	ErrorMatrixMassToEnergy (const ROOT::Math::PxPyPzEVector &p4, const TMatrixDSym &MassErr)
	Convert the error matrix from P-M to P-E.
TMatrixDSym	ErrorMatrixEnergyToMass (const ROOT::Math::PxPyPzEVector &p4, const TMatrixDSym &EnergyErr)
	Convert the error matrix from P-E to P-M.
IOIntercept::InterceptorScopeGuard< IOIntercept::OutputToLogMessages >	captureOutput ()
	Start capturing the output of rave and divert it to log messages.

Detailed Description

The RaveKinematicVertexFitter class is part of the RaveInterface together with RaveSetup.

To fit a vertex the user adds tracks (different formats are supported). Then calls the fit() function where the name of the fitting method and parameters can be passed as a string. RaveKinematicVertexFitter provides the getters to extract the results produced by Rave.

Definition at line 44 of file RaveKinematicVertexFitter.h.

Constructor & Destructor Documentation

RaveKinematicVertexFitter()

RaveKinematicVertexFitter

(

)

The default constructor checks if RaveSetup was initialized and will set the attributes of RaveKinematicVertexFitter.

Definition at line 37 of file RaveKinematicVertexFitter.cc.

                                                    : m_useBeamSpot(false), m_motherParticlePtr(nullptr), m_raveAlgorithm(""),
  m_massConstFit(false), m_vertFit(true)
{
  if (RaveSetup::getRawInstance() == nullptr) {
    B2FATAL("RaveSetup::initialize was not called. It has to be called before RaveSetup or RaveKinematicVertexFitter are used");
  }
  m_useBeamSpot = RaveSetup::getRawInstance()->m_useBeamSpot;
}

Member Function Documentation

addMother()

void addMother

(

const Particle *

aMotherParticlePtr

)

All daughters of the argument of this function will be used as input for the vertex fit.

Writes back the result directly to the mother particle

Definition at line 105 of file RaveKinematicVertexFitter.cc.

{
  vector<Particle*> daughters = aMotherParticlePtr->getDaughters();
 
  int nDaughters = daughters.size();
  for (int i = 0; i not_eq nDaughters; ++i) {
    addTrack(daughters[i]);
  }
 
  //store input pointer so fit results can be written to the mother particle after the fit
  auto* tmp = const_cast<Particle*>(aMotherParticlePtr);
  m_motherParticlePtr = tmp;
 
}

addTrack()

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

Definition at line 68 of file RaveKinematicVertexFitter.cc.

{
  rave::Vector7D raveState(aParticlePtr->getX(), aParticlePtr->getY(), aParticlePtr->getZ(), aParticlePtr->getPx(),
                           aParticlePtr->getPy(), aParticlePtr->getPz(), aParticlePtr->getMass());
  TMatrixDSym covP = aParticlePtr->getMomentumVertexErrorMatrix();
 
  TMatrixDSym covE(7);
  for (int i = 0; i < 7; i++) {
    for (int j = 0; j < 7; j++) {
      if (i < 3 && j < 3) covE(i, j) = covP(i + 4, j + 4);
      if (i > 2 && j > 2) covE(i, j) = covP(i - 3, j - 3);
      if (i < 3 && j > 2) covE(i, j) = covP(i + 4, j - 3);
      if (i > 2 && j < 3) covE(i, j) = covP(i - 3, j + 4);
    }
  }
 
  TMatrixDSym cov = ErrorMatrixEnergyToMass(aParticlePtr->get4Vector(), covE);
 
  rave::Covariance7D raveCov(cov(0, 0), cov(0, 1), cov(0, 2), // x x , x y, x z
                             cov(1, 1), cov(1, 2), cov(2, 2), // y y , y z, z z
                             cov(0, 3), cov(0, 4), cov(0, 5), // x px , x py, x pz
                             cov(1, 3), cov(1, 4), cov(1, 5), // y px , y py, y pz
                             cov(2, 3), cov(2, 4), cov(2, 5), // z px , z py, z pz
                             cov(3, 3), cov(3, 4), cov(3, 5), // px px , px py, px pz
                             cov(4, 4), cov(4, 5), cov(5, 5), // py py , py pz, pz pz
                             cov(0, 6), cov(1, 6), cov(2, 6), // x m , y m, z m
                             cov(3, 6), cov(4, 6), cov(5, 6), // px m, py m, pz m
                             cov(6, 6)); // mm
 
  rave::TransientTrackKinematicParticle aRaveParticle(raveState, raveCov, rave::Charge(aParticlePtr->getCharge()), 1, 1);
  // 1 and 1 are dummy values for chi2 and ndf. the are not used for the vertex fit
 
  m_inputParticles.push_back(aRaveParticle);
  m_belleDaughters.push_back(const_cast<Particle*>(aParticlePtr));
 
}

captureOutput()

IOIntercept::InterceptorScopeGuard< IOIntercept::OutputToLogMessages > captureOutput ( )

private

Start capturing the output of rave and divert it to log messages.

Capturing will finish as soon as the returned object goes out of scope

Definition at line 46 of file RaveKinematicVertexFitter.cc.

{
  static IOIntercept::OutputToLogMessages s_captureOutput("Rave", LogConfig::c_Debug, LogConfig::c_Debug);
  return IOIntercept::start_intercept(s_captureOutput);
}

clearTracks()

void clearTracks ( )

inline

Delete all information of previously added tracks and fitted results.

Definition at line 107 of file RaveKinematicVertexFitter.h.

      {
        m_inputParticles.clear();
        m_motherParticlePtr = nullptr;
      }

ErrorMatrixEnergyToMass()

TMatrixDSym ErrorMatrixEnergyToMass ( const ROOT::Math::PxPyPzEVector &  p4, const TMatrixDSym &  EnergyErr  )

private

Convert the error matrix from P-E to P-M.

It Requires an input error matrix in the form X,P,E

Definition at line 483 of file RaveKinematicVertexFitter.cc.

{
 
  TMatrix jac(7, 7);
  for (int i = 0; i < 7; i++) {
    for (int j = 0; j < 7; j++) {
      if (i == j) jac(i, j) = 1;
      else jac(i, j) = 0;
    }
  }
  jac(6, 3) = -1 * p4.Px() / p4.M();
  jac(6, 4) = -1 * p4.Py() / p4.M();
  jac(6, 5) = -1 * p4.Pz() / p4.M();
  jac(6, 6) = p4.E() / p4.M();
 
  TMatrix jact(7, 7); jact.Transpose(jac);
  TMatrix MassErrPart(7, 7); MassErrPart.Mult(jac, EnergyErr);
  TMatrix MassErrTemp(7, 7); MassErrTemp.Mult(MassErrPart, jact);
 
  TMatrixDSym MassErr(7);
  for (int i = 0; i < 7; i++) {
    for (int j = 0; j < 7; j++) {
      MassErr(i, j) = MassErrTemp(i, j);
    }
  }
 
  return MassErr;
}

ErrorMatrixMassToEnergy()

TMatrixDSym ErrorMatrixMassToEnergy ( const ROOT::Math::PxPyPzEVector &  p4, const TMatrixDSym &  MassErr  )

private

Convert the error matrix from P-M to P-E.

It Requires an input error matrix in the form X,P,M

Definition at line 453 of file RaveKinematicVertexFitter.cc.

{
 
  TMatrix jac(7, 7);
  for (int i = 0; i < 7; i++) {
    for (int j = 0; j < 7; j++) {
      if (i == j) jac(i, j) = 1;
      else jac(i, j) = 0;
    }
  }
  jac(6, 3) = p4.Px() / p4.E();
  jac(6, 4) = p4.Py() / p4.E();
  jac(6, 5) = p4.Pz() / p4.E();
  jac(6, 6) = p4.M() / p4.E();
 
 
  TMatrix jact(7, 7); jact.Transpose(jac);
  TMatrix EnergyErrPart(7, 7); EnergyErrPart.Mult(jac, MassErr);
  TMatrix EnergyErrTemp(7, 7); EnergyErrTemp.Mult(EnergyErrPart, jact);
 
  TMatrixDSym EnergyErr(7);
  for (int i = 0; i < 7; i++) {
    for (int j = 0; j < 7; j++) {
      EnergyErr(i, j) = EnergyErrTemp(i, j);
    }
  }
 
  return EnergyErr;
}

fit()

int fit

(

)

do the kinematic vertex fit with all tracks previously added with the addTrack or addMother function.

The return value is 1 if a vertex is successfully found. Return value 0 means the fit was not successful. -1 means not enough tracks were added

Definition at line 129 of file RaveKinematicVertexFitter.cc.

{
  // make sure all output in this function is converted to log messages
  auto output_capture = captureOutput();
 
  if (m_inputParticles.size() < 2 && m_vertFit) {
    return -1;
  }
  int nOfVertices = -100;
  if (m_useBeamSpot == true) {
    const B2Vector3D& bsPos = RaveSetup::getRawInstance()->m_beamSpot;
    const TMatrixDSym& bsCov = RaveSetup::getRawInstance()->m_beamSpotCov;
    const rave::Covariance3D bsCovRave(bsCov(0, 0), bsCov(0, 1), bsCov(0, 2), bsCov(1, 1), bsCov(1, 2), bsCov(2, 2));
    RaveSetup::getRawInstance()->m_raveVertexFactory->setBeamSpot(rave::Ellipsoid3D(rave::Point3D(bsPos.X(), bsPos.Y(), bsPos.Z()),
        bsCovRave));
  }
 
  if (m_vertFit && m_massConstFit) {
 
    rave::KinematicConstraint cs2 = rave::KinematicConstraintBuilder().createMultiTrackMassKinematicConstraint(
                                      m_motherParticlePtr->getPDGMass(), m_inputParticles.size());
    try {
      m_fittedResult = RaveSetup::getRawInstance()->m_raveKinematicTreeFactory->useVertexFitter(m_inputParticles, cs2, "", m_useBeamSpot);
      m_fittedParticle = m_fittedResult.topParticle();
    } catch (...) {
      nOfVertices = 0;
    }
  } else {
 
    if (m_vertFit) {
      if (!m_massConstFit) {
        try {
          m_fittedResult = RaveSetup::getRawInstance()->m_raveKinematicTreeFactory->useVertexFitter(m_inputParticles, "", m_useBeamSpot);
          m_fittedParticle = m_fittedResult.topParticle();
        } catch (...) {
          nOfVertices = 0;
        }
      }
 
    }
 
    if (!m_vertFit && m_massConstFit) {
 
      try {
 
        Particle* aParticlePtr = m_motherParticlePtr;
 
        rave::Vector7D raveState(aParticlePtr->getX(), aParticlePtr->getY(), aParticlePtr->getZ(), aParticlePtr->getPx(),
                                 aParticlePtr->getPy(), aParticlePtr->getPz(), aParticlePtr->getMass());
        TMatrixFSym covP = aParticlePtr->getMomentumVertexErrorMatrix();
 
        float chi2 = 1;
        float ndf = 1.;
 
        TMatrixDSym covE(7);
        for (int i = 0; i < 7; i++) {
          for (int j = 0; j < 7; j++) {
            if (i < 3 && j < 3) covE(i, j) = covP(i + 4, j + 4);
            if (i > 2 && j > 2) covE(i, j) = covP(i - 3, j - 3);
            if (i < 3 && j > 2) covE(i, j) = covP(i + 4, j - 3);
            if (i > 2 && j < 3) covE(i, j) = covP(i - 3, j + 4);
          }
        }
 
        TMatrixDSym cov = ErrorMatrixEnergyToMass(aParticlePtr->get4Vector(), covE);
 
        rave::Covariance7D raveCov(cov(0, 0), cov(0, 1), cov(0, 2), // x x , x y, x z
                                   cov(1, 1), cov(1, 2), cov(2, 2), // y y , y z, z z
                                   cov(0, 3), cov(0, 4), cov(0, 5), // x px , x py, x pz
                                   cov(1, 3), cov(1, 4), cov(1, 5), // y px , y py, y pz
                                   cov(2, 3), cov(2, 4), cov(2, 5), // z px , z py, z pz
                                   cov(3, 3), cov(3, 4), cov(3, 5), // px px , px py, px pz
                                   cov(4, 4), cov(4, 5), cov(5, 5), // py py , py pz, pz pz
                                   cov(0, 6), cov(1, 6), cov(2, 6), // x m , y m, z m
                                   cov(3, 6), cov(4, 6), cov(5, 6), // px m, py m, pz m
                                   cov(6, 6)); // mm
 
        rave::TransientTrackKinematicParticle aRaveParticle(raveState, raveCov, rave::Charge(aParticlePtr->getCharge()), chi2, ndf);
 
        std::vector< rave::KinematicParticle > parts; parts.push_back(aRaveParticle);
 
        rave::KinematicConstraint constraint = rave::KinematicConstraintBuilder().createMassKinematicConstraint(
                                                 m_motherParticlePtr->getPDGMass(), 0.);
 
        if (m_motherParticlePtr->getMomentumVertexErrorMatrix().Determinant() != 0) {
 
          std::vector< rave::KinematicParticle > refitted = RaveSetup::getRawInstance()->m_raveKinematicTreeFactory->useParticleFitter(parts,
                                                            constraint, "ppf:lppf");
 
          m_fittedParticle = refitted[0];
 
        } else {
          B2ERROR("[RaveKinematicVertexFitter]: VertexException saying ParentParticleFitter::error inverting covariance matrix occurred");
          nOfVertices = 0;
        }
 
 
      } catch (...) {
        nOfVertices = 0;
      }
    }
 
  }
 
 
  rave::Vector7D fittedState;
  rave::Covariance7D fittedCov;
 
  try {
    fittedState = m_fittedParticle.fullstate();
    fittedCov = m_fittedParticle.fullerror();
  } catch (...) {
    nOfVertices = 0;
  }
 
 
  for (auto& i : m_inputParticles) i.unlink();
  m_inputParticles.clear();
  //
 
  if (nOfVertices == 0) { // vertex fit not successful
    return 0;
  }
 
  if (m_vertFit) {
    rave::KinematicVertex fittedVertex = m_fittedResult.currentDecayVertex();
 
    m_fittedNdf = fittedVertex.ndf();
    m_fittedChi2 = fittedVertex.chiSquared();
    m_fittedPValue = ROOT::Math::chisquared_cdf_c(m_fittedChi2, m_fittedNdf);
    m_fittedPos.SetXYZ(fittedVertex.position().x(), fittedVertex.position().y(), fittedVertex.position().z());
 
  } else {
    m_fittedNdf = m_fittedParticle.ndof();
    m_fittedChi2 = m_fittedParticle.chi2();
    m_fittedPValue = ROOT::Math::chisquared_cdf_c(m_fittedChi2, m_fittedNdf);
    m_fittedPos.SetXYZ(m_motherParticlePtr->getX(), m_motherParticlePtr->getY(), m_motherParticlePtr->getZ());
  }
 
 
  m_fitted4Vector.SetXYZT(fittedState.p4().p3().x(), fittedState.p4().p3().y(), fittedState.p4().p3().z(), fittedState.p4().energy());
 
  m_fitted7Cov.ResizeTo(7, 7);
 
  TMatrixDSym fitted7CovM(7);
 
  fitted7CovM(3, 6) = fittedCov.dpxm();  fitted7CovM(6, 3) = fitted7CovM(3, 6);
  fitted7CovM(4, 6) = fittedCov.dpym();  fitted7CovM(6, 4) = fitted7CovM(4, 6);
  fitted7CovM(5, 6) = fittedCov.dpzm();  fitted7CovM(6, 5) = fitted7CovM(5, 6);
  fitted7CovM(6, 6) = fittedCov.dmm();   fitted7CovM(6, 6) = fitted7CovM(6, 6);
  fitted7CovM(0, 6) = fittedCov.dxm();   fitted7CovM(6, 0) = fitted7CovM(0, 6);
  fitted7CovM(1, 6) = fittedCov.dym();   fitted7CovM(6, 1) = fitted7CovM(1, 6);
  fitted7CovM(2, 6) = fittedCov.dzm();   fitted7CovM(6, 2) = fitted7CovM(2, 6);
 
  fitted7CovM(3, 3) = fittedCov.dpxpx(); fitted7CovM(3, 3) = fitted7CovM(3, 3);
  fitted7CovM(3, 4) = fittedCov.dpxpy(); fitted7CovM(4, 3) = fitted7CovM(3, 4);
  fitted7CovM(3, 5) = fittedCov.dpxpz(); fitted7CovM(5, 3) = fitted7CovM(3, 5);
  fitted7CovM(3, 0) = fittedCov.dxpx();  fitted7CovM(0, 3) = fitted7CovM(3, 0);
  fitted7CovM(3, 1) = fittedCov.dypx();  fitted7CovM(1, 3) = fitted7CovM(3, 1);
  fitted7CovM(3, 2) = fittedCov.dzpx();  fitted7CovM(2, 3) = fitted7CovM(3, 2);
 
  fitted7CovM(4, 4) = fittedCov.dpypy(); fitted7CovM(4, 4) = fitted7CovM(4, 4);
  fitted7CovM(4, 5) = fittedCov.dpypz(); fitted7CovM(5, 4) = fitted7CovM(4, 5);
  fitted7CovM(4, 0) = fittedCov.dxpy();  fitted7CovM(0, 4) = fitted7CovM(4, 0);
  fitted7CovM(4, 1) = fittedCov.dypy();  fitted7CovM(1, 4) = fitted7CovM(4, 1);
  fitted7CovM(4, 2) = fittedCov.dzpy();  fitted7CovM(2, 4) = fitted7CovM(4, 2);
 
  fitted7CovM(5, 5) = fittedCov.dpzpz(); fitted7CovM(5, 5) = fitted7CovM(5, 5);
  fitted7CovM(5, 0) = fittedCov.dxpz();  fitted7CovM(0, 5) = fitted7CovM(5, 0);
  fitted7CovM(5, 1) = fittedCov.dypz();  fitted7CovM(1, 5) = fitted7CovM(5, 1);
  fitted7CovM(5, 2) = fittedCov.dzpz();  fitted7CovM(2, 5) = fitted7CovM(5, 2);
 
  fitted7CovM(0, 0) = fittedCov.dxx();   fitted7CovM(0, 0) = fitted7CovM(0, 0);
  fitted7CovM(0, 1) = fittedCov.dxy();   fitted7CovM(1, 0) = fitted7CovM(0, 1);
  fitted7CovM(0, 2) = fittedCov.dxz();   fitted7CovM(2, 0) = fitted7CovM(0, 2);
 
  fitted7CovM(1, 1) = fittedCov.dyy();   fitted7CovM(1, 1) = fitted7CovM(1, 1);
  fitted7CovM(1, 2) = fittedCov.dyz();   fitted7CovM(2, 1) = fitted7CovM(1, 2);
 
  fitted7CovM(2, 2) = fittedCov.dzz();   fitted7CovM(2, 2) = fitted7CovM(2, 2);
 
  TMatrixDSym fitted7CovE = ErrorMatrixMassToEnergy(m_fitted4Vector, fitted7CovM);
 
  for (int i = 0; i < 7; i++) {
    for (int j = 0; j < 7; j++) {
      if (i < 4 && j < 4) m_fitted7Cov(i, j) = fitted7CovE(i + 3, j + 3);
      if (i > 3 && j > 3) m_fitted7Cov(i, j) = fitted7CovE(i - 4, j - 4);
      if (i < 4 && j > 3) m_fitted7Cov(i, j) = fitted7CovE(i + 3, j - 4);
      if (i > 3 && j < 4) m_fitted7Cov(i, j) = fitted7CovE(i - 4, j + 3);
    }
  }
 
  return 1;
}

getChi2()

double getChi2

(

)

get the χ² of the fitted vertex.

Definition at line 426 of file RaveKinematicVertexFitter.cc.

{
  return m_fittedChi2;
}

getCov()

TMatrixDSym getCov

(

)

get the covariance matrix (7x7).

Definition at line 431 of file RaveKinematicVertexFitter.cc.

{
  return m_fitted7Cov;
}

getMother()

Particle * getMother

(

)

returns a pointer to the mother particle

Definition at line 406 of file RaveKinematicVertexFitter.cc.

{
  return m_motherParticlePtr;
}

getNdf()

double getNdf

(

)

get the number of degrees of freedom (NDF) of the fitted vertex.

Definition at line 421 of file RaveKinematicVertexFitter.cc.

{
  return m_fittedNdf;
}

getPos()

ROOT::Math::XYZVector getPos

(

)

get the position of the fitted vertex.

Definition at line 411 of file RaveKinematicVertexFitter.cc.

{
  return m_fittedPos;
}

getPValue()

double getPValue

(

)

get the p value of the fitted vertex.

Definition at line 416 of file RaveKinematicVertexFitter.cc.

{
  return m_fittedPValue;
}

getVertexErrorMatrix()

TMatrixDSym getVertexErrorMatrix

(

)

get the covariance matrix (3x3) of the of the fitted vertex position.

Definition at line 436 of file RaveKinematicVertexFitter.cc.

{
  TMatrixDSym posErr(3);
  posErr(0, 0) = m_fitted7Cov(4, 4);
  posErr(0, 1) = m_fitted7Cov(4, 5);
  posErr(0, 2) = m_fitted7Cov(4, 6);
  posErr(1, 0) = m_fitted7Cov(5, 4);
  posErr(1, 1) = m_fitted7Cov(5, 5);
  posErr(1, 2) = m_fitted7Cov(5, 6);
  posErr(2, 0) = m_fitted7Cov(6, 4);
  posErr(2, 1) = m_fitted7Cov(6, 5);
  posErr(2, 2) = m_fitted7Cov(6, 6);
 
  return posErr;
}

Print()

void Print ( ) const

inline

Print all attributes of this object to terminal.

Definition at line 120 of file RaveKinematicVertexFitter.h.

      {
        B2INFO("useBeamSpot: " << m_useBeamSpot);
        B2INFO("number of rave::Tracks: " << m_inputParticles.size());
      }

setMassConstFit()

void setMassConstFit

(

bool

isConstFit = true

)

Set mass constrained fit

Definition at line 56 of file RaveKinematicVertexFitter.cc.

{
  m_massConstFit = isConstFit;
}

setMother()

void setMother

(

const Particle *

aMotherParticlePtr

)

Set Mother particle for Vertex/momentum update.

Not to be used with addMother

Definition at line 121 of file RaveKinematicVertexFitter.cc.

{
  auto* tmp = const_cast<Particle*>(aMotherParticlePtr);
  m_motherParticlePtr = tmp;
}

setVertFit()

void setVertFit

(

bool

isVertFit = true

)

Set vertex fit: set false in case of mass fit only.

Definition at line 61 of file RaveKinematicVertexFitter.cc.

{
  m_vertFit = isVertFit;
}

updateDaughters()

void updateDaughters

(

)

update the Daughters particles

Definition at line 329 of file RaveKinematicVertexFitter.cc.

{
 
  m_fittedResult.topParticle();
  std::vector< rave::KinematicParticle > rDau = m_fittedResult.daughterParticles();
  std::vector<Belle2::Particle*> bDau = m_belleDaughters;
  if (rDau.size() == bDau.size()) {
    for (unsigned ii = 0; ii < bDau.size(); ii++) {
      rave::Vector7D fittedState;
      rave::Covariance7D fittedCov;
      fittedState = rDau[ii].fullstate();
      fittedCov = rDau[ii].fullerror();
 
      ROOT::Math::PxPyPzEVector p4(fittedState.p4().p3().x(), fittedState.p4().p3().y(), fittedState.p4().p3().z(),
                                   fittedState.p4().energy());
 
      ROOT::Math::XYZVector x3(fittedState.x(), fittedState.y(), fittedState.z());
 
 
      TMatrixDSym fitted7CovM(7);
      fitted7CovM(3, 6) = fittedCov.dpxm();  fitted7CovM(6, 3) = fitted7CovM(3, 6);
      fitted7CovM(4, 6) = fittedCov.dpym();  fitted7CovM(6, 4) = fitted7CovM(4, 6);
      fitted7CovM(5, 6) = fittedCov.dpzm();  fitted7CovM(6, 5) = fitted7CovM(5, 6);
      fitted7CovM(6, 6) = fittedCov.dmm();   fitted7CovM(6, 6) = fitted7CovM(6, 6);
      fitted7CovM(0, 6) = fittedCov.dxm();   fitted7CovM(6, 0) = fitted7CovM(0, 6);
      fitted7CovM(1, 6) = fittedCov.dym();   fitted7CovM(6, 1) = fitted7CovM(1, 6);
      fitted7CovM(2, 6) = fittedCov.dzm();   fitted7CovM(6, 2) = fitted7CovM(2, 6);
 
      fitted7CovM(3, 3) = fittedCov.dpxpx(); fitted7CovM(3, 3) = fitted7CovM(3, 3);
      fitted7CovM(3, 4) = fittedCov.dpxpy(); fitted7CovM(4, 3) = fitted7CovM(3, 4);
      fitted7CovM(3, 5) = fittedCov.dpxpz(); fitted7CovM(5, 3) = fitted7CovM(3, 5);
      fitted7CovM(3, 0) = fittedCov.dxpx();  fitted7CovM(0, 3) = fitted7CovM(3, 0);
      fitted7CovM(3, 1) = fittedCov.dypx();  fitted7CovM(1, 3) = fitted7CovM(3, 1);
      fitted7CovM(3, 2) = fittedCov.dzpx();  fitted7CovM(2, 3) = fitted7CovM(3, 2);
 
      fitted7CovM(4, 4) = fittedCov.dpypy(); fitted7CovM(4, 4) = fitted7CovM(4, 4);
      fitted7CovM(4, 5) = fittedCov.dpypz(); fitted7CovM(5, 4) = fitted7CovM(4, 5);
      fitted7CovM(4, 0) = fittedCov.dxpy();  fitted7CovM(0, 4) = fitted7CovM(4, 0);
      fitted7CovM(4, 1) = fittedCov.dypy();  fitted7CovM(1, 4) = fitted7CovM(4, 1);
      fitted7CovM(4, 2) = fittedCov.dzpy();  fitted7CovM(2, 4) = fitted7CovM(4, 2);
 
      fitted7CovM(5, 5) = fittedCov.dpzpz(); fitted7CovM(5, 5) = fitted7CovM(5, 5);
      fitted7CovM(5, 0) = fittedCov.dxpz();  fitted7CovM(0, 5) = fitted7CovM(5, 0);
      fitted7CovM(5, 1) = fittedCov.dypz();  fitted7CovM(1, 5) = fitted7CovM(5, 1);
      fitted7CovM(5, 2) = fittedCov.dzpz();  fitted7CovM(2, 5) = fitted7CovM(5, 2);
 
      fitted7CovM(0, 0) = fittedCov.dxx();   fitted7CovM(0, 0) = fitted7CovM(0, 0);
      fitted7CovM(0, 1) = fittedCov.dxy();   fitted7CovM(1, 0) = fitted7CovM(0, 1);
      fitted7CovM(0, 2) = fittedCov.dxz();   fitted7CovM(2, 0) = fitted7CovM(0, 2);
 
      fitted7CovM(1, 1) = fittedCov.dyy();   fitted7CovM(1, 1) = fitted7CovM(1, 1);
      fitted7CovM(1, 2) = fittedCov.dyz();   fitted7CovM(2, 1) = fitted7CovM(1, 2);
 
      fitted7CovM(2, 2) = fittedCov.dzz();   fitted7CovM(2, 2) = fitted7CovM(2, 2);
 
      TMatrixDSym fitted7CovE = ErrorMatrixMassToEnergy(m_fitted4Vector, fitted7CovM);
 
      TMatrixDSym fitted7CovDauM(7);
      for (int i = 0; i < 7; i++) {
        for (int j = 0; j < 7; j++) {
          if (i < 4 && j < 4) fitted7CovDauM(i, j) = fitted7CovE(i + 3, j + 3);
          if (i > 3 && j > 3) fitted7CovDauM(i, j) = fitted7CovE(i - 4, j - 4);
          if (i < 4 && j > 3) fitted7CovDauM(i, j) = fitted7CovE(i + 3, j - 4);
          if (i > 3 && j < 4) fitted7CovDauM(i, j) = fitted7CovE(i - 4, j + 3);
        }
      }
 
      float pValDau = rDau[ii].chi2();
 
      bDau[ii]->updateMomentum(p4, x3, fitted7CovDauM, pValDau);
 
    }
 
  } else B2ERROR("Error in Daughters update");
 
}

updateMother()

void updateMother

(

)

update the mother particle

Definition at line 324 of file RaveKinematicVertexFitter.cc.

{
  m_motherParticlePtr->updateMomentum(m_fitted4Vector, m_fittedPos, m_fitted7Cov, m_fittedPValue);
}

useBeamSpot()

void useBeamSpot ( bool  beamSpot = true )

inline

Overwrite the global option in ReveSetup that tells the fitter if beam spot info should be used or not.

The beam spot pos and cov must still be set in the RaveSetup class if you what to use it

Definition at line 85 of file RaveKinematicVertexFitter.h.

      {
        if (beamSpot == true) {
          if (RaveSetup::getRawInstance()->m_useBeamSpot == false) {
            B2FATAL("Beam spot information cannot be used because the beam spot position and covariance was not set in RaveSetup");
          }
          m_useBeamSpot = true;
        } else {
          m_useBeamSpot = false;
        }
      }

Member Data Documentation

m_belleDaughters

std::vector<Particle*> m_belleDaughters

protected

Belle Particle pointers input.

Definition at line 155 of file RaveKinematicVertexFitter.h.

m_fitted4Vector

ROOT::Math::PxPyPzEVector m_fitted4Vector

protected

4 momentum of the mother particle after the fit

Definition at line 170 of file RaveKinematicVertexFitter.h.

m_fitted7Cov

TMatrixFSym m_fitted7Cov

protected

7x7 error matrix of the mother particle after the fit

Definition at line 172 of file RaveKinematicVertexFitter.h.

m_fittedChi2

double m_fittedChi2

protected

chi^2 of the vertex fit

Definition at line 166 of file RaveKinematicVertexFitter.h.

m_fittedNdf

double m_fittedNdf

protected

Ndf of the vertex fit.

Definition at line 162 of file RaveKinematicVertexFitter.h.

m_fittedParticle

rave::KinematicParticle m_fittedParticle

protected

Particle fit output.

Definition at line 152 of file RaveKinematicVertexFitter.h.

m_fittedPos

ROOT::Math::XYZVector m_fittedPos

protected

Fitted vertex position.

Definition at line 168 of file RaveKinematicVertexFitter.h.

m_fittedPValue

double m_fittedPValue

protected

Pvalue of the fit result.

Definition at line 164 of file RaveKinematicVertexFitter.h.

m_fittedResult

rave::KinematicTree m_fittedResult

protected

the output of the kinematic fit

Definition at line 141 of file RaveKinematicVertexFitter.h.

m_inputParticles

std::vector< rave::KinematicParticle > m_inputParticles

protected

input particles for vertex fit in rave format

Definition at line 150 of file RaveKinematicVertexFitter.h.

m_massConstFit

bool m_massConstFit

protected

flag determines if the mass fit is performed

Definition at line 158 of file RaveKinematicVertexFitter.h.

m_motherParticlePtr

Particle* m_motherParticlePtr

protected

pointer to the mother particle who's daughters will be used in the fit.

the fit result will be written back to the mother particle

Definition at line 135 of file RaveKinematicVertexFitter.h.

m_raveAlgorithm

std::string m_raveAlgorithm

protected

Algorithm used by rave (kalman, avr, ...)

Definition at line 138 of file RaveKinematicVertexFitter.h.

m_raveTracks

std::vector<rave::Track> m_raveTracks

protected

holds the tracks that were added to a RaveVertexFitter object in the format used by Rave

Definition at line 144 of file RaveKinematicVertexFitter.h.

m_raveVertices

std::vector< rave::Vertex > m_raveVertices

protected

holds the fitted vertices after fit() was called in the format used by Rave

Definition at line 147 of file RaveKinematicVertexFitter.h.

m_useBeamSpot

bool m_useBeamSpot

protected

flag determines if the beam spot will be used or not.

Overwrites the global flag in RaveSetup

Definition at line 132 of file RaveKinematicVertexFitter.h.

m_vertFit

bool m_vertFit

protected

flag determines if the vertex fit is performed

Definition at line 160 of file RaveKinematicVertexFitter.h.

---

The documentation for this class was generated from the following files:

• analysis/VertexFitting/RaveInterface/include/RaveKinematicVertexFitter.h
• analysis/VertexFitting/RaveInterface/src/RaveKinematicVertexFitter.cc