From 54e4fd740467d15b00008571c783a082cba41950 Mon Sep 17 00:00:00 2001 From: Kei Okada Date: Sun, 25 Jan 2015 01:11:27 +0900 Subject: [PATCH] add joint_state_publisher plugin --- .../joint_state_publisher.py | 106 ++++++++++++++++++ 1 file changed, 106 insertions(+) create mode 100755 dynamixel_controllers/src/dynamixel_controllers/joint_state_publisher.py diff --git a/dynamixel_controllers/src/dynamixel_controllers/joint_state_publisher.py b/dynamixel_controllers/src/dynamixel_controllers/joint_state_publisher.py new file mode 100755 index 0000000..be57e94 --- /dev/null +++ b/dynamixel_controllers/src/dynamixel_controllers/joint_state_publisher.py @@ -0,0 +1,106 @@ +# -*- coding: utf-8 -*- +# +# Software License Agreement (BSD License) +# +# Copyright (c) 2015, Kei Okada. +# 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 Kei Okada 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. + +from threading import Thread + +import rospy + +from sensor_msgs.msg import JointState + +class JointStatePublisher(): + def __init__(self, controller_namespace, controllers): + self.update_rate = 1000 + self.state_update_rate = 50 + self.trajectory = [] + + self.controller_namespace = controller_namespace + self.joint_names = [c.joint_name for c in controllers] + + self.joint_to_controller = {} + for c in controllers: + self.joint_to_controller[c.joint_name] = c + + self.port_to_joints = {} + for c in controllers: + if c.port_namespace not in self.port_to_joints: self.port_to_joints[c.port_namespace] = [] + self.port_to_joints[c.port_namespace].append(c.joint_name) + + self.port_to_io = {} + for c in controllers: + if c.port_namespace in self.port_to_io: continue + self.port_to_io[c.port_namespace] = c.dxl_io + + def initialize(self): + self.msg = JointState() + return True + + + def start(self): + self.running = True + self.state_pub = rospy.Publisher('joint_states', JointState , queue_size=100) + + Thread(target=self.update_state).start() + + + def stop(self): + self.running = False + + def update_state(self): + rate = rospy.Rate(self.state_update_rate) + while self.running and not rospy.is_shutdown(): + self.msg.header.stamp = rospy.Time.now() + self.msg.name = [] + self.msg.position = [] + self.msg.velocity = [] + self.msg.effort = [] + + for port, joints in self.port_to_joints.items(): + vals = [] + for joint in joints: + j = self.joint_names.index(joint) + + motor_id = self.joint_to_controller[joint].motor_id + co = self.joint_to_controller[joint] + io = self.port_to_io[port] + + self.msg.name.append(joint) + self.msg.position.append(self.raw_to_rad(io.get_position(motor_id),co)) + self.msg.velocity.append(self.raw_to_rad(io.get_speed(motor_id),co)) + self.msg.effort.append(io.get_current(motor_id)) + + self.state_pub.publish(self.msg) + rate.sleep() + + def raw_to_rad(self, raw, c): + return (c.initial_position_raw - raw if c.flipped else raw - c.initial_position_raw) * c.RADIANS_PER_ENCODER_TICK