-
Notifications
You must be signed in to change notification settings - Fork 23
/
run_assoc.py
63 lines (58 loc) · 1.94 KB
/
run_assoc.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
""" Run associator
picks --> events
"""
import os, glob
import argparse
import numpy as np
from obspy import UTCDateTime
import associator_pal
import config
import warnings
warnings.filterwarnings("ignore")
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--pick_dir', type=str,
default='output/eg/picks')
parser.add_argument('--time_range', type=str,
default='20171003-20171004')
parser.add_argument('--sta_file', type=str,
default='input/example_pal_format1.sta')
parser.add_argument('--out_ctlg', type=str,
default='output/eg/catalog.dat')
parser.add_argument('--out_pha', type=str,
default='output/eg/phase.dat')
args = parser.parse_args()
# PAL config
cfg = config.Config()
get_picks = cfg.get_picks
sta_dict = cfg.get_sta_dict(args.sta_file)
associator = associator_pal.PS_Pair_Assoc(\
sta_dict,
xy_margin = cfg.xy_margin,
xy_grid = cfg.xy_grid,
z_grids = cfg.z_grids,
min_sta = cfg.min_sta,
ot_dev = cfg.ot_dev,
max_res = cfg.max_res,
max_drop = cfg.max_drop,
vp = cfg.vp)
# i/o paths
out_root = os.path.split(args.out_ctlg)[0]
if not os.path.exists(out_root): os.makedirs(out_root)
out_ctlg = open(args.out_ctlg,'w')
out_pha = open(args.out_pha,'w')
# get date range
start_date, end_date = [UTCDateTime(date) for date in args.time_range.split('-')]
print('run assoc: picks --> events')
print('time range: {} to {}'.format(start_date.date, end_date.date))
# for all days
num_day = (end_date.date - start_date.date).days
for day_idx in range(num_day):
# 1. get picks
date = start_date + day_idx*86400
picks = get_picks(date, args.pick_dir)
picks = picks[[net_sta in sta_dict for net_sta in picks['net_sta']]]
# 2. associate picks: picks --> events
associator.associate(picks, out_ctlg, out_pha)
out_pha.close()
out_ctlg.close()