odometry.h

Go to the documentation of this file.

#ifndef ODOMETRY_H_
#define ODOMETRY_H_

#include <ros/time.h>
#include <boost/accumulators/accumulators.hpp>
#include <boost/accumulators/statistics/stats.hpp>
#include <boost/accumulators/statistics/rolling_mean.hpp>
#include <boost/function.hpp>

namespace ackermann_controller
{
  namespace bacc = boost::accumulators;

  class Odometry
  {
  public:

    typedef boost::function<void(double, double)> IntegrationFunction;

    Odometry(size_t velocity_rolling_window_size = 10);

    void init(const ros::Time &time);

    bool update(double front_wheel_angular_pos, double front_wheel_angular_vel,
                double rear_wheel_angular_pos, double rear_wheel_angular_vel,
                double front_steering, const ros::Time &time);

    void updateOpenLoop(double linear, double angular, const ros::Time &time);

    double getHeading() const
    {
      return heading_;
    }

    double getX() const
    {
      return x_;
    }

    double getY() const
    {
      return y_;
    }

    double getLinear() const
    {
      return linear_;
    }

    double getAngular() const
    {
      return angular_;
    }

    void setWheelParams(double track, double front_wheel_radius, double rear_wheel_radius, double wheel_base);

    void setVelocityRollingWindowSize(size_t velocity_rolling_window_size);

  private:

    typedef bacc::accumulator_set<double, bacc::stats<bacc::tag::rolling_mean> > RollingMeanAcc;
    typedef bacc::tag::rolling_window RollingWindow;

    void integrateRungeKutta2(double linear, double angular);

    void integrateExact(double linear, double angular);

    void resetAccumulators();

    ros::Time timestamp_, last_update_timestamp_;

    double x_;        //   [m]
    double y_;        //   [m]
    double heading_;  // [rad]

    double linear_;  //   [m/s]
    double angular_; // [rad/s]

    double track_;
    double front_wheel_radius_, rear_wheel_radius_;
    double wheel_base_;

    double left_wheel_old_pos_;
    double right_wheel_old_pos_;
    double wheel_old_pos_;

    size_t velocity_rolling_window_size_;
    RollingMeanAcc linear_acc_;
    RollingMeanAcc angular_acc_;
  };
}

#endif /* ODOMETRY_H_ */