four_wheel_steering_controller.h

Go to the documentation of this file.

#include <controller_interface/controller_base.h>
#include <hardware_interface/joint_command_interface.h>
#include <pluginlib/class_list_macros.h>

#include <nav_msgs/Odometry.h>
#include <four_wheel_steering_msgs/FourWheelSteering.h>
#include <tf/tfMessage.h>

#include <realtime_tools/realtime_buffer.h>
#include <realtime_tools/realtime_publisher.h>

#include <four_wheel_steering_controller/odometry.h>
#include <four_wheel_steering_controller/speed_limiter.h>

namespace four_wheel_steering_controller{

  class FourWheelSteeringController
      : public controller_interface::ControllerBase
  {
  public:
    FourWheelSteeringController();

    virtual bool initRequest(hardware_interface::RobotHW *const robot_hw,
              ros::NodeHandle& root_nh,
              ros::NodeHandle &controller_nh,
              std::set<std::string>& claimed_resources);
    std::string getHardwareInterfaceType() const
    {
      return "";
    }

    bool init(hardware_interface::PositionJointInterface* hw_pos,
                                     hardware_interface::VelocityJointInterface* hw_vel,
                                     ros::NodeHandle& root_nh,
                                     ros::NodeHandle &controller_nh);

    void update(const ros::Time& time, const ros::Duration& period);

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

    void stopping(const ros::Time& /*time*/);

  private:
    std::string name_;

    ros::Duration publish_period_;
    ros::Time last_state_publish_time_;
    bool open_loop_;

    std::vector<hardware_interface::JointHandle> front_wheel_joints_;
    std::vector<hardware_interface::JointHandle> rear_wheel_joints_;
    std::vector<hardware_interface::JointHandle> front_steering_joints_;
    std::vector<hardware_interface::JointHandle> rear_steering_joints_;

    struct Commands
    {
      double lin;
      double ang;
      double front_steering;
      double rear_steering;
      ros::Time stamp;

      Commands() : lin(0.0), ang(0.0), front_steering(0.0), rear_steering(0.0), stamp(0.0) {}
    };
    realtime_tools::RealtimeBuffer<Commands> command_;
    Commands command_struct_;
    ros::Subscriber sub_command_;

    realtime_tools::RealtimeBuffer<Commands> command_four_wheel_steering_;
    Commands command_struct_four_wheel_steering_;
    ros::Subscriber sub_command_four_wheel_steering_;

    boost::shared_ptr<realtime_tools::RealtimePublisher<nav_msgs::Odometry> > odom_pub_;
    boost::shared_ptr<realtime_tools::RealtimePublisher<tf::tfMessage> > tf_odom_pub_;
    Odometry odometry_;

    double track_;

    double wheel_radius_;

    double wheel_base_;

    double cmd_vel_timeout_;

    std::string base_frame_id_;

    bool enable_odom_tf_;

    bool enable_twist_cmd_;

    Commands last1_cmd_;
    Commands last0_cmd_;
    SpeedLimiter limiter_lin_;
    SpeedLimiter limiter_ang_;

  private:
    void brake();

    void cmdVelCallback(const geometry_msgs::Twist& command);

    void cmdFourWheelSteeringCallback(const four_wheel_steering_msgs::FourWheelSteering& command);

    bool getWheelNames(ros::NodeHandle& controller_nh,
                       const std::string& wheel_param,
                       std::vector<std::string>& wheel_names);

    void setOdomPubFields(ros::NodeHandle& root_nh, ros::NodeHandle& controller_nh);

  };

  PLUGINLIB_EXPORT_CLASS(four_wheel_steering_controller::FourWheelSteeringController, controller_interface::ControllerBase);
} // namespace four_wheel_steering_controller