###########################################################################
#
# Copyright (c) 2020-2024 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    td_proxy.py
#
# @author  (last)      Zoltan Miskolci
# @date    (last)      04-May-2026ks
# @author  (original)  Peter Lucia
# @date    (original)  02-Apr-2020
#
############################################################################

# Module imports
from logging import Logger

# Project imports
from leahi_dialin.common.generic_defs import DataTypes
from leahi_dialin.common.msg_defs import MsgIds
from leahi_dialin.common.override_templates import cmd_generic_override
from leahi_dialin.protocols.CAN import DenaliCanMessenger, DenaliChannels
from leahi_dialin.utils.abstract_classes import AbstractSubSystem
from leahi_dialin.utils.base import publish
from leahi_dialin.utils.conversions import integer_to_bytearray, float_to_bytearray


class TDProxy(AbstractSubSystem):
    """
    Dialysate Delivery (DD) Dialin API sub-class for TD proxy ( injection ) related commands.
    """

    def __init__(self, can_interface: DenaliCanMessenger, logger: Logger):
        """
        TDProxy constructor

        @param can_interface: the Denali CAN interface object
        """
        super().__init__()
        self.can_interface = can_interface
        self.logger = logger

        if self.can_interface is not None:
            self.can_interface.register_receiving_publication_function(channel_id = DenaliChannels.td_to_dd_ch_id,
                                                                       message_id = MsgIds.MSG_ID_DD_GEN_DIALYSATE_REQUEST_DATA.value,
                                                                       function = self._handler_dialysate_delivery_request_response)

        self.dialysate_delivery_request_start = 0
        self.dialysate_delivery_request_dial_rate = 0.0
        self.dialysate_delivery_request_uf_rate = 0.0
        self.dialysate_delivery_request_dial_temp = 0.0
        self.dialysate_delivery_request_bypass = 0
        self.dialysate_delivery_request_acid = 0
        self.dialysate_delivery_request_bicarb = 0
        self.dd_td_to_dd_request_response_timestamp = 0.0


    @publish(["dd_td_to_dd_request_response_timestamp",
              "dialysate_delivery_request_start","dialysate_delivery_request_dial_rate",
              "dialysate_delivery_request_uf_rate","dialysate_delivery_request_dial_temp",
              "dialysate_delivery_request_bypass","dialysate_delivery_request_acid",
              "dialysate_delivery_request_bicarb",
              ])
    def _handler_dialysate_delivery_request_response(self, message, timestamp=0.0):
        """
        Handles published dialysate delivery request response data messages.

        @param message:  published dialysate delivery request response data message
        @return: None
        """
        msg_list =[]
        msg_list.append(('self.dialysate_delivery_request_start',       DataTypes.U32))
        msg_list.append(('self.dialysate_delivery_request_dial_rate',   DataTypes.F32))
        msg_list.append(('self.dialysate_delivery_request_uf_rate',     DataTypes.F32))
        msg_list.append(('self.dialysate_delivery_request_dial_temp',   DataTypes.F32))
        msg_list.append(('self.dialysate_delivery_request_bypass',      DataTypes.U32))
        msg_list.append(('self.dialysate_delivery_request_acid',        DataTypes.U32))
        msg_list.append(('self.dialysate_delivery_request_bicarb',      DataTypes.U32))

        self.process_into_vars(decoder_list = msg_list,
                               message = message)

        self.dd_td_to_dd_request_response_timestamp = timestamp


    def cmd_td_send_dd_gen_dialysate_request(self, start: bool = 0, dial_rate: float = 0, uf_rate: float = 0,
                                             dial_temp: float = 0, bypass: int = 0, acid_conv_factor: float = 0.0, bicarb_conv_factor: float = 0.0, sodium: int = 0, bicarb_volume: int = 0):
        """
        Constructs and sends a TD gen dialysate request to the DD.

        @return: none
        """
        stt = integer_to_bytearray(start)
        dial = float_to_bytearray(dial_rate)
        uf = float_to_bytearray(uf_rate)
        temp = float_to_bytearray(dial_temp)
        byp = integer_to_bytearray(bypass)
        acd = float_to_bytearray(acid_conv_factor)
        bic = float_to_bytearray(bicarb_conv_factor)
        sod = integer_to_bytearray(sodium)
        bvl = integer_to_bytearray(bicarb_volume)
        payload = stt + dial + uf + temp + byp + acd + bic + sod + bvl

        cmd_generic_override(payload = payload,
                             reset = None,
                             channel_id = DenaliChannels.dialin_to_dd_ch_id,
                             msg_id = MsgIds.MSG_ID_DD_GEN_DIALYSATE_REQUEST_DATA,
                             entity_name = 'TD Gen Dialysate Request',
                             override_text = 'N/A',
                             logger = self.logger,
                             can_interface = self.can_interface)


    def cmd_td_send_dd_start_pre_gen_request(self, start: bool = 0, dialysate_rate: float = 0.0, dialysate_temp: float = 0.0,
                                             acid_conv_factor: float = 0.0, bicarb_conv_factor: float = 0.0, sodium: int = 0, bicarb_volume: int = 0):
        """
                Constructs and sends a TD start pre gen dialysate request to the DD.
        :param start: start and stop boolean
        :param dialysate_rate: dialysate rate in ml/min
        :param dialysate_temp: dialysate temp in c
        :param acid_type: acid type
        :param bicarb_type: bicarb type
        :return:
        """
        stt = integer_to_bytearray(start)
        dial = float_to_bytearray(dialysate_rate)
        temp = float_to_bytearray(dialysate_temp)
        acd = float_to_bytearray(acid_conv_factor)
        bic = float_to_bytearray(bicarb_conv_factor)
        sod = integer_to_bytearray(sodium)
        bvl = integer_to_bytearray(bicarb_volume)
        payload = stt + dial + temp + acd + bic + sod + bvl

        cmd_generic_override(payload = payload,
                             reset = None,
                             channel_id = DenaliChannels.dialin_to_dd_ch_id,
                             msg_id = MsgIds.MSG_ID_DD_PRE_GEN_DIALYSATE_REQUEST_DATA,
                             entity_name = 'TD Pre-Gen Dialysate Request',
                             override_text = 'N/A',
                             logger = self.logger,
                             can_interface = self.can_interface)