mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
229 lines
No EOL
9.1 KiB
Python
229 lines
No EOL
9.1 KiB
Python
import threading
|
|
import arq_session
|
|
import helpers
|
|
from modem_frametypes import FRAME_TYPE
|
|
from codec2 import FREEDV_MODE
|
|
from enum import Enum
|
|
|
|
class IRS_State(Enum):
|
|
NEW = 0
|
|
OPEN_ACK_SENT = 1
|
|
INFO_ACK_SENT = 2
|
|
BURST_REPLY_SENT = 3
|
|
ENDED = 4
|
|
FAILED = 5
|
|
ABORTED = 6
|
|
|
|
class ARQSessionIRS(arq_session.ARQSession):
|
|
|
|
TIMEOUT_CONNECT = 55 #14.2
|
|
TIMEOUT_DATA = 60
|
|
|
|
STATE_TRANSITION = {
|
|
IRS_State.NEW: {
|
|
FRAME_TYPE.ARQ_SESSION_OPEN.value : 'send_open_ack',
|
|
FRAME_TYPE.ARQ_STOP.value: 'send_stop_ack'
|
|
},
|
|
IRS_State.OPEN_ACK_SENT: {
|
|
FRAME_TYPE.ARQ_SESSION_OPEN.value: 'send_open_ack',
|
|
FRAME_TYPE.ARQ_SESSION_INFO.value: 'send_info_ack',
|
|
FRAME_TYPE.ARQ_STOP.value: 'send_stop_ack'
|
|
|
|
},
|
|
IRS_State.INFO_ACK_SENT: {
|
|
FRAME_TYPE.ARQ_SESSION_INFO.value: 'send_info_ack',
|
|
FRAME_TYPE.ARQ_BURST_FRAME.value: 'receive_data',
|
|
FRAME_TYPE.ARQ_STOP.value: 'send_stop_ack'
|
|
|
|
},
|
|
IRS_State.BURST_REPLY_SENT: {
|
|
FRAME_TYPE.ARQ_BURST_FRAME.value: 'receive_data',
|
|
FRAME_TYPE.ARQ_STOP.value: 'send_stop_ack'
|
|
|
|
},
|
|
IRS_State.ENDED: {
|
|
FRAME_TYPE.ARQ_BURST_FRAME.value: 'receive_data',
|
|
FRAME_TYPE.ARQ_STOP.value: 'send_stop_ack'
|
|
|
|
},
|
|
IRS_State.FAILED: {
|
|
FRAME_TYPE.ARQ_BURST_FRAME.value: 'receive_data',
|
|
FRAME_TYPE.ARQ_STOP.value: 'send_stop_ack'
|
|
},
|
|
IRS_State.ABORTED: {
|
|
FRAME_TYPE.ARQ_STOP.value: 'send_stop_ack',
|
|
FRAME_TYPE.ARQ_SESSION_OPEN.value: 'send_open_ack',
|
|
FRAME_TYPE.ARQ_SESSION_INFO.value: 'send_info_ack',
|
|
FRAME_TYPE.ARQ_BURST_FRAME.value: 'receive_data',
|
|
},
|
|
}
|
|
|
|
def __init__(self, config: dict, modem, dxcall: str, session_id: int):
|
|
super().__init__(config, modem, dxcall)
|
|
|
|
self.id = session_id
|
|
self.dxcall = dxcall
|
|
self.version = 1
|
|
|
|
self.state = IRS_State.NEW
|
|
|
|
self.total_length = 0
|
|
self.total_crc = ''
|
|
self.received_data = None
|
|
self.received_bytes = 0
|
|
self.received_crc = None
|
|
|
|
self.transmitted_acks = 0
|
|
|
|
self.abort = False
|
|
|
|
def set_decode_mode(self):
|
|
self.modem.demodulator.set_decode_mode(self.get_mode_by_speed_level(self.speed_level))
|
|
|
|
def all_data_received(self):
|
|
return self.total_length == self.received_bytes
|
|
|
|
def final_crc_matches(self) -> bool:
|
|
match = self.total_crc == helpers.get_crc_32(bytes(self.received_data)).hex()
|
|
return match
|
|
|
|
def transmit_and_wait(self, frame, timeout, mode):
|
|
self.event_frame_received.clear()
|
|
self.transmit_frame(frame, mode)
|
|
self.log(f"Waiting {timeout} seconds...")
|
|
if not self.event_frame_received.wait(timeout):
|
|
self.log("Timeout waiting for ISS. Session failed.")
|
|
self.set_state(IRS_State.FAILED)
|
|
self.event_manager.send_arq_session_finished(False, self.id, self.dxcall, self.total_length, False, self.state.name)
|
|
|
|
def launch_transmit_and_wait(self, frame, timeout, mode):
|
|
thread_wait = threading.Thread(target = self.transmit_and_wait,
|
|
args = [frame, timeout, mode])
|
|
thread_wait.start()
|
|
|
|
def send_open_ack(self, open_frame):
|
|
self.event_manager.send_arq_session_new(
|
|
False, self.id, self.dxcall, 0, self.state.name)
|
|
ack_frame = self.frame_factory.build_arq_session_open_ack(
|
|
self.id,
|
|
self.dxcall,
|
|
self.version,
|
|
self.snr[0], flag_abort=self.abort)
|
|
self.launch_transmit_and_wait(ack_frame, self.TIMEOUT_CONNECT, mode=FREEDV_MODE.signalling)
|
|
if not self.abort:
|
|
self.set_state(IRS_State.OPEN_ACK_SENT)
|
|
|
|
def send_info_ack(self, info_frame):
|
|
# Get session info from ISS
|
|
self.received_data = bytearray(info_frame['total_length'])
|
|
self.total_length = info_frame['total_length']
|
|
self.total_crc = info_frame['total_crc']
|
|
self.dx_snr.append(info_frame['snr'])
|
|
|
|
self.log(f"New transfer of {self.total_length} bytes")
|
|
self.event_manager.send_arq_session_new(False, self.id, self.dxcall, self.total_length, self.state.name)
|
|
|
|
self.calibrate_speed_settings()
|
|
self.set_decode_mode()
|
|
|
|
info_ack = self.frame_factory.build_arq_session_info_ack(
|
|
self.id, self.total_crc, self.snr[0],
|
|
self.speed_level, self.frames_per_burst, flag_abort=self.abort)
|
|
self.launch_transmit_and_wait(info_ack, self.TIMEOUT_CONNECT, mode=FREEDV_MODE.signalling)
|
|
if not self.abort:
|
|
self.set_state(IRS_State.INFO_ACK_SENT)
|
|
|
|
|
|
def process_incoming_data(self, frame):
|
|
if frame['offset'] != self.received_bytes:
|
|
self.log(f"Discarding data offset {frame['offset']}")
|
|
return False
|
|
|
|
remaining_data_length = self.total_length - self.received_bytes
|
|
|
|
# Is this the last data part?
|
|
if remaining_data_length <= len(frame['data']):
|
|
# we only want the remaining length, not the entire frame data
|
|
data_part = frame['data'][:remaining_data_length]
|
|
else:
|
|
# we want the entire frame data
|
|
data_part = frame['data']
|
|
|
|
self.received_data[frame['offset']:] = data_part
|
|
self.received_bytes += len(data_part)
|
|
self.log(f"Received {self.received_bytes}/{self.total_length} bytes")
|
|
self.event_manager.send_arq_session_progress(
|
|
False, self.id, self.dxcall, self.received_bytes, self.total_length, self.state.name)
|
|
|
|
return True
|
|
|
|
def receive_data(self, burst_frame):
|
|
self.process_incoming_data(burst_frame)
|
|
self.calibrate_speed_settings()
|
|
|
|
if not self.all_data_received():
|
|
ack = self.frame_factory.build_arq_burst_ack(
|
|
self.id, self.received_bytes,
|
|
self.speed_level, self.frames_per_burst, self.snr[0], flag_abort=self.abort)
|
|
|
|
self.set_decode_mode()
|
|
|
|
# increase ack counter
|
|
# self.transmitted_acks += 1
|
|
self.set_state(IRS_State.BURST_REPLY_SENT)
|
|
self.launch_transmit_and_wait(ack, self.TIMEOUT_DATA, mode=FREEDV_MODE.signalling)
|
|
return
|
|
|
|
if self.final_crc_matches():
|
|
self.log("All data received successfully!")
|
|
ack = self.frame_factory.build_arq_burst_ack(self.id,
|
|
self.received_bytes,
|
|
self.speed_level,
|
|
self.frames_per_burst,
|
|
self.snr[0],
|
|
flag_final=True,
|
|
flag_checksum=True)
|
|
self.transmit_frame(ack, mode=FREEDV_MODE.signalling)
|
|
self.log("ACK sent")
|
|
self.set_state(IRS_State.ENDED)
|
|
self.event_manager.send_arq_session_finished(
|
|
False, self.id, self.dxcall, self.total_length, True, self.state.name, data=self.received_data)
|
|
|
|
else:
|
|
|
|
ack = self.frame_factory.build_arq_burst_ack(self.id,
|
|
self.received_bytes,
|
|
self.speed_level,
|
|
self.frames_per_burst,
|
|
self.snr[0],
|
|
flag_final=True,
|
|
flag_checksum=False)
|
|
self.transmit_frame(ack, mode=FREEDV_MODE.signalling)
|
|
self.log("CRC fail at the end of transmission!")
|
|
self.set_state(IRS_State.FAILED)
|
|
self.event_manager.send_arq_session_finished(
|
|
False, self.id, self.dxcall, self.total_length, False, self.state.name)
|
|
|
|
|
|
def calibrate_speed_settings(self):
|
|
self.speed_level = 0 # for now stay at lowest speed level
|
|
return
|
|
# if we have two ACKS, then consider increasing speed level
|
|
if self.transmitted_acks >= 2:
|
|
self.transmitted_acks = 0
|
|
new_speed_level = min(self.speed_level + 1, len(self.SPEED_LEVEL_DICT) - 1)
|
|
|
|
# check first if the next mode supports the actual snr
|
|
if self.snr[0] >= self.SPEED_LEVEL_DICT[new_speed_level]["min_snr"]:
|
|
self.speed_level = new_speed_level
|
|
|
|
def abort_transmission(self):
|
|
self.log(f"Aborting transmission... setting abort flag")
|
|
self.abort = True
|
|
|
|
def send_stop_ack(self, stop_frame):
|
|
stop_ack = self.frame_factory.build_arq_stop_ack(self.id)
|
|
self.launch_transmit_and_wait(stop_ack, self.TIMEOUT_CONNECT, mode=FREEDV_MODE.signalling)
|
|
self.set_state(IRS_State.ABORTED)
|
|
self.event_manager.send_arq_session_finished(
|
|
False, self.id, self.dxcall, self.total_length, False, self.state.name) |