mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
126 lines
4.2 KiB
Python
126 lines
4.2 KiB
Python
import threading
|
|
import frame_dispatcher
|
|
import modem
|
|
import structlog
|
|
import audio
|
|
import ujson as json
|
|
import explorer
|
|
import beacon
|
|
|
|
|
|
class SM:
|
|
def __init__(self, app):
|
|
self.log = structlog.get_logger("service")
|
|
|
|
self.modem = False
|
|
self.beacon = False
|
|
self.app = app
|
|
self.config = self.app.config_manager.read()
|
|
self.modem_fft = app.modem_fft
|
|
self.modem_service = app.modem_service
|
|
self.states = app.state_manager
|
|
self.event_manager = app.event_manager
|
|
|
|
|
|
runner_thread = threading.Thread(
|
|
target=self.runner, name="runner thread", daemon=True
|
|
)
|
|
runner_thread.start()
|
|
|
|
self.start_explorer_publishing()
|
|
|
|
def runner(self):
|
|
while True:
|
|
cmd = self.modem_service.get()
|
|
if cmd in ['start'] and not self.modem:
|
|
self.log.info("------------------ FreeDATA ------------------")
|
|
self.log.info("------------------ MODEM ------------------")
|
|
self.start_modem()
|
|
|
|
elif cmd in ['stop'] and self.modem:
|
|
self.stop_modem()
|
|
# we need to wait a bit for avoiding a portaudio crash
|
|
threading.Event().wait(0.5)
|
|
|
|
elif cmd in ['restart']:
|
|
self.stop_modem()
|
|
# we need to wait a bit for avoiding a portaudio crash
|
|
threading.Event().wait(0.5)
|
|
if self.start_modem():
|
|
self.event_manager.modem_restarted()
|
|
elif cmd in ['start_beacon']:
|
|
self.start_beacon()
|
|
|
|
elif cmd in ['stop_beacon']:
|
|
self.stop_beacon()
|
|
|
|
|
|
|
|
else:
|
|
self.log.warning("[SVC] modem command processing failed", cmd=cmd, state=self.states.is_modem_running)
|
|
|
|
|
|
def start_modem(self):
|
|
# read config
|
|
self.config = self.app.config_manager.read()
|
|
|
|
if self.states.is_modem_running:
|
|
self.log.warning("modem already running")
|
|
return False
|
|
|
|
# test audio devices
|
|
audio_test = self.test_audio()
|
|
|
|
if False in audio_test or None in audio_test or self.states.is_modem_running:
|
|
self.log.warning("starting modem failed", input_test=audio_test[0], output_test=audio_test[1])
|
|
self.states.set("is_modem_running", False)
|
|
self.event_manager.modem_failed()
|
|
return False
|
|
|
|
self.log.info("starting modem....")
|
|
self.modem = modem.RF(self.config, self.event_manager, self.modem_fft, self.modem_service, self.states)
|
|
|
|
self.frame_dispatcher = frame_dispatcher.DISPATCHER(self.config,
|
|
self.event_manager,
|
|
self.states,
|
|
self.modem)
|
|
self.frame_dispatcher.start()
|
|
|
|
self.event_manager.modem_started()
|
|
self.states.set("is_modem_running", True)
|
|
self.modem.start_modem()
|
|
|
|
return True
|
|
|
|
def stop_modem(self):
|
|
self.log.info("stopping modem....")
|
|
del self.modem
|
|
self.modem = False
|
|
self.states.set("is_modem_running", False)
|
|
self.event_manager.modem_stopped()
|
|
|
|
def test_audio(self):
|
|
try:
|
|
audio_test = audio.test_audio_devices(self.config['AUDIO']['input_device'],
|
|
self.config['AUDIO']['output_device'])
|
|
self.log.info("tested audio devices", result=audio_test)
|
|
|
|
return audio_test
|
|
except Exception as e:
|
|
self.log.error("Error testing audio devices", e=e)
|
|
return [False, False]
|
|
|
|
def start_beacon(self):
|
|
self.beacon = beacon.Beacon(self.config, self.states, self.event_manager, self.log, self.modem)
|
|
self.beacon.start()
|
|
|
|
def stop_beacon(self):
|
|
del self.beacon
|
|
|
|
def start_explorer_publishing(self):
|
|
try:
|
|
# optionally start explorer module
|
|
if self.config['STATION']['enable_explorer']:
|
|
explorer.explorer(self.app, self.config, self.states)
|
|
except Exception as e:
|
|
self.log.warning("[EXPLORER] Publishin not started because of error", e=e) |