Source code for obspy.clients.earthworm.client

# -*- coding: utf-8 -*-
"""
Earthworm Wave Server client for ObsPy.

:copyright:
    The ObsPy Development Team (devs@obspy.org) & Victor Kress
:license:
    GNU Lesser General Public License, Version 3
    (https://www.gnu.org/copyleft/lesser.html)

.. seealso:: http://www.isti2.com/ew/PROGRAMMER/wsv_protocol.html
"""
from fnmatch import fnmatch

from obspy import Stream, UTCDateTime
from obspy.clients.earthworm.waveserver import get_menu, read_wave_server_v


[docs] class Client(object): """ A Earthworm Wave Server client. :type host: str :param host: Host name of the remote Earthworm Wave Server server. :type port: int :param port: Port of the remote Earthworm Wave Server server. :type timeout: int, optional :param timeout: Seconds before a connection timeout is raised (default is ``None``). :type debug: bool, optional :param debug: Enables verbose output of the connection handling (default is ``False``). """
[docs] def __init__(self, host, port, timeout=None, debug=False): """ Initializes a Earthworm Wave Server client. See :class:`obspy.clients.earthworm.client.Client` for all parameters. """ self.host = host self.port = port self.timeout = timeout self.debug = debug
[docs] def get_waveforms(self, network, station, location, channel, starttime, endtime, cleanup=True): """ Retrieves waveform data from Earthworm Wave Server and returns an ObsPy Stream object. :type filename: str :param filename: Name of the output file. :type network: str :param network: Network code, e.g. ``'UW'``. :type station: str :param station: Station code, e.g. ``'TUCA'``. :type location: str :param location: Location code, e.g. ``'--'``. :type channel: str :param channel: Channel code, e.g. ``'BHZ'``. Last character (i.e. component) can be a wildcard ('?' or '*') to fetch `Z`, `N` and `E` component. :type starttime: :class:`~obspy.core.utcdatetime.UTCDateTime` :param starttime: Start date and time. :type endtime: :class:`~obspy.core.utcdatetime.UTCDateTime` :param endtime: End date and time. :return: ObsPy :class:`~obspy.core.stream.Stream` object. :type cleanup: bool :param cleanup: Specifies whether perfectly aligned traces should be merged or not. See :meth:`obspy.core.stream.Stream.merge` for ``method=-1``. .. rubric:: Example >>> from obspy.clients.earthworm import Client >>> client = Client("pubavo1.wr.usgs.gov", 16022) >>> dt = UTCDateTime() - 15000 # now - 15000 seconds >>> st = client.get_waveforms('AV', 'AKV', '', 'BHE', dt, dt + 10) >>> st.plot() # doctest: +SKIP >>> st = client.get_waveforms('AV', 'AKV', '', 'BH*', dt, dt + 10) >>> st.plot() # doctest: +SKIP .. plot:: from obspy.clients.earthworm import Client from obspy import UTCDateTime client = Client("pubavo1.wr.usgs.gov", 16022, timeout=5) dt = UTCDateTime() - 15000 # now - 15000 seconds st = client.get_waveforms('AV', 'AKV', '', 'BHE', dt, dt + 10) st.plot() st = client.get_waveforms('AV', 'AKV', '', 'BH*', dt, dt + 10) st.plot() """ # replace wildcards in last char of channel and fetch all 3 components if channel[-1] in "?*": st = Stream() for comp in ("Z", "N", "E"): channel_new = channel[:-1] + comp st += self.get_waveforms(network, station, location, channel_new, starttime, endtime, cleanup=cleanup) return st if location == '': location = '--' scnl = (station, channel, network, location) # fetch waveform tbl = read_wave_server_v(self.host, self.port, scnl, starttime, endtime, timeout=self.timeout, cleanup=cleanup) # create new stream st = Stream() for tb in tbl: st.append(tb.get_obspy_trace()) if cleanup: st._cleanup() st.trim(starttime, endtime) return st
[docs] def save_waveforms(self, filename, network, station, location, channel, starttime, endtime, format="MSEED", cleanup=True): """ Writes a retrieved waveform directly into a file. :type filename: str :param filename: Name of the output file. :type network: str :param network: Network code, e.g. ``'UW'``. :type station: str :param station: Station code, e.g. ``'TUCA'``. :type location: str :param location: Location code, e.g. ``''``. :type channel: str :param channel: Channel code, e.g. ``'BHZ'``. Last character (i.e. component) can be a wildcard ('?' or '*') to fetch `Z`, `N` and `E` component. :type starttime: :class:`~obspy.core.utcdatetime.UTCDateTime` :param starttime: Start date and time. :type endtime: :class:`~obspy.core.utcdatetime.UTCDateTime` :param endtime: End date and time. :type format: str, optional :param format: Output format. One of ``"MSEED"``, ``"GSE2"``, ``"SAC"``, ``"SACXY"``, ``"Q"``, ``"SH_ASC"``, ``"SEGY"``, ``"SU"``, ``"WAV"``. See the Supported Formats section in method :meth:`~obspy.core.stream.Stream.write` for a full list of supported formats. Defaults to ``'MSEED'``. :type cleanup: bool :param cleanup: Specifies whether perfectly aligned traces should be merged or not. See :meth:`~obspy.core.stream.Stream.merge`, `method` -1 or :meth:`~obspy.core.stream.Stream._cleanup`. :return: None .. rubric:: Example >>> from obspy.clients.earthworm import Client >>> client = Client("pubavo1.wr.usgs.gov", 16022) >>> t = UTCDateTime() - 2000 # now - 2000 seconds >>> client.save_waveforms('AV.AKV.--.BHE.mseed', ... 'AV', 'AKV', '', 'BHE', ... t, t + 10, format='MSEED') # doctest: +SKIP """ st = self.get_waveforms(network, station, location, channel, starttime, endtime, cleanup=cleanup) st.write(filename, format=format)
[docs] def get_availability(self, network="*", station="*", location="*", channel="*"): """ Gets a list of data available on the server. This method returns information about what time series data is available on the server. The query can optionally be restricted to specific network, station, channel and/or location criteria. :type network: str :param network: Network code, e.g. ``'UW'``, wildcards allowed. :type station: str :param station: Station code, e.g. ``'TUCA'``, wildcards allowed. :type location: str :param location: Location code, e.g. ``'--'``, wildcards allowed. :type channel: str :param channel: Channel code, e.g. ``'BHZ'``, wildcards allowed. :rtype: list :return: List of tuples with information on the available data. One tuple consists of network, station, location, channel (all strings), start time and end time (both as :class:`~obspy.core.utcdatetime.UTCDateTime`). .. rubric:: Example >>> from obspy.clients.earthworm import Client >>> client = Client("pubavo1.wr.usgs.gov", 16022, timeout=5) >>> response = client.get_availability( ... network="AV", station="AKV", channel="BH*") >>> print(response) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE [('AV', 'AKV', '--', 'BHE', UTCDateTime(...), UTCDateTime(...)), ('AV', 'AKV', '--', 'BHN', UTCDateTime(...), UTCDateTime(...)), ('AV', 'AKV', '--', 'BHZ', UTCDateTime(...), UTCDateTime(...))] """ # build up possibly wildcarded trace id pattern for query if location == '': location = '--' pattern = ".".join((network, station, location, channel)) # get overview of all available data, winston wave servers can not # restrict the query via network, station etc. so we do that manually response = get_menu(self.host, self.port, timeout=self.timeout) # reorder items and convert time info to UTCDateTime response = [(x[3], x[1], x[4], x[2], UTCDateTime(x[5]), UTCDateTime(x[6])) for x in response] # restrict results according to user input response = [x for x in response if fnmatch(".".join(x[:4]), pattern)] return response
if __name__ == '__main__': import doctest doctest.testmod(exclude_empty=True)