# -*- coding: utf-8 -*-
"""
Provides the Event class
:copyright:
The ObsPy Development Team (devs@obspy.org)
:license:
GNU Lesser General Public License, Version 3
(https://www.gnu.org/copyleft/lesser.html)
"""
import copy
from itertools import chain
from obspy.core.event.header import (
EventType, EventTypeCertainty, EventDescriptionType)
from obspy.core.event.resourceid import ResourceIdentifier
from obspy.core.util.misc import _yield_resource_id_parent_attr
from obspy.imaging.source import plot_radiation_pattern, _setup_figure_and_axes
from .base import _event_type_class_factory, CreationInfo
__Event = _event_type_class_factory(
"__Event",
class_attributes=[("resource_id", ResourceIdentifier),
("event_type", EventType),
("event_type_certainty", EventTypeCertainty),
("creation_info", CreationInfo),
("preferred_origin_id", ResourceIdentifier),
("preferred_magnitude_id", ResourceIdentifier),
("preferred_focal_mechanism_id", ResourceIdentifier)],
class_contains=['event_descriptions', 'comments', 'picks', 'amplitudes',
'focal_mechanisms', 'origins', 'magnitudes',
'station_magnitudes'])
[docs]
class Event(__Event):
"""
The class Event describes a seismic event which does not necessarily need
to be a tectonic earthquake. An event is usually associated with one or
more origins, which contain information about focal time and geographical
location of the event. Multiple origins can cover automatic and manual
locations, a set of location from different agencies, locations generated
with different location programs and earth models, etc. Furthermore, an
event is usually associated with one or more magnitudes, and with one or
more focal mechanism determinations.
:type resource_id: :class:`~obspy.core.event.resourceid.ResourceIdentifier`
:param resource_id: Resource identifier of Event.
:type force_resource_id: bool, optional
:param force_resource_id: If set to False, the automatic initialization of
`resource_id` attribute in case it is not specified will be skipped.
:type event_type: str, optional
:param event_type: Describes the type of an event.
See :class:`~obspy.core.event.header.EventType` for allowed values.
:type event_type_certainty: str, optional
:param event_type_certainty: Denotes how certain the information on event
type is.
See :class:`~obspy.core.event.header.EventTypeCertainty` for allowed
values.
:type creation_info: :class:`~obspy.core.event.base.CreationInfo`, optional
:param creation_info: Creation information used to describe author,
version, and creation time.
:type event_descriptions: list of
:class:`~obspy.core.event.event.EventDescription`
:param event_descriptions: Additional event description, like earthquake
name, Flinn-Engdahl region, etc.
:type comments: list of :class:`~obspy.core.event.base.Comment`, optional
:param comments: Additional comments.
:type picks: list of :class:`~obspy.core.event.origin.Pick`
:param picks: Picks associated with the event.
:type amplitudes: list of :class:`~obspy.core.event.magnitude.Amplitude`
:param amplitudes: Amplitudes associated with the event.
:type focal_mechanisms: list of
:class:`~obspy.core.event.source.FocalMechanism`
:param focal_mechanisms: Focal mechanisms associated with the event
:type origins: list of :class:`~obspy.core.event.origin.Origin`
:param origins: Origins associated with the event.
:type magnitudes: list of :class:`~obspy.core.event.magnitude.Magnitude`
:param magnitudes: Magnitudes associated with the event.
:type station_magnitudes: list of
:class:`~obspy.core.event.magnitude.StationMagnitude`
:param station_magnitudes: Station magnitudes associated with the event.
.. note::
For handling additional information not covered by the QuakeML
standard and how to output it to QuakeML see the
:ref:`ObsPy Tutorial <quakeml-extra>`.
"""
do_not_warn_on = ["_format", "extra"]
[docs]
def __init__(self, *args, **kwargs):
super(Event, self).__init__(*args, **kwargs)
self.scope_resource_ids()
[docs]
def short_str(self):
"""
Returns a short string representation of the current Event.
Example:
Time | Lat | Long | Magnitude of the first origin, e.g.
2011-03-11T05:46:24.120000Z | +38.297, +142.373 | 9.1 MW
"""
out = ''
origin = None
if self.origins:
origin = self.preferred_origin() or self.origins[0]
# get lat, lon, time and handle if any are None (#2119)
lat, lon, time = origin.latitude, origin.longitude, origin.time
lat_str = '%+7.3f' % lat if lat is not None else 'None'
lon_str = '%+8.3f' % lon if lon is not None else 'None'
out += '%s | %s, %s' % (time, lat_str, lon_str)
if self.magnitudes:
magnitude = self.preferred_magnitude() or self.magnitudes[0]
try:
if round(magnitude.mag, 1) == magnitude.mag:
mag_string = '%3.1f ' % magnitude.mag
else:
mag_string = '%4.2f' % magnitude.mag
except TypeError:
mag_string = str(magnitude.mag)
out += ' | %s %-2s' % (mag_string,
magnitude.magnitude_type)
if origin and origin.evaluation_mode:
out += ' | %s' % (origin.evaluation_mode)
return out
[docs]
def __str__(self):
"""
Print a short summary at the top.
"""
return "Event:\t%s\n\n%s" % (
self.short_str(),
"\n".join(super(Event, self).__str__().split("\n")[1:]))
[docs]
def _repr_pretty_(self, p, cycle):
p.text(str(self))
[docs]
def __repr__(self):
return super(Event, self).__str__(force_one_line=True)
[docs]
def preferred_origin(self):
"""
Returns the preferred origin
"""
if self.preferred_origin_id is None:
return None
return self.preferred_origin_id.get_referred_object()
[docs]
def preferred_magnitude(self):
"""
Returns the preferred magnitude
"""
if self.preferred_magnitude_id is None:
return None
return self.preferred_magnitude_id.get_referred_object()
[docs]
def preferred_focal_mechanism(self):
"""
Returns the preferred focal mechanism
"""
if self.preferred_focal_mechanism_id is None:
return None
return self.preferred_focal_mechanism_id.get_referred_object()
[docs]
def plot(self, kind=[['ortho', 'beachball'], ['p_sphere', 's_sphere']],
subplot_size=4.0, show=True, outfile=None, **kwargs):
"""
Plot event location and/or the preferred focal mechanism
and radiation pattern.
:type kind: list[str] or list[list[str]]
:param kind: A list of strings (for a 1-row plot) or a nested list of
strings (one list of strings per row), with the following keywords
to generate a matplotlib figure:
* ``'ortho'`` (Orthographic plot of event location
see :meth:`~obspy.core.event.catalog.Catalog.plot`)
* ``'global'`` (Global plot of event location
see :meth:`~obspy.core.event.catalog.Catalog.plot`)
* ``'local'`` (Local plot of event location
see :meth:`~obspy.core.event.catalog.Catalog.plot`)
* ``'beachball'`` (Beachball of preferred focal mechanism)
* ``'p_quiver'`` (quiver plot of p wave farfield)
* ``'s_quiver'`` (quiver plot of s wave farfield)
* ``'p_sphere'`` (surface plot of p wave farfield)
* ``'s_sphere'`` (surface plot of s wave farfield)
:type subplot_size: float
:param subplot_size: Width/height of one single subplot cell in inches.
:type show: bool
:param show: Whether to show the figure after plotting or not. Can be
used to do further customization of the plot before
showing it. Has no effect if `outfile` is specified.
:type outfile: str
:param outfile: Output file path to directly save the resulting image
(e.g. ``"/tmp/image.png"``). Overrides the ``show`` option, image
will not be displayed interactively. The given path/filename is
also used to automatically determine the output format. Supported
file formats depend on your matplotlib backend. Most backends
support png, pdf, ps, eps and svg. Defaults to ``None``.
The figure is closed after saving it to file.
:returns: Figure instance with the plot.
.. rubric:: Examples
Default plot includes an orthographic map plot, a beachball plot and
plots of P/S farfield radiation patterns (preferred -- or first --
focal mechanism has to have a moment tensor set).
>>> from obspy import read_events
>>> event = read_events("/path/to/CMTSOLUTION")[0]
>>> event.plot() # doctest:+SKIP
.. plot::
from obspy import read_events
event = read_events("/path/to/CMTSOLUTION")[0]
event.plot()
Individual subplot parts and the setup of the grid of subplots
(rows/columns) can be specified by using certain keywords, see `kind`
parameter description.
>>> event.plot(kind=[['global'],
... ['p_sphere', 'p_quiver']]) # doctest:+SKIP
.. plot::
from obspy import read_events
event = read_events("/path/to/CMTSOLUTION")[0]
event.plot(kind=[['global'], ['p_sphere', 'p_quiver']])
"""
import matplotlib.pyplot as plt
from .catalog import Catalog
try:
fm = self.preferred_focal_mechanism() or self.focal_mechanisms[0]
mtensor = fm.moment_tensor.tensor
except (IndexError, AttributeError) as e:
msg = "Could not access event's moment tensor ({}).".format(str(e))
raise ValueError(msg)
mt = [mtensor.m_rr, mtensor.m_tt, mtensor.m_pp,
mtensor.m_rt, mtensor.m_rp, mtensor.m_tp]
if len(kind) == 1:
kind_ = kind
else:
kind_ = list(chain(*kind))
if any([k_ in ("ortho", "global", "local") for k_ in kind_]):
cat_ = Catalog([self])
kwargs["events"] = cat_
fig, axes, kind_ = _setup_figure_and_axes(kind,
subplot_size=subplot_size,
**kwargs)
for ax, kind__ in zip(axes, kind_):
if kind__ in ("ortho", "global", "local"):
ax.stock_img()
ax.gridlines()
ax.coastlines()
cat_.plot(projection=kind__, fig=ax, show=False,
**kwargs)
# shrink plot a bit to avoid it looking oversized compared to
# 3d axes that have some white space around them
# if kind__ == "ortho":
# scale = 0.8
# for getter, setter in zip((ax.get_xlim, ax.get_ylim),
# (ax.set_xlim, ax.set_ylim)):
# min_, max_ = getter()
# margin = (max_ - min_) * (1 - scale) / 2.0
# setter(min_ - margin, max_ + margin)
plot_radiation_pattern(
mt, kind=kind, coordinate_system='RTP', fig=fig, show=False)
# fig.tight_layout(pad=0.1)
if outfile:
fig.savefig(outfile)
plt.close(fig)
else:
if show:
plt.show()
return fig
[docs]
def __deepcopy__(self, memodict=None):
"""
reset resource_id's object_id after deep copy to allow the
object specific behavior of get_referred_object
"""
memodict = memodict or {}
cls = self.__class__
result = cls.__new__(cls)
memodict[id(self)] = result
for k, v in self.__dict__.items():
setattr(result, k, copy.deepcopy(v, memodict))
result.scope_resource_ids()
return result
[docs]
def __setstate__(self, state_dict):
self.__dict__.update(state_dict)
self.scope_resource_ids()
[docs]
def write(self, filename, format, **kwargs):
"""
Saves event information into a file.
:type filename: str
:param filename: The name of the file to write.
:type format: str
:param format: The file format to use (e.g. ``"QUAKEML"``). See
:meth:`obspy.core.event.catalog.Catalog.write()` for a list of
supported formats.
:param kwargs: Additional keyword arguments passed to the underlying
plugin's writer method.
.. rubric:: Example
>>> from obspy import read_events
>>> event = read_events()[0] # doctest: +SKIP
>>> event.write("example.xml", format="QUAKEML") # doctest: +SKIP
"""
from .catalog import Catalog
Catalog(events=[self]).write(filename, format, **kwargs)
[docs]
def scope_resource_ids(self):
"""
Ensure all resource_ids in event instance are event-scoped.
This will ensure the resource_ids refer to objects in the event
structure when possible.
"""
gen = _yield_resource_id_parent_attr(self)
for resource_id, parent, attr in gen:
if attr == 'resource_id':
resource_id.set_referred_object(parent, parent=self,
warn=False)
else:
resource_id._parent_key = self
resource_id._object_id = None
__EventDescription = _event_type_class_factory(
"__EventDescription",
class_attributes=[("text", str),
("type", EventDescriptionType)])
[docs]
class EventDescription(__EventDescription):
"""
Free-form string with additional event description. This can be a
well-known name, like 1906 San Francisco Earthquake. A number of categories
can be given in type.
:type text: str, optional
:param text: Free-form text with earthquake description.
:type type: str, optional
:param type: Category of earthquake description.
See :class:`~obspy.core.event.header.EventDescriptionType` for allowed
values.
.. note::
For handling additional information not covered by the QuakeML
standard and how to output it to QuakeML see the
:ref:`ObsPy Tutorial <quakeml-extra>`.
"""
if __name__ == '__main__':
import doctest
doctest.testmod(exclude_empty=True)