gazebo_ros_imu.h

Go to the documentation of this file.

/*
 * Copyright 2012 Open Source Robotics Foundation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
*/
#ifndef GAZEBO_ROS_IMU_HH
#define GAZEBO_ROS_IMU_HH

#include <string>
#include <boost/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/bind.hpp>

#include <ros/ros.h>
#include <ros/callback_queue.h>
#include <ros/advertise_options.h>
#include <sensor_msgs/Imu.h>
#include <std_srvs/Empty.h>

#include <gazebo/physics/physics.hh>
#include <gazebo/transport/transport.hh>
#include <gazebo/common/common.hh>

#include <gazebo_plugins/PubQueue.h>

namespace gazebo
{
  class GazeboRosIMU : public ModelPlugin
  {
    public: GazeboRosIMU();

    public: virtual ~GazeboRosIMU();

    public: void Load(physics::ModelPtr _parent, sdf::ElementPtr _sdf);

    protected: virtual void UpdateChild();

    private: physics::WorldPtr world_;

    private: physics::LinkPtr link;

    private: ros::NodeHandle* rosnode_;
    private: ros::Publisher pub_;
    private: PubQueue<sensor_msgs::Imu>::Ptr pub_Queue;

    private: sensor_msgs::Imu imu_msg_;

    private: std::string link_name_;

    private: std::string topic_name_;

    private: math::Pose offset_;

    private: boost::mutex lock_;

    private: common::Time last_time_;
    private: math::Vector3 last_vpos_;
    private: math::Vector3 last_veul_;
    private: math::Vector3 apos_;
    private: math::Vector3 aeul_;
    
    // rate control
    private: double update_rate_;

    private: math::Pose initial_pose_;

    private: double gaussian_noise_;

    private: double GaussianKernel(double mu, double sigma);

    private: std::string robot_namespace_;

    private: bool ServiceCallback(std_srvs::Empty::Request &req,
                                  std_srvs::Empty::Response &res);

    private: ros::ServiceServer srv_;
    private: std::string service_name_;

    private: ros::CallbackQueue imu_queue_;
    private: void IMUQueueThread();
    private: boost::thread callback_queue_thread_;

    // Pointer to the update event connection
    private: event::ConnectionPtr update_connection_;

    // deferred load in case ros is blocking
    private: sdf::ElementPtr sdf;
    private: void LoadThread();
    private: boost::thread deferred_load_thread_;
    private: unsigned int seed;

    // ros publish multi queue, prevents publish() blocking
    private: PubMultiQueue pmq;
  };
}
#endif