pytec: add configuration getters

pytec/plot.py

@@ -6,15 +6,19 @@ from pytec.client import Client
 
 TIME_WINDOW = 300.0
 
+tec = Client()
+target_temperature = tec.get_pid()[0]['target']
+print("Channel 0 target temperature: {:.3f}".format(target_temperature))
+
 class Series:
-    def __init__(self, scale=1.0):
+    def __init__(self, conv=lambda x: x):
-        self.scale = scale
+        self.conv = conv
         self.x_data = []
         self.y_data = []
 
     def append(self, x, y):
         self.x_data.append(x)
-        self.y_data.append(self.scale * y)
+        self.y_data.append(self.conv(y))
 
     def clip(self, min_x):
         drop = 0
@@ -25,8 +29,8 @@ class Series:
         
 series = {
     'adc': Series(),
-    'sens': Series(0.0001),
+    'sens': Series(lambda x: x * 0.0001),
-    'temperature': Series(),
+    'temperature': Series(lambda t: t - target_temperature),
     'i_set': Series(),
     'pid_output': Series(),
     'vref': Series(),
@@ -55,7 +59,6 @@ def recv_data(tec):
         if quit:
             break
 
-tec = Client()
 thread = Thread(target=recv_data, args=(tec,))
 thread.start()
 

pytec/pytec/client.py

@@ -1,6 +1,8 @@
 import socket
 import json
 
+CHANNELS = 2
+
 class Client:
     def __init__(self, host="192.168.1.26", port=23, timeout=None):
         self._socket = socket.create_connection((host, port), timeout)
@@ -22,6 +24,27 @@ class Client:
         self._lines = self._lines[1:]
         return line
 
+    def _get_conf(self, topic):
+        self._command(topic)
+        result = []
+        for channel in range(0, CHANNELS):
+            line = self._read_line()
+            conf = json.loads(line)
+            result.append(conf)
+        return result
+
+    def get_pwm(self):
+        return self._get_conf("pwm")
+
+    def get_pid(self):
+        return self._get_conf("pid")
+
+    def get_steinhart_hart(self):
+        return self._get_conf("s-h")
+
+    def get_postfilter(self):
+        return self._get_conf("postfilter")
+
     def report_mode(self):
         """Start reporting measurement values