day8-ec.py

gen_screen = (True, 50, 6)

inp = []
if gen_screen[0]:
    import random
    inp = [''.join([random.choice(list('#..')) for i in range(gen_screen[1])]) for j in range(gen_screen[2])]
else:
    inp = open('screen.in').read().split('\n')

screen = [[y == '#' for y in row] for row in inp]
W, H = (len(screen[0]), len(screen))

def pscreen():
    print '\n'.join(''.join('#' if a else '.' for a in row) for row in screen)
    
print sum(sum(screen, []))  
pscreen()

cmds = []

for row in reversed(screen):
    for i in range(W):
        cmds.append("rotate column x=" + str(i) + " by 1")
    for blip in reversed(row):
        cmds.append("rotate row y=0 by 1")
        if blip:
            cmds.append("rect 1x1")

out = open('cmds.out', 'w').write('\n'.join(cmds))