#!/usr/bin/python3

import sys
import copy

cubemap={}
start_z=0
min_z=0
max_z=0
min_x=0
max_x=0
min_y=0
max_y=0

filename="p1_example.txt"
if len(sys.argv) > 1:
    filename=sys.argv[1]

def print_map():
    for z in [z for z in sorted([int(k) for k in cubemap.keys()])]:
        print("z = ",z)
        for y in [y for y in sorted([int(k) for k in cubemap[z].keys()])]:
            print("".join([cubemap[z][y][x] for x in sorted([int(x) for x in cubemap[z][y].keys()])]))
        print()

def get_active_neighbours(x,y,z):
    global cubemap
    sum=0
    for cur_z in range(z-1, z+2):
        for cur_y in range(y-1, y+2):
            for cur_x in range(x-1, x+2):
                if x == cur_x and y == cur_y and z == cur_z:
                    continue
                else:
                    if cur_z in cubemap:
                        if cur_y in cubemap[cur_z]:
                            if cur_x in cubemap[cur_z][cur_y]:
                                if cubemap[cur_z][cur_y][cur_x] == "#":
                                    sum+=1

    #print("Location: {},{},{}, Active neighbours: {}".format(x,y,z,sum))

    return sum

def do_iteration():
    global cubemap,min_x,max_x,min_y,max_y,min_z,max_z

    min_x-=1
    max_x+=1
    min_y-=1
    max_y+=1
    min_z-=1
    max_z+=1

    new_cubemap=copy.deepcopy(cubemap)

    for z in range(min_z, max_z+1):
        for y in range(min_y, max_y+1):
            for x in range(min_x, max_x+1):
                active_neighbours=get_active_neighbours(x,y,z)
                if not z in cubemap:
                    new_cubemap[z]={}
                    cubemap[z]={}
                if not y in cubemap[z]:
                    new_cubemap[z][y]={}
                    cubemap[z][y]={}
                if not x in cubemap[z][y]:
                    new_cubemap[z][y][x]="."
                    cubemap[z][y][x]="."
                if cubemap[z][y][x] == "#" and active_neighbours != 2 and active_neighbours != 3:
                    new_cubemap[z][y][x]="."
                elif cubemap[z][y][x] == "." and active_neighbours == 3:
                    new_cubemap[z][y][x]="#"

    cubemap=new_cubemap

# read the file
z=0
y=0
cubemap[z]={}
for line in open(filename, "r"):
    line=line.rstrip()
    cubemap[z][y]={}
    x=0
    for char in line:
        cubemap[z][y][x]=char
        x+=1
    max_x=x-1
    y+=1
max_y=y-1

print_map()

for iteration in range(0,6):
    print("Iteration:", iteration+1)
    do_iteration()
    #print_map()

active=0
for z in cubemap.keys():
    for y in cubemap[z].keys():
        for x in cubemap[z][y].keys():
            if cubemap[z][y][x]=="#":
                active+=1

print("Active cubes:", active)