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