197 lines
8.2 KiB
Python
197 lines
8.2 KiB
Python
from pyqtgraph.Qt import QtGui, QtCore
|
|
import pyqtgraph.parametertree.parameterTypes as pTypes
|
|
from pyqtgraph.parametertree import Parameter, ParameterTree, ParameterItem, registerParameterType
|
|
import numpy as np
|
|
import pyqtgraph as pg
|
|
from pytec.client import Client
|
|
|
|
rec_len = 1000
|
|
refresh_period = 20
|
|
|
|
TECparams = [ [
|
|
{'tag': 'report', 'type': 'parent', 'children': [
|
|
{'tag': 'pid_engaged', 'type': 'bool', 'value': False},
|
|
]},
|
|
{'tag': 'pwm', 'type': 'parent', 'children': [
|
|
{'tag': 'max_i_pos', 'type': 'float', 'value': 0},
|
|
{'tag': 'max_i_neg', 'type': 'float', 'value': 0},
|
|
{'tag': 'max_v', 'type': 'float', 'value': 0},
|
|
{'tag': 'i_set', 'type': 'float', 'value': 0},
|
|
]},
|
|
{'tag': 'pid', 'type': 'parent', 'children': [
|
|
{'tag': 'kp', 'type': 'float', 'value': 0},
|
|
{'tag': 'ki', 'type': 'float', 'value': 0},
|
|
{'tag': 'kd', 'type': 'float', 'value': 0},
|
|
{'tag': 'output_min', 'type': 'float', 'value': 0},
|
|
{'tag': 'output_max', 'type': 'float', 'value': 0},
|
|
]},
|
|
{'tag': 's-h', 'type': 'parent', 'children': [
|
|
{'tag': 't0', 'type': 'float', 'value': 0},
|
|
{'tag': 'r0', 'type': 'float', 'value': 0},
|
|
{'tag': 'b', 'type': 'float', 'value': 0},
|
|
]},
|
|
{'tag': 'PIDtarget', 'type': 'parent', 'children': [
|
|
{'tag': 'target', 'type': 'float', 'value': 0},
|
|
]},
|
|
] for _ in range(2)]
|
|
|
|
|
|
GUIparams = [[
|
|
{'name': 'Enable Output', 'type': 'bool', 'value': False},
|
|
{'name': 'Enable Constant Current', 'type': 'bool', 'value': False, 'children': [
|
|
{'name': 'Set Current', 'type': 'float', 'value': 0, 'step': 0.1, 'siPrefix': True, 'suffix': 'A'},
|
|
]},
|
|
{'name': 'Enable PID', 'type': 'bool', 'value': False, 'children': [
|
|
{'name': 'Set Temperature', 'type': 'float', 'value': 25, 'step': 0.1, 'siPrefix': True, 'suffix': 'C'},
|
|
]},
|
|
{'name': 'Output Config', 'type': 'group', 'children': [
|
|
{'name': 'Max Current', 'type': 'float', 'value': 0, 'step': 0.1, 'siPrefix': True, 'suffix': 'A'},
|
|
{'name': 'Max Voltage', 'type': 'float', 'value': 0, 'step': 0.1, 'siPrefix': True, 'suffix': 'V'},
|
|
]},
|
|
{'name': 'Thermistor Config', 'type': 'group', 'children': [
|
|
{'name': 'T0', 'type': 'float', 'value': 25, 'step': 0.1, 'siPrefix': True, 'suffix': 'C'},
|
|
{'name': 'R0', 'type': 'float', 'value': 10000, 'step': 1, 'siPrefix': True, 'suffix': 'Ohm'},
|
|
{'name': 'Beta', 'type': 'float', 'value': 3950, 'step': 1},
|
|
]},
|
|
{'name': 'PID Config', 'type': 'group', 'children': [
|
|
{'name': 'kP', 'type': 'float', 'value': 0, 'step': 0.1},
|
|
{'name': 'kI', 'type': 'float', 'value': 0, 'step': 0.1},
|
|
{'name': 'kD', 'type': 'float', 'value': 0, 'step': 0.1},
|
|
]},
|
|
{'name': 'Save', 'type': 'action', 'tip': 'Save'},
|
|
] for ch in range(2)]
|
|
|
|
## If anything changes in the tree, print a message
|
|
def change(param, changes):
|
|
print("tree changes:")
|
|
for param, change, data in changes:
|
|
path = paramList0.childPath(param)
|
|
if path is not None:
|
|
childName = '.'.join(path)
|
|
else:
|
|
childName = param.name()
|
|
print(' parameter: %s'% childName)
|
|
print(' change: %s'% change)
|
|
print(' data: %s'% str(data))
|
|
print(' ----------')
|
|
|
|
class Curves:
|
|
def __init__(self, legend: str, key: str, channel: int, color: str, buffer_len: int, period: int):
|
|
self.curveItem = pg.PlotCurveItem(pen=({'color': color, 'width': 1}))
|
|
self.legendStr = legend
|
|
self.keyStr = key
|
|
self.channel = channel
|
|
self.data_buf = np.zeros(buffer_len)
|
|
self.time_stamp = np.zeros(buffer_len)
|
|
self.buffLen = buffer_len
|
|
self.period = period
|
|
|
|
def update(self, tec_data, cnt):
|
|
if cnt == 0:
|
|
np.copyto(self.data_buf, np.full(self.buffLen, tec_data[self.channel][self.keyStr]))
|
|
else:
|
|
self.data_buf[:-1] = self.data_buf[1:]
|
|
self.data_buf[-1] = tec_data[self.channel][self.keyStr]
|
|
self.time_stamp[:-1] = self.time_stamp[1:]
|
|
self.time_stamp[-1] = cnt * self.period / 1000
|
|
self.curveItem.setData(x = self.time_stamp, y = self.data_buf)
|
|
|
|
class Graph:
|
|
def __init__(self, parent: pg.LayoutWidget, title: str, row: int, col: int, curves: list[Curves]):
|
|
self.plotItem = pg.PlotWidget(title=title)
|
|
self.legendItem = pg.LegendItem(offset=(75, 30), brush=(50,50,200,150))
|
|
self.legendItem.setParentItem(self.plotItem.getPlotItem())
|
|
parent.addWidget(self.plotItem, row, col)
|
|
self.curves = curves
|
|
for curve in self.curves:
|
|
self.plotItem.addItem(curve.curveItem)
|
|
self.legendItem.addItem(curve.curveItem, curve.legendStr)
|
|
|
|
def update(self, tec_data, cnt):
|
|
for curve in self.curves:
|
|
curve.update(tec_data, cnt)
|
|
self.plotItem.setRange(xRange=[(cnt - self.curves[0].buffLen) * self.curves[0].period / 1000, cnt * self.curves[0].period / 1000])
|
|
|
|
def TECsync():
|
|
global TECparams
|
|
for channel in range(2):
|
|
for parents in TECparams[channel]:
|
|
if parents['tag'] == 'report':
|
|
for data in tec.report_mode():
|
|
for children in parents['children']:
|
|
children['value'] = data[channel][children['tag']]
|
|
if quit:
|
|
break
|
|
if parents['tag'] == 'pwm':
|
|
for children in parents['children']:
|
|
children['value'] = tec.get_pwm()[channel][children['tag']]['value']
|
|
if parents['tag'] == 'pid':
|
|
for children in parents['children']:
|
|
children['value'] = tec.get_pid()[channel]['parameters'][children['tag']]
|
|
if parents['tag'] == 's-h':
|
|
for children in parents['children']:
|
|
children['value'] = tec.get_steinhart_hart()[channel]['params'][children['tag']]
|
|
if parents['tag'] == 'PIDtarget':
|
|
for children in parents['children']:
|
|
children['value'] = tec.get_pid()[channel]['target']
|
|
|
|
|
|
cnt = 0
|
|
def updateData():
|
|
global cnt
|
|
for data in tec.report_mode():
|
|
|
|
ch0tempGraph.update(data, cnt)
|
|
ch1tempGraph.update(data, cnt)
|
|
ch0currentGraph.update(data, cnt)
|
|
ch1currentGraph.update(data, cnt)
|
|
|
|
if quit:
|
|
break
|
|
cnt += 1
|
|
|
|
|
|
if __name__ == '__main__':
|
|
tec = Client(host="192.168.1.26", port=23, timeout=None)
|
|
TECsync()
|
|
|
|
app = pg.mkQApp()
|
|
pg.setConfigOptions(antialias=True)
|
|
mw = QtGui.QMainWindow()
|
|
mw.setWindowTitle('Thermostat Control Panel')
|
|
mw.resize(1920,1200)
|
|
cw = QtGui.QWidget()
|
|
mw.setCentralWidget(cw)
|
|
l = QtGui.QVBoxLayout()
|
|
layout = pg.LayoutWidget()
|
|
l.addWidget(layout)
|
|
cw.setLayout(l)
|
|
|
|
## Create tree of Parameter objects
|
|
paramList0 = Parameter.create(name='GUIparams', type='group', children=GUIparams[0])
|
|
paramList0.sigTreeStateChanged.connect(change)
|
|
ch0Tree = ParameterTree()
|
|
ch0Tree.setParameters(paramList0, showTop=False)
|
|
|
|
paramList1 = Parameter.create(name='GUIparams', type='group', children=GUIparams[1])
|
|
paramList1.sigTreeStateChanged.connect(change)
|
|
ch1Tree = ParameterTree()
|
|
ch1Tree.setParameters(paramList1, showTop=False)
|
|
|
|
layout.addWidget(ch0Tree, 1, 1, 1, 1)
|
|
layout.addWidget(ch1Tree, 2, 1, 1, 1)
|
|
|
|
ch0tempGraph = Graph(layout, 'Channel 0 Temperature', 1, 2, [Curves('Feedback', 'temperature', 0, 'r', rec_len, refresh_period)])
|
|
ch1tempGraph = Graph(layout, 'Channel 1 Temperature', 2, 2, [Curves('Feedback', 'temperature', 1, 'r', rec_len, refresh_period)])
|
|
ch0currentGraph = Graph(layout, 'Channel 0 Current', 1, 3, [Curves('Feedback', 'tec_i', 0, 'r', rec_len, refresh_period),
|
|
Curves('Setpoint', 'i_set', 0, 'g', rec_len, refresh_period)])
|
|
ch1currentGraph = Graph(layout, 'Channel 1 Current', 2, 3, [Curves('Feedback', 'tec_i', 1, 'r', rec_len, refresh_period),
|
|
Curves('Setpoint', 'i_set', 1, 'g', rec_len, refresh_period)])
|
|
|
|
t = QtCore.QTimer()
|
|
t.timeout.connect(updateData)
|
|
t.start(refresh_period)
|
|
|
|
mw.show()
|
|
|
|
pg.exec() |