KTrack.h

Go to the documentation of this file.

#ifndef KTRACK_H
#define KTRACK_H

#include "lardata/RecoObjects/KalmanLinearAlgebra.h"
#include "lardata/RecoObjects/Surface.h"
#include <iosfwd>
#include <memory>

namespace trkf {

  class KTrack {
  public:

    KTrack();

    KTrack(const std::shared_ptr<const Surface>& psurf);

    KTrack(std::shared_ptr<const Surface> psurf,
           const TrackVector& vec,
           Surface::TrackDirection dir = Surface::UNKNOWN,
           int pdg = 0);

    virtual ~KTrack();

    // Accessors.

    const std::shared_ptr<const Surface>& getSurface() const { return fSurf; } 
    const TrackVector& getVector() const { return fVec; } 
    Surface::TrackDirection getDirection() const;         
    int PdgCode() const { return fPdgCode; }              
    double Mass() const;                                  

    // Modifiers.

    TrackVector& getVector() { return fVec; } 

    void setSurface(const std::shared_ptr<const Surface>& psurf) { fSurf = psurf; }
    void setVector(const TrackVector& vec) { fVec = vec; }         
    void setDirection(Surface::TrackDirection dir) { fDir = dir; } 
    void setPdgCode(int pdg) { fPdgCode = pdg; }                   

    bool isValid() const;

    void getPosition(double xyz[3]) const;

    void getMomentum(double mom[3]) const;

    double XLatitude() const;

    double XLongitude() const;

    virtual std::ostream& Print(std::ostream& out, bool doTitle = true) const;

  private:
    // Attributes.

    std::shared_ptr<const Surface> fSurf; 
    TrackVector fVec;                     
    Surface::TrackDirection fDir;         
    int fPdgCode;                         
  };

  std::ostream& operator<<(std::ostream& out, const KTrack& trk);
}

#endif