git.gag.com Git - debian/gnuradio/blob - gr-msdd6000/src/python_test/udp_stream_rate_test_plot_loop.py

   1 #!/usr/bin/python
   2
   3 from socket import *
   4 import string
   5 import time
   6 import struct;
   7 import random;
   8 import array;
   9 import cmath;
  10 from numpy import *;
  11 from numpy.fft import *;
  12 from pylab import *;
  13
  14 myport = random.randint(1025,65535);
  15 filename = "output.dat";
  16
  17 msdd_port = 10001
  18 #msdd_host = "10.45.4.43"
  19 msdd_host = "10.45.4.45"
  20
  21 buf = 100000;
  22
  23 my_udp_addr = ('',10001);
  24 my_udp_addr = ('10.45.1.229 ',10001);
  25
  26 UDPSock = socket(AF_INET,SOCK_DGRAM);
  27 UDPSock.bind(my_udp_addr);
  28
  29 #f_mhz = 3500;
  30 f_mhz = 1500;
  31 #f_mhz = 1000;
  32 f_hz = 0;
  33 gain = 80;
  34 window = 3;     #0=rect, 1=hanning, 2=hamming, 3=blackman
  35
  36 #samples = 65535;
  37 samples = 16384;
  38 #samples = samples*4;   #bytes of data we are requesting
  39
  40 # decim 0-8 ( 3 - 8 )
  41 #decim = 5;     # rate ok
  42 decim = 4;
  43 decim = decim+16;       # +16 to use 16bit floats instead of 32 bit floats
  44 mode = 0;       #0=IQ, 1=MAG, 2=MAGDB
  45 #sets = 0;
  46 sets = 0xffffffff;
  47
  48 size_int = 4;
  49 request_len = 6*size_int;       # 6 int items not including the 8 bytes for opcode and length fields
  50 print "request len = %d"%(request_len);
  51
  52 raw_data = struct.pack("<IIIIIIII", 0x01, request_len, f_mhz, f_hz, gain, samples, decim, sets);
  53
  54 data = raw_data;
  55
  56 UDPSock.sendto(data, (msdd_host, msdd_port));
  57 start =  time.time();
  58
  59 print "sent"
  60
  61
  62
  63 count = 0;
  64
  65 total_data = [];
  66
  67 state = 0;
  68
  69 vals = [];
  70 mags = [];
  71 re = [];
  72
  73 sample_count = 0;
  74 IQ_bytes=0;
  75
  76
  77 numtocap = 1000;
  78 IQ_bytes = 4 * numtocap;
  79
  80 #numbytes = 65536*100;
  81 numbytes = 65536;
  82
  83
  84
  85 while(True):
  86
  87         data = [];
  88         l = [];
  89         num_rx = 0;
  90         while(num_rx < numbytes):
  91                 data = UDPSock.recv(65536);
  92                 num_rx = num_rx + len(data);
  93                 l.append(data);
  94
  95         end = time.time();
  96
  97         # send stop command
  98         #halt_data = struct.pack("<II", 0x04, 0x00);
  99         #UDPSock.sendto(halt_data, (msdd_host, msdd_port));
 100
 101         # perform timing analysis
 102         print "recieved %d bytes in %f sec"%(numbytes, end-start);
 103         bytes_per_sec = numbytes / (end-start);
 104         samples_per_sec = bytes_per_sec / 4;
 105         MSPS = samples_per_sec / 1000000.0;
 106
 107         print "Got %f MSPS"%(MSPS);
 108         print "Expected %f MSPS"%(102.4/math.pow(2,(decim-16)));
 109
 110
 111         # plot data
 112         val_arr = [];
 113         mag_arr = [];
 114
 115         print "Repacking data..."
 116         for li in l:
 117                 for p in range(0, len(li)/4):
 118                         [i,q] = struct.unpack_from("<hh", li, p);
 119                         val = complex(i,q);
 120                         mag_arr.append(abs(val));
 121                         val_arr.append(val);
 122
 123
 124         print "Calculating Time Domain Power..."
 125         tpwr = [];
 126         for i in val_arr:
 127                 tpwr.append( (i*conj(i)).real );
 128
 129         print "Calculating PSD..."
 130         freqz = fft(val_arr);
 131
 132         #freqz = [];
 133         #
 134         #for i in range(0, floor(len(val_arr)/2048)):
 135         #       tmp = val_arr(range(i,i+2048));
 136         #       if len(freqz) == 0:
 137         #               freqz = tmp;
 138         #
 139
 140         psd = (freqz * conj(freqz)).real;
 141
 142         print "Plotting..."
 143         subplot(2,1,1);
 144         plot(tpwr);
 145         subplot(2,1,2);
 146         plot(10*log10(psd));
 147         show();
 148
 149 UDPSock.close();
 150