allanplot.py
3.37 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
#!/usr/bin/env python
import argparse, allantools, numpy, csv, glob, Gnuplot, fnmatch, os
import matplotlib.pyplot as plt
#==============================================================================
# Default filename
FILENAME = '*.dat'
COLUMNS = '2'
RATE = 1
#==============================================================================
def parse():
"""
Specific parsing procedure for Allan Deviation plotting tool.
:returns: populated namespace (parser)
"""
parser = argparse.ArgumentParser(description = 'Plot Allan Deviation from timeseries file',
epilog = 'Example: \'./allanplot.py -f \'toto.dat\' -c 2 3\' compute and plot Allan Deviation form columns 2 and 3 of toto.dat')
parser.add_argument('-f',
action='store',
dest='filename',
default=FILENAME,
help='File(s) to import (default '+FILENAME+')')
parser.add_argument('-c',
nargs = '+',
action='store',
dest='columns',
default=COLUMNS,
help='Columns to import (default '+str(COLUMNS)+')')
parser.add_argument('-r',
action='store',
dest='rate',
default=RATE,
help='Time rate in seconds (default '+str(RATE)+')')
parser.add_argument('-R',
action='store_true',
dest='rel',
default=False,
help='Relative ADev')
args = parser.parse_args()
return args
#==============================================================================
def main():
"""
Main script
"""
# Parse command line
args = parse()
# filename
filename = args.filename
# columns
columns = map(int, args.columns)
# rate
rate = int(args.rate)
# rel
rel = bool(args.rel)
try:
list_files = []
filename = filename.split()
for name in filename:
list_files.extend(sorted(glob.glob(name)))
for name in list_files:
print(name)
data = []
for f in list_files:
with open(f, 'r') as dest_f:
data_iter = csv.reader(dest_f, delimiter = '\t', quotechar = '"')
temp_data = [value for value in data_iter]
data.extend(temp_data)
data = numpy.asarray(data, dtype = float)
del(temp_data, list_files, value, f)
g = Gnuplot.Gnuplot(persist = 1)
g('set logscale x')
g('set logscale y')
g('set grid')
g('set format y "%0.1e"')
g.xlabel('Tau (s)')
g.ylabel('Adev')
for i in columns:
if rel:
(tau2, ad, ade, adn) = allantools.adev(data[:,i]/data[:,i].mean(), rate=rate, data_type="freq", taus='decade')
else:
(tau2, ad, ade, adn) = allantools.adev(data[:,i], rate=rate, data_type="freq", taus='decade')
g.replot(Gnuplot.Data(tau2, ad, ade, with_='yerrorbars', title='%s : %s'%(name, str(i))))
except Exception as ex:
print 'Oups '+str(ex)
print 'Program ending\n'
#==============================================================================
if __name__ == "__main__":
main()