#!/usr/bin/env python3
from functools import reduce
from re import findall
from copy import deepcopy
import sys
# import numpy as np

filename = sys.argv[1] if len(sys.argv) == 2 \
    else "in/" + sys.argv[0].split('/')[-1].rstrip(".py") + ".pzl"
data = open(filename).read()
blocks = data.rstrip('\n').split('\n\n')

res1 = 0
res2 = 0

S = []
for i, block in enumerate(blocks[:-1]):
    s = 0
    for line in block.split('\n')[1:]:
        for c in line:
            if c == '#':
                s += 1
    S.append(s)

print(S)

for line in blocks[-1].split('\n'):
    l, r = line.split(':')
    x, y = l.split('x')
    x, y = int(x), int(y)

    sh = [int(i) for i in r.strip().split(' ')]

    a = sum([i*j for i, j in zip(sh, S)])
    print(a / (x*y))



print('res1:', res1)
print('res2:', res2)