Christopher Stone · Christopher Stone
Showing 1 changed file Show diff stats
robots/little_john/telemetry/code/monitor/graph_plotter_rewrite.py
-#!/usr/bin/env python
+#!/usr/bin/env python2
  
 # Tool to draw live graphs of data coming in from serial port
 # Written as a telemetry tool by:
@@ -11,17 +11,58 @@ import pyglet
 #import time
 import serial
 import numpy
+import os
  
 from colours import *
  
-datafeed = serial.Serial(
-port='/dev/ttyUSB0',
-baudrate = 9600,
-parity=serial.PARITY_NONE,
-stopbits=serial.STOPBITS_ONE,
-bytesize=serial.EIGHTBITS,
-timeout=1
-)
+devpatterns = ['ttyACM', 'ttyUSB', 'rfcomm']
+
+alldevs = os.listdir("/dev/")
+targetdevs = []
+for dev in alldevs:
+    for pattern in devpatterns:
+        if pattern in dev:
+            targetdevs.append(dev)
+            
+
+if len(targetdevs) == 0:
+    print("Sorry, no serial devices found.")
+    print("Exiting...")
+elif len(targetdevs) > 1:
+    print("Found multiple serial devices: ")
+    for i, dev in enumerate(targetdevs):
+        print("  " + str(i) + ": /dev/" + dev)
+    while True:
+        try:
+            selection = int(input("Please enter your selection (as a digit):\n > "))
+            if selection >= 0 and selection < len(targetdevs):
+                break
+        except KeyboardInterrupt:
+            exit()
+        except:
+            print("Choice unrecognised: please try again:")
+else:
+    print("Only found one likely serial device: /dev/" + dev)
+    selection = 0
+
+serialport = "/dev/" + targetdevs[selection]
+
+try:
+    datafeed = serial.Serial(
+    port=serialport,
+    baudrate = 9600,
+    parity=serial.PARITY_NONE,
+    stopbits=serial.STOPBITS_ONE,
+    bytesize=serial.EIGHTBITS,
+    timeout=1
+    )
+
+    print("Sucessfully opened " + serialport + " as data source!")
+except Exception, e:
+    print("Failed to access " + serialport + " because:")
+    print(e)
+    print("Exiting...")
+    exit()
  
 class Series:
     def __init__(self, points=100, title="Series title", xname="x-axis name", yname="y-axis name"):
@@ -154,4 +195,4 @@ def pollSerial(elapsed):
 # Pyglet looks after the main event loop, but this ensures that data keeps being read in
 pyglet.clock.schedule_interval(pollSerial, 0.1)
  
-pyglet.app.run()
 \ No newline at end of file
+pyglet.app.run()