Merged changeset r9241:9289 from jblum/glwxgui into trunk. Adds OpenGL versions of fftsink, waterfallsink, and scopesink, and new constsink. See README.gl for use. (Josh Blum)

git-svn-id: http://gnuradio.org/svn/gnuradio/trunk@9290 221aa14e-8319-0410-a670-987f0aec2ac5

1 files changed, 183 insertions, 0 deletions

diff --git a/gr-wxgui/src/python/scopesink_gl.py b/gr-wxgui/src/python/scopesink_gl.py
new file mode 100644
index 000000000..727e1dc0a
--- /dev/null
+++ b/gr-wxgui/src/python/scopesink_gl.py
@@ -0,0 +1,183 @@
+#
+# Copyright 2008 Free Software Foundation, Inc.
+#
+# This file is part of GNU Radio
+#
+# GNU Radio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3, or (at your option)
+# any later version.
+#
+# GNU Radio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with GNU Radio; see the file COPYING.  If not, write to
+# the Free Software Foundation, Inc., 51 Franklin Street,
+# Boston, MA 02110-1301, USA.
+#
+
+##################################################
+# Imports
+##################################################
+import scope_window
+import common
+from gnuradio import gr
+from pubsub import pubsub
+from constants import *
+
+##################################################
+# Scope sink block (wrapper for old wxgui)
+##################################################
+class _scope_sink_base(gr.hier_block2, common.prop_setter):
+	"""!
+	A scope block with a gui window.
+	"""
+
+	def __init__(
+		self,
+		parent,
+		title='',
+		sample_rate=1,
+		size=scope_window.DEFAULT_WIN_SIZE,
+		frame_decim=None, #ignore (old wrapper)
+		v_scale=scope_window.DEFAULT_V_SCALE,
+		t_scale=None,
+		num_inputs=1,
+		ac_couple=False,
+		xy_mode=False,
+		frame_rate=scope_window.DEFAULT_FRAME_RATE,
+	):
+		if t_scale is None: t_scale = 0.001
+		#init
+		gr.hier_block2.__init__(
+			self,
+			"scope_sink",
+			gr.io_signature(num_inputs, num_inputs, self._item_size),
+			gr.io_signature(0, 0, 0),
+		)
+		#scope
+		msgq = gr.msg_queue(2)
+		scope = gr.oscope_sink_f(sample_rate, msgq)
+		#connect
+		if self._real:
+			for i in range(num_inputs):
+				self.connect((self, i), (scope, i))
+		else:
+			for i in range(num_inputs):
+				c2f = gr.complex_to_float() 
+				self.connect((self, i), c2f)
+				self.connect((c2f, 0), (scope, 2*i+0))
+				self.connect((c2f, 1), (scope, 2*i+1))
+			num_inputs *= 2
+		#controller
+		self.controller = pubsub()
+		self.controller.subscribe(SAMPLE_RATE_KEY, scope.set_sample_rate)
+		self.controller.publish(SAMPLE_RATE_KEY, scope.sample_rate)
+		def set_trigger_level(level):
+			if level == '': scope.set_trigger_level_auto()
+			else: scope.set_trigger_level(level)
+		self.controller.subscribe(SCOPE_TRIGGER_LEVEL_KEY, set_trigger_level)
+		def set_trigger_mode(mode):
+			if mode == 0: mode = gr.gr_TRIG_AUTO
+			elif mode < 0: mode = gr.gr_TRIG_NEG_SLOPE
+			elif mode > 0: mode = gr.gr_TRIG_POS_SLOPE
+			else: return
+			scope.set_trigger_mode(mode)
+		self.controller.subscribe(SCOPE_TRIGGER_MODE_KEY, set_trigger_mode)
+		self.controller.subscribe(SCOPE_TRIGGER_CHANNEL_KEY, scope.set_trigger_channel)
+		#start input watcher
+		def setter(p, k, x): # lambdas can't have assignments :(
+		    p[k] = x
+		common.input_watcher(msgq, lambda x: setter(self.controller, MSG_KEY, x))
+		#create window
+		self.win = scope_window.scope_window(
+			parent=parent,
+			controller=self.controller,
+			size=size,
+			title=title,
+			frame_rate=frame_rate,
+			num_inputs=num_inputs,
+			sample_rate_key=SAMPLE_RATE_KEY,
+			t_scale=t_scale,
+			v_scale=v_scale,
+			ac_couple=ac_couple,
+			xy_mode=xy_mode,
+			scope_trigger_level_key=SCOPE_TRIGGER_LEVEL_KEY,
+			scope_trigger_mode_key=SCOPE_TRIGGER_MODE_KEY,
+			scope_trigger_channel_key=SCOPE_TRIGGER_CHANNEL_KEY,
+			msg_key=MSG_KEY,
+		)
+		#register callbacks from window for external use
+		for attr in filter(lambda a: a.startswith('set_'), dir(self.win)):
+			setattr(self, attr, getattr(self.win, attr))
+		self._register_set_prop(self.controller, SAMPLE_RATE_KEY)
+
+class scope_sink_f(_scope_sink_base):
+	_item_size = gr.sizeof_float
+	_real = True
+
+class scope_sink_c(_scope_sink_base):
+	_item_size = gr.sizeof_gr_complex
+	_real = False
+
+#backwards compadible wrapper (maybe only grc uses this)
+class constellation_sink(scope_sink_c):
+	def __init__(self, **kwargs):
+		kwargs['xy_mode'] = True
+		scope_sink_c.__init__(self, **kwargs)
+
+# ----------------------------------------------------------------
+# Stand-alone test application
+# ----------------------------------------------------------------
+
+import wx
+from gnuradio.wxgui import stdgui2
+
+class test_top_block (stdgui2.std_top_block):
+    def __init__(self, frame, panel, vbox, argv):
+        stdgui2.std_top_block.__init__ (self, frame, panel, vbox, argv)
+
+        if len(argv) > 1:
+            frame_decim = int(argv[1]) 
+        else:
+            frame_decim = 1
+
+        if len(argv) > 2:
+            v_scale = float(argv[2])  # start up at this v_scale value
+        else:
+            v_scale = None  # start up in autorange mode, default
+
+        if len(argv) > 3:
+            t_scale = float(argv[3])  # start up at this t_scale value
+        else:
+            t_scale = .00003  # old behavior
+
+        print "frame decim %s  v_scale %s  t_scale %s" % (frame_decim,v_scale,t_scale)
+            
+        input_rate = 1e6
+
+        # Generate a complex sinusoid
+        self.src0 = gr.sig_source_c (input_rate, gr.GR_SIN_WAVE, 25.1e3, 1e3)
+
+        # We add this throttle block so that this demo doesn't suck down
+        # all the CPU available.  You normally wouldn't use it...
+        self.thr = gr.throttle(gr.sizeof_gr_complex, input_rate)
+
+        scope = scope_sink_c (panel,"Secret Data",sample_rate=input_rate,
+                              frame_decim=frame_decim,
+                              v_scale=v_scale, t_scale=t_scale)
+        vbox.Add (scope.win, 1, wx.EXPAND)
+
+        # Ultimately this will be
+        # self.connect("src0 throttle scope")
+	self.connect(self.src0, self.thr, scope) 
+
+def main ():
+    app = stdgui2.stdapp (test_top_block, "O'Scope Test App")
+    app.MainLoop ()
+
+if __name__ == '__main__':
+    main ()