118 lines
3.6 KiB
Python
118 lines
3.6 KiB
Python
|
import threading
|
||
|
|
from owrx.source import SdrService
|
||
|
|
from owrx.bands import Bandplan
|
||
|
|
from csdr import dsp, output
|
||
|
|
from owrx.wsjt import WsjtParser
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class ServiceOutput(output):
|
||
|
|
def __init__(self, frequency):
|
||
|
|
self.frequency = frequency
|
||
|
|
|
||
|
|
def add_output(self, t, read_fn):
|
||
|
|
logger.debug("got output of type {0}".format(t))
|
||
|
|
|
||
|
|
def pump(read, write):
|
||
|
|
def copy():
|
||
|
|
run = True
|
||
|
|
while run:
|
||
|
|
data = read()
|
||
|
|
if data is None or (isinstance(data, bytes) and len(data) == 0):
|
||
|
|
logger.warning("zero read on {0}".format(t))
|
||
|
|
run = False
|
||
|
|
else:
|
||
|
|
write(data)
|
||
|
|
|
||
|
|
return copy
|
||
|
|
|
||
|
|
if t == "wsjt_demod":
|
||
|
|
parser = WsjtParser(WsjtHandler())
|
||
|
|
parser.setDialFrequency(self.frequency)
|
||
|
|
target = pump(read_fn, parser.parse)
|
||
|
|
else:
|
||
|
|
# dump everything else
|
||
|
|
# TODO rewrite the output mechanism in a way that avoids producing unnecessary data
|
||
|
|
target = pump(read_fn, lambda x: None)
|
||
|
|
threading.Thread(target=target).start()
|
||
|
|
|
||
|
|
|
||
|
|
class ServiceHandler(object):
|
||
|
|
def __init__(self, source):
|
||
|
|
self.services = []
|
||
|
|
self.source = source
|
||
|
|
self.source.addClient(self)
|
||
|
|
self.source.getProps().collect("center_freq", "samp_rate").wire(self.onFrequencyChange)
|
||
|
|
self.onFrequencyChange("", "")
|
||
|
|
|
||
|
|
def onSdrAvailable(self):
|
||
|
|
logger.debug("sdr {0} is available".format(self.source.getName()))
|
||
|
|
self.onFrequencyChange("", "")
|
||
|
|
|
||
|
|
def onSdrUnavailable(self):
|
||
|
|
logger.debug("sdr {0} is unavailable".format(self.source.getName()))
|
||
|
|
self.stopServices()
|
||
|
|
|
||
|
|
def isSupported(self, mode):
|
||
|
|
return mode in ["ft8", "ft4", "wspr"]
|
||
|
|
|
||
|
|
def stopServices(self):
|
||
|
|
for service in self.services:
|
||
|
|
service.stop()
|
||
|
|
self.services = []
|
||
|
|
|
||
|
|
def startServices(self):
|
||
|
|
for service in self.services:
|
||
|
|
service.start()
|
||
|
|
|
||
|
|
def onFrequencyChange(self, key, value):
|
||
|
|
if not self.source.isAvailable():
|
||
|
|
return
|
||
|
|
logger.debug("sdr {0} is changing frequency".format(self.source.getName()))
|
||
|
|
self.stopServices()
|
||
|
|
cf = self.source.getProps()["center_freq"]
|
||
|
|
srh = self.source.getProps()["samp_rate"] / 2
|
||
|
|
frequency_range = (cf - srh, cf + srh)
|
||
|
|
self.services = [self.setupService(dial["mode"], dial["frequency"]) for dial in Bandplan.getSharedInstance().collectDialFrequencies(frequency_range) if self.isSupported(dial["mode"])]
|
||
|
|
|
||
|
|
def setupService(self, mode, frequency):
|
||
|
|
logger.debug("setting up service {0} on frequency {1}".format(mode, frequency))
|
||
|
|
d = dsp(ServiceOutput(frequency))
|
||
|
|
d.set_offset_freq(frequency - self.source.getProps()["center_freq"])
|
||
|
|
d.set_demodulator("usb")
|
||
|
|
d.set_bpf(0, 3000)
|
||
|
|
d.set_secondary_demodulator(mode)
|
||
|
|
d.set_audio_compression("none")
|
||
|
|
d.set_samp_rate(self.source.getProps()["samp_rate"])
|
||
|
|
d.start()
|
||
|
|
return d
|
||
|
|
|
||
|
|
|
||
|
|
class WsjtHandler(object):
|
||
|
|
def write_wsjt_message(self, msg):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class ServiceManager(object):
|
||
|
|
sharedInstance = None
|
||
|
|
@staticmethod
|
||
|
|
def getSharedInstance():
|
||
|
|
if ServiceManager.sharedInstance is None:
|
||
|
|
ServiceManager.sharedInstance = ServiceManager()
|
||
|
|
return ServiceManager.sharedInstance
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
for source in SdrService.getSources().values():
|
||
|
|
ServiceHandler(source)
|
||
|
|
|
||
|
|
|
||
|
|
class Service(object):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class WsjtService(Service):
|
||
|
|
pass
|