KFTrackState.h

Go to the documentation of this file.

#ifndef KFTRACKSTATE_H
#define KFTRACKSTATE_H

#include "lardata/RecoObjects/TrackState.h"

namespace trkf {


  class KFTrackState {
  public:
    //
  KFTrackState(const SVector5& trackStatePar, const SMatrixSym55& trackStateCov, const Plane& plane, bool trackAlongPlaneDir, int pid)
    : fTrackState(trackStatePar, trackStateCov, plane, trackAlongPlaneDir, pid) {}
  KFTrackState(TrackState&& trackState)
    : fTrackState(std::move(trackState)) { }

    bool updateWithHitState(const HitState& hitstate);

    bool combineWithTrackState(const TrackState& trackstate);

    const TrackState& trackState() const { return fTrackState; }
    void setTrackState(TrackState&& s) { fTrackState = std::move(s); }
    //
    const SVector5&     parameters()           const { return fTrackState.parameters(); }
    const SMatrixSym55& covariance()           const { return fTrackState.covariance(); }
    const Plane&        plane()                const { return fTrackState.plane(); }
    const Point_t&      position()             const { return fTrackState.position(); }
    const Vector_t&     momentum()             const { return fTrackState.momentum(); }
    int                 pID()                  const { return fTrackState.pID(); }
    double              mass()                 const { return fTrackState.mass(); }
    const SVector6      parameters6D()         const { return fTrackState.parameters6D(); }
    bool                isTrackAlongPlaneDir() const { return fTrackState.isTrackAlongPlaneDir(); }
    //
    double              residual      (const HitState& hitstate) const { return fTrackState.residual(hitstate); }
    double              combinedError2(const HitState& hitstate) const { return fTrackState.combinedError2(hitstate); }
    double              combinedError (const HitState& hitstate) const { return fTrackState.combinedError(hitstate); }
    double              chi2          (const HitState& hitstate) const { return fTrackState.chi2(hitstate); }
    //
    void                setCovariance (const SMatrixSym55& trackStateCov) { fTrackState.setCovariance(trackStateCov); }
    void                setParameters (const SVector5&     trackStatePar) { fTrackState.setParameters(trackStatePar); }
    //
    std::ostream& dump(std::ostream& out = std::cout) const {
      out << "KFTrackState with pID=" << pID() << " mass=" << mass()
          << "\npars=" << parameters() << " position=" << position() << " momentum=" << momentum()
          << "\ncov=\n" << covariance()
          << "\non plane with pos=" << plane().position() << " and dir=" << plane().direction() << " along=" << isTrackAlongPlaneDir() << "\n";
      return out;
    }
    //
  private:
    TrackState fTrackState;
  };

}
#endif