def update_state(self):
# TODO: apply this directly from observations
# No use applying this to base confs
self.base_conf = FConf(self.world.robot,
self.world.base_joints,
init=True)
arm_conf = FConf(self.world.robot, self.world.arm_joints, init=True)
if (self.arm_conf is None) or not are_confs_close(
arm_conf, self.arm_conf, tol=ARM_TOLERANCE):
self.arm_conf = arm_conf
else:
print('At anticipated arm conf')
gripper_conf = FConf(self.world.robot,
self.world.gripper_joints,
init=True)
if (self.gripper_conf is None) or not are_confs_close(
gripper_conf, self.gripper_conf, tol=GRIPPER_TOLERANCE):
self.gripper_conf = gripper_conf
else:
print('At anticipated gripper conf')
# TODO: do I still need to test if the current values are equal to the last ones?
for joint in self.world.kitchen_joints:
name = get_joint_name(self.world.kitchen, joint)
position = get_joint_position(self.world.kitchen, joint)
self.update_door_conf(name, position)
self.update_door_conf(name, position)
#wait_for_user()
return self.check_consistent()
def _update_initial(self):
# TODO: store initial poses as well?
self.initial_saver = WorldSaver()
self.goal_bq = FConf(self.robot, self.base_joints)
self.goal_aq = FConf(self.robot, self.arm_joints)
if are_confs_close(self.goal_aq, self.carry_conf):
self.goal_aq = self.carry_conf
self.goal_gq = FConf(self.robot, self.gripper_joints)
self.initial_confs = [self.goal_bq, self.goal_aq, self.goal_gq]
set_all_static()
def update_door_conf(self, name, position):
joint = joint_from_name(self.world.kitchen, name)
conf = FConf(self.world.kitchen, [joint], [position], init=True)
if (name not in self.door_confs) or not are_confs_close(
conf, self.door_confs[name], tol=1e-3):
# TODO: different threshold for drawers and doors
self.door_confs[name] = conf
else:
print('At anticipated conf for door {}'.format(name))
return self.door_confs[name]
def get_reachability_test(world, **kwargs):
base_motion_fn = get_base_motion_fn(world,
restarts=2,
iterations=50,
smooth=0,
**kwargs)
bq0 = FConf(world.robot, world.base_joints)
# TODO: can check for arm motions as well
def test(bq):
aq = world.carry_conf
outputs = base_motion_fn(aq, bq0, bq, fluents=[])
return outputs is not None
return test
def inverse_reachability(world,
base_generator,
obstacles=set(),
max_attempts=25,
**kwargs):
min_distance = 0.01 #if world.is_real() else 0.0
min_nearby_distance = 0.1 # if world.is_real() else 0.0
lower_limits, upper_limits = get_custom_limits(world.robot,
world.base_joints,
world.custom_limits)
while True:
attempt = 0
for base_conf in islice(base_generator, max_attempts):
attempt += 1
if not all_between(lower_limits, base_conf, upper_limits):
continue
bq = FConf(world.robot, world.base_joints, base_conf)
#wait_for_user()
if not test_base_conf(
world, bq, obstacles, min_distance=min_distance):
continue
if world.is_real():
# TODO: could also rotate in place
# TODO: restrict orientation to face the counter
nearby_values = translate_linearly(world,
distance=-REVERSE_DISTANCE)
bq.nearby_bq = FConf(world.robot, world.base_joints,
nearby_values)
if not test_base_conf(world,
bq.nearby_bq,
obstacles,
min_distance=min_nearby_distance):
continue
#if PRINT_FAILURES: print('Success after {} IR attempts:'.format(attempt))
bq.assign()
#wait_for_user()
yield bq
break
else:
if PRINT_FAILURES:
print('Failed after {} IR attempts:'.format(attempt))
if attempt < max_attempts - 1:
return
yield None
def __init__(self,
robot_name=FRANKA_CARTER,
use_gui=True,
full_kitchen=False):
self.task = None
self.interface = None
self.client = connect(use_gui=use_gui)
set_real_time(False)
#set_caching(False) # Seems to make things worse
disable_gravity()
add_data_path()
set_camera_pose(camera_point=[2, -1.5, 1])
if DEBUG:
draw_pose(Pose(), length=3)
#self.kitchen_yaml = load_yaml(KITCHEN_YAML)
with HideOutput(enable=True):
self.kitchen = load_pybullet(KITCHEN_PATH,
fixed_base=True,
cylinder=True)
self.floor = load_pybullet('plane.urdf', fixed_base=True)
z = stable_z(self.kitchen, self.floor) - get_point(self.floor)[2]
point = np.array(get_point(self.kitchen)) - np.array([0, 0, z])
set_point(self.floor, point)
self.robot_name = robot_name
if self.robot_name == FRANKA_CARTER:
urdf_path, yaml_path = FRANKA_CARTER_PATH, None
#urdf_path, yaml_path = FRANKA_CARTER_PATH, FRANKA_YAML
#elif self.robot_name == EVE:
# urdf_path, yaml_path = EVE_PATH, None
else:
raise ValueError(self.robot_name)
self.robot_yaml = yaml_path if yaml_path is None else load_yaml(
yaml_path)
with HideOutput(enable=True):
self.robot = load_pybullet(urdf_path)
#dump_body(self.robot)
#chassis_pose = get_link_pose(self.robot, link_from_name(self.robot, 'chassis_link'))
#wheel_pose = get_link_pose(self.robot, link_from_name(self.robot, 'left_wheel_link'))
#wait_for_user()
set_point(self.robot, Point(z=stable_z(self.robot, self.floor)))
self.set_initial_conf()
self.gripper = create_gripper(self.robot)
self.environment_bodies = {}
if full_kitchen:
self._initialize_environment()
self._initialize_ik(urdf_path)
self.initial_saver = WorldSaver()
self.body_from_name = {}
# self.path_from_name = {}
self.names_from_type = {}
self.custom_limits = {}
self.base_limits_handles = []
self.cameras = {}
self.disabled_collisions = set()
if self.robot_name == FRANKA_CARTER:
self.disabled_collisions.update(
tuple(link_from_name(self.robot, link) for link in pair)
for pair in DISABLED_FRANKA_COLLISIONS)
self.carry_conf = FConf(self.robot, self.arm_joints, self.default_conf)
#self.calibrate_conf = Conf(self.robot, self.arm_joints, load_calibrate_conf(side='left'))
self.calibrate_conf = FConf(
self.robot, self.arm_joints,
self.default_conf) # Must differ from carry_conf
self.special_confs = [self.carry_conf] #, self.calibrate_conf]
self.open_gq = FConf(self.robot, self.gripper_joints,
get_max_limits(self.robot, self.gripper_joints))
self.closed_gq = FConf(self.robot, self.gripper_joints,
get_min_limits(self.robot, self.gripper_joints))
self.gripper_confs = [self.open_gq, self.closed_gq]
self.open_kitchen_confs = {
joint: FConf(self.kitchen, [joint], [self.open_conf(joint)])
for joint in self.kitchen_joints
}
self.closed_kitchen_confs = {
joint: FConf(self.kitchen, [joint], [self.closed_conf(joint)])
for joint in self.kitchen_joints
}
self._update_custom_limits()
self._update_initial()
class World(object):
def __init__(self,
robot_name=FRANKA_CARTER,
use_gui=True,
full_kitchen=False):
self.task = None
self.interface = None
self.client = connect(use_gui=use_gui)
set_real_time(False)
#set_caching(False) # Seems to make things worse
disable_gravity()
add_data_path()
set_camera_pose(camera_point=[2, -1.5, 1])
if DEBUG:
draw_pose(Pose(), length=3)
#self.kitchen_yaml = load_yaml(KITCHEN_YAML)
with HideOutput(enable=True):
self.kitchen = load_pybullet(KITCHEN_PATH,
fixed_base=True,
cylinder=True)
self.floor = load_pybullet('plane.urdf', fixed_base=True)
z = stable_z(self.kitchen, self.floor) - get_point(self.floor)[2]
point = np.array(get_point(self.kitchen)) - np.array([0, 0, z])
set_point(self.floor, point)
self.robot_name = robot_name
if self.robot_name == FRANKA_CARTER:
urdf_path, yaml_path = FRANKA_CARTER_PATH, None
#urdf_path, yaml_path = FRANKA_CARTER_PATH, FRANKA_YAML
#elif self.robot_name == EVE:
# urdf_path, yaml_path = EVE_PATH, None
else:
raise ValueError(self.robot_name)
self.robot_yaml = yaml_path if yaml_path is None else load_yaml(
yaml_path)
with HideOutput(enable=True):
self.robot = load_pybullet(urdf_path)
#dump_body(self.robot)
#chassis_pose = get_link_pose(self.robot, link_from_name(self.robot, 'chassis_link'))
#wheel_pose = get_link_pose(self.robot, link_from_name(self.robot, 'left_wheel_link'))
#wait_for_user()
set_point(self.robot, Point(z=stable_z(self.robot, self.floor)))
self.set_initial_conf()
self.gripper = create_gripper(self.robot)
self.environment_bodies = {}
if full_kitchen:
self._initialize_environment()
self._initialize_ik(urdf_path)
self.initial_saver = WorldSaver()
self.body_from_name = {}
# self.path_from_name = {}
self.names_from_type = {}
self.custom_limits = {}
self.base_limits_handles = []
self.cameras = {}
self.disabled_collisions = set()
if self.robot_name == FRANKA_CARTER:
self.disabled_collisions.update(
tuple(link_from_name(self.robot, link) for link in pair)
for pair in DISABLED_FRANKA_COLLISIONS)
self.carry_conf = FConf(self.robot, self.arm_joints, self.default_conf)
#self.calibrate_conf = Conf(self.robot, self.arm_joints, load_calibrate_conf(side='left'))
self.calibrate_conf = FConf(
self.robot, self.arm_joints,
self.default_conf) # Must differ from carry_conf
self.special_confs = [self.carry_conf] #, self.calibrate_conf]
self.open_gq = FConf(self.robot, self.gripper_joints,
get_max_limits(self.robot, self.gripper_joints))
self.closed_gq = FConf(self.robot, self.gripper_joints,
get_min_limits(self.robot, self.gripper_joints))
self.gripper_confs = [self.open_gq, self.closed_gq]
self.open_kitchen_confs = {
joint: FConf(self.kitchen, [joint], [self.open_conf(joint)])
for joint in self.kitchen_joints
}
self.closed_kitchen_confs = {
joint: FConf(self.kitchen, [joint], [self.closed_conf(joint)])
for joint in self.kitchen_joints
}
self._update_custom_limits()
self._update_initial()
def _initialize_environment(self):
# wall to fridge: 4cm
# fridge to goal: 1.5cm
# hitman to range: 3.5cm
# range to indigo: 3.5cm
self.environment_poses = read_json(POSES_PATH)
root_from_world = get_link_pose(self.kitchen, self.world_link)
for name, world_from_part in self.environment_poses.items():
if name in ['range']:
continue
visual_path = os.path.join(KITCHEN_LEFT_PATH,
'{}.obj'.format(name))
collision_path = os.path.join(KITCHEN_LEFT_PATH,
'{}_collision.obj'.format(name))
mesh_path = None
for path in [collision_path, visual_path]:
if os.path.exists(path):
mesh_path = path
break
if mesh_path is None:
continue
body = load_pybullet(mesh_path, fixed_base=True)
root_from_part = multiply(root_from_world, world_from_part)
if name in ['axe', 'dishwasher', 'echo', 'fox', 'golf']:
(pos, quat) = root_from_part
# TODO: still not totally aligned
pos = np.array(pos) + np.array([0, -0.035, 0]) # , -0.005])
root_from_part = (pos, quat)
self.environment_bodies[name] = body
set_pose(body, root_from_part)
# TODO: release bounding box or convex hull
# TODO: static object nonconvex collisions
if TABLE_NAME in self.environment_bodies:
body = self.environment_bodies[TABLE_NAME]
_, (w, l, _) = approximate_as_prism(body)
_, _, z = get_point(body)
new_pose = Pose(Point(TABLE_X + l / 2, -TABLE_Y, z),
Euler(yaw=np.pi / 2))
set_pose(body, new_pose)
def _initialize_ik(self, urdf_path):
if not USE_TRACK_IK:
self.ik_solver = None
return
from trac_ik_python.trac_ik import IK # killall -9 rosmaster
base_link = get_link_name(
self.robot, parent_link_from_joint(self.robot, self.arm_joints[0]))
tip_link = get_link_name(self.robot,
child_link_from_joint(self.arm_joints[-1]))
# limit effort and velocities are required
# solve_type: Speed, Distance, Manipulation1, Manipulation2
# TODO: fast solver and slow solver
self.ik_solver = IK(base_link=str(base_link),
tip_link=str(tip_link),
timeout=0.01,
epsilon=1e-5,
solve_type="Speed",
urdf_string=read(urdf_path))
if not CONSERVITIVE_LIMITS:
return
lower, upper = self.ik_solver.get_joint_limits()
buffer = JOINT_LIMITS_BUFFER * np.ones(len(self.ik_solver.joint_names))
lower, upper = lower + buffer, upper - buffer
lower[6] = -MAX_FRANKA_JOINT7
upper[6] = +MAX_FRANKA_JOINT7
self.ik_solver.set_joint_limits(lower, upper)
def _update_initial(self):
# TODO: store initial poses as well?
self.initial_saver = WorldSaver()
self.goal_bq = FConf(self.robot, self.base_joints)
self.goal_aq = FConf(self.robot, self.arm_joints)
if are_confs_close(self.goal_aq, self.carry_conf):
self.goal_aq = self.carry_conf
self.goal_gq = FConf(self.robot, self.gripper_joints)
self.initial_confs = [self.goal_bq, self.goal_aq, self.goal_gq]
set_all_static()
def is_real(self):
return (self.task is not None) and self.task.real
@property
def constants(self):
return self.special_confs + self.gripper_confs + self.initial_confs
#########################
@property
def base_joints(self):
return joints_from_names(self.robot, BASE_JOINTS)
@property
def arm_joints(self):
#if self.robot_name == EVE:
# return get_eve_arm_joints(self.robot, arm=DEFAULT_ARM)
joint_names = ['panda_joint{}'.format(1 + i) for i in range(7)]
#joint_names = self.robot_yaml['cspace']
return joints_from_names(self.robot, joint_names)
@property
def gripper_joints(self):
#if self.robot_yaml is None:
# return []
joint_names = ['panda_finger_joint{}'.format(1 + i) for i in range(2)]
#joint_names = [joint_from_name(self.robot, rule['name'])
# for rule in self.robot_yaml['cspace_to_urdf_rules']]
return joints_from_names(self.robot, joint_names)
@property
def kitchen_joints(self):
joint_names = get_joint_names(self.kitchen,
get_movable_joints(self.kitchen))
#joint_names = self.kitchen_yaml['cspace']
#return joints_from_names(self.kitchen, joint_names)
return joints_from_names(self.kitchen,
filter(ALL_JOINTS.__contains__, joint_names))
@property
def base_link(self):
return child_link_from_joint(self.base_joints[-1])
@property
def franka_link(self):
return parent_link_from_joint(self.robot, self.arm_joints[0])
@property
def gripper_link(self):
return parent_link_from_joint(self.robot, self.gripper_joints[0])
@property
def tool_link(self):
return link_from_name(self.robot, get_tool_link(self.robot))
@property
def world_link(self): # for kitchen
return BASE_LINK
@property
def door_links(self):
door_links = set()
for joint in self.kitchen_joints:
door_links.update(get_link_subtree(self.kitchen, joint))
return door_links
@property
def static_obstacles(self):
# link=None is fine
# TODO: decompose obstacles
#return [(self.kitchen, frozenset(get_links(self.kitchen)) - self.door_links)]
return {(self.kitchen, frozenset([link])) for link in
set(get_links(self.kitchen)) - self.door_links} | \
{(body, None) for body in self.environment_bodies.values()}
@property
def movable(self): # movable base
return set(self.body_from_name) # frozenset?
@property
def fixed(self): # fixed base
return set(self.environment_bodies.values()) | {self.kitchen}
@property
def all_bodies(self):
return self.movable | self.fixed | {self.robot}
@property
def default_conf(self):
# if self.robot_name == EVE:
# # Eve starts outside of joint limits
# # Eve starts outside of joint limits
# conf = [np.average(get_joint_limits(self.robot, joint)) for joint in self.arm_joints]
# #conf = np.zeros(len(self.arm_joints))
# #conf[3] -= np.pi / 2
# return conf
return DEFAULT_ARM_CONF
#conf = np.array(self.robot_yaml['default_q'])
##conf[1] += np.pi / 4
##conf[3] -= np.pi / 4
#return conf
#########################
# TODO: could perform base motion planning without free joints
def get_base_conf(self):
return get_joint_positions(self.robot, self.base_joints)
def set_base_conf(self, conf):
set_joint_positions(self.robot, self.base_joints, conf)
def get_base_aabb(self):
franka_links = get_link_subtree(self.robot, self.franka_link)
base_links = get_link_subtree(self.robot, self.base_link)
return aabb_union(
get_aabb(self.robot, link)
for link in set(base_links) - set(franka_links))
def get_world_aabb(self):
return aabb_union(get_aabb(body)
for body in self.fixed) # self.all_bodies
def _update_custom_limits(self, min_extent=0.0):
#robot_extent = get_aabb_extent(get_aabb(self.robot))
# Scaling by 0.5 to prevent getting caught in corners
#min_extent = 0.5 * min(robot_extent[:2]) * np.ones(2) / 2
world_aabb = self.get_world_aabb()
full_lower, full_upper = world_aabb
base_limits = (full_lower[:2] - min_extent,
full_upper[:2] + min_extent)
base_limits[1][0] = COMPUTER_X - min_extent # TODO: robot radius
base_limits[0][1] += 0.1
base_limits[1][1] -= 0.1
for handle in self.base_limits_handles:
remove_debug(handle)
self.base_limits_handles = []
#self.base_limits_handles.extend(draw_aabb(world_aabb))
z = get_point(self.floor)[2] + 1e-2
if DEBUG:
self.base_limits_handles.extend(draw_base_limits(base_limits, z=z))
self.custom_limits = custom_limits_from_base_limits(
self.robot, base_limits)
return self.custom_limits
def solve_trac_ik(self, world_from_tool, nearby_tolerance=INF):
assert self.ik_solver is not None
init_lower, init_upper = self.ik_solver.get_joint_limits()
base_link = link_from_name(self.robot, self.ik_solver.base_link)
world_from_base = get_link_pose(self.robot, base_link)
tip_link = link_from_name(self.robot, self.ik_solver.tip_link)
tool_from_tip = multiply(
invert(get_link_pose(self.robot, self.tool_link)),
get_link_pose(self.robot, tip_link))
world_from_tip = multiply(world_from_tool, tool_from_tip)
base_from_tip = multiply(invert(world_from_base), world_from_tip)
joints = joints_from_names(
self.robot,
self.ik_solver.joint_names) # self.ik_solver.link_names
seed_state = get_joint_positions(self.robot, joints)
# seed_state = [0.0] * self.ik_solver.number_of_joints
lower, upper = init_lower, init_upper
if nearby_tolerance < INF:
tolerance = nearby_tolerance * np.ones(len(joints))
lower = np.maximum(lower, seed_state - tolerance)
upper = np.minimum(upper, seed_state + tolerance)
self.ik_solver.set_joint_limits(lower, upper)
(x, y, z), (rx, ry, rz, rw) = base_from_tip
# TODO: can also adjust tolerances
conf = self.ik_solver.get_ik(seed_state, x, y, z, rx, ry, rz, rw)
self.ik_solver.set_joint_limits(init_lower, init_upper)
if conf is None:
return conf
# if nearby_tolerance < INF:
# print(lower.round(3))
# print(upper.round(3))
# print(conf)
# print(get_difference(seed_state, conf).round(3))
set_joint_positions(self.robot, joints, conf)
return get_configuration(self.robot)
def solve_pybullet_ik(self, world_from_tool, nearby_tolerance):
start_time = time.time()
# Most of the time is spent creating the robot
# TODO: use the waypoint version that doesn't repeatedly create the robot
current_conf = get_joint_positions(self.robot, self.arm_joints)
full_conf = sub_inverse_kinematics(
self.robot,
self.arm_joints[0],
self.tool_link,
world_from_tool,
custom_limits=self.custom_limits
) # , max_iterations=1) # , **kwargs)
conf = get_joint_positions(self.robot, self.arm_joints)
max_distance = get_distance(current_conf, conf, norm=INF)
if nearby_tolerance < max_distance:
return None
print('Nearby) time: {:.3f} | distance: {:.5f}'.format(
elapsed_time(start_time), max_distance))
return full_conf
def solve_inverse_kinematics(self,
world_from_tool,
nearby_tolerance=INF,
**kwargs):
if self.ik_solver is not None:
return self.solve_trac_ik(world_from_tool, **kwargs)
#if nearby_tolerance != INF:
# return self.solve_pybullet_ik(world_from_tool, nearby_tolerance=nearby_tolerance)
current_conf = get_joint_positions(self.robot, self.arm_joints)
start_time = time.time()
if nearby_tolerance == INF:
generator = ikfast_inverse_kinematics(self.robot,
PANDA_INFO,
self.tool_link,
world_from_tool,
max_attempts=10,
use_halton=True)
else:
generator = closest_inverse_kinematics(
self.robot,
PANDA_INFO,
self.tool_link,
world_from_tool,
max_time=0.01,
max_distance=nearby_tolerance,
use_halton=True)
conf = next(generator, None)
#conf = sample_tool_ik(self.robot, world_from_tool, max_attempts=100)
if conf is None:
return conf
max_distance = get_distance(current_conf, conf, norm=INF)
#print('Time: {:.3f} | distance: {:.5f} | max: {:.5f}'.format(
# elapsed_time(start_time), max_distance, nearby_tolerance))
set_joint_positions(self.robot, self.arm_joints, conf)
return get_configuration(self.robot)
#########################
def set_initial_conf(self):
set_joint_positions(self.robot, self.base_joints, [2.0, 0, np.pi])
#for rule in self.robot_yaml['cspace_to_urdf_rules']: # gripper: max is open
# joint = joint_from_name(self.robot, rule['name'])
# set_joint_position(self.robot, joint, rule['value'])
set_joint_positions(self.robot, self.arm_joints,
self.default_conf) # active_task_spaces
# if self.robot_name == EVE:
# for arm in ARMS:
# joints = get_eve_arm_joints(self.robot, arm)[2:4]
# set_joint_positions(self.robot, joints, -0.2*np.ones(len(joints)))
def set_gripper(self, value):
positions = value * np.ones(len(self.gripper_joints))
set_joint_positions(self.robot, self.gripper_joints, positions)
def close_gripper(self):
self.closed_gq.assign()
def open_gripper(self):
self.open_gq.assign()
#########################
def get_door_sign(self, joint):
return -1 if 'left' in get_joint_name(self.kitchen, joint) else +1
def closed_conf(self, joint):
lower, upper = get_joint_limits(self.kitchen, joint)
if 'drawer' in get_joint_name(self.kitchen, joint):
fraction = 0.9
return fraction * lower + (1 - fraction) * upper
if 'left' in get_joint_name(self.kitchen, joint):
return upper
return lower
def open_conf(self, joint):
joint_name = get_joint_name(self.kitchen, joint)
if 'left' in joint_name:
open_position = get_min_limit(self.kitchen, joint)
else:
open_position = get_max_limit(self.kitchen, joint)
#print(get_joint_name(self.kitchen, joint), get_min_limit(self.kitchen, joint), get_max_limit(self.kitchen, joint))
# drawers: [0.0, 0.4]
# left doors: [-1.57, 0.0]
# right doors: [0.0, 1.57]
if joint_name in CABINET_JOINTS:
# TODO: could make fraction of max
return CABINET_OPEN_ANGLE * open_position / abs(open_position)
if joint_name in DRAWER_JOINTS:
return DRAWER_OPEN_FRACTION * open_position
return open_position
def close_door(self, joint):
set_joint_position(self.kitchen, joint, self.closed_conf(joint))
def open_door(self, joint):
set_joint_position(self.kitchen, joint, self.open_conf(joint))
#########################
def add_camera(self,
name,
pose,
camera_matrix,
max_depth=KINECT_DEPTH,
display=False):
color = apply_alpha(RED, 0.1 if DEBUG else 0)
cone = get_viewcone(depth=max_depth,
camera_matrix=camera_matrix,
color=color,
mass=0,
collision=False)
set_pose(cone, pose)
if display:
kinect = load_pybullet(KINECT_URDF, fixed_base=True)
set_pose(kinect, pose)
set_color(kinect, BLACK)
self.add(name, kinect)
self.cameras[name] = Camera(cone, camera_matrix, max_depth)
if DEBUG:
draw_pose(pose)
step_simulation()
return name
def get_supporting(self, obj_name):
# is_placed_on_aabb | is_center_on_aabb
# Only want to generate stable placements, but can operate on initially unstable ones
# TODO: could filter orientation as well
body = self.get_body(obj_name)
supporting = {
surface
for surface in ALL_SURFACES
if is_center_on_aabb(body,
compute_surface_aabb(self, surface),
above_epsilon=5e-2,
below_epsilon=5e-2)
}
if ('range' in supporting) and (len(supporting) == 2):
# TODO: small hack for now
supporting -= {'range'}
if len(supporting) != 1:
print('{} is not supported by a single surface ({})!'.format(
obj_name, supporting))
return None
[surface_name] = supporting
return surface_name
def fix_pose(self, name, pose=None, fraction=0.5):
body = self.get_body(name)
if pose is None:
pose = get_pose(body)
else:
set_pose(body, pose)
# TODO: can also drop in simulation
x, y, z = point_from_pose(pose)
roll, pitch, yaw = euler_from_quat(quat_from_pose(pose))
quat = quat_from_euler(Euler(yaw=yaw))
set_quat(body, quat)
surface_name = self.get_supporting(name)
if surface_name is None:
return None, None
if fraction == 0:
new_pose = (Point(x, y, z), quat)
return new_pose, surface_name
surface_aabb = compute_surface_aabb(self, surface_name)
new_z = (1 - fraction) * z + fraction * stable_z_on_aabb(
body, surface_aabb)
point = Point(x, y, new_z)
set_point(body, point)
print('{} error: roll={:.3f}, pitch={:.3f}, z-delta: {:.3f}'.format(
name, roll, pitch, new_z - z))
new_pose = (point, quat)
return new_pose, surface_name
# def fix_geometry(self):
# for name in self.movable:
# fixed_pose, _ = self.fix_pose(name)
# if fixed_pose is not None:
# set_pose(self.get_body(name), fixed_pose)
#########################
def add(self, name, body):
assert name not in self.body_from_name
if DEBUG:
add_body_name(body, name)
self.body_from_name[name] = body
return name
def add_body(self, name, **kwargs):
obj_type = type_from_name(name)
self.names_from_type.setdefault(obj_type, []).append(name)
path = get_obj_path(obj_type)
#self.path_from_name[name] = path
print('Loading', path)
body = load_pybullet(path, **kwargs)
assert body is not None
self.add(name, body)
def get_body(self, name):
return self.body_from_name[name]
# def get_body_path(self, name):
# return self.path_from_name[name]
# def get_body_type(self, name):
# filename, _ = os.path.splitext(os.path.basename(self.get_body_path(name)))
# return filename
def get_name(self, name):
inverse = {v: k for k, v in self.body_from_name.items()}
return inverse.get(name, None)
def remove_body(self, name):
body = self.get_body(name)
remove_body(body)
del self.body_from_name[name]
def reset(self):
#remove_all_debug()
for camera in self.cameras.values():
remove_body(camera.body)
#remove_body(camera.kinect)
self.cameras = {}
for name in list(self.body_from_name):
self.remove_body(name)
def destroy(self):
reset_simulation()
disconnect()
def gen(bowl_name, wp, cup_name, grasp, bq):
# https://github.mit.edu/Learning-and-Intelligent-Systems/ltamp_pr2/blob/d1e6024c5c13df7edeab3a271b745e656a794b02/plan_tools/samplers/pour.py
if bowl_name == cup_name:
return
#attachment = get_grasp_attachment(world, arm, grasp)
bowl_body = world.get_body(bowl_name)
#cup_body = world.get_body(cup_name)
obstacles = (world.static_obstacles
| {bowl_body}) if collisions else set()
cup_path_bowl = pour_path_from_parameter(world, bowl_name, cup_name)
while True:
for _ in range(max_attempts):
bowl_pose = wp.get_world_from_body()
rotate_bowl = Pose(euler=Euler(
yaw=random.uniform(-np.pi, np.pi)))
rotate_cup = Pose(euler=Euler(
yaw=random.uniform(-np.pi, np.pi)))
cup_path = [
multiply(bowl_pose, invert(rotate_bowl), cup_pose_bowl,
rotate_cup) for cup_pose_bowl in cup_path_bowl
]
#visualize_cartesian_path(cup_body, cup_path)
#if cartesian_path_collision(cup_body, cup_path, obstacles + [bowl_body]):
# continue
tool_path = [
multiply(p, invert(grasp.grasp_pose)) for p in cup_path
]
# TODO: extra collision test for visibility
# TODO: orientation constraint while moving
bq.assign()
grasp.set_gripper()
world.carry_conf.assign()
arm_path = plan_workspace(world,
tool_path,
obstacles,
randomize=True) # tilt to upright
if arm_path is None:
continue
assert MOVE_ARM
aq = FConf(world.robot, world.arm_joints, arm_path[-1])
robot_saver = BodySaver(world.robot)
obj_type = type_from_name(cup_name)
duration = 5.0 if obj_type in [MUSTARD] else 1.0
objects = [bowl_name, cup_name]
cmd = Sequence(State(world, savers=[robot_saver]),
commands=[
ApproachTrajectory(objects, world,
world.robot,
world.arm_joints,
arm_path[::-1]),
Wait(world, duration=duration),
ApproachTrajectory(objects, world,
world.robot,
world.arm_joints,
arm_path),
],
name='pour')
yield (
aq,
cmd,
)
break
else:
yield None
def plan_pull(world,
door_joint,
door_plan,
base_conf,
randomize=True,
collisions=True,
teleport=False,
**kwargs):
door_path, handle_path, handle_plan, tool_path = door_plan
handle_link, handle_grasp, handle_pregrasp = handle_plan
door_obstacles = get_descendant_obstacles(
world.kitchen, door_joint) # if collisions else set()
obstacles = (world.static_obstacles | door_obstacles
) # if collisions else set()
base_conf.assign()
world.open_gripper()
world.carry_conf.assign()
robot_saver = BodySaver(world.robot) # TODO: door_saver?
if not is_pull_safe(world, door_joint, door_plan):
return
arm_path = plan_workspace(world,
tool_path,
world.static_obstacles,
randomize=randomize,
teleport=collisions)
if arm_path is None:
return
approach_paths = []
for index in [0, -1]:
set_joint_positions(world.kitchen, [door_joint], door_path[index])
set_joint_positions(world.robot, world.arm_joints, arm_path[index])
tool_pose = multiply(handle_path[index], invert(handle_pregrasp))
approach_path = plan_approach(world,
tool_pose,
obstacles=obstacles,
teleport=teleport,
**kwargs)
if approach_path is None:
return
approach_paths.append(approach_path)
if MOVE_ARM:
aq1 = FConf(world.robot, world.arm_joints, approach_paths[0][0])
aq2 = FConf(world.robot, world.arm_joints, approach_paths[-1][0])
else:
aq1 = world.carry_conf
aq2 = aq1
set_joint_positions(world.kitchen, [door_joint], door_path[0])
set_joint_positions(world.robot, world.arm_joints, arm_path[0])
grasp_width = close_until_collision(world.robot,
world.gripper_joints,
bodies=[(world.kitchen, [handle_link])
])
gripper_motion_fn = get_gripper_motion_gen(world,
teleport=teleport,
collisions=collisions,
**kwargs)
gripper_conf = FConf(world.robot, world.gripper_joints,
[grasp_width] * len(world.gripper_joints))
finger_cmd, = gripper_motion_fn(world.open_gq, gripper_conf)
objects = []
commands = [
ApproachTrajectory(objects, world, world.robot, world.arm_joints,
approach_paths[0]),
DoorTrajectory(world, world.robot, world.arm_joints, arm_path,
world.kitchen, [door_joint], door_path),
ApproachTrajectory(objects, world, world.robot, world.arm_joints,
reversed(approach_paths[-1])),
]
door_path, _, _, _ = door_plan
sign = world.get_door_sign(door_joint)
pull = (sign * door_path[0][0] < sign * door_path[-1][0])
if pull:
commands.insert(1, finger_cmd.commands[0])
commands.insert(3, finger_cmd.commands[0].reverse())
cmd = Sequence(State(world, savers=[robot_saver]), commands, name='pull')
yield (
aq1,
aq2,
cmd,
)
def plan_pick(world,
obj_name,
pose,
grasp,
base_conf,
obstacles,
randomize=True,
**kwargs):
# TODO: check if within database convex hull
# TODO: flag to check if initially in collision
obj_body = world.get_body(obj_name)
pose.assign()
base_conf.assign()
world.open_gripper()
robot_saver = BodySaver(world.robot)
obj_saver = BodySaver(obj_body)
if randomize:
sample_fn = get_sample_fn(world.robot, world.arm_joints)
set_joint_positions(world.robot, world.arm_joints, sample_fn())
else:
world.carry_conf.assign()
world_from_body = pose.get_world_from_body()
gripper_pose = multiply(world_from_body, invert(
grasp.grasp_pose)) # w_f_g = w_f_o * (g_f_o)^-1
full_grasp_conf = world.solve_inverse_kinematics(gripper_pose)
if full_grasp_conf is None:
if PRINT_FAILURES: print('Grasp kinematic failure')
return
moving_links = get_moving_links(world.robot, world.arm_joints)
robot_obstacle = (world.robot, frozenset(moving_links))
#robot_obstacle = get_descendant_obstacles(world.robot, child_link_from_joint(world.arm_joints[0]))
#robot_obstacle = world.robot
if any(pairwise_collision(robot_obstacle, b) for b in obstacles):
if PRINT_FAILURES: print('Grasp collision failure')
#set_renderer(enable=True)
#wait_for_user()
#set_renderer(enable=False)
return
approach_pose = multiply(world_from_body, invert(grasp.pregrasp_pose))
approach_path = plan_approach(
world,
approach_pose, # attachments=[grasp.get_attachment()],
obstacles=obstacles,
**kwargs)
if approach_path is None:
if PRINT_FAILURES: print('Approach plan failure')
return
if MOVE_ARM:
aq = FConf(world.robot, world.arm_joints, approach_path[0])
else:
aq = world.carry_conf
gripper_motion_fn = get_gripper_motion_gen(world, **kwargs)
finger_cmd, = gripper_motion_fn(world.open_gq, grasp.get_gripper_conf())
attachment = create_surface_attachment(world, obj_name, pose.support)
objects = [obj_name]
cmd = Sequence(State(world,
savers=[robot_saver, obj_saver],
attachments=[attachment]),
commands=[
ApproachTrajectory(objects, world, world.robot,
world.arm_joints, approach_path),
finger_cmd.commands[0],
Detach(world, attachment.parent, attachment.parent_link,
attachment.child),
AttachGripper(world, obj_body, grasp=grasp),
ApproachTrajectory(objects, world, world.robot,
world.arm_joints,
reversed(approach_path)),
],
name='pick')
yield (
aq,
cmd,
)
def plan_press(world,
knob_name,
pose,
grasp,
base_conf,
obstacles,
randomize=True,
**kwargs):
base_conf.assign()
world.close_gripper()
robot_saver = BodySaver(world.robot)
if randomize:
sample_fn = get_sample_fn(world.robot, world.arm_joints)
set_joint_positions(world.robot, world.arm_joints, sample_fn())
else:
world.carry_conf.assign()
gripper_pose = multiply(pose, invert(
grasp.grasp_pose)) # w_f_g = w_f_o * (g_f_o)^-1
#set_joint_positions(world.gripper, get_movable_joints(world.gripper), world.closed_gq.values)
#set_tool_pose(world, gripper_pose)
full_grasp_conf = world.solve_inverse_kinematics(gripper_pose)
#wait_for_user()
if full_grasp_conf is None:
# if PRINT_FAILURES: print('Grasp kinematic failure')
return
robot_obstacle = (world.robot,
frozenset(get_moving_links(world.robot,
world.arm_joints)))
if any(pairwise_collision(robot_obstacle, b) for b in obstacles):
#if PRINT_FAILURES: print('Grasp collision failure')
return
approach_pose = multiply(pose, invert(grasp.pregrasp_pose))
approach_path = plan_approach(world,
approach_pose,
obstacles=obstacles,
**kwargs)
if approach_path is None:
return
aq = FConf(world.robot, world.arm_joints,
approach_path[0]) if MOVE_ARM else world.carry_conf
#gripper_motion_fn = get_gripper_motion_gen(world, **kwargs)
#finger_cmd, = gripper_motion_fn(world.open_gq, world.closed_gq)
objects = []
cmd = Sequence(
State(world, savers=[robot_saver]),
commands=[
#finger_cmd.commands[0],
ApproachTrajectory(objects, world, world.robot, world.arm_joints,
approach_path),
ApproachTrajectory(objects, world, world.robot, world.arm_joints,
reversed(approach_path)),
#finger_cmd.commands[0].reverse(),
Wait(world, duration=1.0),
],
name='press')
yield (
aq,
cmd,
)