virtual_shadowhand.cpp

Go to the documentation of this file.

#include "sr_hand/hand/virtual_shadowhand.h"

#include <time.h>
#include <ros/ros.h>

#include <boost/algorithm/string.hpp>

#ifdef GAZEBO
#include <std_msgs/Float64.h>
#endif

#include <urdf/model.h>
#include <sr_utilities/sr_math_utils.hpp>

namespace shadowrobot
{
  VirtualShadowhand::VirtualShadowhand() :
    SRArticulatedRobot(), n_tilde("~")
  {
#ifdef GAZEBO
    ROS_INFO("This ROS interface is built for Gazebo.");
    //initialises the subscriber to the Gazebo joint_states messages
    std::string prefix;
    std::string searched_param;

    n_tilde.searchParam("gazebo_joint_states_prefix", searched_param);
    n_tilde.param(searched_param, prefix, std::string());
    std::string full_topic = prefix + "joint_states";
    gazebo_subscriber = node.subscribe(full_topic, 2, &VirtualShadowhand::gazeboCallback, this);
#else
    ROS_INFO("This ROS interface is not built for Gazebo.");
#endif

    srand(time(NULL));
    initializeMap();
  }

  VirtualShadowhand::~VirtualShadowhand()
  {
  }

  void VirtualShadowhand::initializeMap()
  {
    joints_map_mutex.lock();
    parameters_map_mutex.lock();
    controllers_map_mutex.lock();


    JointData tmpData;
    JointData tmpDataZero;
    JointControllerData tmpController;
    tmpDataZero.isJointZero = 1;
    tmpDataZero.max = 180.0;

#ifdef GAZEBO
    std::string topic_prefix = "/sh_";
    std::string topic_suffix = "_mixed_position_velocity_controller/command";
    std::string full_topic = "";
#endif

    //Get the urdf model from the parameter server
    std::string robot_desc_string;
    node.param("hand_description", robot_desc_string, std::string());
    urdf::Model robot_model;
    if (!robot_model.initString(robot_desc_string)){
      ROS_ERROR("Failed to parse urdf file");
      return;
    }

    std::map<std::string, boost::shared_ptr<urdf::Joint> > all_joints = robot_model.joints_;
    std::map<std::string, boost::shared_ptr<urdf::Joint> >::const_iterator iter = all_joints.begin();
    ROS_DEBUG("All the Hand joints: ");

#ifdef GAZEBO
    //index of the publisher to the Gazebo controller
    int tmp_index = 0;
#endif

    for (; iter != all_joints.end(); ++iter )
    {
      if( iter->second->type == urdf::Joint::REVOLUTE)
      {
        std::string current_joint_name = iter->first;

        ROS_DEBUG_STREAM(" joint: " << current_joint_name << '\t' << iter->second );

        //check if we need to create a joint 0
        // create them when we have the joint 2

#ifdef GAZEBO
        //The joint 1 is not controlled directly in Gazebo (the controller
        // controls joint 0)
        bool no_controller = false;
#endif

        bool create_joint_zero = false;
        boost::algorithm::to_lower(current_joint_name);
        if( current_joint_name == "ffj2" || current_joint_name == "mfj2"
            || current_joint_name == "rfj2" || current_joint_name == "lfj2" )
        {
          create_joint_zero = true;

#ifdef GAZEBO
          no_controller = true;
#endif
        }

#ifdef GAZEBO
        if( current_joint_name == "ffj1" || current_joint_name == "mfj1"
            || current_joint_name == "rfj1" || current_joint_name == "lfj1" )
        {
          no_controller = true;
        }
#endif

        if( create_joint_zero )
        {
#ifdef GAZEBO
          full_topic = topic_prefix + std::string(1, current_joint_name[0]) + "fj0" + topic_suffix;
          gazebo_publishers.push_back(node.advertise<std_msgs::Float64>(full_topic, 2));
          tmpDataZero.publisher_index = tmp_index;

          ++tmp_index;
#endif
          boost::algorithm::to_upper(current_joint_name);
          joints_map[std::string(1, current_joint_name[0]) + "FJ0"] = tmpDataZero;
          controllers_map[std::string(1, current_joint_name[0]) + "FJ0"] = tmpController;

          tmpData.min = sr_math_utils::to_degrees(iter->second->limits->lower);
          tmpData.max = sr_math_utils::to_degrees(iter->second->limits->upper);
          joints_map[current_joint_name] = tmpData;
          controllers_map[current_joint_name] = tmpController;
        }
        else
        {
#ifdef GAZEBO
          if( !no_controller )
          {
            boost::algorithm::to_lower(current_joint_name);
            full_topic = topic_prefix + current_joint_name + topic_suffix;
            gazebo_publishers.push_back(node.advertise<std_msgs::Float64>(full_topic, 2));
            tmpData.publisher_index = tmp_index;
            ++tmp_index;
          }
#endif
          tmpData.min = sr_math_utils::to_degrees(iter->second->limits->lower);
          tmpData.max = sr_math_utils::to_degrees(iter->second->limits->upper);
          boost::algorithm::to_upper(current_joint_name);
          joints_map[ current_joint_name ] = tmpData;
          controllers_map[ current_joint_name ] = tmpController;
        }
      }
    }

    parameters_map["d"] = PARAM_d;
    parameters_map["i"] = PARAM_i;
    parameters_map["p"] = PARAM_p;
    parameters_map["target"] = PARAM_target;
    parameters_map["sensor"] = PARAM_sensor;

    parameters_map["valve"] = PARAM_valve;
    parameters_map["dead"] = PARAM_deadband;
    parameters_map["deadband"] = PARAM_deadband;
    parameters_map["imax"] = PARAM_imax;
    parameters_map["offset"] = PARAM_output_offset;
    parameters_map["shift"] = PARAM_shift;
    parameters_map["max"] = PARAM_output_max;

    parameters_map["motor_maxforce"] = PARAM_motor_maxforce;
    parameters_map["motor_safeforce"] = PARAM_motor_safeforce;

    parameters_map["force_p"] = PARAM_force_p;
    parameters_map["force_i"] = PARAM_force_i;
    parameters_map["force_d"] = PARAM_force_d;

    parameters_map["force_imax"] = PARAM_force_imax;
    parameters_map["force_out_shift"] = PARAM_force_out_shift;
    parameters_map["force_deadband"] = PARAM_force_deadband;
    parameters_map["force_offset"] = PARAM_force_offset;

    parameters_map["sensor_imax"] = PARAM_sensor_imax;
    parameters_map["sensor_out_shift"] = PARAM_sensor_out_shift;
    parameters_map["sensor_deadband"] = PARAM_sensor_deadband;
    parameters_map["sensor_offset"] = PARAM_sensor_offset;
    parameters_map["max_temp"] = PARAM_max_temperature;
    parameters_map["max_temperature"] = PARAM_max_temperature;
    parameters_map["max_current"] = PARAM_max_current;

    controllers_map_mutex.unlock();
    parameters_map_mutex.unlock();
    joints_map_mutex.unlock();
  }

  short VirtualShadowhand::sendupdate( std::string joint_name, double target )
  {
    joints_map_mutex.lock();

    JointsMap::iterator iter = joints_map.find(joint_name);
#ifdef GAZEBO
    std_msgs::Float64 target_msg;
#endif

    //not found
    if( iter == joints_map.end() )
    {
      ROS_DEBUG("Joint %s not found", joint_name.c_str());

      joints_map_mutex.unlock();
      return -1;
    }
    JointData joint_0_data = JointData(iter->second);

    //in total simulation:
    //if joint 0, send 1/2 of the target to joint 1 and other half to
    //2;
    // if using gazebo, we just send the target to the joint 0 controller
    // which is then controlling both joints.
    if( iter->second.isJointZero == 1 )
    {
      //push target and position to the given target for Joint 0
      JointData tmpData0 = JointData(iter->second);
      if( target < tmpData0.min )
        target = tmpData0.min;
      if( target > tmpData0.max )
        target = tmpData0.max;

#ifndef GAZEBO
      tmpData0.position = target;
#endif
      tmpData0.target = target;

      joints_map[joint_name] = tmpData0;

      ++iter;
      JointData tmpData1 = JointData(iter->second);
#ifdef GAZEBO
      //gazebo targets are in radians
      target_msg.data = toRad( target );
      gazebo_publishers[joint_0_data.publisher_index].publish(target_msg);
#else
      tmpData1.position = target / 2.0;
#endif
      tmpData1.target = target / 2.0;

      joints_map[iter->first] = tmpData1;

      ++iter;
      JointData tmpData2 = JointData(iter->second);
#ifdef GAZEBO
      //we've already send the data to the joint 0 controller
#else
      tmpData2.position = target / 2.0;
#endif
      tmpData2.target = target / 2.0;

      joints_map[iter->first] = tmpData2;

      joints_map_mutex.unlock();
      return 0;
    }

    //joint found
    JointData tmpData(iter->second);

    if( target < tmpData.min )
      target = tmpData.min;
    if( target > tmpData.max )
      target = tmpData.max;

#ifdef GAZEBO
    //gazebo targets are in radians
    target_msg.data = toRad(target);
    gazebo_publishers[tmpData.publisher_index].publish(target_msg);
#else
    tmpData.position = target;
#endif
    tmpData.target = target;

    joints_map[joint_name] = tmpData;

    joints_map_mutex.unlock();
    return 0;
  }

  JointData VirtualShadowhand::getJointData( std::string joint_name )
  {
    joints_map_mutex.lock();

    JointsMap::iterator iter = joints_map.find(joint_name);

    //joint found
    if( iter != joints_map.end() )
    {
      //return the position
      iter->second.temperature = ((double)(rand() % 100) / 100.0);
      iter->second.current = ((double)(rand() % 100) / 100.0);
#ifndef GAZEBO
      iter->second.force = ((double)(rand() % 100) / 100.0);
#endif

      JointData tmp = JointData(iter->second);

      joints_map_mutex.unlock();
      return tmp;
    }

    ROS_ERROR("Joint %s not found.", joint_name.c_str());
    JointData noData;
    joints_map_mutex.unlock();
    return noData;
  }

  SRArticulatedRobot::JointsMap VirtualShadowhand::getAllJointsData()
  {
    joints_map_mutex.lock();

    for( JointsMap::const_iterator it = joints_map.begin(); it != joints_map.end(); ++it )
    {
      JointData tmpData = it->second;
      tmpData.temperature = ((double)(rand() % 100) / 100.0);
      tmpData.current = ((double)(rand() % 100) / 100.0);
#ifndef GAZEBO
      tmpData.force = ((double)(rand() % 100) / 100.0);
#endif
      tmpData.jointIndex = 0;
      tmpData.flags = "";

      joints_map[it->first] = tmpData;
    }

    JointsMap tmp_map = JointsMap(joints_map);
    joints_map_mutex.unlock();
    return tmp_map;
  }

  short VirtualShadowhand::setContrl( std::string contrlr_name, JointControllerData ctrlr_data )
  {
    controllers_map_mutex.lock();

    ControllersMap::iterator iter = controllers_map.find(contrlr_name);

    //joint found
    if( iter != controllers_map.end() )
    {
      controllers_map[iter->first] = ctrlr_data;
    }
    else
    {
      ROS_ERROR("Controller %s not found", contrlr_name.c_str());
    }

    controllers_map_mutex.unlock();
    return 0;
  }

  JointControllerData VirtualShadowhand::getContrl( std::string contrlr_name )
  {
    controllers_map_mutex.lock();
    ControllersMap::iterator iter = controllers_map.find(contrlr_name);

    //joint found
    if( iter != controllers_map.end() )
    {
      JointControllerData tmp = JointControllerData(iter->second);
      controllers_map_mutex.unlock();
      return tmp;
    }

    ROS_ERROR("Controller %s not found", contrlr_name.c_str() );
    JointControllerData no_result;
    controllers_map_mutex.unlock();
    return no_result;
  }

  short VirtualShadowhand::setConfig( std::vector<std::string> myConfig )
  {
    ROS_WARN("The set config function is not implemented in the virtual shadowhand.");
    return 0;
  }

  void VirtualShadowhand::getConfig( std::string joint_name )
  {
    ROS_WARN("The get config function is not implemented in the virtual shadowhand.");
  }

  std::vector<DiagnosticData> VirtualShadowhand::getDiagnostics()
  {
    joints_map_mutex.lock();
    std::vector<DiagnosticData> returnVect;

    for( JointsMap::const_iterator it = joints_map.begin(); it != joints_map.end(); ++it )
    {
      DiagnosticData tmpDiag;
      tmpDiag.joint_name = it->first;
      tmpDiag.level = 0;
      tmpDiag.flags = "";
      tmpDiag.target_sensor_num = 0;
      tmpDiag.target = it->second.target;
      tmpDiag.position_sensor_num = 0;
      tmpDiag.position = it-> second.position;

      returnVect.push_back(tmpDiag);
    }

    joints_map_mutex.unlock();
    return returnVect;
  }

#ifdef GAZEBO
  void VirtualShadowhand::gazeboCallback(const sensor_msgs::JointStateConstPtr& msg)
  {
    joints_map_mutex.lock();

    //loop on all the names in the joint_states message
    for(unsigned int index = 0; index < msg->name.size(); ++index)
    {
      std::string joint_name = msg->name[index];
      JointsMap::iterator iter = joints_map.find(joint_name);
      //not found => can be a joint from the arm / hand
      if(iter == joints_map.end())
        continue;

      //joint found
      JointData tmpData(iter->second);

      tmpData.position = toDegrees(msg->position[index]);
      tmpData.force = msg->effort[index];

      joints_map[joint_name] = tmpData;
    }

    //push the sum of J1+J2 to the J0s
    for(JointsMap::const_iterator it = joints_map.begin(); it != joints_map.end(); ++it)
    {
      JointData tmpData = it->second;
      if( tmpData.isJointZero == 1 )
      {
        std::string joint_name = it->first;
        double position = 0.0;

        //get the position from joint 1
        ++it;
        JointData tmpData1 = JointData(it->second);
        position += tmpData1.position;

        //get the position from joint 2
        ++it;
        JointData tmpData2 = JointData(it->second);
        position += tmpData2.position;

        tmpData.position = position;

        joints_map[joint_name] = tmpData;
      }
    }

    joints_map_mutex.unlock();
  }

#endif
} //end namespace


/* For the emacs weenies in the crowd.
Local Variables:
   c-basic-offset: 2
End:
*/