Merge branch 'upstream' into dfsg-orig

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

diff --git a/gr-msdd6000/src/python_test/newtest.py b/gr-msdd6000/src/python_test/newtest.py
new file mode 100644 (file)
index 0000000..9596a06
--- /dev/null
+++ b/gr-msdd6000/src/python_test/newtest.py
@@ -0,0 +1,140 @@
+#!/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"
+
+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 = 3500; 
+f_mhz = 100;   
+f_hz = 0;      
+gain = 0;      
+window = 3;    #0=rect, 1=hanning, 2=hamming, 3=blackman
+
+samples = 12000;
+samples = samples*4;   #bytes of data we are requesting
+
+decim = 2;     #0-8   (3 => 2^3 = 8)  
+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, 0x18, f_mhz, f_hz, gain, samples, decim, sets);
+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));
+
+print "sent"
+
+
+
+count = 0;
+
+total_data = [];
+
+state = 0;
+
+vals = [];
+mags = [];
+re = [];
+
+sample_count = 0;
+IQ_bytes=0;
+
+while(True):
+       print state;
+       if(state==0):
+               data = UDPSock.recv(4);
+               [opcode] = struct.unpack("<I", data);
+               print "Opcode = %d"%(opcode);
+               if(opcode==1):
+
+                       # if UDP mode and sets_stream requested,
+                       # we do not get a header reply back, only data
+                       if(sets == 0):
+                               state = 1;
+                       else:
+                               state = 2;
+
+       elif(state==1):
+               data = UDPSock.recv(7*4);
+               args = struct.unpack("<IIIIIII", data);
+               print ["reply_len", "freq_mhz", "offset_hz", "gain", "sample_bytes", "decim", "sets_remain"];
+               print args;
+               IQ_bytes = args[0] - 7*4;
+               state =2;
+
+       elif(state==2):
+               data = UDPSock.recv(4);
+               [i,q] = struct.unpack("<hh", data);
+               tmp = complex(i,q);
+
+               re.append(i);
+               vals.append(tmp);
+               mags.append(abs(tmp));
+
+
+               sample_count = sample_count + 1;
+#              print "sample count %d"%(sample_count)  
+
+               IQ_bytes = IQ_bytes - 4;
+               if(IQ_bytes < 4):
+                       print "got all data (total %d)"%(sample_count);
+                       print "remaining: %d"%(IQ_bytes);
+                       break;
+
+
+UDPSock.close();
+
+print "done"
+nmags = []
+for i in mags:
+       if i == 0:
+               i=1;
+       nmags.append(i);
+
+
+subplot(2,1,1);
+plot(20*log10(nmags));
+
+fft_data = 20*log10(fftshift(fft(vals)));
+
+subplot(2,1,2);
+plot(fft_data);
+show();
+
+f = open(filename, "w");
+for sample in vals:
+       binchunk = struct.pack("<ff",float(sample.real), float(sample.imag) );
+       f.write(binchunk);
+f.close();
+
+