bmarechal / FPC1000_viewer

Blame view

FPC1000_viewer.py 3.06 KB
edit raw normal view history
9534feb52   bmarechal   first commit 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
 
  #!/usr/bin/env python
  
  # -*- coding: utf-8 -*-
  
  from pyqtgraph.Qt import QtGui, QtCore
  import argparse, numpy, pyqtgraph, socket, sys
  
  #==============================================================================
  
  # Default filename
  IP = '192.168.0.11'
  PORT = 5555
  
  #==============================================================================
  
  def parse():
86ff3fbf6   bmarechal   replace 4 spaces ... 
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
  	"""
  	Specific parsing procedure for Allan Deviation plotting tool.
  	:returns: populated namespace (parser)
  	"""
  	parser = argparse.ArgumentParser(description = 'R&S FPC1000 spectrum viewer',
  									 epilog = 'Example: \'./FPC1000_viewer.py -ip \'192.168.0.2\' -p 1234\' plot current spectrum from given device')
  
  	parser.add_argument('-ip',
  						action='store',
  						dest='ip',
  						default=IP,
  						help='device IP (default '+IP+')')
  
  	parser.add_argument('-p',
  						action='store',
  						dest='port',
  						default=PORT,
  						help='device port (default '+str(PORT)+')')
  
  	args = parser.parse_args()
  	return args
9534feb52   bmarechal   first commit 
38
39
40
41
 
  
  #==============================================================================
  
  def send(sock, command):
86ff3fbf6   bmarechal   replace 4 spaces ... 
42
43
 
  	sock.send("%s
  "%command)
9534feb52   bmarechal   first commit 
44
45
46
47
 
  
  #==============================================================================
  
  def read(sock):
86ff3fbf6   bmarechal   replace 4 spaces ... 
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
  	ans = ''
  	nb_data_list = []
  	nb_data = ''
  	try:
  		while ans != '
  ':
  			ans = sock.recv(1)
  			nb_data_list.append(ans) # Return the number of data
  		list_size = len(nb_data_list)
  		for j in range (0, list_size):
  			nb_data = nb_data+nb_data_list[j]
  		return nb_data
  	except socket.timeout:
  		print "Socket timeout error when reading."
9534feb52   bmarechal   first commit 
62
63
64
65
 
  
  #==============================================================================
  
  def get_spectrum(sock):
86ff3fbf6   bmarechal   replace 4 spaces ... 
66
67
68
69
70
71
72
73
74
75
76
77
78
79
 
  	send(sock, 'FREQ:STAR?')
  	x1 = read(sock)
  	x1 = float(x1.replace('
  ',''))
  	send(sock, 'FREQ:STOP?')
  	x2 = read(sock)
  	x2 = float(x2.replace('
  ',''))
  	send(sock, 'TRAC?')
  	y = read(sock)
  	y = numpy.array(y.replace('
  ','').split(','), dtype='float')
  	x = numpy.linspace(x1, x2, len(y))
  	return (x, y)
9534feb52   bmarechal   first commit 
80
81
82
83
 
  
  #==============================================================================
  
  def main():
86ff3fbf6   bmarechal   replace 4 spaces ... 
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
 
  	"""
  	Main script
  	"""
  	# Parse command line
  	args = parse()
  	# ip
  	ip = str(args.ip)
  	# port
  	port = int(args.port)
  
  	try:
  		sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM,socket.IPPROTO_TCP)
  		sock.settimeout(10.0)
  		sock.connect((ip, port))
  
  		app = QtGui.QApplication([])
  
  		win = pyqtgraph.GraphicsWindow(title='Basic FPC1000 viewer (%s:%i)'%(ip, port))
  
  		pyqtgraph.setConfigOptions(antialias=True)
  
  		global curve, ptr, p1
  		p1 = win.addPlot(title="Spectrum")
  		curve = p1.plot(pen='y')
  		ptr = 0
  		def update():
  			global curve, ptr, p1
  			(x_data, y_data) = get_spectrum(sock)
  			curve.setData(x = x_data, y = y_data)
  			if ptr == 0:
  				p1.enableAutoRange('xy', False)  ## stop auto-scaling after the first data set is plotted
  			ptr += 1
  		timer = QtCore.QTimer()
  		timer.timeout.connect(update)
  		timer.start(500)
  
  		if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
  			QtGui.QApplication.instance().exec_()
  
  	except:
  		pass
  
  		## Start Qt event loop unless running in interactive mode or using pyside.
9534feb52   bmarechal   first commit 
127
 
  if __name__ == '__main__':
86ff3fbf6   bmarechal   replace 4 spaces ... 
128
 
  	main()