def test_000(self): # Just run some data through and make sure it doesn't puke. src_data = (1, 2, 3, 4, 5, 6, 7, 8, 9, 10) src = blocks.pdu_to_tagged_stream(blocks.byte_t) snk3 = blocks.tagged_stream_to_pdu(blocks.byte_t) snk2 = gr.vector_sink_b() snk = gr.tag_debug(1, "test") snk.set_display(False) dbg = gr.message_debug() # Test that the right number of ports exist. pi = snk3.message_ports_in() po = snk3.message_ports_out() self.assertEqual(pmt.pmt_length(pi), 0) self.assertEqual(pmt.pmt_length(po), 1) time.sleep(0.1) self.tb.connect(src, snk) self.tb.connect(src, snk2) self.tb.connect(src, snk3) self.tb.msg_connect(snk3, "pdus", dbg, "store") self.tb.start() # make our reference and message pmts port = pmt.pmt_intern("pdus") msg = pmt.pmt_cons(pmt.PMT_NIL, pmt.pmt_make_u8vector(16, 0xFF)) # post the message src.to_basic_block()._post(port, msg) # eww, what's that smell? while dbg.num_messages() < 1: time.sleep(0.5) self.tb.stop() self.tb.wait() # Get the vector of data from the vector sink result_data = snk2.data() # Get the vector of data from the message sink # Convert the message PMT as a pair into its vector result_msg = dbg.get_message(0) msg_vec = pmt.pmt_cdr(result_msg) #pmt.pmt_print(msg_vec) # Convert the PMT vector into a Python list msg_data = [] for i in xrange(16): msg_data.append(pmt.pmt_u8vector_ref(msg_vec, i)) actual_data = 16 * [ 0xFF, ] self.assertEqual(actual_data, list(result_data)) self.assertEqual(actual_data, msg_data)
def test_000(self): # Just run some data through and make sure it doesn't puke. src_data = (1, 2, 3, 4, 5, 6, 7, 8, 9, 10) src = blocks.pdu_to_tagged_stream(blocks.byte_t) snk3 = blocks.tagged_stream_to_pdu(blocks.byte_t) snk2 = gr.vector_sink_b() snk = gr.tag_debug(1, "test") snk.set_display(False) dbg = gr.message_debug() # Test that the right number of ports exist. pi = snk3.message_ports_in() po = snk3.message_ports_out() self.assertEqual(pmt.pmt_length(pi), 0) self.assertEqual(pmt.pmt_length(po), 1) time.sleep(0.1) self.tb.connect(src, snk) self.tb.connect(src, snk2) self.tb.connect(src, snk3) self.tb.msg_connect(snk3, "pdus", dbg, "store") self.tb.start() # make our reference and message pmts port = pmt.pmt_intern("pdus") msg = pmt.pmt_cons( pmt.PMT_NIL, pmt.pmt_make_u8vector(16, 0xFF)) # post the message src.to_basic_block()._post(port, msg) # eww, what's that smell? while dbg.num_messages() < 1: time.sleep(0.5) self.tb.stop() self.tb.wait() # Get the vector of data from the vector sink result_data = snk2.data() # Get the vector of data from the message sink # Convert the message PMT as a pair into its vector result_msg = dbg.get_message(0) msg_vec = pmt.pmt_cdr(result_msg) #pmt.pmt_print(msg_vec) # Convert the PMT vector into a Python list msg_data = [] for i in xrange(16): msg_data.append(pmt.pmt_u8vector_ref(msg_vec, i)) actual_data = 16*[0xFF,] self.assertEqual(actual_data, list(result_data)) self.assertEqual(actual_data, msg_data)
def pmt_to_dict(p): d = dict() items = pmt.pmt_dict_items(p) for i in range(pmt.pmt_length(items)): pair = pmt.pmt_nth(i, items) k = pmt.pmt_car(pair) v = pmt.pmt_cdr(pair) d[pmt_to_python(k)] = pmt_to_python(v) return d
def parse_extra_dict(p, info, VERBOSE=False): if(pmt.pmt_is_dict(p) is False): sys.stderr.write("Extra header is not a PMT dictionary: invalid or corrupt data file.\n") sys.exit(1) items = pmt.pmt_dict_items(p) nitems = pmt.pmt_length(items) for i in xrange(nitems): item = pmt.pmt_nth(i, items) key = pmt.pmt_symbol_to_string(pmt.pmt_car(item)) val = pmt.pmt_cdr(item) info[key] = val if(VERBOSE): print "{0}: ".format(key) pmt.pmt_print(val) return info
def test_001_t (self): """ Simplest possible test: put in zeros, then header, then payload, trigger signal, try to demux. The return signal from the header parser is faked via _post() """ n_zeros = 100 header = (1, 2, 3) payload = tuple(range(17)) data_signal = (0,) * n_zeros + header + payload trigger_signal = [0,] * len(data_signal) trigger_signal[n_zeros] = 1 data_src = gr.vector_source_f(data_signal, False) trigger_src = gr.vector_source_b(trigger_signal, False) hpd = digital.header_payload_demux( len(header), 1, 0, "frame_len", "detect", False, gr.sizeof_float ) self.assertEqual(pmt.pmt_length(hpd.message_ports_in()), 1) header_sink = gr.vector_sink_f() payload_sink = gr.vector_sink_f() self.tb.connect(data_src, (hpd, 0)) self.tb.connect(trigger_src, (hpd, 1)) self.tb.connect((hpd, 0), header_sink) self.tb.connect((hpd, 1), payload_sink) self.tb.start() time.sleep(.2) # Need this, otherwise, the next message is ignored hpd.to_basic_block()._post( pmt.pmt_intern('header_data'), pmt.pmt_from_long(len(payload)) ) while len(payload_sink.data()) < len(payload): time.sleep(.2) self.tb.stop() self.tb.wait() self.assertEqual(header_sink.data(), header) self.assertEqual(payload_sink.data(), payload)
def pmt_to_vector(p): v = list() for i in range(pmt.pmt_length(p)): elem = pmt.pmt_vector_ref(p, i) v.append(pmt_to_python(elem)) return v
def pmt_to_tuple(p): elems = list() for i in range(pmt.pmt_length(p)): elem = pmt.pmt_tuple_ref(p, i) elems.append(pmt_to_python(elem)) return tuple(elems)