mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
127 lines
3.7 KiB
Python
127 lines
3.7 KiB
Python
import helpers
|
|
from event_manager import EventManager
|
|
from state_manager import StateManager
|
|
from queue import Queue
|
|
import structlog
|
|
import time, uuid
|
|
from codec2 import FREEDV_MODE
|
|
|
|
class FrameHandler():
|
|
|
|
def __init__(self, name: str, config, states: StateManager, event_manager: EventManager,
|
|
tx_frame_queue: Queue) -> None:
|
|
|
|
self.name = name
|
|
self.config = config
|
|
self.states = states
|
|
self.event_manager = event_manager
|
|
self.tx_frame_queue = tx_frame_queue
|
|
self.logger = structlog.get_logger("Frame Handler")
|
|
|
|
self.details = {
|
|
'frame' : None,
|
|
'snr' : 0,
|
|
'freq_offset': 0,
|
|
'freedv_inst': None,
|
|
'bytes_per_frame': 0
|
|
}
|
|
|
|
|
|
|
|
def add_to_activity_list(self):
|
|
frame = self.details['frame']
|
|
|
|
activity = {
|
|
"direction": "received",
|
|
"snr": self.details['snr'],
|
|
"freq_offset": self.details['freq_offset'],
|
|
"activity_type": frame["frame_type"]
|
|
}
|
|
if "origin" in frame:
|
|
activity["origin"] = frame["origin"]
|
|
|
|
if "destination" in frame:
|
|
activity["destination"] = frame["destination"]
|
|
|
|
if "gridsquare" in frame:
|
|
activity["gridsquare"] = frame["gridsquare"]
|
|
|
|
if "session_id" in frame:
|
|
activity["session_id"] = frame["session_id"]
|
|
|
|
self.states.add_activity(activity)
|
|
|
|
|
|
def add_to_heard_stations(self):
|
|
frame = self.details['frame']
|
|
|
|
if 'origin' not in frame:
|
|
return
|
|
|
|
dxgrid = frame['gridsquare'] if 'gridsquare' in frame else "------"
|
|
helpers.add_to_heard_stations(
|
|
frame['origin'],
|
|
dxgrid,
|
|
self.name,
|
|
self.details['snr'],
|
|
self.details['freq_offset'],
|
|
self.states.radio_frequency,
|
|
self.states.heard_stations,
|
|
)
|
|
|
|
def make_event(self):
|
|
event = {
|
|
"freedata": "modem-message",
|
|
"received": self.details['frame']['frame_type'],
|
|
"uuid": str(uuid.uuid4()),
|
|
"timestamp": int(time.time()),
|
|
"mycallsign": self.config['STATION']['mycall'],
|
|
"snr": str(self.details['snr']),
|
|
}
|
|
if 'origin' in self.details['frame']:
|
|
event['dxcallsign'] = self.details['frame']['origin']
|
|
return event
|
|
|
|
def emit_event(self):
|
|
event_data = self.make_event()
|
|
self.event_manager.broadcast(event_data)
|
|
|
|
def make_modem_queue_item(self, mode, repeat, repeat_delay, frame):
|
|
return {
|
|
'mode': self.get_tx_mode(),
|
|
'repeat': repeat,
|
|
'repeat_delay': repeat_delay,
|
|
'frame': frame,
|
|
}
|
|
|
|
def get_tx_mode(self):
|
|
return (
|
|
FREEDV_MODE.fsk_ldpc_0.value
|
|
if self.config['MODEM']['enable_fsk']
|
|
else FREEDV_MODE.sig0.value
|
|
)
|
|
|
|
def transmit(self, frame):
|
|
tx_queue_item = self.make_modem_queue_item(self.get_tx_mode(), 1, 0, frame)
|
|
self.tx_frame_queue.put(tx_queue_item)
|
|
|
|
def follow_protocol(self):
|
|
pass
|
|
|
|
def log(self):
|
|
self.logger.info(f"[Frame Handler] Handling frame {self.details['frame']}")
|
|
pass
|
|
|
|
def handle(self, frame, snr, freq_offset, freedv_inst, bytes_per_frame):
|
|
self.details['frame'] = frame
|
|
self.details['snr'] = snr
|
|
self.details['freq_offset'] = freq_offset
|
|
self.details['freedv_inst'] = freedv_inst
|
|
self.details['bytes_per_frame'] = bytes_per_frame
|
|
|
|
self.log()
|
|
self.add_to_heard_stations()
|
|
self.add_to_activity_list()
|
|
self.emit_event()
|
|
self.follow_protocol()
|