Spaces:
Sleeping
Sleeping
File size: 1,875 Bytes
3fd98da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
from time import sleep
from playsound import playsound
from frontend import Frontend
from leds import LEDs
FRONTEND_CONFIG = [
0x3E, # Overwrite ID?
0x95, # Datarate = 500 SPS
0xC0, # No tests
0xE1, # Power-down reference buffer, no bias
0x00, # No lead-off
0x60, # Channel 1 active, 24 gain, no SRB2 & normal input
0x60, # Channel 2 active, 24 gain, no SRB2 & normal input
0x60, # Channel 3 active, 24 gain, no SRB2 & normal input
0x60, # Channel 4 active, 24 gain, no SRB2 & normal input
0x60, # Channel 5 active, 24 gain, no SRB2 & normal input
0x60, # Channel 6 active, 24 gain, no SRB2 & normal input
0x60, # Channel 7 active, 24 gain, no SRB2 & normal input
0xE0, # Channel 8 disabled, 24 gain, no SRB2 & normal input
0x00, # No bias
0x00, # No bias
0xFF, # Lead-off on all positive pins?
0xFF, # Lead-off on all negative pins?
0x00, # Normal lead-off
0x00, # Lead-off positive status (RO) ?
0x01, # Lead-off negative status (RO) ?
0x00, # All GPIOs as output ???
0x20, # Disable SRB1
]
frontend = Frontend()
leds = LEDs()
playsound('sample.mp3')
try:
data = frontend.read_reg(0x00, 1)
assert data == [0x3E], "Wrong output"
print("EEG Frontend responsive")
print("Configuring EEG Frontend")
data = frontend.write_reg(0x00, FRONTEND_CONFIG)
print("EEG Frontend configured")
leds.aquisition(True)
sleep(0.5)
leds.aquisition(False)
sleep(0.5)
leds.aquisition(True)
for i in range(200):
red = (i % 10) * 10
blue = ((i % 100) // 10) * 10
leds.led1(red, 0, blue)
sleep(0.02)
for state in [RED, BLUE, PURPLE, CLOSED] * 3:
leds.led2(state)
sleep(0.2)
for state in [RED, CLOSED] * 3:
leds.led3(state)
sleep(0.2)
finally:
frontend.close()
leds.close()
|