From 7c0ebe4c50f995427238e8cd5e3a0f13cde83099 Mon Sep 17 00:00:00 2001
From: Caleb Webber <caleb@codingthemsoftly.com>
Date: Sat, 11 Nov 2023 21:12:19 -0500
Subject: [PATCH] day 10 part 2

---
 lib/day10.ex | 44 +++++++++++++++++++++++++++++++++++++++-----
 main.exs     |  2 +-
 2 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/lib/day10.ex b/lib/day10.ex
index 9459950..8ff711b 100644
--- a/lib/day10.ex
+++ b/lib/day10.ex
@@ -3,15 +3,19 @@ import Integer, only: [mod: 2]
 defmodule Day10 do
   @behaviour Solution
 
-  def init_state(_) do
-    %{ vm: VirtualMachine.new(), acc: 0 }
+  def init_state(p) when p == :part1 do
+    %{ vm: VirtualMachine.new(), acc: 0, p: p }
+  end
+
+  def init_state(p) when p == :part2 do
+    %{ vm: VirtualMachine.new(), crt: "", p: p}
   end
 
   def execute(line, state) when line == "" do
     state
   end
 
-  def execute(line, state) do
+  def execute(line, state) when state.p == :part1 do
     VirtualMachine.plan_execute(line) |>
     Enum.reduce(
       state,
@@ -21,12 +25,42 @@ defmodule Day10 do
         vm = tock.(vm)
         %{ vm: vm, acc: acc}
       end
-    )
+    ) |> Map.put(:p, state.p)
   end
 
-  def get_answer(state) do
+  def execute(line, state) when state.p == :part2 do
+    VirtualMachine.plan_execute(line) |>
+    Enum.reduce(
+      state,
+      fn (tock, %{vm: vm, crt: crt}) ->
+        next_tick = vm.tick + 1
+
+        char = case mod(vm.tick, 40) do
+          x when x >= (vm.x - 1) and x <= (vm.x + 1) -> "#"
+          _ -> "."
+        end
+
+        crt = crt <> char
+
+        crt = crt <> if mod(next_tick, 40) == 0 do
+          "\n"
+        else ""
+        end
+
+        vm = tock.(vm)
+
+        %{ vm: vm, crt: crt}
+      end
+    ) |> Map.put(:p, state.p)
+  end
+
+  def get_answer(state) when state.p == :part1 do
     state.acc
   end
+
+  def get_answer(state) when state.p == :part2 do
+    state.crt
+  end
 end
 
 defmodule VirtualMachine do
diff --git a/main.exs b/main.exs
index 832bec1..6cb5124 100644
--- a/main.exs
+++ b/main.exs
@@ -1 +1 @@
-AOCRunner.run(Day10, :part1)
+AOCRunner.run(Day10, :part2)