Merge branch 'qtplotters' into next

author: Tom Rondeau <trondeau@vt.edu> 2013-01-22 13:57:37 -0500
committer: Tom Rondeau <trondeau@vt.edu> 2013-01-22 13:57:37 -0500
commit: b3860befa302bb723f74dce84d7ed7ab768d5854 (patch)
tree: 6bfaccf901745f125df6f1e47485cef97192bef9 /gr-qtgui/apps/plot_form.py
parent: b55aab99aa9305eafcfa628774f0a4056bf763eb (diff)
parent: 0f9ad0a68173762b83bbe7ed38f1708be294bcc6 (diff)
1 files changed, 288 insertions, 31 deletions
diff --git a/gr-qtgui/apps/plot_form.py b/gr-qtgui/apps/plot_form.py
index f6841811a8..5b034407a8 100644
--- a/gr-qtgui/apps/plot_form.py
+++ b/gr-qtgui/apps/plot_form.py
@@ -27,42 +27,22 @@ except ImportError:
     print "Error: Program requires PyQt4."
     sys.exit(1)
 
-class dialog_box(QtGui.QWidget):
+class plot_form(QtGui.QWidget):
     def __init__(self, top_block, title=''):
         QtGui.QWidget.__init__(self, None)
-
+        
         self._start = 0
         self._end = 0
+        self._y_min = 0
+        self._y_max = 0
 
         self.top_block = top_block
+        self.top_block.gui_y_axis = self.gui_y_axis
         
         self.setWindowTitle(title)
 
-        self.size_val = QtGui.QIntValidator(0, top_block._max_nsamps, self)
-
-        self.start_form = QtGui.QFormLayout()
-        self.start_edit = QtGui.QLineEdit(self)
-        self.start_edit.setMinimumWidth(100)
-        self.start_edit.setText(QtCore.QString("%1").arg(top_block._start))
-        self.start_edit.setValidator(self.size_val)
-        self.start_form.addRow("Start:", self.start_edit)
-        self.connect(self.start_edit, QtCore.SIGNAL("returnPressed()"),
-                     self.update_points)
-
-        end = top_block._start + top_block._nsamps
-        self.end_form = QtGui.QFormLayout()
-        self.end_edit = QtGui.QLineEdit(self)
-        self.end_edit.setMinimumWidth(100)
-        self.end_edit.setText(QtCore.QString("%1").arg(end))
-        self.end_edit.setValidator(self.size_val)
-        self.end_form.addRow("End:", self.end_edit)
-        self.connect(self.end_edit, QtCore.SIGNAL("returnPressed()"),
-                     self.update_points)
-
         self.layout = QtGui.QGridLayout(self)
-        self.layout.addWidget(top_block.get_gui(), 1,0,1,2)
-        self.layout.addLayout(self.start_form, 2,0,1,1)
-        self.layout.addLayout(self.end_form, 2,1,1,1)
+        self.layout.addWidget(top_block.get_gui(), 1,2,1,2)
 
         # Create a save action
         self.save_act = QtGui.QAction("Save", self)
@@ -81,11 +61,241 @@ class dialog_box(QtGui.QWidget):
         self.menu.addAction(self.save_act)
         self.menu.addAction(self.exit_act)
 
-        self.layout.addWidget(self.menu, 0,0,1,2)
-        
-        self.resize(800, 500)
+        self.layout.addWidget(self.menu, 0,0,1,4)
+
+        self.left_col_form = QtGui.QFormLayout()
+        self.layout.addLayout(self.left_col_form, 1,0,1,1)
+        self.layout.setColumnStretch(0, 0)
+        self.layout.setColumnStretch(2, 1)
+
+        # Create Edit boxes for X-axis start/stop
+        self.size_val = QtGui.QIntValidator(0, top_block._max_nsamps, self)
+
+        self.start_edit = QtGui.QLineEdit(self)
+        self.start_edit.setMinimumWidth(100)
+        self.start_edit.setMaximumWidth(100)
+        self.start_edit.setText(QtCore.QString("%1").arg(top_block._start))
+        self.start_edit.setValidator(self.size_val)
+        self.left_col_form.addRow("Start:", self.start_edit)
+        self.connect(self.start_edit, QtCore.SIGNAL("returnPressed()"),
+                     self.update_xaxis_pos)
+
+        end = top_block._start + top_block._nsamps
+        self.end_edit = QtGui.QLineEdit(self)
+        self.end_edit.setMinimumWidth(100)
+        self.end_edit.setMaximumWidth(100)
+        self.end_edit.setText(QtCore.QString("%1").arg(end))
+        self.end_edit.setValidator(self.size_val)
+        self.left_col_form.addRow("End:", self.end_edit)
+        self.connect(self.end_edit, QtCore.SIGNAL("returnPressed()"),
+                     self.update_xaxis_pos)
+
+        # Create a slider to move the position in the file
+        self.posbar = QtGui.QSlider(QtCore.Qt.Horizontal, self)
+        self.posbar.setMaximum(self.top_block._max_nsamps)
+        self.posbar.setPageStep(self.top_block._nsamps)
+        self.connect(self.posbar, QtCore.SIGNAL("valueChanged(int)"),
+                     self.update_xaxis_slider)
+        self.layout.addWidget(self.posbar, 2,2,1,1)
+
+        # Create Edit boxes for Y-axis min/max
+        self.y_range_val = QtGui.QDoubleValidator(top_block._y_min,
+                                                  top_block._y_max,
+                                                  4, self)
 
-    def update_points(self):
+        self.y_max_edit = QtGui.QLineEdit(self)
+        self.y_max_edit.setValidator(self.y_range_val)
+        self.y_max_edit.setMinimumWidth(100)
+        self.y_max_edit.setMaximumWidth(100)
+        self.left_col_form.addRow("Y Max:", self.y_max_edit)
+        self.connect(self.y_max_edit, QtCore.SIGNAL("returnPressed()"),
+                     self.update_yaxis_pos)
+
+        self.y_min_edit = QtGui.QLineEdit(self)
+        self.y_min_edit.setValidator(self.y_range_val)
+        self.y_min_edit.setMinimumWidth(100)
+        self.y_min_edit.setMaximumWidth(100)
+        self.left_col_form.addRow("Y Min:", self.y_min_edit)
+        self.connect(self.y_min_edit, QtCore.SIGNAL("returnPressed()"),
+                     self.update_yaxis_pos)
+
+        self.grid_check = QtGui.QCheckBox("Grid", self)
+        self.connect(self.grid_check, QtCore.SIGNAL("stateChanged(int)"),
+                     self.set_grid_check)
+        self.left_col_form.addWidget(self.grid_check)
+
+        self.gui_y_axis(top_block._y_value-top_block._y_range, top_block._y_value)
+
+        # Create a slider to move the plot's y-axis offset
+        self.ybar = QtGui.QSlider(QtCore.Qt.Vertical, self)
+        self.ybar.setMinimum(self.top_block._y_min)
+        self.ybar.setMaximum(self.top_block._y_max)
+        self.ybar.setPageStep(self.top_block._y_range)
+        self.ybar.setValue(self.top_block._y_value)
+        self.connect(self.ybar, QtCore.SIGNAL("valueChanged(int)"),
+                     self.update_yaxis_slider)
+        self.layout.addWidget(self.ybar, 1,1,1,1)
+
+        # Create an edit box for the Sample Rate
+        sr = top_block._samp_rate
+        self.samp_rate_edit = QtGui.QLineEdit(self)
+        self.samp_rate_edit.setMinimumWidth(100)
+        self.samp_rate_edit.setMaximumWidth(100)
+        self.samp_rate_edit.setText(QtCore.QString("%1").arg(sr))
+        self.left_col_form.addRow("Sample Rate:", self.samp_rate_edit)
+        self.connect(self.samp_rate_edit, QtCore.SIGNAL("returnPressed()"),
+                     self.update_samp_rate)
+
+        # Create an edit box for the center frequency
+        freq = top_block._center_freq
+        self.freq_edit = QtGui.QLineEdit(self)
+        self.freq_edit.setMinimumWidth(100)
+        self.freq_edit.setMaximumWidth(100)
+        self.freq_edit.setText(QtCore.QString("%1").arg(freq))
+        self.left_col_form.addRow("Frequency:", self.freq_edit)
+        self.connect(self.freq_edit, QtCore.SIGNAL("returnPressed()"),
+                     self.update_samp_rate)
+
+        self.resize(1000, 500)
+
+    def add_line_control(self, layout):
+        self._line_tabs = QtGui.QTabWidget()
+
+        self._line_pages = []
+        self._line_forms = []
+        self._label_edit = []
+        self._size_edit = []
+        self._color_edit = []
+        self._style_edit = []
+        self._marker_edit = []
+        self._alpha_edit = []
+        for n in xrange(self.top_block._nsigs):
+            self._line_pages.append(QtGui.QDialog())
+            self._line_forms.append(QtGui.QFormLayout(self._line_pages[-1]))
+
+            label = self.top_block.gui_snk.line_label(n)
+            self._label_edit.append(QtGui.QLineEdit(self))
+            self._label_edit[-1].setMinimumWidth(125)
+            self._label_edit[-1].setMaximumWidth(125)
+            self._label_edit[-1].setText(QtCore.QString("%1").arg(label))
+            self._line_forms[-1].addRow("Label:", self._label_edit[-1])
+            self.connect(self._label_edit[-1], QtCore.SIGNAL("returnPressed()"),
+                         self.update_line_label)
+
+            width = self.top_block.gui_snk.line_width(n)
+            self._size_edit.append(QtGui.QLineEdit(self))
+            self._size_edit[-1].setMinimumWidth(100)
+            self._size_edit[-1].setMaximumWidth(100)
+            self._size_edit[-1].setText(QtCore.QString("%1").arg(width))
+            self._line_forms[-1].addRow("Width:", self._size_edit[-1])
+            self.connect(self._size_edit[-1], QtCore.SIGNAL("returnPressed()"),
+                         self.update_line_size)
+
+            color = self.top_block.gui_snk.line_color(n)
+            self._color_edit.append(QtGui.QLineEdit(self))
+            self._color_edit[-1].setMinimumWidth(100)
+            self._color_edit[-1].setMaximumWidth(100)
+            self._color_edit[-1].setText(QtCore.QString("%1").arg(color))
+            self._line_forms[-1].addRow("Color:", self._color_edit[-1])
+            self.connect(self._color_edit[-1], QtCore.SIGNAL("returnPressed()"),
+                         self.update_line_color)
+
+            self._qtstyles = {"None": QtCore.Qt.NoPen,
+                              "Solid": QtCore.Qt.SolidLine,
+                              "Dash": QtCore.Qt.DashLine,
+                              "Dot": QtCore.Qt.DotLine,
+                              "DashDot": QtCore.Qt.DashDotLine,
+                              "DashDotDot": QtCore.Qt.DashDotDotLine}
+            self._style_edit.append(QtGui.QComboBox(self))
+            self._style_edit[-1].addItems(["None", "Solid", "Dash",
+                                          "Dot", "DashDot", "DashDotDot"])
+            self._style_edit[-1].setCurrentIndex(1)
+            self._line_forms[-1].addRow("Style:", self._style_edit[-1])
+            self.connect(self._style_edit[-1],
+                         QtCore.SIGNAL("currentIndexChanged(int)"),
+                         self.update_line_style)
+
+            # A bit dangerous, this. If QWT ever changes the lineup,
+            # we will have to adjust this, too. But we also can't
+            # really rely on PyQwt right now.
+            self._qwtmarkers = {"None": -1,
+                                "Circle": 0,
+                                "Rectangle": 1,
+                                "Diamond": 2,
+                                "Triangle": 3,
+                                "Down Triangle": 4,
+                                "Left Triangle": 6,
+                                "Right Triangle": 7,
+                                "Cross": 8,
+                                "X-Cross": 9,
+                                "Horiz. Line": 10,
+                                "Vert. Line": 11,
+                                "Star 1": 12,
+                                "Star 2": 13,
+                                "Hexagon": 14}
+            self._marker_edit.append(QtGui.QComboBox(self))
+            self._marker_edit[-1].addItems(["None", "Circle", "Rectangle", "Diamond",
+                                            "Triangle", "Down Triangle", "Left Triangle",
+                                            "Right Triangle", "Cross", "X-Cross",
+                                            "Horiz. Line", "Vert. Line", "Star 1",
+                                            "Star 2", "Hexagon"])
+            self._line_forms[-1].addRow("Marker:", self._marker_edit[-1])
+            self.connect(self._marker_edit[-1],
+                         QtCore.SIGNAL("currentIndexChanged(int)"),
+                         self.update_line_marker)
+
+            alpha_val = QtGui.QDoubleValidator(0, 1.0, 2, self)
+            alpha_val.setTop(1.0)
+            alpha = self.top_block.gui_snk.line_alpha(n)
+            self._alpha_edit.append(QtGui.QLineEdit(self))
+            self._alpha_edit[-1].setMinimumWidth(50)
+            self._alpha_edit[-1].setMaximumWidth(100)
+            self._alpha_edit[-1].setText(QtCore.QString("%1").arg(alpha))
+            self._alpha_edit[-1].setValidator(alpha_val)
+            self._line_forms[-1].addRow("Alpha:", self._alpha_edit[-1])
+            self.connect(self._alpha_edit[-1], QtCore.SIGNAL("returnPressed()"),
+                         self.update_line_alpha)
+
+            self._line_tabs.addTab(self._line_pages[-1], "{0}".format(label))
+
+        layout.addWidget(self._line_tabs)
+
+    def update_line_label(self):
+        index = self._line_tabs.currentIndex()
+        label = self._label_edit[index].text()
+        self._line_tabs.setTabText(index, label)
+        self.top_block.gui_snk.set_line_label(index, str(label))
+
+    def update_line_size(self):
+        index = self._line_tabs.currentIndex()
+        width = self._size_edit[index].text().toUInt()[0]
+        self.top_block.gui_snk.set_line_width(index, width)
+        self.update_line_alpha()
+
+    def update_line_color(self):
+        index = self._line_tabs.currentIndex()
+        color = str(self._color_edit[index].text())
+        self.top_block.gui_snk.set_line_color(index, color)
+        self.update_line_alpha()
+
+    def update_line_style(self, s_index):
+        index = self._line_tabs.currentIndex()
+        style_str = self._style_edit[index].itemText(s_index)
+        style = self._qtstyles[str(style_str)]
+        self.top_block.gui_snk.set_line_style(index, int(style))
+
+    def update_line_marker(self, m_index):
+        index = self._line_tabs.currentIndex()
+        marker_str = self._marker_edit[index].itemText(m_index)
+        marker = self._qwtmarkers[str(marker_str)]
+        self.top_block.gui_snk.set_line_marker(index, int(marker))
+
+    def update_line_alpha(self):
+        index = self._line_tabs.currentIndex()
+        alpha = self._alpha_edit[index].text().toDouble()[0]
+        self.top_block.gui_snk.set_line_alpha(index, alpha)
+
+    def update_xaxis_pos(self):
         newstart = self.start_edit.text().toUInt()[0]
         newend = self.end_edit.text().toUInt()[0]
         if(newstart != self._start or newend != self._end):
@@ -99,6 +309,54 @@ class dialog_box(QtGui.QWidget):
                 self.top_block.reset(newstart, newnsamps)
         self._start = newstart
         self._end = newend
+        self.posbar.setPageStep(self.top_block._nsamps)
+        self.posbar.setValue(self._start)
+
+    def update_xaxis_slider(self, value):
+        self._start = value
+        self._end = value + self.posbar.pageStep()
+
+        self.start_edit.setText("{0}".format(self._start))
+        self.end_edit.setText("{0}".format(self._end))
+
+        if(value != self.top_block._max_nsamps):
+            self.top_block.reset(self._start, self.posbar.pageStep())
+
+    def update_yaxis_pos(self):
+        newmin = self.y_min_edit.text().toDouble()[0]
+        newmax = self.y_max_edit.text().toDouble()[0]
+        if(newmin != self._y_min or newmax != self._y_max):
+            self._y_min = newmin
+            self._y_max = newmax
+            self.top_block._y_range = newmax - newmin
+            self.ybar.setValue(self._y_max)
+            self.top_block.set_y_axis(self._y_min, self._y_max)
+
+    def update_yaxis_slider(self, value):
+        self.top_block._y_value = value
+        self._y_min = value - self.top_block._y_range
+        self._y_max = value
+
+        self._y_min, self._y_max = self.top_block.set_y_axis(self._y_min, self._y_max)
+        
+        self.gui_y_axis(self._y_min, self._y_max)
+
+    def update_samp_rate(self):
+        sr = self.samp_rate_edit.text().toDouble()[0]
+        fr = self.freq_edit.text().toDouble()[0]
+        self.top_block.gui_snk.set_frequency_range(fr, sr)
+        self.top_block.reset(self.top_block._start,
+                             self.top_block._nsamps)
+
+    def gui_y_axis(self, ymin, ymax):
+        self.y_min_edit.setText("{0:.4f}".format(ymin))
+        self.y_max_edit.setText("{0:.4f}".format(ymax))
+
+    def set_grid_check(self, state):
+        if(state):
+            self.top_block.gui_snk.enable_grid(True)
+        else:
+            self.top_block.gui_snk.enable_grid(False)
 
     def save_figure(self):
         qpix = QtGui.QPixmap.grabWidget(self.top_block.pyWin)
@@ -124,4 +382,3 @@ class dialog_box(QtGui.QWidget):
             qpix.save(filename, "TIFF");
         else:
             qpix.save(filename, "JPEG");
-
author	Tom Rondeau <trondeau@vt.edu>	2013-01-22 13:57:37 -0500
committer	Tom Rondeau <trondeau@vt.edu>	2013-01-22 13:57:37 -0500
commit	b3860befa302bb723f74dce84d7ed7ab768d5854 (patch)
tree	6bfaccf901745f125df6f1e47485cef97192bef9 /gr-qtgui/apps/plot_form.py
parent	b55aab99aa9305eafcfa628774f0a4056bf763eb (diff)
parent	0f9ad0a68173762b83bbe7ed38f1708be294bcc6 (diff)