"""
    The Pre-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

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


class Loader(DynamicLoader):
    """
    The Pre-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.tbSend         = self.find_widget(QtWidgets.QToolButton    , 'tbSend'      )

        self.tblSubMode     = self.find_widget(QtWidgets.QTableWidget   , 'tblSubMode'  )


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

        self.tblSubMode     .currentCellChanged.connect(self.do_pre_treatment_states)


    @Slot()
    def _init_widgets(self):
        """
        initializes the widgets' properties
        :return: none
        """
        self.tblSubMode .setCurrentCell(0, 0)


    @Slot()
    def do_pre_treatment_states(self):
        """
        the slot for Treatment state changes
        :return: none
        """
        sub_mode              = int(self.tblSubMode            .verticalHeaderItem(self.tblSubMode            .currentRow()).text())

        self.td_interface.td_pre_tx_state(
            sub_mode    ,
            0           ,
            0           ,
            0           ,
            0           ,
            0           ,       
            0           ,
            0         
        )