-
Notifications
You must be signed in to change notification settings - Fork 51
/
track_mha.py
executable file
·38 lines (33 loc) · 1.16 KB
/
track_mha.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
#!/usr/bin/env python
import mht
import os
import sys
def usage():
print 'Usage: {} [srcdir] [srcpattern] [lookahead] [load data] [destmat]'.format(
os.path.basename(__file__))
print ' srdir: Directory in which to find saved eddy frames'
print ' srcpattern: Filename pattern used for eddies (ex. anticyc_ for'
print ' anticyc_19921014.mat)'
print ' lookahead: Boolean value to run the scan with lookahead enabled [0 or 1]'
print ' load data: Path to pre-generated data. Use 0 to load no data.'
print ' destmat: Path to save mat-file(s). (should NOT end in .mat)'
if len(sys.argv) != 6:
usage()
sys.exit(1)
srcdir = sys.argv[1]
srcpattern = sys.argv[2]
do_look = sys.argv[3].lower() in ['1', 'yes', 'true']
if sys.argv[4].lower() in ['0', 'no', 'false']:
load_path = None
else:
load_path = sys.argv[4]
destmat = sys.argv[5]
if load_path is None:
data = None
else:
print 'Loading old data...'
data = mht.load_tracks(load_path)
print 'Tracking...'
roots = mht.build_mht(mht.list_eddies(srcdir, srcpattern), do_lookahead=do_look, prev_data=data)
print 'Saving...'
mht.write_tracks(roots, destmat, mht.list_dates(srcdir, srcpattern))