ledbar/host_python/cellular.py

76 lines
2 KiB
Python

#!/usr/bin/python
"""
An elementary 2D celluar autonoma implementation for the ledbar in brmlab.
For some fun rules, try:
30: near-random behavior
22: gives a symmetric triangle pattern. It just looks like splitting cells and gets
empty quickly, though.
142: neat waves
73: provides a downwards pattern with some fixed columns.
51: likes to blink.
The color mode encodes individual bits into, well, colors. Not too exciting,
but sure more colorful.
You can choose between a single pixel or a random starting row.
Possible fun stuff: Automatically pick new rules, detect patterns and restart.
Also, implement grayscare totalistic autonoma.
"""
import sys
import random
from ledbar import Ledbar
PIXELS = 20
PIXEL_MODE = ('bw', 'color')[0]
START = ('single', 'random')[0]
RULE = 30
WIDTH = PIXELS
if PIXEL_MODE == 'color': WIDTH *= 3
def bits(num, align=8):
for i in range(align)[::-1]:
yield bool(num & (1 << i))
rules = dict(zip(((1,1,1), (1,1,0), (1,0,1), (1,0,0), (0,1,1), (0,1,0), (0,0,1), (0,0,0)), bits(RULE)))
iteration = [0]*WIDTH
if START == 'single':
iteration[WIDTH//2] = 1
elif START == 'random':
iteration = list(random.randint(0, 1) for i in iteration)
def iterate(iteration):
new = []
iteration.insert(0, 0)
iteration.append(0)
for i in xrange(len(iteration)):
if 0 < i < len(iteration)-1:
top = (iteration[i-1], iteration[i], iteration[i+1])
new.append(rules[top])
else:
new.append(0)
return new
def update(i):
visible = iteration[(len(iteration)//2)-(WIDTH//2):(len(iteration)//2)+(WIDTH//2)]
if PIXEL_MODE == 'bw':
return (visible[i], visible[i], visible[i])
elif PIXEL_MODE == 'color':
return (visible[3*i], visible[3*i+1], visible[3*i+2])
l = Ledbar(PIXELS)
work = True
t = 0
while work:
for i in xrange(PIXELS):
c = update(i)
l.set_pixel(i, c[0], c[1], c[2])
work = l.update()
t += 1
if not (t % 50):
iteration = iterate(iteration)