def pick(self, block_coord: Coordinate, robot: BuilderState):
if block_coord in self.sim_state.s_blocks:
del self.sim_state.s_blocks[block_coord]
robot.block = HeldBlock.SCAFFOLD
elif block_coord in self.sim_state.b_blocks:
self.sim_state.b_blocks.remove(block_coord)
robot.block = HeldBlock.BUILD
else:
raise RobotActionError(
f'Attempting to pick an empty block at {block_coord}')
def drop(self, block_coord: Coordinate, robot: BuilderState):
if robot.block is HeldBlock.NONE:
raise RobotActionError('Cannot drop NONE Block')
sim_state = self.sim_state
if block_coord in sim_state.b_blocks or block_coord in sim_state.s_blocks:
raise RobotActionError('Block already present')
if robot.block is HeldBlock.BUILD:
self.sim_state.b_blocks.add(block_coord)
elif robot.block is HeldBlock.SCAFFOLD:
self.sim_state.s_blocks[block_coord] = ScaffoldState()
robot.block = HeldBlock.NONE
states = SimulationStateList.create_with_empty_states(400)
s_blocks = {}
b_blocks = {}
robots = {}
for instruction in SInstruction:
s_blocks[get_next_coord()] = ScaffoldState(instruction)
b_blocks[get_next_coord()] = BuildingBlockState()
for direction in Direction:
for held_block in HeldBlock:
robot = BuilderState()
robot.block = held_block
robot.direction = direction
coord = get_next_coord()
s_blocks[coord] = ScaffoldState(SInstruction.DRIVE_RIGHT)
robots[coord] = robot
states.states[0].s_blocks = s_blocks
states.states[0].b_blocks = b_blocks
states.states[0].robots = robots
goal = Goal(get_next_coord(), GoalType.PLACE_SCAFFOLD)
states.states[0].target_structure = [get_next_coord()]
states.states[0].goal_stack.append(goal)
s_blocks = {}
b_blocks = {}
robots = {}
get_next_coord(): ScaffoldState(SInstruction.NONE),
get_next_coord(): ScaffoldState(SInstruction.STOP),
get_next_coord(): ScaffoldState(SInstruction.STOP),
get_next_coord(): ScaffoldState(SInstruction.NONE),
# row 9
get_next_coord(): ScaffoldState(SInstruction.DRIVE_RIGHT),
get_next_coord(): ScaffoldState(SInstruction.NONE),
get_next_coord(): ScaffoldState(SInstruction.NONE),
get_next_coord(): ScaffoldState(SInstruction.DRIVE_RIGHT),
get_next_coord(): ScaffoldState(SInstruction.STOP),
get_next_coord(): ScaffoldState(SInstruction.STOP),
get_next_coord(): ScaffoldState(SInstruction.DRIVE_RIGHT),
get_next_coord(): ScaffoldState(SInstruction.NONE),
get_next_coord(): ScaffoldState(SInstruction.NONE),
get_next_coord(): ScaffoldState(SInstruction.DRIVE_RIGHT),
}
sim = SimulationState()
sim.s_blocks = s_blocks
robot = BuilderState()
robot.direction = Direction.EAST
robot.block = HeldBlock.BUILD
sim.robots = {Coordinate(1, 0): copy.copy(robot)}
root = tk.Tk()
player = BasicPlayer(root, sim)
player.grid()
root.mainloop()