RecoTrackStateProvider.h

Go to the documentation of this file.

// ******************************************************
// DecayTreeFitter Package
// We thank the original author Wouter Hulsbergen
// (BaBar, LHCb) for providing the sources.
// http://arxiv.org/abs/physics/0503191v1 (2005)
// Adaptation & Development for PANDA: Ralf Kliemt (2015)
// ******************************************************
#ifndef DECAYTREEFITTER_RECOTRACKSTATEPROVIDER_H
#define DECAYTREEFITTER_RECOTRACKSTATEPROVIDER_H 1

//#include "TrackInterfaces/ITrackStateProvider.h"
//#include "TrackKernel/TrackTraj.h"
// FIXME Make a Panda state provider...
#include "RhoCalculationTools.h"
//#include "LineTool.h"
#include "State.h"
#include "Rtypes.h"
#include "RhoCandidate.h"
#include "TVectorD.h"
#include "TMatrixDSym.h"

namespace DecayTreeFitter {
// wrapper around ITrackStateProviderTool
class RecoTrackStateProvider {
 public:
  RecoTrackStateProvider(double ztolerance) : m_ztolerance(ztolerance){};

  RecoTrackStateProvider() : m_ztolerance(0){};

  virtual ~RecoTrackStateProvider(){};

  void state(DecayTreeFitter::State &aState, RhoCandidate *track) const { return state(aState, track, track->Pos().x(), track->Pos().y(), track->Pos().z()); };

  double ToleranceZ() const { return m_ztolerance; };
  void SetToleranceZ(double z) { m_ztolerance = z; };

  void state(DecayTreeFitter::State &aState, RhoCandidate *track, double vx, double vy, double vz) const
  {
    TVectorD statevec = aState.stateVector();
    TMatrixDSym statecov = aState.covariance();
    double newz = RhoCalculationTools::StateFromTrajectory(statevec, statecov, track, vx, vy, vz, m_ztolerance);
    aState.setState(statevec);
    aState.setZ(newz);
    aState.setCovariance(statecov);
  };

  void P6FromTrack(TVectorD &p6, TMatrixDSym &cov6, RhoCandidate *track, double z) const { RhoCalculationTools::P6FromTrajectory(p6, cov6, track, z, m_ztolerance); };

 private:
  double m_ztolerance;
  bool m_usetraj;
  ClassDef(RecoTrackStateProvider, 1)
};
} // namespace DecayTreeFitter

#endif