-
Notifications
You must be signed in to change notification settings - Fork 1
/
process.py
68 lines (56 loc) · 2.27 KB
/
process.py
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""This script runs processing functions."""
from __future__ import print_function
import os
from pyrm2tt.processing import *
import matplotlib.pyplot as plt
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Process data from the UNH "
"RM2 tow tank experiment.")
parser.add_argument("--single-run", "-r", nargs=2, help="Process a single "
"run; args=[section, nrun]")
parser.add_argument("--single-tare-drag", help="Process a tare drag run")
parser.add_argument("--single-tare-torque",
help="Process a tare torque run")
parser.add_argument("--section", help="Process a test matrix section")
parser.add_argument("--tare-drag", help="Process tare drag runs",
action="store_true")
parser.add_argument("--tare-torque", help="Process tare torque runs",
action="store_true")
parser.add_argument("--all", "-a", action="store_true",
help="Process all data")
parser.add_argument("--plot", action="store_true", default=False,
help="Create plots (if applicable)")
args = parser.parse_args()
if args.plot:
from pxl.styleplot import set_sns
set_sns()
plt.rcParams["axes.grid"] = True
if args.single_run:
# Dealing with individual runs
section = args.single_run[0]
nrun = int(args.single_run[1])
print("Processing {} run {}".format(section, nrun))
r = Run(section, nrun)
if args.plot:
r.plot_perf("cp")
r.plot_wake()
r.print_perf_stats()
r.print_wake_stats()
if args.single_tare_drag:
process_tare_drag(args.single_tare_drag, plot=args.plot)
if args.tare_drag:
batch_process_tare_drag(plot=args.plot)
if args.single_tare_torque:
process_tare_torque(args.single_tare_torque, plot=args.plot)
if args.tare_torque:
batch_process_tare_torque(plot=args.plot)
if args.section:
print("Processing {}".format(args.section))
batch_process_section(args.section)
if args.all:
batch_process_all()
if args.plot:
plt.show()