FreeDATA/modem/arq_session_iss.py

177 lines
6.8 KiB
Python
Raw Normal View History

2023-12-05 14:40:04 +00:00
import threading
import data_frame_factory
import queue
import random
2023-12-05 17:50:39 +00:00
from codec2 import FREEDV_MODE
2023-12-14 16:29:04 +00:00
from modem_frametypes import FRAME_TYPE
2023-12-05 17:50:39 +00:00
import arq_session
2023-12-12 21:05:32 +00:00
import helpers
2023-12-19 14:01:08 +00:00
from enum import Enum
2023-12-05 14:40:04 +00:00
2023-12-19 14:01:08 +00:00
class ISS_State(Enum):
NEW = 0
OPEN_SENT = 1
INFO_SENT = 2
BURST_SENT = 3
ENDED = 4
FAILED = 5
2023-12-29 18:25:59 +00:00
ABORTING = 6 # state while running abort sequence and waiting for stop ack
ABORTED = 7 # stop ack received
2023-12-05 14:40:04 +00:00
2023-12-19 14:01:08 +00:00
class ARQSessionISS(arq_session.ARQSession):
2023-12-05 14:40:04 +00:00
RETRIES_CONNECT = 10
# DJ2LS: 3 seconds seems to be too small for radios with a too slow PTT toggle time
# DJ2LS: 3.5 seconds is working well WITHOUT a channel busy detection delay
2023-12-30 20:47:16 +00:00
TIMEOUT_CHANNEL_BUSY = 2
TIMEOUT_CONNECT_ACK = 3.5 + TIMEOUT_CHANNEL_BUSY
TIMEOUT_TRANSFER = 3.5 + TIMEOUT_CHANNEL_BUSY
TIMEOUT_STOP_ACK = 3.5 + TIMEOUT_CHANNEL_BUSY
2023-12-05 14:40:04 +00:00
2023-12-14 16:29:04 +00:00
STATE_TRANSITION = {
2023-12-19 14:01:08 +00:00
ISS_State.OPEN_SENT: {
2023-12-14 16:29:04 +00:00
FRAME_TYPE.ARQ_SESSION_OPEN_ACK.value: 'send_info',
},
2023-12-19 14:01:08 +00:00
ISS_State.INFO_SENT: {
2023-12-14 16:29:04 +00:00
FRAME_TYPE.ARQ_SESSION_OPEN_ACK.value: 'send_info',
FRAME_TYPE.ARQ_SESSION_INFO_ACK.value: 'send_data',
},
2023-12-19 14:01:08 +00:00
ISS_State.BURST_SENT: {
2023-12-14 16:29:04 +00:00
FRAME_TYPE.ARQ_SESSION_INFO_ACK.value: 'send_data',
FRAME_TYPE.ARQ_BURST_ACK.value: 'send_data',
},
2023-12-21 14:05:22 +00:00
ISS_State.FAILED:{
2023-12-29 18:25:59 +00:00
FRAME_TYPE.ARQ_STOP_ACK.value: 'transmission_aborted'
2023-12-24 12:20:51 +00:00
},
ISS_State.ABORTING: {
FRAME_TYPE.ARQ_STOP_ACK.value: 'transmission_aborted',
2023-12-24 12:20:51 +00:00
},
ISS_State.ABORTED: {
FRAME_TYPE.ARQ_STOP_ACK.value: 'transmission_aborted',
2023-12-21 14:05:22 +00:00
}
2023-12-14 16:29:04 +00:00
}
def __init__(self, config: dict, modem, dxcall: str, data: bytearray):
super().__init__(config, modem, dxcall)
2023-12-05 14:40:04 +00:00
self.data = data
2023-12-14 16:29:04 +00:00
self.data_crc = ''
2023-12-05 14:40:04 +00:00
2023-12-14 16:29:04 +00:00
self.confirmed_bytes = 0
2023-12-05 14:40:04 +00:00
2023-12-19 14:01:08 +00:00
self.state = ISS_State.NEW
2023-12-14 16:29:04 +00:00
self.id = self.generate_id()
2023-12-05 14:40:04 +00:00
self.frame_factory = data_frame_factory.DataFrameFactory(self.config)
def generate_id(self):
return random.randint(1,255)
2023-12-15 13:41:11 +00:00
def transmit_wait_and_retry(self, frame_or_burst, timeout, retries, mode):
2023-12-29 18:25:59 +00:00
while retries > 0:
2023-12-15 15:40:05 +00:00
self.event_frame_received = threading.Event()
2023-12-14 16:29:04 +00:00
if isinstance(frame_or_burst, list): burst = frame_or_burst
else: burst = [frame_or_burst]
for f in burst:
2023-12-15 13:41:11 +00:00
self.transmit_frame(f, mode)
2023-12-19 22:32:04 +00:00
self.event_frame_received.clear()
2023-12-14 16:57:58 +00:00
self.log(f"Waiting {timeout} seconds...")
2023-12-14 16:29:04 +00:00
if self.event_frame_received.wait(timeout):
2023-12-14 16:57:58 +00:00
return
self.log("Timeout!")
2023-12-05 14:40:04 +00:00
retries = retries - 1
2023-12-24 12:20:51 +00:00
2023-12-19 14:01:08 +00:00
self.set_state(ISS_State.FAILED)
2023-12-20 18:30:38 +00:00
self.transmission_failed()
2023-12-05 14:40:04 +00:00
2023-12-15 13:41:11 +00:00
def launch_twr(self, frame_or_burst, timeout, retries, mode):
twr = threading.Thread(target = self.transmit_wait_and_retry, args=[frame_or_burst, timeout, retries, mode])
2023-12-14 16:29:04 +00:00
twr.start()
2023-12-05 14:40:04 +00:00
2023-12-14 16:29:04 +00:00
def start(self):
session_open_frame = self.frame_factory.build_arq_session_open(self.dxcall, self.id)
2023-12-16 09:28:30 +00:00
self.launch_twr(session_open_frame, self.TIMEOUT_CONNECT_ACK, self.RETRIES_CONNECT, mode=FREEDV_MODE.signalling)
2023-12-19 14:01:08 +00:00
self.set_state(ISS_State.OPEN_SENT)
2023-12-12 21:05:32 +00:00
2023-12-14 16:29:04 +00:00
def set_speed_and_frames_per_burst(self, frame):
self.speed_level = frame['speed_level']
self.log(f"Speed level set to {self.speed_level}")
self.frames_per_burst = frame['frames_per_burst']
self.log(f"Frames per burst set to {self.frames_per_burst}")
2023-12-15 15:22:38 +00:00
def send_info(self, frame):
2023-12-12 21:05:32 +00:00
info_frame = self.frame_factory.build_arq_session_info(self.id, len(self.data),
helpers.get_crc_32(self.data),
2023-12-14 16:29:04 +00:00
self.snr[0])
2023-12-16 16:21:07 +00:00
2023-12-16 09:28:30 +00:00
self.launch_twr(info_frame, self.TIMEOUT_CONNECT_ACK, self.RETRIES_CONNECT, mode=FREEDV_MODE.signalling)
2023-12-19 14:01:08 +00:00
self.set_state(ISS_State.INFO_SENT)
2023-12-12 21:05:32 +00:00
2023-12-14 16:29:04 +00:00
def send_data(self, irs_frame):
2023-12-24 12:20:51 +00:00
2023-12-14 16:29:04 +00:00
self.set_speed_and_frames_per_burst(irs_frame)
2023-12-05 14:40:04 +00:00
2023-12-14 16:29:04 +00:00
if 'offset' in irs_frame:
self.confirmed_bytes = irs_frame['offset']
self.log(f"IRS confirmed {self.confirmed_bytes}/{len(self.data)} bytes")
2023-12-19 14:35:07 +00:00
self.event_manager.send_arq_session_progress(
2023-12-25 12:26:51 +00:00
True, self.id, self.dxcall, self.confirmed_bytes, len(self.data), self.state.name)
2023-12-05 14:40:04 +00:00
2023-12-28 21:27:49 +00:00
if irs_frame["flag"]["ABORT"]:
self.transmission_aborted(irs_frame)
2023-12-28 21:27:49 +00:00
return
2023-12-20 18:30:38 +00:00
if irs_frame["flag"]["FINAL"]:
if self.confirmed_bytes == len(self.data) and irs_frame["flag"]["CHECKSUM"]:
self.transmission_ended(irs_frame)
2023-12-20 18:30:38 +00:00
return
else:
2023-12-24 12:20:51 +00:00
self.transmission_failed()
2023-12-20 18:30:38 +00:00
return
2023-12-20 15:43:08 +00:00
2023-12-14 16:29:04 +00:00
payload_size = self.get_data_payload_size()
burst = []
for f in range(0, self.frames_per_burst):
offset = self.confirmed_bytes
payload = self.data[offset : offset + payload_size]
2023-12-12 21:33:17 +00:00
data_frame = self.frame_factory.build_arq_burst_frame(
2023-12-14 21:53:32 +00:00
self.SPEED_LEVEL_DICT[self.speed_level]["mode"],
2023-12-14 16:29:04 +00:00
self.id, self.confirmed_bytes, payload)
burst.append(data_frame)
2023-12-15 15:22:38 +00:00
self.launch_twr(burst, self.TIMEOUT_TRANSFER, self.RETRIES_CONNECT, mode='auto')
2023-12-19 14:01:08 +00:00
self.set_state(ISS_State.BURST_SENT)
2023-12-20 18:30:38 +00:00
def transmission_ended(self, irs_frame):
2023-12-28 21:27:49 +00:00
# final function for sucessfully ended transmissions
2023-12-20 18:30:38 +00:00
self.set_state(ISS_State.ENDED)
self.log(f"All data transfered! flag_final={irs_frame['flag']['FINAL']}, flag_checksum={irs_frame['flag']['CHECKSUM']}")
2023-12-25 12:26:51 +00:00
self.event_manager.send_arq_session_finished(True, self.id, self.dxcall, len(self.data),True, self.state.name)
2023-12-20 18:30:38 +00:00
2023-12-24 15:15:37 +00:00
def transmission_failed(self, irs_frame=None):
2023-12-28 21:27:49 +00:00
# final function for failed transmissions
2023-12-20 18:30:38 +00:00
self.set_state(ISS_State.FAILED)
2023-12-24 12:20:51 +00:00
self.log(f"Transmission failed!")
2023-12-25 12:26:51 +00:00
self.event_manager.send_arq_session_finished(True, self.id, self.dxcall, len(self.data),False, self.state.name)
2023-12-21 14:05:22 +00:00
2023-12-24 12:20:51 +00:00
def abort_transmission(self, irs_frame=None):
2023-12-28 21:27:49 +00:00
# function for starting the abort sequence
2023-12-25 12:26:51 +00:00
self.log(f"aborting transmission...")
2023-12-29 18:25:59 +00:00
self.set_state(ISS_State.ABORTING)
2023-12-25 12:26:51 +00:00
self.event_manager.send_arq_session_finished(
2023-12-28 21:27:49 +00:00
True, self.id, self.dxcall, len(self.data), False, self.state.name)
# break actual retries
2023-12-29 18:25:59 +00:00
self.event_frame_received.set()
2023-12-28 21:27:49 +00:00
# start with abort sequence
2023-12-24 12:20:51 +00:00
self.send_stop()
2023-12-21 14:44:54 +00:00
def send_stop(self):
2023-12-21 14:05:22 +00:00
stop_frame = self.frame_factory.build_arq_stop(self.id)
2023-12-29 18:25:59 +00:00
self.launch_twr(stop_frame, self.TIMEOUT_STOP_ACK, self.RETRIES_CONNECT, mode=FREEDV_MODE.signalling)
2023-12-24 12:20:51 +00:00
2023-12-28 21:27:49 +00:00
def transmission_aborted(self, irs_frame):
2023-12-24 12:20:51 +00:00
self.log("session aborted")
2023-12-25 12:26:51 +00:00
self.set_state(ISS_State.ABORTED)
self.event_manager.send_arq_session_finished(
2023-12-28 21:27:49 +00:00
True, self.id, self.dxcall, len(self.data), False, self.state.name)