#!/usr/bin/env python
# 2021 - Advent Of Code - 13
# 2022: cheat for pylint to start fresh for 2022
# pylint: disable=unnecessary-list-index-lookup


def parse_file(file):
    parsed_dots = []
    parsed_folds = []
    with open(file, encoding="utf-8") as f:
        for line in f.readlines():
            if line.startswith('fold along '):
                _, f = line.strip().split('fold along ')
                parsed_folds.append(f.split('='))
            elif line == '\n':
                pass
            else:
                dot = [int(d) for d in line.strip().split(',')]
                parsed_dots.append(dot)
        return parsed_dots, parsed_folds


def find_max(dot_list):
    mx = 0
    my = 0
    for d in dot_list:
        if d[0] > mx:
            mx = d[0]
        if d[1] > my:
            my = d[1]

    return mx, my


def place_dots(g, dots_list):
    for dot in dots_list:
        g[dot[1]][dot[0]] = 1


def print_grid(g, fold=-1):
    for y, _ in enumerate(g):
        if y == fold:
            for x, _ in enumerate(g[y]):
                print('-', end='')
        else:
            for x, _ in enumerate(g[y]):
                if g[y][x] == 1:
                    print('#', end='')
                else:
                    print('.', end='')
        print()


def fold_horizontally(g, fold_y):
    print(f'folding y={fold_y}')
    for y in range(fold_y, maxy+1):
        folding_dist = y - fold_y
        new_y = fold_y - folding_dist

        # place new dots
        for x, _ in enumerate(g[y]):
            if g[y][x] == 1:
                g[new_y][x] = 1

    # cut the other half of the grid
    for _ in range(fold_y+1):
        g.pop()


def fold_vertically(g, fold_x):
    print(f'folding x={fold_x}')
    for y, _ in enumerate(g):
        # place new dots
        for x in range(fold_x, maxx+1):
            folding_dist = x - fold_x
            new_x = fold_x - folding_dist
            if g[y][x] == 1:
                g[y][new_x] = 1

        # cut the other half of the grid
        g[y] = g[y][:fold_x]


def count_dots(g):
    count = 0
    for y, _ in enumerate(g):
        for x, _ in enumerate(g[y]):
            if g[y][x] == 1:
                count += 1

    return count


# dots, folds = parse_file('input_example.txt')
dots, folds = parse_file('input.txt')

maxx, maxy = find_max(dots)
grid = [[0] * (maxx+1) for _ in range(maxy+1)]

# for f in folds:
#     print(f'folding {f[0]} = {f[1]}')
# for d in dots:
#     print(f'dot: {d}')

place_dots(grid, dots)
# print_grid(grid, 7)
print(f'grid size: {maxx}, {maxy}')

# folding_y = int(folds[0][1])
# print()
# fold_horizontally(grid, folding_y)
# print()
fold_vertically(grid, 655)

# print_grid(grid)
print(f'number of dots: {count_dots(grid)}')