thermostat/pytec/tecQT.py

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': 'Disable Output', 'type': 'action', 'tip': 'Disable Output'},
    {'name': 'Constant Current', 'type': 'group', 'children': [
        {'name': 'Set Current', 'type': 'float', 'value': 0, 'step': 0.1, 'siPrefix': True, 'suffix': 'A'},
    ]},    
    {'name': 'Temperature PID', 'type': 'bool', 'value': False, 'children': [
        {'name': 'Set Temperature', 'type': 'float', 'value': 25, 'step': 0.1, 'siPrefix': True, 'suffix': 'C'},
    ]},    
    {'name': 'Output Config', 'expanded': False, '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', 'expanded': False,  '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', 'expanded': False,  '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 _ in range(2)]

## If anything changes in the tree, print a message
def change(param, changes, ch):
    print("tree changes:")
    for param, change, data in changes:
        path = paramList[ch].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('  ----------')

        if (childName == 'Disable Output'):
            tec.set_param('pwm', ch, 'i_set', 0)
            paramList[ch].child('Constant Current').child('Set Current').setValue(0) 
            paramList[ch].child('Temperature PID').setValue(False)          

        if (childName == 'Temperature PID'):
            if (data):
                tec.set_param("pwm", ch, "pid")
            else:
                tec.set_param('pwm', ch, 'i_set', paramList[ch].child('Constant Current').child('Set Current').value())  
        
        if (childName == 'Constant Current.Set Current'):
            tec.set_param('pwm', ch, 'i_set', data)
            paramList[ch].child('Temperature PID').setValue(False)  

        if (childName == 'Temperature PID.Set Temperature'):
            tec.set_param('pid', ch, 'target', data)

        if (childName == 'Output Config.Max Current'):
            tec.set_param('pwm', ch, 'max_i_pos', data)
            tec.set_param('pwm', ch, 'max_i_neg', data)
            tec.set_param('pid', ch, 'output_min', -data)
            tec.set_param('pid', ch, 'output_max', data)

        if (childName == 'Output Config.Max Voltage'):
            tec.set_param('pwm', ch, 'max_v', data)

        if (childName == 'Thermistor Config.T0'):
            tec.set_param('s-h', ch, 't0', data)

        if (childName == 'Thermistor Config.R0'):
            tec.set_param('s-h', ch, 'r0', data)

        if (childName == 'Thermistor Config.Beta'):
            tec.set_param('s-h', ch, 'b', data)

        if (childName == 'PID Config.kP'):
            tec.set_param('pid', ch, 'kp', data)

        if (childName == 'PID Config.kI'):
            tec.set_param('pid', ch, 'ki', data)

        if (childName == 'PID Config.kD'):
            tec.set_param('pid', ch, 'kd', data)

        if (childName == 'Save'):
            tec.save_config()
        
def change0(param, changes):
    change(param, changes, 0)

def change1(param, changes):
    change(param, changes, 1)

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']
    
def refreshTreeParam(tempTree:dict, channel:int) -> dict:
    # tempTree['children']['Constant Current']['value'] = not TECparams[channel][0]['children'][0]['value']
    tempTree['children']['Constant Current']['children']['Set Current']['value'] = TECparams[channel][1]['children'][3]['value']
    tempTree['children']['Temperature PID']['value'] = TECparams[channel][0]['children'][0]['value']
    tempTree['children']['Temperature PID']['children']['Set Temperature']['value'] = TECparams[channel][4]['children'][0]['value']
    tempTree['children']['Output Config']['children']['Max Current']['value'] = TECparams[channel][1]['children'][0]['value']
    tempTree['children']['Output Config']['children']['Max Voltage']['value'] = TECparams[channel][1]['children'][2]['value']
    tempTree['children']['Thermistor Config']['children']['T0']['value'] = TECparams[channel][3]['children'][0]['value'] - 273.15
    tempTree['children']['Thermistor Config']['children']['R0']['value'] = TECparams[channel][3]['children'][1]['value']
    tempTree['children']['Thermistor Config']['children']['Beta']['value'] = TECparams[channel][3]['children'][2]['value']
    tempTree['children']['PID Config']['children']['kP']['value'] = TECparams[channel][2]['children'][0]['value']
    tempTree['children']['PID Config']['children']['kI']['value'] = TECparams[channel][2]['children'][1]['value']
    tempTree['children']['PID Config']['children']['kD']['value'] = TECparams[channel][2]['children'][2]['value']
    return tempTree

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)

    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
    paramList = [Parameter.create(name='GUIparams', type='group', children=GUIparams[0]),
                 Parameter.create(name='GUIparams', type='group', children=GUIparams[1])]

    paramList[0].sigTreeStateChanged.connect(change0)
    print(paramList[0].children())
    ch0Tree = ParameterTree()
    ch0Tree.setParameters(paramList[0], showTop=False)

    paramList[1].sigTreeStateChanged.connect(change1)
    ch1Tree = ParameterTree()
    ch1Tree.setParameters(paramList[1], showTop=False)

    TECsync()
    paramList[0].restoreState(refreshTreeParam(paramList[0].saveState(), 0))
    paramList[1].restoreState(refreshTreeParam(paramList[1].saveState(), 1))

    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()