Add wrapper for frontend reading

frontend.py

@@ -15,6 +15,36 @@ RDATA = 0x12
 RREG = 0x20
 WREG = 0x40
 
+class Reading:
+    def __init__(self, values: [int]):
+        print(values)
+        assert values[0] & 0xF0 == 0xC0, "Invalid readback"
+        self.loff_statp = (values[0] & 0x0F) << 4 | (values[1] & 0xF0) >> 4
+        self.loff_statn = (values[1] & 0x0F) << 4 | (values[2] & 0xF0) >> 4
+        self.gpios = values[2] & 0x0F
+
+        self._channels = [
+            (values[3 + i * 3] << 16) | (values[4 + i * 3] << 8) | values[5 + i * 3]
+            for i in range(8)
+        ]
+
+        print(self.loff_statp, self.loff_statn, self.gpios, self._channels)
+
+    def channels(self):
+        return self._channels
+
+    def gpio(idx: int):
+        assert 0 <= idx <= 3, "Invalid gpio index"
+        return (self.gpios >> idx) & 0x01 == 0x01
+
+    def loff_p(idx: int):
+        assert 0 <= idx <= 7, "Invalid loff index"
+        return (self.loff_statp >> idx) & 0x01 == 0x01
+
+    def loff_n(idx: int):
+        assert 0 <= idx <= 7, "Invalid loff index"
+        return (self.loff_statn >> idx) & 0x01 == 0x01
+
 class Frontend:
     def __init__(self):
         self.nrst = GPIO("/dev/gpiochip2", 9, "out")
@@ -50,9 +80,9 @@ class Frontend:
     def write_regs(self, start, values):
         self.dev.xfer([WREG | (start & 0x1F), (len(values) - 1) & 0x1F] + values)
 
-    def read(self, len):
-        values = self.dev.xfer([RDATA] + [0x00] * len)
-        return values[1:]
+    def read(self):
+        values = self.dev.xfer([RDATA] + [0x00] * 27)
+        return Reading(values[1:])
 
     def start_continuous(self):
         self.dev.xfer([RDATAC])
@@ -60,8 +90,9 @@ class Frontend:
     def stop_continuous(self):
         self.dev.xfer([SDATAC])
 
-    def read_continuous(self, len):
-        values = self.dev.xfer([0x00] * len)
+    def read_continuous(self):
+        values = self.dev.xfer([0x00] * 27)
+        return Reading(values)
 
     def close(self):
         self.dev.close()

prog.py

@@ -1,8 +1,11 @@
 from time import sleep
 from playsound import playsound
+import numpy as np
+import matplotlib as plt
+import os
 
 from frontend import Frontend
-from leds import LEDs
+from leds import LEDs, Color
 
 FRONTEND_CONFIG = [
     0x3E, # Overwrite ID?
@@ -32,16 +35,16 @@ FRONTEND_CONFIG = [
 frontend = Frontend()
 leds = LEDs()
 
-playsound('sample.mp3')
-
 try:
-    data = frontend.read_reg(0x00, 1)
+    data = frontend.read_regs(0x00, 1)
     assert data == [0x3E], "Wrong output"
     print("EEG Frontend responsive")
+    leds.led2(Color.BLUE)
 
     print("Configuring EEG Frontend")
-    data = frontend.write_reg(0x00, FRONTEND_CONFIG)
+    data = frontend.write_regs(0x00, FRONTEND_CONFIG)
     print("EEG Frontend configured")
+    leds.led2(Color.PURPLE)
 
     leds.aquisition(True)
     sleep(0.5)
@@ -49,19 +52,11 @@ try:
     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)
+    points = []
+    for x in range(2000):
+        sleep(0.001)
+        values = frontend.read()
+        print(values.channels())
 
 finally:
     frontend.close()