FreeDATA/modem/state_manager.py

import time
import ujson as json
class STATES:
    def __init__(self, statequeue):

        self.modem_version = 0.0

        # state related settings
        self.statequeue = statequeue
        self.newstate = None
        self.last = time.time()

        # modem related states
        # not every state is needed to publish, yet
        # TODO can we reduce them?
        self.channel_busy = False
        self.channel_busy_slot = [False, False, False, False, False]
        self.is_codec2_traffic = False
        self.is_modem_running = False
        self.is_modem_busy = False
        self.is_beacon_running = False
        self.is_arq_state = False
        self.is_arq_session = False
        self.is_transmitting = False
        self.audio_dbfs = 0
        self.dxcallsign: bytes = b"ZZ9YY-0"
        self.dxgrid: bytes = b"------"
        self.heard_stations = []

        self.arq_session_state = 'disconnected'
        self.arq_speed_level = 0
        self.arq_total_bytes = 0
        self.arq_bits_per_second = 0
        self.arq_bytes_per_minute = 0
        self.arq_transmission_percent = 0
        self.arq_compression_factor = 0
        self.arq_speed_list = []
        self.arq_seconds_until_timeout = 0

        self.mesh_routing_table = []

        self.radio_frequency = 0
        self.radio_mode = None
        self.radio_bandwidth = 0
        self.radio_rf_power = 0
        self.radio_strength = 0

    def sendState (self):
        currentState = self.getAsJSON(False)
        self.statequeue.put(currentState)
        return currentState

    def sendStateUpdate (self):
        self.statequeue.put(self.newstate)
            

    def set(self, key, value):
        setattr(self, key, value)
        # only process data if changed
        # but also send an update if more than a 'updateCadence' second(s) has lapsed
        # Otherwise GUI can't tell if modem is active due to lack of state messages on startup
        new_state = self.getAsJSON(True)
        if new_state != self.newstate:
            self.newstate = new_state
            self.sendStateUpdate()
            

    def getAsJSON(self, isChangedState):
        
        msgtype = "state-change"
        if (not isChangedState):
            msgtype = "state"

        return json.dumps({
            "freedata-message": msgtype,
            "channel_busy": self.channel_busy,
            "is_codec2_traffic": self.is_codec2_traffic,
            "is_modem_running": self.is_modem_running,
            "is_beacon_running": self.is_beacon_running,

        })