class IK(object):
def __init__(self,
base_link,
tip_link,
timeout=0.005,
epsilon=1e-5,
solve_type="Speed",
urdf_string=None):
"""
Create a TRAC_IK instance and keep track of it.
:param str base_link: Starting link of the chain.
:param str tip_link: Last link of the chain.
:param float timeout: Timeout in seconds for the IK calls.
:param float epsilon: Error epsilon.
:param solve_type str: Type of solver, can be:
Speed (default), Distance, Manipulation1, Manipulation2
:param urdf_string str: Optional arg, if not given URDF is taken from
the param server at /robot_description.
"""
if urdf_string is None:
raise NotImplementedError("Can't load urdf_string")
#urdf_string = rospy.get_param('/robot_description')
self._urdf_string = urdf_string
self._timeout = timeout
self._epsilon = epsilon
self._solve_type = solve_type
self.base_link = base_link
self.tip_link = tip_link
self._ik_solver = TRAC_IK(self.base_link, self.tip_link,
self._urdf_string, self._timeout,
self._epsilon, self._solve_type)
self.number_of_joints = self._ik_solver.getNrOfJointsInChain()
self.joint_names = self._ik_solver.getJointNamesInChain(
self._urdf_string)
self.link_names = self._ik_solver.getLinkNamesInChain()
def get_ik(self,
qinit,
x,
y,
z,
rx,
ry,
rz,
rw,
bx=1e-5,
by=1e-5,
bz=1e-5,
brx=1e-3,
bry=1e-3,
brz=1e-3):
"""
Do the IK call.
:param list of float qinit: Initial status of the joints as seed.
:param float x: X coordinates in base_frame.
:param float y: Y coordinates in base_frame.
:param float z: Z coordinates in base_frame.
:param float rx: X quaternion coordinate.
:param float ry: Y quaternion coordinate.
:param float rz: Z quaternion coordinate.
:param float rw: W quaternion coordinate.
:param float bx: X allowed bound.
:param float by: Y allowed bound.
:param float bz: Z allowed bound.
:param float brx: rotation over X allowed bound.
:param float bry: rotation over Y allowed bound.
:param float brz: rotation over Z allowed bound.
:return: joint values or None if no solution found.
:rtype: tuple of float.
"""
if len(qinit) != self.number_of_joints:
raise Exception(
"qinit has length %i and it should have length %i" %
(len(qinit), self.number_of_joints))
solution = self._ik_solver.CartToJnt(qinit, x, y, z, rx, ry, rz, rw,
bx, by, bz, brx, bry, brz)
if solution:
return solution
else:
return None
def get_joint_limits(self):
"""
Return lower bound limits and upper bound limits for all the joints
in the order of the joint names.
"""
lb = self._ik_solver.getLowerBoundLimits()
ub = self._ik_solver.getUpperBoundLimits()
return lb, ub
def set_joint_limits(self, lower_bounds, upper_bounds):
"""
Set joint limits for all the joints.
:arg list lower_bounds: List of float of the lower bound limits for
all joints.
:arg list upper_bounds: List of float of the upper bound limits for
all joints.
"""
if len(lower_bounds) != self.number_of_joints:
raise Exception(
"lower_bounds array size mismatch, it's size %i, should be %i"
% (len(lower_bounds), self.number_of_joints))
if len(upper_bounds) != self.number_of_joints:
raise Exception(
"upper_bounds array size mismatch, it's size %i, should be %i"
% (len(upper_bounds), self.number_of_joints))
self._ik_solver.setKDLLimits(lower_bounds, upper_bounds)
if __name__ == '__main__':
# roslaunch pr2_description upload_pr2.launch
# Needed beforehand
urdf = rospy.get_param('/robot_description')
# params of constructor:
# base_link, tip_link, urdf_string, timeout, epsilon, solve_type="Speed"
# solve_type can be: Distance, Speed, Manipulation1, Manipulation2
ik_solver = TRAC_IK(
"torso_lift_link",
"r_wrist_roll_link",
urdf,
0.005, # default seconds
1e-5, # default epsilon
"Speed")
print("Number of joints:")
print(ik_solver.getNrOfJointsInChain())
print("Joint names:")
print(ik_solver.getJointNamesInChain(urdf))
print("Link names:")
print(ik_solver.getLinkNamesInChain())
qinit = [0.] * 7
x = y = z = 0.0
rx = ry = rz = 0.0
rw = 1.0
bx = by = bz = 0.001
brx = bry = brz = 0.1
# Generate a set of random coords in the arm workarea approx
NUM_COORDS = 200
rand_coords = []