def plan_cartesian_motion_async(self, callback, errback, robot, frames,
start_configuration, group, max_step,
jump_threshold, avoid_collisions,
path_constraints,
attached_collision_meshes):
"""Asynchronous handler of MoveIt cartesian motion planner service."""
base_link = robot.get_base_link_name(group)
ee_link = robot.get_end_effector_link_name(group)
joint_names = robot.get_configurable_joint_names(group)
header = Header(frame_id=base_link)
waypoints = [Pose.from_frame(frame) for frame in frames]
joint_state = JointState(header=header,
name=joint_names,
position=start_configuration.values)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header))
if attached_collision_meshes:
for acm in attached_collision_meshes:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
path_constraints = self._convert_constraints_to_rosmsg(
path_constraints, header)
request = dict(header=header,
start_state=start_state,
group_name=group,
link_name=ee_link,
waypoints=waypoints,
max_step=float(max_step),
jump_threshold=float(jump_threshold),
avoid_collisions=bool(avoid_collisions),
path_constraints=path_constraints)
def convert_to_trajectory(response):
try:
trajectory = JointTrajectory()
trajectory.source_message = response
trajectory.fraction = response.fraction
trajectory.joint_names = response.solution.joint_trajectory.joint_names
joint_types = robot.get_joint_types_by_names(
trajectory.joint_names)
trajectory.points = convert_trajectory_points(
response.solution.joint_trajectory.points, joint_types)
start_state = response.start_state.joint_state
start_state_types = robot.get_joint_types_by_names(
start_state.name)
trajectory.start_configuration = Configuration(
start_state.position, start_state_types)
callback(trajectory)
except Exception as e:
errback(e)
self.GET_CARTESIAN_PATH(self, request, convert_to_trajectory, errback)
def inverse_kinematics_async(self,
callback,
errback,
frame,
base_link,
group,
joint_names,
joint_positions,
avoid_collisions=True,
constraints=None,
attempts=8,
attached_collision_meshes=None):
"""Asynchronous handler of MoveIt IK service."""
header = Header(frame_id=base_link)
pose = Pose.from_frame(frame)
pose_stamped = PoseStamped(header, pose)
joint_state = JointState(name=joint_names,
position=joint_positions,
header=header)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header))
if attached_collision_meshes:
for acm in attached_collision_meshes:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
ik_request = PositionIKRequest(group_name=group,
robot_state=start_state,
constraints=constraints,
pose_stamped=pose_stamped,
avoid_collisions=avoid_collisions,
attempts=attempts)
self.GET_POSITION_IK(self, (ik_request, ), callback, errback)
def remove_attached_collision_mesh_async(self, callback, errback, id):
aco = AttachedCollisionObject()
aco.object.id = id
aco.object.operation = CollisionObject.REMOVE
world = PlanningSceneWorld(collision_objects=[aco.object])
scene = PlanningScene(world=world, is_diff=True)
request = dict(scene=scene)
self.APPLY_PLANNING_SCENE(self.ros_client, request, callback, errback)
def add_attached_collision_mesh_async(self, callback, errback, attached_collision_mesh):
aco = AttachedCollisionObject.from_attached_collision_mesh(
attached_collision_mesh)
aco.object.operation = CollisionObject.ADD
world = PlanningSceneWorld(collision_objects=[aco.object])
scene = PlanningScene(world=world, is_diff=True)
request = dict(scene=scene)
self.APPLY_PLANNING_SCENE(self.ros_client, request, callback, errback)
def add_attached_collision_mesh_async(self, callback, errback, attached_collision_mesh):
aco = AttachedCollisionObject.from_attached_collision_mesh(
attached_collision_mesh)
aco.object.operation = CollisionObject.ADD
robot_state = RobotState(attached_collision_objects=[aco], is_diff=True)
scene = PlanningScene(robot_state=robot_state, is_diff=True)
request = scene.to_request(self.ros_client.ros_distro)
self.APPLY_PLANNING_SCENE(self.ros_client, request, callback, errback)
def remove_attached_collision_mesh_async(self, callback, errback, id):
aco = AttachedCollisionObject()
aco.object.id = id
aco.object.operation = CollisionObject.REMOVE
robot_state = RobotState(attached_collision_objects=[aco],
is_diff=True)
scene = PlanningScene(robot_state=robot_state, is_diff=True)
request = dict(scene=scene)
self.APPLY_PLANNING_SCENE(self.ros_client, request, callback, errback)
def plan_cartesian_motion_async(self, callback, errback,
frames_WCF, start_configuration=None, group=None, options=None):
"""Asynchronous handler of MoveIt cartesian motion planner service."""
joint_names = options['joint_names']
joint_types = options['joint_types']
joint_type_by_name = dict(zip(joint_names, joint_types))
header = Header(frame_id=options['base_link'])
waypoints = [Pose.from_frame(frame) for frame in frames_WCF]
joint_state = JointState(header=header,
name=start_configuration.joint_names,
position=start_configuration.joint_values)
start_state = RobotState(joint_state, MultiDOFJointState(header=header), is_diff=True)
if options.get('attached_collision_meshes'):
for acm in options['attached_collision_meshes']:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
path_constraints = convert_constraints_to_rosmsg(options.get('path_constraints'), header)
request = dict(header=header,
start_state=start_state,
group_name=group,
link_name=options['link'],
waypoints=waypoints,
max_step=float(options.get('max_step', 0.01)),
jump_threshold=float(options.get('jump_threshold', 1.57)),
avoid_collisions=bool(options.get('avoid_collisions', True)),
path_constraints=path_constraints)
def convert_to_trajectory(response):
try:
trajectory = JointTrajectory()
trajectory.source_message = response
trajectory.fraction = response.fraction
trajectory.joint_names = response.solution.joint_trajectory.joint_names
joint_types = [joint_type_by_name[name] for name in trajectory.joint_names]
trajectory.points = convert_trajectory_points(
response.solution.joint_trajectory.points, joint_types)
start_state = response.start_state.joint_state
start_state_types = [joint_type_by_name[name] for name in start_state.name]
trajectory.start_configuration = Configuration(start_state.position, start_state_types, start_state.name)
trajectory.attached_collision_meshes = list(itertools.chain(*[
aco.to_attached_collision_meshes()
for aco in response.start_state.attached_collision_objects]))
callback(trajectory)
except Exception as e:
errback(e)
self.GET_CARTESIAN_PATH(self.ros_client, request, convert_to_trajectory, errback)
def inverse_kinematics_async(self,
callback,
errback,
frame_WCF,
start_configuration=None,
group=None,
options=None):
"""Asynchronous handler of MoveIt IK service."""
base_link = options['base_link']
header = Header(frame_id=base_link)
pose_stamped = PoseStamped(header, Pose.from_frame(frame_WCF))
joint_state = JointState(name=start_configuration.joint_names,
position=start_configuration.joint_values,
header=header)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header))
if options.get('attached_collision_meshes'):
for acm in options['attached_collision_meshes']:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
constraints = convert_constraints_to_rosmsg(options.get('constraints'),
header)
timeout_in_secs = options.get('timeout', 2)
timeout_duration = Duration(timeout_in_secs, 0).to_data()
ik_request = PositionIKRequest(group_name=group,
robot_state=start_state,
constraints=constraints,
pose_stamped=pose_stamped,
avoid_collisions=options.get(
'avoid_collisions', True),
attempts=options.get('attempts', 8),
timeout=timeout_duration)
# The field `attempts` was removed in Noetic (and higher)
# so it needs to be removed from the message otherwise it causes a serialization error
# https://github.com/ros-planning/moveit/pull/1288
if self.ros_client.ros_distro not in (RosDistro.KINETIC,
RosDistro.MELODIC):
del ik_request.attempts
def convert_to_positions(response):
callback((response.solution.joint_state.position,
response.solution.joint_state.name))
self.GET_POSITION_IK(self.ros_client, (ik_request, ),
convert_to_positions, errback)
def inverse_kinematics_async(self,
callback,
errback,
robot,
frame,
group,
start_configuration,
avoid_collisions=True,
constraints=None,
attempts=8,
attached_collision_meshes=None):
"""Asynchronous handler of MoveIt IK service."""
base_link = robot.model.root.name
header = Header(frame_id=base_link)
pose = Pose.from_frame(frame)
pose_stamped = PoseStamped(header, pose)
joint_state = JointState(name=start_configuration.joint_names,
position=start_configuration.values,
header=header)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header))
if attached_collision_meshes:
for acm in attached_collision_meshes:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
constraints = self._convert_constraints_to_rosmsg(constraints, header)
ik_request = PositionIKRequest(group_name=group,
robot_state=start_state,
constraints=constraints,
pose_stamped=pose_stamped,
avoid_collisions=avoid_collisions,
attempts=attempts)
def convert_to_positions(response):
callback((response.solution.joint_state.position,
response.solution.joint_state.name))
self.GET_POSITION_IK(self, (ik_request, ), convert_to_positions,
errback)
def inverse_kinematics_async(self,
callback,
errback,
frame_WCF,
start_configuration=None,
group=None,
options=None):
"""Asynchronous handler of MoveIt IK service."""
base_link = options['base_link']
header = Header(frame_id=base_link)
pose_stamped = PoseStamped(header, Pose.from_frame(frame_WCF))
joint_state = JointState(name=start_configuration.joint_names,
position=start_configuration.values,
header=header)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header))
if options.get('attached_collision_meshes'):
for acm in options['attached_collision_meshes']:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
constraints = convert_constraints_to_rosmsg(options.get('constraints'),
header)
ik_request = PositionIKRequest(group_name=group,
robot_state=start_state,
constraints=constraints,
pose_stamped=pose_stamped,
avoid_collisions=options.get(
'avoid_collisions', True),
attempts=options.get('attempts', 8))
def convert_to_positions(response):
callback((response.solution.joint_state.position,
response.solution.joint_state.name))
self.GET_POSITION_IK(self.ros_client, (ik_request, ),
convert_to_positions, errback)
def build_attached_collision_mesh(self,
ee_link,
id_name,
compas_mesh,
operation,
touch_links=None):
"""
"""
co = self.build_collision_object(ee_link, id_name, compas_mesh,
operation)
aco = AttachedCollisionObject()
aco.link_name = ee_link
# The set of links that the attached objects are allowed to touch by default.
if not touch_links:
aco.touch_links = [ee_link]
else:
aco.touch_links = touch_links
aco.object = co
return aco
def plan_motion_async(self,
callback,
errback,
goal_constraints,
start_configuration=None,
group=None,
options=None):
"""Asynchronous handler of MoveIt motion planner service."""
# http://docs.ros.org/jade/api/moveit_core/html/utils_8cpp_source.html
# TODO: if list of frames (goals) => receive multiple solutions?
joints = options['joints']
header = Header(frame_id=options['base_link'])
joint_state = JointState(header=header,
name=start_configuration.joint_names,
position=start_configuration.joint_values)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header),
is_diff=True)
if options.get('attached_collision_meshes'):
for acm in options['attached_collision_meshes']:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
# convert constraints
goal_constraints = [
convert_constraints_to_rosmsg(goal_constraints, header)
]
path_constraints = convert_constraints_to_rosmsg(
options.get('path_constraints'), header)
trajectory_constraints = options.get('trajectory_constraints')
if trajectory_constraints is not None:
trajectory_constraints = TrajectoryConstraints(
constraints=convert_constraints_to_rosmsg(
options['trajectory_constraints'], header))
request = dict(
start_state=start_state,
goal_constraints=goal_constraints,
path_constraints=path_constraints,
trajectory_constraints=trajectory_constraints,
planner_id=options.get('planner_id', 'RRTConnect'),
group_name=group,
num_planning_attempts=options.get('num_planning_attempts', 1),
allowed_planning_time=options.get('allowed_planning_time', 2.),
max_velocity_scaling_factor=options.get(
'max_velocity_scaling_factor', 1.),
max_acceleration_scaling_factor=options.get(
'max_acceleration_scaling_factor', 1.))
# workspace_parameters=options.get('workspace_parameters')
def response_handler(response):
try:
trajectory = convert_trajectory(joints, response.trajectory,
response.trajectory_start, 1.,
response.planning_time,
response)
callback(trajectory)
except Exception as e:
errback(e)
self.GET_MOTION_PLAN(self.ros_client, request, response_handler,
errback)
def plan_motion_async(self,
callback,
errback,
goal_constraints,
base_link,
ee_link,
group,
joint_names,
joint_types,
start_configuration,
path_constraints=None,
trajectory_constraints=None,
planner_id='',
num_planning_attempts=8,
allowed_planning_time=2.,
max_velocity_scaling_factor=1.,
max_acceleration_scaling_factor=1.,
attached_collision_meshes=None,
workspace_parameters=None):
"""Asynchronous handler of MoveIt motion planner service."""
# http://docs.ros.org/jade/api/moveit_core/html/utils_8cpp_source.html
# TODO: if list of frames (goals) => receive multiple solutions?
header = Header(frame_id=base_link)
joint_state = JointState(header=header,
name=joint_names,
position=start_configuration.values)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header))
if attached_collision_meshes:
for acm in attached_collision_meshes:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
# goal constraints
constraints = Constraints()
for c in goal_constraints:
if c.type == c.JOINT:
constraints.joint_constraints.append(
JointConstraint.from_joint_constraint(c))
elif c.type == c.POSITION:
constraints.position_constraints.append(
PositionConstraint.from_position_constraint(header, c))
elif c.type == c.ORIENTATION:
constraints.orientation_constraints.append(
OrientationConstraint.from_orientation_constraint(
header, c))
else:
raise NotImplementedError
goal_constraints = [constraints]
# path constraints
if path_constraints:
constraints = Constraints()
for c in path_constraints:
if c.type == c.JOINT:
constraints.joint_constraints.append(
JointConstraint.from_joint_constraint(c))
elif c.type == c.POSITION:
constraints.position_constraints.append(
PositionConstraint.from_position_constraint(header, c))
elif c.type == c.ORIENTATION:
constraints.orientation_constraints.append(
OrientationConstraint.from_orientation_constraint(
header, c))
else:
raise NotImplementedError
path_constraints = constraints
request = dict(
start_state=start_state,
goal_constraints=goal_constraints,
path_constraints=path_constraints,
trajectory_constraints=trajectory_constraints,
planner_id=planner_id,
group_name=group,
num_planning_attempts=num_planning_attempts,
allowed_planning_time=allowed_planning_time,
max_velocity_scaling_factor=max_velocity_scaling_factor,
max_acceleration_scaling_factor=max_velocity_scaling_factor)
# workspace_parameters=workspace_parameters
def convert_to_trajectory(response):
trajectory = JointTrajectory()
trajectory.source_message = response
trajectory.fraction = 1.
trajectory.points = convert_trajectory_points(
response.trajectory.joint_trajectory.points, joint_types)
trajectory.start_configuration = Configuration(
response.trajectory_start.joint_state.position,
start_configuration.types)
trajectory.planning_time = response.planning_time
callback(trajectory)
self.GET_MOTION_PLAN(self, request, convert_to_trajectory, errback)
def plan_motion_async(self,
callback,
errback,
goal_constraints,
start_configuration=None,
group=None,
options=None):
"""Asynchronous handler of MoveIt motion planner service."""
# http://docs.ros.org/jade/api/moveit_core/html/utils_8cpp_source.html
# TODO: if list of frames (goals) => receive multiple solutions?
joint_names = options['joint_names']
joint_types = options['joint_types']
joint_type_by_name = dict(
zip(joint_names, joint_types)
) # !!! should this go somewhere else? does it already exist somewhere else?
header = Header(frame_id=options['base_link'])
joint_state = JointState(header=header,
name=start_configuration.joint_names,
position=start_configuration.joint_values)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header),
is_diff=True)
if options.get('attached_collision_meshes'):
for acm in options['attached_collision_meshes']:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
# convert constraints
goal_constraints = [
convert_constraints_to_rosmsg(goal_constraints, header)
]
path_constraints = convert_constraints_to_rosmsg(
options.get('path_constraints'), header)
trajectory_constraints = options.get('trajectory_constraints')
if trajectory_constraints is not None:
trajectory_constraints = TrajectoryConstraints(
constraints=convert_constraints_to_rosmsg(
options['trajectory_constraints'], header))
request = dict(
start_state=start_state,
goal_constraints=goal_constraints,
path_constraints=path_constraints,
trajectory_constraints=trajectory_constraints,
planner_id=options.get('planner_id', 'RRTConnectkConfigDefault'),
group_name=group,
num_planning_attempts=options.get('num_planning_attempts', 1),
allowed_planning_time=options.get('allowed_planning_time', 2.),
max_velocity_scaling_factor=options.get(
'max_velocity_scaling_factor', 1.),
max_acceleration_scaling_factor=options.get(
'max_acceleration_scaling_factor', 1.))
# workspace_parameters=options.get('workspace_parameters')
def convert_to_trajectory(response):
trajectory = JointTrajectory()
trajectory.source_message = response
trajectory.fraction = 1.
trajectory.joint_names = response.trajectory.joint_trajectory.joint_names
trajectory.planning_time = response.planning_time
joint_types = [
joint_type_by_name[name] for name in trajectory.joint_names
]
trajectory.points = convert_trajectory_points(
response.trajectory.joint_trajectory.points, joint_types)
start_state = response.trajectory_start.joint_state
start_state_types = [
joint_type_by_name[name] for name in start_state.name
]
trajectory.start_configuration = Configuration(
start_state.position, start_state_types, start_state.name)
trajectory.attached_collision_meshes = list(
itertools.chain(*[
aco.to_attached_collision_meshes() for aco in
response.trajectory_start.attached_collision_objects
]))
callback(trajectory)
self.GET_MOTION_PLAN(self.ros_client, request, convert_to_trajectory,
errback)
def plan_motion_async(self,
callback,
errback,
robot,
goal_constraints,
start_configuration,
group,
path_constraints=None,
trajectory_constraints=None,
planner_id='',
num_planning_attempts=8,
allowed_planning_time=2.,
max_velocity_scaling_factor=1.,
max_acceleration_scaling_factor=1.,
attached_collision_meshes=None,
workspace_parameters=None):
"""Asynchronous handler of MoveIt motion planner service."""
# http://docs.ros.org/jade/api/moveit_core/html/utils_8cpp_source.html
# TODO: if list of frames (goals) => receive multiple solutions?
base_link = robot.model.root.name # use world coords
header = Header(frame_id=base_link)
joint_state = JointState(header=header,
name=start_configuration.joint_names,
position=start_configuration.values)
start_state = RobotState(joint_state,
MultiDOFJointState(header=header))
if attached_collision_meshes:
for acm in attached_collision_meshes:
aco = AttachedCollisionObject.from_attached_collision_mesh(acm)
start_state.attached_collision_objects.append(aco)
# convert constraints
goal_constraints = [
self._convert_constraints_to_rosmsg(goal_constraints, header)
]
path_constraints = self._convert_constraints_to_rosmsg(
path_constraints, header)
if trajectory_constraints is not None:
trajectory_constraints = TrajectoryConstraints(
constraints=self._convert_constraints_to_rosmsg(
path_constraints, header))
request = dict(
start_state=start_state,
goal_constraints=goal_constraints,
path_constraints=path_constraints,
trajectory_constraints=trajectory_constraints,
planner_id=planner_id,
group_name=group,
num_planning_attempts=num_planning_attempts,
allowed_planning_time=allowed_planning_time,
max_velocity_scaling_factor=max_velocity_scaling_factor,
max_acceleration_scaling_factor=max_velocity_scaling_factor)
# workspace_parameters=workspace_parameters
def convert_to_trajectory(response):
trajectory = JointTrajectory()
trajectory.source_message = response
trajectory.fraction = 1.
trajectory.joint_names = response.trajectory.joint_trajectory.joint_names
trajectory.planning_time = response.planning_time
joint_types = robot.get_joint_types_by_names(
trajectory.joint_names)
trajectory.points = convert_trajectory_points(
response.trajectory.joint_trajectory.points, joint_types)
start_state = response.trajectory_start.joint_state
start_state_types = robot.get_joint_types_by_names(
start_state.name)
trajectory.start_configuration = Configuration(
start_state.position, start_state_types, start_state.name)
callback(trajectory)
self.GET_MOTION_PLAN(self, request, convert_to_trajectory, errback)
def add_attached_collision_mesh(self, attached_collision_mesh):
"""Add a collision mesh attached to the robot."""
aco = AttachedCollisionObject.from_attached_collision_mesh(
attached_collision_mesh)
self._attached_collision_object(aco, operation=CollisionObject.ADD)
def remove_attached_collision_mesh(self, id):
"""Add an attached collision mesh from the robot."""
aco = AttachedCollisionObject()
aco.object.id = id
return self._attached_collision_object(
aco, operation=CollisionObject.REMOVE)
