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

Move Trace Extraction to __init__ and copy over existing hor. trace if one is missing

Browse Source
This commit is contained in:
Darius Arnold 2018-06-19 18:42:32 +02:00
parent 7b6f64b46a
commit 5b5da2e40d
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pylot/core/pick/autopick.py
View File

@ -225,6 +225,21 @@ class AutopickStation(object):
self.network_name = wfstream[0].stats.network self.network_name = wfstream[0].stats.network
self.station_id = '{}.{}'.format(self.network_name, self.station_name) self.station_id = '{}.{}'.format(self.network_name, self.station_name)
# save streams and traces
# TODO: error handling of missing traces
self.zstream, self.nstream, self.estream = self.get_components_from_waveformstream()
self.ztrace = self.zstream[0]
try:
self.ntrace = self.nstream[0]
except IndexError:
self.nstream = self.estream
self.ntrace = self.nstream[0]
try:
self.etrace = self.estream[0]
except IndexError:
self.estream = self.nstream
self.etrace = self.estream[0]
# default values used in old autopickstation function #TODO way for user to set those # default values used in old autopickstation function #TODO way for user to set those
self.detrend_type = 'demean' self.detrend_type = 'demean'
self.filter_type = 'bandpass' self.filter_type = 'bandpass'
@ -376,9 +391,6 @@ class AutopickStation(object):
self.p_params.pstop = min(self.p_params.pstop, len(self.ztrace) * self.ztrace.stats.delta) self.p_params.pstop = min(self.p_params.pstop, len(self.ztrace) * self.ztrace.stats.delta)
def autopickstation(self): def autopickstation(self):
self.zstream, self.nstream, self.estream = self.get_components_from_waveformstream()
self.ztrace, self.ntrace, self.etrace = self.zstream[0], self.nstream[0], self.estream[0]
try: try:
self.pick_p_phase() self.pick_p_phase()
#TODO handle exceptions correctly #TODO handle exceptions correctly