segway_rmp400_odom_fused_alg_node.h

Go to the documentation of this file.

// Copyright (C) 2010-2011 Institut de Robotica i Informatica Industrial, CSIC-UPC.
// Author 
// All rights reserved.
//
// This file is part of iri-ros-pkg
// iri-ros-pkg is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
// 
// IMPORTANT NOTE: This code has been generated through a script from the 
// iri_ros_scripts. Please do NOT delete any comments to guarantee the correctness
// of the scripts. ROS topics can be easly add by using those scripts. Please
// refer to the IRI wiki page for more information:
// http://wikiri.upc.es/index.php/Robotics_Lab

#ifndef _segway_rmp400_odom_alg_node_h_
#define _segway_rmp400_odom_alg_node_h_

#include <iri_base_algorithm/iri_base_algorithm.h>
#include "segway_rmp400_odom_alg.h"

#include <geometry_msgs/PoseWithCovariance.h>
#include <geometry_msgs/TwistWithCovariance.h>
#include <geometry_msgs/Transform.h>

#include <tf/transform_broadcaster.h>

#include <Eigen/Core>

// [publisher subscriber headers]
#include <nav_msgs/Odometry.h>
#include <sensor_msgs/Imu.h>
#include <iri_segway_rmp_msgs/SegwayRMP400Status.h>

// [service client headers]

// [action server client headers]

class SegwayRmp400OdomAlgNode : public algorithm_base::IriBaseAlgorithm<SegwayRmp400OdomAlgorithm>
{
  private:

    ros::Publisher odom_publisher_;
    nav_msgs::Odometry Odometry_msg_;
    ros::Publisher odom_rel_publisher_;
    nav_msgs::Odometry Odometry_rel_msg_;


    ros::Subscriber segway_status_subscriber_;
    void segway_status_callback(const iri_segway_rmp_msgs::SegwayRMP400Status::ConstPtr& msg);
    CMutex segway_status_mutex_;

    ros::Subscriber imu_subscriber_;
    void imu_callback(const sensor_msgs::Imu::ConstPtr& msg);
    CMutex imu_mutex_;

    geometry_msgs::PoseWithCovariance pose_;
    geometry_msgs::TwistWithCovariance twist_;
    geometry_msgs::Transform transform_;

    // odometry variables
    double vT_;
    double r_,vrimu_,rimu_;
    double p_,vpimu_,pimu_;
    double th_,dth_,vthimu_,thimu_;
    double dx_local_accum_;

    bool six_d_;
    ros::Time last_time_;
    ros::Time last_time_imu_;

    double sigma_dx_,sigma2_dth_,sigma_thimu_,sigma_dx_fwd_,sigma_dx_turn_;
    double offset_fwd_, offset_turn_;
    bool first_segway_status_, first_imu_;

    Eigen::Matrix3f Jp_,Jd_,Q_,cov_;

    // parameter variables
    std::string tf_prefix_;
    std::string odom_id_;
    std::string base_link_id_;
    bool publish_tf_;
    bool publish_odom_rel_;

  public:
    SegwayRmp400OdomAlgNode(void);

    ~SegwayRmp400OdomAlgNode(void);

  protected:

    tf::TransformBroadcaster odom_broadcaster_;

    void mainNodeThread(void);

    void node_config_update(Config &config, uint32_t level);

    void addNodeDiagnostics(void);

    // [diagnostic functions]
    
    // [test functions]
};

#endif