Newer
Older
Alex Diebold
committed
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
import os
import argparse
import pandas as pd
import matplotlib.pyplot as plt
from scipy import arange
from aeri_tools.io.dmv import housekeeping
def main(path, bounds = None):
#plt.rcParams['figure.figsize'] = [25, 15]
data = housekeeping.get_all_housekeeping(path)
if bounds != None:
data = data[(data['Time'] > bounds[0]) & (data['Time'] < bounds[1])]
variables = ['BBsupportStructureTemp', 'HBBapexTemp', 'HBBbottomTemp', 'HBBtopTemp', 'ABBapexTemp', 'ABBbottomTemp', 'ABBtopTemp', 'airNearBBsTemp', 'SCEtemp', 'sceneMirrorPosition']
fig, ax = plt.subplots(len(variables),figsize=(25,15),sharex=True)
for x,v in enumerate(variables):
name = v + ' '*30
plt.sca(ax[x])
plt.grid()
#ax = plt.subplot(len(vars), 1, x+1)
plt.xticks(arange(0,25))
#ax.grid(True)
plt.ylabel(name, rotation=0)
plt.plot(data['Time'], data[v])
'''
plt.sca(ax[0])
plt.plot(data['Time'], data['BBsupportStructureTemp'])
plt.sca(ax[1])
plt.plot(dmv['Time'], data['sceneMirrorPosition'])
'''
plt.savefig('/Users/adiebold/aeri_quality_control/testing/testing_bst/graphs/sgp_' + path[-12:-6] + '.png')
# plt.savefig('/Users/adiebold/aeri_quality_control/testing/bst_' + path[-12:-6] + '.png')
# print('/Users/adiebold/aeri_quality_control/testing/testing_bst/graphs/HBB' + path[-12:-6] + '.png')
# plt.show()
plt.clf()
if __name__ == '__main__':
'''
files = (['151130', (9, 13)], ['151201', (18, 20)], ['151202', (7, 11)],
['151205', (9, 11)], ['151207', (9, 11)])
for f in files:
f[0] = 'awr/AE' + f[0] + '/' + f[0] + 'B1.CXS'
main(f[0], f[1])
'''
parser = argparse.ArgumentParser()
parser.add_argument('filepath')
args = parser.parse_args()
print(args.filepath)
#amount of files to skip
skip_num = 85
curr_num = 0
print('skip_num = ', skip_num, '\n')
if os.path.isdir(args.filepath):
for filename_1 in os.listdir(args.filepath):
filename_1 = args.filepath + '/' + filename_1
filename_1 = filename_1.replace('//', '/')
if os.path.isdir(filename_1):
for filename_2 in os.listdir(filename_1):
filename_2 = filename_1 + '/' + filename_2
filename_2 = filename_2.replace('//', '/')
if (os.path.isfile(filename_2)
and filename_2.endswith('B1.CXS')):
curr_num += 1
if curr_num >= skip_num:
print(curr_num, ': ', filename_2)
main(filename_2)
else:
print(curr_num, ': ', filename_2, ' -- SKIPPED')
elif os.path.isfile(filename_1) and filename_1.endswith('B1.CXS'):
curr_num += 1
if curr_num >= skip_num:
print(curr_num, ': ', filename_1)
main(filename_1)
else:
print(curr_num, ': ', filename_1, ' -- SKIPPED')
elif os.path.isfile(args.filepath):
if args.filepath.endswith('B1.CXS'):
print(args.filepath)
main(args.filepath)