"""
###########################################################################
#
# Copyright (c) 2020-2023 Diality Inc. - All Rights Reserved.
#
# THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
# WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
#
# @file    loader.py
#
# @author  (last)      Peter Lucia
# @date    (last)      22-Jan-2021
# @author  (original)  Behrouz NematiPour
# @date    (original)  23-Nov-2020
#
############################################################################
    The Ultrafiltration ui loader
"""
import os
from simulator.dynamicloader import DynamicLoader
from PySide2 import QtWidgets
from PySide2.QtCore import Slot
from dialin.ui import TXStates


class Loader(DynamicLoader):
    """
    The Ultrafiltration ui loader
    """
    tblSubMode: QtWidgets.QTableWidget
    tblUFStates: QtWidgets.QTableWidget
    tblSalineStates: QtWidgets.QTableWidget
    tblHeparinStates: QtWidgets.QTableWidget

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

    def _init_loader(self):
        """
        finds and creates widgets
        :return: none
        """
        self.tblSubMode = self.find_table_widget('tblSubMode')
        self.tblUFStates = self.find_table_widget('tblUFStates')
        self.tblSalineStates = self.find_table_widget('tblSalineStates')
        self.tblHeparinStates = self.find_table_widget('tblHeparinStates')

    def _init_connections(self):
        """
        initializes the widgets connections
        :return: none
        """
        self.tblSubMode.cellClicked.connect(self.do_treatment_states)
        self.tblUFStates.cellClicked.connect(self.do_treatment_states)
        self.tblSalineStates.cellClicked.connect(self.do_treatment_states)
        self.tblHeparinStates.cellClicked.connect(self.do_treatment_states)
        # apply/send the initial states
        self.do_treatment_states()

    def _init_widgets(self):
        """
        initializes the widgets' properties
        :return: none
        """
        self.tblSubMode.setCurrentCell(TXStates.TREATMENT_DIALYSIS_STATE, 0)
        self.tblUFStates.setCurrentCell(TXStates.UF_OFF_STATE, 0)
        self.tblSalineStates.setCurrentCell(TXStates.SALINE_BOLUS_STATE_IDLE, 0)
        self.tblHeparinStates.setCurrentCell(TXStates.HEPARIN_STATE_OFF, 0)

    @Slot()
    def do_treatment_states(self):
        """
        the slot for saline bolus state change
        :return: none
        """
        sub_mode = int(self.tblSubMode.verticalHeaderItem(self.tblSubMode.currentRow()).text())
        uf_state = int(self.tblUFStates.verticalHeaderItem(self.tblUFStates.currentRow()).text())
        saline = int(self.tblSalineStates.verticalHeaderItem(self.tblSalineStates.currentRow()).text())
        heparin = int(self.tblHeparinStates.verticalHeaderItem(self.tblHeparinStates.currentRow()).text())
        self.hd_simulator.cmd_set_treatment_states_data(sub_mode, uf_state, saline, heparin)