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_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 = gr.pdu_to_tagged_stream(gr.pdu_byte) snk3 = gr.tagged_stream_to_pdu(gr.pdu_byte) snk2 = gr.vector_sink_b() snk = gr.tag_debug(1, "test") dbg = gr.message_debug() # Test that the right number of ports exist. pi = dbg.message_ports_in() po = dbg.message_ports_out() self.assertEqual(pmt.pmt_length(pi), 2) self.assertEqual(pmt.pmt_length(po), 0) pi = snk3.message_ports_in() po = snk3.message_ports_out() self.assertEqual(pmt.pmt_length(pi), 0) self.assertEqual(pmt.pmt_length(po), 1) #print "Message Debug input ports: " #pmt.pmt_print(pi) #print "Message Debug output ports: " #pmt.pmt_print(po) #print "Stream to PDU input ports: " #pmt.pmt_print(pi) #print "Stream to PDU output ports: " #pmt.pmt_print(po) 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) ) #print "printing port & msg" #pmt.pmt_print(port) #pmt.pmt_print(msg) # post the message src.to_basic_block()._post( port, msg ) 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 = gr.pdu_to_tagged_stream(gr.pdu_byte) snk3 = gr.tagged_stream_to_pdu(gr.pdu_byte) snk2 = gr.vector_sink_b() snk = gr.tag_debug(1, "test") dbg = gr.message_debug() # Test that the right number of ports exist. pi = dbg.message_ports_in() po = dbg.message_ports_out() self.assertEqual(pmt.pmt_length(pi), 3) self.assertEqual(pmt.pmt_length(po), 0) pi = snk3.message_ports_in() po = snk3.message_ports_out() self.assertEqual(pmt.pmt_length(pi), 0) self.assertEqual(pmt.pmt_length(po), 1) #print "Message Debug input ports: " #pmt.pmt_print(pi) #print "Message Debug output ports: " #pmt.pmt_print(po) #print "Stream to PDU input ports: " #pmt.pmt_print(pi) #print "Stream to PDU output ports: " #pmt.pmt_print(po) 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)) #print "printing port & msg" #pmt.pmt_print(port) #pmt.pmt_print(msg) # post the message src.to_basic_block()._post(port, msg) while (dbg.num_messages() < 1): time.sleep(0.1) 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)