"""
    The Ultrafiltration ui loader
"""
# Python
import os
# Qt
from PySide2 import QtCore, QtWidgets
from PySide2.QtCore import Slot
# parent
from simulator.dynamicloader import DynamicLoader
# hd Simulator
from dialin.ui.hd_simulator import HDSimulator
# plugin specific
from dialin.common.hd_defs import *


class Loader(DynamicLoader):
    """
    The Ultrafiltration ui loader
    """
    tbReset: QtWidgets.QToolButton
    tbSend: QtWidgets.QToolButton

    tblSubMode: QtWidgets.QTableWidget
    tblWaterSampleStates: QtWidgets.QTableWidget
    tblConsumablesSelfTestStates: QtWidgets.QTableWidget
    tblNoCartridgeSelfTestStates: QtWidgets.QTableWidget
    tblInstallationStates: QtWidgets.QTableWidget
    tblDrySelfTestStates: QtWidgets.QTableWidget
    tblPrimeStates: QtWidgets.QTableWidget
    tblRecirculateStates: QtWidgets.QTableWidget
    tblPatientConnectionStates: QtWidgets.QTableWidget

    def __init__(self, hd_simulator: HDSimulator):
        super().__init__(os.path.dirname(__file__), hd_simulator)

    def _init_loader(self):
        """
        finds and creates widgets
        :return: none
        """
        self.tbReset = self.find_tool_button('tbReset')
        self.tbSend = self.find_tool_button('tbSend')

        self.tblSubMode = self.find_table_widget('tblSubMode')
        self.tblWaterSampleStates = self.find_table_widget('tblWaterSampleStates')
        self.tblConsumablesSelfTestStates = self.find_table_widget('tblConsumablesSelfTestStates')
        self.tblNoCartridgeSelfTestStates = self.find_table_widget('tblNoCartridgeSelfTestStates')
        self.tblInstallationStates = self.find_table_widget('tblInstallationStates')
        self.tblDrySelfTestStates = self.find_table_widget('tblDrySelfTestStates')
        self.tblPrimeStates = self.find_table_widget('tblPrimeStates')
        self.tblRecirculateStates = self.find_table_widget('tblRecirculateStates')
        self.tblPatientConnectionStates = self.find_table_widget('tblPatientConnectionStates')

        self._init_widgets()

    def _init_connections(self):
        """
        initializes the widgets connections
        :return: none
        """
        self.tbReset.clicked.connect(self._init_widgets)
        self.tbSend.clicked.connect(self.do_pre_treatment_states)

        self.tblSubMode.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblWaterSampleStates.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblConsumablesSelfTestStates.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblNoCartridgeSelfTestStates.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblInstallationStates.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblDrySelfTestStates.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblPrimeStates.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblRecirculateStates.currentCellChanged.connect(self.do_pre_treatment_states)
        self.tblPatientConnectionStates.currentCellChanged.connect(self.do_pre_treatment_states)

    @Slot()
    def _init_widgets(self):
        """
        initializes the widgets' properties
        :return: none
        """
        self.tblSubMode.setCurrentCell(HD_Pre_Treatment_Mode_States.HD_PRE_TREATMENT_START_STATE.value, 0)
        self.tblWaterSampleStates.setCurrentCell(HD_Pre_Treatment_Sample_Water_States.SAMPLE_WATER_SETUP_STATE.value, 0)
        self.tblConsumablesSelfTestStates.setCurrentCell(HD_Pre_Treatment_Consumable_Self_Test_States.CONSUMABLE_SELF_TESTS_INSTALL_STATE.value, 0)
        self.tblNoCartridgeSelfTestStates.setCurrentCell(HD_Pre_Treatment_No_Cart_Self_Tests_States.NO_CART_SELF_TESTS_START_STATE.value, 0)
        self.tblInstallationStates.setCurrentCell(HD_Pre_Treatment_Cartridge_Install_States.CARTRIDGE_INSTALL_STATE.value, 0)
        self.tblDrySelfTestStates.setCurrentCell(HD_Pre_Treatment_Dry_Self_Tests_States.DRY_SELF_TESTS_START_STATE.value, 0)
        self.tblPrimeStates.setCurrentCell(HD_Pre_Treatment_Prime_States.HD_PRIME_START_STATE.value, 0)
        self.tblRecirculateStates.setCurrentCell(HD_Pre_Treatment_Recirc_States.PRE_TREATMENT_RECIRC_STATE.value, 0)
        self.tblPatientConnectionStates.setCurrentCell(HD_Pre_Treatment_Patient_Connection_States.PATIENT_CONNECTION_STATE.value, 0)

        self.tblSubMode.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblWaterSampleStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblConsumablesSelfTestStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblNoCartridgeSelfTestStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblInstallationStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblDrySelfTestStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblPrimeStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblRecirculateStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
        self.tblPatientConnectionStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)

    @Slot()
    def do_pre_treatment_states(self):
        """
        the slot for saline bolus state change
        :return: none
        """
        sub_mode = int(self.tblSubMode.verticalHeaderItem(self.tblSubMode.currentRow()).text())
        water_sample = int(self.tblWaterSampleStates.verticalHeaderItem(self.tblWaterSampleStates.currentRow()).text())
        consumables_self_test = int(self.tblConsumablesSelfTestStates.verticalHeaderItem(self.tblConsumablesSelfTestStates.currentRow()).text())
        no_cartridge_self_test = int(self.tblNoCartridgeSelfTestStates.verticalHeaderItem(self.tblNoCartridgeSelfTestStates.currentRow()).text())
        installation = int(self.tblInstallationStates.verticalHeaderItem(self.tblInstallationStates.currentRow()).text())
        dry_self_test = int(self.tblDrySelfTestStates.verticalHeaderItem(self.tblDrySelfTestStates.currentRow()).text())
        prime = int(self.tblPrimeStates.verticalHeaderItem(self.tblPrimeStates.currentRow()).text())
        recirculate = int(self.tblRecirculateStates.verticalHeaderItem(self.tblRecirculateStates.currentRow()).text())
        patient_connection = int(self.tblPatientConnectionStates.verticalHeaderItem(self.tblPatientConnectionStates.currentRow()).text())
        self.hd_simulator.cmd_send_pre_treatment_state_data(
            sub_mode,
            water_sample,
            consumables_self_test,
            no_cartridge_self_test,
            installation,
            dry_self_test,
            prime,
            recirculate,
            patient_connection
        )