day 08

08/part02.py

@@ -0,0 +1,65 @@
+#!/usr/bin/env python
+
+import re
+import argparse
+
+SIZE_X = 50
+SIZE_Y = 6
+
+RE_RECT = re.compile(r'rect (\d+)x(\d+)$')
+RE_ROTATE_C = re.compile(r'rotate column x=(\d+) by (\d+)$')
+RE_ROTATE_R = re.compile(r'rotate row y=(\d+) by (\d+)$')
+
+def load_file(filename):
+    with open(filename) as f:
+        for line in f:
+            yield line.strip()
+
+def do_rect(display, size_x, size_y):
+    for x in range(size_x):
+        for y in range(size_y):
+            display[y][x] = 'X'
+
+def do_rotate_row(display, y, p):
+    p = SIZE_X - p
+    display[y] = display[y][p:] + display[y][:p]
+
+def do_rotate_column(display, x, p):
+    data = [display[i][x] for i in range(SIZE_Y)]
+    p = SIZE_Y - p
+    data = data[p:] + data[:p]
+    for i in range(SIZE_Y):
+        display[i][x] = data[i]
+
+def main(args):
+    DISPLAY = []
+    for y in range(SIZE_Y):
+        line = []
+        for x in range(SIZE_X):
+            line.append(" ")
+        DISPLAY.append(line)
+
+    for LINE in load_file(args.input):
+        match = RE_RECT.search(LINE)
+        if match:
+            do_rect(DISPLAY, int(match.group(1)), int(match.group(2)))
+            continue
+        match = RE_ROTATE_C.search(LINE)
+        if match:
+            do_rotate_column(DISPLAY, int(match.group(1)), int(match.group(2)))    
+            continue
+        match = RE_ROTATE_R.search(LINE)
+        if match:
+            do_rotate_row(DISPLAY, int(match.group(1)), int(match.group(2)))
+            continue
+        raise Exception("unkown instruction: {}".format(LINE))
+
+    for y in range(SIZE_Y):
+        print("".join(DISPLAY[y]))
+ 
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='AdventOfCode 2016 Day 01')
+    parser.add_argument('--input', '-i', action='store',  required=True, help='input file')
+    args = parser.parse_args()
+
+    main(args)