Add correct way to put picking results even when picking fails, now passing all tests

This commit is contained in:
Darius Arnold 2018-06-19 19:55:09 +02:00
parent f4750472c7
commit fe1e76f53a

View File

@ -184,14 +184,18 @@ class PickingResults(dict):
self.Perror = None # symmetrized picking error P onset
self.Serror = None # symmetrized picking error S onset
self.aicSflag = 0
self.aicPflag = 0
self.Pflag = 0
self.Sflag = 0
#self.aicSflag = 0
#self.aicPflag = 0
#self.Pflag = 0
self.Pmarker = []
self.Ao = None # Wood-Anderson peak-to-peak amplitude
self.picker = 'auto' # type of picks
# these get added during the construction of the p pick results dictionary
self.w0 = None
self.fc = None
self.Mw = None
def __setattr__(self, key, value):
self[key] = value
@ -253,6 +257,9 @@ class AutopickStation(object):
self.taper_max_percentage = 0.05
self.taper_type = 'hann'
# initialize picking results
self.p_results = PickingResults()
self.s_results = PickingResults()
def vprint(self, s):
"""Only print statement if verbose picking is set to true."""
@ -400,11 +407,25 @@ class AutopickStation(object):
def autopickstation(self):
try:
self.pick_p_phase()
#TODO handle exceptions correctly
#TODO handle exceptions correctly (goal is to be compatible with old code first)
# requires an overlook of what should be returned in case picking fails at various stages
except MissingTraceException as mte:
print(mte)
return
except PickingFailedException as pfe:
print(pfe)
if self.estream is not None and self.nstream is not None and len(self.estream) > 0 and len(self.nstream) > 0 and self.p_results.Pweight is not None and self.p_results.Pweight < 4:
try:
self.pick_s_phase()
# TODO: when an exception occurs, return picking results so far. This requires that the pick methods save their results in the instance member of PickingResults
except MissingTraceException as mte:
print(mte)
except PickingFailedException as pfe:
print(pfe)
self.finish_picking()
return {'P': self.p_results, 'S':self.s_results, 'station':self.ztrace.stats.station} #TODO method to format picking results as a dict correctly
def finish_picking(self):
# calculate "real" onset times, save them in PickingResults
@ -492,18 +513,22 @@ class AutopickStation(object):
self.vprint(msg)
tr_filt, z_copy = self.prepare_wfstream(self.zstream, self.p_params.bpz1[0], self.p_params.bpz1[1])
# save filtered trace in instance for later plotting
self.tr_filt_z = tr_filt
if self.p_params.use_taup is True and self.origin is not None:
Lc = np.inf # what is Lc? DA
try:
self.modify_starttimes_taupy()
except AttributeError as ae:
print(ae)
else:
#TODO handle case of no metadata/origin. Picking should continue without taupy.
if self.p_params.use_taup is False or self.origin:
Lc = self.p_params.pstop - self.p_params.pstart
Lwf = self.ztrace.stats.endtime - self.ztrace.stats.starttime
if Lwf < 0:
print('autopickstation: empty trace! Return!')
return
#Todo add correct exception here
Ldiff = Lwf - abs(Lc)
if Ldiff < 0 or self.p_params.pstop <= self.p_params.pstart:
@ -522,6 +547,8 @@ class AutopickStation(object):
self.p_params.addnoise)
else:
cf1 = None
# save cf1 for plotting
self.cf1 = cf1
assert isinstance(cf1, CharacteristicFunction), 'cf2 is not set ' \
'correctly: maybe the algorithm name ({algoP}) is ' \
'corrupted'.format(algoP=self.p_params.algoP)
@ -562,21 +589,21 @@ class AutopickStation(object):
minsiglength,
self.signal_length_params.noisefactor,
self.signal_length_params.minpercent, self.iplot, fig, linecolor)
if Pflag == 0:
Pmarker = 'shortsignallength'
Pweight = 9
if Pflag == 1:
if len(self.nstream) == 0 or len(self.estream) == 0:
if self.nstream == self.estream:
msg = 'One or more horizontal components missing!\n' \
'Skipping control function checkZ4S.'
self.vprint(msg)
else:
if self.iplot > 1:
fig, linecolor = get_fig_from_figdict(self.fig_dict, 'checkZ4s')
Pflag = checkZ4S(zne, aicpick.getpick(), self.s_params.zfac, self.p_params.tsnrz[2], self.iplot, fig, linecolor)
if Pflag == 0:
Pmarker = 'SinsteadP'
Pweight = 9
Pflag = checkZ4S(zne, aicpick.getpick(), self.s_params.zfac, self.p_params.tsnrz[2], self.iplot, fig, linecolor)
if Pflag == 0:
self.p_results.Pmarker = 'SinsteadP'
self.p_results.Pweight = 9
else:
self.p_results.Pmarker = 'shortsignallength'
self.p_results.Pweight = 9
# go on with processing if AIC onset passes quality control
slope = aicpick.getSlope()
if not slope: slope = 0
@ -598,6 +625,8 @@ class AutopickStation(object):
cf2 = ARZcf(z_copy, cuttimes2, self.p_params.tpred1z, self.p_params.Parorder, self.p_params.tdet1z, self.p_params.addnoise)
else:
cf2 = None
# save cf2 for plotting
self.cf2 = cf2
# get refined onset time from CF2
assert isinstance(cf2, CharacteristicFunction), 'cf2 is not set ' \
'correctly: maybe the algorithm name ({algoP}) is ' \
@ -605,42 +634,30 @@ class AutopickStation(object):
fig, linecolor = get_fig_from_figdict(self.fig_dict, 'refPpick')
refPpick = PragPicker(cf2, self.p_params.tsnrz, self.p_params.pickwinP, self.iplot, self.p_params.ausP,
self.p_params.tsmoothP, aicpick.getpick(), fig, linecolor)
mpickP = refPpick.getpick()
if mpickP is not None:
self.p_results.mpickP = refPpick.getpick()
if self.p_results.mpickP is not None:
# quality assessment, get earliest/latest pick and symmetrized uncertainty
fig, linecolor = get_fig_from_figdict(self.fig_dict, 'el_Ppick')
epickP, lpickP, Perror = earllatepicker(z_copy, self.p_params.nfacP, self.p_params.tsnrz, mpickP,
self.p_results.epickP, self.p_results.lpickP, self.p_results.Perror = earllatepicker(z_copy, self.p_params.nfacP, self.p_params.tsnrz, self.p_results.mpickP,
self.iplot, fig=fig, linecolor=linecolor)
SNRP, SNRPdB, Pnoiselevel = getSNR(z_copy, self.p_params.tsnrz, mpickP)
self.p_results.SNRP, self.p_results.SNRPdB, self.p_results.Pnoiselevel = getSNR(z_copy, self.p_params.tsnrz, self.p_results.mpickP)
# weight P-onset using symmetric error
#todo shorter expression for this
if Perror is None:
Pweight = 4
else:
if Perror <= self.p_params.timeerrorsP[0]:
Pweight = 0
elif self.p_params.timeerrorsP[0] < Perror <= self.p_params.timeerrorsP[1]:
Pweight = 1
elif self.p_params.timeerrorsP[1] < Perror <= self.p_params.timeerrorsP[2]:
Pweight = 2
elif self.p_params.timeerrorsP[2] < Perror <= self.p_params.timeerrorsP[3]:
Pweight = 3
elif Perror > self.p_params.timeerrorsP[3]:
Pweight = 4
if Pweight <= self.first_motion_params.minfmweight and SNRP >= self.first_motion_params.minFMSSNR:
self.p_results.Pweight = get_quality_class(self.p_results.Perror, self.p_params.timeerrorsP)
if self.p_results.Pweight <= self.first_motion_params.minfmweight and self.p_results.SNRP >= self.first_motion_params.minFMSNR:
fig, linecolor = get_fig_from_figdict(self.fig_dict, 'fm_picker')
FM = fmpicker(self.zstream, z_copy, self.first_motion_params.fmpickwin, mpickP, self.iplot,
self.p_results.FM = fmpicker(self.zstream, z_copy, self.first_motion_params.fmpickwin, self.p_results.mpickP, self.iplot,
fig, linecolor)
else:
FM = 'N'
self.p_results.FM = 'N'
msg = "autopickstation: P-weight: {0}, " \
"SNR: {1}, SNR[dB]: {2}, Polarity: {3}".format(Pweight, SNRP, SNRPdB, FM)
"SNR: {1}, SNR[dB]: {2}, Polarity: {3}".format(self.p_results.Pweight, self.p_results.SNRP, self.p_results.SNRPdB, self.p_results.FM)
print(msg)
msg = 'autopickstation: Refined P-Pick: {} s | P-Error: {} s'.format(mpickP, Perror)
msg = 'autopickstation: Refined P-Pick: {} s | P-Error: {} s'.format(self.p_results.mpickP, self.p_results.Perror)
print(msg)
Sflag = 1
self.p_results.aicpick = aicpick
else:
msg = 'Bad initial (AIC) P-pick, skipping this onset!\n' \
'AIC-SNR={0}, AIC-Slope={1}counts/s\n' \
@ -649,6 +666,10 @@ class AutopickStation(object):
self.p_params.minAICPSNR, self.p_params.minAICPslope)
self.vprint(msg)
Sflag = 0
else:
#todo add why did picking fail, which should be saved in the pick dictionary
raise PickingFailedException("Why did it fail")
def pick_s_phase(self):
def check_existence_ne_traces():
@ -852,7 +873,7 @@ class AutopickStation(object):
def get_fig_from_figdict(figdict, figkey):
"""
Helper method to extract a figure by name from dictionary
:param figdict:
:type figdict: dict
:param figkey:
@ -860,15 +881,20 @@ def get_fig_from_figdict(figdict, figkey):
:return:
:rtype:
"""
if figdict is None:
return None, None
fig = figdict.get(figkey, None)
linecolor = figdict.get('plot_style', 'k')
linecolor = linecolor['linecolor']['rgba_mpl']
return fig, linecolor
def autopickstation(wfstream, pickparam, verbose=False, iplot=0, fig_dict=None, metadata=None, origin=None):
station = AutopickStation(wfstream, pickparam, verbose, iplot, fig_dict, metadata, origin)
return station.autopickstation()
def autopickstation(wfstream, pickparam, verbose=False,
def nautopickstation(wfstream, pickparam, verbose=False,
iplot=0, fig_dict=None, metadata=None, origin=None):
"""
picks a single station