import struct import datetime import time from collections import OrderedDict from ..common.msg_defs import MsgIds, MsgFieldPositions from ..protocols.CAN import (DenaliMessage, DenaliChannels) from ..utils.base import _AbstractSubSystem, _publish, DialinEnum from ..utils.nv_ops_utils import NVOpsUtils from logging import Logger from enum import unique @unique class ServiceLocation(DialinEnum): SERVICE_LOCATION_FACTORY = 0 SERVICE_LOCATION_FIELD = 1 class HDServiceNVRecords(_AbstractSubSystem): """ Hemodialysis Device (HD) Dialin API sub-class for service record commands. """ _RECORD_SPECS_BYTES = 12 _DEFAULT_SERVICE_LOCATION = ServiceLocation.SERVICE_LOCATION_FACTORY.value _DEFAULT_TIME_VALUE = 0 _DEFAULT_CRC_VALUE = 0 # Maximum allowed bytes to be written to RTC RAM _RTC_RAM_MAX_BYTES_TO_WRITE = 64 # Delay in between each payload transfer _PAYLOAD_TRANSFER_DELAY_S = 0.2 _DIALIN_RECORD_UPDATE_DELAY_S = 0.2 def __init__(self, can_interface, logger: Logger): """ @param can_interface: Denali CAN Messenger object """ super().__init__() self.can_interface = can_interface self.logger = logger self.current_message = 0 self.total_messages = 0 self.received_msg_length = 0 self._is_getting_service_in_progress = False self._write_fw_data_to_excel = True self.service_data = 0 self._raw_service_record = [] self._utilities = NVOpsUtils(logger=self.logger) self.hd_service_record = self._prepare_hd_service_record() if self.can_interface is not None: channel_id = DenaliChannels.hd_to_dialin_ch_id msg_id = MsgIds.MSG_ID_HD_SEND_SERVICE_RECORD.value self.can_interface.register_receiving_publication_function(channel_id, msg_id, self._handler_hd_service_sync) def cmd_request_hd_service_record(self): """ Handles getting HD service record from firmware. @return: 1 upon success, False otherwise """ if self._is_getting_service_in_progress is not True: self._is_getting_service_in_progress = True # Clear the list for the next call self._raw_service_record.clear() message = DenaliMessage.build_message(channel_id=DenaliChannels.dialin_to_hd_ch_id, message_id=MsgIds.MSG_ID_HD_GET_SERVICE_RECORD.value) self.logger.debug('Getting HD service record') received_message = self.can_interface.send(message) # If there is content... if received_message is not None: # response payload is OK or not OK return received_message['message'][DenaliMessage.PAYLOAD_START_INDEX] else: self.logger.debug("Timeout!!!!") return False self.logger.debug("Request cancelled: an existing request is in progress.") return False def _handler_hd_service_sync(self, message): """ Handles published HD system record messages. HD system records are captured for processing and updating the HD system record. @param message: published HD system record data message @return: None """ curr = struct.unpack('i', bytearray( message['message'][MsgFieldPositions.START_POS_FIELD_1:MsgFieldPositions.END_POS_FIELD_1]))[0] total = struct.unpack('i', bytearray( message['message'][MsgFieldPositions.START_POS_FIELD_2:MsgFieldPositions.END_POS_FIELD_2]))[0] length = struct.unpack('i', bytearray( message['message'][MsgFieldPositions.START_POS_FIELD_3:MsgFieldPositions.END_POS_FIELD_3]))[0] self.current_message = curr self.total_messages = total self.received_msg_length = length # The end of calibration_record record payload is from the start index + 12 bytes for the current message +total # messages + the length of calibration_record. The rest is the CAN messaging CRC that is not needed # to be kept end_of_data_index = MsgFieldPositions.START_POS_FIELD_1 + self._RECORD_SPECS_BYTES + self.received_msg_length # Get the data only and not specs of it (i.e current message number) self.service_data = message['message'][MsgFieldPositions.START_POS_FIELD_1:end_of_data_index] # Continue getting calibration_record records until the all the calibration_record messages are received. # Concatenate the calibration_record records to each other if self.current_message <= self.total_messages: self._raw_service_record += (message['message'][MsgFieldPositions.START_POS_FIELD_1 + self._RECORD_SPECS_BYTES:end_of_data_index]) if self.current_message == self.total_messages: # Done with receiving the messages self._is_getting_service_in_progress = False # If all the messages have been received, call another function to process the raw data self._utilities.process_received_record_from_fw(self.hd_service_record, self._raw_service_record) self._handler_received_complete_hd_service_record() @_publish(["hd_service_record"]) def _handler_received_complete_hd_service_record(self): """ Publishes the received service record @return: None """ self.logger.debug("Received a complete hd service record.") def cmd_set_hd_service_record(self, hd_service_record: OrderedDict): """ Handles updating the HD system and sends it to FW. @param hd_service_record: (OrderedDict) the hd service record to be sent @return: True upon success, False otherwise """ self.logger.debug('Setting HD service record') record_packets = self._utilities.prepare_record_to_send_to_fw(hd_service_record) # Update all the data packets with the last message count since is the number of messages that firmware # should receive for packet in record_packets: # Sleep to let the firmware receive and process the data time.sleep(self._PAYLOAD_TRANSFER_DELAY_S) # Convert the list packet to a bytearray payload = b''.join(packet) message = DenaliMessage.build_message(channel_id=DenaliChannels.dialin_to_hd_ch_id, message_id=MsgIds.MSG_ID_HD_SET_SERVICE_RECORD.value, payload=payload) received_message = self.can_interface.send(message) # If there is no content... if received_message is None: self.logger.debug("Timeout!!!!") return False self.logger.debug("Finished sending HD service record.") return True def _prepare_hd_service_record(self): """ Handles assembling the sub dictionaries of each group to make the main HD service record. @return: (OrderedDict) an assembled hd service record """ result = OrderedDict() groups_byte_size = 0 # create a list of the functions of the sub dictionaries functions = [self._prepare_service_record()] for function in functions: # Update the groups bytes size so far to be use to padding later groups_byte_size += function[1] # Update the calibration record result.update(function[0]) # Build the CRC of the main calibration_record record record_crc = OrderedDict({'crc': ['