Merge branch 'master' of git://github.com/acm-uiuc/chroma-scripts

Author: root

animations/boxy/main.py

@@ -3,25 +3,29 @@
 import random
 import math
 import time
-from animations import FadeAnimation 
+from animations import FadeAnimation
 
-#author: Harsh Singh
+# author: Harsh Singh
+light_count = 48
 nice_pixels = [
-    (1023.0,0.0,0.0),
-    (0.0,1023.0,0.0),
-    (0.0,0.0,1023.0),
-    (0.0,1023.0,1023.0),
-    (1023.0,0.,1023.0),
-    (1023.0,1023.0,0.),    
-    (823.0,823.0,0.0),
-    (823.0,0.0,823.0),
-    (0.0,823.0,823.0),    
+    (1023.0, 0.0, 0.0),
+    (0.0, 1023.0, 0.0),
+    (0.0, 0.0, 1023.0),
+    (0.0, 1023.0, 1023.0),
+    (1023.0, 0., 1023.0),
+    (1023.0, 1023.0, 0.),
+    (823.0, 823.0, 0.0),
+    (823.0, 0.0, 823.0),
+    (0.0, 823.0, 823.0),
 ]
-bgColor=(0.0,0.0,0.0)
-template = [[0,1,4,5],[2,3,6,7],[8,9,12,13],[10,11,14,15],[16,17,20,21],[18,19,22,23],[0,1,2,3,4,7,8,11,12,13,14,15],[5,6,9,10],]
+bgColor = (0.0, 0.0, 0.0)
+template = [
+    [0, 1, 4, 5], [2, 3, 6, 7], [8, 9, 12, 13], [
+        10, 11, 14, 15], [16, 17, 20, 21],
+    [18, 19, 22, 23], [0, 1, 2, 3, 4, 7, 8, 11, 12, 13, 14, 15], [5, 6, 9, 10], ]
 out = FadeAnimation()
 out.start()
-layout = [bgColor]*24
+layout = [bgColor] * light_count
 
 
 def repeat(**options):
@@ -30,124 +34,130 @@ def repeat(**options):
     if (options.get("times") > 1):
         times = options.get("times")
     if (options.get("frames")):
-        while (n<times):
+        while (n < times):
             options.get("frames")
-            n+=1
+            n += 1
 
 # set color in box 1,2,3,4 with .4 second interval
+
+
 def frame1():
-    rand = random.sample(nice_pixels,5)
-    layout = [rand[4]]*24
+    rand = random.sample(nice_pixels, 5)
+    layout = [rand[4]] * light_count
     out.write(layout)
-    time.sleep(0.4)    
-    for i in [0,1,3,2]:
+    time.sleep(0.4)
+    for i in [0, 1, 3, 2]:
         for j in range(4):
             layout[template[i][j]] = rand[i]
-        #printlayout(layout)
+        # printlayout(layout)
         out.write(layout)
         time.sleep(0.4)
 
 
 # set color outer box, then inner box .2 delay
 def frame2():
-    rand = random.sample(nice_pixels,3)
-    layout = [rand[2]]*24
-    for i in [6,7]:
+    rand = random.sample(nice_pixels, 3)
+    layout = [rand[2]] * light_count
+    for i in [6, 7]:
         for j in range(len(template[i])):
-            layout[template[i][j]] = rand[i-6]
+            layout[template[i][j]] = rand[i - 6]
         out.write(layout)
         time.sleep(0.2)
 
 
-
-
 # spin colors at 5,6,9,10 at .1 second interval
 def frame3():
-    rand = random.sample(nice_pixels,5)
-    layout = [rand[4]]*24
-    #printlayout(layout)
-    #out.write(layout)
-    #time.sleep(0.4)  
-    i = 0  
+    rand = random.sample(nice_pixels, 5)
+    layout = [rand[4]] * light_count
+    # printlayout(layout)
+    # out.write(layout)
+    # time.sleep(0.4)
+    i = 0
     for j in range(len(template[i])):
 
-        #print "i= "+str(i)+" j ="+str(j)
+        # print "i= "+str(i)+" j ="+str(j)
         layout[template[7][j]] = rand[i]
         out.write(layout)
         time.sleep(0.1)
-        i+=1
-        #printlayout(layout)
-        
+        i += 1
+        # printlayout(layout)
+
 # spin colors at 5,6,9,10 at .1 second interval
+
+
 def frame3():
-    rand = random.sample(nice_pixels,5)
-    layout = [rand[4]]*24
-    #printlayout(layout)
-    #out.write(layout)
-    #time.sleep(0.4)  
-    i = 0  
+    rand = random.sample(nice_pixels, 5)
+    layout = [rand[4]] * light_count
+    # printlayout(layout)
+    # out.write(layout)
+    # time.sleep(0.4)
+    i = 0
     for j in range(len(template[i])):
 
-        #print "i= "+str(i)+" j ="+str(j)
+        # print "i= "+str(i)+" j ="+str(j)
         layout[template[7][j]] = rand[i]
         out.write(layout)
         time.sleep(0.1)
-        i+=1
-        #printlayout(layout)
+        i += 1
+        # printlayout(layout)
+
+# spiral function from 0 to 9
+
 
-#spiral function from 0 to 9
 def frame4():
-    path = [0,1,2,3,7,11,15,14,13,12,8,4,5,6,10,9]
-    
-    rand = random.sample(nice_pixels,4)
+    path = [0, 1, 2, 3, 7, 11, 15, 14, 13, 12, 8, 4, 5, 6, 10, 9]
+
+    rand = random.sample(nice_pixels, 4)
     for i in range(len(path)):
         layout[path[i]] = rand[0]
         if (i != 0):
-            layout[path[i-1]] = rand[1]
+            layout[path[i - 1]] = rand[1]
         # else:
         #     layout[path[len(path)-1]] = rand[1]
-        #printlayout(layout)    
+        # printlayout(layout)
         out.write(layout)
         time.sleep(0.05)
     for i in range(len(path)):
-        layout[path[len(path)-1-i]] = rand[2]
+        layout[path[len(path) - 1 - i]] = rand[2]
         if (i != 0):
-            layout[path[len(path)-i]] = rand[3]
+            layout[path[len(path) - i]] = rand[3]
         # else:
         #     layout[path[len(path)-1]] = rand[1]
-        #printlayout(layout)    
+        # printlayout(layout)
         out.write(layout)
         time.sleep(0.05)
 
+
 def color_picker():
-    r = random.random()*1000
+    r = random.random() * 1000
     b, g = (0., 0.)
     while True:
         if (b > 1000):
-            r = random.random()*1000
+            r = random.random() * 1000
             b = 0
-            g = random.random()*1000
+            g = random.random() * 1000
         if (g > 1000):
-            r = random.random()*1000
-            b = random.random()*1000
+            r = random.random() * 1000
+            b = random.random() * 1000
             g = 0
         if (r > 1000):
-            g = random.random()*1000
-            b = random.random()*1000
+            g = random.random() * 1000
+            b = random.random() * 1000
             r = 0
 
-        r+= 10.
-        b+= 10.
-        g+= 10.
-        layout = [(r,g,b)]*24
-        print (r,g,b)
+        r += 10.
+        b += 10.
+        g += 10.
+        layout = [(r, g, b)] * light_count
+        print (r, g, b)
         out.write(layout)
         time.sleep(.01)
 
 
 def printlayout(layout):
-    for i in range(0,5):
-        print (str(layout[i*4+0])+ " " +str(layout[i*4+1])+ " " +str(layout[i*4+2])+ " " +str(layout[i*4+3]))
+    for i in range(0, 5):
+        print (str(layout[i * 4 + 0]) + " " + str(layout[i * 4 + 1])
+               + " " + str(layout[i * 4 + 2]) + " " + str(layout[i * 4 + 3]))
     print
 
 if __name__ == "__main__":

animations/spirals/main.py

@@ -15,7 +15,8 @@
 bgColor=(0.0,0.0,0.0)
 out = FadeAnimation()
 out.start()
-layout = [bgColor]*24
+light_count = 48
+layout = [bgColor]* light_count
 
 """
 Spirals! Start from 0-3,3-15,15-12,12-4,4-6,6-10,10-9, then do it in reverse
@@ -43,7 +44,7 @@ def spirals():
 """
 
 def blinky():
-    layout = [bgColor]*24
+    layout = [bgColor]* light_count
 
     rand = random.sample(nice_pixels,2)
     for i in template[len(template)-2]:
@@ -55,11 +56,11 @@ def blinky():
     out.write(layout)
     time.sleep(0.3)
     temp = layout
-    layout = [bgColor]*24
+    layout = [bgColor]* light_count
     #recycle code later
     # for i in range(3):
     #     if (i%2 == 0):
-    #         layout = [bgColor]*24
+    #         layout = [bgColor]* light_count
     #     else :
     #         layout = temp
     #     out.write(layout)

animations/traffic/main.py

@@ -16,17 +16,18 @@
 scale = 1
 out = FadeAnimation()
 out.start()
+light_count = 48
 
 
 def frame1():
     scale = random.randint(1,4)
-    layout = [nice_pixels[0]]*24
+    layout = [nice_pixels[0]]*light_count
     out.write(layout)
     time.sleep(3*scale)
-    layout = [nice_pixels[1]]*24
+    layout = [nice_pixels[1]]*light_count
     out.write(layout)
     time.sleep(5*scale)
-    layout = [nice_pixels[2]]*24
+    layout = [nice_pixels[2]]*light_count
     out.write(layout)
     time.sleep(1*scale)
 

animations/well/main.py

@@ -6,37 +6,48 @@
 from animations import FadeAnimation 
 import operator
 
+#NOTE: This is some pretty shitty code i hacked up. Don't use it as an example, still a work in progress
+
 #author: Harsh Singh
-red = (1023.0,0.0,0.0)
-yello = (1023.0,1023.0,0)
+cool_trans = [[(1023.0,0.0,0.0),(1023.0,1023.0,0)], [(0.0,0.0,1023.0),(1023.0,0,1023.0)]]
 
 bgColor=(0.0,0.0,0.0)
 layout = [[bgColor for col in range(4)] for row in range(4)]
 out = FadeAnimation()
 out.start()
+times = 30
 
-def well():
+def well(col1, col2):
     global layout
-    times = 40
+
+    for a in xrange(times):
+        set_layout(col1, col2, a, times)
+        out.write(makeLayout(layout))
+        time.sleep(.05)
+    for a in xrange(times):
+        set_layout(col2, col1, a, times)
+        out.write(makeLayout(layout))
+        time.sleep(.05)
+
+def fade_out():
     for a in xrange(times):
         for x in xrange(4):
             for y in xrange(4):
-                if (0<x<3 and 0<y<3):
-                    layout[x][y] = transition_color(red, yello, a, times)
-                else:
-                    layout[x][y] = transition_color(yello, red, a, times)
+                layout[x][y] = transition_color(layout[x][y], bgColor, a, times)
         out.write(makeLayout(layout))
         time.sleep(.05)
+
+def fade_in():
+ 
     for a in xrange(times):
         for x in xrange(4):
             for y in xrange(4):
-                if (0<x<3 and 0<y<3):
-                    layout[x][y] = transition_color(yello, red, a, times)
-                else:
-                    layout[x][y] = transition_color(red, yello, a, times)
+                layout[x][y] = transition_color(bgColor, layout[x][y], a, times)
+        printLayout(layout)
         out.write(makeLayout(layout))
         time.sleep(.05)
 
+
 def transition_color(col1, col2, step, steps):
     step +=1
     col1 = tuple( [ int( e- (e * step/steps) ) for e in col1 ] )
@@ -55,7 +66,20 @@ def printLayout(lay):
         print str(lay[x][0])+" "+str(lay[x][1])+" "+str(lay[x][2])+" "+str(lay[x][3])
     print "---------------------------------"
 
+def set_layout(col1, col2, a, times):
+
+    for x in xrange(4):
+        for y in xrange(4):
+            if (0<x<3 and 0<y<3):
+                layout[x][y] = transition_color(col1, col2, a, times)
+            else:
+                layout[x][y] = transition_color(col2, col1, a, times)
 
 if __name__ == "__main__":
     while True:
-        well()
+        # well(cool_trans[0][0], cool_trans[0][1])
+        # fade_out()
+        # set_layout(cool_trans[1][0], cool_trans[1][1], 4, 4)
+        # fade_in()
+        # well(cool_trans[1][0], cool_trans[1][1])
+        well()