From 3fede9fba23c2841f61bcdc8072cbb8850e2effc Mon Sep 17 00:00:00 2001
From: Daniel Flanagan <daniel@lyte.dev>
Date: Tue, 21 Dec 2021 12:18:23 -0600
Subject: [PATCH] Ok, I think I figured out day 20s cleverness after many angry
 wrong guesses

---
 2021/nim/day20.nim | 84 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 2021/nim/day20.nim

diff --git a/2021/nim/day20.nim b/2021/nim/day20.nim
new file mode 100644
index 0000000..63fcf66
--- /dev/null
+++ b/2021/nim/day20.nim
@@ -0,0 +1,84 @@
+import ./common, std/[sequtils, algorithm, sugar, sets, strformat, strutils, tables, options, json, hashes]
+
+type
+  Point = (int, int)
+  Image = object
+    pixels: TableRef[Point, bool]
+    topLeft: Point
+    bottomRight: Point
+    infinite: bool
+
+proc parse(s: openArray[string]): Image =
+  var pixels = newTable[Point, bool]()
+  for y,l in s.pairs:
+    for x,c in l:
+      pixels[(x, y)] = c == '#'
+  return Image(topLeft: (0, 0), bottomRight: (s[0].len-1, s.len-1), pixels: pixels, infinite: false)
+
+proc enhance(image: var Image, algo: string) =
+  var (tlx, tly) = image.topLeft
+  var (brx, bry) = image.bottomRight
+  var newPixels = newTable[Point, bool]()
+  var xr = tlx-1..brx+1
+  var yr = tly-1..bry+1
+  for x in xr:
+    for y in yr:
+      var sample = 0
+      for sy in y-1..y+1:
+        for sx in x-1..x+1:
+          if image.pixels.getOrDefault((sx, sy), image.infinite):
+            inc sample
+          sample = sample shl 1
+      sample = sample shr 1
+      # echo &"{x}, {y} => {sample}"
+      newPixels[(x, y)] = algo[sample] == '#'
+      if x < tlx: tlx = x
+      if y < tly: tly = y
+      if x > brx: brx = x
+      if y > bry: bry = y
+  image.topLeft = (tlx, tly)
+  image.bottomRight = (brx, bry)
+  image.pixels = newPixels
+  if image.infinite:
+    image.infinite = algo[^1] == '#'
+  else:
+    image.infinite = algo[0] == '#'
+
+proc `$`(image: Image): string =
+  var s: seq[string] = @[]
+  var (tlx, tly) = image.topLeft
+  var (brx, bry) = image.bottomRight
+  for _ in tly..bry:
+    s.add ".".repeat((brx-tlx)+1)
+  for p in image.pixels.keys:
+    let (x, y) = p
+    if image.pixels[p]: s[y-tly][x-tlx] = '#'
+  s.join("\n")
+
+
+proc p1(input: Lines): uint64 =
+  var image = input[2..^1].parse()
+  echo &"{image.pixels.len} dim: {image.topLeft} {image.bottomRight}:"
+  echo image
+  image.enhance(input[0])
+  echo &"{image.pixels.len} dim: {image.topLeft} {image.bottomRight}:"
+  echo image
+  image.enhance(input[0])
+  echo &"{image.pixels.len} dim: {image.topLeft} {image.bottomRight}:"
+  echo image
+  for v in image.pixels.values:
+    if v: inc result
+
+proc p2(input: Lines): uint64 =
+  0
+
+const input = """
+..#.#..#####.#.#.#.###.##.....###.##.#..###.####..#####..#....#..#..##..###..######.###...####..#..#####..##..#.#####...##.#.#..#.##..#.#......#.###.######.###.####...#.##.##..#..#..#####.....#.#....###..#.##......#.....#..#..#..##..#...##.######.####.####.#.#...#.......#..#.#.#...####.##.#......#..#...##.#.##..#...##.#.##..###.#......#.#.......#.#.#.####.###.##...#.....####.#..#..#.##.#....##..#.####....##...##..#...#......#.#.......#.......##..####..#...#.#.#...##..#.#..###..#####........#..####......#..#
+
+#..#.
+#....
+##..#
+..#..
+..###
+""".strip().split('\n')
+doDayX 20, (n: int) => n.loadInput, p1, p2, (input, 35'u64, 0'u64)