def packed(arm='left', grasp_type='top', num=5):
# TODO: packing problem where you have to place in one direction
base_extent = 5.0
base_limits = (-base_extent / 2. * np.ones(2),
base_extent / 2. * np.ones(2))
block_width = 0.07
block_height = 0.1
#block_height = 2*block_width
block_area = block_width * block_width
#plate_width = 2*math.sqrt(num*block_area)
plate_width = 0.27
#plate_width = 0.28
#plate_width = 0.3
print('Width:', plate_width)
plate_width = min(plate_width, 0.6)
plate_height = 0.001
other_arm = get_other_arm(arm)
initial_conf = get_carry_conf(arm, grasp_type)
add_data_path()
floor = load_pybullet("plane.urdf")
pr2 = create_pr2()
set_arm_conf(pr2, arm, initial_conf)
open_arm(pr2, arm)
set_arm_conf(pr2, other_arm, arm_conf(other_arm, REST_LEFT_ARM))
close_arm(pr2, other_arm)
set_group_conf(pr2, 'base',
[-1.0, 0, 0]) # Be careful to not set the pr2's pose
table = create_table()
plate = create_box(plate_width, plate_width, plate_height, color=GREEN)
plate_z = stable_z(plate, table)
set_point(plate, Point(z=plate_z))
surfaces = [table, plate]
blocks = [
create_box(block_width, block_width, block_height, color=BLUE)
for _ in range(num)
]
initial_surfaces = {block: table for block in blocks}
min_distances = {block: 0.05 for block in blocks}
sample_placements(initial_surfaces, min_distances=min_distances)
return Problem(
robot=pr2,
movable=blocks,
arms=[arm],
grasp_types=[grasp_type],
surfaces=surfaces,
#goal_holding=[(arm, block) for block in blocks])
goal_on=[(block, plate) for block in blocks],
base_limits=base_limits)
def main():
parser = argparse.ArgumentParser() # Automatically includes help
parser.add_argument('-viewer', action='store_true', help='enable viewer.')
args = parser.parse_args()
connect(use_gui=True)
#ycb_path = os.path.join(DRAKE_PATH, DRAKE_YCB)
#ycb_path = os.path.join(YCB_PATH, YCB_TEMPLATE.format('003_cracker_box'))
#print(ycb_path)
#load_pybullet(ycb_path)
with LockRenderer():
draw_pose(unit_pose(), length=1, width=1)
floor = create_floor()
set_point(floor, Point(z=-EPSILON))
table = create_table(width=TABLE_WIDTH, length=TABLE_WIDTH/2, height=TABLE_WIDTH/2, top_color=TAN, cylinder=False)
#set_euler(table, Euler(yaw=np.pi/2))
with HideOutput(False):
# data_path = add_data_path()
# robot_path = os.path.join(data_path, WSG_GRIPPER)
robot_path = get_model_path(WSG_50_URDF) # WSG_50_URDF | PANDA_HAND_URDF
#robot_path = get_file_path(__file__, 'mit_arch_suction_gripper/mit_arch_suction_gripper.urdf')
robot = load_pybullet(robot_path, fixed_base=True)
#dump_body(robot)
#robot = create_cylinder(radius=0.5*BLOCK_SIDE, height=4*BLOCK_SIDE) # vacuum gripper
block1 = create_box(w=BLOCK_SIDE, l=BLOCK_SIDE, h=BLOCK_SIDE, color=RED)
block_z = stable_z(block1, table)
set_point(block1, Point(z=block_z))
block2 = create_box(w=BLOCK_SIDE, l=BLOCK_SIDE, h=BLOCK_SIDE, color=GREEN)
set_point(block2, Point(x=+0.25, z=block_z))
block3 = create_box(w=BLOCK_SIDE, l=BLOCK_SIDE, h=BLOCK_SIDE, color=BLUE)
set_point(block3, Point(x=-0.15, z=block_z))
blocks = [block1, block2, block3]
add_line(Point(x=-TABLE_WIDTH/2, z=block_z - BLOCK_SIDE/2 + EPSILON),
Point(x=+TABLE_WIDTH/2, z=block_z - BLOCK_SIDE/2 + EPSILON), color=RED)
set_camera_pose(camera_point=Point(y=-1, z=block_z+1), target_point=Point(z=block_z))
wait_for_user()
block_pose = get_pose(block1)
open_gripper(robot)
tool_link = link_from_name(robot, 'tool_link')
base_from_tool = get_relative_pose(robot, tool_link)
#draw_pose(unit_pose(), parent=robot, parent_link=tool_link)
y_grasp, x_grasp = get_top_grasps(block1, tool_pose=unit_pose(), grasp_length=0.03, under=False)
grasp = y_grasp # fingers won't collide
gripper_pose = multiply(block_pose, invert(grasp))
set_pose(robot, multiply(gripper_pose, invert(base_from_tool)))
wait_for_user('Finish?')
disconnect()
def blocked(arm='left', grasp_type='side', num=1):
x_extent = 10.0
base_limits = (-x_extent / 2. * np.ones(2), x_extent / 2. * np.ones(2))
block_width = 0.07
#block_height = 0.1
block_height = 2 * block_width
#block_height = 0.2
plate_height = 0.001
table_x = (x_extent - 1) / 2.
other_arm = get_other_arm(arm)
initial_conf = get_carry_conf(arm, grasp_type)
add_data_path()
floor = load_pybullet("plane.urdf")
pr2 = create_pr2()
set_arm_conf(pr2, arm, initial_conf)
open_arm(pr2, arm)
set_arm_conf(pr2, other_arm, arm_conf(other_arm, REST_LEFT_ARM))
close_arm(pr2, other_arm)
set_group_conf(
pr2, 'base',
[x_extent / 4, 0, 0]) # Be careful to not set the pr2's pose
table1 = create_table()
set_point(table1, Point(x=+table_x, y=0))
table2 = create_table()
set_point(table2, Point(x=-table_x, y=0))
#table3 = create_table()
#set_point(table3, Point(x=0, y=0))
plate = create_box(0.6, 0.6, plate_height, color=GREEN)
x, y, _ = get_point(table1)
plate_z = stable_z(plate, table1)
set_point(plate, Point(x=x, y=y - 0.3, z=plate_z))
#surfaces = [table1, table2, table3, plate]
surfaces = [table1, table2, plate]
green1 = create_box(block_width, block_width, block_height, color=BLUE)
green1_z = stable_z(green1, table1)
set_point(green1, Point(x=x, y=y + 0.3, z=green1_z))
# TODO: can consider a fixed wall here instead
spacing = 0.15
#red_directions = [(-1, 0), (+1, 0), (0, -1), (0, +1)]
red_directions = [(-1, 0)]
#red_directions = []
red_bodies = []
for red_direction in red_directions:
red = create_box(block_width, block_width, block_height, color=RED)
red_bodies.append(red)
x, y = get_point(green1)[:2] + spacing * np.array(red_direction)
z = stable_z(red, table1)
set_point(red, Point(x=x, y=y, z=z))
wall1 = create_box(0.01, 2 * spacing, block_height, color=GREY)
wall2 = create_box(spacing, 0.01, block_height, color=GREY)
wall3 = create_box(spacing, 0.01, block_height, color=GREY)
z = stable_z(wall1, table1)
x, y = get_point(green1)[:2]
set_point(wall1, Point(x=x + spacing, y=y, z=z))
set_point(wall2, Point(x=x + spacing / 2, y=y + spacing, z=z))
set_point(wall3, Point(x=x + spacing / 2, y=y - spacing, z=z))
green_bodies = [
create_box(block_width, block_width, block_height, color=BLUE)
for _ in range(num)
]
body_types = [(b, 'green')
for b in [green1] + green_bodies] # + [(table1, 'sink')]
movable = [green1] + green_bodies + red_bodies
initial_surfaces = {block: table2 for block in green_bodies}
sample_placements(initial_surfaces)
return Problem(
robot=pr2,
movable=movable,
arms=[arm],
grasp_types=[grasp_type],
surfaces=surfaces,
#sinks=[table1],
#goal_holding=[(arm, '?green')],
#goal_cleaned=['?green'],
goal_on=[('?green', plate)],
body_types=body_types,
base_limits=base_limits,
costs=True)
def main():
parser = argparse.ArgumentParser() # Automatically includes help
parser.add_argument('-viewer', action='store_true', help='enable viewer.')
args = parser.parse_args()
connect(use_gui=True)
with LockRenderer():
draw_pose(unit_pose(), length=1, width=1)
floor = create_floor()
set_point(floor, Point(z=-EPSILON))
table1 = create_table(width=TABLE_WIDTH,
length=TABLE_WIDTH / 2,
height=TABLE_WIDTH / 2,
top_color=TAN,
cylinder=False)
set_point(table1, Point(y=+0.5))
table2 = create_table(width=TABLE_WIDTH,
length=TABLE_WIDTH / 2,
height=TABLE_WIDTH / 2,
top_color=TAN,
cylinder=False)
set_point(table2, Point(y=-0.5))
tables = [table1, table2]
#set_euler(table1, Euler(yaw=np.pi/2))
with HideOutput():
# data_path = add_data_path()
# robot_path = os.path.join(data_path, WSG_GRIPPER)
robot_path = get_model_path(
WSG_50_URDF) # WSG_50_URDF | PANDA_HAND_URDF
robot = load_pybullet(robot_path, fixed_base=True)
#dump_body(robot)
block1 = create_box(w=BLOCK_SIDE,
l=BLOCK_SIDE,
h=BLOCK_SIDE,
color=RED)
block_z = stable_z(block1, table1)
set_point(block1, Point(y=-0.5, z=block_z))
block2 = create_box(w=BLOCK_SIDE,
l=BLOCK_SIDE,
h=BLOCK_SIDE,
color=GREEN)
set_point(block2, Point(x=-0.25, y=-0.5, z=block_z))
block3 = create_box(w=BLOCK_SIDE,
l=BLOCK_SIDE,
h=BLOCK_SIDE,
color=BLUE)
set_point(block3, Point(x=-0.15, y=+0.5, z=block_z))
blocks = [block1, block2, block3]
set_camera_pose(camera_point=Point(x=-1, z=block_z + 1),
target_point=Point(z=block_z))
block_pose = get_pose(block1)
open_gripper(robot)
tool_link = link_from_name(robot, 'tool_link')
base_from_tool = get_relative_pose(robot, tool_link)
#draw_pose(unit_pose(), parent=robot, parent_link=tool_link)
grasps = get_side_grasps(block1,
tool_pose=Pose(euler=Euler(yaw=np.pi / 2)),
top_offset=0.02,
grasp_length=0.03,
under=False)[1:2]
for grasp in grasps:
gripper_pose = multiply(block_pose, invert(grasp))
set_pose(robot, multiply(gripper_pose, invert(base_from_tool)))
wait_for_user()
wait_for_user('Finish?')
disconnect()