########################################################################### # # 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 pressure_occlusion.py # # @author (last) Zoltan Miskolci # @date (last) 05-May-2026 # @author (original) Peter Lucia # @date (original) 02-Apr-2020 # ############################################################################ # Module imports from logging import Logger # Project imports from leahi_dialin.common.constants import NO_RESET from leahi_dialin.common.generic_defs import DataTypes from leahi_dialin.common.msg_ids import MsgIds from leahi_dialin.common.override_templates import cmd_generic_broadcast_interval_override, cmd_generic_override from leahi_dialin.common import td_enum_repository from leahi_dialin.protocols.CAN import CanMessenger, CanChannels 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 TDPressureSensors(AbstractSubSystem): """ Treatment Delivery (TD) Dialin API sub-class for pressure related commands. """ def __init__(self, can_interface: CanMessenger, logger: Logger): """ TDPressureSensors constructor """ 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 = CanChannels.td_sync_broadcast_ch_id, message_id = MsgIds.MSG_ID_TD_PRESSURE_DATA.value, function = self._handler_pressure_sync) self.td_pressure_timestamp = 0.0 #: The timestamp of the latest message self.td_pressure_sensors = { td_enum_repository.TDPressureSensorNames.H2_PRES.name: { td_enum_repository.TDPressureSensorAttributes.PRES.name: 0.0, td_enum_repository.TDPressureSensorAttributes.LONG_FILTERED_PRESSURE.name: 0.0, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MIN.name: 0, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MAX.name: 0 }, td_enum_repository.TDPressureSensorNames.H14_PRES.name: { td_enum_repository.TDPressureSensorAttributes.PRES.name: 0.0, td_enum_repository.TDPressureSensorAttributes.LONG_FILTERED_PRESSURE.name: 0.0, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MIN.name: 0, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MAX.name: 0 }, td_enum_repository.TDPressureSensorNames.H23_PRES.name: { td_enum_repository.TDPressureSensorAttributes.PRES.name : 0.0 }, 'TMP_PRES': { td_enum_repository.TDPressureSensorAttributes.PRES.name: 0.0, td_enum_repository.TDPressureSensorAttributes.LONG_FILTERED_PRESSURE.name: 0.0, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MIN.name: 0, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MAX.name: 0 } } self.pressure_limits_state = 0 # ============================================================ Properties ============================================================ @property def td_pressure_sensors(self) -> dict: """ The current Pressure data """ return self._pressure_sensors @td_pressure_sensors.setter def td_pressure_sensors(self, value): self._pressure_sensors = value @property def pressure_limits_state(self) -> dict: """ Pressure limits state value """ return self._pressure_limits_state @pressure_limits_state.setter def pressure_limits_state(self, value): self._pressure_limits_state = value # ============================================================ Handlers ============================================================ @publish([ "msg_id_td_pressure_data", "td_pressure_sensors", "pressure_limits_state", "tmp_pressure", "tmp_long_filtered_pressure", "tmp_min_limit", "tmp_max_limit", "td_pressure_timestamp" ]) def _handler_pressure_sync(self, message, timestamp=0.0): """ Handles published pressure & occlusion data messages. Pressure data are captured for reference. @param message: published pressure & occlusion data message @return: none """ msg_list = [] msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H2_PRES.name, td_enum_repository.TDPressureSensorAttributes.PRES.name, DataTypes.F32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H14_PRES.name, td_enum_repository.TDPressureSensorAttributes.PRES.name, DataTypes.F32)) msg_list.append((type(self).pressure_limits_state, DataTypes.U32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H2_PRES.name, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MIN.name, DataTypes.S32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H2_PRES.name, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MAX.name, DataTypes.S32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H14_PRES.name, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MIN.name, DataTypes.S32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H14_PRES.name, td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MAX.name, DataTypes.S32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H2_PRES.name, td_enum_repository.TDPressureSensorAttributes.LONG_FILTERED_PRESSURE.name, DataTypes.F32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H14_PRES.name, td_enum_repository.TDPressureSensorAttributes.LONG_FILTERED_PRESSURE.name, DataTypes.F32)) msg_list.append((self.td_pressure_sensors, 'TMP_PRES', td_enum_repository.TDPressureSensorAttributes.PRES.name, DataTypes.F32)) msg_list.append((self.td_pressure_sensors, 'TMP_PRES', td_enum_repository.TDPressureSensorAttributes.LONG_FILTERED_PRESSURE.name, DataTypes.F32)) msg_list.append((self.td_pressure_sensors, 'TMP_PRES', td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MIN.name, DataTypes.F32)) msg_list.append((self.td_pressure_sensors, 'TMP_PRES', td_enum_repository.TDPressureSensorAttributes.PRES_LIMIT_MAX.name, DataTypes.F32)) msg_list.append((self.td_pressure_sensors, td_enum_repository.TDPressureSensorNames.H23_PRES.name, td_enum_repository.TDPressureSensorAttributes.PRES.name, DataTypes.F32)) self.process_into_vars(decoder_list = msg_list, message = message) self.td_pressure_timestamp = timestamp # ============================================================ Overrides and Requests ============================================================ def cmd_pressure_sensors_broadcast_interval_override(self, ms: int, reset: int = NO_RESET) -> int: """ Constructs and sends the measured pressure sensors broadcast interval override command Constraints: Must be logged into TD. Given interval must be non-zero and a multiple of the TD general task interval (50 ms). @param ms: integer - interval (in ms) to override with @param reset: integer - 1 to reset a previous override, 0 to override @return: 1 if successful, zero otherwise """ return cmd_generic_broadcast_interval_override( ms = ms, reset = reset, channel_id = CanChannels.dialin_to_td_ch_id, msg_id = MsgIds.MSG_ID_TD_PRESSURE_PUBLISH_INTERVAL_OVERRIDE_REQUEST, module_name = 'TD Pressure Sensors', logger = self.logger, can_interface = self.can_interface) def cmd_pressure_sensor_override(self, sensor: int, pressure: float, reset: int = NO_RESET) -> int: """ Constructs and sends the measured pressure sensor override command . Constraints: Must be logged into TD. @param sensor: int - sensor ID @param pressure: float - measured arterial or venous pressure (in mmHg) to override with @param reset: integer - 1 to reset a previous override, 0 to override @return: 1 if successful, zero otherwise """ rst = integer_to_bytearray(reset) sen = integer_to_bytearray(sensor) prs = float_to_bytearray(pressure) payload = rst + prs + sen sensor_name = td_enum_repository.TDPressureSensorNames(sensor).name.split('_')[0] return cmd_generic_override( payload = payload, reset = reset, channel_id = CanChannels.dialin_to_td_ch_id, msg_id = MsgIds.MSG_ID_TD_PRESSURE_OVERRIDE_REQUEST, entity_name = f'TD {sensor_name} Pressure Sensor pressure', override_text = str(prs) + " mmHg. ", logger = self.logger, can_interface = self.can_interface) def cmd_tmp_sensor_override(self, pressure: float, reset: int = NO_RESET) -> int: """ Constructs and sends the tmp sensor override command . Constraints: Must be logged into TD. @param pressure: float - measured pressure (in mmHg) to override with @param reset: integer - 1 to reset a previous override, 0 to override @return: 1 if successful, zero otherwise """ rst = integer_to_bytearray(reset) prs = float_to_bytearray(pressure) payload = rst + prs return cmd_generic_override( payload = payload, reset = reset, channel_id = CanChannels.dialin_to_td_ch_id, msg_id = MsgIds.MSG_ID_TD_TMP_PRESSURE_OVERRIDE_REQUEST, entity_name = 'TD TMP Pressure Sensor pressure', override_text = str(prs) + " mmHg. ", logger = self.logger, can_interface = self.can_interface) def cmd_baro_sensor_override(self, pressure: float, reset: int = NO_RESET) -> int: """ Constructs and sends the barometric sensor override command . Constraints: Must be logged into TD. @param pressure: float - measured pressure (in PSI) to override with @param reset: integer - 1 to reset a previous override, 0 to override @return: 1 if successful, zero otherwise """ rst = integer_to_bytearray(reset) prs = float_to_bytearray(pressure) payload = rst + prs return cmd_generic_override( payload = payload, reset = reset, channel_id = CanChannels.dialin_to_td_ch_id, msg_id = MsgIds.MSG_ID_TD_BARO_PRESSURE_OVERRIDE, entity_name = 'TD Barometric Pressure Sensor pressure', override_text = str(prs) + " PSI. ", logger = self.logger, can_interface = self.can_interface)