summaryrefslogtreecommitdiff
path: root/gr-msdd6000/src/python_test/udp_stream_test.py
blob: 6136d16c599152ce8956a166bb7b94ed484b8dc9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#!/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.46"

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 = 2500;	
f_hz = 0;	
gain = 0; 	
window = 3; 	#0=rect, 1=hanning, 2=hamming, 3=blackman

samples = 65535;
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, 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;


numtocap = 1000;
IQ_bytes = 4 * numtocap;

while(True):
	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 % 200 == 0):
		print IQ_bytes;
	if(IQ_bytes < 4):
		print "got all data (total %d)"%(sample_count);
		print "remaining: %d"%(IQ_bytes);
		break;


halt_data = struct.pack("<II", 0x04, 0x00);
UDPSock.sendto(halt_data, (msdd_host, msdd_port));



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();