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
|
#!/usr/bin/env python3
import csv
import getopt
import numpy as np
import os
import re
import struct
import sys
import tarfile
import matplotlib.pyplot as plt
from dfatool import running_mean, MIMOSA
opt = dict()
if __name__ == '__main__':
try:
optspec = (
'export= '
)
raw_opts, args = getopt.getopt(sys.argv[1:], '', optspec.split())
for option, parameter in raw_opts:
optname = re.sub(r'^--', '', option)
opt[optname] = parameter
if 'export' in opt:
opt['export'] = list(map(int, opt['export'].split(':')))
except getopt.GetoptError as err:
print(err)
sys.exit(2)
voltage = float(args[0])
shunt = float(args[1])
mimfile = args[2]
mim = MIMOSA(voltage, shunt)
charges, triggers = mim.load_file(mimfile)
charges = charges[:3000000]
currents = running_mean(mim.charge_to_current_nocal(charges), 10) * 1e-6
powers = currents * voltage
xr = np.arange(len(currents)) * 1e-5
if 'export' in opt:
xr = xr[opt['export'][0] : opt['export'][1]]
currents = currents[opt['export'][0] : opt['export'][1]]
powers = powers[opt['export'][0] : opt['export'][1]]
for pair in zip(xr, powers):
print('{} {}'.format(*pair))
plt.plot( xr, powers, "r-")
plt.xlabel('Time [s]')
plt.ylabel('Power [W]')
plt.grid(True)
plt.show()
|
