1 files changed, 100 insertions, 0 deletions
diff --git a/firmware/offset_test.py b/firmware/offset_test.py
new file mode 100644
index 0000000..5f87d20
--- /dev/null
+++ b/firmware/offset_test.py
@@ -0,0 +1,100 @@
+
+import time
+import statistics
+import sqlite3
+from olsndot import Olsndot, Driver
+from datetime import datetime
+
+from pyBusPirateLite import Buspirate
+
+AVERAGE_READINGS = 16
+
+if __name__ == '__main__':
+    import argparse
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('channels', type=str, help='olsndot channels to test, format: 0-3,5,7,8-10')
+    parser.add_argument('run_name')
+    parser.add_argument('olsndot_port', nargs='?', default='/dev/serial/by-id/usb-FTDI_FT232R_USB_UART_A50285BI-if00-port0')
+    parser.add_argument('buspirate_port', nargs='?', default='/dev/serial/by-id/usb-FTDI_FT232R_USB_UART_AD01W1RF-if00-port0') # FIXME
+    parser.add_argument('-d', '--database', nargs='?', default='results.sqlite3', help='sqlite3 database file to store results in')
+    parser.add_argument('-m', '--mac', type=int, nargs='?', default=0xDEBE10BB, help='olsndot MAC address')
+    args = parser.parse_args()
+
+    def parse_channels(channels):
+        for spec in channels.split(','):
+            if str.isnumeric(spec):
+                yield int(spec)
+            else:
+                low, high = spec.split('-')
+                yield from range(int(low), int(high)+1)
+    channels = list(parse_channels(args.channels))
+
+    db = sqlite3.connect(args.database)
+    db.execute("""
+        CREATE TABLE IF NOT EXISTS runs (
+                run_id INTEGER PRIMARY KEY,
+                name TEXT,
+                comment TEXT,
+                uut_mac TEXT, -- hex-string formatted 32-bit mac of the uut
+                timestamp REAL -- unix timestamp in fractional seconds
+                )""")
+    db.execute("""
+        CREATE TABLE IF NOT EXISTS measurements (
+                measurement_id INTEGER PRIMARY KEY,
+                run_id INTEGER,
+                channel INTEGER,
+                duty_cycle REAL, -- setpoint duty cycle as a float between 0.0 and 1.0
+                voltage REAL, -- volts
+                voltage_stdev REAL, -- volts
+                timestamp REAL, -- unix timestamp in fractional seconds
+                FOREIGN KEY (run_id) REFERENCES runs)""")
+
+    bp = Buspirate(args.buspirate_port)
+    bp.power_on = True
+
+    uut = Olsndot(args.mac)
+    d = Driver(args.olsndot_port, devices=[uut])
+
+    print('Connected to uut:', uut)
+
+    with db:
+        cur = db.cursor()
+        cur.execute('INSERT INTO runs(name, uut_mac, timestamp) VALUES (?, ?, ?)',
+                (args.run_name, args.mac, time.time()))
+        run_id = cur.lastrowid
+    print('Starting run {} at {:%y-%m-%d %H:%M:%S:%f}'.format(run_id, datetime.now()))
+    print('mac={:08x} channels={}'.format(args.mac, ','.join('{:02d}'.format(ch) for ch in channels)))
+    print('[measurement id] " " [hex setpoint value] "(" [float duty cycle] ")" " " [reading (V)]')
+
+    for ch in channels:
+        for i in range(-1, uut.nbits):
+            fb = [0]*uut.nchannels
+            if i == -1:
+                val = 0
+            else:
+                val = 1<<i
+            duty_cycle = val/(2**uut.nbits)
+            extra_shift = 16-uut.nbits
+            val <<= extra_shift
+
+            fb[ch] = val
+            uut.send_framebuf(fb)
+            
+            time.sleep(0.2)
+            readings = [ bp.adc_value for _ in range(AVERAGE_READINGS) ]
+            mean, stdev = statistics.mean(readings), statistics.stdev(readings)
+
+            with db:
+                cur = db.cursor()
+                cur.execute('''
+                    INSERT INTO measurements (
+                            run_id, channel, duty_cycle, voltage, voltage_stdev, timestamp
+                        ) VALUES (?, ?, ?, ?, ?, ?)''',
+                        (run_id, ch, duty_cycle, mean, stdev, time.time()))
+                print('{:08d} ch={} {:04x}({:6.5f}): {:5.4f} stdev {:5.4}'.format(
+                    cur.lastrowid, ch, val, duty_cycle, mean, stdev))
+
+    uut.send_framebuf([0]*uut.nchannels)
+    bp.power_on = False
+