from mindwave import BluetoothHeadset
def GetColorsForAttentionLevel(attention):
redlevel = int((point.attention - 33) * 255 / 33.0)
redlevel = max(min(redlevel, 255), 0)
bluelevel = 255 - redlevel
return redlevel, 0, bluelevel
led_strip = LedStrip("/dev/spidev0.0", 20)
# Set lights to a soft white to indicate the program is starting,
# but not reading your mind yet.
led_strip.setAll((64, 64, 64))
led_strip.update()
headset = BluetoothHeadset()
while True:
point = headset.readDatapoint()
if not point or not point.clean():
print(
"Not getting clean data yet. If you see this at startup, wait ~20s. "
"If it persists, adjust the headset for a better fit.")
led_strip.setAll((64, 64, 64))
led_strip.update()
continue
print "Attention:", point.attention
r, g, b = GetColorsForAttentionLevel(point.attention)
print "Coloring LEDS to RGB:", (r, g, b)
led_strip.setAll((g, b, r)) # LED strip uses funky ordering
led_strip.update()
