cellular.py: Colors, starting conditions, any rule number

host_python/cellular.py

@@ -1,27 +1,55 @@
 #!/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
 
-rules = {(1,1,1): 0,
-    (1,1,0): 0,
-    (1,0,1): 0,
-    (1,0,0): 1,
-    (0,1,1): 1,
-    (0,1,0): 1,
-    (0,0,1): 1,
-    (0,0,0): 0}
+WIDTH = PIXELS
+if PIXEL_MODE == 'color': WIDTH *= 3
 
-iteration = [0]*PIXELS
-iteration[PIXELS/2] = 1
+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 = []
-    for i in range(len(iteration)):
-        if 0 < i < PIXELS-1:
+    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:
@@ -29,7 +57,11 @@ def iterate(iteration):
     return new
 
 def update(i):
-    return (iteration[i], iteration[i], iteration[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