trajectory_planner_ros.h

Go to the documentation of this file.

/*********************************************************************
*
* Software License Agreement (BSD License)
*
*  Copyright (c) 2008, Willow Garage, Inc.
*  All rights reserved.
*
*  Redistribution and use in source and binary forms, with or without
*  modification, are permitted provided that the following conditions
*  are met:
*
*   * Redistributions of source code must retain the above copyright
*     notice, this list of conditions and the following disclaimer.
*   * Redistributions in binary form must reproduce the above
*     copyright notice, this list of conditions and the following
*     disclaimer in the documentation and/or other materials provided
*     with the distribution.
*   * Neither the name of the Willow Garage nor the names of its
*     contributors may be used to endorse or promote products derived
*     from this software without specific prior written permission.
*
*  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
*  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
*  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
*  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
*  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
*  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
*  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
*  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
*  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
*  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
*  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
*  POSSIBILITY OF SUCH DAMAGE.
*
* Author: Eitan Marder-Eppstein
*********************************************************************/
#ifndef TRAJECTORY_ROLLOUT_TRAJECTORY_PLANNER_ROS_H_
#define TRAJECTORY_ROLLOUT_TRAJECTORY_PLANNER_ROS_H_

#include <ros/ros.h>
#include <costmap_2d/costmap_2d.h>
#include <costmap_2d/costmap_2d_publisher.h>
#include <costmap_2d/costmap_2d_ros.h>
#include <world_model.h>
#include <point_grid.h>
#include <costmap_model.h>
#include <voxel_grid_model.h>
#include <trajectory_planner.h>
#include <map_grid_visualizer.h>

#include <planar_laser_scan.h>

#include <tf/transform_datatypes.h>

#include <nav_msgs/Odometry.h>
#include <geometry_msgs/PoseStamped.h>
#include <geometry_msgs/Twist.h>
#include <geometry_msgs/Point.h>

#include <tf/transform_listener.h>

#include <boost/thread.hpp>

#include <string>

#include <angles/angles.h>

#include <nav_core/base_local_planner.h>

#include <dynamic_reconfigure/server.h>
#include <iri_ackermann_local_planner/AckermannLocalPlannerConfig.h>

namespace iri_ackermann_local_planner {
  class TrajectoryPlannerROS : public nav_core::BaseLocalPlanner {
    public:
      TrajectoryPlannerROS();

      TrajectoryPlannerROS(std::string name, tf::TransformListener* tf,
          costmap_2d::Costmap2DROS* costmap_ros);

      void initialize(std::string name, tf::TransformListener* tf,
          costmap_2d::Costmap2DROS* costmap_ros);

      ~TrajectoryPlannerROS();
      
      bool computeVelocityCommands(geometry_msgs::Twist& cmd_vel);

      bool setPlan(const std::vector<geometry_msgs::PoseStamped>& orig_global_plan);

      bool isGoalReached();

      bool checkTrajectory(double vx_samp, double vy_samp, double vtheta_samp, bool update_map = true);

      double scoreTrajectory(double vx_samp, double vy_samp, double vtheta_samp, bool update_map = true);

    private:
      void reconfigureCB(AckermannLocalPlannerConfig &config, uint32_t level);

      void odomCallback(const nav_msgs::Odometry::ConstPtr& msg);

      double sign(double x){
        return x < 0.0 ? -1.0 : 1.0;
      }

      WorldModel* world_model_; 
      TrajectoryPlanner* tc_; 
      costmap_2d::Costmap2DROS* costmap_ros_; 
      costmap_2d::Costmap2D costmap_; 
      MapGridVisualizer map_viz_; 
      tf::TransformListener* tf_; 
      std::string global_frame_; 
      double max_sensor_range_; 
      nav_msgs::Odometry base_odom_; 
      std::string robot_base_frame_; 
      double rot_stopped_velocity_, trans_stopped_velocity_;
      double xy_goal_tolerance_, yaw_goal_tolerance_;
      double inflation_radius_; 
      std::vector<geometry_msgs::PoseStamped> global_plan_;
      bool prune_plan_;
      ros::Publisher g_plan_pub_, l_plan_pub_;
      ros::Subscriber odom_sub_;
      boost::recursive_mutex odom_lock_;
      bool initialized_;
      bool rotating_to_goal_;
      bool latch_xy_goal_tolerance_, xy_tolerance_latch_;
      dynamic_reconfigure::Server<AckermannLocalPlannerConfig> *dsrv_;
      iri_ackermann_local_planner::AckermannLocalPlannerConfig default_config_;
      bool setup_;
      
      std::vector < std::vector < geometry_msgs::PoseStamped > > subPathList;
      std::vector < geometry_msgs::PoseStamped > subPath;
      unsigned int subPathIndex;

      geometry_msgs::Twist ackermann_state_;
  };

};

#endif