import time

filename = "data"

try:
    with open(filename, "r") as file:
        lines = file.read().splitlines()
        print("File opened successfully.\n")
except:
    input("File '{filename}' not found. Program will now exit.\n")
    exit()

for i, line in enumerate(lines):
    lines[i] = [char for char in line]

y_range = range(len(lines))
x_range = range(len(lines[0]))
x_max = max(x_range)
y_max = max(y_range)
def tilt(_dir):

    if -1 in _dir:
        y = 0

        for line in lines:
            x = 0
            for char in line:
                if char not in "#.":
                    _y = y
                    _x = x
                    while ((_y + _dir[0]) in y_range) and ((_x + _dir[1]) in x_range) and lines[_y + _dir[0]][_x + _dir[1]] == '.':
                        lines[_y][_x] = '.'
                        lines[_y + _dir[0]][_x + _dir[1]] = 'O'
                        _y += _dir[0]
                        _x += _dir[1]
                x += 1
            y += 1
    else:
        y = y_max
        for line in reversed(lines):
            x = x_max
            for char in reversed(line):
                if char not in "#.":
                    _y = y
                    _x = x
                    while ((_y + _dir[0]) in y_range) and ((_x + _dir[1]) in x_range) and lines[_y + _dir[0]][_x + _dir[1]] == '.':
                        lines[_y][_x] = '.'
                        lines[_y + _dir[0]][_x + _dir[1]] = 'O'
                        _y += _dir[0]
                        _x += _dir[1]
                x -= 1
            y -= 1

def cycle():
    tilt([-1, 0])
    tilt([0, -1])
    tilt([1, 0])
    tilt([0, 1])

def print_map():
    for line in lines:
        string = ""
        for char in line:
            string += char
        print(string)
    print("\n")

def print_index(i):
    for line in states[i]:
        string = ""
        for char in line:
            string += char
        print(string)
    print("\n")

def copy_2d_list(_list):
    new = []
    for sub_list in _list:
        new.append(sub_list.copy())
    return new

#save the current data for use in part 2
lines2 = copy_2d_list(lines)

start = time.time()
tilt([-1, 0])
result = 0
distance = len(lines)
for line in lines:
    for char in line:
        if char == "O": result += distance
    distance -= 1
end = time.time()
time_needed = end-start
print(f"Part 1:\n Time needed: {round(time_needed, 3)}s\n Load: {result}\n")

#load the saved data
lines = copy_2d_list(lines2)

start = time.time()

cycles_max = 1000000000
states = []
cycles = 0

while True:
    cycle()
    if lines in states:
        cycle_start = states.index(lines)
        cycle_end = cycles
        cycle_length = cycle_end - cycle_start
        break
    states.append(copy_2d_list(lines))
    cycles += 1

left_over = (cycles_max - cycle_start) % cycle_length
lines = states[cycle_start + left_over-1]

result = 0
distance = len(lines)
for line in lines:
    for char in line:
        if char == "O": result += distance
    distance -= 1

end = time.time()
time_needed = end-start
input(f"Part 2:\n Time needed: {round(time_needed, 3)}s\n Load: {result}\n")