trajectory_planner.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_H_
#define TRAJECTORY_ROLLOUT_TRAJECTORY_PLANNER_H_

#include <vector>
#include <string>
#include <sstream>
#include <math.h>
#include <ros/console.h>
#include <angles/angles.h>

//for creating a local cost grid
#include <iri_diff_local_planner/map_cell.h>
#include <iri_diff_local_planner/map_grid.h>

//for obstacle data access
#include <costmap_2d/costmap_2d.h>
#include <iri_diff_local_planner/world_model.h>

#include <iri_diff_local_planner/trajectory.h>

//we'll take in a path as a vector of poses
#include <geometry_msgs/PoseStamped.h>
#include <geometry_msgs/Point.h>
#include <iri_diff_local_planner/Position2DInt.h>

//for computing path distance
#include <queue>

//for some datatypes
#include <tf/transform_datatypes.h>

#include <iri_diff_local_planner/IriDiffLocalPlannerConfig.h>
#include <boost/algorithm/string.hpp>

namespace iri_diff_local_planner {
  class TrajectoryPlanner{
    friend class TrajectoryPlannerTest; //Need this for gtest to work
    public:
      TrajectoryPlanner(WorldModel& world_model, 
          const costmap_2d::Costmap2D& costmap, 
          std::vector<geometry_msgs::Point> footprint_spec,
          double acc_lim_x = 1.0, double acc_lim_y = 1.0, double acc_lim_theta = 1.0,
          double sim_time = 1.0, double sim_granularity = 0.025, 
          int vx_samples = 20, int vtheta_samples = 20,
          double pdist_scale = 0.6, double gdist_scale = 0.8, double occdist_scale = 0.2,
          double heading_lookahead = 0.325, double oscillation_reset_dist = 0.05, 
          double escape_reset_dist = 0.10, double escape_reset_theta = M_PI_2,
          bool holonomic_robot = true,
          double *max_vel_x = NULL, double min_vel_x = 0.1, 
          double max_vel_th = 1.0, double min_vel_th = -1.0, double min_in_place_vel_th = 0.4,
          double backup_vel = -0.1,
          bool dwa = false, bool heading_scoring = false, double heading_scoring_timestep = 0.1,
          bool simple_attractor = false,
          std::vector<double> y_vels = std::vector<double>(0),
          double stop_time_buffer = 0.2,
          double sim_period = 0.1, double angular_sim_granularity = 0.025);

      ~TrajectoryPlanner();

      void reconfigure(IriDiffLocalPlannerConfig &cfg);

      Trajectory findBestPath(tf::Stamped<tf::Pose> global_pose, tf::Stamped<tf::Pose> global_vel,
          tf::Stamped<tf::Pose>& drive_velocities);

      void updatePlan(const std::vector<geometry_msgs::PoseStamped>& new_plan, bool compute_dists = false);

      void getLocalGoal(double& x, double& y);

      bool checkTrajectory(double x, double y, double theta, double vx, double vy, 
          double vtheta, double vx_samp, double vy_samp, double vtheta_samp);

      double scoreTrajectory(double x, double y, double theta, double vx, double vy, 
          double vtheta, double vx_samp, double vy_samp, double vtheta_samp);

      bool getCellCosts(int cx, int cy, float &path_cost, float &goal_cost, float &occ_cost, float &total_cost);
    private:
      Trajectory createTrajectories(double x, double y, double theta, double vx, double vy, double vtheta, 
          double acc_x, double acc_y, double acc_theta);

      void generateTrajectory(double x, double y, double theta, double vx, double vy, 
          double vtheta, double vx_samp, double vy_samp, double vtheta_samp, double acc_x, double acc_y,
          double acc_theta, double impossible_cost, Trajectory& traj);

      double footprintCost(double x_i, double y_i, double theta_i);

      std::vector<iri_diff_local_planner::Position2DInt> getFootprintCells(double x_i, double y_i, double theta_i, bool fill);

      void getLineCells(int x0, int x1, int y0, int y1, std::vector<iri_diff_local_planner::Position2DInt>& pts);

      void getFillCells(std::vector<iri_diff_local_planner::Position2DInt>& footprint);

      MapGrid map_; 
      const costmap_2d::Costmap2D& costmap_; 
      WorldModel& world_model_; 

      std::vector<geometry_msgs::Point> footprint_spec_; 

      std::vector<geometry_msgs::PoseStamped> global_plan_; 

      bool stuck_left, stuck_right; 
      bool rotating_left, rotating_right; 

      bool stuck_left_strafe, stuck_right_strafe; 
      bool strafe_right, strafe_left; 

      bool escaping_; 

      double goal_x_,goal_y_; 


      double sim_time_; 
      double sim_granularity_; 
      double angular_sim_granularity_; 

      int vx_samples_; 
      int vtheta_samples_; 

      double pdist_scale_, gdist_scale_, occdist_scale_; 
      double acc_lim_x_, acc_lim_y_, acc_lim_theta_; 

      double prev_x_, prev_y_; 
      double escape_x_, escape_y_, escape_theta_; 

      Trajectory traj_one, traj_two; 

      double heading_lookahead_; 
      double oscillation_reset_dist_; 
      double escape_reset_dist_, escape_reset_theta_; 
      bool holonomic_robot_; 
      
      double min_vel_x_, max_vel_th_, min_vel_th_, min_in_place_vel_th_; 
      double *max_vel_x_;

      double backup_vel_; 

      bool dwa_;  
      bool heading_scoring_; 
      double heading_scoring_timestep_; 
      bool simple_attractor_;  

      std::vector<double> y_vels_; 

      double stop_time_buffer_; 
      double sim_period_; 

      boost::mutex configuration_mutex_;

      inline double computeNewXPosition(double xi, double vx, double vy, double theta, double dt){
        return xi + (vx * cos(theta) + vy * cos(M_PI_2 + theta)) * dt;
      }

      inline double computeNewYPosition(double yi, double vx, double vy, double theta, double dt){
        return yi + (vx * sin(theta) + vy * sin(M_PI_2 + theta)) * dt;
      }

      inline double computeNewThetaPosition(double thetai, double vth, double dt){
        return thetai + vth * dt;
      }

      //compute velocity based on acceleration
      inline double computeNewVelocity(double vg, double vi, double a_max, double dt){
        if((vg - vi) >= 0)
          return std::min(vg, vi + a_max * dt);
        return std::max(vg, vi - a_max * dt);
      }

      void getMaxSpeedToStopInTime(double time, double& vx, double& vy, double& vth){
        vx = acc_lim_x_ * std::max(time, 0.0);
        vy = acc_lim_y_ * std::max(time, 0.0);
        vth = acc_lim_theta_ * std::max(time, 0.0);
      }

      double lineCost(int x0, int x1, int y0, int y1);
      double pointCost(int x, int y);
      double headingDiff(int cell_x, int cell_y, double x, double y, double heading);
  };
};

#endif