"""
    The Treatment States ui loader
"""
# Python
import os
# Qt
from PySide2 import QtWidgets
from PySide2.QtCore import Slot
# parent
from engine.dynamicloader   import DynamicLoader

# plugin specific
from leahi_dialin.common.td_defs  import TDTreatmentStates as TXStates
from leahi_dialin.common.td_defs  import TDDialysisStates as DialStates

# hd Simulator
from leahi_dialin.ui.td_messaging import TD_Messaging
# -- none --


class Loader(DynamicLoader):
    """
    The Treatment States ui loader
    """

    def __init__(self):
        self.td_interface = TD_Messaging()
        super().__init__(os.path.dirname(__file__))

    def _init_loader(self):
        """
        finds and creates widgets
        :return: none
        """
        self.tbReset    = self.find_widget(QtWidgets.QToolButton    , 'tbReset'                 )
        self.tbSend     = self.find_widget(QtWidgets.QToolButton    , 'tbSend'                  )

        self.tblSubMode             = self.find_widget(QtWidgets.QTableWidget       , 'tblSubMode')
        self.tblBloodPrimeStates    = self.find_widget(QtWidgets.QTableWidget       , 'tblBloodPrimeStates')
        self.tblDialysisStates      = self.find_widget(QtWidgets.QTableWidget       , 'tblDialysisStates')
        self.tblUFStates            = self.find_widget(QtWidgets.QTableWidget       , 'tblUFStates')
        self.tblTreatmentStopStates = self.find_widget(QtWidgets.QTableWidget       , 'tblTreatmentStopStates')
        self.tblRinsebackStates     = self.find_widget(QtWidgets.QTableWidget       , 'tblRinsebackStates')
        self.tblRecirculateStates   = self.find_widget(QtWidgets.QTableWidget       , 'tblRecirculateStates')
        self.tblTreatmentEndStates  = self.find_widget(QtWidgets.QTableWidget       , 'tblTreatmentEndStates')
        self.tblSalineStates        = self.find_widget(QtWidgets.QTableWidget       , 'tblSalineStates')
        self.tblHeparinStates       = self.find_widget(QtWidgets.QTableWidget       , 'tblHeparinStates')

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

        self.tblSubMode            .currentCellChanged.connect(self.do_treatment_states)
        self.tblBloodPrimeStates   .currentCellChanged.connect(self.do_treatment_states)
        self.tblDialysisStates     .currentCellChanged.connect(self.do_treatment_states)
        self.tblUFStates           .currentCellChanged.connect(self.do_treatment_states)
        self.tblTreatmentStopStates.currentCellChanged.connect(self.do_treatment_states)
        self.tblRinsebackStates    .currentCellChanged.connect(self.do_treatment_states)
        self.tblRecirculateStates  .currentCellChanged.connect(self.do_treatment_states)
        self.tblTreatmentEndStates .currentCellChanged.connect(self.do_treatment_states)
        self.tblSalineStates       .currentCellChanged.connect(self.do_treatment_states)
        self.tblHeparinStates      .currentCellChanged.connect(self.do_treatment_states)


    @Slot()
    def _init_widgets(self):
        """
        initializes the widgets' properties
        :return: none
        """
        self.tblSubMode            .setCurrentCell(TXStates.TREATMENT_START_STATE   .value  , 0)
        self.tblBloodPrimeStates   .setCurrentCell(0                                        , 0)
        self.tblDialysisStates     .setCurrentCell(DialStates.DIALYSIS_UF_STATE     .value  , 0)
        self.tblUFStates           .setCurrentCell(0                                        , 0)
        self.tblTreatmentStopStates.setCurrentCell(0                                        , 0)
        self.tblRinsebackStates    .setCurrentCell(0                                        , 0)
        self.tblRecirculateStates  .setCurrentCell(0                                        , 0)
        self.tblTreatmentEndStates .setCurrentCell(0                                        , 0)
        self.tblSalineStates       .setCurrentCell(0                                        , 0)
        self.tblHeparinStates      .setCurrentCell(0                                        , 0)

    @Slot()
    def do_treatment_states(self):
        """
        the slot for Treatment state changes
        :return: none
        """
        sub_mode              = int(self.tblSubMode            .verticalHeaderItem(self.tblSubMode            .currentRow()).text())
        blood_prime_state     = int(self.tblBloodPrimeStates   .verticalHeaderItem(self.tblBloodPrimeStates   .currentRow()).text())
        dialysis_state        = int(self.tblDialysisStates     .verticalHeaderItem(self.tblDialysisStates     .currentRow()).text())
        isolated_uf_state     = int(self.tblUFStates           .verticalHeaderItem(self.tblUFStates           .currentRow()).text())
        tx_stop_state         = int(self.tblTreatmentStopStates.verticalHeaderItem(self.tblTreatmentStopStates.currentRow()).text())
        tx_rinseback_state    = int(self.tblRinsebackStates    .verticalHeaderItem(self.tblRinsebackStates    .currentRow()).text())
        tx_recirculate_state  = int(self.tblRecirculateStates  .verticalHeaderItem(self.tblRecirculateStates  .currentRow()).text())
        tx_end_state          = int(self.tblTreatmentEndStates .verticalHeaderItem(self.tblTreatmentEndStates .currentRow()).text())
        tx_saline_state       = int(self.tblSalineStates       .verticalHeaderItem(self.tblSalineStates       .currentRow()).text())
        tx_heparin_state      = int(self.tblHeparinStates      .verticalHeaderItem(self.tblHeparinStates      .currentRow()).text())
    
        self.td_interface.td_tx_state(
            sub_mode             ,
            blood_prime_state    ,
            dialysis_state       ,
            isolated_uf_state    ,
            tx_stop_state        ,
            tx_rinseback_state   ,
            tx_recirculate_state ,
            tx_end_state         ,
            tx_saline_state      ,
            tx_heparin_state 
        )