caiyu/qspectrumanalyzer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated for recent changes in hackrf_sweep -B output format

Browse Source
This commit is contained in:
Michael Ossmann 2017-02-09 19:05:46 -07:00
parent 4e9ecdb64d
commit adb3b53058
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
qspectrumanalyzer/backend.py
View File

@ -2,6 +2,8 @@ import subprocess, math, pprint
import numpy as np import numpy as np
from PyQt4 import QtCore from PyQt4 import QtCore
import struct
import time
class RtlPowerBaseThread(QtCore.QThread): class RtlPowerBaseThread(QtCore.QThread):
@ -279,7 +281,7 @@ class HackRFSweepThread(RtlPowerBaseThread):
"hops": 0, "hops": 0,
"device_index": 0, "device_index": 0,
"sample_rate": 20e6, "sample_rate": 20e6,
"bin_size": 1e6, "bin_size": int(bin_size*1000),
"interval": 0, "interval": 0,
"gain": 0, "gain": 0,
"ppm": 0, "ppm": 0,
@ -301,7 +303,7 @@ class HackRFSweepThread(RtlPowerBaseThread):
settings.value("rtl_power_executable", "hackrf_sweep"), settings.value("rtl_power_executable", "hackrf_sweep"),
"-f", "{}:{}".format(int(self.params["start_freq"]), "-f", "{}:{}".format(int(self.params["start_freq"]),
int(self.params["stop_freq"])), int(self.params["stop_freq"])),
"-B", "-w", "1000000", "-B", "-w", "{}".format(self.params["bin_size"]),
] ]
if self.params["single_shot"]: if self.params["single_shot"]:
@ -312,23 +314,19 @@ class HackRFSweepThread(RtlPowerBaseThread):
def parse_output(self, buf): def parse_output(self, buf):
"""Parse one buf of output from hackrf_sweep""" """Parse one buf of output from hackrf_sweep"""
data = np.fromstring(buf, dtype='<f4') (low_edge, high_edge, bin_width) = struct.unpack('QQI', buf[:20])
low_edge = data[0] * 1e6 data = np.fromstring(buf[20:], dtype='<f4')
high_edge = low_edge + self.params["sample_rate"] // 4
bin_width = self.params["sample_rate"] / self.fft_size
x_axis = list(np.arange(low_edge, high_edge, bin_width)) x_axis = list(np.arange(low_edge, high_edge, bin_width))
self.databuffer["x"].extend(x_axis) self.databuffer["x"].extend(x_axis)
for i in range(self.fft_size//4): for i in range(len(data)):
self.databuffer["y"].append(data[1+i]) self.databuffer["y"].append(data[i])
# The -1 below works around floating point errors if (high_edge / 1e6) >= (self.params["stop_freq"]):
if (high_edge / 1e6) >= (self.params["stop_freq"] - 1):
sorted_data = sorted(zip(self.databuffer["x"], self.databuffer["y"])) sorted_data = sorted(zip(self.databuffer["x"], self.databuffer["y"]))
self.databuffer["x"], self.databuffer["y"] = [list(x) for x in zip(*sorted_data)] self.databuffer["x"], self.databuffer["y"] = [list(x) for x in zip(*sorted_data)]
self.data_storage.update(self.databuffer) self.data_storage.update(self.databuffer)
self.databuffer = {"timestamp": [], "x": [], "y": []} self.databuffer = {"timestamp": [], "x": [], "y": []}
def run(self): def run(self):
"""hackrf_sweep thread main loop""" """hackrf_sweep thread main loop"""
self.process_start() self.process_start()
@ -336,7 +334,10 @@ class HackRFSweepThread(RtlPowerBaseThread):
self.rtlPowerStarted.emit() self.rtlPowerStarted.emit()
while self.alive: while self.alive:
buf = self.process.stdout.read(4*(1+self.fft_size//4)) buf = self.process.stdout.read(4)
if buf:
(record_length,) = struct.unpack('I', buf)
buf = self.process.stdout.read(record_length)
if buf: if buf:
self.parse_output(buf) self.parse_output(buf)