added fade to leds

lighting.py

@@ -8,19 +8,27 @@ import zmq
 import spidev
 import ws2812
 
+activeFlag = True
+c = 0
+rand = 1
+r = 255
 
 def get_color():
- rand = randint(0, 10)
- if rand < 5:
- return (248, 255, 255)
- elif rand == 6:
- return (111, 255, 0)
- elif rand == 7:
- return (234, 255, 0)
- elif rand == 8:
- return (87, 255, 34)
+ if activeFlag == True :
+ global c
+ global r
+ global rand
+ if c+rand > 80:
+ rand = randint(1, 8)*(-1)
+ return(80, 255, 0)
+ elif c+rand < 0:
+ rand = randint(1, 8)
+ return(0, 255, 0)
+ else:
+ c = c + rand
+ return (c, 255, 0)
  else:
- return (0, 213, 0)
+ fade()
 
 
 def light_flame():
@@ -29,7 +37,14 @@ def light_flame():
 
  while True:
  ws2812.write2812(spi, [get_color() * 4])
- sleep(0.2)
+ sleep(0.001)
+
+def fade():
+ if r > 0:
+ r--
+ if c > 0:
+ c--
+ return (c, r, 0)
 
 
 def main():
@@ -58,6 +73,9 @@ def main():
  light_flame()
  elif messagedata == b"stage_1_separation":
  light_flame()
+ elif messagedata == b"stage_2_separation":
+ light_flame()
+ activeFlag = False
  else:
  print("Unhandled message")