52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import random
|
|
|
|
from cellular_automaton.cellular_automaton import CellularAutomaton, CellularAutomatonEvolver
|
|
from cellular_automaton.ca_rule import Rule
|
|
from cellular_automaton.ca_neighborhood import MooreNeighborhood, EdgeRule
|
|
from cellular_automaton.ca_display import PyGameFor2D
|
|
from cellular_automaton.ca_cell_state import CellState
|
|
from cellular_automaton.ca_grid import Grid
|
|
|
|
|
|
class TestRule(Rule):
|
|
def evolve_cell(self, cell, iteration_index):
|
|
active = False
|
|
neighbors = cell.neighbours
|
|
if cell.state is None:
|
|
rand = random.randrange(0, 101, 1)
|
|
if rand <= 99:
|
|
cell.state = MyStatus(0)
|
|
else:
|
|
cell.state = MyStatus(1)
|
|
cell.is_set_for_redraw = True
|
|
active = True
|
|
elif len(neighbors) == 8:
|
|
left_neighbour_state = neighbors[0].state.get_status_of_iteration(iteration_index - 1)
|
|
active = cell.state.set_status_of_iteration(left_neighbour_state, iteration_index)
|
|
if active:
|
|
cell.is_set_for_redraw = True
|
|
return active
|
|
|
|
|
|
class MyStatus(CellState):
|
|
def __init__(self, initial_state):
|
|
super().__init__([initial_state])
|
|
|
|
def get_state_draw_color(self, iteration):
|
|
red = 0
|
|
if self._state_slots[iteration % 2][0]:
|
|
red = 255
|
|
return [red, 0, 0]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
random.seed(1000)
|
|
rule = TestRule()
|
|
ca = CellularAutomaton(Grid([400, 400], MooreNeighborhood(EdgeRule.FIRST_AND_LAST_CELL_OF_DIMENSION_ARE_NEIGHBORS)),
|
|
rule)
|
|
ca_evolver = CellularAutomatonEvolver(2)
|
|
ca_window = PyGameFor2D([1000, 800], ca, ca_evolver, 5)
|
|
ca_window.main_loop()
|