Merge commit 'v3.3.0' into upstream

[debian/gnuradio] / gr-msdd6000 / src / python_test / udp_stream_rate_test_plot_loop.py

diff --git a/gr-msdd6000/src/python_test/udp_stream_rate_test_plot_loop.py b/gr-msdd6000/src/python_test/udp_stream_rate_test_plot_loop.py
new file mode 100644 (file)
index 0000000..185afc4
--- /dev/null
+++ b/gr-msdd6000/src/python_test/udp_stream_rate_test_plot_loop.py
@@ -0,0 +1,150 @@
+#!/usr/bin/python
+
+from socket import *
+import string
+import time
+import struct;
+import random;
+import array;
+import cmath;
+from numpy import *;
+from numpy.fft import *;
+from pylab import *;
+
+myport = random.randint(1025,65535);
+filename = "output.dat";
+
+msdd_port = 10001
+#msdd_host = "10.45.4.43"
+msdd_host = "10.45.4.45"
+
+buf = 100000;
+
+my_udp_addr = ('',10001);
+my_udp_addr = ('10.45.1.229 ',10001);
+
+UDPSock = socket(AF_INET,SOCK_DGRAM);
+UDPSock.bind(my_udp_addr);
+
+#f_mhz = 3500; 
+f_mhz = 1500;  
+#f_mhz = 1000; 
+f_hz = 0;      
+gain = 80;     
+window = 3;    #0=rect, 1=hanning, 2=hamming, 3=blackman
+
+#samples = 65535;
+samples = 16384;
+#samples = samples*4;  #bytes of data we are requesting
+
+# decim 0-8 ( 3 - 8 )
+#decim = 5;    # rate ok
+decim = 4;
+decim = decim+16;      # +16 to use 16bit floats instead of 32 bit floats
+mode = 0;      #0=IQ, 1=MAG, 2=MAGDB
+#sets = 0;
+sets = 0xffffffff;
+
+size_int = 4;
+request_len = 6*size_int;      # 6 int items not including the 8 bytes for opcode and length fields
+print "request len = %d"%(request_len);
+
+raw_data = struct.pack("<IIIIIIII", 0x01, request_len, f_mhz, f_hz, gain, samples, decim, sets);
+
+data = raw_data;
+
+UDPSock.sendto(data, (msdd_host, msdd_port));
+start =  time.time();
+
+print "sent"
+
+
+
+count = 0;
+
+total_data = [];
+
+state = 0;
+
+vals = [];
+mags = [];
+re = [];
+
+sample_count = 0;
+IQ_bytes=0;
+
+
+numtocap = 1000;
+IQ_bytes = 4 * numtocap;
+
+#numbytes = 65536*100;
+numbytes = 65536;
+
+
+
+while(True):
+
+       data = [];
+       l = [];
+       num_rx = 0;
+       while(num_rx < numbytes):
+               data = UDPSock.recv(65536);
+               num_rx = num_rx + len(data);
+               l.append(data);
+
+       end = time.time();
+       
+       # send stop command
+       #halt_data = struct.pack("<II", 0x04, 0x00);
+       #UDPSock.sendto(halt_data, (msdd_host, msdd_port));
+       
+       # perform timing analysis
+       print "recieved %d bytes in %f sec"%(numbytes, end-start);
+       bytes_per_sec = numbytes / (end-start);
+       samples_per_sec = bytes_per_sec / 4;
+       MSPS = samples_per_sec / 1000000.0;
+       
+       print "Got %f MSPS"%(MSPS);
+       print "Expected %f MSPS"%(102.4/math.pow(2,(decim-16)));
+       
+       
+       # plot data
+       val_arr = [];
+       mag_arr = [];
+       
+       print "Repacking data..."
+       for li in l:
+               for p in range(0, len(li)/4):
+                       [i,q] = struct.unpack_from("<hh", li, p);
+                       val = complex(i,q);
+                       mag_arr.append(abs(val));
+                       val_arr.append(val);
+       
+       
+       print "Calculating Time Domain Power..."
+       tpwr = [];
+       for i in val_arr:
+               tpwr.append( (i*conj(i)).real );
+       
+       print "Calculating PSD..."
+       freqz = fft(val_arr);
+       
+       #freqz = [];
+       #
+       #for i in range(0, floor(len(val_arr)/2048)):
+       #       tmp = val_arr(range(i,i+2048));
+       #       if len(freqz) == 0:
+       #               freqz = tmp;
+       #       
+       
+       psd = (freqz * conj(freqz)).real;
+       
+       print "Plotting..."
+       subplot(2,1,1);
+       plot(tpwr);
+       subplot(2,1,2);
+       plot(10*log10(psd));
+       show();
+       
+UDPSock.close();
+