summaryrefslogtreecommitdiff
path: root/gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py
diff options
context:
space:
mode:
authorJosh Morman <jmorman@gnuradio.org>2021-11-24 12:53:58 -0500
committermormj <34754695+mormj@users.noreply.github.com>2021-11-24 14:41:53 -0500
commit51ec89501552c35b2bd0721c0501302f224fe44d (patch)
tree442036c424c286ece0976dcd3c36891cddefcd25 /gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py
parente426b9ad20dbbf41326c103b31a7418a80362b06 (diff)
runtime: pep8 formatting
Signed-off-by: Josh Morman <jmorman@gnuradio.org>
Diffstat (limited to 'gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py')
-rw-r--r--gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py52
1 files changed, 29 insertions, 23 deletions
diff --git a/gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py b/gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py
index f363393c36..ae3857d2f9 100644
--- a/gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py
+++ b/gnuradio-runtime/python/gnuradio/ctrlport/GrDataPlotter.py
@@ -13,7 +13,9 @@ from gnuradio import gr
from gnuradio import blocks
from gnuradio import filter, fft
from gnuradio.ctrlport.GNURadio import ControlPort
-import sys, time, struct
+import sys
+import time
+import struct
try:
from gnuradio import qtgui
@@ -23,6 +25,7 @@ except ImportError:
print("Error: Program requires PyQt5 and gr-qtgui.")
sys.exit(1)
+
class GrDataPlotParent(gr.top_block, Qt.QWidget):
# Setup signals
plotupdated = QtCore.pyqtSignal(Qt.QWidget)
@@ -34,7 +37,7 @@ class GrDataPlotParent(gr.top_block, Qt.QWidget):
self._name = name
self._npts = 500
self._rate = rate
- self.knobnames = [name,]
+ self.knobnames = [name, ]
self.layout = Qt.QVBoxLayout()
self.setLayout(self.layout)
@@ -52,10 +55,10 @@ class GrDataPlotParent(gr.top_block, Qt.QWidget):
self.disconnect(self.thr, (self.snk, 0))
self.disconnect(self.src[0], self.thr)
for n in range(1, self._ncons):
- self.disconnect(self.src[n], (self.snk,n))
+ self.disconnect(self.src[n], (self.snk, n))
self._ncons = nconnections
- self._data_len = self._ncons*[0,]
+ self._data_len = self._ncons * [0, ]
self.thr = blocks.throttle(self._datasize, self._rate)
self.snk = self.get_qtsink()
@@ -67,13 +70,13 @@ class GrDataPlotParent(gr.top_block, Qt.QWidget):
for n in range(self._ncons):
self.set_line_label(n, self.knobnames[n])
- self._last_data.append(int(self._npts)*[0,])
+ self._last_data.append(int(self._npts) * [0, ])
self.src.append(self.get_vecsource())
if(n == 0):
self.connect(self.src[n], self.thr)
else:
- self.connect(self.src[n], (self.snk,n))
+ self.connect(self.src[n], (self.snk, n))
self.py_window = sip.wrapinstance(self.snk.qwidget(), Qt.QWidget)
@@ -122,7 +125,7 @@ class GrDataPlotParent(gr.top_block, Qt.QWidget):
if(self._iscomplex):
data_r = data[0::2]
data_i = data[1::2]
- data = [complex(r,i) for r,i in zip(data_r, data_i)]
+ data = [complex(r, i) for r, i in zip(data_r, data_i)]
return data
def update(self, data):
@@ -136,10 +139,11 @@ class GrDataPlotParent(gr.top_block, Qt.QWidget):
if(self._data_len[n] < npts):
self._last_data[n] = self._last_data[n][0:npts]
else:
- self._last_data[n] = self._last_data[n][self._data_len[n]-npts:self._data_len[n]]
+ self._last_data[n] = self._last_data[n][self._data_len[n] -
+ npts:self._data_len[n]]
self._data_len[n] = npts
else:
- self._last_data[n] += (npts - self._npts)*[0,]
+ self._last_data[n] += (npts - self._npts) * [0, ]
self._npts = npts
self.snk.reset()
@@ -152,12 +156,13 @@ class GrDataPlotParent(gr.top_block, Qt.QWidget):
self.src[n].set_data(data[n])
self._last_data[n] = data[n][-self._npts:]
else:
- newdata = self._last_data[n][-(self._npts-len(data)):]
+ newdata = self._last_data[n][-(
+ self._npts - len(data)):]
newdata += data[n]
self.src[n].set_data(newdata)
self._last_data[n] = newdata
- else: # single value update
+ else: # single value update
if(self._iscomplex):
data[n] = complex(data[n][0], data[n][1])
if(self._data_len[n] < self._npts):
@@ -170,12 +175,11 @@ class GrDataPlotParent(gr.top_block, Qt.QWidget):
else:
for n in range(self._ncons):
if(type(data[n]) != list):
- data[n] = [data[n],]
+ data[n] = [data[n], ]
data[n] = self.data_to_complex(data[n])
self.src[n].set_data(data[n])
-
class GrDataPlotterC(GrDataPlotParent):
def __init__(self, name, rate, pmin=None, pmax=None, stripchart=False):
GrDataPlotParent.__init__(self, name, rate, pmin, pmax)
@@ -203,8 +207,8 @@ class GrDataPlotterC(GrDataPlotParent):
return self._npts
def set_line_label(self, n, name):
- self.snk.set_line_label(2*n+0, "Re{" + self.knobnames[n] + "}")
- self.snk.set_line_label(2*n+1, "Im{" + self.knobnames[n] + "}")
+ self.snk.set_line_label(2 * n + 0, "Re{" + self.knobnames[n] + "}")
+ self.snk.set_line_label(2 * n + 1, "Im{" + self.knobnames[n] + "}")
class GrDataPlotterF(GrDataPlotParent):
@@ -366,6 +370,7 @@ class GrTimeRasterF(GrDataPlotParent):
def set_line_label(self, n, name):
self.snk.set_line_label(n, self.knobnames[n])
+
class GrTimeRasterB(GrDataPlotParent):
def __init__(self, name, rate, pmin=None, pmax=None):
GrDataPlotParent.__init__(self, name, rate, pmin, pmax)
@@ -404,7 +409,7 @@ class GrDataPlotterValueTable(object):
self.uid = uid
self.treeWidget = Qt.QTreeWidget(parent)
self.treeWidget.setColumnCount(len(headers))
- self.treeWidget.setGeometry(x,y,xsize,ysize)
+ self.treeWidget.setGeometry(x, y, xsize, ysize)
self.treeWidget.setHeaderLabels(headers)
self.treeWidget.resizeColumnToContents(0)
@@ -432,19 +437,19 @@ class GrDataPlotterValueTable(object):
descr = str(knobprops[itemKey].description)
if(type(v) == ControlPort.complex):
- v = v.re + v.im*1j
+ v = v.re + v.im * 1j
# If it's a byte stream, Python thinks it's a string.
# Unpack and convert to floats for plotting.
# Ignore the edge list knob if it's being exported
elif(type(v) == str and itemKey.find('probe2_b') == 0):
- v = struct.unpack(len(v)*'b', v)
+ v = struct.unpack(len(v) * 'b', v)
# Convert the final value to a string for displaying
v = str(v)
if (item.text(1) != v or
item.text(2) != units or
- item.text(3) != descr):
+ item.text(3) != descr):
item.setText(1, v)
item.setText(2, units)
@@ -458,20 +463,21 @@ class GrDataPlotterValueTable(object):
if k not in foundKeys:
v = knobs[k].value
if(type(v) == ControlPort.complex):
- v = v.re + v.im*1j
+ v = v.re + v.im * 1j
# If it's a byte stream, Python thinks it's a string.
# Unpack and convert to floats for plotting.
# Ignore the edge list knob if it's being exported
elif(type(v) == str and k.find('probe2_b') == 0):
- v = struct.unpack(len(v)*'b', v)
+ v = struct.unpack(len(v) * 'b', v)
item = Qt.QTreeWidgetItem([k, str(v),
- knobprops[k].units, knobprops[k].description])
+ knobprops[k].units, knobprops[k].description])
self.treeWidget.addTopLevelItem(item)
# Remove items currently in tree that are not in the knob list.
for itemKey in deleteKeys:
- qtwiList = self.treeWidget.findItems(itemKey, Qt.Qt.MatchFixedString)
+ qtwiList = self.treeWidget.findItems(
+ itemKey, Qt.Qt.MatchFixedString)
if (len(qtwiList) > 1):
raise Exception('More than one item with key %s in tree' %
itemKey)
generated by cgit v1.2.3 (git 2.39.1) at 2025-03-25 00:25:55 +0000