mirror of
https://github.com/janet-lang/janet
synced 2024-11-28 11:09:54 +00:00
184fe31e0c
Similar to python generator, but with the same syntax as the loop macro.
47 lines
1.3 KiB
Plaintext
47 lines
1.3 KiB
Plaintext
# A game of life implementation
|
|
|
|
(def- window
|
|
(fora [x :range [-1 2]
|
|
y :range [-1 2]
|
|
:when (not (and (zero? x) (zero? y)))]
|
|
(tuple x y)))
|
|
|
|
(defn- neighbors
|
|
[[x y]]
|
|
(mapa (fn [[x1 y1]] (tuple (+ x x1) (+ y y1))) window))
|
|
|
|
(defn tick
|
|
"Get the next state in the Game Of Life."
|
|
[state]
|
|
(def cell-set (frequencies state))
|
|
(def neighbor-set (frequencies (mapcat neighbors state)))
|
|
(fora [coord :keys neighbor-set
|
|
:let [count (get neighbor-set coord)]
|
|
:when (or (= count 3) (and (get cell-set coord) (= count 2)))]
|
|
coord))
|
|
|
|
(defn draw
|
|
"Draw cells in the game of life from (x1, y1) to (x2, y2)"
|
|
[state x1 y1 x2 y2]
|
|
(def cellset @{})
|
|
(loop [cell :in state] (put cellset cell true))
|
|
(loop [:before (print "+" (string.repeat "--" (inc (- y2 y1))) "+")
|
|
:after (print "+" (string.repeat "--" (inc (- y2 y1))) "+")
|
|
x :range [x1 (+ 1 x2)]
|
|
:before (file.write stdout "|")
|
|
:after (file.write stdout "|\n")
|
|
y :range [y1 (+ 1 y2)]]
|
|
(file.write stdout (if (get cellset (tuple x y)) "X " ". ")))
|
|
(print))
|
|
|
|
#
|
|
# Run the example
|
|
#
|
|
|
|
(var *state* '[(0 0) (-1 0) (1 0) (1 1) (0 2)])
|
|
|
|
(loop [i :range [0 20]]
|
|
(print "generation " i)
|
|
(draw *state* -7 -7 7 7)
|
|
(:= *state* (tick *state*)))
|