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

[change] autopickstation works with new Metadata class

Browse Source
This commit is contained in:
Darius Arnold 2018-08-03 13:47:46 +02:00
parent 86f6584d91
commit 2b9cf655ae
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -446,6 +446,17 @@ class AutopickStation(object):
:rtype: None :rtype: None
""" """
def get_seed_id():
"""
Returns seed id of ztrace
:return: Seed id with format Network.Station.Location.Channel
:rtype: str
"""
stats = self.ztrace.stats
id = "{network}.{station}.{location}.{channel}"
id = id.format(network=stats.network, station=stats.station, location=stats.location, channel=stats.channel)
return id
def create_arrivals(metadata, origin, station_id, taup_model): def create_arrivals(metadata, origin, station_id, taup_model):
""" """
Create List of arrival times for all phases for a given origin and station Create List of arrival times for all phases for a given origin and station
@ -462,8 +473,8 @@ class AutopickStation(object):
:raises: :raises:
AttributeError when no metadata or source origins is given AttributeError when no metadata or source origins is given
""" """
parser = metadata[1] id = get_seed_id()
station_coords = get_source_coords(parser, station_id) station_coords = metadata.get_coordinates(id)
source_origin = origin[0] source_origin = origin[0]
model = TauPyModel(taup_model) model = TauPyModel(taup_model)
arrivals = model.get_travel_times_geo(source_depth_in_km=source_origin.depth, arrivals = model.get_travel_times_geo(source_depth_in_km=source_origin.depth,
@ -500,7 +511,7 @@ class AutopickStation(object):
print('autopickstation: use_taup flag active.') print('autopickstation: use_taup flag active.')
# catch missing metadata or origin information. Onset calculation is stopped, given cuttimes are then used. # catch missing metadata or origin information. Onset calculation is stopped, given cuttimes are then used.
if not self.metadata[1]: if not self.metadata:
raise AttributeError('Warning: Could not use TauPy to estimate onsets as there are no metadata given.') raise AttributeError('Warning: Could not use TauPy to estimate onsets as there are no metadata given.')
if not self.origin: if not self.origin:
raise AttributeError('No source origins given!') raise AttributeError('No source origins given!')