about summary refs log tree commit diff stats
path: root/swift/life/Sources
diff options
context:
space:
mode:
Diffstat (limited to 'swift/life/Sources')
-rw-r--r--swift/life/Sources/main.swift53
1 files changed, 53 insertions, 0 deletions
diff --git a/swift/life/Sources/main.swift b/swift/life/Sources/main.swift
new file mode 100644
index 0000000..4b3e7a6
--- /dev/null
+++ b/swift/life/Sources/main.swift
@@ -0,0 +1,53 @@
+import Foundation
+
+let rows = 10
+let cols = 10
+var board = Array(repeating: Array(repeating: 0, count: cols), count: rows)
+
+func printBoard(_ board: [[Int]]) {
+    board.forEach { row in
+        row.forEach { cell in
+            print(cell == 1 ? "@" : ".", terminator: "")
+        }
+        print()
+    }
+    print("\n")
+}
+
+func countLiveNeighbors(_ board: [[Int]], x: Int, y: Int) -> Int {
+    let directions = [(-1, -1), (-1, 0), (-1, 1), (0, -1), (0, 1), (1, -1), (1, 0), (1, 1)]
+    return directions.reduce(0) { count, dir in
+        let newX = x + dir.0
+        let newY = y + dir.1
+        if newX >= 0 && newX < rows && newY >= 0 && newY < cols {
+            return count + board[newX][newY]
+        }
+        return count
+    }
+}
+
+func nextGeneration(_ currentBoard: [[Int]]) -> [[Int]] {
+    return currentBoard.enumerated().map { (x, row) in
+        return row.enumerated().map { (y, cell) in
+            let liveNeighbors = countLiveNeighbors(currentBoard, x: x, y: y)
+            if cell == 1 && (liveNeighbors < 2 || liveNeighbors > 3) {
+                return 0
+            } else if cell == 0 && liveNeighbors == 3 {
+                return 1
+            }
+            return cell
+        }
+    }
+}
+
+board[1][2] = 1
+board[2][3] = 1
+board[3][1] = 1
+board[3][2] = 1
+board[3][3] = 1
+
+for _ in 0..<10 {
+    printBoard(board)
+    board = nextGeneration(board)
+    sleep(1) // FIXME: figure out how to sleep for periods shorter than a full second
+}