segway_rmp400_odom_6d_alg_node.cpp

Go to the documentation of this file.

#include "segway_rmp400_odom_6d_alg_node.h"

SegwayRmp400OdomAlgNode::SegwayRmp400OdomAlgNode(void) :
  algorithm_base::IriBaseAlgorithm<SegwayRmp400OdomAlgorithm>()
{
  //init class attributes if necessary
  loop_rate_ = 100;//in [Hz]

  public_node_handle_.param<std::string>("tf_prefix",    tf_prefix_,    "");
  public_node_handle_.param<std::string>("odom_id",      odom_id_,      "odom" );
  public_node_handle_.param<std::string>("base_link_id", base_link_id_, "base_footprint" );
  public_node_handle_.param<bool>       ("publish_tf",   publish_tf_,   true  );
  
  // push down frame ids if necessary
  if( tf_prefix_.size()!=0 )
  {
    odom_id_      = tf_prefix_ + "/" + odom_id_;
    base_link_id_ = tf_prefix_ + "/" + base_link_id_;
  }

  // [init publishers]
  odom_publisher_ = public_node_handle_.advertise<nav_msgs::Odometry>("odom", 100);
  
  // [init subscribers]
  segway_status_subscriber_ = public_node_handle_.subscribe("segway_status", 100, &SegwayRmp400OdomAlgNode::segway_status_callback, this);
  
  // [init services]
  
  // [init clients]
  
  // [init action servers]
  
  // [init action clients]
}

SegwayRmp400OdomAlgNode::~SegwayRmp400OdomAlgNode(void)
{
  // [free dynamic memory]
}

void SegwayRmp400OdomAlgNode::mainNodeThread(void)
{
  ros::Time current_time = ros::Time::now();
  geometry_msgs::TransformStamped odom_trans_msg;
  static ros::Time last_time = ros::Time::now();

  //compute differencial
  double dt = (current_time - last_time).toSec();

  //get current translational velocity and component velocities
  segway_status_mutex_.enter(); 
  double vT = (left_wheels_velocity_ + right_wheels_velocity_) / 2;
  double vy = - yaw_rate_;
  double vr = roll_rate_;
  double vp = pitch_rate_;
  segway_status_mutex_.exit(); 
  
  // First, we will only use the half of time. See header file for info.
  r_ += vr * dt/2;
  p_ += vp * dt/2;
  y_ += vy * dt/2;
  
  // split translational velocity in cartesian components
  double vTx  = vT * cos(y_) * cos(p_);
  double vTy  = vT * sin(y_) * cos(r_);
  double vTz  = vT * sin(r_) * sin(p_);

  // Integrate velocities with time to get current position
  double dx  = vTx * dt;
  double dy  = vTy * dt;
  double dz  = vTz * dt;

  //update second half angles
  r_ += vr * dt/2;
  p_ += vp * dt/2;
  y_ += vy * dt/2;

  // [fill msg structures]
  geometry_msgs::Quaternion rot = tf::createQuaternionMsgFromRollPitchYaw (r_,p_,y_);
  transform_.translation.x += dx;
  transform_.translation.y += dy;
  transform_.translation.z += dz;
  transform_.rotation       = rot;

  pose_.pose.position.x += dx;
  pose_.pose.position.y += dy;
  pose_.pose.position.z += dz;
  pose_.pose.orientation = rot;

  pose_.covariance[0]  =
  pose_.covariance[7]  = 0.01;
  pose_.covariance[14] =
  pose_.covariance[21] =
  pose_.covariance[28] =
  pose_.covariance[35] = 1e10;

  twist_.twist.linear.x  = vT;
  twist_.twist.linear.y  = 0.0;
  twist_.twist.linear.z  = 0.0;
  twist_.twist.angular.x = vr;
  twist_.twist.angular.y = vp;
  twist_.twist.angular.z = vy;

  // Add the covariances to use with EKF
  // for x velocity , Error ~= +-5%
  twist_.covariance[0]  = 0.0001;//pow(0.0001*vT,2);
  twist_.covariance[7]  = 0.0001;
  twist_.covariance[14] =
  twist_.covariance[21] =
  twist_.covariance[28] =
  // for theta velocity, Error ~= +-90%
  twist_.covariance[35] = 999; //pow(0.9*vyaw,2);
  
  //update last time
  last_time = current_time;

  //send the transform
  if(publish_tf_)
  {
    //first, we'll publish the transform over tf
    odom_trans_msg.header.stamp    = current_time;
    odom_trans_msg.header.frame_id = odom_id_;
    odom_trans_msg.child_frame_id  = base_link_id_;
    odom_trans_msg.transform       = transform_;

    odom_broadcaster_.sendTransform(odom_trans_msg);
  }

  //next, we'll publish the odometry message over ROS
  Odometry_msg_.header.stamp    = current_time;
  Odometry_msg_.header.frame_id = odom_id_;
  Odometry_msg_.child_frame_id  = base_link_id_;
  Odometry_msg_.pose            = pose_;
  Odometry_msg_.twist           = twist_;

  // [fill msg structures]
  //Odometry_msg_.data = my_var;
  
  // [fill srv structure and make request to the server]
  
  // [fill action structure and make request to the action server]

  // [publish messages]
  odom_publisher_.publish(Odometry_msg_);
}

/*  [subscriber callbacks] */
void SegwayRmp400OdomAlgNode::segway_status_callback(const iri_segway_rmp_msgs::SegwayRMP400Status::ConstPtr& msg) 
{ 
  ROS_DEBUG("SegwayRmp400OdomAlgNode::segway_status_callback: New Message Received"); 

  //use appropiate mutex to shared variables if necessary 
  //alg_.lock(); 
  segway_status_mutex_.enter(); 

  //std::cout << msg->data << std::endl;
  left_wheels_velocity_  = (msg->rmp200[0].left_wheel_velocity  + msg->rmp200[1].left_wheel_velocity)/2;
  right_wheels_velocity_ = (msg->rmp200[0].right_wheel_velocity + msg->rmp200[1].right_wheel_velocity)/2;
  yaw_rate_              = (msg->rmp200[0].yaw_rate   + msg->rmp200[1].yaw_rate)/2;
  pitch_rate_            = (msg->rmp200[0].pitch_rate + msg->rmp200[1].pitch_rate) / 2;
  roll_rate_             = (msg->rmp200[0].roll_rate  + msg->rmp200[1].roll_rate ) / 2;

  //unlock previously blocked shared variables 
  //alg_.unlock(); 
  segway_status_mutex_.exit(); 
}

/*  [service callbacks] */

/*  [action callbacks] */

/*  [action requests] */

void SegwayRmp400OdomAlgNode::node_config_update(Config &config, uint32_t level)
{
  alg_.lock();

  alg_.unlock();
}

void SegwayRmp400OdomAlgNode::addNodeDiagnostics(void)
{
}

/* main function */
int main(int argc,char *argv[])
{
  return algorithm_base::main<SegwayRmp400OdomAlgNode>(argc, argv, "segway_rmp400_odom_alg_node");
}