Modularize & add untested main program

demo.py

@@ -1,115 +1,39 @@
-from periphery import GPIO, PWM, SPI, I2C
-from spidev import SpiDev
 from time import sleep
+from playsound import playsound
 
-led1_R = PWM(0, 0)
-led1_B = PWM(2, 0)
-led1_gnd = PWM(1, 0)
+from frontend import Frontend
+from leds import LEDs
 
-led2_R = GPIO("/dev/gpiochip0", 8, "out")
-led2_B = GPIO("/dev/gpiochip0", 6, "out")
-led2_gnd = GPIO("/dev/gpiochip0", 7, "out")
+frontend = Frontend()
+leds = LEDs()
 
-led3_R = GPIO("/dev/gpiochip2", 0, "out")
-#led3_B = GPIO("/dev/gpiochip2", 5, "out")
-led3_gnd = GPIO("/dev/gpiochip2", 8, "out")
-
-aquisition = GPIO("/dev/gpiochip2", 20, "out")
-
-frontend_nrst = GPIO("/dev/gpiochip2", 9, "out")
-frontend_npwdn = GPIO("/dev/gpiochip2", 12, "out")
-frontend_start = GPIO("/dev/gpiochip3", 29, "out")
-frontend_ = SpiDev()
-frontend_.open(0, 0)
-frontend_.max_speed_hz = 8000000
-frontend_.mode = 0b01
-
-audio = I2C("/dev/i2c-1")
-AUDIO_ADDR = 0x68
-
-aquisition.write(True)
-sleep(0.5)
-aquisition.write(False)
-sleep(0.5)
-aquisition.write(True)
+playsound('sample.mp3')
 
 try:
-  frontend_npwdn.write(True)
-  frontend_start.write(False)
-  sleep(0.1)
-  frontend_nrst.write(False)
-  sleep(0.01)
-  frontend_nrst.write(True)
-  sleep(0.1)
-  frontend_.xfer([0x11])
-
-  data_out = [0x20, 0x00, 0x00]
-  data_in = frontend_.xfer(list(data_out))
+    data = frontend.read_reg(0x00, 1)
+    assert data == [0x3E], "Wrong output"
+    print("EEG Frontend responsive")
+
+    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)
 
-  assert data_in[2] == 0x3E, "Wrong output"
-  print("EEG Frontend responsive")
-finally:
-  frontend_.close()
-
-try:
-  for reg in [0x0000, 0x0002, 0x0004, 0x0006, 0x000A, 0x000E, 0x0010, 0x0014, 0x0020, 0x0022, 0x0024, 0x0026, 0x0028, 0x002A, 0x002C, 0x002E, 0x0030, 0x0032, 0x0034, 0x0036, 0x0038, 0x003A, 0x003C, 0x0100, 0x0102, 0x0104, 0x0106, 0x0108, 0x010A, 0x010C, 0x010E, 0x0110, 0x0116]:
-    msgs = [I2C.Message([reg >> 8, reg & 0xFF]), I2C.Message([0x00, 0x00], read=True)]
-    audio.transfer(AUDIO_ADDR, msgs)
-    print("0x{:04x}: 0x{:02x} 0x{:02x}".format(reg, msgs[1].data[0], msgs[1].data[1]))
-finally:
-  audio.close()
-
-state = 0
-try:
-  led1_R.frequency = 1e3
-  led1_R.enable()
-  led1_B.frequency = 1e3
-  led1_B.enable()
-  led1_gnd.frequency = 1e3
-  led1_gnd.enable()
-  led1_gnd.duty_cycle = 0.0
-  for i in range(200):
-    led1_R.duty_cycle = (state % 10) / 10
-    led1_B.duty_cycle = (state // 10) / 10
-    state = (state + 1) % 100
-    sleep(0.02)
-finally:
-  led1_R.duty_cycle = 0.0
-  led1_R.close()
-  led1_B.duty_cycle = 0.0
-  led1_B.close()
-  led1_gnd.duty_cycle = 0.0
-  led1_gnd.close()
-
-state = 0
-try:
-  led2_gnd.write(False)
-  for i in range(12):
-    led2_R.write(state % 2 == 1)
-    led2_B.write(state // 2 == 1)
-    state = (state + 1) % 4
-    sleep(0.2)
-finally:
-  led2_R.write(False)
-  led2_R.close()
-  led2_B.write(False)
-  led2_B.close()
-  led2_gnd.write(False)
-  led2_gnd.close()
-
-state = 0
-try:
-  led3_gnd.write(False)
-  for i in range(12):
-    state = 3
-    led3_R.write(state % 2 == 1)
-    #led3_B.write(state // 2 == 1)
-    state = (state + 1) % 4
-    sleep(0.2)
 finally:
-  led3_R.write(False)
-  led3_R.close()
-  #led3_B.write(False)
-  #led3_B.close()
-  led3_gnd.write(False)
-  led3_gnd.close()
+    frontend.close()
+    leds.close()

frontend.py

@@ -0,0 +1,66 @@
+from periphery import GPIO
+from spidev import SpiDev
+
+WAKEUP = 0x02
+STANDBY = 0x04
+RESET = 0x06
+START = 0x08
+STOP = 0x0A
+
+RDATAC = 0x10
+SDATAC = 0x11
+RDATA = 0x12
+
+RREG = 0x20
+WREG = 0x40
+
+class Frontend:
+    def __init__():
+        self.nrst = GPIO("/dev/gpiochip2", 9, "out")
+        self.npwdn = GPIO("/dev/gpiochip2", 12, "out")
+        self.start = GPIO("/dev/gpiochip3", 29, "out")
+        self.dev = SpiDev()
+        self.dev.open(0, 0)
+        self.dev.max_speed_hz = 8000000
+        self.dev.mode = 0b01
+
+        self.start.write(False)
+        self.powerup()
+        self.reset()
+        self.stop_continuous()
+
+    def powerup():
+        self.pwdn.write(True)
+        sleep(0.1)
+
+    def powerdown():
+        self.pwdn.write(False)
+
+    def reset():
+        self.nrst.write(False)
+        sleep(0.01)
+        self.nrst.write(True)
+        sleep(0.1)
+
+    def read_regs(start, len):
+        values = self.dev.xfer([RREG | (start & 0x1F), (len - 1) & 0x1F] + [0x00] * len)
+        return values[2:]
+
+    def write_regs(start, values):
+        self.dev.xfer([WREG | (start & 0x1F), (len(values) - 1) & 0x1F] + values)
+
+    def read(len):
+        values = self.dev.xfer([RDATA] + [0x00] * len)
+        return values[1:]
+
+    def start_continuous():
+        self.dev.xfer([RDATAC])
+
+    def stop_continuous():
+        self.dev.xfer([SDATAC])
+
+    def read_continuous(len):
+        values = self.dev.xfer([0x00] * len)
+
+    def close():
+        self.dev.close()

leds.py

@@ -0,0 +1,103 @@
+from periphery import GPIO, PWM
+
+class LEDs:
+    def __init__():
+        self.led1_R = PWM(0, 0)
+        self.led1_B = PWM(2, 0)
+        self.led1_gnd = PWM(1, 0)
+
+        self.led2_R = GPIO("/dev/gpiochip0", 8, "out")
+        self.led2_B = GPIO("/dev/gpiochip0", 6, "out")
+        self.led2_gnd = GPIO("/dev/gpiochip0", 7, "out")
+
+        self.led3_R = GPIO("/dev/gpiochip2", 0, "out")
+        #self.led3_B = GPIO("/dev/gpiochip2", 5, "out")
+        self.led3_gnd = GPIO("/dev/gpiochip2", 8, "out")
+
+        self.aquisition = GPIO("/dev/gpiochip2", 20, "out")
+
+        # Init LEDs
+        # RGBs
+        self.led1_R.frequency = 1e3
+        self.led1_R.duty_cycle = 0.0
+        self.led1_R.enable()
+
+        self.led1_B.frequency = 1e3
+        self.led1_B.duty_cycle = 0.0
+        self.led1_B.enable()
+
+        self.led1_gnd.frequency = 1e3
+        self.led1_gnd.duty_cycle = 0.0
+        self.led1_gnd.enable()
+
+        # LED2
+        self.led2_R.write(False)
+        self.led2_B.write(False)
+        self.led2_gnd.write(False)
+
+        # LED3
+        self.led3_R.write(False)
+        #self.led3_B.write(False)
+        self.led3_gnd.write(False)
+
+    def aquisition(val):
+        self.aquisition.write(val)
+
+    # red, green & blue are between 0 and 100 inclusively
+    def led1(red, green, blue):
+        self.led1_R.duty_cycle = red / 100
+        self.led1_B.duty_cycle = blue / 100
+
+    def led2(value):
+        if value == RED:
+            self.led2_R.write(True)
+            self.led2_B.write(False)
+        elif value == BLUE:
+            self.led2_R.write(False)
+            self.led2_B.write(True)
+        elif value == PURPLE:
+            self.led2_R.write(True)
+            self.led2_B.write(True)
+        elif value == CLOSED:
+            self.led2_R.write(False)
+            self.led2_B.write(False)
+        else:
+            assert False, "Unknown color"
+
+    def led3(value):
+        if value == RED:
+            self.led3_R.write(True)
+        elif value == CLOSED:
+            self.led3_R.write(False)
+        else:
+            assert False, "Unknown color"
+
+    def close():
+        # LED1
+        self.led1_R.disable()
+        self.led1_B.disable()
+        self.led1_gnd.disable()
+        self.led1_R.close()
+        self.led1_B.close()
+        self.led1_gnd.close()
+
+        # LED2
+        self.led2_R.write(False)
+        self.led2_B.write(False)
+        self.led2_gnd.write(False)
+        self.led2_R.close()
+        self.led2_B.close()
+        self.led2_gnd.close()
+
+        # LED3
+        self.led3_R.write(False)
+        #self.led3_B.write(False)
+        self.led3_gnd.write(False)
+        self.led3_R.close()
+        #self.led3_B.close()
+        self.led3_gnd.close()
+
+        # AQUISITION
+        self.aquisition.write(False)
+        self.aquisition.close()
+

prog.py

@@ -0,0 +1,68 @@
+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()