diff options
Diffstat (limited to 'swift/life/Sources')
-rw-r--r-- | swift/life/Sources/main.swift | 53 |
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 +} |