pendulum_plugin.h

Go to the documentation of this file.

/*******************************************************************/
/*    AUTHOR: Paal Arthur S. Thorseth                              */
/*    ORGN:   Dept of Eng Cybernetics, NTNU Trondheim              */
/*    FILE:   pendulum_plugin.h                                    */
/*    DATE:   May 5, 2021                                          */
/*                                                                 */
/* Copyright (C) 2021 Paal Arthur S. Thorseth,                     */
/*                    Adrian B. Ghansah                            */
/*                                                                 */
/* This program is free software: you can redistribute it          */
/* and/or modify it under the terms of the GNU 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 General Public License for more details.            */
/*                                                                 */
/* You should have received a copy of the GNU General Public       */
/* License along with this program. If not, see                    */
/* <https://www.gnu.org/licenses/>.                                */
/*                                                                 */
/*******************************************************************/
 
#pragma once
 
// Gazebo
#include <gazebo/gazebo.hh>
#include <gazebo/physics/physics.hh>
#include <gazebo/transport/transport.hh>
#include <gazebo/msgs/msgs.hh>
 
// ROS
#include <thread>
#include "ros/ros.h"
#include "ros/package.h"
#include "ros/console.h"
#include "ros/callback_queue.h"
#include "ros/subscribe_options.h"
#include "eigen_conversions/eigen_msg.h"
#include "std_msgs/Float64.h"
#include "std_msgs/Float64MultiArray.h"
#include "sensor_msgs/JointState.h"
 
// ROS Package Libraries
#include<math_utils/angle_utils.h>
 
// Eigen
#include <Eigen/Core>
 
// Standard library
#include <vector>
#include <map>
 
namespace gazebo
{
    static constexpr unsigned int NUMJOINTS = 1;
 
    class PendulumPlugin : public ModelPlugin
    {
        public: enum ControlMode { position = 1, velocity = 2, torque = 3 };
 
        public: PendulumPlugin();
        
        public: virtual ~PendulumPlugin();
 
        public: virtual void Load(physics::ModelPtr _model, sdf::ElementPtr _sdf);
 
        public: double GetJointForce();
 
        public: void SetJointForce(const double &_force);
 
        public: void SetJointVelocity(const double &_vel);
 
        public: void SetJointPosition(const double &_pos);
 
        public: void OnJointStateMsg(const sensor_msgs::JointStateConstPtr &_msg);
 
        public: void ProcessQueueThread();
 
        public: void PublishQueueThread();
 
        protected: void InitRos();
 
        protected: void InitRosQueueThreads();
 
        protected: void InitJointController();
 
        protected: void InitJointConfiguration();
 
        private: physics::ModelPtr model;
 
        private: std::string model_name;
 
        private: physics::JointPtr joint;
 
        private: double vel_p_gain;
 
        private: double vel_i_gain;
 
        private: double vel_d_gain;
 
        private: double pos_p_gain;
 
        private: double pos_i_gain;
 
        private: double pos_d_gain;
 
        private: ControlMode controlMode;
 
        private: std::unique_ptr<ros::NodeHandle> rosNode;
 
        private: ros::Subscriber jointStateSub;
 
        private: ros::Publisher jointForcePub;
 
        private: ros::CallbackQueue rosProcessQueue;
 
        private: ros::CallbackQueue rosPublishQueue;
 
        private: std::thread rosProcessQueueThread;
 
        private: std::thread rosPublishQueueThread;
 
    };
 
    // Tell Gazebo about this plugin, so that Gazebo can call Load on this plugin.
    GZ_REGISTER_MODEL_PLUGIN(PendulumPlugin)
} // namespace gazebo