Commit b2e5864fc469da8688e70361999ec929c78613b1
1 parent
2e009c12df
Exists in
master
allanplot: add relative ADEV option `-R`
Showing 1 changed file with 12 additions and 1 deletions Inline Diff
allanplot.py
#!/usr/bin/env python | 1 | 1 | #!/usr/bin/env python | |
2 | 2 | |||
import argparse, allantools, numpy, csv, glob, Gnuplot, fnmatch, os | 3 | 3 | import argparse, allantools, numpy, csv, glob, Gnuplot, fnmatch, os | |
import matplotlib.pyplot as plt | 4 | 4 | import matplotlib.pyplot as plt | |
5 | 5 | |||
#============================================================================== | 6 | 6 | #============================================================================== | |
7 | 7 | |||
# Default filename | 8 | 8 | # Default filename | |
FILENAME = '*.dat' | 9 | 9 | FILENAME = '*.dat' | |
COLUMNS = '2' | 10 | 10 | COLUMNS = '2' | |
RATE = 1 | 11 | 11 | RATE = 1 | |
12 | 12 | |||
#============================================================================== | 13 | 13 | #============================================================================== | |
14 | 14 | |||
def parse(): | 15 | 15 | def parse(): | |
""" | 16 | 16 | """ | |
Specific parsing procedure for Allan Deviation plotting tool. | 17 | 17 | Specific parsing procedure for Allan Deviation plotting tool. | |
:returns: populated namespace (parser) | 18 | 18 | :returns: populated namespace (parser) | |
""" | 19 | 19 | """ | |
parser = argparse.ArgumentParser(description = 'Plot Allan Deviation from timeseries file', | 20 | 20 | 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') | 21 | 21 | epilog = 'Example: \'./allanplot.py -f \'toto.dat\' -c 2 3\' compute and plot Allan Deviation form columns 2 and 3 of toto.dat') | |
22 | 22 | |||
parser.add_argument('-f', | 23 | 23 | parser.add_argument('-f', | |
action='store', | 24 | 24 | action='store', | |
dest='filename', | 25 | 25 | dest='filename', | |
default=FILENAME, | 26 | 26 | default=FILENAME, | |
help='File(s) to import (default '+FILENAME+')') | 27 | 27 | help='File(s) to import (default '+FILENAME+')') | |
28 | 28 | |||
parser.add_argument('-c', | 29 | 29 | parser.add_argument('-c', | |
nargs = '+', | 30 | 30 | nargs = '+', | |
action='store', | 31 | 31 | action='store', | |
dest='columns', | 32 | 32 | dest='columns', | |
default=COLUMNS, | 33 | 33 | default=COLUMNS, | |
help='Columns to import (default '+str(COLUMNS)+')') | 34 | 34 | help='Columns to import (default '+str(COLUMNS)+')') | |
35 | 35 | |||
parser.add_argument('-r', | 36 | 36 | parser.add_argument('-r', | |
action='store', | 37 | 37 | action='store', | |
dest='rate', | 38 | 38 | dest='rate', | |
default=RATE, | 39 | 39 | default=RATE, | |
help='Time rate in seconds (default '+str(RATE)+')') | 40 | 40 | help='Time rate in seconds (default '+str(RATE)+')') | |
41 | 41 | |||
42 | parser.add_argument('-R', | |||
43 | action='store_true', | |||
44 | dest='rel', | |||
45 | default=False, | |||
46 | help='Relative ADev') | |||
47 | ||||
args = parser.parse_args() | 42 | 48 | args = parser.parse_args() | |
return args | 43 | 49 | return args | |
44 | 50 | |||
#============================================================================== | 45 | 51 | #============================================================================== | |
46 | 52 | |||
def main(): | 47 | 53 | def main(): | |
""" | 48 | 54 | """ | |
Main script | 49 | 55 | Main script | |
""" | 50 | 56 | """ | |
# Parse command line | 51 | 57 | # Parse command line | |
args = parse() | 52 | 58 | args = parse() | |
# filename | 53 | 59 | # filename | |
filename = args.filename | 54 | 60 | filename = args.filename | |
# columns | 55 | 61 | # columns | |
columns = map(int, args.columns) | 56 | 62 | columns = map(int, args.columns) | |
# rate | 57 | 63 | # rate | |
rate = int(args.rate) | 58 | 64 | rate = int(args.rate) | |
65 | # rel | |||
66 | rel = bool(args.rel) | |||
59 | 67 | |||
try: | 60 | 68 | try: | |
list_files = [] | 61 | 69 | list_files = [] | |
filename = filename.split() | 62 | 70 | filename = filename.split() | |
for name in filename: | 63 | 71 | for name in filename: | |
list_files.extend(sorted(glob.glob(name))) | 64 | 72 | list_files.extend(sorted(glob.glob(name))) | |
65 | 73 | |||
for name in list_files: | 66 | 74 | for name in list_files: | |
print(name) | 67 | 75 | print(name) | |
data = [] | 68 | 76 | data = [] | |
69 | 77 | |||
for f in list_files: | 70 | 78 | for f in list_files: | |
with open(f, 'r') as dest_f: | 71 | 79 | with open(f, 'r') as dest_f: | |
data_iter = csv.reader(dest_f, delimiter = '\t', quotechar = '"') | 72 | 80 | data_iter = csv.reader(dest_f, delimiter = '\t', quotechar = '"') | |
temp_data = [value for value in data_iter] | 73 | 81 | temp_data = [value for value in data_iter] | |
data.extend(temp_data) | 74 | 82 | data.extend(temp_data) | |
75 | 83 | |||
data = numpy.asarray(data, dtype = float) | 76 | 84 | data = numpy.asarray(data, dtype = float) | |
77 | 85 | |||
del(temp_data, list_files, value, f) | 78 | 86 | del(temp_data, list_files, value, f) | |
79 | 87 | |||
g = Gnuplot.Gnuplot(persist = 1) | 80 | 88 | g = Gnuplot.Gnuplot(persist = 1) | |
g('set logscale x') | 81 | 89 | g('set logscale x') | |
g('set logscale y') | 82 | 90 | g('set logscale y') |