コード例 #1
0
def ur5():
    I = Interface()
    I.connect()
    MOVABLE_ARM_JOINTS = [1, 2, 3, 4, 5, 6]
    ROBOT_URDF_PATH = '../models/ur5_description/urdf/ur5.urdf'
    END_EFFECTOR_LINK = 'end_effector_link'
    TABLE = '../models/table/table.urdf'
    CUBE_URDF = '../models/cube/%s_cube.urdf'

    # ------------------------UR5--------------------------#
    x_offset = 0.33
    y_offset = 0.15

    robot = I.load_model(ROBOT_URDF_PATH)

    redBlock1 = I.load_model(CUBE_URDF % 'red',
                             Pose(Point(0.3 + x_offset, -0.1 + y_offset,
                                        0.265)),
                             fixed_base=False)
    redBlock2 = I.load_model(CUBE_URDF % 'red',
                             Pose(
                                 Point(0.35 + x_offset, 0.05 + y_offset,
                                       0.265)),
                             fixed_base=False)
    table = I.load_model(TABLE,
                         Pose(Point(0.35 + x_offset, 0.0 + y_offset, 0.05),
                              Euler(0.0, 0.0, 1.57)),
                         fixed_base=False,
                         scaling=0.3)

    # ------------------------Jaco---------------------------#

    velocities = Csv('UR5_velocities.csv', [
        'joints_1', 'joints_2', 'joints_3', 'joints_4', 'joints_5', 'joints_6',
        'joints_11', 'joints_13', 'joints_15', 'joints_16', 'joints_17',
        'joints_18'
    ],
                     mode='r').read()
    torques = Csv('UR5_torques.csv', [
        'joints_1', 'joints_2', 'joints_3', 'joints_4', 'joints_5', 'joints_6',
        'joints_11', 'joints_13', 'joints_15', 'joints_16', 'joints_17',
        'joints_18'
    ],
                  mode='r').read()
    action = Csv('UR5_action.csv', ['action'], mode='r').read()

    for joint, values in velocities.items():
        velocities[joint] = map(float, values)
    velocities = np.vstack(velocities.values())

    for joint, values in torques.items():
        torques[joint] = map(float, values)

    torques = np.vstack(torques.values())

    I.replay(robot, [velocities.T, torques.T, action.values()[0]])
    #
    time.sleep(5.0)

    I.disconnect()
コード例 #2
0
def jaco():
    I = Interface()
    I.connect()

    # ------------------------Jaco---------------------------#
    ROBOT_URDF_PATH = '../models/jaco_pkg/jaco_description/urdf/test_jaco.urdf'
    BLOCK_URDF = "../ss-pybullet-master/models/drake/objects/block_for_pick_and_place_mid_size.urdf"
    END_EFFECTOR_LINK = 'jaco_eef_link'
    TABLE = '../models/table/table.urdf'
    CUBE_URDF = '../models/cube/%s_cube.urdf'
    TRAY_URDF = '../models/tray/tray.urdf'

    robot = I.load_model(ROBOT_URDF_PATH)

    redBlock1 = I.load_model(CUBE_URDF % 'red', Pose(Point(0.22, -0.2, 0.265)), fixed_base = False)
    redBlock2 = I.load_model(CUBE_URDF % 'red', Pose(Point(0.35, 0.05, 0.265)), fixed_base = False)
    greenBlock1 = I.load_model(CUBE_URDF % 'green', Pose(Point(0.38, -0.1, 0.2599)), fixed_base = False)
    greenBlock2 = I.load_model(CUBE_URDF % 'green', Pose(Point(0.38, 0.2, 0.2599)), fixed_base = False)
    table = I.load_model(TABLE, Pose(Point(0.35, 0.0, 0.05), Euler(0.0, 0.0, 1.57)), fixed_base = False, scaling = 0.3)

    # ------------------------Jaco---------------------------#

    velocities = Csv('JACO_velocities.csv',
                     ['joints_3', 'joints_5', 'joints_7', 'joints_9', 'joints_11', 'joints_13', 'joints_16',
                      'joints_19', 'joints_22'], mode = 'r').read()

    for joint, values in velocities.items():
        velocities[joint] = map(float, values)
    velocities = np.vstack(velocities.values())

    I.replay(robot, velocities.T)

    time.sleep(2.0)
    I.disconnect()
コード例 #3
0
def main():
    folders = get_folders(path=DATA_PATH)
    p_mode_velocity_file = 'p_mode_velocities.csv'
    scene_objects = 'scene_objects.npy'
    success = 0
    for i in range(507, 999):
        folder = DATA_PATH + 'Trial_%d/' % i
        with Interface() as I:
            # LOAD ROBOT
            robot = I.load_model(ROBOT_URDF_PATH)

            # LOAD TABLE
            table = I.load_model(TABLE,
                                 Pose(Point(0.4, 0.0, 0.05),
                                      Euler(1.57, 0.0, 0.0)),
                                 fixed_base=True,
                                 scaling=1.0)

            # TODO set camera
            I.set_camera_pose(0.6, 155, -50, [0.2, 0.0, 0.5])

            scene = np.load(os.path.join(folder, scene_objects))[2:, 0]

            block = I.load_model(CUBE_URDF,
                                 Pose(Point(*scene[0])),
                                 fixed_base=False)
            basket = I.load_model(BASKET_URDF,
                                  Pose(Point(*scene[1])),
                                  fixed_base=False)

            I.save_body_conf()

            if os.path.isfile(os.path.join(folder, p_mode_velocity_file)):
                velocities = Pd(os.path.join(
                    folder, p_mode_velocity_file)).read().values
                I.replay_velocities(robot, velocities)

                if inside_basket(I, block, basket):
                    I.restore_body_conf()
                    time.sleep(1.0)

                    I.replay_velocities(robot,
                                        velocities,
                                        filename=os.path.join(folder, 'img'))
                    success += 1

        if success == 270:
            break

        print('Trials : ', folder, ' , Success: ', success)
コード例 #4
0
def plan(I,
         robot,
         gripper_link,
         b,
         neutralPose,
         initialize=False,
         x_offset=0.0,
         y_offset=0.0):

    initial_pose = [neutralPose] if initialize else []

    target_pose = initial_pose + \
                    [Pose(Point(b[0].pose[0],b[0].pose[1],TABLE_HEIGHT + 0.2),Euler(0.0, 3.14, 1.57)),
                    Pose(Point(b[0].pose[0],b[0].pose[1],TABLE_HEIGHT + 0.02),Euler(0.0, 3.14, 1.57)),
                    Pose(Point(b[0].pose[0], b[0].pose[1], TABLE_HEIGHT + 0.2), Euler(0.0, 3.14, 1.57)),

                   Pose(Point(b[1].pose[0] + x_offset, b[1].pose[1] + y_offset, TABLE_HEIGHT + 0.3), Euler(0.0, 3.14, 1.57)),
                   Pose(Point(b[1].pose[0] + x_offset, b[1].pose[1] + y_offset, TABLE_HEIGHT + 0.08), Euler(0.0, 3.14, 1.57)),
                   Pose(Point(b[1].pose[0], b[1].pose[1], TABLE_HEIGHT + 0.3), Euler(0.0, 3.14, 1.57)),

                     ]
    stack_path = []
    for poses in target_pose:
        end_conf = I.calculate_inverse_kinematics(robot, gripper_link, poses)

        if end_conf is not None:
            path = I.plan_motion(robot, end_conf, [2, 3, 4, 5])
            if path is None:
                return None
            else:
                stack_path.append(path)
        else:
            return None
        I.set_joint_positions(robot, MOVABLE_ARM_JOINTS, end_conf)

    return stack_path
コード例 #5
0
def main():
    for i in range(TRIALS):
        SAVE_PATH = DATA_PATH + 'Trial_%d/' % i
        mkdir(SAVE_PATH)
        p_mode_velocity_file = SAVE_PATH + 'p_mode_velocities.csv'
        scene_object_file = SAVE_PATH + 'scene_objects'
        start = time.time()
        while time.time() - start <= 120.0:
            with Interface(save=True, filename=p_mode_velocity_file) as I:
                # LOAD ROBOT
                robot = I.load_model(ROBOT_URDF_PATH)

                # LOAD TABLE
                table = I.load_model(TABLE,
                                     Pose(Point(0.4, 0.0, 0.05),
                                          Euler(1.57, 0.0, 0.0)),
                                     fixed_base=True,
                                     scaling=1.0)

                #TODO set camera
                I.set_camera_pose(1.6, 60, -25, [0.2, 0.0, 0.7])

                #LOAD_BLOCKS
                blocks = load_blocks(I)
                if len(blocks) != len(BLOCKS):
                    continue
                time.sleep(1.0)

                # Gripper ID
                gripper_link = I.get_link_from_name(robot, END_EFFECTOR_LINK)

                # Neutral Pose
                neutralPose = Pose(Point(0.35, 0.0, 0.5),
                                   Euler(0.0, 3.14, 1.57))

                I.save_body_conf(save=True, filename=scene_object_file)
                offset = 0.005
                x_offset = [
                    0.0, 0.0, 0.0, offset, offset, offset, -offset, -offset,
                    -offset
                ]
                y_offset = [
                    0.0, offset, -offset, 0.0, offset, -offset, 0.0, offset,
                    -offset
                ]

                for k in range(9):
                    # Set initial state
                    I.restore_body_conf()
                    I.set_initial_state(robot)
                    success = False

                    # Plan motion
                    paths = {}
                    for j in range(0, len(blocks), 2):
                        initialize = True if j == 0 else False
                        path = plan(I,
                                    robot,
                                    gripper_link,
                                    blocks[j:j + 2],
                                    neutralPose,
                                    initialize=initialize,
                                    x_offset=x_offset[k],
                                    y_offset=y_offset[k])
                        if path is not None:
                            paths[j] = path
                        else:
                            paths = {}
                            break

                    if len(paths) > 0:
                        I.reset_simulation(robot)
                        time.sleep(1.0)

                        I.command(robot, paths[0][0])
                        I.command(robot, paths[0][1])
                        I.command(robot, paths[0][2])
                        I.gripper(robot, 'close')
                        I.command(robot, paths[0][3])
                        I.command(robot, paths[0][4])
                        I.command(robot, paths[0][5])

                        I.gripper(robot, 'open')
                        I.command(robot, paths[0][6])

                        time.sleep(1.0)

                        if on_top(I, *blocks[0:2]):

                            I.command(robot, paths[2][0])
                            I.command(robot, paths[2][1])
                            I.gripper(robot, 'close')
                            I.command(robot, paths[2][2])
                            I.command(robot, paths[2][3])
                            I.command(robot, paths[2][4])

                            I.gripper(robot, 'open')
                            I.command(robot, paths[2][5])

                            time.sleep(1.0)
                            if on_top(I, *blocks[2:4]):
                                I.command(robot, paths[4][0])
                                I.command(robot, paths[4][1])
                                I.gripper(robot, 'close')
                                I.command(robot, paths[4][2])
                                I.command(robot, paths[4][3])
                                I.command(robot, paths[4][4])

                                I.gripper(robot, 'open')
                                I.command(robot, paths[4][5])

                                time.sleep(1.0)
                                if on_top(I, *blocks[4:6]):
                                    success = True
                                    time.sleep(1.0)
                                    I.vel.write_csv()
                                    break

            if success:
                break
コード例 #6
0
I.connect()

x_offset = 0.33
y_offset = 0.15

robot = I.load_model(ROBOT_URDF_PATH)

redBlock1 = I.load_model(CUBE_URDF % 'red',
                         Pose(Point(0.3 + x_offset, -0.1 + y_offset, 0.265)),
                         fixed_base=False)
redBlock2 = I.load_model(CUBE_URDF % 'red',
                         Pose(Point(0.35 + x_offset, 0.05 + y_offset, 0.265)),
                         fixed_base=False)
table = I.load_model(TABLE,
                     Pose(Point(0.35 + x_offset, 0.0 + y_offset, 0.05),
                          Euler(0.0, 0.0, 1.57)),
                     fixed_base=False,
                     scaling=0.3)


def move_to(target_pose, gripper_link):
    paths = []
    for poses in target_pose:
        end_conf = I.calculate_inverse_kinematics(robot, gripper_link, poses)

        if end_conf is not None:
            path = I.plan_motion(robot, end_conf)
            if path is None:
                return None
            else:
                paths.append(path)
コード例 #7
0
def main():
    with open(pathname(os.getcwd(), '..', '..', 'machines.json'), 'r') as f:
        js = json.load(f)
    DATA_PATH = js[IP]['DATA_PATH']
    trial = js[IP]['trial_start']
    while True:
        if trial == js[IP]['trial_stop']:
            break
        start = time.time()
        while time.time() - start <= 120.0:
            SAVE_PATH = DATA_PATH + 'Trial_%d/' % trial
            mkdir(SAVE_PATH)
            p_mode_velocity_file = SAVE_PATH + 'p_mode_velocities.csv'
            scene_object_file = SAVE_PATH + 'scene'

            with Interface(mode='direct',
                           save=True,
                           filename=p_mode_velocity_file) as I:
                # LOAD ROBOT
                robot = I.load_model(ROBOT_URDF_PATH)

                # LOAD TABLE
                table = I.load_model(TABLE,
                                     Pose(Point(0.4, 0.0, 0.05),
                                          Euler(1.57, 0.0, 0.0)),
                                     fixed_base=True,
                                     scaling=1.0)

                # TODO set camera
                I.set_camera_pose(1.6, 60, -25, [0.2, 0.0, 0.7])

                # LOAD_BLOCKS
                blocks = load_blocks(I)
                if len(blocks) != len(BLOCKS):
                    continue
                time.sleep(1.0)

                # Gripper ID
                gripper_link = I.get_link_from_name(robot, END_EFFECTOR_LINK)

                # Neutral Pose
                neutralPose = Pose(Point(0.35, 0.0, 0.5),
                                   Euler(0.0, 3.14, 1.57))

                I.save_body_conf(save=True, filename=scene_object_file)
                I.set_initial_state(robot)

                offset = 0.005
                x_offset = [0.0, offset, -offset, -offset]
                y_offset = [0.0, offset, -offset, 0.0]

                shuffle(blocks)

                color = ['red']

                for k in range(1):
                    # Set initial state
                    I.restore_body_conf()
                    success = False

                    # Plan motion
                    paths = []
                    velocities = OrderedDict()

                    for j in range(1):
                        initialize = True if j == 0 else False
                        path = plan(I,
                                    robot,
                                    gripper_link,
                                    blocks,
                                    neutralPose,
                                    initialize=initialize,
                                    x_offset=x_offset[k],
                                    y_offset=y_offset[k])
                        if path is not None:
                            paths.append(path)
                        else:
                            paths = []
                            break

                    if len(paths) > 0:
                        I.reset_simulation(robot)
                        # time.sleep(1.0)

                        I.command(robot, paths[0][0])
                        I.command(robot, paths[0][1])
                        I.command(robot, paths[0][2])
                        I.gripper(robot, 'close')
                        I.command(robot, paths[0][3])
                        I.command(robot, paths[0][4])
                        I.command(robot, paths[0][5])

                        I.gripper(robot, 'open')
                        I.command(robot, paths[0][6])

                        if on_top(I, *blocks[0:2]):
                            I.keep_pose(body=robot)
                            I.vel.write_csv()

                            velocities = I.vel.array
                            I.restore_body_conf()
                            I.reset_simulation(robot)

                            I.replay_velocities(robot, velocities)

                            if on_top(I, *blocks[0:2]):
                                I.restore_body_conf()
                                I.reset_simulation(robot)
                                time.sleep(1.0)
                                I.replay_velocities(
                                    robot,
                                    velocities,
                                    path=os.path.abspath(SAVE_PATH))

                                if on_top(I, *blocks[0:2]):
                                    success = True
                                    np.save(
                                        os.path.join(SAVE_PATH, 'velocities'),
                                        np.array(I.target_vel))
                                    np.save(
                                        os.path.join(SAVE_PATH,
                                                     'joints_configurations'),
                                        np.array(I.joint_val))
                                    trial += 1
                                    break

                else:
                    shutil.rmtree(SAVE_PATH)
            if success:
                break
            else:
                if folder_exists(SAVE_PATH):
                    shutil.rmtree(SAVE_PATH)
コード例 #8
0
from pblt import Interface, MOVABLE_ARM_JOINTS
import time

from geometry import Point, Pose, Euler

ROBOT_URDF_PATH = '../models/jaco_pkg/jaco_description/urdf/test_jaco.urdf'
END_EFFECTOR_LINK = 'jaco_eef_link'
TABLE = '../models/table/table.urdf'
CUBE_URDF = '../models/cube/%s_cube.urdf'
BASKET_URDF = '../models/basket/basket.urdf'

with Interface() as I:

    robot = I.load_model(ROBOT_URDF_PATH)
    table = I.load_model(TABLE,
                         Pose(Point(0.35, 0.0, 0.05), Euler(0.0, 0.0, 1.57)),
                         fixed_base=False,
                         scaling=0.3)
    redBlock1 = I.load_model(CUBE_URDF % 'red',
                             Pose(Point(0.22, -0.2, 0.265)),
                             fixed_base=False)

    basket = I.load_model(BASKET_URDF,
                          Pose(Point(0.35, 0.05, 0.265)),
                          fixed_base=False)

    def move_to(target_pose, gripper_link):
        paths = []
        for poses in target_pose:
            end_conf = I.calculate_inverse_kinematics(robot, gripper_link,
                                                      poses)
コード例 #9
0
    "%(color)sBlock%(indice)d" % ({
        'color': name,
        'indice': indice
    }) for name in ['red', 'green', 'blue'] for indice in [1, 2]
]
BLOCK_SIZE = 0.05
Block = namedtuple('Block', ['pose', 'handle'])

I = Interface()
I.connect()
# LOAD ROBOT
robot = I.load_model(ROBOT_URDF_PATH)

# LOAD TABLE
table = I.load_model(TABLE,
                     Pose(Point(0.4, 0.0, 0.05), Euler(1.57, 0.0, 0.0)),
                     fixed_base=True,
                     scaling=1.0)

# TODO set camera
camTargetPos = [0, 0, 0]
cameraUp = [0, 0, 1]
cameraPos = [1, 1, 1]

pitch = -50
roll = 10
yaw = 70
upAxisIndex = 2
camDistance = 1.5
pixelWidth = 256
pixelHeight = 256
コード例 #10
0
def main():
    with open(pathname(os.getcwd(), '..', '..', 'machines.json'), 'r') as f:
        js = json.load(f)
    DATA_PATH = js[IP]['DATA_PATH']
    trial = js[IP]['trial_start']
    while True:
        if trial == js[IP]['trial_stop']:
            break

        start = time.time()
        while time.time() - start <= 120.0:
            SAVE_PATH = DATA_PATH + 'Trial_%d/' % trial
            mkdir(SAVE_PATH)
            p_mode_velocity_file = SAVE_PATH + 'p_mode_velocities.csv'
            scene_object_file = SAVE_PATH + 'scene_objects'
            with Interface(mode='direct',
                           save=True,
                           filename=p_mode_velocity_file) as I:
                # LOAD ROBOT
                robot = I.load_model(ROBOT_URDF_PATH)

                # LOAD TABLE
                table = I.load_model(TABLE,
                                     Pose(Point(0.4, 0.0, 0.05),
                                          Euler(1.57, 0.0, 0.0)),
                                     fixed_base=True,
                                     scaling=1.0)

                # TODO set camera
                I.set_camera_pose(1.6, 60, -25, [0.2, 0.0, 0.7])

                # LOAD SCENE OBJECTS
                scene_object = load_objects(
                    I, **dict(zip(SCENE_OBJECT, [CUBE_URDF, BASKET_URDF])))

                if len(scene_object) != len(SCENE_OBJECT):
                    continue

                # Gripper ID
                gripper_link = I.get_link_from_name(robot, END_EFFECTOR_LINK)

                # Neutral Pose
                neutralPose = Pose(Point(0.35, 0.0, 0.5),
                                   Euler(0.0, 3.14, 1.57))

                I.save_body_conf(save=True, filename=scene_object_file)
                I.set_initial_state(robot)

                offset = 0.005
                x_offset = [
                    0.0, 0.0, 0.0, offset, offset, offset, -offset, -offset,
                    -offset
                ]
                y_offset = [
                    0.0, offset, -offset, 0.0, offset, -offset, 0.0, offset,
                    -offset
                ]

                for k in range(2):
                    #    Set initial state
                    I.restore_body_conf()
                    success = False

                    initialize = True
                    path = plan(I,
                                robot,
                                gripper_link,
                                scene_object,
                                neutralPose,
                                initialize=initialize,
                                x_offset=x_offset[k],
                                y_offset=y_offset[k])

                    if len(path) > 0:
                        I.reset_simulation(robot)
                        time.sleep(1.0)

                        I.command(robot, path[0])
                        I.command(robot, path[1])
                        I.command(robot, path[2])
                        I.gripper(robot)

                        I.command(robot, path[3])
                        I.command(robot, path[4])
                        I.command(robot, path[5])

                        I.gripper(robot, 'open')
                        I.command(robot, path[6])

                        if inside_basket(I, *scene_object):
                            success = True
                            I.keep_pose(body=robot)
                            I.vel.write_csv()

                            velocities = I.vel.array
                            I.restore_body_conf()
                            I.reset_simulation(robot)

                            I.replay_velocities(robot, velocities)

                            if inside_basket(I, *scene_object):
                                I.restore_body_conf()
                                I.reset_simulation(robot)
                                time.sleep(1.0)
                                I.replay_velocities(
                                    robot,
                                    velocities,
                                    path=os.path.abspath(SAVE_PATH))
                                if inside_basket(I, *scene_object):
                                    np.save(
                                        os.path.join(SAVE_PATH, 'velocities'),
                                        np.array(I.target_vel))
                                    np.save(
                                        os.path.join(SAVE_PATH,
                                                     'joints_configurations'),
                                        np.array(I.joint_val))
                                    trial += 1
                                else:
                                    shutil.rmtree(SAVE_PATH)

                            else:
                                shutil.rmtree(SAVE_PATH)

                            break
                else:

                    shutil.rmtree(SAVE_PATH)
            if success:
                break
コード例 #11
0
def main():

    trial = 2495
    while True:
        if trial == 3000:
            break

        start = time.time()
        while time.time() - start <= 120.0:
            SAVE_PATH = DATA_PATH + 'Trial_%d/' % trial
            mkdir(SAVE_PATH)
            p_mode_velocity_file = SAVE_PATH + 'VELOCITIES/'

            scene_object_file = SAVE_PATH + 'SCENE_OBJECTS/'
            blocks_goals_file = SAVE_PATH + 'GOALS/'
            mkdir(scene_object_file)
            mkdir(p_mode_velocity_file)
            with Interface(mode='direct',
                           save=True,
                           filename=p_mode_velocity_file +
                           'p_mode_velocities.csv') as I:
                # LOAD ROBOT
                robot = I.load_model(ROBOT_URDF_PATH)

                # LOAD TABLE
                table = I.load_model(TABLE,
                                     Pose(Point(0.4, 0.0, 0.05),
                                          Euler(1.57, 0.0, 0.0)),
                                     fixed_base=True,
                                     scaling=1.0)

                #TODO set camera
                I.set_camera_pose(1.6, 60, -25, [0.2, 0.0, 0.7])

                #LOAD_BLOCKS
                blocks = load_blocks(I)
                if len(blocks) != len(BLOCKS):
                    continue
                time.sleep(1.0)

                # Gripper ID
                gripper_link = I.get_link_from_name(robot, END_EFFECTOR_LINK)

                # Neutral Pose
                neutralPose = Pose(Point(0.35, 0.0, 0.5),
                                   Euler(0.0, 3.14, 1.57))

                I.save_body_conf(save=True,
                                 filename=scene_object_file + 'scene')
                I.set_initial_state(robot)

                offset = 0.005
                x_offset = [
                    0.0, 0.0, 0.0, offset, offset, offset, -offset, -offset,
                    -offset
                ]
                y_offset = [
                    0.0, offset, -offset, 0.0, offset, -offset, 0.0, offset,
                    -offset
                ]

                blocks = [blocks[k * 2:2 * (k + 1)] for k in range(3)]
                shuffle(blocks)
                blocks = blocks[0] + blocks[1] + blocks[2]
                color = [blocks[0].name, blocks[2].name, blocks[4].name]
                for k in range(9):
                    # Set initial state
                    I.restore_body_conf()
                    success = False

                    # Plan motion
                    paths = {}
                    velocities = OrderedDict()

                    for j in range(0, len(blocks), 2):
                        initialize = True if j == 0 else False
                        path = plan(I,
                                    robot,
                                    gripper_link,
                                    blocks[j:j + 2],
                                    neutralPose,
                                    initialize=initialize,
                                    x_offset=x_offset[k],
                                    y_offset=y_offset[k])
                        if path is not None:
                            paths[j] = path
                        else:
                            paths = {}
                            break

                    if len(paths) > 0:
                        I.reset_simulation(robot)
                        # time.sleep(1.0)

                        I.command(robot, paths[0][0])
                        I.command(robot, paths[0][1])
                        I.command(robot, paths[0][2])
                        I.gripper(robot, 'close')
                        I.command(robot, paths[0][3])
                        I.command(robot, paths[0][4])
                        I.command(robot, paths[0][5])

                        I.gripper(robot, 'open')
                        I.command(robot, paths[0][6])

                        # time.sleep(1.0)

                        if on_top(I, *blocks[0:2]):
                            velocities[color[0]] = I.vel.array
                            shape = I.vel.array.shape[0]

                            I.command(robot, paths[2][0])
                            I.command(robot, paths[2][1])
                            I.gripper(robot, 'close')
                            I.command(robot, paths[2][2])
                            I.command(robot, paths[2][3])
                            I.command(robot, paths[2][4])

                            I.gripper(robot, 'open')
                            I.command(robot, paths[2][5])

                            # time.sleep(1.0)
                            if on_top(I, *blocks[2:4]):
                                velocities[color[1]] = I.vel.array[shape:]
                                shape = I.vel.array.shape[0]

                                I.command(robot, paths[4][0])
                                I.command(robot, paths[4][1])
                                I.gripper(robot, 'close')
                                I.command(robot, paths[4][2])
                                I.command(robot, paths[4][3])
                                I.command(robot, paths[4][4])

                                I.gripper(robot, 'open')
                                I.command(robot, paths[4][5])

                                # time.sleep(1.0)
                                if on_top(I, *blocks[4:6]):
                                    velocities[color[2]] = I.vel.array[shape:]

                                    success = True
                                    time.sleep(1.0)

                                    for e, colors in enumerate(
                                            velocities.keys()):
                                        mkdir(blocks_goals_file + '%d_' % e +
                                              colors)
                                        np.save(
                                            blocks_goals_file + '%d_' % e +
                                            colors + '/' + colors,
                                            velocities[colors])
                                        I.vel.write_csv()
                                    I.restore_body_conf()
                                    I.reset_simulation(robot)
                                    time.sleep(1.0)

                                    task_velocities = np.concatenate(
                                        velocities.values())

                                    I.replay_velocities(robot, task_velocities)

                                    final_stacking = []
                                    for k in range(0, len(blocks), 2):
                                        final_stacking.append(
                                            on_top(I, *blocks[k:k + 2]))

                                    if all(final_stacking):
                                        I.restore_body_conf()
                                        I.reset_simulation(robot)
                                        time.sleep(1.0)
                                        replay_final_stacking = []
                                        I.replay_velocities(
                                            robot,
                                            task_velocities,
                                            path=os.path.abspath(
                                                os.path.join(
                                                    SAVE_PATH, 'VELOCITIES')))
                                        for k in range(0, len(blocks), 2):
                                            replay_final_stacking.append(
                                                on_top(I, *blocks[k:k + 2]))
                                        if all(replay_final_stacking):
                                            #save
                                            np.save(
                                                os.path.join(
                                                    SAVE_PATH, 'VELOCITIES',
                                                    'velocities'),
                                                np.array(I.target_vel))
                                            np.save(
                                                os.path.join(
                                                    SAVE_PATH, 'VELOCITIES',
                                                    'joints_configurations'),
                                                np.array(I.joint_val))

                                            trial += 1
                                        else:
                                            shutil.rmtree(SAVE_PATH)

                                    else:

                                        shutil.rmtree(SAVE_PATH)

                                    break

                else:
                    shutil.rmtree(SAVE_PATH)
            if success:
                break