GeoTrajectory.h

Go to the documentation of this file.

#ifndef BASICTOOL_GEOTRAJECTORY_H
#define BASICTOOL_GEOTRAJECTORY_H

#include "larcorealg/GeoAlgo/GeoVector.h"

#include <ostream>
#include <stddef.h>
#include <vector>

namespace geoalgo {

  class Trajectory : public std::vector<geoalgo::Vector> {

  public:
    Trajectory(size_t npoints = 0, size_t ndimension = 0);

    virtual ~Trajectory() {}

    Trajectory(const std::vector<std::vector<double>>& obj);

    Trajectory(const std::vector<geoalgo::Point_t>& obj);

    //
    // Getters
    //
    double Length(size_t start_step = 0,
                  size_t end_step = 0) const; 
    bool IsLonger(double) const;    
    Vector Dir(size_t i = 0) const; 

    //
    // Setters
    //
    void push_back(const Point_t& obj); 

    inline Trajectory& operator+=(const Point_t& rhs)
    {
      push_back(rhs);
      return *this;
    }

    //
    // utility
    //
    void compat(const Point_t& obj) const;    
    void compat(const Trajectory& obj) const; 

  protected:
    Vector _Dir_(size_t i) const;

  public:
    //
    // templates
    //
    template <class T>
    void push_back(const T& obj)
    {
      Point_t pt(obj);
      push_back(pt);
    }

  public:
#ifndef __CINT__
    friend std::ostream& operator<<(std::ostream& o, Trajectory const& a)
    {
      o << "Trajectory with " << a.size() << " points " << std::endl;
      for (auto const& p : a)
        o << " " << p << std::endl;
      return o;
    }
#endif
  };

  typedef Trajectory Trajectory_t;

}

#endif
 // end of doxygen group