Hugo Stephens · Hugo Stephens
Showing 6 changed files Show diff stats
robots/little_john/telemetry/code/monitor/version1/colours.py
robots/little_john/telemetry/code/monitor/version1/getdata.py
robots/little_john/telemetry/code/monitor/version1/main.py
robots/little_john/telemetry/code/monitor/version1/plot.py
robots/little_john/telemetry/code/monitor/version1/serialselect.py
robots/little_john/telemetry/code/monitor/version1/series.py
@@ -2,4 +2,4 @@ WHITE = (255, 255, 255, 255)
 BLACK = (0, 0, 0, 255)
 RED = (255, 0, 0, 255)
 GREEN = (0, 255, 0, 255)
-BLUE = (0, 0, 255, 255)
 \ No newline at end of file
+BLUE = (0, 0, 255, 255)
@@ -7,6 +7,7 @@ import os
 import time
 import math
+
 def getData(starttime, datafeed, testseries, testseries2, fake=False):
     """Check serial port for incoming data"""
     # Note: 'elapsed' is time since last call of this function
@@ -15,7 +16,7 @@ def getData(starttime, datafeed, testseries, testseries2, fake=False):
     if fake:
         timefromstart = (time.time()-starttime)
         values = [timefromstart, 100*math.sin(timefromstart)]
-        #logging.info("Generated test data: " + str(values))
+        # logging.info("Generated test data: " + str(values))
         testseries.addpoint(values)
         testseries2.addpoint(values)
     else:
@@ -23,9 +24,9 @@ def getData(starttime, datafeed, testseries, testseries2, fake=False):
             incoming = datafeed.readline()
         except:
             logging.error("Failed to read input data")
-            
+
         try:
-            if os=='Windows':
+            if os == 'Windows':
                 values = incoming.strip().split(b", ")
             else:
                 values = incoming.strip()
@@ -33,15 +34,13 @@ def getData(starttime, datafeed, testseries, testseries2, fake=False):
         except:
             logging.error("Failed to parse input data")
             return
-        
+
         try:
             for n, value in enumerate(values):
                 values[n] = float(value)
         except:
             logging.error("Failed to convert input to float")
             return
-        #logging.info("Recieved data: " + str(values))
+        # logging.info("Recieved data: " + str(values))
         testseries.addpoint([time.time()-starttime] + values)
         testseries2.addpoint(values)
-
-    
@@ -31,7 +31,7 @@ starttime = time.time()
 datafeed = selectserial()
-if datafeed == None:
+if datafeed is None:
     logging.critical("Failed to open serial port")
     sys.exit()
@@ -43,9 +43,11 @@ logging.info(&#39;OS = &#39; + os)
 testseries = Series(points=150, xauto=True, ylimits=(0, 1024),
-                    title="Data from serial (time)", xname="Time (s)", yname="ADC output")
+                    title="Data from serial (time)",
+                    xname="Time (s)", yname="ADC output")
 testseries2 = Series(points=150, xlimits=(0, 1024), ylimits=(0, 1024),
-                     title="Data from serial (xy)", xname="ADC0 output", yname="ADC7 output")
+                     title="Data from serial (xy)",
+                     xname="ADC0 output", yname="ADC7 output")
 plots = []
 plots.append(Plot(testseries))
@@ -8,24 +8,27 @@ import logging
 from colours import *
 import sys
+
 class Plot(pyglet.window.Window):
     def __init__(self, series):
         """Setup a the details of a plot, and create a corresponding window"""
         pyglet.window.Window.__init__(self, resizable=True)
         self.set_icon(pyglet.image.load('32x32.png'))
-        self.set_minimum_size(320,320)
+        self.set_minimum_size(320, 320)
         self.series = series
         self.font = 'Arkhip'
-        self.margins = (0.09, 0.08) # Fractions of window size
+        self.margins = (0.09, 0.08)  # Fractions of window size
         self.lines = (10, 8)
-        #self.resizable = True
+        # self.resizable = True
         self.set_caption(self.series.title)
     def on_resize(self, width, height):
         """Handle a resize event from the pyglet event loop"""
         try:
-            self.bounds = ((int(self.width * self.margins[0]), int(self.width * (1 - self.margins[0]))),
-                (int(self.height * self.margins[1]), int(self.height * (1 - self.margins[1]))))
+            self.bounds = ((int(self.width * self.margins[0]),
+                            int(self.width * (1 - self.margins[0]))),
+                           (int(self.height * self.margins[1]),
+                            int(self.height * (1 - self.margins[1]))))
         except Exception as e:
             logging.critical(str(e))
             self.close()
@@ -54,11 +57,13 @@ class Plot(pyglet.window.Window):
                 self.width, self.height).blit(0, 0)
     def drawHeading(self):
-        """Draw a title for the graph (duplicated in the window titlebar, if present"""
+        """Draw a title for the graph (duplicated in the window titlebar"""
         heading = pyglet.text.Label(self.series.title, color=BLACK,
-                            font_name=self.font, font_size=self.height*self.margins[0]*0.5,
-                            x=self.width/2, y=self.height-(self.margins[1]),
-                            anchor_x='center', anchor_y='top')
+                                    font_name=self.font,
+                                    font_size=self.height*self.margins[0]*0.5,
+                                    x=self.width/2,
+                                    y=self.height-(self.margins[1]),
+                                    anchor_x='center', anchor_y='top')
         heading.draw()
     def drawLines(self, series):
@@ -78,18 +83,20 @@ class Plot(pyglet.window.Window):
         try:
             for m in range(len(series.data[0])-1):
                 for n in range(len(series.data) - 1):
-                    x1, y1 = series.data[n][0]-series.xlimits[0], series.data[n][m+1]-series.ylimits[0]
-                    x2, y2 = series.data[n+1][0]-series.xlimits[0], series.data[n+1][m+1]-series.ylimits[0]
+                    x1 = series.data[n][0]-series.xlimits[0]
+                    y1 = series.data[n][m+1]-series.ylimits[0]
+                    x2 = series.data[n+1][0]-series.xlimits[0]
+                    y2 = series.data[n+1][m+1]-series.ylimits[0]
                     x1 = int((x1/xscale)+lmar)
                     y1 = int((y1/yscale)+bmar)
                     x2 = int((x2/xscale)+lmar)
                     y2 = int((y2/yscale)+bmar)
                     pyglet.graphics.draw(2, pyglet.gl.GL_LINES,
-                                        ('v2i', (x1, y1, x2, y2)),
-                                        ('c3B', linecolours[m]))
+                                         ('v2i', (x1, y1, x2, y2)),
+                                         ('c3B', linecolours[m]))
         except:
-            logging.error("Failed to plot lines, possibly due to corrupt/missing data. (This happens at startup)")
-            
+            logging.error("Failed to plot lines: corrupt/missing data?")
+
         pyglet.gl.glLineWidth(1)
     def drawAxis(self, axis):  # axis=0 is x, 1 is y
@@ -108,27 +115,34 @@ class Plot(pyglet.window.Window):
                         stop-start)
                 tagvalue = ((pos-start) * scale) + self.series.xlimits[0]
                 tagtext = str(int(tagvalue))
-                pyglet.graphics.draw(2, pyglet.gl.GL_LINES, ('v2i', (pos, limita, pos, limitb)),
-                                 ('c3B', (0, 0, 0, 0, 0, 0)))
+                pyglet.graphics.draw(2, pyglet.gl.GL_LINES,
+                                     ('v2i', (pos, limita, pos, limitb)),
+                                     ('c3B', (0, 0, 0, 0, 0, 0)))
                 tag = pyglet.text.Label(tagtext, color=BLACK,
-                                        font_name=self.font, font_size=self.tag_size,
+                                        font_name=self.font,
+                                        font_size=self.tag_size,
                                         x=pos, y=self.height*self.margins[1],
                                         anchor_x='left', anchor_y='top')
                 axistitle = pyglet.text.Label(self.series.xname, color=BLACK,
-                                                font_name=self.font, font_size=self.tag_size,
-                                                x=self.width/2, y=0,
-                                                anchor_x='center', anchor_y='bottom')
+                                              font_name=self.font,
+                                              font_size=self.tag_size,
+                                              x=self.width/2, y=0,
+                                              anchor_x='center',
+                                              anchor_y='bottom')
                 axistitle.draw()
             if axis == 1:  # y axis, horizontal lines
                 scale = float(self.series.ylimits[1]-self.series.ylimits[0])/(
                         stop-start)
                 tagvalue = ((pos-start) * scale) + self.series.ylimits[0]
                 tagtext = str(int(tagvalue))
-                pyglet.graphics.draw(2, pyglet.gl.GL_LINES, ('v2i', (limita, pos, limitb, pos)),
-                                 ('c3B', (0, 0, 0, 0, 0, 0)))
+                pyglet.graphics.draw(2, pyglet.gl.GL_LINES,
+                                     ('v2i', (limita, pos, limitb, pos)),
+                                     ('c3B', (0, 0, 0, 0, 0, 0)))
                 tag = pyglet.text.Label(tagtext, color=BLACK,
-                                        font_name=self.font, font_size=self.tag_size,
-                                        x=self.width*self.margins[0]*0.9, y=pos,
+                                        font_name=self.font,
+                                        font_size=self.tag_size,
+                                        x=self.width*self.margins[0]*0.9,
+                                        y=pos,
                                         anchor_x='right', anchor_y='center')
                 axistitle = pyglet.text.Label(self.series.yname, color=BLACK,
                                               font_name=self.font,
@@ -2,27 +2,29 @@
 # Written as a telemetry tool by:
 # The UoN Robot Wars Project, 2018
+
 def selectserial():
-    """Cross-platform function to find appropriate serial ports, query the user if necessary, and open one of them"""
+    """Cross-platform function to find appropriate serial ports, query the
+    user if necessary, and open one of them"""
     import platform
     import serial
     import os
     import easygui
     import logging
-    
+
     devpatterns = ['ttyACM', 'ttyUSB', 'rfcomm']
     targetdevs = []
-    if platform.system()=='Windows':
+    if platform.system() == 'Windows':
         com_ports = ['COM%s' % (i + 1) for i in range(256)]
         for port in com_ports:
             try:
                 s = serial.Serial(port)
                 s.close()
                 targetdevs.append(port)
-            # Temporarily broadened exception in attempt to make this work on uni PCs
-            except: # (OSError, serial.SerialException):
+            # Temporarily broadened exception to make this work on uni PCs
+            except:  # (OSError, serial.SerialException):
                 pass
-        os='Windows'    #may be useful
+        os = 'Windows'    # may be useful
     else:
         alldevs = os.listdir("/dev/")
         targetdevs = []
@@ -30,7 +32,7 @@ def selectserial():
             for pattern in devpatterns:
                 if pattern in dev:
                     targetdevs.append("/dev/" + dev)
-        os='Other'     #may be useful
+        os = 'Other'     # may be useful
     if len(targetdevs) == 0:
         logging.info("No serial device found.")
@@ -42,13 +44,13 @@ def selectserial():
         message = "Please choose a serial port to recieve data through:"
         title = "Found multiple serial ports!"
         serialport = easygui.choicebox(message, title, targetdevs)
-        if serialport == None:
+        if serialport is None:
             logging.info("User cancelled selection dialogue")
             return None
     else:
         logging.info("Only found one likely serial device: " + targetdevs[0])
         serialport = targetdevs[0]
-        
+
     try:
         datafeed = serial.Serial(
                 port=serialport,
@@ -4,10 +4,12 @@
 import logging
+
 class Series:
     def __init__(self, points=100, xlimits=(0, 100), ylimits=(0, 100),
                  xauto=False, yauto=False,
-                 title="Series title", xname="x-axis name", yname="y-axis name"):
+                 title="Series title",
+                 xname="x-axis name", yname="y-axis name"):
         """Set up an object to store a 2D data series"""
         # Proposal:
         #   In order to neatly handle multiple lines on the same graph
@@ -28,9 +30,9 @@ class Series:
         self.points = points
         self.xauto = xauto
         self.yauto = yauto
-        
+
         logging.info("Created series: " + title)
-        
+
     def addpoint(self, point):
         """Add a point to the dataset, and remove the oldest, if necessary"""
         self.data.append(point)
@@ -43,15 +45,15 @@ class Series:
                 self.autoscale(1)
         except:
             logging.error("Series autoscale failed")
-         
-    def autoscale(self, axis): # axis=0 is x, 1 is y
+
+    def autoscale(self, axis):  # axis=0 is x, 1 is y
         minval = self.data[0][axis]
         maxval = self.data[0][axis]
         for value in self.data:
             if value[axis] < minval:
                 minval = value[axis]
             if value[axis] > maxval:
-                maxval = value[axis] 
+                maxval = value[axis]
         if axis == 0:
             self.xlimits = (minval, maxval)
         else: