Plot-sink added

1 files changed, 57 insertions, 0 deletions

diff --git a/gr-input/python/plot_sink.py b/gr-input/python/plot_sink.py
new file mode 100755
index 000000000..1c1def3fc
--- /dev/null
+++ b/gr-input/python/plot_sink.py
@@ -0,0 +1,57 @@
+# Hacked from blks2/variable_sink.py
+# Requires modified Matplotsink code
+
+from gnuradio import gr
+import threading
+import numpy
+import matplotsink
+import Queue
+
+class _plot_sink_base(gr.hier_block2, threading.Thread):
+	"""
+	The thread polls the message queue for values and writes to matplotsink callback
+	"""
+
+	def __init__(self, parent, title, vlen, decim,gsz,zoom):
+		self._vlen = vlen
+		self._parent = parent
+		self._title = title
+		print "Initing block: %s" % title
+
+		self.plotQueue = Queue.Queue()
+		self.win = matplotsink.matplotsink(parent,title, self.plotQueue,gsz,zoom)		
+
+		self._item_size = self._size*self._vlen
+		#init hier block
+		gr.hier_block2.__init__(
+			self, 'plot_sink',
+			gr.io_signature(1, 1, self._item_size),
+			gr.io_signature(0, 0, 0),
+		)
+		#create blocks		
+		self._msgq = gr.msg_queue(2)
+		message_sink = gr.message_sink(self._item_size, self._msgq, False)
+		#connect
+		self.connect(self, message_sink)
+		#setup thread
+		threading.Thread.__init__(self)
+		self.setDaemon(True)
+		self.start() 
+	
+	def set_decim(self, decim): self._decimator.set_n(decim)
+
+	def run(self):
+		while True: #truncate to item size, convert to array, callback
+			msg = self._msgq.delete_head().to_string()[-self._item_size:]
+			arr = map(self._cast, numpy.fromstring(msg, self._numpy))
+			print "Sending value:" , arr
+			self.plotQueue.put(self._vlen > 1 and arr or arr[0])
+
+	def print_callback(self, array):
+		print array
+
+class plot_sink_b(_plot_sink_base): _numpy, _size, _cast = numpy.int8, gr.sizeof_char, int
+class plot_sink_s(_plot_sink_base): _numpy, _size, _cast = numpy.int16, gr.sizeof_short, int
+class plot_sink_i(_plot_sink_base): _numpy, _size, _cast = numpy.int32, gr.sizeof_int, int
+class plot_sink_f(_plot_sink_base): _numpy, _size, _cast = numpy.float32, gr.sizeof_float, float
+class plot_sink_c(_plot_sink_base): _numpy, _size, _cast = numpy.complex64, gr.sizeof_gr_complex, complex