TrackVariables.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
// Own header.
#include <analysis/variables/TrackVariables.h>
 
// include VariableManager
#include <analysis/VariableManager/Manager.h>
 
#include <analysis/dataobjects/Particle.h>
#include <analysis/utility/DetectorSurface.h>
 
// framework - DataStore
#include <framework/datastore/StoreObjPtr.h>
#include <framework/dataobjects/Helix.h>
 
// dataobjects from the MDST
#include <mdst/dbobjects/BeamSpot.h>
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/MCParticle.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <mdst/dataobjects/EventLevelTrackingInfo.h>
#include <mdst/dataobjects/HitPatternVXD.h>
#include <mdst/dataobjects/ECLCluster.h>
 
// framework aux
#include <framework/logging/Logger.h>
 
#include <cmath>
 
namespace Belle2 {
  namespace Variable {
 
    static const ROOT::Math::XYZVector vecNaN(Const::doubleNaN, Const::doubleNaN, Const::doubleNaN);
 
    double trackNHits(const Particle* part, const Const::EDetector& det)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
 
      if (det == Const::EDetector::CDC) {
        return trackFit->getHitPatternCDC().getNHits();
      } else if (det == Const::EDetector::SVD) {
        return trackFit->getHitPatternVXD().getNSVDHits();
      } else if (det == Const::EDetector::PXD) {
        return trackFit->getHitPatternVXD().getNPXDHits();
      } else {
        return Const::doubleNaN;
      }
    }
 
    double trackNCDCHits(const Particle* part)
    {
      return trackNHits(part, Const::EDetector::CDC);
    }
 
    double trackNSVDHits(const Particle* part)
    {
      return trackNHits(part, Const::EDetector::SVD);
    }
 
    double trackNPXDHits(const Particle* part)
    {
      return trackNHits(part, Const::EDetector::PXD);
    }
 
    double trackNVXDHits(const Particle* part)
    {
      return trackNPXDHits(part) + trackNSVDHits(part);
    }
 
    double trackNDF(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getNDF();
    }
 
    double trackChi2(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getChi2();
    }
 
    double trackFirstSVDLayer(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getHitPatternVXD().getFirstSVDLayer();
    }
 
    double trackFirstPXDLayer(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getHitPatternVXD().getFirstPXDLayer(HitPatternVXD::PXDMode::normal);
    }
 
    double trackFirstCDCLayer(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getHitPatternCDC().getFirstLayer();
    }
 
    double trackLastCDCLayer(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getHitPatternCDC().getLastLayer();
    }
 
    double trackD0(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getD0();
    }
 
    double trackPhi0(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getPhi0();
    }
 
    double trackOmega(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getOmega();
    }
 
    double trackZ0(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getZ0();
    }
 
    double trackTanLambda(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getTanLambda();
    }
 
    double trackD0FromIP(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      static DBObjPtr<BeamSpot> beamSpotDB;
      if (!beamSpotDB.isValid())
        return Const::doubleNaN;
      auto helix = trackFit->getHelix();
      helix.passiveMoveBy(beamSpotDB->getIPPosition());
      return helix.getD0();
    }
 
    double trackZ0FromIP(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      static DBObjPtr<BeamSpot> beamSpotDB;
      if (!beamSpotDB.isValid())
        return Const::doubleNaN;
      auto helix = trackFit->getHelix();
      helix.passiveMoveBy(beamSpotDB->getIPPosition());
      return helix.getZ0();
    }
 
    double trackPhi0FromIP(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      static DBObjPtr<BeamSpot> beamSpotDB;
      if (!beamSpotDB.isValid())
        return Const::doubleNaN;
      auto helix = trackFit->getHelix();
      helix.passiveMoveBy(beamSpotDB->getIPPosition());
      return helix.getPhi0();
    }
 
    double trackD0Error(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      double errorSquared = trackFit->getCovariance5()[0][0];
      if (errorSquared <= 0) return Const::doubleNaN;
      return sqrt(errorSquared);
    }
 
    double trackPhi0Error(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
 
      double errorSquared = trackFit->getCovariance5()[1][1];
      if (errorSquared <= 0) return Const::doubleNaN;
      return sqrt(errorSquared);
    }
 
    double trackOmegaError(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
 
      double errorSquared = trackFit->getCovariance5()[2][2];
      if (errorSquared <= 0) return Const::doubleNaN;
      return sqrt(errorSquared);
    }
 
    double trackZ0Error(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
 
      double errorSquared = trackFit->getCovariance5()[3][3];
      if (errorSquared <= 0) return Const::doubleNaN;
      return sqrt(errorSquared);
    }
 
    double trackTanLambdaError(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
 
      double errorSquared = trackFit->getCovariance5()[4][4];
      if (errorSquared <= 0) return Const::doubleNaN;
      return sqrt(errorSquared);
    }
 
    double trackFitCovariance(const Particle* particle, const std::vector<double>& indices)
    {
      if (indices.size() != 2) {
        B2FATAL("Exactly two indices must be provided to the variable trackFitCovariance!");
      }
      if (*(std::min_element(indices.begin(), indices.end())) < 0 or *(std::max_element(indices.begin(), indices.end())) > 4) {
        B2FATAL("The indices provided to the variable trackFitCovariance must be in the range 0 - 4!");
      }
      auto trackFit = particle->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getCovariance5()[indices[0]][indices[1]];
    }
 
    double trackPValue(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getPValue();
    }
 
    double trackFitHypothesisPDG(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
      return trackFit->getParticleType().getPDGCode();
    }
 
    double trackNECLClusters(const Particle* part)
    {
      const Track* track = part->getTrack();
      if (!track) return Const::doubleNaN;
 
      // count the number of nPhotons hypothesis ecl clusters
      int count = 0;
      for (const ECLCluster& cluster : track->getRelationsTo<ECLCluster>())
        if (cluster.hasHypothesis(ECLCluster::EHypothesisBit::c_nPhotons))
          ++count;
      return count;
    }
 
    // used in trackHelixExtTheta and trackHelixExtPhi
    ROOT::Math::XYZVector getPositionOnHelix(const TrackFitResult* trackFit, const std::vector<double>& pars)
    {
      const double r = pars[0];
      const double zfwd = pars[1];
      const double zbwd = pars[2];
 
      // get helix and parameters
      const double z0 = trackFit->getZ0();
      const double tanlambda = trackFit->getTanLambda();
      const Helix h = trackFit->getHelix();
 
      // extrapolate to radius
      const double arcLength = h.getArcLength2DAtCylindricalR(r);
      const double lHelixRadius = arcLength > 0 ? arcLength : std::numeric_limits<double>::max();
 
      // extrapolate to FWD z
      const double lFWD = (zfwd - z0) / tanlambda > 0 ? (zfwd - z0) / tanlambda : std::numeric_limits<double>::max();
 
      // extrapolate to BWD z
      const double lBWD = (zbwd - z0) / tanlambda > 0 ? (zbwd - z0) / tanlambda : std::numeric_limits<double>::max();
 
      // pick smallest arclength
      const double l = std::min({lHelixRadius, lFWD, lBWD});
 
      return h.getPositionAtArcLength2D(l);
    }
 
    ROOT::Math::XYZVector getPositionOnHelix(const Particle* part, const std::vector<double>& pars)
    {
      if (pars.size() == 4 and pars[3]) {
        const Track* track = part->getTrack();
        if (!track)
          return vecNaN;
 
        auto highestProbMass = part->getMostLikelyTrackFitResult().first;
        const TrackFitResult* trackFit = track->getTrackFitResultWithClosestMass(highestProbMass);
        if (!trackFit) return vecNaN;
        else return getPositionOnHelix(trackFit, pars);
      } else {
        const TrackFitResult* trackFit = part->getTrackFitResult();
        if (!trackFit) return vecNaN;
        else return getPositionOnHelix(trackFit, pars);
      }
    }
 
    // returns extrapolated theta position based on helix parameters
    double trackHelixExtTheta(const Particle* part, const std::vector<double>& pars)
    {
      const auto nParams = pars.size();
      if (nParams != 3 && nParams != 4) {
        B2FATAL("Exactly three (+1 optional) parameters (r, zfwd, zbwd, [useHighestProbMass]) required for helixExtTheta.");
      }
 
      ROOT::Math::XYZVector position = getPositionOnHelix(part, pars);
      if (position == vecNaN) return Const::doubleNaN;
      return position.Theta();
    }
 
    // returns extrapolated phi position based on helix parameters
    double trackHelixExtPhi(const Particle* part, const std::vector<double>& pars)
    {
      const auto nParams = pars.size();
      if (nParams != 3 && nParams != 4) {
        B2FATAL("Exactly three (+1 optional) parameters (r, zfwd, zbwd, [useHighestProbMass]) required for helixExtPhi.");
      }
 
      ROOT::Math::XYZVector position = getPositionOnHelix(part, pars);
      if (position == vecNaN) return Const::doubleNaN;
      return position.Phi();
    }
 
    Manager::FunctionPtr trackHelixExtThetaOnDet(const std::vector<std::string>& arguments)
    {
      if (arguments.size() != 1 && arguments.size() != 2)
        B2FATAL("Exactly one (+1 optional) parameter (detector_surface_name, [useHighestProbMass]) is required for helixExtThetaOnDet.");
 
      std::vector<double> parameters(3);
      const std::string det = arguments[0];
      if (DetectorSurface::detToSurfBoundaries.find(det) != DetectorSurface::detToSurfBoundaries.end()) {
        parameters[0] = DetectorSurface::detToSurfBoundaries.at(det).m_rho;
        parameters[1] = DetectorSurface::detToSurfBoundaries.at(det).m_zfwd;
        parameters[2] = DetectorSurface::detToSurfBoundaries.at(det).m_zbwd;
      } else if (DetectorSurface::detLayerToSurfBoundaries.find(det) != DetectorSurface::detLayerToSurfBoundaries.end()) {
        parameters[0] = DetectorSurface::detLayerToSurfBoundaries.at(det).m_rho;
        parameters[1] = DetectorSurface::detLayerToSurfBoundaries.at(det).m_zfwd;
        parameters[2] = DetectorSurface::detLayerToSurfBoundaries.at(det).m_zbwd;
      } else
        B2FATAL("Given detector surface name is not supported.");
 
      if (arguments.size() == 2)
        parameters.push_back(std::stod(arguments[1]));
 
      auto func = [parameters](const Particle * part) -> double {
 
        ROOT::Math::XYZVector position = getPositionOnHelix(part, parameters);
        if (position == vecNaN) return Const::doubleNaN;
        return position.Theta();
      };
      return func;
    }
 
    Manager::FunctionPtr trackHelixExtPhiOnDet(const std::vector<std::string>& arguments)
    {
      if (arguments.size() != 1 && arguments.size() != 2)
        B2FATAL("Exactly one (+1 optional) parameter (detector_surface_name, [useHighestProbMass]) is required for helixExtPhiOnDet.");
 
      std::vector<double> parameters(3);
      const std::string det = arguments[0];
      if (DetectorSurface::detToSurfBoundaries.find(det) != DetectorSurface::detToSurfBoundaries.end()) {
        parameters[0] = DetectorSurface::detToSurfBoundaries.at(det).m_rho;
        parameters[1] = DetectorSurface::detToSurfBoundaries.at(det).m_zfwd;
        parameters[2] = DetectorSurface::detToSurfBoundaries.at(det).m_zbwd;
      } else if (DetectorSurface::detLayerToSurfBoundaries.find(det) != DetectorSurface::detLayerToSurfBoundaries.end()) {
        parameters[0] = DetectorSurface::detLayerToSurfBoundaries.at(det).m_rho;
        parameters[1] = DetectorSurface::detLayerToSurfBoundaries.at(det).m_zfwd;
        parameters[2] = DetectorSurface::detLayerToSurfBoundaries.at(det).m_zbwd;
      } else
        B2FATAL("Given detector surface name is not supported.");
 
      if (arguments.size() == 2)
        parameters.push_back(std::stod(arguments[1]));
 
      auto func = [parameters](const Particle * part) -> double {
 
        ROOT::Math::XYZVector position = getPositionOnHelix(part, parameters);
        if (position == vecNaN) return Const::doubleNaN;
        return position.Phi();
      };
      return func;
    }
 
 
    /***************************************************
     * Event level tracking quantities
     */
 
    // The number of CDC hits in the event not assigned to any track
    double nExtraCDCHits(const Particle*)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      return elti->getNCDCHitsNotAssigned();
    }
 
    // The number of CDC hits in the event not assigned to any track nor very
    // likely beam background (i.e. hits that survive a cleanup selection)
    double nExtraCDCHitsPostCleaning(const Particle*)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      return elti->getNCDCHitsNotAssignedPostCleaning();
    }
 
    // Check for the presence of a non-assigned hit in the specified CDC layer
    double hasExtraCDCHitsInLayer(const Particle*, const std::vector<double>& layer)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      int ilayer = std::lround(layer[0]);
      return elti->hasCDCLayer(ilayer);
    }
 
    // Check for the presence of a non-assigned hit in the specified CDC SuperLayer
    double hasExtraCDCHitsInSuperLayer(const Particle*, const std::vector<double>& layer)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      int ilayer = std::lround(layer[0]);
      return elti->hasCDCSLayer(ilayer);
    }
 
    // The number of segments that couldn't be assigned to any track
    double nExtraCDCSegments(const Particle*)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      return elti->getNCDCSegments();
    }
 
    //  The number of VXD hits not assigned to any track in the specified layer
    double nExtraVXDHitsInLayer(const Particle*, const std::vector<double>& layer)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      int ilayer = std::lround(layer[0]);
      return elti->getNVXDClustersInLayer(ilayer);
    }
 
    //  The number of VXD hits not assigned to any track
    double nExtraVXDHits(const Particle*)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      double out = 0.0;
      for (uint16_t ilayer = 1; ilayer < 7; ++ilayer)
        out += elti->getNVXDClustersInLayer(ilayer);
      return out;
    }
 
    //  The number of PXD hits not assigned to any track
    double nExtraPXDHits(const Particle*)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      double out = 0.0;
      for (uint16_t ilayer = 1; ilayer < 3; ++ilayer)
        out += elti->getNVXDClustersInLayer(ilayer);
      return out;
    }
 
    // time of first SVD sample relative to event T0
    double svdFirstSampleTime(const Particle*)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      return elti->getSVDFirstSampleTime();
    }
 
    // A flag set by the tracking if there is reason to assume there was a track
    // in the event missed by the tracking or the track finding was (partly) aborted
    // for this event. Further information about this can be obtained from the flagBlock
    // of the EventLevelTrackingInfo object.
    double trackFindingFailureFlag(const Particle*)
    {
      StoreObjPtr<EventLevelTrackingInfo> elti;
      if (!elti) return Const::doubleNaN;
      return elti->hasAnErrorFlag();
    }
 
    // Build a Helix from a MCParticle's kinematics. Does NOT check for null pointer!!!
    Helix getMCHelix(const MCParticle* mcparticle)
    {
      const ROOT::Math::XYZVector mcProdVertex = mcparticle->getVertex();
      const ROOT::Math::XYZVector mcMomentum = mcparticle->getMomentum();
      const double BzAtProdVertex = BFieldManager::getFieldInTesla(mcProdVertex).Z();
      const double mcParticleCharge = mcparticle->getCharge();
      return Helix(mcProdVertex, mcMomentum, mcParticleCharge, BzAtProdVertex);
    }
 
    double getMCHelixParameterAtIndex(const Particle* particle, const int index)
    {
      if (!particle) return Const::doubleNaN;
 
      const MCParticle* mcparticle = particle->getMCParticle();
      if (!mcparticle) return Const::doubleNaN;
 
      const Helix mcHelix(getMCHelix(mcparticle));
      const std::vector<double> mcHelixPars{mcHelix.getD0(), mcHelix.getPhi0(), mcHelix.getOmega(), mcHelix.getZ0(), mcHelix.getTanLambda()};
      return mcHelixPars.at(index);
    }
 
    double getHelixParameterPullAtIndex(const Particle* particle, const int index)
    {
      if (!particle) return Const::doubleNaN;
 
      const MCParticle* mcparticle = particle->getMCParticle();
      if (!mcparticle) return Const::doubleNaN;
 
      const TrackFitResult* trackfit =  particle->getTrackFitResult();
      if (!trackfit) return Const::doubleNaN;
 
      const UncertainHelix measHelix = trackfit->getUncertainHelix();
      const TMatrixDSym& measCovariance = measHelix.getCovariance();
      const Helix mcHelix(getMCHelix(mcparticle));
 
      const std::vector<double> mcHelixPars   = {mcHelix.getD0(), mcHelix.getPhi0(), mcHelix.getOmega(), mcHelix.getZ0(), mcHelix.getTanLambda()};
      const std::vector<double> measHelixPars = {measHelix.getD0(), measHelix.getPhi0(), measHelix.getOmega(), measHelix.getZ0(), measHelix.getTanLambda()};
 
      return (mcHelixPars.at(index) - measHelixPars.at(index)) / std::sqrt(measCovariance(index, index));
    }
 
    double getHelixMCD0(const Particle* part) { return getMCHelixParameterAtIndex(part, 0); }
    double getHelixMCPhi0(const Particle* part) { return getMCHelixParameterAtIndex(part, 1); }
    double getHelixMCOmega(const Particle* part) { return getMCHelixParameterAtIndex(part, 2); }
    double getHelixMCZ0(const Particle* part) { return getMCHelixParameterAtIndex(part, 3); }
    double getHelixMCTanLambda(const Particle* part) { return getMCHelixParameterAtIndex(part, 4); }
 
    double getHelixD0Pull(const Particle* part)
    {
      return getHelixParameterPullAtIndex(part, 0);
    }
 
    double getHelixPhi0Pull(const Particle* part)
    {
      return getHelixParameterPullAtIndex(part, 1);
    }
 
    double getHelixOmegaPull(const Particle* part)
    {
      return getHelixParameterPullAtIndex(part, 2);
    }
 
    double getHelixZ0Pull(const Particle* part)
    {
      return getHelixParameterPullAtIndex(part, 3);
    }
 
    double getHelixTanLambdaPull(const Particle* part)
    {
      return getHelixParameterPullAtIndex(part, 4);
    }
 
    double getTrackTime(const Particle* part)
    {
      const Track* track = part->getTrack();
      if (!track) return Const::doubleNaN;
      return track->getTrackTime();
    }
 
    double isTrackFlippedAndRefitted(const Particle* part)
    {
      auto track = part->getTrack();
      if (!track) return Const::doubleNaN;
      return track->isFlippedAndRefitted() ? 1 : 0;
    }
 
    double getTrackLength(const Particle* part)
    {
      auto trackFit = part->getTrackFitResult();
      if (!trackFit) return Const::doubleNaN;
 
      const double lastCDCLayer = trackLastCDCLayer(part);
      if (std::isnan(lastCDCLayer) or lastCDCLayer < 0)
        return Const::doubleNaN;
 
      const double r = DetectorSurface::cdcWireRadiuses.at((int)lastCDCLayer);
 
      return trackFit->getHelix().getArcLength2DAtCylindricalR(r);
    }
 
 
    VARIABLE_GROUP("Tracking");
 
    REGISTER_VARIABLE("mcD0", getHelixMCD0, R"DOC(
Returns the MC value of :math:`d_0`, the signed distance to the
point-of-closest-approach (POCA) in the :math:`r-\phi` plane.
 
.. seealso:: :b2:var:`d0`
 
Returns NaN if the particle is not related to any MCParticle.
 
)DOC", "cm");
    REGISTER_VARIABLE("mcPhi0", getHelixMCPhi0, R"DOC(
Returns the MC value of :math:`\phi_0`, the angle of the transverse momentum
in the :math:`r-\phi` plane.
 
.. seealso:: :b2:var:`phi0`
 
Returns NaN if the particle is not related to any MCParticle.
 
)DOC", "rad");
    REGISTER_VARIABLE("mcOmega", getHelixMCOmega, R"DOC(
Returns the MC value of :math:`\omega`, the curvature of the track.
 
.. seealso:: :b2:var:`omega`
 
Returns NaN if the particle is not related to any MCParticle.
 
)DOC", ":math:`\\text{cm}^{-1}`");
    REGISTER_VARIABLE("mcZ0", getHelixMCZ0, R"DOC(
Returns the MC value of :math:`z_0`, the z-coordinate of the
point-of-closest-approach (POCA).
 
.. seealso:: :b2:var:`z0`
 
Returns NaN if the particle is not related to any MCParticle.
 
)DOC", "cm");
    REGISTER_VARIABLE("mcTanLambda", getHelixMCTanLambda, R"DOC(
Returns the MC value of :math:`\tan\lambda`, the slope of the track in the
:math:`r-z` plane.
 
.. seealso:: :b2:var:`tanLambda`
 
Returns NaN if the particle is not related to any MCParticle.
)DOC");
 
    REGISTER_VARIABLE("d0Pull", getHelixD0Pull,     R"DOC(
The pull of the tracking parameter :math:`d_0` for the reconstructed
pattern-recognition track, with respect to the MC track. That is:
 
.. math::
 
      \frac{d_0^\textrm{MC} - d_0^\textrm{PR}}{\sigma_{d_0; \textrm{PR}}}
 
.. seealso:: :b2:var:`d0`, :b2:var:`d0Err`
 
Returns NaN if no MC particle is related or if called on something other than a
track-based particle.
    )DOC");
    REGISTER_VARIABLE("phi0Pull", getHelixPhi0Pull, R"DOC(
The pull of the tracking parameter :math:`\phi_0` for the reconstructed
pattern-recognition track, with respect to the MC track.  That is:
 
.. math::
 
      \frac{\phi_0^\textrm{MC} - \phi_0^\textrm{PR}}{\sigma_{\phi_0; \textrm{PR}}}
 
.. seealso:: :b2:var:`phi0`, :b2:var:`phi0Err`
 
Returns NaN if no MC particle is related or if called on something other than a
track-based particle.
    )DOC");
    REGISTER_VARIABLE("omegaPull", getHelixOmegaPull, R"DOC(
The pull of the tracking parameter :math:`\omega` for the reconstructed
pattern-recognition track, with respect to the MC track.  That is:
 
.. math::
 
      \frac{\omega^\textrm{MC} - \omega^\textrm{PR}}{\sigma_{\omega; \textrm{PR}}}
 
.. seealso:: :b2:var:`omega`, :b2:var:`omegaErr`
 
Returns NaN if no MC particle is related or if called on something other than a
track-based particle.
    )DOC");
    REGISTER_VARIABLE("z0Pull", getHelixZ0Pull, R"DOC(
The pull of the tracking parameter :math:`z_0` for the reconstructed
pattern-recognition track, with respect to the MC track.  That is:
 
.. math::
 
      \frac{z_0^\textrm{MC} - z_0^\textrm{PR}}{\sigma_{z_0; \textrm{PR}}}
 
.. seealso:: :b2:var:`z0`, :b2:var:`z0Err`
 
Returns NaN if no MC particle is related or if called on something other than a
track-based particle.
    )DOC");
    REGISTER_VARIABLE("tanLambdaPull", getHelixTanLambdaPull, R"DOC(
The pull of the tracking parameter :math:`\tan\lambda` for the reconstructed
pattern-recognition track, with respect to the MC track.  That is:
 
.. math::
 
      \frac{(\tan\lambda)^\textrm{MC} - (\tan\lambda)^\textrm{PR}}{\sigma_{\tan\lambda; \textrm{PR}}}
 
.. seealso:: :b2:var:`tanLambda`, :b2:var:`tanLambdaErr`
 
Returns NaN if no MC particle is related or if called on something other than a
track-based particle.
    )DOC");
    REGISTER_VARIABLE("nCDCHits", trackNCDCHits,
                      "The number of CDC hits associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("nSVDHits", trackNSVDHits,
                      "The number of SVD hits associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("nPXDHits", trackNPXDHits,
                      "The number of PXD hits associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("nVXDHits", trackNVXDHits,
                      "The number of PXD and SVD hits associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("ndf",      trackNDF, R"DOC(
Returns the number of degrees of freedom of the track fit.
 
.. note::
 
        Note that this is not simply the number of hits -5 due to outlier hit
        rejection.
 
Returns NaN if called for something other than a track-based particle, or for
mdst files processed with basf2 versions older than ``release-05-01``.
    )DOC");
    REGISTER_VARIABLE("chi2",      trackChi2, R"DOC(
Returns the :math:`\chi^2` of the track fit.  This is actually computed based on
:b2:var:`pValue` and :b2:var:`ndf`.
 
.. note:: Note that for :b2:var:`pValue` exactly equal to 0 it returns infinity.
 
Returns NaN if called for something other than a track-based particle, or for
mdst files processed with basf2 versions older than ``release-05-01``.
    )DOC");
    REGISTER_VARIABLE("firstSVDLayer", trackFirstSVDLayer,
                      "The first activated SVD layer associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("firstPXDLayer", trackFirstPXDLayer,
                      "The first activated PXD layer associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("firstCDCLayer", trackFirstCDCLayer,
                      "The first activated CDC layer associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("lastCDCLayer", trackLastCDCLayer,
                      "The last CDC layer associated to the track. Returns NaN if called for something other than a track-based particle.");
    REGISTER_VARIABLE("d0", trackD0, R"DOC(
Returns the tracking parameter :math:`d_0`, the signed distance to the
point-of-closest-approach (POCA) in the :math:`r-\phi` plane.
 
.. note::
 
        Tracking parameters are with respect to the origin (0,0,0). For the
        POCA with respect to the measured beam interaction point, see
        :b2:var:`dr` (you probably want this unless you're doing a tracking
        study or some debugging) and :b2:var:`d0FromIP`.
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "cm");
    REGISTER_VARIABLE("phi0", trackPhi0, R"DOC(
Returns the tracking parameter :math:`\phi_0`, the angle of the transverse
momentum in the :math:`r-\phi` plane.
 
.. note::
 
        Tracking parameters are with respect to the origin (0,0,0). For the
        POCA with respect to the measured beam interaction point, see
        :b2:var:`phi0FromIP`.
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "rad");
    REGISTER_VARIABLE("omega", trackOmega, R"DOC(
Returns the tracking parameter :math:`\omega`, the curvature of the track.
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", ":math:`\\text{cm}^{-1}`");
    REGISTER_VARIABLE("z0", trackZ0, R"DOC(
Returns the tracking parameter :math:`z_0`, the z-coordinate of the
point-of-closest-approach (POCA).
 
.. note::
 
        Tracking parameters are with respect to the origin (0,0,0). For the
        POCA with respect to the measured beam interaction point, see
        :b2:var:`dz` (you probably want this unless you're doing a tracking
        study or some debugging) and :b2:var:`z0FromIP`.
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "cm");
    REGISTER_VARIABLE("tanLambda", trackTanLambda, R"DOC(
Returns :math:`\tan\lambda`, the slope of the track in the :math:`r-z` plane.
 
Returns NaN if called for something other than a track-based particle.
    )DOC");
    REGISTER_VARIABLE("d0FromIP", trackD0FromIP, R"DOC(
Returns the tracking parameter :math:`d_0`, the signed distance to the
point-of-closest-approach (POCA) in the :math:`r-\phi` plane, with respect to the measured beam interaction point.
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "cm");
    REGISTER_VARIABLE("z0FromIP", trackZ0FromIP, R"DOC(
Returns the tracking parameter :math:`z_0`, the z-coordinate of the
point-of-closest-approach (POCA), with respect to the measured beam interaction point.
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "cm");
    REGISTER_VARIABLE("phi0FromIP", trackPhi0FromIP, R"DOC(
Returns the tracking parameter :math:`\phi_0`, the angle of the transverse
momentum in the :math:`r-\phi` plane, with respect to the measured beam interaction point.
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "rad");
    REGISTER_VARIABLE("d0Err", trackD0Error, R"DOC(
Returns the uncertainty on :math:`d_0`, the signed distance to the
point-of-closest-approach (POCA) in the :math:`r-\phi` plane.
 
.. seealso:: :b2:var:`d0`, :b2:var:`d0Pull`
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "cm");
    REGISTER_VARIABLE("phi0Err", trackPhi0Error, R"DOC(
Returns the uncertainty on :math:`\phi_0`, the angle of the transverse momentum
in the :math:`r-\phi` plane.
 
.. seealso:: :b2:var:`phi0`, :b2:var:`phi0Pull`
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", "rad");
    REGISTER_VARIABLE("omegaErr", trackOmegaError, R"DOC(
Returns the uncertainty on :math:`\omega`, the curvature of the track.
 
.. seealso:: :b2:var:`omega`, :b2:var:`omegaPull`
 
Returns NaN if called for something other than a track-based particle.
 
)DOC", ":math:`\\text{cm}^{-1}`");
    REGISTER_VARIABLE("z0Err", trackZ0Error, R"DOC(
Returns the uncertainty on :math:`z_0`, the z-coordinate of the
point-of-closest-approach (POCA).
 
.. seealso:: :b2:var:`z0`, :b2:var:`z0Pull`
 
Returns NaN if called for something other than a track-based particle."
 
)DOC", "cm");
    REGISTER_VARIABLE("tanLambdaErr", trackTanLambdaError, R"DOC(
Returns the uncertainty on :math:`\tan\lambda`, the slope of the track in the
:math:`r-z` plane.
 
.. seealso:: :b2:var:`tanLambda`, :b2:var:`tanLambdaPull`
 
Returns NaN if called for something other than a track-based particle.
    )DOC");
    REGISTER_VARIABLE("trackFitCovariance(i, j)", trackFitCovariance, R"DOC(
      The track fit covariance matrix element corresponding to the two indices is returned.
      This is the association between integers and parameters:
 
      * 0: :math:`d_0`
      * 1: :math:`\phi_0`
      * 2: :math:`\omega`
      * 3: :math:`z_0`
      * 4: :math:`\tan\lambda`
 
      .. note::
 
              The covariance is returned. This means that the return value can be negative.
              Furthermore, it's the squared value of the track fit error variables :b2:var:`d0Err`, etc.
              when selecting the diagonal entries.
 
      )DOC");
    REGISTER_VARIABLE("pValue", trackPValue, R"DOC(
The :math:`\chi^2` probability of the **track** fit.
 
.. note::
 
        This is the p-value of the track-fit. It does not get updated after
        vertex fitting or kinematic fitting, and is meaningless for composite
        particles.
 
        See :b2:var:`chiProb` (you probably want this for high-level analysis).
 
Returns NaN if called for something other than a track-based particle.
    )DOC");
    REGISTER_VARIABLE("trackFitHypothesisPDG", trackFitHypothesisPDG, R"DOC(
Returns the PDG code of the track hypothesis actually used for the fit.
Returns NaN if called for something other than a track-based particle.
    )DOC");
    REGISTER_VARIABLE("trackNECLClusters", trackNECLClusters, R"DOC(
Returns a count of the number of ECLClusters matched to the track.  This is
always 0 or 1 with newer versions of ECL reconstruction.
 
.. note::
 
        For high-level analysis it is recommended to require the presence of a
        matched ECL cluster along with a minimum energy requirement.  A
        track-based particle will have a clusterE if it is matched (NaN if
        there is no cluster match for the track.
 
        .. code-block:: python
 
              import modularAnalysis as ma
              # minimum energy of 200 MeV
              ma.fillParticleList("e+:clusters", "clusterE > 0.2", path)
 
              # these two are equivalent
              ma.fillParticleList("e+:unmatched", "isNAN(clusterE) == 1", path)
              ma.fillParticleList("e+:unmatched2", "trackNECLClusters == 0", path)
 
Returns NaN if called for something other than a track-based particle.
    )DOC");
    REGISTER_VARIABLE("helixExtTheta(radius [cm], z fwd [cm], z bwd [cm], useHighestProbMass=0)", trackHelixExtTheta, R"DOC(
                      Returns theta of extrapolated helix parameters. If ``useHighestProbMass=1`` is set, the extrapolation will
                      use the track fit result for the mass hypothesis with the highest pValue.
 
                      )DOC", "rad");
    REGISTER_VARIABLE("helixExtPhi(radius, z fwd, z bwd, useHighestProbMass=0)", trackHelixExtPhi,
                      "Returns phi of extrapolated helix parameters. If ``useHighestProbMass=1`` is set, the extrapolation will use the track fit result for the mass hypothesis with the highest pValue.\n\n",
                      "rad");
 
    REGISTER_METAVARIABLE("helixExtThetaOnDet(detector_surface_name, useHighestProbMass=0)", trackHelixExtThetaOnDet, R"DOC(
                          Returns theta of extrapolated helix parameters on the given detector surface. The unit of angle is ``rad``.
                          If ``useHighestProbMass=1`` is set, the extrapolation will use the track fit result for the mass hypothesis with the highest pValue.
                          The supported detector surface names are ``{'CDC', 'TOP', 'ARICH', 'ECL', 'KLM'}``.
                          Also, the detector name with number of meaningful-layer is supported, e.g. ``'CDC8'``: last superlayer of CDC, ``'ECL1'``: mid-point of ECL.
 
                          .. note:: You can find more information in `modularAnalysis.calculateTrackIsolation`.
                          )DOC", Manager::VariableDataType::c_double);
    REGISTER_METAVARIABLE("helixExtPhiOnDet(detector_surface_name, useHighestProbMass=0)", trackHelixExtPhiOnDet, R"DOC(
                          Returns phi of extrapolated helix parameters on the given detector surface. The unit of angle is ``rad``.
                          If ``useHighestProbMass=1`` is set, the extrapolation will use the track fit result for the mass hypothesis with the highest pValue.
                          The supported detector surface names are ``{'CDC', 'TOP', 'ARICH', 'ECL', 'KLM'}``.
                          Also, the detector name with number of meaningful-layer is supported, e.g. ``'CDC8'``: last superlayer of CDC, ``'ECL1'``: mid-point of ECL.
 
                          .. note:: You can find more information in `modularAnalysis.calculateTrackIsolation`.
                          )DOC", Manager::VariableDataType::c_double);
 
 
    REGISTER_VARIABLE("nExtraCDCHits", nExtraCDCHits, R"DOC(
[Eventbased] The number of CDC hits in the event not assigned to any track.
 
Returns NaN if there is no event-level tracking information available.
    )DOC");
    REGISTER_VARIABLE("nExtraCDCHitsPostCleaning", nExtraCDCHitsPostCleaning, R"DOC(
[Eventbased] Returns a count of the number of CDC hits in the event not assigned
to any track nor very likely beam background (i.e. hits that survive a cleanup
selection).
 
Returns NaN if there is no event-level tracking information available.
    )DOC");
    REGISTER_VARIABLE("hasExtraCDCHitsInLayer(i)", hasExtraCDCHitsInLayer, R"DOC(
[Eventbased] Returns 1 if a non-assigned hit exists in the specified CDC layer,
0 otherwise.
 
Returns NaN if there is no event-level tracking information available.
    )DOC");
    REGISTER_VARIABLE("hasExtraCDCHitsInSuperLayer(i)", hasExtraCDCHitsInSuperLayer, R"DOC(
[Eventbased] Returns 1 if a non-assigned hit exists in the specified CDC
SuperLayer, 0 otherwise.
 
Returns NaN if there is no event-level tracking information available.
    )DOC");
    REGISTER_VARIABLE("nExtraCDCSegments", nExtraCDCSegments, R"DOC(
[Eventbased] Returns the number of CDC segments not assigned to any track.
 
Returns NaN if there is no event-level tracking information available.
    )DOC");
    // TODO: once the Tracking group fill the dataobject these can be
    // uncommented - at the moment they are not filled, so leave out
    //REGISTER_VARIABLE("nExtraVXDHitsInLayer(i)", nExtraVXDHitsInLayer,
    //"[Eventbased] The number VXD hits not assigned in the specified VXD layer");
    //REGISTER_VARIABLE("nExtraVXDHits", nExtraVXDHits, "[Eventbased] The number of VXD hits not assigned to any track");
    REGISTER_VARIABLE("nExtraPXDHits", nExtraPXDHits, "[Eventbased] The number of PXD hits not assigned to any track");
    //REGISTER_VARIABLE("svdFirstSampleTime", svdFirstSampleTime, "[Eventbased] The time of first SVD sample relatvie to event T0");
    REGISTER_VARIABLE("trackFindingFailureFlag", trackFindingFailureFlag, R"DOC(
[Eventbased] Returns a flag set by the tracking if there is reason to assume
there was a track in the event missed by the tracking, or the track finding was
(partly) aborted for this event.
 
Returns NaN if there is no event-level tracking information available.
    )DOC");
 
    REGISTER_VARIABLE("isTrackFlippedAndRefitted", isTrackFlippedAndRefitted, R"DOC(
Returns 1 if the charged final state particle comes from a track that has been flipped and refitted
at the end of the tracking chain, in particular before the outer detector reconstruction.
    )DOC");
 
    REGISTER_VARIABLE("trackTime", getTrackTime, R"DOC(
Returns the time at which the track is produced relative to the time of the collision (given by SVD EventT0).
Both the time of the collision and the track time are computed using only SVD hits.
Returns NaN if SVD EventT0 is NaN, or if no SVD Hits are attached to the track.
For more details, see :ref:`Time Extraction <tracking_eventTimeExtraction>` page.
 
)DOC", "ns");
 
    REGISTER_VARIABLE("trackLength", getTrackLength, R"DOC(
Returns the arc length of the helix for the TrackFitResult associated with the particle.
The arc length is measured from the track origin to the radius of the CDC layer in which the Track has a hit.
Returns NaN if the particle has no CDC Hits.
 
)DOC", "cm");
 
 
  }
}