1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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
|
