marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

[edit] rearrangement fastens up a bit

Browse Source
This commit is contained in:
Sebastian Wehling-Benatelli 2016-05-11 14:50:52 +02:00
parent 67c11558f3
commit 79e429db8d
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pylot/core/io/phases.py
View File

@ -234,7 +234,7 @@ def picks_from_picksdict(picks):
for station, onsets in picks.items(): for station, onsets in picks.items():
print('Reading picks on station %s' % station) print('Reading picks on station %s' % station)
for label, phase in onsets.items(): for label, phase in onsets.items():
if not isinstance(phase, dict): if not isinstance(phase, dict) or len(phase) < 3:
continue continue
onset = phase['mpp'] onset = phase['mpp']
epp = phase['epp'] epp = phase['epp']
@ -276,7 +276,7 @@ def reassess_pilot_db(root_dir, out_dir=None, fn_param=None):
reassess_pilot_event(root_dir, evt, out_dir, fn_param) reassess_pilot_event(root_dir, evt, out_dir, fn_param)
def reassess_pilot_event(root_dir, event_id, out_dir=None, fn_param=None): def reassess_pilot_event(root_dir, event_id, out_dir=None, fn_param=None, verbosity=0):
from obspy import read from obspy import read
from pylot.core.io.inputs import AutoPickParameter from pylot.core.io.inputs import AutoPickParameter
@ -298,12 +298,10 @@ def reassess_pilot_event(root_dir, event_id, out_dir=None, fn_param=None):
for station in picks_dict.keys(): for station in picks_dict.keys():
fn_pattern = os.path.join(search_base, '{0}*'.format(station)) fn_pattern = os.path.join(search_base, '{0}*'.format(station))
try: try:
st = read(fn_pattern) st = read(fn_pattern, format='GSE2')
except TypeError as e: except TypeError as e:
print(e.message) print(e.message)
st = read(fn_pattern, format='GSE2') st = read(fn_pattern)
if not st:
raise RuntimeError('no waveform data found for station {station}'.format(station=station))
for phase in picks_dict[station].keys(): for phase in picks_dict[station].keys():
try: try:
mpp = picks_dict[station][phase]['mpp'] mpp = picks_dict[station][phase]['mpp']
@ -311,6 +309,11 @@ def reassess_pilot_event(root_dir, event_id, out_dir=None, fn_param=None):
print(e.message, station) print(e.message, station)
continue continue
sel_st = select_for_phase(st, phase) sel_st = select_for_phase(st, phase)
if not sel_st:
raise warnings.formatwarning(
'no waveform data found for station {station}'.format(
station=station), category=RuntimeWarning)
print(sel_st)
stime, etime = getGlobalTimes(sel_st) stime, etime = getGlobalTimes(sel_st)
rel_pick = mpp - stime rel_pick = mpp - stime
epp, lpp, spe = earllatepicker(sel_st, epp, lpp, spe = earllatepicker(sel_st,
@ -319,6 +322,8 @@ def reassess_pilot_event(root_dir, event_id, out_dir=None, fn_param=None):
Pick1=rel_pick, Pick1=rel_pick,
iplot=None, iplot=None,
) )
if epp is None or lpp is None:
continue
epp = stime + epp epp = stime + epp
lpp = stime + lpp lpp = stime + lpp
min_diff = 3 * st[0].stats.delta min_diff = 3 * st[0].stats.delta
@ -334,7 +339,7 @@ def reassess_pilot_event(root_dir, event_id, out_dir=None, fn_param=None):
if not out_dir: if not out_dir:
fnout_prefix = os.path.join(root_dir, event_id, '{0}.'.format(event_id)) fnout_prefix = os.path.join(root_dir, event_id, '{0}.'.format(event_id))
else: else:
fnout_prefix = os.path.join(out_dir, event_id, '{0}.'.format(event_id)) fnout_prefix = os.path.join(out_dir, '{0}.'.format(event_id))
evt.write(fnout_prefix + 'xml', format='QUAKEML') evt.write(fnout_prefix + 'xml', format='QUAKEML')
#evt.write(fnout_prefix + 'cnv', format='VELEST') #evt.write(fnout_prefix + 'cnv', format='VELEST')