def task2(actions): ship_coords = (0, 0) wayp_coords = (10, 1) for action_str in actions: log(action_str, ship_coords, wayp_coords) action = action_str[0] value = int(action_str[1:]) if action in ("N", "S", "E", "W"): if action == "N": move_dir = Direction.NORTH() elif action == "S": move_dir = Direction.SOUTH() elif action == "E": move_dir = Direction.EAST() elif action == "W": move_dir = Direction.WEST() for _ in range(value): wayp_coords = move_dir.move_grid_coordinates(wayp_coords) elif action == "F": ship_coords = (ship_coords[0] + value * wayp_coords[0], ship_coords[1] + value * wayp_coords[1]) elif action == "L": wayp_coords = rotate_counterclockwise(wayp_coords, value, origin=(0,0)) elif action == "R": wayp_coords = rotate_clockwise(wayp_coords, value, origin=(0,0)) log(f"New coordinates {ship_coords} {wayp_coords}\n") return abs(int(ship_coords[0])) + abs(int(ship_coords[1]))
def task1(actions): coords = (0, 0) cur_dir = Direction.EAST() for action_str in actions: log(action_str, coords) action = action_str[0] value = int(action_str[1:]) if action in ("N", "S", "E", "W", "F"): if action == "N": move_dir = Direction.NORTH() elif action == "S": move_dir = Direction.SOUTH() elif action == "E": move_dir = Direction.EAST() elif action == "W": move_dir = Direction.WEST() elif action == "F": move_dir = cur_dir for _ in range(value): coords = move_dir.move_grid_coordinates(coords) elif action == "L": cur_dir.rotate_counterclockwise(value) elif action == "R": cur_dir.rotate_clockwise(value) log(f"New coordinates {coords}\n") return abs(coords[0]) + abs(coords[1])