Debug wam_gazebo_ros_control.
authorWalter Fetter Lages <w.fetter@ieee.org>
Wed, 28 Nov 2018 20:47:57 +0000 (18:47 -0200)
committerWalter Fetter Lages <w.fetter@ieee.org>
Wed, 28 Nov 2018 20:47:57 +0000 (18:47 -0200)
12 files changed:
ufrgs_wam/package.xml
wam_controllers/README
wam_controllers/include/wam_controllers/computed_torque_controller.h
wam_controllers/package.xml
wam_controllers/scripts/set_home.sh
wam_controllers/src/computed_torque_controller.cpp
wam_description/xacro/wam.urdf.xacro
wam_gazebo_ros_control/CMakeLists.txt
wam_gazebo_ros_control/package.xml
wam_gazebo_ros_control/src/default_robot_hw_sim.cpp [new file with mode: 0644]
wam_gazebo_ros_control/src/wam_robot_hw_sim.cpp
wam_gazebo_ros_control/src/wam_robot_hw_sim.cpp1 [new file with mode: 0644]

index 2bbf3d1..f9af292 100644 (file)
@@ -3,17 +3,56 @@
   <name>ufrgs_wam</name>
   <version>2.0.0</version>
   <description>The ufrgs_wam package</description>
-  <maintainer email="fetter@ece.ufrgs.br">Walter Fetter Lages
-       </maintainer>
+
+  <!-- One maintainer tag required, multiple allowed, one person per tag --> 
+  <!-- Example:  -->
+  <!-- <maintainer email="jane.doe@example.com">Jane Doe</maintainer> -->
+  <maintainer email="fetter@ece.ufrgs.br">Walter Fetter Lages</maintainer>
+
+
+  <!-- One license tag required, multiple allowed, one license per tag -->
+  <!-- Commonly used license strings: -->
+  <!--   BSD, MIT, Boost Software License, GPLv2, GPLv3, LGPLv2.1, LGPLv3 -->
   <license>GPLv3</license>
+
+
+  <!-- Url tags are optional, but mutiple are allowed, one per tag -->
+  <!-- Optional attribute type can be: website, bugtracker, or repository -->
+  <!-- Example: -->
+  <!-- <url type="website">http://wiki.ros.org/ufrgs_wam</url> -->
   <url type="website">http://www.ece.ufrgs.br/~fetter/ufrgs_wam</url>
 
+
+  <!-- Author tags are optional, mutiple are allowed, one per tag -->
+  <!-- Authors do not have to be maintianers, but could be -->
+  <!-- Example: -->
+  <!-- <author email="jane.doe@example.com">Jane Doe</author> -->
+  <author email="fetter@ece.ufrgs.br">Walter Fetter Lages</author>
+
+  <!-- The *_depend tags are used to specify dependencies -->
+  <!-- Dependencies can be catkin packages or system dependencies -->
+  <!-- Examples: -->
+  <!-- Use build_depend for packages you need at compile time: -->
+  <!--   <build_depend>message_generation</build_depend> -->
+  <!-- Use buildtool_depend for build tool packages: -->
+  <!--   <buildtool_depend>catkin</buildtool_depend> -->
+  <!-- Use run_depend for packages you need at runtime: -->
+  <!--   <run_depend>message_runtime</run_depend> -->
+  <!-- Use test_depend for packages you need only for testing: -->
+  <!--   <test_depend>gtest</test_depend> -->
   <buildtool_depend>catkin</buildtool_depend>
 
   <run_depend>wam_description</run_depend>
+  <run_depend>wam_gazebo_ros_control</run_depend>
   <run_depend>wam_controllers</run_depend>
   
+  <!-- The export tag contains other, unspecified, tags -->
   <export>
-    <metapackage/>
+    <!-- You can specify that this package is a metapackage here: -->
+    <!-- <metapackage/> -->
+  <metapackage/>
+
+    <!-- Other tools can request additional information be placed here -->
+
   </export>
 </package>
index 07f6e1b..861d535 100644 (file)
@@ -5,3 +5,6 @@ rostopic pub /wam/computed_torque_controller/command trajectory_msgs/JointTrajec
 Arguments are positions, velocities, accelerations, effort, time from start in
 seconds and nanoseconds.
 
+Set starting position:
+
+rosservice call /gazebo/set_model_configuration wam joint ['wam_joint_1','wam_joint_2','wam_joint_3','wam_joint_4','wam_joint_5','wam_joint_6','wam_joint_7'] [0.0,0.75,0.0,1.5,0.0,0.9,0.0]
\ No newline at end of file
index 6abd28c..2bbbe22 100644 (file)
@@ -1,24 +1,3 @@
-/******************************************************************************
-                              UFRGS WAM
-                        Computed Torque Controller
-          Copyright (C) 2013-2015 Walter Fetter Lages <w.fetter@ieee.org>
-
-        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
-        <http://www.gnu.org/licenses/>.
-        
-*******************************************************************************/
-
 #ifndef WAM_CONTROLLERS_COMPUTED_TORQUE_CONTROLLER_H
 #define WAM_CONTROLLERS_COMPUTED_TORQUE_CONTROLLER_H
 
index 67e159c..118d020 100644 (file)
@@ -3,14 +3,46 @@
   <name>wam_controllers</name>
   <version>2.0.0</version>
   <description>The wam_controllers package</description>
-  <maintainer email="fetter@ece.ufrgs.br">Walter Fetter Lages
-       </maintainer>
+
+  <!-- One maintainer tag required, multiple allowed, one person per tag --> 
+  <!-- Example:  -->
+  <!-- <maintainer email="jane.doe@example.com">Jane Doe</maintainer> -->
+  <maintainer email="fetter@ece.ufrgs.br">Walter Fetter Lages</maintainer>
+
+
+  <!-- One license tag required, multiple allowed, one license per tag -->
+  <!-- Commonly used license strings: -->
+  <!--   BSD, MIT, Boost Software License, GPLv2, GPLv3, LGPLv2.1, LGPLv3 -->
   <license>GPLv3</license>
+
+
+  <!-- Url tags are optional, but mutiple are allowed, one per tag -->
+  <!-- Optional attribute type can be: website, bugtracker, or repository -->
+  <!-- Example: -->
+  <!-- <url type="website">http://wiki.ros.org/wam_controllers</url> -->
   <url type="website">http://www.ece.ufrgs.br/ufrgs_wam</url>
+
+
+  <!-- Author tags are optional, mutiple are allowed, one per tag -->
+  <!-- Authors do not have to be maintianers, but could be -->
+  <!-- Example: -->
+  <!-- <author email="jane.doe@example.com">Jane Doe</author> -->
   <author email="fetter@ece.ufrgs.br">Walter Fetter Lages</author>
 
+  <!-- The *_depend tags are used to specify dependencies -->
+  <!-- Dependencies can be catkin packages or system dependencies -->
+  <!-- Examples: -->
+  <!-- Use build_depend for packages you need at compile time: -->
+  <!--   <build_depend>message_generation</build_depend> -->
+  <!-- Use buildtool_depend for build tool packages: -->
+  <!--   <buildtool_depend>catkin</buildtool_depend> -->
+  <!-- Use run_depend for packages you need at runtime: -->
+  <!--   <run_depend>message_runtime</run_depend> -->
+  <!-- Use test_depend for packages you need only for testing: -->
+  <!--   <test_depend>gtest</test_depend> -->
   <buildtool_depend>catkin</buildtool_depend>
   
+  <build_depend>wam_gazebo_ros_control</build_depend>
   <build_depend>controller_interface</build_depend>
   <build_depend>orocos_kdl</build_depend>
   <build_depend>kdl_parser</build_depend>
   <run_depend>controller_interface</run_depend>
   <run_depend>controller_manager</run_depend>
   <run_depend>control_msgs</run_depend>
-  <run_depend>joint_state_controller</run_depend>
+  <run_depend> joint_state_controller</run_depend>
   <run_depend>urdf</run_depend>
   <run_depend>kdl_parser</run_depend>
 
+  <!-- The export tag contains other, unspecified, tags -->
   <export>
+    <!-- You can specify that this package is a metapackage here: -->
+    <!-- <metapackage/> -->
+
+    <!-- Other tools can request additional information be placed here -->
     <controller_interface plugin="${prefix}/wam_controllers_plugins.xml"/>
   </export>
-
-</package>
+</package>
\ No newline at end of file
index 1ff2f52..4641b95 100755 (executable)
@@ -4,10 +4,10 @@ rosservice call /gazebo/set_model_configuration wam joint \
 "['wam_joint_1','wam_joint_2','wam_joint_3','wam_joint_4','wam_joint_5','wam_joint_6','wam_joint_7']" \
 "[0.0,-2.0,0.0,3.1,0.0,0.0,0.0]"
 
-rostopic pub /wam/computed_torque_controller/command \
-trajectory_msgs/JointTrajectoryPoint \
-"[0.0,-2.0,0.0,3.1,0.0,0.0,0.0]" \
-"[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]" \
-"[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]" \
-"[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]" \
-"[0.0, 0.0]" "-1"
+#rostopic pub /wam/computed_torque_controller/command \
+#trajectory_msgs/JointTrajectoryPoint \
+#"[0.0,-2.0,0.0,3.1,0.0,0.0,0.0]" \
+#"[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]" \
+#"[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]" \
+#"[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]" \
+#"[0.0, 0.0]" "-1"
index de09786..c880ed0 100644 (file)
@@ -1,24 +1,3 @@
-/******************************************************************************
-                              UFRGS WAM
-                        Computed Torque Controller
-          Copyright (C) 2013-2015 Walter Fetter Lages <w.fetter@ieee.org>
-
-        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
-        <http://www.gnu.org/licenses/>.
-        
-*******************************************************************************/
-
 #include <sys/mman.h>
 
 #include <wam_controllers/computed_torque_controller.h>
index 7a2f79f..c0db1f6 100644 (file)
   <gazebo>
     <plugin name="gazebo_ros_control" filename="libgazebo_ros_control.so" >
       <robotNamespace>/wam</robotNamespace>
-      <robotSimType>gazebo_ros_control/DefaultRobotHWSim</robotSimType>
+
+      <!-- Custom plugin -->
+      <!-- robotSimType>wam_gazebo_ros_control/WamRobotHWSim</robotSimType -->
+
+      <!-- Default plugin -->
+      <!-- robotSimType>gazebo_ros_control/DefaultRobotHWSim</robotSimType -->
+
       <controlPeriod>0.001</controlPeriod>
     </plugin>
   </gazebo>
index 6b84ae6..f8d4956 100644 (file)
@@ -78,7 +78,10 @@ find_package(gazebo REQUIRED)
 catkin_package(
 #  INCLUDE_DIRS include
 #  LIBRARIES wam_gazebo_ros_control
-#  CATKIN_DEPENDS other_catkin_pkg
+   CATKIN_DEPENDS
+   controller_manager
+   pluginlib
+   gazebo_ros_control
 #  DEPENDS system_lib
 )
 
@@ -105,9 +108,9 @@ add_library(wam_gazebo_ros_control
 # add_dependencies(wam_gazebo_ros_control_node wam_gazebo_ros_control_generate_messages_cpp)
 
 ## Specify libraries to link a library or executable target against
-#target_link_libraries(${PROJECT_NAME}
-#   ${catkin_LIBRARIES} 
-#)
+target_link_libraries(${PROJECT_NAME}
+   ${catkin_LIBRARIES} 
+)
 
 #############
 ## Install ##
@@ -138,7 +141,7 @@ install(TARGETS ${PROJECT_NAME}
 # )
 
 ## Mark other files for installation (e.g. launch and bag files, etc.)
-install(FILES robot_sim_plugins.xml
+install(FILES wam_gazebo_ros_control_plugins.xml
 #   # myfile1
 #   # myfile2
    DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}
index b6d8a17..ac4211e 100644 (file)
   <!--   <test_depend>gtest</test_depend> -->
   <buildtool_depend>catkin</buildtool_depend>
 
+   <build_depend>controller_manager</build_depend>
+   <build_depend>pluginlib</build_depend>
    <build_depend>gazebo_ros_control</build_depend>
    <!--build_depend>gazebo_ros_control_plugin</build_depend-->
    <build_depend>wam_description</build_depend>
    <build_depend>gazebo</build_depend>
    
-    <!--run_depend>roscpp</run_depend-->
+    <run_depend>roscpp</run_depend>
     <!--run_depend>gazebo</run_depend-->
     <!--run_depend>gazebo_ros</run_depend-->
-    <!--run_depend>controller_manager</run_depend-->
-    <!--run_depend>pluginlib</run_depend-->
+    <run_depend>controller_manager</run_depend>
+    <run_depend>pluginlib</run_depend>
     <!--run_depend>transmission_interface</run_depend-->
     <!--run_depend>urdf</run_depend-->
     <run_depend>gazebo_ros_control</run_depend>
diff --git a/wam_gazebo_ros_control/src/default_robot_hw_sim.cpp b/wam_gazebo_ros_control/src/default_robot_hw_sim.cpp
new file mode 100644 (file)
index 0000000..1a45eb9
--- /dev/null
@@ -0,0 +1,509 @@
+/*********************************************************************
+ * Software License Agreement (BSD License)
+ *
+ *  Copyright (c) 2013, Open Source Robotics Foundation
+ *  Copyright (c) 2013, The Johns Hopkins University
+ *  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 Open Source Robotics Foundation
+ *     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: Dave Coleman, Johnathan Bohren
+   Desc:   Hardware Interface for any simulated robot in Gazebo
+*/
+
+#ifndef _GAZEBO_ROS_CONTROL___DEFAULT_ROBOT_HW_SIM_H_
+#define _GAZEBO_ROS_CONTROL___DEFAULT_ROBOT_HW_SIM_H_
+
+// ros_control
+#include <control_toolbox/pid.h>
+#include <hardware_interface/joint_command_interface.h>
+#include <hardware_interface/robot_hw.h>
+#include <joint_limits_interface/joint_limits.h>
+#include <joint_limits_interface/joint_limits_interface.h>
+#include <joint_limits_interface/joint_limits_rosparam.h>
+#include <joint_limits_interface/joint_limits_urdf.h>
+
+// Gazebo
+#include <gazebo/common/common.hh>
+#include <gazebo/physics/physics.hh>
+#include <gazebo/gazebo.hh>
+
+// ROS
+#include <ros/ros.h>
+#include <angles/angles.h>
+#include <pluginlib/class_list_macros.h>
+
+// gazebo_ros_control
+#include <gazebo_ros_control/robot_hw_sim.h>
+
+// URDF
+#include <urdf/model.h>
+
+namespace
+{
+
+double clamp(const double val, const double min_val, const double max_val)
+{
+  return std::min(std::max(val, min_val), max_val);
+}
+
+}
+
+namespace gazebo_ros_control
+{
+
+class DefaultRobotHWSim : public gazebo_ros_control::RobotHWSim
+{
+public:
+
+  bool initSim(
+    const std::string& robot_namespace,
+    ros::NodeHandle model_nh,
+    gazebo::physics::ModelPtr parent_model,
+    const urdf::Model *const urdf_model,
+    std::vector<transmission_interface::TransmissionInfo> transmissions)
+  {
+    // getJointLimits() searches joint_limit_nh for joint limit parameters. The format of each
+    // parameter's name is "joint_limits/<joint name>". An example is "joint_limits/axle_joint".
+    const ros::NodeHandle joint_limit_nh(model_nh, robot_namespace);
+
+    // Resize vectors to our DOF
+    n_dof_ = transmissions.size();
+    joint_names_.resize(n_dof_);
+    joint_types_.resize(n_dof_);
+    joint_lower_limits_.resize(n_dof_);
+    joint_upper_limits_.resize(n_dof_);
+    joint_effort_limits_.resize(n_dof_);
+    joint_control_methods_.resize(n_dof_);
+    pid_controllers_.resize(n_dof_);
+    joint_position_.resize(n_dof_);
+    joint_velocity_.resize(n_dof_);
+    joint_effort_.resize(n_dof_);
+    joint_effort_command_.resize(n_dof_);
+    joint_position_command_.resize(n_dof_);
+    joint_velocity_command_.resize(n_dof_);
+
+    // Initialize values
+    for(unsigned int j=0; j < n_dof_; j++)
+    {
+      // Check that this transmission has one joint
+      if(transmissions[j].joints_.size() == 0)
+      {
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has no associated joints.");
+        continue;
+      }
+      else if(transmissions[j].joints_.size() > 1)
+      {
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has more than one joint. Currently the default robot hardware simulation "
+          << " interface only supports one.");
+        continue;
+      }
+
+      // Check that this transmission has one actuator
+      if(transmissions[j].actuators_.size() == 0)
+      {
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has no associated actuators.");
+        continue;
+      }
+      else if(transmissions[j].actuators_.size() > 1)
+      {
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has more than one actuator. Currently the default robot hardware simulation "
+          << " interface only supports one.");
+        continue;
+      }
+
+      // Add data from transmission
+      joint_names_[j] = transmissions[j].joints_[0].name_;
+      joint_position_[j] = 1.0;
+      joint_velocity_[j] = 0.0;
+      joint_effort_[j] = 1.0;  // N/m for continuous joints
+      joint_effort_command_[j] = 0.0;
+      joint_position_command_[j] = 0.0;
+      joint_velocity_command_[j] = 0.0;
+
+#if ROS_VERSION_MINOR > 10 || ROS_VERSION_MAJOR > 1
+      const std::string &hardware_interface =
+        transmissions[j].actuators_[0].hardware_interfaces_[0];
+#else
+      const std::string &hardware_interface =
+        transmissions[j].actuators_[0].hardware_interface_;
+#endif
+
+      // Debug
+      ROS_DEBUG_STREAM_NAMED("default_robot_hw_sim","Loading joint '" << joint_names_[j]
+        << "' of type '" << hardware_interface << "'");
+
+      // Create joint state interface for all joints
+      js_interface_.registerHandle(hardware_interface::JointStateHandle(
+          joint_names_[j], &joint_position_[j], &joint_velocity_[j], &joint_effort_[j]));
+
+      // Decide what kind of command interface this actuator/joint has
+      hardware_interface::JointHandle joint_handle;
+      if(hardware_interface == "EffortJointInterface")
+      {
+        // Create effort joint interface
+        joint_control_methods_[j] = EFFORT;
+        joint_handle = hardware_interface::JointHandle(js_interface_.getHandle(joint_names_[j]),
+                                                       &joint_effort_command_[j]);
+        ej_interface_.registerHandle(joint_handle);
+      }
+      else if(hardware_interface == "PositionJointInterface")
+      {
+        // Create position joint interface
+        joint_control_methods_[j] = POSITION;
+        joint_handle = hardware_interface::JointHandle(js_interface_.getHandle(joint_names_[j]),
+                                                       &joint_position_command_[j]);
+        pj_interface_.registerHandle(joint_handle);
+      }
+      else if(hardware_interface == "VelocityJointInterface")
+      {
+        // Create velocity joint interface
+        joint_control_methods_[j] = VELOCITY;
+        joint_handle = hardware_interface::JointHandle(js_interface_.getHandle(joint_names_[j]),
+                                                       &joint_velocity_command_[j]);
+        vj_interface_.registerHandle(joint_handle);
+      }
+      else
+      {
+        ROS_FATAL_STREAM_NAMED("default_robot_hw_sim","No matching hardware interface found for '"
+          << hardware_interface );
+        return false;
+      }
+
+      // Get the gazebo joint that corresponds to the robot joint.
+      //ROS_DEBUG_STREAM_NAMED("default_robot_hw_sim", "Getting pointer to gazebo joint: "
+      //  << joint_names_[j]);
+      gazebo::physics::JointPtr joint = parent_model->GetJoint(joint_names_[j]);
+      if (!joint)
+      {
+        ROS_ERROR_STREAM("This robot has a joint named \"" << joint_names_[j]
+          << "\" which is not in the gazebo model.");
+        return false;
+      }
+      sim_joints_.push_back(joint);
+
+      registerJointLimits(joint_names_[j], joint_handle, joint_control_methods_[j],
+                          joint_limit_nh, urdf_model,
+                          &joint_types_[j], &joint_lower_limits_[j], &joint_upper_limits_[j],
+                          &joint_effort_limits_[j]);
+      if (joint_control_methods_[j] != EFFORT)
+      {
+        // Initialize the PID controller. If no PID gain values are found, use joint->SetAngle() or
+        // joint->SetVelocity() to control the joint.
+        const ros::NodeHandle nh(model_nh, robot_namespace + "/gazebo_ros_control/pid_gains/" +
+                                 joint_names_[j]);
+        if (pid_controllers_[j].init(nh))
+        {
+          switch (joint_control_methods_[j])
+          {
+            case POSITION:
+              joint_control_methods_[j] = POSITION_PID;
+              break;
+            case VELOCITY:
+              joint_control_methods_[j] = VELOCITY_PID;
+              break;
+          }
+        }
+        else
+        {
+          // joint->SetMaxForce() must be called if joint->SetAngle() or joint->SetVelocity() are
+          // going to be called. joint->SetMaxForce() must *not* be called if joint->SetForce() is
+          // going to be called.
+          joint->SetMaxForce(0, joint_effort_limits_[j]);
+        }
+      }
+    }
+
+    // Register interfaces
+    registerInterface(&js_interface_);
+    registerInterface(&ej_interface_);
+    registerInterface(&pj_interface_);
+    registerInterface(&vj_interface_);
+
+    return true;
+  }
+
+  void readSim(ros::Time time, ros::Duration period)
+  {
+    for(unsigned int j=0; j < n_dof_; j++)
+    {
+      // Gazebo has an interesting API...
+      if (joint_types_[j] == urdf::Joint::PRISMATIC)
+      {
+        joint_position_[j] = sim_joints_[j]->GetAngle(0).Radian();
+      }
+      else
+      {
+        joint_position_[j] += angles::shortest_angular_distance(joint_position_[j],
+                              sim_joints_[j]->GetAngle(0).Radian());
+      }
+      joint_velocity_[j] = sim_joints_[j]->GetVelocity(0);
+      joint_effort_[j] = sim_joints_[j]->GetForce((unsigned int)(0));
+    }
+  }
+
+  void writeSim(ros::Time time, ros::Duration period)
+  {
+    ej_sat_interface_.enforceLimits(period);
+    ej_limits_interface_.enforceLimits(period);
+    pj_sat_interface_.enforceLimits(period);
+    pj_limits_interface_.enforceLimits(period);
+    vj_sat_interface_.enforceLimits(period);
+    vj_limits_interface_.enforceLimits(period);
+
+    for(unsigned int j=0; j < n_dof_; j++)
+    {
+      switch (joint_control_methods_[j])
+      {
+        case EFFORT:
+          {
+            const double effort = joint_effort_command_[j];
+            sim_joints_[j]->SetForce(0, effort);
+          }
+          break;
+
+        case POSITION:
+#if GAZEBO_MAJOR_VERSION >= 4
+          sim_joints_[j]->SetPosition(0, joint_position_command_[j]);
+#else
+          sim_joints_[j]->SetAngle(0, joint_position_command_[j]);
+#endif
+          break;
+
+        case POSITION_PID:
+          {
+            double error;
+            switch (joint_types_[j])
+            {
+              case urdf::Joint::REVOLUTE:
+                angles::shortest_angular_distance_with_limits(joint_position_[j],
+                                                              joint_position_command_[j],
+                                                              joint_lower_limits_[j],
+                                                              joint_upper_limits_[j],
+                                                              error);
+                break;
+              case urdf::Joint::CONTINUOUS:
+                error = angles::shortest_angular_distance(joint_position_[j],
+                                                          joint_position_command_[j]);
+                break;
+              default:
+                error = joint_position_command_[j] - joint_position_[j];
+            }
+
+            const double effort_limit = joint_effort_limits_[j];
+            const double effort = clamp(pid_controllers_[j].computeCommand(error, period),
+                                        -effort_limit, effort_limit);
+            sim_joints_[j]->SetForce(0, effort);
+          }
+          break;
+
+        case VELOCITY:
+          sim_joints_[j]->SetVelocity(0, joint_velocity_command_[j]);
+          break;
+
+        case VELOCITY_PID:
+          const double error = joint_velocity_command_[j] - joint_velocity_[j];
+          const double effort_limit = joint_effort_limits_[j];
+          const double effort = clamp(pid_controllers_[j].computeCommand(error, period),
+                                      -effort_limit, effort_limit);
+          sim_joints_[j]->SetForce(0, effort);
+          break;
+      }
+    }
+  }
+
+private:
+  // Methods used to control a joint.
+  enum ControlMethod {EFFORT, POSITION, POSITION_PID, VELOCITY, VELOCITY_PID};
+
+  // Register the limits of the joint specified by joint_name and joint_handle. The limits are
+  // retrieved from joint_limit_nh. If urdf_model is not NULL, limits are retrieved from it also.
+  // Return the joint's type, lower position limit, upper position limit, and effort limit.
+  void registerJointLimits(const std::string& joint_name,
+                           const hardware_interface::JointHandle& joint_handle,
+                           const ControlMethod ctrl_method,
+                           const ros::NodeHandle& joint_limit_nh,
+                           const urdf::Model *const urdf_model,
+                           int *const joint_type, double *const lower_limit,
+                           double *const upper_limit, double *const effort_limit)
+  {
+    *joint_type = urdf::Joint::UNKNOWN;
+    *lower_limit = -std::numeric_limits<double>::max();
+    *upper_limit = std::numeric_limits<double>::max();
+    *effort_limit = std::numeric_limits<double>::max();
+
+    joint_limits_interface::JointLimits limits;
+    bool has_limits = false;
+    joint_limits_interface::SoftJointLimits soft_limits;
+    bool has_soft_limits = false;
+
+    if (urdf_model != NULL)
+    {
+      const boost::shared_ptr<const urdf::Joint> urdf_joint = urdf_model->getJoint(joint_name);
+      if (urdf_joint != NULL)
+      {
+        *joint_type = urdf_joint->type;
+        // Get limits from the URDF file.
+        if (joint_limits_interface::getJointLimits(urdf_joint, limits))
+          has_limits = true;
+        if (joint_limits_interface::getSoftJointLimits(urdf_joint, soft_limits))
+          has_soft_limits = true;
+      }
+    }
+    // Get limits from the parameter server.
+    if (joint_limits_interface::getJointLimits(joint_name, joint_limit_nh, limits))
+      has_limits = true;
+
+    if (!has_limits)
+      return;
+
+    if (*joint_type == urdf::Joint::UNKNOWN)
+    {
+      // Infer the joint type.
+
+      if (limits.has_position_limits)
+      {
+        *joint_type = urdf::Joint::REVOLUTE;
+      }
+      else
+      {
+        if (limits.angle_wraparound)
+          *joint_type = urdf::Joint::CONTINUOUS;
+        else
+          *joint_type = urdf::Joint::PRISMATIC;
+      }
+    }
+
+    if (limits.has_position_limits)
+    {
+      *lower_limit = limits.min_position;
+      *upper_limit = limits.max_position;
+    }
+    if (limits.has_effort_limits)
+      *effort_limit = limits.max_effort;
+
+    if (has_soft_limits)
+    {
+      switch (ctrl_method)
+      {
+        case EFFORT:
+          {
+            const joint_limits_interface::EffortJointSoftLimitsHandle
+              limits_handle(joint_handle, limits, soft_limits);
+            ej_limits_interface_.registerHandle(limits_handle);
+          }
+          break;
+        case POSITION:
+          {
+            const joint_limits_interface::PositionJointSoftLimitsHandle
+              limits_handle(joint_handle, limits, soft_limits);
+            pj_limits_interface_.registerHandle(limits_handle);
+          }
+          break;
+        case VELOCITY:
+          {
+            const joint_limits_interface::VelocityJointSoftLimitsHandle
+              limits_handle(joint_handle, limits, soft_limits);
+            vj_limits_interface_.registerHandle(limits_handle);
+          }
+          break;
+      }
+    }
+    else
+    {
+      switch (ctrl_method)
+      {
+        case EFFORT:
+          {
+            const joint_limits_interface::EffortJointSaturationHandle
+              sat_handle(joint_handle, limits);
+            ej_sat_interface_.registerHandle(sat_handle);
+          }
+          break;
+        case POSITION:
+          {
+            const joint_limits_interface::PositionJointSaturationHandle
+              sat_handle(joint_handle, limits);
+            pj_sat_interface_.registerHandle(sat_handle);
+          }
+          break;
+        case VELOCITY:
+          {
+            const joint_limits_interface::VelocityJointSaturationHandle
+              sat_handle(joint_handle, limits);
+            vj_sat_interface_.registerHandle(sat_handle);
+          }
+          break;
+      }
+    }
+  }
+
+  unsigned int n_dof_;
+
+  hardware_interface::JointStateInterface    js_interface_;
+  hardware_interface::EffortJointInterface   ej_interface_;
+  hardware_interface::PositionJointInterface pj_interface_;
+  hardware_interface::VelocityJointInterface vj_interface_;
+
+  joint_limits_interface::EffortJointSaturationInterface   ej_sat_interface_;
+  joint_limits_interface::EffortJointSoftLimitsInterface   ej_limits_interface_;
+  joint_limits_interface::PositionJointSaturationInterface pj_sat_interface_;
+  joint_limits_interface::PositionJointSoftLimitsInterface pj_limits_interface_;
+  joint_limits_interface::VelocityJointSaturationInterface vj_sat_interface_;
+  joint_limits_interface::VelocityJointSoftLimitsInterface vj_limits_interface_;
+
+  std::vector<std::string> joint_names_;
+  std::vector<int> joint_types_;
+  std::vector<double> joint_lower_limits_;
+  std::vector<double> joint_upper_limits_;
+  std::vector<double> joint_effort_limits_;
+  std::vector<ControlMethod> joint_control_methods_;
+  std::vector<control_toolbox::Pid> pid_controllers_;
+  std::vector<double> joint_position_;
+  std::vector<double> joint_velocity_;
+  std::vector<double> joint_effort_;
+  std::vector<double> joint_effort_command_;
+  std::vector<double> joint_position_command_;
+  std::vector<double> joint_velocity_command_;
+
+  std::vector<gazebo::physics::JointPtr> sim_joints_;
+};
+
+typedef boost::shared_ptr<DefaultRobotHWSim> DefaultRobotHWSimPtr;
+
+}
+
+PLUGINLIB_EXPORT_CLASS(gazebo_ros_control::DefaultRobotHWSim, gazebo_ros_control::RobotHWSim)
+
+#endif // #ifndef __GAZEBO_ROS_CONTROL_PLUGIN_DEFAULT_ROBOT_HW_SIM_H_
index 413c4a3..d9655d0 100644 (file)
-#include <pluginlib/class_list_macros.h>
+#ifndef _GAZEBO_ROS_CONTROL___WAM_ROBOT_HW_SIM_H_
+#define _GAZEBO_ROS_CONTROL___WAM_ROBOT_HW_SIM_H_
 
+// ros_control
+#include <control_toolbox/pid.h>
 #include <hardware_interface/joint_command_interface.h>
 #include <hardware_interface/robot_hw.h>
+#include <joint_limits_interface/joint_limits.h>
+#include <joint_limits_interface/joint_limits_interface.h>
+#include <joint_limits_interface/joint_limits_rosparam.h>
+#include <joint_limits_interface/joint_limits_urdf.h>
 
-#include <gazebo_ros_control/robot_hw_sim.h>
+// Gazebo
+#include <gazebo/common/common.hh>
+#include <gazebo/physics/physics.hh>
+#include <gazebo/gazebo.hh>
 
+// ROS
+#include <ros/ros.h>
 #include <angles/angles.h>
+#include <pluginlib/class_list_macros.h>
 
-#include <gazebo/gazebo.hh>
-#include <gazebo/physics/physics.hh>
-#include <gazebo/common/common.hh>
+// gazebo_ros_control
+#include <gazebo_ros_control/robot_hw_sim.h>
+
+// URDF
+#include <urdf/model.h>
+
+namespace
+{
 
-namespace wam_gazebo_ros_control
+double clamp(const double val, const double min_val, const double max_val)
 {
-  class WamRobotHWSim:public gazebo_ros_control::RobotHWSim
+  return std::min(std::max(val, min_val), max_val);
+}
+
+}
+
+namespace gazebo_ros_control
+{
+
+class WamRobotHWSim : public gazebo_ros_control::RobotHWSim
+{
+public:
+
+  bool initSim(
+    const std::string& robot_namespace,
+    ros::NodeHandle model_nh,
+    gazebo::physics::ModelPtr parent_model,
+    const urdf::Model *const urdf_model,
+    std::vector<transmission_interface::TransmissionInfo> transmissions)
   {
-    unsigned int n_dof_;
-
-    hardware_interface::JointStateInterface js_interface_;
-    hardware_interface::EffortJointInterface ej_interface_;
-
-    std::vector<std::string> joint_name_;
-    std::vector<double> joint_position_;
-    std::vector<double> joint_velocity_;
-    std::vector<double> joint_effort_;
-    std::vector<double> joint_effort_command_;
-
-    std::vector<gazebo::physics::JointPtr> sim_joints_;
-    
-    public:
-    
-    WamRobotHWSim(void):n_dof_(7),joint_name_(n_dof_),joint_position_(n_dof_),
-      joint_velocity_(n_dof_),joint_effort_(n_dof_),joint_effort_command_(n_dof_)
-    {
-      joint_name_[0]="wam_joint_1";
-      joint_name_[1]="wam_joint_2";
-      joint_name_[2]="wam_joint_3";
-      joint_name_[3]="wam_joint_4";
-      joint_name_[4]="wam_joint_5";
-      joint_name_[5]="wam_joint_6";
-      joint_name_[6]="wam_joint_7";
-      
-      for(unsigned int j=0;j < n_dof_;j++)
-      {
-        joint_position_[j]=0.0;
-        joint_velocity_[j]=0.0;
-        joint_effort_[j]=0.0;
-        js_interface_.registerHandle(hardware_interface::JointStateHandle(joint_name_[j],&joint_position_[j],
-          &joint_velocity_[j],&joint_effort_[j]));
-
-        joint_effort_command_[j] = 0.0;
-        ej_interface_.registerHandle(hardware_interface::JointHandle(js_interface_.getHandle(joint_name_[j]),
-          &joint_effort_command_[j]));
-      }
-
-      registerInterface(&js_interface_);
-      registerInterface(&ej_interface_);
-    }
+    // getJointLimits() searches joint_limit_nh for joint limit parameters. The format of each
+    // parameter's name is "joint_limits/<joint name>". An example is "joint_limits/axle_joint".
+    const ros::NodeHandle joint_limit_nh(model_nh, robot_namespace);
 
-    bool initSim(const std::string& robot_namespace,
-      ros::NodeHandle nh,gazebo::physics::ModelPtr model,
-      const urdf::Model *const urdf_model,
-      std::vector<transmission_interface::TransmissionInfo> transmissions)
+    // Resize vectors to our DOF
+    n_dof_ = transmissions.size();
+    joint_names_.resize(n_dof_);
+    joint_types_.resize(n_dof_);
+    joint_lower_limits_.resize(n_dof_);
+    joint_upper_limits_.resize(n_dof_);
+    joint_effort_limits_.resize(n_dof_);
+    joint_control_methods_.resize(n_dof_);
+    pid_controllers_.resize(n_dof_);
+    joint_position_.resize(n_dof_);
+    joint_velocity_.resize(n_dof_);
+    joint_effort_.resize(n_dof_);
+    joint_effort_command_.resize(n_dof_);
+    joint_position_command_.resize(n_dof_);
+    joint_velocity_command_.resize(n_dof_);
+
+    // Initialize values
+    for(unsigned int j=0; j < n_dof_; j++)
     {
-      for(unsigned int j=0;j < n_dof_;j++)
+      // Check that this transmission has one joint
+      if(transmissions[j].joints_.size() == 0)
+      {
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has no associated joints.");
+        continue;
+      }
+      else if(transmissions[j].joints_.size() > 1)
       {
-        ROS_INFO_STREAM("Getting pointer to gazebo joint: " << joint_name_[j]);
-        gazebo::physics::JointPtr joint=model->GetJoint(joint_name_[j]);
-        if(joint) sim_joints_.push_back(joint);
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has more than one joint. Currently the default robot hardware simulation "
+          << " interface only supports one.");
+        continue;
+      }
+
+      // Check that this transmission has one actuator
+      if(transmissions[j].actuators_.size() == 0)
+      {
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has no associated actuators.");
+        continue;
+      }
+      else if(transmissions[j].actuators_.size() > 1)
+      {
+        ROS_WARN_STREAM_NAMED("default_robot_hw_sim","Transmission " << transmissions[j].name_
+          << " has more than one actuator. Currently the default robot hardware simulation "
+          << " interface only supports one.");
+        continue;
+      }
+
+      // Add data from transmission
+      joint_names_[j] = transmissions[j].joints_[0].name_;
+      joint_position_[j] = 1.0;
+      joint_velocity_[j] = 0.0;
+      joint_effort_[j] = 1.0;  // N/m for continuous joints
+      joint_effort_command_[j] = 0.0;
+      joint_position_command_[j] = 0.0;
+      joint_velocity_command_[j] = 0.0;
+
+#if ROS_VERSION_MINOR > 10 || ROS_VERSION_MAJOR > 1
+      const std::string &hardware_interface =
+        transmissions[j].actuators_[0].hardware_interfaces_[0];
+#else
+      const std::string &hardware_interface =
+        transmissions[j].actuators_[0].hardware_interface_;
+#endif
+
+      // Debug
+      ROS_DEBUG_STREAM_NAMED("default_robot_hw_sim","Loading joint '" << joint_names_[j]
+        << "' of type '" << hardware_interface << "'");
+
+      // Create joint state interface for all joints
+      js_interface_.registerHandle(hardware_interface::JointStateHandle(
+          joint_names_[j], &joint_position_[j], &joint_velocity_[j], &joint_effort_[j]));
+
+      // Decide what kind of command interface this actuator/joint has
+      hardware_interface::JointHandle joint_handle;
+      if(hardware_interface == "EffortJointInterface")
+      {
+        // Create effort joint interface
+        joint_control_methods_[j] = EFFORT;
+        joint_handle = hardware_interface::JointHandle(js_interface_.getHandle(joint_names_[j]),
+                                                       &joint_effort_command_[j]);
+        ej_interface_.registerHandle(joint_handle);
+      }
+      else if(hardware_interface == "PositionJointInterface")
+      {
+        // Create position joint interface
+        joint_control_methods_[j] = POSITION;
+        joint_handle = hardware_interface::JointHandle(js_interface_.getHandle(joint_names_[j]),
+                                                       &joint_position_command_[j]);
+        pj_interface_.registerHandle(joint_handle);
+      }
+      else if(hardware_interface == "VelocityJointInterface")
+      {
+        // Create velocity joint interface
+        joint_control_methods_[j] = VELOCITY;
+        joint_handle = hardware_interface::JointHandle(js_interface_.getHandle(joint_names_[j]),
+                                                       &joint_velocity_command_[j]);
+        vj_interface_.registerHandle(joint_handle);
+      }
+      else
+      {
+        ROS_FATAL_STREAM_NAMED("default_robot_hw_sim","No matching hardware interface found for '"
+          << hardware_interface );
+        return false;
+      }
+
+      // Get the gazebo joint that corresponds to the robot joint.
+      //ROS_DEBUG_STREAM_NAMED("default_robot_hw_sim", "Getting pointer to gazebo joint: "
+      //  << joint_names_[j]);
+      gazebo::physics::JointPtr joint = parent_model->GetJoint(joint_names_[j]);
+      if (!joint)
+      {
+        ROS_ERROR_STREAM("This robot has a joint named \"" << joint_names_[j]
+          << "\" which is not in the gazebo model.");
+        return false;
+      }
+      sim_joints_.push_back(joint);
+
+      registerJointLimits(joint_names_[j], joint_handle, joint_control_methods_[j],
+                          joint_limit_nh, urdf_model,
+                          &joint_types_[j], &joint_lower_limits_[j], &joint_upper_limits_[j],
+                          &joint_effort_limits_[j]);
+      if (joint_control_methods_[j] != EFFORT)
+      {
+        // Initialize the PID controller. If no PID gain values are found, use joint->SetAngle() or
+        // joint->SetVelocity() to control the joint.
+        const ros::NodeHandle nh(model_nh, robot_namespace + "/gazebo_ros_control/pid_gains/" +
+                                 joint_names_[j]);
+        if (pid_controllers_[j].init(nh))
+        {
+          switch (joint_control_methods_[j])
+          {
+            case POSITION:
+              joint_control_methods_[j] = POSITION_PID;
+              break;
+            case VELOCITY:
+              joint_control_methods_[j] = VELOCITY_PID;
+              break;
+          }
+        }
         else
         {
-          ROS_ERROR_STREAM("This robot has a joint named \"" << joint_name_[j]
-              <<"\" which is not in the gazebo model.");
-          return false;
+          // joint->SetMaxForce() must be called if joint->SetAngle() or joint->SetVelocity() are
+          // going to be called. joint->SetMaxForce() must *not* be called if joint->SetForce() is
+          // going to be called.
+          joint->SetMaxForce(0, joint_effort_limits_[j]);
         }
       }
-      return true;
     }
 
-    void readSim(ros::Time time,ros::Duration period)
+    // Register interfaces
+    registerInterface(&js_interface_);
+    registerInterface(&ej_interface_);
+    registerInterface(&pj_interface_);
+    registerInterface(&vj_interface_);
+
+    return true;
+  }
+
+  void readSim(ros::Time time, ros::Duration period)
+  {
+    for(unsigned int j=0; j < n_dof_; j++)
+    {
+      // Gazebo has an interesting API...
+      if (joint_types_[j] == urdf::Joint::PRISMATIC)
+      {
+        joint_position_[j] = sim_joints_[j]->GetAngle(0).Radian();
+      }
+      else
+      {
+        joint_position_[j] += angles::shortest_angular_distance(joint_position_[j],
+                              sim_joints_[j]->GetAngle(0).Radian());
+      }
+      joint_velocity_[j] = sim_joints_[j]->GetVelocity(0);
+      joint_effort_[j] = sim_joints_[j]->GetForce((unsigned int)(0));
+    }
+  }
+
+  void writeSim(ros::Time time, ros::Duration period)
+  {
+    ej_sat_interface_.enforceLimits(period);
+    ej_limits_interface_.enforceLimits(period);
+    pj_sat_interface_.enforceLimits(period);
+    pj_limits_interface_.enforceLimits(period);
+    vj_sat_interface_.enforceLimits(period);
+    vj_limits_interface_.enforceLimits(period);
+
+    for(unsigned int j=0; j < n_dof_; j++)
+    {
+      switch (joint_control_methods_[j])
+      {
+        case EFFORT:
+          {
+            const double effort = joint_effort_command_[j];
+            sim_joints_[j]->SetForce(0, effort);
+          }
+          break;
+
+        case POSITION:
+#if GAZEBO_MAJOR_VERSION >= 4
+          sim_joints_[j]->SetPosition(0, joint_position_command_[j]);
+#else
+          sim_joints_[j]->SetAngle(0, joint_position_command_[j]);
+#endif
+          break;
+
+        case POSITION_PID:
+          {
+            double error;
+            switch (joint_types_[j])
+            {
+              case urdf::Joint::REVOLUTE:
+                angles::shortest_angular_distance_with_limits(joint_position_[j],
+                                                              joint_position_command_[j],
+                                                              joint_lower_limits_[j],
+                                                              joint_upper_limits_[j],
+                                                              error);
+                break;
+              case urdf::Joint::CONTINUOUS:
+                error = angles::shortest_angular_distance(joint_position_[j],
+                                                          joint_position_command_[j]);
+                break;
+              default:
+                error = joint_position_command_[j] - joint_position_[j];
+            }
+
+            const double effort_limit = joint_effort_limits_[j];
+            const double effort = clamp(pid_controllers_[j].computeCommand(error, period),
+                                        -effort_limit, effort_limit);
+            sim_joints_[j]->SetForce(0, effort);
+          }
+          break;
+
+        case VELOCITY:
+          sim_joints_[j]->SetVelocity(0, joint_velocity_command_[j]);
+          break;
+
+        case VELOCITY_PID:
+          const double error = joint_velocity_command_[j] - joint_velocity_[j];
+          const double effort_limit = joint_effort_limits_[j];
+          const double effort = clamp(pid_controllers_[j].computeCommand(error, period),
+                                      -effort_limit, effort_limit);
+          sim_joints_[j]->SetForce(0, effort);
+          break;
+      }
+    }
+  }
+
+private:
+  // Methods used to control a joint.
+  enum ControlMethod {EFFORT, POSITION, POSITION_PID, VELOCITY, VELOCITY_PID};
+
+  // Register the limits of the joint specified by joint_name and joint_handle. The limits are
+  // retrieved from joint_limit_nh. If urdf_model is not NULL, limits are retrieved from it also.
+  // Return the joint's type, lower position limit, upper position limit, and effort limit.
+  void registerJointLimits(const std::string& joint_name,
+                           const hardware_interface::JointHandle& joint_handle,
+                           const ControlMethod ctrl_method,
+                           const ros::NodeHandle& joint_limit_nh,
+                           const urdf::Model *const urdf_model,
+                           int *const joint_type, double *const lower_limit,
+                           double *const upper_limit, double *const effort_limit)
+  {
+    *joint_type = urdf::Joint::UNKNOWN;
+    *lower_limit = -std::numeric_limits<double>::max();
+    *upper_limit = std::numeric_limits<double>::max();
+    *effort_limit = std::numeric_limits<double>::max();
+
+    joint_limits_interface::JointLimits limits;
+    bool has_limits = false;
+    joint_limits_interface::SoftJointLimits soft_limits;
+    bool has_soft_limits = false;
+
+    if (urdf_model != NULL)
     {
-      for(unsigned int j=0; j < n_dof_;j++)
+      const boost::shared_ptr<const urdf::Joint> urdf_joint = urdf_model->getJoint(joint_name);
+      if (urdf_joint != NULL)
       {
-        joint_position_[j]=sim_joints_[j]->GetAngle(0).Radian();
-        joint_velocity_[j]=sim_joints_[j]->GetVelocity(0);
-        joint_effort_[j]=joint_effort_command_[j];
+        *joint_type = urdf_joint->type;
+        // Get limits from the URDF file.
+        if (joint_limits_interface::getJointLimits(urdf_joint, limits))
+          has_limits = true;
+        if (joint_limits_interface::getSoftJointLimits(urdf_joint, soft_limits))
+          has_soft_limits = true;
       }
     }
+    // Get limits from the parameter server.
+    if (joint_limits_interface::getJointLimits(joint_name, joint_limit_nh, limits))
+      has_limits = true;
 
-    void writeSim(ros::Time time,ros::Duration period) 
+    if (!has_limits)
+      return;
+
+    if (*joint_type == urdf::Joint::UNKNOWN)
+    {
+      // Infer the joint type.
+
+      if (limits.has_position_limits)
+      {
+        *joint_type = urdf::Joint::REVOLUTE;
+      }
+      else
+      {
+        if (limits.angle_wraparound)
+          *joint_type = urdf::Joint::CONTINUOUS;
+        else
+          *joint_type = urdf::Joint::PRISMATIC;
+      }
+    }
+
+    if (limits.has_position_limits)
+    {
+      *lower_limit = limits.min_position;
+      *upper_limit = limits.max_position;
+    }
+    if (limits.has_effort_limits)
+      *effort_limit = limits.max_effort;
+
+    if (has_soft_limits)
     {
-      for(unsigned int j=0;j < n_dof_;j++)
-        sim_joints_[j]->SetForce(0,joint_effort_command_[j]);
+      switch (ctrl_method)
+      {
+        case EFFORT:
+          {
+            const joint_limits_interface::EffortJointSoftLimitsHandle
+              limits_handle(joint_handle, limits, soft_limits);
+            ej_limits_interface_.registerHandle(limits_handle);
+          }
+          break;
+        case POSITION:
+          {
+            const joint_limits_interface::PositionJointSoftLimitsHandle
+              limits_handle(joint_handle, limits, soft_limits);
+            pj_limits_interface_.registerHandle(limits_handle);
+          }
+          break;
+        case VELOCITY:
+          {
+            const joint_limits_interface::VelocityJointSoftLimitsHandle
+              limits_handle(joint_handle, limits, soft_limits);
+            vj_limits_interface_.registerHandle(limits_handle);
+          }
+          break;
+      }
     }
+    else
+    {
+      switch (ctrl_method)
+      {
+        case EFFORT:
+          {
+            const joint_limits_interface::EffortJointSaturationHandle
+              sat_handle(joint_handle, limits);
+            ej_sat_interface_.registerHandle(sat_handle);
+          }
+          break;
+        case POSITION:
+          {
+            const joint_limits_interface::PositionJointSaturationHandle
+              sat_handle(joint_handle, limits);
+            pj_sat_interface_.registerHandle(sat_handle);
+          }
+          break;
+        case VELOCITY:
+          {
+            const joint_limits_interface::VelocityJointSaturationHandle
+              sat_handle(joint_handle, limits);
+            vj_sat_interface_.registerHandle(sat_handle);
+          }
+          break;
+      }
+    }
+  }
+
+  unsigned int n_dof_;
+
+  hardware_interface::JointStateInterface    js_interface_;
+  hardware_interface::EffortJointInterface   ej_interface_;
+  hardware_interface::PositionJointInterface pj_interface_;
+  hardware_interface::VelocityJointInterface vj_interface_;
+
+  joint_limits_interface::EffortJointSaturationInterface   ej_sat_interface_;
+  joint_limits_interface::EffortJointSoftLimitsInterface   ej_limits_interface_;
+  joint_limits_interface::PositionJointSaturationInterface pj_sat_interface_;
+  joint_limits_interface::PositionJointSoftLimitsInterface pj_limits_interface_;
+  joint_limits_interface::VelocityJointSaturationInterface vj_sat_interface_;
+  joint_limits_interface::VelocityJointSoftLimitsInterface vj_limits_interface_;
+
+  std::vector<std::string> joint_names_;
+  std::vector<int> joint_types_;
+  std::vector<double> joint_lower_limits_;
+  std::vector<double> joint_upper_limits_;
+  std::vector<double> joint_effort_limits_;
+  std::vector<ControlMethod> joint_control_methods_;
+  std::vector<control_toolbox::Pid> pid_controllers_;
+  std::vector<double> joint_position_;
+  std::vector<double> joint_velocity_;
+  std::vector<double> joint_effort_;
+  std::vector<double> joint_effort_command_;
+  std::vector<double> joint_position_command_;
+  std::vector<double> joint_velocity_command_;
+
+  std::vector<gazebo::physics::JointPtr> sim_joints_;
+};
+
+typedef boost::shared_ptr<WamRobotHWSim> WamRobotHWSimPtr;
 
-  };
 }
 
-PLUGINLIB_EXPORT_CLASS(wam_gazebo_ros_control::WamRobotHWSim,gazebo_ros_control::RobotHWSim)
+PLUGINLIB_EXPORT_CLASS(gazebo_ros_control::WamRobotHWSim, gazebo_ros_control::RobotHWSim)
+
+#endif // #ifndef __GAZEBO_ROS_CONTROL_PLUGIN_WAM_ROBOT_HW_SIM_H_
diff --git a/wam_gazebo_ros_control/src/wam_robot_hw_sim.cpp1 b/wam_gazebo_ros_control/src/wam_robot_hw_sim.cpp1
new file mode 100644 (file)
index 0000000..fd8ee43
--- /dev/null
@@ -0,0 +1,101 @@
+#include <pluginlib/class_list_macros.h>
+
+#include <hardware_interface/joint_command_interface.h>
+#include <hardware_interface/robot_hw.h>
+
+#include <gazebo_ros_control/robot_hw_sim.h>
+
+#include <angles/angles.h>
+
+#include <gazebo/gazebo.hh>
+#include <gazebo/physics/physics.hh>
+#include <gazebo/common/common.hh>
+
+namespace wam_gazebo_ros_control
+{
+  class WamRobotHWSim:public gazebo_ros_control::RobotHWSim
+  {
+    unsigned int n_dof_;
+
+    hardware_interface::JointStateInterface js_interface_;
+    hardware_interface::EffortJointInterface ej_interface_;
+
+    std::vector<std::string> joint_name_;
+    std::vector<double> joint_position_;
+    std::vector<double> joint_velocity_;
+    std::vector<double> joint_effort_;
+    std::vector<double> joint_effort_command_;
+
+    std::vector<gazebo::physics::JointPtr> sim_joints_;
+    
+    public:
+    
+    WamRobotHWSim(void):n_dof_(7),joint_name_(n_dof_),joint_position_(n_dof_),
+      joint_velocity_(n_dof_),joint_effort_(n_dof_),joint_effort_command_(n_dof_)
+    {
+      joint_name_[0]="wam_joint_1";
+      joint_name_[1]="wam_joint_2";
+      joint_name_[2]="wam_joint_3";
+      joint_name_[3]="wam_joint_4";
+      joint_name_[4]="wam_joint_5";
+      joint_name_[5]="wam_joint_6";
+      joint_name_[6]="wam_joint_7";
+      
+      for(unsigned int j=0;j < n_dof_;j++)
+      {
+        joint_position_[j]=0.0;
+        joint_velocity_[j]=0.0;
+        joint_effort_[j]=0.0;
+
+        js_interface_.registerHandle(hardware_interface::JointStateHandle(joint_name_[j],&joint_position_[j],
+          &joint_velocity_[j],&joint_effort_[j]));
+
+        joint_effort_command_[j] = 0.0;
+        ej_interface_.registerHandle(hardware_interface::JointHandle(js_interface_.getHandle(joint_name_[j]),
+          &joint_effort_command_[j]));
+      }
+
+      registerInterface(&js_interface_);
+      registerInterface(&ej_interface_);
+    }
+
+    bool initSim(const std::string& robot_namespace,
+      ros::NodeHandle nh,gazebo::physics::ModelPtr model,
+      const urdf::Model *const urdf_model,
+      std::vector<transmission_interface::TransmissionInfo> transmissions)
+    {
+      for(unsigned int j=0;j < n_dof_;j++)
+      {
+        ROS_INFO_STREAM("Getting pointer to gazebo joint: " << joint_name_[j]);
+        gazebo::physics::JointPtr joint=model->GetJoint(joint_name_[j]);
+        if(joint) sim_joints_.push_back(joint);
+        else
+        {
+          ROS_ERROR_STREAM("This robot has a joint named \"" << joint_name_[j]
+              <<"\" which is not in the gazebo model.");
+          return false;
+        }
+      }
+      return true;
+    }
+
+    void readSim(ros::Time time,ros::Duration period)
+    {
+      for(unsigned int j=0; j < n_dof_;j++)
+      {
+        joint_position_[j]=sim_joints_[j]->GetAngle(0).Radian();
+        joint_velocity_[j]=sim_joints_[j]->GetVelocity(0);
+        joint_effort_[j]=joint_effort_command_[j];
+      }
+    }
+
+    void writeSim(ros::Time time,ros::Duration period) 
+    {
+      for(unsigned int j=0;j < n_dof_;j++)
+        sim_joints_[j]->SetForce(0,joint_effort_command_[j]);
+    }
+
+  };
+}
+
+PLUGINLIB_EXPORT_CLASS(wam_gazebo_ros_control::WamRobotHWSim,gazebo_ros_control::RobotHWSim)