we got fft
This commit is contained in:
Jakob Ketterl
@ -21,3 +21,6 @@ class PropertyManager(object):
|
||||
if not name in self.properties:
|
||||
self.properties[name] = Property()
|
||||
return self.properties[name]
|
||||
|
||||
def getPropertyValue(self, name):
|
||||
return self.getProperty(name).getValue()
|
||||
|
||||
@ -1,6 +1,9 @@
|
||||
import mimetypes
|
||||
from owrx.websocket import WebSocketConnection
|
||||
from owrx.config import PropertyManager
|
||||
from owrx.source import SpectrumThread
|
||||
import csdr
|
||||
import json
|
||||
|
||||
class Controller(object):
|
||||
def __init__(self, handler, matches):
|
||||
@ -44,19 +47,66 @@ class AssetsController(Controller):
|
||||
filename = self.matches.group(1)
|
||||
self.serve_file(filename)
|
||||
|
||||
class SpectrumForwarder(object):
|
||||
def __init__(self, conn):
|
||||
self.conn = conn
|
||||
def write_spectrum_data(self, data):
|
||||
self.conn.send(bytes([0x01]) + data)
|
||||
|
||||
class WebSocketMessageHandler(object):
|
||||
def __init__(self):
|
||||
self.forwarder = None
|
||||
|
||||
def handleTextMessage(self, conn, message):
|
||||
if (message[:16] == "SERVER DE CLIENT"):
|
||||
config = {}
|
||||
pm = PropertyManager.getSharedInstance()
|
||||
|
||||
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"]:
|
||||
|
||||
config[key] = pm.getProperty(key).getValue()
|
||||
|
||||
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)
|
||||
|
||||
else:
|
||||
try:
|
||||
message = json.loads(message)
|
||||
if message["type"] == "start":
|
||||
self.dsp.set_samp_rate(message["params"]["output_rate"])
|
||||
self.dsp.start()
|
||||
except json.JSONDecodeError:
|
||||
print("message is not json: {0}".format(message))
|
||||
|
||||
def handleBinaryMessage(self, conn, data):
|
||||
print("unsupported binary message, discarding")
|
||||
|
||||
def handleClose(self, conn):
|
||||
if self.forwarder:
|
||||
SpectrumThread.getSharedInstance().remove_client(self.forwarder)
|
||||
|
||||
class WebSocketController(Controller):
|
||||
def handle_request(self):
|
||||
conn = WebSocketConnection(self.handler)
|
||||
conn = WebSocketConnection(self.handler, WebSocketMessageHandler())
|
||||
conn.send("CLIENT DE SERVER openwebrx.py")
|
||||
|
||||
config = {}
|
||||
pm = PropertyManager.getSharedInstance()
|
||||
|
||||
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"]:
|
||||
|
||||
config[key] = pm.getProperty(key).getValue()
|
||||
|
||||
conn.send({"type":"config","value":config})
|
||||
# enter read loop
|
||||
conn.read_loop()
|
||||
|
||||
101
owrx/source.py
Normal file
101
owrx/source.py
Normal file
@ -0,0 +1,101 @@
|
||||
import subprocess
|
||||
from owrx.config import PropertyManager
|
||||
import threading
|
||||
import csdr
|
||||
import time
|
||||
|
||||
class RtlNmuxSource(object):
|
||||
def __init__(self):
|
||||
pm = PropertyManager.getSharedInstance()
|
||||
|
||||
nmux_bufcnt = nmux_bufsize = 0
|
||||
while nmux_bufsize < pm.getPropertyValue("samp_rate")/4: nmux_bufsize += 4096
|
||||
while nmux_bufsize * nmux_bufcnt < pm.getPropertyValue("nmux_memory") * 1e6: nmux_bufcnt += 1
|
||||
if nmux_bufcnt == 0 or nmux_bufsize == 0:
|
||||
print("[openwebrx-main] Error: nmux_bufsize or nmux_bufcnt is zero. These depend on nmux_memory and samp_rate options in config_webrx.py")
|
||||
return
|
||||
print("[openwebrx-main] nmux_bufsize = %d, nmux_bufcnt = %d" % (nmux_bufsize, nmux_bufcnt))
|
||||
cmd = pm.getPropertyValue("start_rtl_command") + "| nmux --bufsize %d --bufcnt %d --port %d --address 127.0.0.1" % (nmux_bufsize, nmux_bufcnt, pm.getPropertyValue("iq_server_port"))
|
||||
subprocess.Popen(cmd, shell=True)
|
||||
print("[openwebrx-main] Started rtl source: " + cmd)
|
||||
|
||||
class SpectrumThread(threading.Thread):
|
||||
sharedInstance = None
|
||||
@staticmethod
|
||||
def getSharedInstance():
|
||||
if SpectrumThread.sharedInstance is None:
|
||||
SpectrumThread.sharedInstance = SpectrumThread()
|
||||
SpectrumThread.sharedInstance.start()
|
||||
return SpectrumThread.sharedInstance
|
||||
|
||||
def __init__(self):
|
||||
self.clients = []
|
||||
self.doRun = True
|
||||
super().__init__()
|
||||
|
||||
def run(self):
|
||||
pm = PropertyManager.getSharedInstance()
|
||||
|
||||
samp_rate = pm.getPropertyValue("samp_rate")
|
||||
fft_size = pm.getPropertyValue("fft_size")
|
||||
fft_fps = pm.getPropertyValue("fft_fps")
|
||||
fft_voverlap_factor = pm.getPropertyValue("fft_voverlap_factor")
|
||||
fft_compression = pm.getPropertyValue("fft_compression")
|
||||
format_conversion = pm.getPropertyValue("format_conversion")
|
||||
|
||||
spectrum_dsp=dsp=csdr.dsp()
|
||||
dsp.nc_port = pm.getPropertyValue("iq_server_port")
|
||||
dsp.set_demodulator("fft")
|
||||
dsp.set_samp_rate(samp_rate)
|
||||
dsp.set_fft_size(fft_size)
|
||||
dsp.set_fft_fps(fft_fps)
|
||||
dsp.set_fft_averages(int(round(1.0 * samp_rate / fft_size / fft_fps / (1.0 - fft_voverlap_factor))) if fft_voverlap_factor>0 else 0)
|
||||
dsp.set_fft_compression(fft_compression)
|
||||
dsp.set_format_conversion(format_conversion)
|
||||
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")
|
||||
sleep_sec=0.87/fft_fps
|
||||
print("[openwebrx-spectrum] Spectrum thread initialized successfully.")
|
||||
dsp.start()
|
||||
if pm.getPropertyValue("csdr_dynamic_bufsize"):
|
||||
dsp.read(8) #dummy read to skip bufsize & preamble
|
||||
print("[openwebrx-spectrum] Note: CSDR_DYNAMIC_BUFSIZE_ON = 1")
|
||||
print("[openwebrx-spectrum] Spectrum thread started.")
|
||||
bytes_to_read=int(dsp.get_fft_bytes_to_read())
|
||||
spectrum_thread_counter=0
|
||||
while self.doRun:
|
||||
data=dsp.read(bytes_to_read)
|
||||
#print("gotcha",len(data),"bytes of spectrum data via spectrum_thread_function()")
|
||||
if spectrum_thread_counter >= fft_fps:
|
||||
spectrum_thread_counter=0
|
||||
else: spectrum_thread_counter+=1
|
||||
for c in self.clients:
|
||||
c.write_spectrum_data(data)
|
||||
'''
|
||||
correction=0
|
||||
for i in range(0,len(clients)):
|
||||
i-=correction
|
||||
if (clients[i].ws_started):
|
||||
if clients[i].spectrum_queue.full():
|
||||
print "[openwebrx-spectrum] client spectrum queue full, closing it."
|
||||
close_client(i, False)
|
||||
correction+=1
|
||||
else:
|
||||
clients[i].spectrum_queue.put([data]) # add new string by "reference" to all clients
|
||||
'''
|
||||
|
||||
print("spectrum thread shut down")
|
||||
|
||||
def add_client(self, c):
|
||||
self.clients.append(c)
|
||||
|
||||
def remove_client(self, c):
|
||||
self.clients.remove(c)
|
||||
if not self.clients:
|
||||
self.shutdown()
|
||||
|
||||
def shutdown(self):
|
||||
print("shutting down spectrum thread")
|
||||
SpectrumThread.sharedInstance = None
|
||||
self.doRun = False
|
||||
@ -3,8 +3,9 @@ import hashlib
|
||||
import json
|
||||
|
||||
class WebSocketConnection(object):
|
||||
def __init__(self, handler):
|
||||
def __init__(self, handler, messageHandler):
|
||||
self.handler = handler
|
||||
self.messageHandler = messageHandler
|
||||
my_headers = self.handler.headers.items()
|
||||
my_header_keys = list(map(lambda x:x[0],my_headers))
|
||||
h_key_exists = lambda x:my_header_keys.count(x)
|
||||
@ -40,8 +41,35 @@ class WebSocketConnection(object):
|
||||
else:
|
||||
header = self.get_header(len(data), 2)
|
||||
self.handler.wfile.write(header)
|
||||
self.handler.wfile.write(data.encode())
|
||||
self.handler.wfile.write(data)
|
||||
self.handler.wfile.flush()
|
||||
|
||||
def read_loop(self):
|
||||
open = True
|
||||
while (open):
|
||||
header = self.handler.rfile.read(2)
|
||||
opcode = header[0] & 0x0F
|
||||
length = header[1] & 0x7F
|
||||
mask = (header[1] & 0x80) >> 7
|
||||
if (length == 126):
|
||||
header = self.handler.rfile.read(2)
|
||||
length = (header[0] << 8) + header[1]
|
||||
if (mask):
|
||||
masking_key = self.handler.rfile.read(4)
|
||||
data = self.handler.rfile.read(length)
|
||||
print("opcode: {0}, length: {1}, mask: {2}".format(opcode, length, mask))
|
||||
if (mask):
|
||||
data = bytes([b ^ masking_key[index % 4] for (index, b) in enumerate(data)])
|
||||
if (opcode == 1):
|
||||
message = data.decode('utf-8')
|
||||
self.messageHandler.handleTextMessage(self, message)
|
||||
elif (opcode == 2):
|
||||
self.messageHandler.handleBinaryMessage(self, data)
|
||||
elif (opcode == 8):
|
||||
open = False
|
||||
self.messageHandler.handleClose(self)
|
||||
else:
|
||||
print("unsupported opcode: {0}".format(opcode))
|
||||
|
||||
class WebSocketException(Exception):
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user