refactor: rewrote examples/games/game_of_life.ark to use more stdlib constructs

Author: SuperFola

examples/games/game_of_life.ark

@@ -1,75 +1,68 @@
-(let board [0 0 0 0 0 1 1 1 1 1 1 0 0 0 0 0])
-(let width 4)
-(let height 4)
+(import std.List)
+(import std.String)
+(import std.Switch)
+
+(let board [0 0 0 1 1 1 0 0 0])
+(let width 3)
+(let height 3)
 (let dead 0)
 (let alive 1)
 
-(let get (fun (board_ i width height)
-  (if (and (>= i 0) (< i (* width height)))
-    (@ board_ i)
+(let get (fun (board_ x y)
+  (if (and (>= x 0) (>= y 0) (< x width) (< y height))
+    (@ board_ (+ x (* y width)))
     dead)))
 
-(let neigh (fun (board_ index width height) {
+(let neigh (fun (board_ index) {
   (let x (math:floor (mod index width)))
   (let y (math:floor (/ index width)))
-  (mut count 0)
-
-  (if (>= (- y 1) 0)
-    (set count (+ count (get board_ (- index width) width height))))
-
-  (if (< (+ y 1) height)
-    (set count (+ count (get board_ (+ index width) width height))))
-
-  (if (>= (- x 1) 0)
-    (set count (+ count (get board_ (- index 1) width height))))
-
-  (if (< (+ x 1) width)
-    (set count (+ count (get board_ (+ index 1) width height))))
-
-  (if (and (>= (- x 1) 0) (>= (- y 1) 0))
-    (set count (+ count (get board_ (- index 1 width) width height))))
-
-  (if (and (< (+ x 1) width) (< (+ y 1) height))
-    (set count (+ count (get board_ (+ index width 1) width height))))
-
-  (if (and (>= (- x 1) 0) (< (+ y 1) height))
-    (set count (+ count (get board_ (+ index width -1) width height))))
-
-  (if (and (< (+ x 1) width) (>= (- y 1) 0))
-    (set count (+ count (get board_ (- index width -1) width height))))
-  count }))
+  (+
+    (get board_ (- x 1) y)
+    (get board_ (+ x 1) y)
+    (get board_ x (- y 1))
+    (get board_ x (+ y 1))
+    (get board_ (- x 1) (- y 1))
+    (get board_ (+ x 1) (- y 1))
+    (get board_ (- x 1) (+ y 1))
+    (get board_ (+ x 1) (+ y 1))) }))
+
+(let indices (list:iota 0 (* height width)))
+(let update (fun (board) {
+  (mut copy (list:fill (* height width) dead))
+
+  (list:forEach indices
+    (fun (i) {
+      (let neighs (neigh board i))
+      (switch neighs
+        2 (@= copy i (@ board i))
+        3 (@= copy i alive)
+        _ (@= copy i dead)) }))
+
+  copy }))
+
+(let display (fun (board width height)
+  (print (string:join
+    (list:map
+      (list:chunkBy
+        (list:map board
+          (fun (cell)
+            (if (= alive cell) "x" "•")))
+        3)
+      (fun (sublist)
+        (string:join sublist "")))
+    "\n"))))
 
-(mut copy (list:fill (* height width) dead))
-(mut i 0)
-(while (< i (* width height)) {
-  (mut neighs (neigh board i width height))
-
-  (if (= 3 neighs)
-    (set copy (list:setAt copy i alive)))
-
-  (if (= 2 neighs)
-    (set copy (list:setAt copy i (@ board i))))
-
-  (if (or (< neighs 2) (> neighs 3))
-    (set copy (list:setAt copy i dead)))
-  (set i (+ 1 i)) })
-
-(let display (fun (board width height) {
-  (mut i 0)
-
-  (while (< i (* width height)) {
-    (mut y (math:floor (/ i width)))
-    (mut x (math:floor (mod i width)))
+(print "initial board:")
+(display board width height)
 
-    (if (= 0 x) (puts "\n"))
+(print "\ngen 2:")
+(mut new (update board))
+(display new width height)
 
-    (if (= alive (@ board i))
-      (puts "x")
-      (puts " "))
-    (set i (+ 1 i)) })
-  (puts "\n") }))
+(print "\ngen 3:")
+(set new (update new))
+(display new width height)
 
-(print "initial board:")
-(display board width height)
-(print "new board:")
-(display copy width height)
+(print "\ngen 4:")
+(set new (update new))
+(display new width height)