mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-28 17:32:35 +00:00
205 lines
7.3 KiB
Python
Executable File
205 lines
7.3 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
##################################################
|
|
# GNU Radio Python Flow Graph
|
|
# Title: Pulse Detect
|
|
# GNU Radio version: 3.7.13.5
|
|
##################################################
|
|
hackrf_sn='000000000000000087c867dc2a44625f'
|
|
center_freq=5180e6
|
|
gnuradio_points=100000
|
|
|
|
if __name__ == '__main__':
|
|
import ctypes
|
|
import sys
|
|
import os
|
|
import argparse
|
|
|
|
# test for QT_X11_NO_MITSHM=1
|
|
if 'QT_X11_NO_MITSHM' not in os.environ:
|
|
print("QT_X11_NO_MITSHM not set. Please export QT_X11_NO_MITSHM=1")
|
|
exit(1);
|
|
if os.environ['QT_X11_NO_MITSHM'] is not "1":
|
|
print("QT_X11_NO_MITSHM not 1. Please export QT_X11_NO_MITSHM=1")
|
|
exit(1);
|
|
|
|
if sys.platform.startswith('linux'):
|
|
try:
|
|
x11 = ctypes.cdll.LoadLibrary('libX11.so')
|
|
x11.XInitThreads()
|
|
except:
|
|
print("Warning: failed to XInitThreads()")
|
|
|
|
unixOptions = "f:s:p:"
|
|
gnuOptions = [ "freq=", "serno=", "points=" ]
|
|
parser = argparse.ArgumentParser(description='pulse watcher')
|
|
parser.add_argument("--serno", type=str, help="hackrf serial number")
|
|
parser.add_argument("--freq", type=int, help="center frequency in Hz")
|
|
parser.add_argument("--points", type=int, help="number of x-axis points")
|
|
args = parser.parse_args()
|
|
hackrf_sn = args.serno
|
|
center_freq = args.freq
|
|
gnuradio_points = args.points
|
|
|
|
from PyQt4 import Qt
|
|
from gnuradio import blocks
|
|
from gnuradio import eng_notation
|
|
from gnuradio import gr
|
|
from gnuradio import qtgui
|
|
from gnuradio.eng_option import eng_option
|
|
from gnuradio.filter import firdes
|
|
from optparse import OptionParser
|
|
import osmosdr
|
|
import os
|
|
import sip
|
|
import sys
|
|
import argparse
|
|
import time
|
|
from gnuradio import qtgui
|
|
|
|
class pulse_detect(gr.top_block, Qt.QWidget):
|
|
|
|
# divide sample rate by points to get time scale in millisec
|
|
gnuradio_points = args.points
|
|
|
|
def __init__(self):
|
|
gr.top_block.__init__(self, "Pulse Detect")
|
|
Qt.QWidget.__init__(self)
|
|
self.setWindowTitle("Pulse Detect")
|
|
qtgui.util.check_set_qss()
|
|
try:
|
|
self.setWindowIcon(Qt.QIcon.fromTheme('gnuradio-grc'))
|
|
except:
|
|
pass
|
|
self.top_scroll_layout = Qt.QVBoxLayout()
|
|
self.setLayout(self.top_scroll_layout)
|
|
self.top_scroll = Qt.QScrollArea()
|
|
self.top_scroll.setFrameStyle(Qt.QFrame.NoFrame)
|
|
self.top_scroll_layout.addWidget(self.top_scroll)
|
|
self.top_scroll.setWidgetResizable(True)
|
|
self.top_widget = Qt.QWidget()
|
|
self.top_scroll.setWidget(self.top_widget)
|
|
self.top_layout = Qt.QVBoxLayout(self.top_widget)
|
|
self.top_grid_layout = Qt.QGridLayout()
|
|
self.top_layout.addLayout(self.top_grid_layout)
|
|
|
|
self.settings = Qt.QSettings("GNU Radio", "pulse_detect")
|
|
self.restoreGeometry(self.settings.value("geometry").toByteArray())
|
|
|
|
|
|
##################################################
|
|
# Variables
|
|
##################################################
|
|
self.samp_rate = samp_rate = 3e6
|
|
|
|
##################################################
|
|
# Blocks
|
|
##################################################
|
|
self.qtgui_time_sink_x_0 = qtgui.time_sink_f(
|
|
self.gnuradio_points,
|
|
samp_rate, #samp_rate
|
|
"", #name
|
|
1 #number of inputs
|
|
)
|
|
self.qtgui_time_sink_x_0.set_update_time(0.10)
|
|
self.qtgui_time_sink_x_0.set_y_axis(0, 2)
|
|
|
|
self.qtgui_time_sink_x_0.set_y_label('Amplitude', "")
|
|
|
|
self.qtgui_time_sink_x_0.enable_tags(-1, True)
|
|
self.qtgui_time_sink_x_0.set_trigger_mode(qtgui.TRIG_MODE_NORM, qtgui.TRIG_SLOPE_POS, 0.5, 0, 0, "")
|
|
self.qtgui_time_sink_x_0.enable_autoscale(False)
|
|
self.qtgui_time_sink_x_0.enable_grid(False)
|
|
self.qtgui_time_sink_x_0.enable_axis_labels(True)
|
|
self.qtgui_time_sink_x_0.enable_control_panel(True)
|
|
self.qtgui_time_sink_x_0.enable_stem_plot(False)
|
|
|
|
if not True:
|
|
self.qtgui_time_sink_x_0.disable_legend()
|
|
|
|
labels = ['', '', '', '', '',
|
|
'', '', '', '', '']
|
|
widths = [1, 1, 1, 1, 1,
|
|
1, 1, 1, 1, 1]
|
|
colors = ["blue", "red", "green", "black", "cyan",
|
|
"magenta", "yellow", "dark red", "dark green", "blue"]
|
|
styles = [1, 1, 1, 1, 1,
|
|
1, 1, 1, 1, 1]
|
|
markers = [-1, -1, -1, -1, -1,
|
|
-1, -1, -1, -1, -1]
|
|
alphas = [1.0, 1.0, 1.0, 1.0, 1.0,
|
|
1.0, 1.0, 1.0, 1.0, 1.0]
|
|
|
|
for i in xrange(1):
|
|
if len(labels[i]) == 0:
|
|
self.qtgui_time_sink_x_0.set_line_label(i, "Data {0}".format(i))
|
|
else:
|
|
self.qtgui_time_sink_x_0.set_line_label(i, labels[i])
|
|
self.qtgui_time_sink_x_0.set_line_width(i, widths[i])
|
|
self.qtgui_time_sink_x_0.set_line_color(i, colors[i])
|
|
self.qtgui_time_sink_x_0.set_line_style(i, styles[i])
|
|
self.qtgui_time_sink_x_0.set_line_marker(i, markers[i])
|
|
self.qtgui_time_sink_x_0.set_line_alpha(i, alphas[i])
|
|
|
|
self._qtgui_time_sink_x_0_win = sip.wrapinstance(self.qtgui_time_sink_x_0.pyqwidget(), Qt.QWidget)
|
|
self.top_grid_layout.addWidget(self._qtgui_time_sink_x_0_win)
|
|
osmodr_source_args = "numchan=1 hackrf=%s"%(hackrf_sn)
|
|
self.osmosdr_source_0 = osmosdr.source( args=osmodr_source_args ) # serno into variable
|
|
self.osmosdr_source_0.set_sample_rate(samp_rate)
|
|
self.osmosdr_source_0.set_center_freq(center_freq, 0) # turn into variable
|
|
self.osmosdr_source_0.set_freq_corr(0, 0)
|
|
self.osmosdr_source_0.set_dc_offset_mode(0, 0)
|
|
self.osmosdr_source_0.set_iq_balance_mode(0, 0)
|
|
self.osmosdr_source_0.set_gain_mode(False, 0)
|
|
self.osmosdr_source_0.set_gain(10, 0)
|
|
self.osmosdr_source_0.set_if_gain(32, 0)
|
|
self.osmosdr_source_0.set_bb_gain(32, 0)
|
|
self.osmosdr_source_0.set_antenna('', 0)
|
|
self.osmosdr_source_0.set_bandwidth(0, 0)
|
|
|
|
self.blocks_complex_to_mag_0 = blocks.complex_to_mag(1)
|
|
|
|
|
|
|
|
##################################################
|
|
# Connections
|
|
##################################################
|
|
self.connect((self.blocks_complex_to_mag_0, 0), (self.qtgui_time_sink_x_0, 0))
|
|
self.connect((self.osmosdr_source_0, 0), (self.blocks_complex_to_mag_0, 0))
|
|
|
|
def closeEvent(self, event):
|
|
self.settings = Qt.QSettings("GNU Radio", "pulse_detect")
|
|
self.settings.setValue("geometry", self.saveGeometry())
|
|
event.accept()
|
|
|
|
def get_samp_rate(self):
|
|
return self.samp_rate
|
|
|
|
def set_samp_rate(self, samp_rate):
|
|
self.samp_rate = samp_rate
|
|
self.qtgui_time_sink_x_0.set_samp_rate(self.samp_rate)
|
|
self.osmosdr_source_0.set_sample_rate(self.samp_rate)
|
|
|
|
|
|
def main(top_block_cls=pulse_detect, options=None):
|
|
|
|
from distutils.version import StrictVersion
|
|
if StrictVersion(Qt.qVersion()) >= StrictVersion("4.5.0"):
|
|
style = gr.prefs().get_string('qtgui', 'style', 'raster')
|
|
Qt.QApplication.setGraphicsSystem(style)
|
|
qapp = Qt.QApplication(sys.argv)
|
|
|
|
tb = top_block_cls()
|
|
tb.start()
|
|
tb.show()
|
|
|
|
def quitting():
|
|
tb.stop()
|
|
tb.wait()
|
|
qapp.connect(qapp, Qt.SIGNAL("aboutToQuit()"), quitting)
|
|
qapp.exec_()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|