joachim/openwebrx-clone
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

get the audio going as well

Browse Source
This commit is contained in:
Jakob Ketterl
2019-05-04 23:11:13 +02:00
parent 6ec21e6716
commit f05afc4b0a
4 changed files with 109 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
owrx/controllers.py
View File

@ -1,8 +1,7 @@
import mimetypes
from owrx.websocket import WebSocketConnection
from owrx.config import PropertyManager
from owrx.source import SpectrumThread
import csdr
from owrx.source import SpectrumThread, DspThread
import json
class Controller(object):
@ -52,6 +51,8 @@ class SpectrumForwarder(object):
self.conn = conn
def write_spectrum_data(self, data):
self.conn.send(bytes([0x01]) + data)
def write_dsp_data(self, data):
self.conn.send(bytes([0x02]) + data)
class WebSocketMessageHandler(object):
def __init__(self):
@ -64,7 +65,7 @@ class WebSocketMessageHandler(object):
for key in ["waterfall_colors", "waterfall_min_level", "waterfall_max_level", "waterfall_auto_level_margin",
"shown_center_freq", "samp_rate", "fft_size", "fft_fps", "audio_compression", "fft_compression",
"max_clients", "start_mod"]:
"max_clients", "start_mod", "client_audio_buffer_size"]:
config[key] = pm.getPropertyValue(key)
@ -73,29 +74,30 @@ class WebSocketMessageHandler(object):
conn.send({"type":"config","value":config})
print("client connection intitialized")
dsp = self.dsp = csdr.dsp()
dsp_initialized=False
dsp.set_audio_compression(pm.getPropertyValue("audio_compression"))
dsp.set_fft_compression(pm.getPropertyValue("fft_compression")) #used by secondary chains
dsp.set_format_conversion(pm.getPropertyValue("format_conversion"))
dsp.set_offset_freq(0)
dsp.set_bpf(-4000,4000)
dsp.set_secondary_fft_size(pm.getPropertyValue("digimodes_fft_size"))
dsp.nc_port=pm.getPropertyValue("iq_server_port")
dsp.csdr_dynamic_bufsize = pm.getPropertyValue("csdr_dynamic_bufsize")
dsp.csdr_print_bufsizes = pm.getPropertyValue("csdr_print_bufsizes")
dsp.csdr_through = pm.getPropertyValue("csdr_through")
do_secondary_demod=False
self.forwarder = SpectrumForwarder(conn)
SpectrumThread.getSharedInstance().add_client(self.forwarder)
self.dsp = DspThread(self.forwarder)
else:
try:
message = json.loads(message)
if message["type"] == "start":
self.dsp.set_samp_rate(message["params"]["output_rate"])
self.dsp.start()
if message["type"] == "dspcontrol":
if "params" in message:
params = message["params"]
for key in params:
methodname = "set_" + key
if hasattr(self.dsp, methodname):
method = getattr(self.dsp, methodname)
if callable(method):
method(params[key])
else:
print("method {0} is not callable".format(methodname))
else:
print("dsp has no method {0}".format(methodname))
if "action" in message and message["action"] == "start":
self.dsp.start()
except json.JSONDecodeError:
print("message is not json: {0}".format(message))
@ -105,6 +107,8 @@ class WebSocketMessageHandler(object):
def handleClose(self, conn):
if self.forwarder:
SpectrumThread.getSharedInstance().remove_client(self.forwarder)
if self.dsp:
self.dsp.stop()
class WebSocketController(Controller):
def handle_request(self):

56
owrx/source.py
View File

@ -98,4 +98,58 @@ class SpectrumThread(threading.Thread):
def shutdown(self):
print("shutting down spectrum thread")
SpectrumThread.sharedInstance = None
self.doRun = False
self.doRun = False
class DspThread(threading.Thread):
def __init__(self, handler):
self.doRun = True
self.handler = handler
pm = PropertyManager.getSharedInstance()
self.dsp = csdr.dsp()
#dsp_initialized=False
self.dsp.set_audio_compression(pm.getPropertyValue("audio_compression"))
self.dsp.set_fft_compression(pm.getPropertyValue("fft_compression")) #used by secondary chains
self.dsp.set_format_conversion(pm.getPropertyValue("format_conversion"))
self.dsp.set_offset_freq(0)
self.dsp.set_bpf(-4000,4000)
self.dsp.set_secondary_fft_size(pm.getPropertyValue("digimodes_fft_size"))
self.dsp.nc_port=pm.getPropertyValue("iq_server_port")
self.dsp.csdr_dynamic_bufsize = pm.getPropertyValue("csdr_dynamic_bufsize")
self.dsp.csdr_print_bufsizes = pm.getPropertyValue("csdr_print_bufsizes")
self.dsp.csdr_through = pm.getPropertyValue("csdr_through")
self.dsp.set_samp_rate(pm.getPropertyValue("samp_rate"))
#do_secondary_demod=False
super().__init__()
def run(self):
self.dsp.start()
while (self.doRun):
data = self.dsp.read(256)
self.handler.write_dsp_data(data)
def stop(self):
self.doRun = False
def set_output_rate(self, samp_rate):
self.dsp.set_output_rate(samp_rate)
def set_low_cut(self, cut):
bpf = self.dsp.get_bpf()
bpf[0] = cut
self.dsp.set_bpf(*bpf)
def set_high_cut(self, cut):
bpf = self.dsp.get_bpf()
bpf[1] = cut
self.dsp.set_bpf(*bpf)
def set_offset_freq(self, freq):
self.dsp.set_offset_freq(freq)
def set_mod(self, mod):
if (self.dsp.get_demodulator() == mod): return
self.dsp.stop()
self.dsp.set_demodulator(mod)
self.dsp.start()

8
owrx/websocket.py
View File

@ -20,7 +20,7 @@ class WebSocketConnection(object):
def get_header(self, size, opcode):
ws_first_byte = 0b10000000 | (opcode & 0x0F)
if(size>125):
if (size > 125):
return bytes([ws_first_byte, 126, (size>>8) & 0xff, size & 0xff])
else:
# 256 bytes binary message in a single unmasked frame
@ -34,14 +34,12 @@ class WebSocketConnection(object):
# string-type messages are sent as text frames
if (type(data) == str):
header = self.get_header(len(data), 1)
self.handler.wfile.write(header)
self.handler.wfile.write(data.encode('utf-8'))
self.handler.wfile.write(header + data.encode('utf-8'))
self.handler.wfile.flush()
# anything else as binary
else:
header = self.get_header(len(data), 2)
self.handler.wfile.write(header)
self.handler.wfile.write(data)
self.handler.wfile.write(header + data)
self.handler.wfile.flush()
def read_loop(self):