Index: simulator/run.py =================================================================== diff -u -r867655840cada2d79fa5028c8fab68516ea59673 -rcb8d48033154f4b36a6a05b6f31bd10390e84ac7 --- simulator/run.py (.../run.py) (revision 867655840cada2d79fa5028c8fab68516ea59673) +++ simulator/run.py (.../run.py) (revision cb8d48033154f4b36a6a05b6f31bd10390e84ac7) @@ -1,402 +1,128 @@ - # import system modules import sys # import project classes -# import names from dialin.ui import utils from dialin import HDSimulator from dialin.ui.hd_simulator import TXStates, EResponse # Import PySide2 classes -from PySide2.QtUiTools import QUiLoader from PySide2 import QtCore, QtWidgets -from PySide2.QtCore import QFile, Slot -from PySide2.QtCore import QTimer +from PySide2.QtCore import Qt +from runtimewidget import RunTimeWidget +# later it's planed to load all the classes in plugin folder dynamically +# so used * in here to load all +from plugins import * HDSimulator = HDSimulator() -class RunTimeWidget: +class Simulator(RunTimeWidget): """ - the parent class of all the run time loadable widgets + The simulator class which loads the ui file dynamically and initializes the objects + and can be eventually shown. + Note: this class is growing fast and seems like needs to be multiple classes """ + mdiArea: QtWidgets.QMdiArea - loader: QUiLoader - window: QtWidgets.QWidget + obj_ultrafiltration: Ultrafiltration + sub_ultrafiltration: QtWidgets.QMdiSubWindow + wgt_ultrafiltration: QtWidgets.QWidget + act_ultrafiltration: QtWidgets.QAction - def __init__(self, ui_name: str): - self.ui_name = ui_name - self.loader = QUiLoader() - self.__load_ui() + obj_inlinebloodpressures: InlineBloodPressures + sub_inlinebloodpressures: QtWidgets.QMdiSubWindow + wgt_inlinebloodpressures: QtWidgets.QWidget + act_inlinebloodpressures: QtWidgets.QAction - def __load_ui(self): - """ - loads the ui file of the GUI at run time - :return: none - """ - ui_file = QFile(self.ui_name) - ui_file.open(QFile.ReadOnly) - self.window = self.loader.load(ui_file) - # assert if the ui file can't be loaded - error = self.loader.errorString() - assert len(error) == 0, error - ui_file.close() + obj_salinebolus: SalineBolus + sub_salinebolus: QtWidgets.QMdiSubWindow + wgt_salinebolus: QtWidgets.QWidget + act_salinebolus: QtWidgets.QAction - def show(self): - """ - shows the main container window - :return: none - """ - self.window.show() + obj_treatmentstates: TreatmentStates + sub_treatmentstates: QtWidgets.QMdiSubWindow + wgt_treatmentstates: QtWidgets.QWidget + act_treatmentstates: QtWidgets.QAction - def __find_child(self, child_type, child_name: str) -> QtWidgets.QWidget: - """ - finds a child in the loaded ui and returns the reference to it if found - otherwise quits the application - :param child_type: (var) type of the child - :param child_name: (str) name of the child - :return: (QtWidgets.QWidget) reference to the child - """ - child: QtWidgets.QWidget = self.window.findChild(child_type, child_name) - assert child is not None, "child name '{}' with type '{}' can't be found.".format(child_name, child_type) - return child + obj_treatmentranges: TreatmentRanges + sub_treatmentranges: QtWidgets.QMdiSubWindow + wgt_treatmentranges: QtWidgets.QWidget + act_treatmentranges: QtWidgets.QAction - def find_label(self, name: str) -> QtWidgets.QLabel: - """ - convenient method of find_child for QLabel - :param name: (str) name of the QLabel Object - :return: (QLabel) reference to the QLabel - """ - child = self.__find_child(QtWidgets.QLabel, name) - return child + obj_heparin: Heparin + sub_heparin: QtWidgets.QMdiSubWindow + wgt_heparin: QtWidgets.QWidget + act_heparin: QtWidgets.QAction - def find_button(self, name: str) -> QtWidgets.QPushButton: - """ - convenient method of find_child for QPushButton - :param name: (str) name of the QPushButton Object - :return: (QPushButton) reference to the QPushButton - """ - child = self.__find_child(QtWidgets.QPushButton, name) - return child - - def find_combobox(self, name: str) -> QtWidgets.QComboBox: - """ - convenient method of find_child for QComboBox - :param name: (str) name of the QComboBox Object - :return: (QPushButton) reference to the QComboBox - """ - child = self.__find_child(QtWidgets.QComboBox, name) - return child - - def find_spinbox(self, name: str) -> QtWidgets.QSpinBox: - """ - convenient method of find_child for QSpinBox - :param name: (str) name of the QSpinBox Object - :return: (QSpinBox) reference to the QSpinBox - """ - child = self.__find_child(QtWidgets.QSpinBox, name) - return child - - def find_slider(self, name: str) -> QtWidgets.QSlider: - """ - convenient method of find_child for QSlider - :param name: (str) name of the QSlider Object - :return: (QSlider) reference to the QSlider - """ - child = self.__find_child(QtWidgets.QSlider, name) - return child - - def find_table_widget(self, name: str) -> QtWidgets.QTableWidget: - """ - convenient method of find_child for QTableWidget - :param name: (str) name of the QTableWidget Object - :return: (QTableWidget) reference to the QTableWidget - """ - child: QtWidgets.QTableWidget = self.__find_child(QtWidgets.QTableWidget, name) - return child - - -class Simulator(RunTimeWidget): - """ - The simulator class which loads the ui file dynamically and initializes the objects - and can be eventually shown. - """ - # global variables declarations - ui_file_name = "Simulator.ui" - timer: QTimer - - # pushbutton - btnSalineAccept: QtWidgets.QPushButton - btnSalineReject: QtWidgets.QPushButton - btnUfPauseAccept: QtWidgets.QPushButton - btnUfPauseReject: QtWidgets.QPushButton - btnUfResumeAccept: QtWidgets.QPushButton - btnUfResumeReject: QtWidgets.QPushButton - btnUfEditAccept: QtWidgets.QPushButton - btnUfEditReject: QtWidgets.QPushButton - # label - lblSalineAction: QtWidgets.QLabel - lblUfPauseAction: QtWidgets.QLabel - lblUfResumeAction: QtWidgets.QLabel - lblUfEditAction: QtWidgets.QLabel - # spinbox - spnSalineRejectReason: QtWidgets.QSpinBox - spnUfPauseRejectReason: QtWidgets.QSpinBox - spnUfResumeRejectReason: QtWidgets.QSpinBox - spnUfEditRejectReason: QtWidgets.QSpinBox - spnDurationMin: QtWidgets.QSpinBox - spnDurationMax: QtWidgets.QSpinBox - spnUFVolumeMin: QtWidgets.QSpinBox - spnUFVolumeMax: QtWidgets.QSpinBox - spnDialysateMin: QtWidgets.QSpinBox - spnDialysateMax: QtWidgets.QSpinBox - # combobox - cmbSalineAcceptTarget: QtWidgets.QComboBox - # sliders - sldSalineTarget: QtWidgets.QSlider - sldSalineCumulative: QtWidgets.QSlider - sldSalineVolume: QtWidgets.QSlider - sldUfVolume: QtWidgets.QSlider - # tables - tblSalineSubMode: QtWidgets.QTableWidget - tblSalineUFStates: QtWidgets.QTableWidget - tblSalineSalineStates: QtWidgets.QTableWidget - - # static class variables - saline_requested_state: TXStates - def __init__(self): - super().__init__(Simulator.ui_file_name) + super().__init__("simulator.ui") self.saline_requested_state = TXStates.SALINE_BOLUS_STATE_IDLE self.initialize() + self.init_widgets() + self.init_connections() - def setup_ranges(self): + def initialize(self): """ - sets up the treatment ranges timer and 1 sec interval + initializes the class by calling it's initializer methods to make objects ready :return: none """ - self.spnDurationMin = self.find_spinbox('spnDurationMin') - self.spnDurationMax = self.find_spinbox('spnDurationMax') - self.spnUFVolumeMin = self.find_spinbox('spnUFVolumeMin') - self.spnUFVolumeMax = self.find_spinbox('spnUFVolumeMax') - self.spnDialysateMin = self.find_spinbox('spnDialysateMin') - self.spnDialysateMax = self.find_spinbox('spnDialysateMax') - self.timer = QTimer() - self.timer.start(1000) - self.timer.timeout.connect(self.do_ranges_data) + self.mdiArea = self.find_widget(QtWidgets.QMdiArea, 'mdiArea') + # TODO: this section needs to change to go over the folders in the plugins folder and load each automatically - def setup_uf_adjustment(self): - """ - sets up the treatment Ultrafiltration adjustment GUI section - :return: none - """ - self.btnUfPauseAccept = self.find_button('btnUfPauseAccept') - self.btnUfPauseReject = self.find_button('btnUfPauseReject') - self.btnUfResumeAccept = self.find_button('btnUfResumeAccept') - self.btnUfResumeReject = self.find_button('btnUfResumeReject') - self.btnUfEditAccept = self.find_button('btnUfEditAccept') - self.btnUfEditReject = self.find_button('btnUfEditReject') - self.lblUfPauseAction = self.find_label('lblUfPauseAction') - self.lblUfResumeAction = self.find_label('lblUfResumeAction') - self.lblUfEditAction = self.find_label('lblUfEditAction') - self.spnUfPauseRejectReason = self.find_spinbox('spnUfPauseRejectReason') - self.spnUfResumeRejectReason = self.find_spinbox('spnUfResumeRejectReason') - self.spnUfEditRejectReason = self.find_spinbox('spnUfEditRejectReason') - self.sldUfVolume = self.find_slider('sldUfVolume') - self.btnUfPauseAccept.clicked.connect(self.do_uf_pause_accept) - self.btnUfPauseReject.clicked.connect(self.do_uf_pause_reject) - self.btnUfResumeAccept.clicked.connect(self.do_uf_resume_accept) - self.btnUfResumeReject.clicked.connect(self.do_uf_resume_reject) - self.btnUfEditAccept.clicked.connect(self.do_uf_edit_accept) - self.btnUfEditReject.clicked.connect(self.do_uf_edit_reject) - self.sldUfVolume.valueChanged.connect(self.do_uf_volume_data) + self.act_treatmentstates = self.find_action('actTreatmentStates') + self.act_treatmentranges = self.find_action('actTreatmentRanges') + self.act_ultrafiltration = self.find_action('actUltrafiltration') + self.act_salinebolus = self.find_action('actSalineBolus') + self.act_inlinebloodpressures = self.find_action('actInLineBloodPressures') + self.act_heparin = self.find_action('actionHeparin') - def setup_saline_adjustment(self): + def init_widgets(self): """ - sets up the treatment saline bolus adjustment GUI section + initializes the widgets' properties :return: none """ - self.btnSalineAccept = self.find_button('btnSalineAccept') - self.btnSalineReject = self.find_button('btnSalineReject') - self.lblSalineAction = self.find_label('lblSalineAction') - self.spnSalineRejectReason = self.find_spinbox('spnSalineRejectReason') - self.cmbSalineAcceptTarget = self.find_combobox('cmbSalineAcceptTarget') - self.btnSalineAccept.clicked.connect(self.do_sb_accept) - self.btnSalineReject.clicked.connect(self.do_sb_reject) + # ultrafiltration + self.obj_ultrafiltration = Ultrafiltration() + self.wgt_ultrafiltration = self.obj_ultrafiltration.window + self.sub_ultrafiltration = self.mdiArea.addSubWindow(self.wgt_ultrafiltration) - def setup_saline_data(self): - """ - sets up the treatment saline bolus data sender GUI section - :return: none - """ - self.sldSalineTarget = self.find_slider('sldSalineTarget') - self.sldSalineCumulative = self.find_slider('sldSalineCumulative') - self.sldSalineVolume = self.find_slider('sldSalineVolume') - self.sldSalineTarget.valueChanged.connect(self.do_saline_data) - self.sldSalineCumulative.valueChanged.connect(self.do_saline_data) - self.sldSalineVolume.valueChanged.connect(self.do_saline_data) + self.obj_inlinebloodpressures = InlineBloodPressures() + self.wgt_inlinebloodpressures = self.obj_inlinebloodpressures.window + self.sub_inlinebloodpressures = self.mdiArea.addSubWindow(self.wgt_inlinebloodpressures) - def setup_treatment_states(self): - """ - sets up the treatment saline bolus states GUI section - :return: none - """ - self.tblSalineSubMode = self.find_table_widget('tblSalineSubMode') - self.tblSalineUFStates = self.find_table_widget('tblSalineUFStates') - self.tblSalineSalineStates = self.find_table_widget('tblSalineSalineStates') - self.tblSalineSubMode.setCurrentCell(0, 0) - self.tblSalineUFStates.setCurrentCell(0, 0) - self.tblSalineSalineStates.setCurrentCell(0, 0) - self.tblSalineSubMode.cellClicked.connect(self.do_saline_saline_state) - self.tblSalineUFStates.cellClicked.connect(self.do_saline_saline_state) - self.tblSalineSalineStates.cellClicked.connect(self.do_saline_saline_state) + self.obj_salinebolus = SalineBolus() + self.wgt_salinebolus = self.obj_salinebolus.window + self.sub_salinebolus = self.mdiArea.addSubWindow(self.wgt_salinebolus) - @Slot() - def do_sb_accept(self): - """ - the slot for accept saline bolus button - :return: none - """ - # toggle the saline requested state - if self.saline_requested_state == TXStates.SALINE_BOLUS_STATE_IN_PROGRESS: - self.saline_requested_state = TXStates.SALINE_BOLUS_STATE_IDLE - else: - self.saline_requested_state = TXStates.SALINE_BOLUS_STATE_IN_PROGRESS + self.obj_treatmentstates = TreatmentStates() + self.wgt_treatmentstates = self.obj_treatmentstates.window + self.sub_treatmentstates = self.mdiArea.addSubWindow(self.wgt_treatmentstates) - target = self.cmbSalineAcceptTarget.currentText() - HDSimulator.cmd_set_saline_bolus_response(True, 0, target, self.saline_requested_state) - self.lblSalineAction.setText('Accepted ' + target) + self.obj_treatmentranges = TreatmentRanges() + self.wgt_treatmentranges = self.obj_treatmentranges.window + self.sub_treatmentranges = self.mdiArea.addSubWindow(self.wgt_treatmentranges) - @Slot() - def do_sb_reject(self): - """ - the slot for accept saline bolus button - :return: none - """ - reason = self.spnSalineRejectReason.value() - HDSimulator.cmd_set_saline_bolus_response(False, reason, 0, self.saline_requested_state) - self.lblSalineAction.setText('Rejected ' + "{}".format(reason)) + self.obj_heparin = Heparin() + self.wgt_heparin = self.obj_heparin.window + self.sub_heparin = self.mdiArea.addSubWindow(self.wgt_heparin) - @Slot() - def do_uf_pause_accept(self): - """ - the slot for accept ultrafiltration pause button - :return: none - """ - HDSimulator.cmd_set_treatment_adjust_ultrafiltration_state_response( - EResponse.Accepted, 0, TXStates.UF_PAUSED_STATE) - self.lblUfPauseAction.setText('Accepted ') + for sub in self.mdiArea.subWindowList(): + sub.setWindowFlags(Qt.WindowMinimizeButtonHint | Qt.WindowTitleHint) + sub.setVisible(False) - @Slot() - def do_uf_pause_reject(self): + def init_connections(self): """ - the slot for reject ultrafiltration pause button + initializes the widgets connections :return: none """ - reason = self.spnUfPauseRejectReason.value() - HDSimulator.cmd_set_treatment_adjust_ultrafiltration_state_response( - EResponse.Rejected, reason, TXStates.UF_RUNNING_STATE) - self.lblUfPauseAction.setText('Rejected ' + "{}".format(reason)) + self.act_treatmentstates.toggled.connect(self.sub_treatmentstates.setVisible) + self.act_treatmentranges.toggled.connect(self.sub_treatmentranges.setVisible) + self.act_ultrafiltration.toggled.connect(self.sub_ultrafiltration.setVisible) + self.act_salinebolus.toggled.connect(self.sub_salinebolus.setVisible) + self.act_inlinebloodpressures.toggled.connect(self.sub_inlinebloodpressures.setVisible) + self.act_heparin.toggled.connect(self.sub_heparin.setVisible) - @Slot() - def do_uf_resume_accept(self): - """ - the slot for accept ultrafiltration resume accept - :return: none - """ - HDSimulator.cmd_set_treatment_adjust_ultrafiltration_state_response( - EResponse.Accepted, 0, TXStates.UF_RUNNING_STATE) - self.lblUfResumeAction.setText('Accepted ') - @Slot() - def do_uf_resume_reject(self): - """ - the slot for reject ultrafiltration resume button - :return: none - """ - reason = self.spnUfResumeRejectReason.value() - HDSimulator.cmd_set_treatment_adjust_ultrafiltration_state_response( - EResponse.Rejected, reason, TXStates.UF_PAUSED_STATE) - self.lblUfResumeAction.setText('Rejected ' + "{}".format(reason)) - - @Slot() - def do_uf_edit_accept(self): - """ - the slot for accept ultrafiltration next button - :return: none - """ - HDSimulator.cmd_set_treatment_adjust_ultrafiltration_edit_response( - EResponse.Accepted, 0, 2500, 60, 0, 10, 0, 10) - self.lblUfEditAction.setText('Accepted ') - - @Slot() - def do_uf_edit_reject(self): - """ - the slot for reject ultrafiltration next button - :return: none - """ - reason = self.spnUfEditRejectReason.value() - HDSimulator.cmd_set_treatment_adjust_ultrafiltration_edit_response( - EResponse.Rejected, reason, 2500, 60, 0, 10, 0, 10) - self.lblUfEditAction.setText('Rejected ' + "{}".format(reason)) - - @Slot() - def do_saline_saline_state(self): - """ - the slot for saline bolus state change - :return: none - """ - sub_mode = self.tblSalineSubMode.verticalHeaderItem(self.tblSalineSubMode.currentRow()).text() - uf_state = self.tblSalineUFStates.verticalHeaderItem(self.tblSalineUFStates.currentRow()).text() - saline = self.tblSalineSalineStates.verticalHeaderItem(self.tblSalineSalineStates.currentRow()).text() - HDSimulator.cmd_set_treatment_states_data(sub_mode, uf_state, saline) - - @Slot() - def do_saline_data(self): - """ - the slot which is called to send the saline bolus data - by calling the denaliMessage API cmd_set_treatment_saline_bolus_data - :return: none - """ - HDSimulator.cmd_set_treatment_saline_bolus_data(self.sldSalineTarget.value(), - self.sldSalineCumulative.value(), - self.sldSalineVolume.value()) - - @Slot() - def do_uf_volume_data(self): - """ - sends the ultrafiltration delivered volume message - :return: none - """ - HDSimulator.cmd_set_treatment_ultrafiltration_outlet_flow_data(self.sldUfVolume.value(), 0, 0, 0, 0, 0, 0) - - @Slot() - def do_ranges_data(self): - """ - sends the treatment ranges message with given value on the screen - :return: none - """ - HDSimulator.cmd_set_treatment_parameter_ranges( - self.spnDurationMin.value(), - self.spnDurationMax.value(), - self.spnUFVolumeMin.value(), - self.spnUFVolumeMax.value(), - self.spnDialysateMin.value(), - self.spnDialysateMax.value() - ) - - def initialize(self): - """ - initializes the class by calling it's initializer methods to make objects ready - :return: none - """ - self.setup_ranges() - self.setup_saline_adjustment() - self.setup_saline_data() - self.setup_uf_adjustment() - self.setup_treatment_states() - - def main(): """ the main function which initializes the Simulator and starts it. @@ -405,15 +131,15 @@ utils.tstStart(__file__) # create qt application - qapp = QtWidgets.QApplication(sys.argv) + app = QtWidgets.QApplication(sys.argv) simulator = Simulator() simulator.show() utils.tstDone() # start qt application main loop - sys.exit(qapp.exec_()) + sys.exit(app.exec_()) if __name__ == "__main__":