TrackBuilder Class Reference

TrackBuilder class to create the Track/TrackFitResult mdst output from the RecoTrack. More...

#include <TrackBuilder.h>

Collaboration diagram for TrackBuilder:

Public Member Functions
	TrackBuilder (const std::string &trackColName, const std::string &trackFitResultColName, const std::string &mcParticleColName, const TVector3 &beamSpot=TVector3(0., 0., 0.), const TVector3 &beamAxis=TVector3(0., 0., 1.))
	Constructor of the class. More...
bool	storeTrackFromRecoTrack (RecoTrack &recoTrack, const bool useClosestHitToIP=false, const bool useBFieldAtHit=false)
	Stores a Belle2 Track from a Reco Track. More...

Static Public Member Functions
static uint32_t	getHitPatternVXDInitializer (const RecoTrack &recoTrack)
	Get the HitPattern in the VXD.
static uint64_t	getHitPatternCDCInitializer (const RecoTrack &recoTrack)
	Get the HitPattern in the CDC.

Private Attributes
std::string	m_trackColName
	TrackColName (output).
std::string	m_trackFitResultColName
	TrackFitResultColName (output).
std::string	m_mcParticleColName
	MCParticleColName (input, optional).
TVector3	m_beamSpot
	Extrapolation target, origin.
TVector3	m_beamAxis
	Extrapolation target, positive z direction.

Detailed Description

TrackBuilder class to create the Track/TrackFitResult mdst output from the RecoTrack.

To use this class, iterate over all Reco tracks and call trackBuilder.storeTrackFromRecoTrack. All fitted hypotheses will be stored into one Track with the TrackFitResult array indices stored in the Track. for (auto& recoTrack : recoTracks) { trackBuilder.storeTrackFromRecoTrack(recoTrack); }

If you want to make sure that all hypotheses are fitted and stored, call the fitter again. If the tracks are fitted already, this produces no overhead. TrackFitter trackFitter; TrackBuilder trackBuilder; for (auto& recoTrack : recoTracks) { for (const auto& pdg : m_pdgCodes) { trackFitter.fit(recoTrack, Const::ParticleType(abs(pdg))); } trackBuilder.storeTrackFromRecoTrack(recoTrack); }

Definition at line 40 of file TrackBuilder.h.

Constructor & Destructor Documentation

TrackBuilder()

TrackBuilder ( const std::string &  trackColName, const std::string &  trackFitResultColName, const std::string &  mcParticleColName, const TVector3 &  beamSpot = TVector3(0., 0., 0.), const TVector3 &  beamAxis = TVector3(0., 0., 1.)  )

inline

Constructor of the class.

Parameters

trackColName	Name of the store array for tracks (output).
trackFitResultColName	Name of the store array for track fit results (output).
mcParticleColName	Name of the store array for MC particles (input, optional). If given, the tracks are matched to MCParticles.
beamSpot	Origin.
beamAxis	Positive z-direction.

Definition at line 51 of file TrackBuilder.h.

      :
      m_trackColName(trackColName),
      m_trackFitResultColName(trackFitResultColName),
      m_mcParticleColName(mcParticleColName),
      m_beamSpot(beamSpot),
      m_beamAxis(beamAxis)
    {};

Member Function Documentation

storeTrackFromRecoTrack()

bool storeTrackFromRecoTrack	(	RecoTrack &	recoTrack,
		const bool	useClosestHitToIP = false,
		const bool	useBFieldAtHit = false
	)

Stores a Belle2 Track from a Reco Track.

Every fitted hypothesis will be extrapolated to the perigee and stored as a TrackFitResult when the fit and the extrapolation were successful. We will only create a Track mdst object, when we are sure that we have at least one valid hypothesis available. If we do not have this, we discard the track no matter what.

The StoreArrayIndex is stored in the Belle2 Track, no relation is set.

Parameters

recoTrack	The reco track the fit results are stored for
useClosestHitToIP	Flag to turn on special handling which measurement to choose; especially useful for Cosmics
useBFieldAtHit	Flag to calculate the BField at the used hit (closest to IP or first one), instead of the one at the POCA. This is also useful for cosmics only.

Returns

Definition at line 35 of file TrackBuilder.cc.

{
  StoreArray<Track> tracks(m_trackColName);
  StoreArray<TrackFitResult> trackFitResults(m_trackFitResultColName);
 
  const auto& trackReps = recoTrack.getRepresentations();
  B2DEBUG(100, trackReps.size() << " track representations available.");
  Track newTrack(recoTrack.getQualityIndicator());
 
  bool repAlreadySet = false;
  unsigned int repIDPlusOne = 0;
  for (const auto& trackRep : trackReps) {
    repIDPlusOne++;
 
    // Check if the fitted particle type is in our charged stable set.
    const Const::ParticleType particleType(std::abs(trackRep->getPDG()));
    if (not Const::chargedStableSet.contains(particleType)) {
      B2DEBUG(100, "Track fitted with hypothesis that is not a ChargedStable (PDG code = " << particleType.getPDGCode() << ")");
      continue;
    }
 
    // Check if the fit worked.
    if (not recoTrack.wasFitSuccessful(trackRep)) {
      B2DEBUG(100, "The fit with the given track representation (" << std::abs(trackRep->getPDG()) <<
              ") was not successful. Skipping ...");
      continue;
    }
 
    if (not repAlreadySet) {
      RecoTrackGenfitAccess::getGenfitTrack(recoTrack).setCardinalRep(repIDPlusOne - 1);
      repAlreadySet = true;
    }
 
    // Extrapolate the tracks to the perigee.
    genfit::MeasuredStateOnPlane msop;
    try {
      if (useClosestHitToIP) {
        msop = recoTrack.getMeasuredStateOnPlaneClosestTo(TVector3(0, 0, 0), trackRep);
      } else {
        msop = recoTrack.getMeasuredStateOnPlaneFromFirstHit(trackRep);
      }
    } catch (genfit::Exception& exception) {
      B2WARNING(exception.what());
      continue;
    }
 
    genfit::MeasuredStateOnPlane extrapolatedMSoP = msop;
    try {
      extrapolatedMSoP.extrapolateToLine(m_beamSpot, m_beamAxis);
    } catch (...) {
      B2WARNING("Could not extrapolate the fit result for pdg " << particleType.getPDGCode() <<
                " to the perigee point. Why, I don't know.");
      continue;
    }
 
    // Build track fit result.
 
    TVector3 poca(0., 0., 0.);
    TVector3 dirInPoca(0., 0., 0.);
    TMatrixDSym cov(6);
    extrapolatedMSoP.getPosMomCov(poca, dirInPoca, cov);
    B2DEBUG(149, "Point of closest approach: " << poca.x() << "  " << poca.y() << "  " << poca.z());
    B2DEBUG(149, "Track direction in POCA: " << dirInPoca.x() << "  " << dirInPoca.y() << "  " << dirInPoca.z());
 
    const int charge = recoTrack.getTrackFitStatus(trackRep)->getCharge();
    const double pValue = recoTrack.getTrackFitStatus(trackRep)->getPVal();
    const int nDF = recoTrack.getTrackFitStatus(trackRep)->getNdf();
 
    double Bx, By, Bz;  // In cgs units
    if (useBFieldAtHit) {
      const TVector3& hitPosition = msop.getPos();
      genfit::FieldManager::getInstance()->getFieldVal(hitPosition.X(), hitPosition.Y(), hitPosition.Z(), Bx, By, Bz);
    } else {
      genfit::FieldManager::getInstance()->getFieldVal(poca.X(), poca.Y(), poca.Z(), Bx, By, Bz);
    }
    Bz = Bz / 10.; // In SI-Units
 
    const uint64_t hitPatternCDCInitializer = getHitPatternCDCInitializer(recoTrack);
    const uint32_t hitPatternVXDInitializer = getHitPatternVXDInitializer(recoTrack);
 
    const auto newTrackFitResult = trackFitResults.appendNew(
                                     poca, dirInPoca, cov, charge, particleType, pValue, Bz,
                                     hitPatternCDCInitializer, hitPatternVXDInitializer, nDF
                                   );
 
    const int newTrackFitResultArrayIndex = newTrackFitResult->getArrayIndex();
    newTrack.setTrackFitResultIndex(particleType, newTrackFitResultArrayIndex);
  }
 
  B2DEBUG(100, "Number of fitted hypothesis = " << newTrack.getNumberOfFittedHypotheses());
  if (newTrack.getNumberOfFittedHypotheses() > 0) {
    Track* addedTrack = tracks.appendNew(newTrack);
    addedTrack->addRelationTo(&recoTrack);
    const auto& mcParticleWithWeight = recoTrack.getRelatedToWithWeight<MCParticle>(m_mcParticleColName);
    const MCParticle* mcParticle = mcParticleWithWeight.first;
    if (mcParticle) {
      B2DEBUG(200, "Relation to MCParticle set.");
      addedTrack->addRelationTo(mcParticle, mcParticleWithWeight.second);
    } else {
      B2DEBUG(200, "Relation to MCParticle not set. No related MCParticle to RecoTrack.");
    }
    return true;
  } else {
    B2DEBUG(200, "Relation to MCParticle not set. No related MCParticle to RecoTrack.");
  }
  return true;
}

---

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

• tracking/trackFitting/trackBuilder/factories/include/TrackBuilder.h
• tracking/trackFitting/trackBuilder/factories/src/TrackBuilder.cc