conway-life-lab

1#!/usr/bin/env python3
2"""
3Conway's Game of Life - Terminal Implementation
4
5Run cellular automata with classic or custom rules.
6Usage: python life.py [pattern] [generations] [delay_ms]
7
8Patterns: random, glider, blinker, beacon, pulsar, gosper
9"""
10
11import os
12import sys
13import time
14import random
15
16# Classic patterns
17PATTERNS = {
18    'glider': [(0, 1), (1, 2), (2, 0), (2, 1), (2, 2)],
19    'blinker': [(1, 0), (1, 1), (1, 2)],
20    'beacon': [(0, 0), (0, 1), (1, 0), (2, 3), (3, 2), (3, 3)],
21    'toad': [(1, 0), (1, 1), (1, 2), (2, 1), (2, 2), (2, 3)],
22    'block': [(0, 0), (0, 1), (1, 0), (1, 1)],
23    'pulsar': [
24        (0, 2), (0, 3), (0, 4), (0, 8), (0, 9), (0, 10),
25        (2, 0), (3, 0), (4, 0), (2, 5), (3, 5), (4, 5),
26        (2, 7), (3, 7), (4, 7), (2, 12), (3, 12), (4, 12),
27        (5, 2), (5, 3), (5, 4), (5, 8), (5, 9), (5, 10),
28        (7, 2), (7, 3), (7, 4), (7, 8), (7, 9), (7, 10),
29        (8, 0), (9, 0), (10, 0), (8, 5), (9, 5), (10, 5),
30        (8, 7), (9, 7), (10, 7), (8, 12), (9, 12), (10, 12),
31        (12, 2), (12, 3), (12, 4), (12, 8), (12, 9), (12, 10),
32    ],
33    'gosper': [  # Gosper Glider Gun
34        (0, 24), (1, 22), (1, 24), (2, 12), (2, 13), (2, 20), (2, 21), (2, 34), (2, 35),
35        (3, 11), (3, 15), (3, 20), (3, 21), (3, 34), (3, 35), (4, 0), (4, 1), (4, 10),
36        (4, 16), (4, 20), (4, 21), (5, 0), (5, 1), (5, 10), (5, 14), (5, 16), (5, 17),
37        (5, 22), (5, 24), (6, 10), (6, 16), (6, 24), (7, 11), (7, 15), (8, 12), (8, 13),
38    ],
39    'lwss': [  # Lightweight Spaceship
40        (0, 1), (0, 4), (1, 0), (2, 0), (2, 4), (3, 0), (3, 1), (3, 2), (3, 3),
41    ],
42    'hwss': [  # Heavyweight Spaceship
43        (0, 3), (0, 4), (1, 1), (1, 6), (2, 0), (3, 0), (3, 6), (4, 0), (4, 1), (4, 2), (4, 3), (4, 4), (4, 5),
44    ],
45    'diehard': [  # Diehard pattern - disappears after 130 generations
46        (0, 6), (1, 0), (1, 1), (2, 1), (2, 5), (2, 6), (2, 7),
47    ],
48    'pentadecathlon': [  # Period-15 oscillator
49        (0, 1), (1, 1), (2, 0), (2, 2), (3, 1), (4, 1), (5, 1), (6, 1), (7, 0), (7, 2), (8, 1), (9, 1),
50    ],
51    'loafe': [  # Still life - Loaf
52        (0, 1), (0, 2), (1, 0), (1, 3), (2, 1), (2, 3), (3, 2),
53    ],
54    'boat': [  # Still life - Boat
55        (0, 0), (0, 1), (1, 0), (1, 2), (2, 1),
56    ],
57    'beehive': [  # Still life - Beehive
58        (0, 1), (0, 2), (1, 0), (1, 3), (2, 1), (2, 2),
59    ],
60    'tub': [  # Still life - Tub
61        (0, 1), (1, 0), (1, 2), (2, 1),
62    ],
63}
64
65
66class GameOfLife:
67    """Conway's Game of Life simulation."""
68    
69    def __init__(self, width: int = 40, height: int = 20):
70        self.width = width
71        self.height = height
72        self.cells: set[tuple[int, int]] = set()
73    
74    def set_pattern(self, pattern: str, offset_y: int = 5, offset_x: int = 5) -> None:
75        """Load a named pattern or random cells."""
76        self.cells.clear()
77        if pattern == 'random':
78            for y in range(self.height):
79                for x in range(self.width):
80                    if random.random() < 0.3:
81                        self.cells.add((y, x))
82        elif pattern in PATTERNS:
83            for y, x in PATTERNS[pattern]:
84                self.cells.add((y + offset_y, x + offset_x))
85        else:
86            print(f"Unknown pattern: {pattern}")
87            print(f"Available: {', '.join(PATTERNS.keys())}, random")
88            sys.exit(1)
89    
90    def count_neighbors(self, y: int, x: int) -> int:
91        """Count live neighbors for a cell."""
92        count = 0
93        for dy in [-1, 0, 1]:
94            for dx in [-1, 0, 1]:
95                if dy == 0 and dx == 0:
96                    continue
97                ny, nx = y + dy, x + dx
98                if (ny, nx) in self.cells:
99                    count += 1
100        return count
101    
102    def step(self) -> None:
103        """Advance one generation."""
104        new_cells: set[tuple[int, int]] = set()
105        
106        # Check all cells and their neighbors
107        candidates: set[tuple[int, int]] = set()
108        for y, x in self.cells:
109            for dy in [-1, 0, 1]:
110                for dx in [-1, 0, 1]:
111                    candidates.add((y + dy, x + dx))
112        
113        for y, x in candidates:
114            neighbors = self.count_neighbors(y, x)
115            if (y, x) in self.cells:
116                # Survive with 2-3 neighbors
117                if neighbors in (2, 3):
118                    new_cells.add((y, x))
119            else:
120                # Birth with exactly 3 neighbors
121                if neighbors == 3:
122                    new_cells.add((y, x))
123        
124        self.cells = new_cells
125    
126    def render(self) -> str:
127        """Render current state as ASCII."""
128        lines = []
129        border = '+' + '-' * self.width + '+'
130        lines.append(border)
131        for y in range(self.height):
132            row = '|'
133            for x in range(self.width):
134                row += '█' if (y, x) in self.cells else ' '
135            row += '|'
136            lines.append(row)
137        lines.append(border)
138        return '\n'.join(lines)
139    
140    def run(self, generations: int = 100, delay_ms: int = 100) -> None:
141        """Run simulation with terminal animation."""
142        for gen in range(generations):
143            # Clear screen and move cursor to top
144            print('\033[H\033[J', end='')
145            print(f"Generation {gen + 1}/{generations} | Cells: {len(self.cells)}")
146            print(self.render())
147            
148            if len(self.cells) == 0:
149                print("All cells died. Simulation ended.")
150                break
151            
152            time.sleep(delay_ms / 1000)
153            self.step()
154
155
156def main():
157    pattern = sys.argv[1] if len(sys.argv) > 1 else 'glider'
158    generations = int(sys.argv[2]) if len(sys.argv) > 2 else 50
159    delay_ms = int(sys.argv[3]) if len(sys.argv) > 3 else 150
160    
161    # Adjust size for larger patterns
162    width, height = 40, 20
163    if pattern == 'gosper':
164        width, height = 50, 25
165    elif pattern == 'pulsar':
166        width, height = 30, 20
167    
168    game = GameOfLife(width, height)
169    game.set_pattern(pattern)
170    
171    print(f"Starting Conway's Game of Life")
172    print(f"Pattern: {pattern} | Generations: {generations} | Delay: {delay_ms}ms")
173    time.sleep(1)
174    
175    game.run(generations, delay_ms)
176    print("\nSimulation complete. Press Ctrl+C to exit.")
177
178
179if __name__ == '__main__':
180    main()
181