pm100d_waist.py
4.91 KB
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
#!/usr/bin/env python
import argparse, time, os
import matplotlib.pyplot as plt
#==============================================================================
# Path
PATH = os.getcwd()
# File footer
OFILENAME = '130'
#==============================================================================
def parse():
"""
Specific parsing procedure for transfering data from Thorlabs PM100D.
:returns: populated namespace (parser)
"""
parser = argparse.ArgumentParser(description = 'Acquire data from Thorlabs PM100D',
epilog = 'Example: \'./pm100d.py -st 10 -o toto\' logs PM100D every 10 seconds to output file YYYYMMDD-HHMMSS-toto.dat')
parser.add_argument('-p',
action='store',
dest='path',
default=PATH,
help='Absolute path (default '+PATH+')')
parser.add_argument('-o',
action='store',
dest='ofile',
default=OFILENAME,
help='Output data filename (default '+OFILENAME+')')
args = parser.parse_args()
return args
#==============================================================================
class usbtmc:
def __init__(self, device):
self.device = device
self.FILE = os.open(device, os.O_RDWR)
def write(self, command):
os.write(self.FILE, command);
def read(self, length = 4000):
return os.read(self.FILE, length)
def getName(self):
self.write("*IDN?")
return self.read(300)
def sendReset(self):
self.write("*RST")
#==============================================================================
class instrument:
"""Class to control a SCPI compatible instrument"""
def __init__(self, device):
print 'Connecting to device %s...' %device
self.meas = usbtmc(device)
self.name = self.meas.getName()
print self.name
print ' --> Ok'
def write(self, command):
"""Send an arbitrary command directly to the scope"""
self.meas.write(command)
def read(self, command):
"""Read an arbitrary amount of data directly from the scope"""
return self.meas.read(command)
def reset(self):
"""Reset the instrument"""
self.meas.sendReset()
def value(self):
self.write('MEAS?')
return self.read(300)
#==============================================================================
def acqu_PM100D(instrument, path, ofile):
t0 = time.time()
filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(t0)) + '-' + ofile + '.dat'
print('Opening %s' %filename)
data_file = open(filename, 'wr', 0)
dx = 0.
xmes = []
Pmes = []
plt.ion()
fig = plt.figure()
ax1 = fig.add_subplot(111)
line1, = ax1.plot(xmes, Pmes, 'o')
# Infinite loop
while True:
try:
keypressed = raw_input("Press Enter to continue...\n")
if keypressed=='':
# Power values
sensors_values = instrument.value()
sensors_values = sensors_values.replace('E', 'e')
string = "%f\t%s" % (dx , sensors_values)
data_file.write(string) # Write in a file
print(string)
xmes.append(dx)
Pmes.append(float(sensors_values))
dx = dx + 0.05
elif keypressed.isdigit():
print('Closing %s' %filename)
data_file.close()
t0 = time.time()
filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(t0)) + '-' + keypressed + '.dat'
print('Opening %s' %filename)
data_file = open(filename, 'wr', 0)
dx = 0.
xmes = []
Pmes = []
line1, = ax1.plot(xmes, Pmes, 'o')
else:
raise KeyboardInterrupt
line1.set_data(xmes, Pmes)
ax1.relim()
ax1.autoscale_view()
fig.canvas.draw()
except Exception as ex:
print 'Exception during controler data reading: ' + str(ex)
except KeyboardInterrupt:
print '\n --> Disconnected'
data_file.close()
# To stop the loop in a clean way
break
#==============================================================================
def main():
"""
Main script
"""
# Parse command line
args = parse()
# path
path = args.path
# Data output filename
ofile = args.ofile
try:
pm100 = instrument("/dev/usbtmc0")
# acquisition with 2 sec. timeout
acqu_PM100D(pm100, path, ofile)
except Exception as ex:
print 'Oups '+str(ex)
print 'Program ending\n'
#==============================================================================
if __name__ == "__main__":
main()