163 lines
5.9 KiB
Python
163 lines
5.9 KiB
Python
from owrx.config import Config
|
|
from owrx.meta import MetaParser
|
|
from owrx.wsjt import WsjtParser
|
|
from owrx.aprs import AprsParser
|
|
from owrx.pocsag import PocsagParser
|
|
from owrx.source import SdrSource
|
|
from owrx.property import PropertyStack
|
|
from csdr import csdr
|
|
import threading
|
|
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DspManager(csdr.output):
|
|
def __init__(self, handler, sdrSource):
|
|
self.handler = handler
|
|
self.sdrSource = sdrSource
|
|
self.parsers = {
|
|
"meta": MetaParser(self.handler),
|
|
"wsjt_demod": WsjtParser(self.handler),
|
|
"packet_demod": AprsParser(self.handler),
|
|
"pocsag_demod": PocsagParser(self.handler),
|
|
}
|
|
|
|
stack = PropertyStack()
|
|
stack.addLayer(0, self.sdrSource.getProps())
|
|
stack.addLayer(1, Config.get())
|
|
self.localProps = stack.collect(
|
|
"audio_compression",
|
|
"fft_compression",
|
|
"digimodes_fft_size",
|
|
"csdr_dynamic_bufsize",
|
|
"csdr_print_bufsizes",
|
|
"csdr_through",
|
|
"digimodes_enable",
|
|
"samp_rate",
|
|
"digital_voice_unvoiced_quality",
|
|
"dmr_filter",
|
|
"temporary_directory",
|
|
"center_freq",
|
|
|
|
# TODO: following properties are set from the client
|
|
"output_rate",
|
|
"squelch_level",
|
|
"secondary_mod",
|
|
"low_cut",
|
|
"high_cut",
|
|
"offset_freq",
|
|
)
|
|
|
|
self.dsp = csdr.dsp(self)
|
|
self.dsp.nc_port = self.sdrSource.getPort()
|
|
|
|
def set_low_cut(cut):
|
|
bpf = self.dsp.get_bpf()
|
|
bpf[0] = cut
|
|
self.dsp.set_bpf(*bpf)
|
|
|
|
def set_high_cut(cut):
|
|
bpf = self.dsp.get_bpf()
|
|
bpf[1] = cut
|
|
self.dsp.set_bpf(*bpf)
|
|
|
|
def set_dial_freq(key, value):
|
|
freq = self.localProps["center_freq"] + self.localProps["offset_freq"]
|
|
for parser in self.parsers.values():
|
|
parser.setDialFrequency(freq)
|
|
|
|
self.subscriptions = [
|
|
self.localProps.wireProperty("audio_compression", self.dsp.set_audio_compression),
|
|
self.localProps.wireProperty("fft_compression", self.dsp.set_fft_compression),
|
|
self.localProps.wireProperty("digimodes_fft_size", self.dsp.set_secondary_fft_size),
|
|
self.localProps.wireProperty("samp_rate", self.dsp.set_samp_rate),
|
|
self.localProps.wireProperty("output_rate", self.dsp.set_output_rate),
|
|
self.localProps.wireProperty("offset_freq", self.dsp.set_offset_freq),
|
|
self.localProps.wireProperty("squelch_level", self.dsp.set_squelch_level),
|
|
self.localProps.wireProperty("low_cut", set_low_cut),
|
|
self.localProps.wireProperty("high_cut", set_high_cut),
|
|
self.localProps.wireProperty("mod", self.dsp.set_demodulator),
|
|
self.localProps.wireProperty("digital_voice_unvoiced_quality", self.dsp.set_unvoiced_quality),
|
|
self.localProps.wireProperty("dmr_filter", self.dsp.set_dmr_filter),
|
|
self.localProps.wireProperty("temporary_directory", self.dsp.set_temporary_directory),
|
|
self.localProps.collect("center_freq", "offset_freq").wire(set_dial_freq),
|
|
]
|
|
|
|
self.dsp.set_offset_freq(0)
|
|
self.dsp.set_bpf(-4000, 4000)
|
|
self.dsp.csdr_dynamic_bufsize = self.localProps["csdr_dynamic_bufsize"]
|
|
self.dsp.csdr_print_bufsizes = self.localProps["csdr_print_bufsizes"]
|
|
self.dsp.csdr_through = self.localProps["csdr_through"]
|
|
|
|
if self.localProps["digimodes_enable"]:
|
|
|
|
def set_secondary_mod(mod):
|
|
if mod == False:
|
|
mod = None
|
|
self.dsp.set_secondary_demodulator(mod)
|
|
if mod is not None:
|
|
self.handler.write_secondary_dsp_config(
|
|
{
|
|
"secondary_fft_size": self.localProps["digimodes_fft_size"],
|
|
"if_samp_rate": self.dsp.if_samp_rate(),
|
|
"secondary_bw": self.dsp.secondary_bw(),
|
|
}
|
|
)
|
|
|
|
self.subscriptions += [
|
|
self.localProps.collect("secondary_mod").wire(set_secondary_mod),
|
|
self.localProps.collect("secondary_offset_freq").wire(self.dsp.set_secondary_offset_freq),
|
|
]
|
|
|
|
self.sdrSource.addClient(self)
|
|
|
|
super().__init__()
|
|
|
|
def start(self):
|
|
if self.sdrSource.isAvailable():
|
|
self.dsp.start()
|
|
|
|
def receive_output(self, t, read_fn):
|
|
logger.debug("adding new output of type %s", t)
|
|
writers = {
|
|
"audio": self.handler.write_dsp_data,
|
|
"smeter": self.handler.write_s_meter_level,
|
|
"secondary_fft": self.handler.write_secondary_fft,
|
|
"secondary_demod": self.handler.write_secondary_demod,
|
|
}
|
|
for demod, parser in self.parsers.items():
|
|
writers[demod] = parser.parse
|
|
|
|
write = writers[t]
|
|
|
|
threading.Thread(target=self.pump(read_fn, write)).start()
|
|
|
|
def stop(self):
|
|
self.dsp.stop()
|
|
self.sdrSource.removeClient(self)
|
|
for sub in self.subscriptions:
|
|
sub.cancel()
|
|
self.subscriptions = []
|
|
|
|
def setProperty(self, prop, value):
|
|
self.localProps[prop] = value
|
|
|
|
def getClientClass(self):
|
|
return SdrSource.CLIENT_USER
|
|
|
|
def onStateChange(self, state):
|
|
if state == SdrSource.STATE_RUNNING:
|
|
logger.debug("received STATE_RUNNING, attempting DspSource restart")
|
|
self.dsp.start()
|
|
elif state == SdrSource.STATE_STOPPING:
|
|
logger.debug("received STATE_STOPPING, shutting down DspSource")
|
|
self.dsp.stop()
|
|
elif state == SdrSource.STATE_FAILED:
|
|
logger.debug("received STATE_FAILED, shutting down DspSource")
|
|
self.dsp.stop()
|
|
|
|
def onBusyStateChange(self, state):
|
|
pass
|