data/task_scripts/main/task01001.py [21:102]: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - @creator_lib.define_task_template( seed=range(1000), version="2", search_params=dict( required_flags=['BALL:GOOD_STABLE'], excluded_flags=['BALL:TRIVIAL'], ), ) def build_task(C, seed): rng = np.random.RandomState(seed=seed) leftWall = [80, 180] breakPoint = [0.2, 0.8] bridgeLength = [150, 300] slopeL = [60, 200] slopeR = [250, 400] flip_lr = rng.uniform(0, 1) < 0.5 ## Make the goal container height = rng.uniform(slopeL[0], slopeL[1]) lW = rng.uniform(leftWall[0], leftWall[1]) container, _ = vt.add_container( C, [[5, height], [5, 5], [lW - 5, 5], [lW - 5, height]], 10, False, True, flip_lr=flip_lr) #'green', 0) ## Make the bridge bL = rng.uniform(bridgeLength[0], bridgeLength[1]) bP = rng.uniform(breakPoint[0], breakPoint[1]) bridgeL = vt.add_box( C, [lW + 20 + 2, height + 10, lW + 20 + 2 + bL * bP, height + 20], True, flip_lr=flip_lr) bridgeR = vt.add_box(C, [ lW + 20 + 2 + bL * bP + 2, height + 10, lW + 20 + 2 + bL + 2, height + 20 ], True, flip_lr=flip_lr) ## Make the left wall lw1 = vt.add_box(C, [lW, 0, lW + 40, height - 10], False, flip_lr=flip_lr) lw2 = vt.add_box(C, [lW, height - 10, lW + 20, height], False, flip_lr=flip_lr) ## Make the right wall sW = rng.uniform(min([lW + 20 + 2 + bL + 2, vt.VT_SCALE - 30]), max([vt.VT_SCALE - 30, lW + 20 + 2 + bL + 2])) rw1 = vt.add_box(C, [lW + 20 + 2 + bL + 2 - 50, 0, vt.VT_SCALE, height - 10], False, flip_lr=flip_lr) rw2 = vt.add_box(C, [lW + 20 + 2 + bL + 2, 0, sW, height], False, flip_lr=flip_lr) ## Make the slope sR = rng.uniform(slopeR[0], slopeR[1]) slopeVerts = [(sW, height - 10), (sW, height), (vt.VT_SCALE, sR), (vt.VT_SCALE, height - 10)] slopeVerts.reverse() if flip_lr: slopeVerts = vt.flip_left_right(slopeVerts) center_x = vt.flip_left_right(rng.uniform(500, 575)) else: center_x = rng.uniform(500, 575) rw2 = C.add_convex_polygon(vt.convert_phyre_tools_vertices(slopeVerts), False) center_y = rng.uniform(425, 525) ball = C.add('dynamic ball', 15 * 2. / vt.VT_SCALE, center_x=center_x * C.scene.width / vt.VT_SCALE, center_y=center_y * C.scene.width / vt.VT_SCALE) C.update_task(body1=ball, body2=container, relationships=[C.SpatialRelationship.TOUCHING]) C.set_meta(C.SolutionTier.VIRTUAL_TOOLS) - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - src/python/phyre/data/task_scripts/main/task01001.py [21:102]: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - @creator_lib.define_task_template( seed=range(1000), version="2", search_params=dict( required_flags=['BALL:GOOD_STABLE'], excluded_flags=['BALL:TRIVIAL'], ), ) def build_task(C, seed): rng = np.random.RandomState(seed=seed) leftWall = [80, 180] breakPoint = [0.2, 0.8] bridgeLength = [150, 300] slopeL = [60, 200] slopeR = [250, 400] flip_lr = rng.uniform(0, 1) < 0.5 ## Make the goal container height = rng.uniform(slopeL[0], slopeL[1]) lW = rng.uniform(leftWall[0], leftWall[1]) container, _ = vt.add_container( C, [[5, height], [5, 5], [lW - 5, 5], [lW - 5, height]], 10, False, True, flip_lr=flip_lr) #'green', 0) ## Make the bridge bL = rng.uniform(bridgeLength[0], bridgeLength[1]) bP = rng.uniform(breakPoint[0], breakPoint[1]) bridgeL = vt.add_box( C, [lW + 20 + 2, height + 10, lW + 20 + 2 + bL * bP, height + 20], True, flip_lr=flip_lr) bridgeR = vt.add_box(C, [ lW + 20 + 2 + bL * bP + 2, height + 10, lW + 20 + 2 + bL + 2, height + 20 ], True, flip_lr=flip_lr) ## Make the left wall lw1 = vt.add_box(C, [lW, 0, lW + 40, height - 10], False, flip_lr=flip_lr) lw2 = vt.add_box(C, [lW, height - 10, lW + 20, height], False, flip_lr=flip_lr) ## Make the right wall sW = rng.uniform(min([lW + 20 + 2 + bL + 2, vt.VT_SCALE - 30]), max([vt.VT_SCALE - 30, lW + 20 + 2 + bL + 2])) rw1 = vt.add_box(C, [lW + 20 + 2 + bL + 2 - 50, 0, vt.VT_SCALE, height - 10], False, flip_lr=flip_lr) rw2 = vt.add_box(C, [lW + 20 + 2 + bL + 2, 0, sW, height], False, flip_lr=flip_lr) ## Make the slope sR = rng.uniform(slopeR[0], slopeR[1]) slopeVerts = [(sW, height - 10), (sW, height), (vt.VT_SCALE, sR), (vt.VT_SCALE, height - 10)] slopeVerts.reverse() if flip_lr: slopeVerts = vt.flip_left_right(slopeVerts) center_x = vt.flip_left_right(rng.uniform(500, 575)) else: center_x = rng.uniform(500, 575) rw2 = C.add_convex_polygon(vt.convert_phyre_tools_vertices(slopeVerts), False) center_y = rng.uniform(425, 525) ball = C.add('dynamic ball', 15 * 2. / vt.VT_SCALE, center_x=center_x * C.scene.width / vt.VT_SCALE, center_y=center_y * C.scene.width / vt.VT_SCALE) C.update_task(body1=ball, body2=container, relationships=[C.SpatialRelationship.TOUCHING]) C.set_meta(C.SolutionTier.VIRTUAL_TOOLS) - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -