1 files changed, 53 insertions, 0 deletions

diff --git a/visualizer/visualizer.py b/visualizer/visualizer.py
new file mode 100644
index 0000000..ed65384
--- /dev/null
+++ b/visualizer/visualizer.py
@@ -0,0 +1,53 @@
+
+BINS = 128
+
+
+import serial
+
+s = serial.Serial("/dev/ttyACM0", 115200)
+
+
+def acquire_data():
+
+    nb_val = 0
+    while nb_val != BINS+1:
+        l = s.readline()
+        d = l.decode("utf8")
+        v = d.split(' ')
+        nb_val = len(v)
+
+    n = [0] * BINS
+    for i in range(BINS):
+        try:
+            n[i] = int(v[i])
+
+        except ValueError:
+            n[i] = 0
+
+    return n
+
+
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+fig = plt.figure()
+ax = fig.add_subplot(111)
+
+li, = ax.plot(range(BINS))
+plt.axis([0, BINS, 0, 250])
+
+fig.canvas.draw()
+plt.show(block=False)
+
+
+while True:
+    try:
+        y = acquire_data()
+        print(y)
+        li.set_ydata(y)
+        fig.canvas.draw()
+
+    except KeyboardInterrupt:
+        break
+