143 lines
3.8 KiB
Python
143 lines
3.8 KiB
Python
from typing import Dict
|
|
from typing_extensions import TypedDict
|
|
|
|
from PySide6.QtCore import QObject, Qt, Slot
|
|
from PySide6.QtGui import QFontDatabase
|
|
from PySide6.QtWidgets import QButtonGroup, QDockWidget, QGridLayout, QGroupBox, QHBoxLayout, QLabel, QRadioButton, QVBoxLayout, QWidget
|
|
|
|
ContactorStates = TypedDict("ContactorStates", {})
|
|
|
|
|
|
class BMSState(QObject):
|
|
"""Represents the BMS state, including history."""
|
|
main_voltage: float
|
|
aux_voltage: float
|
|
current: float
|
|
|
|
def __init__(self, parent=None, upstream=None):
|
|
super().__init__(parent)
|
|
# uhh, take a connection to the upstream?
|
|
|
|
|
|
class BMSModuleViewer(QWidget):
|
|
"""BMS module status viewer (temp and voltage)"""
|
|
|
|
# use graphics view for rendering.
|
|
|
|
temps: list[float] = []
|
|
volts: list[float] = []
|
|
|
|
def __init__(self, parent = None) -> None:
|
|
super().__init__(parent)
|
|
|
|
layout = QGridLayout()
|
|
|
|
bg = QButtonGroup(self)
|
|
|
|
self.volts_btn = QRadioButton("Voltage", self)
|
|
self.temps_btn = QRadioButton("Temperatures", self)
|
|
bg.addButton(self.volts_btn)
|
|
bg.addButton(self.temps_btn)
|
|
|
|
|
|
layout.addWidget(self.volts_btn, 0, 0)
|
|
layout.addWidget(self.temps_btn, 0, 1)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BMSOverview(QWidget):
|
|
|
|
current: QLabel
|
|
main_voltage: QLabel
|
|
aux_voltage: QLabel
|
|
|
|
def __init__(self, parent=None) -> None:
|
|
super().__init__(parent)
|
|
self.setMaximumWidth
|
|
layout = QGridLayout()
|
|
layout.setRowStretch(0, 80)
|
|
layout.setRowStretch(1, 20)
|
|
|
|
number_font = QFontDatabase.systemFont(QFontDatabase.SystemFont.FixedFont)
|
|
number_font.setPointSize(18)
|
|
hcenter = Qt.AlignmentFlag.AlignHCenter
|
|
|
|
self.main_voltage = QLabel("0.000", self)
|
|
self.main_voltage.setAlignment(hcenter)
|
|
self.main_voltage.setFont(number_font)
|
|
layout.addWidget(self.main_voltage, 0, 0)
|
|
|
|
main_v_label = QLabel("Main Voltage", self)
|
|
main_v_label.setAlignment(hcenter)
|
|
layout.addWidget(main_v_label, 1, 0)
|
|
|
|
|
|
self.aux_voltage = QLabel("0.000", self)
|
|
self.aux_voltage.setAlignment(hcenter)
|
|
self.aux_voltage.setFont(number_font)
|
|
layout.addWidget(self.aux_voltage, 0, 1)
|
|
|
|
aux_v_label = QLabel("Aux Voltage", self)
|
|
aux_v_label.setAlignment(hcenter)
|
|
layout.addWidget(aux_v_label, 1, 1)
|
|
|
|
self.current = QLabel("0.000", self)
|
|
self.current.setAlignment(hcenter)
|
|
self.current.setFont(number_font)
|
|
layout.addWidget(self.current, 0, 2)
|
|
|
|
current_label = QLabel("Battery Current", self)
|
|
current_label.setAlignment(hcenter)
|
|
layout.addWidget(current_label, 1, 2)
|
|
|
|
# now add widgets that display the numeric values.
|
|
# then make slots that take floats and display them.
|
|
self.setLayout(layout)
|
|
|
|
|
|
@Slot(float)
|
|
def update_main_v(self, value: float):
|
|
self.main_voltage.setText(f"{value:.2f}")
|
|
|
|
@Slot(float)
|
|
def set_aux_v(self, value:float):
|
|
self.aux_voltage.setText(f"{value:.3f}")
|
|
|
|
@Slot(float)
|
|
def set_current(self, value: float):
|
|
self.current.setText(f"{value:.3f}")
|
|
|
|
|
|
|
|
class BMSStatus(QWidget):
|
|
|
|
contactor_items: Dict[str, QLabel] = dict()
|
|
"A mapping of string names to the label, used to set open/closed"
|
|
|
|
def __init__(self, parent: QWidget | None = None, contactors: list[str] = []):
|
|
super().__init__(parent)
|
|
|
|
layout = QVBoxLayout(self)
|
|
|
|
self.contactors_grp = QGroupBox("Contactor State", self)
|
|
contactor_layout = QGridLayout()
|
|
self.contactors_grp.setLayout(contactor_layout)
|
|
layout.addWidget(self.contactors_grp)
|
|
|
|
for c in contactors:
|
|
label = QLabel(c, self)
|
|
|
|
self.contactor_items[c] = label
|
|
contactor_layout.addWidget(label)
|
|
|
|
|
|
|
|
|
|
class BMSPlotsWidget(QWidget):
|
|
pass
|
|
|
|
|