Source code for obspy.io.sac.sactrace

# -*- coding: utf-8 -*-
"""
Python interface to the Seismic Analysis Code (SAC) file format.

:copyright:
    The Los Alamos National Security, LLC, Yannik Behr, C. J. Ammon,
    C. Satriano, L. Krischer, and J. MacCarthy
:license:
    GNU Lesser General Public License, Version 3
    (https://www.gnu.org/copyleft/lesser.html)


The SACTrace object maintains consistency between SAC headers and manages
header values in a user-friendly way. This includes some value-checking, native
Python logicals (True, False) and nulls (None) instead of SAC's 0, 1, or
-12345...

SAC headers are implemented as properties, with appropriate getters and
setters.


Features
--------

1. **Read and write SAC binary or ASCII**

   -  autodetect or specify expected byteorder
   -  optional file size checking and/or header consistency checks
   -  header-only reading and writing
   -  "overwrite OK" checking ('lovrok' header)

2. **Convenient access and manipulation of relative and absolute time
   headers**
3. **User-friendly header printing/viewing**
4. **Fast access to header values from attributes**

   -  With type checking, null handling, and enumerated value checking

5. **Convert to/from ObsPy Traces**

   -  Conversion from ObsPy Trace to SAC trace retains detected previous
      SAC header values.
   -  Conversion to ObsPy Trace retains the *complete* SAC header.


Usage examples
--------------

Read/write SAC files
~~~~~~~~~~~~~~~~~~~~

.. code:: python

    # read from a binary file
    sac = SACTrace.read(filename)

    # read header only
    sac = SACTrace.read(filename, headonly=True)

    # write header-only, file must exist
    sac.write(filename, headonly=True)

    # read from an ASCII file
    sac = SACTrace.read(filename, ascii=True)

    # write a binary SAC file for a Sun machine
    sac.write(filename, byteorder='big')

Reference-time and relative time headers
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. code:: python

    sac = SACTrace(nzyear=2000, nzjday=1, nzhour=0, nzmin=0, nzsec=0, nzmsec=0,
                   t1=23.5, data=numpy.arange(100))

    sac.reftime
    sac.b, sac.e, sac.t1

::

    2000-01-01T00:00:00.000000Z
    (0.0, 99.0, 23.5)

Move reference time by relative seconds, relative time headers are
preserved.

.. code:: python

    sac.reftime -= 2.5
    sac.b, sac.e, sac.t1

::

    (2.5, 101.5, 26.0)

Set reference time to new absolute time, relative time headers are
preserved.

.. code:: python

    sac.reftime = UTCDateTime(2000, 1, 1, 0, 2, 0, 0)
    sac.b, sac.e

::

    (-120.0, -21.0, -96.5)

Quick header viewing
~~~~~~~~~~~~~~~~~~~~

Print non-null header values.

.. code:: python

    sac = SACTrace()
    print sac

::

    Reference Time = 01/01/2000 (001) 00:00:00.000000
        iztype IB: begin time
    b          = 0.0
    cmpaz      = 0.0
    cmpinc     = 0.0
    delta      = 1.0
    e          = 99.0
    iftype     = itime
    internal0  = 2.0
    iztype     = ib
    kcmpnm     = Z
    lcalda     = False
    leven      = True
    lovrok     = True
    lpspol     = True
    npts       = 100
    nvhdr      = 6
    nzhour     = 0
    nzjday     = 1
    nzmin      = 0
    nzmsec     = 0
    nzsec      = 0
    nzyear     = 2000

Print relative time header values.

.. code:: python

    sac.lh('picks')

::

    Reference Time = 01/01/1970 (001) 00:00:00.000000
        iztype IB: begin time
        a          = None
        b          = 0.0
        e          = 0.0
        f          = None
        o          = None
        t0         = None
        t1         = None
        t2         = None
        t3         = None
        t4         = None
        t5         = None
        t6         = None
        t7         = None
        t8         = None
        t9         = None

Header values as attributes
~~~~~~~~~~~~~~~~~~~~~~~~~~~

Great for interactive use, with (ipython) tab-completion...

.. code:: python

    sac.<tab>

::

    sac.a                 sac.kevnm             sac.nzsec
    sac.az                sac.kf                sac.nzyear
    sac.b                 sac.khole             sac.o
    sac.baz               sac.kinst             sac.odelta
    sac.byteorder         sac.knetwk            sac.read
    sac.cmpaz             sac.ko                sac.reftime
    sac.cmpinc            sac.kstnm             sac.scale
    sac.copy              sac.kt0               sac.stdp
    sac.data              sac.kt1               sac.stel
    sac.delta             sac.kt2               sac.stla
    sac.depmax            sac.kt3               sac.stlo
    sac.depmen            sac.kt4               sac.t0
    sac.depmin            sac.kt5               sac.t1
    sac.dist              sac.kt6               sac.t2
    sac.e                 sac.kt7               sac.t3
    sac.evdp              sac.kt8               sac.t4
    sac.evla              sac.kt9               sac.t5
    sac.evlo              sac.kuser0            sac.t6
    sac.f                 sac.kuser1            sac.t7
    sac.from_obspy_trace  sac.kuser2            sac.t8
    sac.gcarc             sac.lcalda            sac.t9
    sac.idep              sac.leven             sac.to_obspy_trace
    sac.ievreg            sac.lh                sac.unused23
    sac.ievtyp            sac.listhdr           sac.user0
    sac.iftype            sac.lovrok            sac.user1
    sac.iinst             sac.lpspol            sac.user2
    sac.imagsrc           sac.mag               sac.user3
    sac.imagtyp           sac.nevid             sac.user4
    sac.internal0         sac.norid             sac.user5
    sac.iqual             sac.npts              sac.user6
    sac.istreg            sac.nvhdr             sac.user7
    sac.isynth            sac.nwfid             sac.user8
    sac.iztype            sac.nzhour            sac.user9
    sac.ka                sac.nzjday            sac.validate
    sac.kcmpnm            sac.nzmin             sac.write
    sac.kdatrd            sac.nzmsec

...and documentation!

.. code:: python

    sac.iztype?

::

    Type:        property
    String form: <property object at 0x106404940>
    Docstring:
    I    Reference time equivalence:
    * IUNKN (5): Unknown
    * IB (9): Begin time
    * IDAY (10): Midnight of reference GMT day
    * IO (11): Event origin time
    * IA (12): First arrival time
    * ITn (13-22): User defined time pick n, n=0,9

Convert to/from ObsPy Traces
~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. code:: python

    from obspy import read
    tr = read()[0]
    print tr.stats

::

             network: BW
             station: RJOB
            location:
             channel: EHZ
           starttime: 2009-08-24T00:20:03.000000Z
             endtime: 2009-08-24T00:20:32.990000Z
       sampling_rate: 100.0
               delta: 0.01
                npts: 3000
               calib: 1.0
        back_azimuth: 100.0
         inclination: 30.0

.. code:: python

    sac = SACTrace.from_obspy_trace(tr)
    print sac

::

    Reference Time = 08/24/2009 (236) 00:20:03.000000
        iztype IB: begin time
    b          = 0.0
    cmpaz      = 0.0
    cmpinc     = 0.0
    delta      = 0.00999999977648
    depmax     = 1293.77099609
    depmen     = -4.49556303024
    depmin     = -1515.81311035
    e          = 29.9899993297
    iftype     = itime
    internal0  = 2.0
    iztype     = ib
    kcmpnm     = EHZ
    knetwk     = BW
    kstnm      = RJOB
    lcalda     = False
    leven      = True
    lovrok     = True
    lpspol     = True
    npts       = 3000
    nvhdr      = 6
    nzhour     = 0
    nzjday     = 236
    nzmin      = 20
    nzmsec     = 0
    nzsec      = 3
    nzyear     = 2009
    scale      = 1.0

.. code:: python

    tr2 = sac.to_obspy_trace()
    print tr2.stats

::

             network: BW
             station: RJOB
            location:
             channel: EHZ
           starttime: 2009-08-24T00:20:03.000000Z
             endtime: 2009-08-24T00:20:32.990000Z
       sampling_rate: 100.0
               delta: 0.01
                npts: 3000
               calib: 1.0
                 sac: AttribDict({'cmpaz': 0.0, 'nzyear': 2009, 'nzjday': 236,
                 'iztype': 9, 'evla': 0.0, 'nzhour': 0, 'lcalda': 0,
                 'evlo': 0.0, 'scale': 1.0, 'nvhdr': 6, 'depmin': -1515.8131,
                 'kcmpnm': 'EHZ', 'nzsec': 3, 'internal0': 2.0,
                 'depmen': -4.495563, 'cmpinc': 0.0,
                 'depmax': 1293.771, 'iftype': 1, 'delta': 0.0099999998,
                 'nzmsec': 0, 'lpspol': 1, 'b': 0.0, 'e': 29.99, 'leven': 1,
                 'kstnm': 'RJOB', 'nzmin': 20, 'lovrok': 1, 'npts': 3000,
                 'knetwk': 'BW'})


"""
from __future__ import (absolute_import, division, print_function,
                        unicode_literals)
from future.builtins import *  # NOQA

import sys
import warnings
from copy import deepcopy
from itertools import chain

import numpy as np
from obspy import Trace, UTCDateTime
from obspy.geodetics import gps2dist_azimuth, kilometer2degrees

from . import header as HD  # noqa
from .util import SacError, SacHeaderError
from . import util as _ut
from . import arrayio as _io


# ------------- HEADER GETTER/SETTERS -----------------------------------------
#
# These are functions used to set up header properties on the SACTrace class.
# Properties are accessed like class attributes, but use getter and/or setter
# functions.  They're defined here, outside the class, b/c there are lots of
# properties and we want to keep the class clean.
#
# getter/setter factories:
# Property getters/setters must be defined for _each_ header, even if they work
# the exact same way for similar headers, so we define function factories here
# for groups of headers that will be gotten and set in a similar fashion.
#
# Usage: delta_getter = _floatgetter('delta')
#        delta_setter = _floatsetter('delta')
# These factories produce functions that simply index into their array to get
# or set.  Use them for header values in hf, hi, and hs that need no special
# handling. Values that depend on other values will need their own
# getters/setters.
#
# See:
# https://stackoverflow.com/q/2123585
#
# floats
[docs]def _floatgetter(hdr): def get_float(self): value = self._hf[HD.FLOATHDRS.index(hdr)] if value == HD.FNULL: value = None return value return get_float
[docs]def _floatsetter(hdr): def set_float(self, value): if value is None: value = HD.FNULL self._hf[HD.FLOATHDRS.index(hdr)] = value return set_float # ints
[docs]def _intgetter(hdr): def get_int(self): value = self._hi[HD.INTHDRS.index(hdr)] if value == HD.INULL: value = None return value return get_int
[docs]def _intsetter(hdr): def set_int(self, value): if value % 1: warnings.warn("Non-integers may be truncated. ({}: {})".format( hdr, value)) if value is None: value = HD.INULL self._hi[HD.INTHDRS.index(hdr)] = value return set_int # logicals/bools (subtype of ints)
[docs]def _boolgetter(hdr): def get_bool(self): value = self._hi[HD.INTHDRS.index(hdr)] return bool(value) return get_bool
[docs]def _boolsetter(hdr): def set_bool(self, value): if value not in (True, False, 1, 0): msg = "Logical header values must be {True, False, 1, 0}" raise ValueError(msg) # booleans are subclasses of integers. They will be set (cast) # directly into an integer array as 0 or 1. self._hi[HD.INTHDRS.index(hdr)] = value return set_bool # enumerated values (stored as ints, represented by strings)
[docs]def _enumgetter(hdr): def get_enum(self): value = self._hi[HD.INTHDRS.index(hdr)] if value == HD.INULL: name = None elif _ut.is_valid_enum_int(hdr, value): name = HD.ENUM_NAMES[value] else: msg = """Unrecognized enumerated value {} for header "{}". See .header for allowed values.""".format(value, hdr) warnings.warn(msg) name = None return name return get_enum
[docs]def _enumsetter(hdr): def set_enum(self, value): if value is None: value = HD.INULL elif _ut.is_valid_enum_str(hdr, value): value = HD.ENUM_VALS[value] else: msg = 'Unrecognized enumerated value "{}" for header "{}"' raise ValueError(msg.format(value, hdr)) self._hi[HD.INTHDRS.index(hdr)] = value return set_enum # strings
[docs]def _strgetter(hdr): def get_str(self): try: # value is a bytes value = self._hs[HD.STRHDRS.index(hdr)].decode() except AttributeError: # value is a str value = self._hs[HD.STRHDRS.index(hdr)] if value == HD.SNULL: value = None try: value = value.strip() except AttributeError: # it's None. no .strip method pass return value return get_str
[docs]def _strsetter(hdr): def set_str(self, value): if value is None: value = HD.SNULL elif len(value) > 8: msg = "Alphanumeric headers longer than 8 characters are "\ "right-truncated." warnings.warn(msg) # they will truncate themselves, since _hs is dtype '|S8' try: self._hs[HD.STRHDRS.index(hdr)] = value.encode('ascii', 'strict') except AttributeError: self._hs[HD.STRHDRS.index(hdr)] = value return set_str # Factory for functions of .data (min, max, mean, len)
[docs]def _make_data_func(func, hdr): # returns a method that returns the value of func(self.data), or the # corresponding array header value, if data is None def do_data_func(self): try: value = func(self.data) except TypeError: # data=None (headonly=True) try: value = self._hf[HD.FLOATHDRS.index(hdr)] null = HD.INULL except ValueError: # hdr is 'npts', the only integer # Will this also trip if a data-centric header is misspelled? value = self._hi[HD.INTHDRS.index(hdr)] null = HD.INULL if value == null: value = None return value return do_data_func # TODO: a data setter the requires a float32 array # Factory function for setting relative time headers with either a relative # time float or an absolute UTCDateTime # used for: b, o, a, f, t0-t9
[docs]def _reltime_setter(hdr): def set_reltime(self, value): if isinstance(value, UTCDateTime): # get the offset from reftime offset = value - self.reftime else: # value is already a reftime offset. offset = value # make and use a _floatsetter to actually set it floatsetter = _floatsetter(hdr) floatsetter(self, offset) return set_reltime # Factory function for setting geographic header values # (evlo, evla, stalo, stalat) # that will check lcalda and calculate and set dist, az, baz, gcarc
[docs]def _geosetter(hdr): def set_geo(self, value): # make and use a _floatsetter set_geo_float = _floatsetter(hdr) set_geo_float(self, value) if self.lcalda: # check and maybe set lcalda try: self._set_distances() except SacHeaderError: pass return set_geo # OTHER GETTERS/SETTERS
[docs]def _set_lcalda(self, value): # make and use a bool setter for lcalda lcalda_setter = _boolsetter('lcalda') lcalda_setter(self, value) # try to set set distances if value evaluates to True if value: try: self._set_distances() except SacHeaderError: pass
[docs]def _get_e(self): try: if self.npts: e = self.b + (self.npts - 1) * self.delta else: e = self.b except TypeError: # b, npts, and/or delta are None/null # TODO: assume "b" is 0.0? e = None return e
[docs]def _set_iztype(self, iztype): """ Set the iztype, which describes what the reftime is. Setting the iztype will shift the relative time headers, such that the header that iztype points to is (near) zero, and all others are shifted together by the difference. Affected headers: b, o, a, f, t0-t9 :param iztype: One of the following strings: 'iunkn' 'ib', begin time 'iday', midnight of reference GMT day 'io', event origin time 'ia', first arrival time 'it0'-'it9', user defined pick t0-t9. :type iztype: str """ # The Plan: # 1. find the seconds needed to shift the old reftime to the new one. # 2. shift reference time onto the iztype header using that shift value. # 3. this triggers an _allt shift of all relative times by the same amount. # 4. If all goes well, actually set the iztype in the header. # 1. if iztype == 'iunkn': # no shift ref_val = 0.0 elif iztype == 'iday': # seconds since midnight of reference day reftime = self.reftime ref_val = reftime - UTCDateTime(year=reftime.year, julday=reftime.julday) else: # a relative time header. # remove the 'i' (first character) in the iztype to get the header name ref_val = getattr(self, iztype[1:]) if ref_val is None: msg = "Reference header for iztype '{}' is not set".format(iztype) raise SacError(msg) # 2. set a new reference time, # 3. which also shifts all non-null relative times (self._allt). # remainder microseconds may be in the reference header value, because # nzmsec can't hold them. self.reftime = self.reftime + ref_val # 4. no exceptions yet. actually set the iztype # make an _enumsetter for iztype and use it, for its enum checking. izsetter = _enumsetter('iztype') izsetter(self, iztype) # kevnm is 16 characters, split into two 8-character fields # intercept and handle in while getting and setting
[docs]def _get_kevnm(self): kevnm = self._hs[HD.STRHDRS.index('kevnm')] kevnm2 = self._hs[HD.STRHDRS.index('kevnm2')] try: kevnm = kevnm.decode() kevnm2 = kevnm2.decode() except AttributeError: # kevnm is a str pass if kevnm == HD.SNULL: kevnm = '' if kevnm2 == HD.SNULL: kevnm2 = '' value = (kevnm + kevnm2).strip() if not value: value = None return value
[docs]def _set_kevnm(self, value): if value is None: value = HD.SNULL + HD.SNULL elif len(value) > 16: msg = "kevnm over 16 characters. Truncated to {}.".format(value[:16]) warnings.warn(msg) kevnm = '{:<8s}'.format(value[0:8]) kevnm2 = '{:<8s}'.format(value[8:16]) self._hs[HD.STRHDRS.index('kevnm')] = kevnm self._hs[HD.STRHDRS.index('kevnm2')] = kevnm2 # TODO: move get/set reftime up here, make it a property # -------------------------- SAC OBJECT INTERFACE -----------------------------
[docs]class SACTrace(object): __doc__ = """ Convenient and consistent in-memory representation of Seismic Analysis Code (SAC) files. This is the human-facing interface for making a valid instance. For file-based or other constructors, see class methods .read and .from_obspy_trace. SACTrace instances preserve relationships between header values. :param data: Associated time-series data vector. Optional. If omitted, None is set as the instance data attribute. :type data: :class:`numpy.ndarray` of float32 Any valid header key/value pair is also an optional input keyword argument. If not provided, minimum required headers are set to valid default values. The default instance is an evenly-space trace, with a sample rate of 1.0, and len(data) or 0 npts, starting at 1970-01-01T00:00:00.000000. :var reftime: Read-only reference time. Calculated from nzyear, nzjday, nzhour, nzmin, nzsec, nzmsec. :var byteorder: The byte order of the underlying header/data arrays. Raises :class:`SacError` if array byte orders are inconsistent, even in the case where '<' is your native order and byteorders look like '<', '=', '='. Any valid header name is also an attribute. See below, :mod:`header`, or individial attribution docstrings for more header information. THE SAC HEADER NOTE: All header names and string values are lowercase. Header value access should be through instance attributes. """ + HD.HEADER_DOCSTRING
[docs] def __init__(self, leven=True, delta=1.0, b=0.0, e=0.0, iztype='ib', nvhdr=6, npts=0, iftype='itime', nzyear=1970, nzjday=1, nzhour=0, nzmin=0, nzsec=0, nzmsec=0, lcalda=False, lpspol=True, lovrok=True, internal0=2.0, data=None, **kwargs): """ Initialize a SACTrace object using header key-value pairs and a numpy.ndarray for the data, both optional. ..rubric:: Example >>> sac = SACTrace(nzyear=1995, nzmsec=50, data=np.arange(100)) >>> print(sac) # doctest: +SKIP Reference Time = 01/01/1995 (001) 00:00:00.050000 iztype IB: begin time b = 0.0 cmpaz = 0.0 cmpinc = 0.0 delta = 1.0 e = 99.0 iftype = itime internal0 = 2.0 iztype = ib kcmpnm = Z lcalda = False leven = True lovrok = True lpspol = True npts = 100 nvhdr = 6 nzhour = 0 nzjday = 1 nzmin = 0 nzmsec = 50 nzsec = 0 nzyear = 1995 """ # The Plan: # 1. Build the default header dictionary and update with provided # values. # 2. Convert header dict to arrays (util.dict_to_header_arrays # initializes the arrays and fills in without checking. # 3. set the _h[fis] and data arrays on self. # 1. # build the required header from provided or default values header = {'leven': leven, 'npts': npts, 'delta': delta, 'b': b, 'e': e, 'iztype': iztype, 'nvhdr': nvhdr, 'iftype': iftype, 'nzyear': nzyear, 'nzjday': nzjday, 'nzhour': nzhour, 'nzmin': nzmin, 'nzsec': nzsec, 'nzmsec': nzmsec, 'lcalda': lcalda, 'lpspol': lpspol, 'lovrok': lovrok, 'internal0': internal0} # required = ['delta', 'b', 'npts', ...] # provided = locals() # for hdr in required: # header[hdr] = kwargs.pop(hdr, provided[hdr]) # combine header with remaining non-required args. # XXX: user can put non-SAC key:value pairs into the header. header.update(kwargs) # -------------------------- DATA ARRAY ------------------------------- if data is None: # this is like "headonly=True" pass else: if not isinstance(data, np.ndarray): raise TypeError("data needs to be a numpy.ndarray") else: # Only copy the data if they are not of the required type # XXX: why require little endian instead of native byte order? # data = np.require(data, native_str('<f4')) pass # --------------------------- HEADER ARRAYS --------------------------- # 2. # TODO: this is done even when we're reading a file. # if it's too much overhead, it may need to change # swap enum names for integer values in the header dictionary header = _ut.enum_string_to_int(header) # XXX: will these always be little endian? hf, hi, hs = _io.dict_to_header_arrays(header) # we now have data and headers, either default or provided. # 3. # this completely sidesteps any checks provided by class properties self._hf = hf self._hi = hi self._hs = hs self.data = data # ---------------------------- SET UP HEADERS ----------------------------- # SAC header values are set up as attributes, with getters and setters # format: header = property(getter_function, setter_function) # # Attributes are outside of __init__ b/c properties are just instructions # about how to retrieve attributes (getters/setters), and therefore shared # across all instances, not attribute data themselves. # # This section looks ugly, but it allows for the following: # 1. Relationships/checks between header variables can be done in setters # 2. The underlying header array structure is retained, for quick writing. # 3. Header access looks like simple attribute-access syntax. # Looks funny to read here, but natural to use. # NOTE: To make something read-only, omit second argument to "property". # FLOATS
delta = property(_floatgetter('delta'), _floatsetter('delta'), doc=HD.DOC['delta']) depmin = property(_make_data_func(min, 'depmin'), doc=HD.DOC['depmin']) depmax = property(_make_data_func(max, 'depmax'), doc=HD.DOC['depmax']) scale = property(_floatgetter('scale'), _floatsetter('scale'), doc=HD.DOC['scale']) odelta = property(_floatgetter('odelta'), _floatsetter('odelta'), doc=HD.DOC['odelta']) b = property(_floatgetter('b'), _reltime_setter('b'), doc=HD.DOC['b']) e = property(_get_e, doc=HD.DOC['e']) o = property(_floatgetter('o'), _reltime_setter('o'), doc=HD.DOC['o']) a = property(_floatgetter('a'), _reltime_setter('a'), doc=HD.DOC['a']) internal0 = property(_floatgetter('internal0'), _floatsetter('internal0')) t0 = property(_floatgetter('t0'), _reltime_setter('t0'), doc=HD.DOC['t0']) t1 = property(_floatgetter('t1'), _reltime_setter('t1'), doc=HD.DOC['t1']) t2 = property(_floatgetter('t2'), _reltime_setter('t2'), doc=HD.DOC['t2']) t3 = property(_floatgetter('t3'), _reltime_setter('t3'), doc=HD.DOC['t3']) t4 = property(_floatgetter('t4'), _reltime_setter('t4'), doc=HD.DOC['t4']) t5 = property(_floatgetter('t5'), _reltime_setter('t5'), doc=HD.DOC['t5']) t6 = property(_floatgetter('t6'), _reltime_setter('t6'), doc=HD.DOC['t6']) t7 = property(_floatgetter('t7'), _reltime_setter('t7'), doc=HD.DOC['t7']) t8 = property(_floatgetter('t8'), _reltime_setter('t8'), doc=HD.DOC['t8']) t9 = property(_floatgetter('t9'), _reltime_setter('t9'), doc=HD.DOC['t9']) f = property(_floatgetter('f'), _reltime_setter('f'), doc=HD.DOC['f']) stla = property(_floatgetter('stla'), _geosetter('stla'), doc=HD.DOC['stla']) stlo = property(_floatgetter('stlo'), _geosetter('stlo'), doc=HD.DOC['stlo']) stel = property(_floatgetter('stel'), _floatsetter('stel'), doc=HD.DOC['stel']) stdp = property(_floatgetter('stdp'), _floatsetter('stdp'), doc=HD.DOC['stdp']) evla = property(_floatgetter('evla'), _geosetter('evla'), doc=HD.DOC['evla']) evlo = property(_floatgetter('evlo'), _geosetter('evlo'), doc=HD.DOC['evlo']) evdp = property(_floatgetter('evdp'), _floatsetter('evsp'), doc=HD.DOC['evdp']) mag = property(_floatgetter('mag'), _floatsetter('mag'), doc=HD.DOC['mag']) user0 = property(_floatgetter('user0'), _floatsetter('user0'), doc=HD.DOC['user0']) user1 = property(_floatgetter('user1'), _floatsetter('user1'), doc=HD.DOC['user1']) user2 = property(_floatgetter('user2'), _floatsetter('user2'), doc=HD.DOC['user2']) user3 = property(_floatgetter('user3'), _floatsetter('user3'), doc=HD.DOC['user3']) user4 = property(_floatgetter('user4'), _floatsetter('user4'), doc=HD.DOC['user4']) user5 = property(_floatgetter('user5'), _floatsetter('user5'), doc=HD.DOC['user5']) user6 = property(_floatgetter('user6'), _floatsetter('user6'), doc=HD.DOC['user6']) user7 = property(_floatgetter('user7'), _floatsetter('user7'), doc=HD.DOC['user7']) user8 = property(_floatgetter('user8'), _floatsetter('user8'), doc=HD.DOC['user8']) user9 = property(_floatgetter('user9'), _floatsetter('user9'), doc=HD.DOC['user9']) dist = property(_floatgetter('dist'), _floatsetter('dist'), doc=HD.DOC['dist']) az = property(_floatgetter('az'), _floatsetter('az'), doc=HD.DOC['az']) baz = property(_floatgetter('baz'), _floatsetter('baz'), doc=HD.DOC['baz']) gcarc = property(_floatgetter('gcarc'), _floatsetter('gcarc'), doc=HD.DOC['gcarc']) depmen = property(_make_data_func(np.mean, 'depmen'), doc=HD.DOC['depmen']) cmpaz = property(_floatgetter('cmpaz'), _floatsetter('cmpaz'), doc=HD.DOC['cmpaz']) cmpinc = property(_floatgetter('cmpinc'), _floatsetter('cmpinc'), doc=HD.DOC['cmpinc']) # # INTS nzyear = property(_intgetter('nzyear'), _intsetter('nzyear'), doc=HD.DOC['nzyear']) nzjday = property(_intgetter('nzjday'), _intsetter('nzjday'), doc=HD.DOC['nzjday']) nzhour = property(_intgetter('nzhour'), _intsetter('nzhour'), doc=HD.DOC['nzhour']) nzmin = property(_intgetter('nzmin'), _intsetter('nzmin'), doc=HD.DOC['nzmin']) nzsec = property(_intgetter('nzsec'), _intsetter('nzsec'), doc=HD.DOC['nzsec']) nzmsec = property(_intgetter('nzmsec'), _intsetter('nzmsec'), doc=HD.DOC['nzmsec']) nvhdr = property(_intgetter('nvhdr'), _intsetter('nvhdr'), doc=HD.DOC['nvhdr']) norid = property(_intgetter('norid'), _intsetter('norid'), doc=HD.DOC['norid']) nevid = property(_intgetter('nevid'), _intsetter('nevid'), doc=HD.DOC['nevid']) npts = property(_make_data_func(len, 'npts'), doc=HD.DOC['npts']) nwfid = property(_intgetter('nwfid'), _intsetter('nwfid'), doc=HD.DOC['nwfid']) iftype = property(_enumgetter('iftype'), _enumsetter('iftype'), doc=HD.DOC['iftype']) idep = property(_enumgetter('idep'), _enumsetter('idep'), doc=HD.DOC['idep']) iztype = property(_enumgetter('iztype'), _set_iztype, doc=HD.DOC['iztype']) iinst = property(_intgetter('iinst'), _intsetter('iinst'), doc=HD.DOC['iinst']) istreg = property(_intgetter('istreg'), _intsetter('istreg'), doc=HD.DOC['istreg']) ievreg = property(_intgetter('ievreg'), _intsetter('ievreg'), doc=HD.DOC['ievreg']) ievtyp = property(_enumgetter('ievtyp'), _enumsetter('ievtyp'), doc=HD.DOC['ievtyp']) iqual = property(_enumgetter('iqual'), _enumsetter('iqual'), doc=HD.DOC['iqual']) isynth = property(_enumgetter('isythn'), _enumsetter('isynth'), doc=HD.DOC['isynth']) imagtyp = property(_enumgetter('imagtyp'), _enumsetter('imagtyp'), doc=HD.DOC['imagtyp']) imagsrc = property(_enumgetter('imagsrc'), _enumsetter('imagsrc'), doc=HD.DOC['imagsrc']) leven = property(_boolgetter('leven'), _boolsetter('leven'), doc=HD.DOC['leven']) lpspol = property(_boolgetter('lpspol'), _boolsetter('lpspol'), doc=HD.DOC['lpspol']) lovrok = property(_boolgetter('lovrok'), _boolsetter('lovrok'), doc=HD.DOC['lovrok']) lcalda = property(_boolgetter('lcalda'), _set_lcalda, doc=HD.DOC['lcalda']) unused23 = property(_intgetter('unused23'), _intsetter('unused23')) # # STRINGS kstnm = property(_strgetter('kstnm'), _strsetter('kstnm'), doc=HD.DOC['kstnm']) kevnm = property(_get_kevnm, _set_kevnm, doc=HD.DOC['kevnm']) khole = property(_strgetter('khole'), _strsetter('khole'), doc=HD.DOC['khole']) ko = property(_strgetter('ko'), _strsetter('ko'), doc=HD.DOC['ko']) ka = property(_strgetter('ka'), _strsetter('ka'), doc=HD.DOC['ka']) kt0 = property(_strgetter('kt0'), _strsetter('kt0'), doc=HD.DOC['kt0']) kt1 = property(_strgetter('kt1'), _strsetter('kt1'), doc=HD.DOC['kt1']) kt2 = property(_strgetter('kt2'), _strsetter('kt2'), doc=HD.DOC['kt2']) kt3 = property(_strgetter('kt3'), _strsetter('kt3'), doc=HD.DOC['kt3']) kt4 = property(_strgetter('kt4'), _strsetter('kt4'), doc=HD.DOC['kt4']) kt5 = property(_strgetter('kt5'), _strsetter('kt5'), doc=HD.DOC['kt5']) kt6 = property(_strgetter('kt6'), _strsetter('kt6'), doc=HD.DOC['kt6']) kt7 = property(_strgetter('kt7'), _strsetter('kt7'), doc=HD.DOC['kt7']) kt8 = property(_strgetter('kt8'), _strsetter('kt8'), doc=HD.DOC['kt8']) kt9 = property(_strgetter('kt9'), _strsetter('kt9'), doc=HD.DOC['kt9']) kf = property(_strgetter('kf'), _strsetter('kf'), doc=HD.DOC['kf']) kuser0 = property(_strgetter('kuser0'), _strsetter('kuser0'), doc=HD.DOC['kuser0']) kuser1 = property(_strgetter('kuser1'), _strsetter('kuser1'), doc=HD.DOC['kuser1']) kuser2 = property(_strgetter('kuser2'), _strsetter('kuser2'), doc=HD.DOC['kuser2']) kcmpnm = property(_strgetter('kcmpnm'), _strsetter('kcmpnm'), doc=HD.DOC['kcmpnm']) knetwk = property(_strgetter('knetwk'), _strsetter('knetwk'), doc=HD.DOC['knetwk']) kdatrd = property(_strgetter('kdatrd'), _strsetter('kdatrd')) kinst = property(_strgetter('kinst'), _strsetter('kinst'), doc=HD.DOC['kinst']) @property def _header(self): """ Convenient read-only dictionary of non-null header array values. Header value access should be through instance attributes. All header names and string values are lowercase. Computed every time, so use frugally. See class docstring for header descriptions. """ out = _io.header_arrays_to_dict(self._hf, self._hi, self._hs, nulls=False) return out @property def byteorder(self): """ The byte order of the underlying header/data arrays. Raises SacError if array byte orders are inconsistent, even in the case where '<' is your native order and byteorders look like '<', '=', '='. """ try: if self.data is None: assert self._hf.dtype.byteorder == self._hi.dtype.byteorder else: assert self._hf.dtype.byteorder == self._hi.dtype.byteorder ==\ self.data.dtype.byteorder except AssertionError: msg = 'Inconsistent header/data byteorders.' raise SacError(msg) bo = self._hf.dtype.byteorder if bo == '=': byteorder = sys.byteorder elif bo == '<': byteorder = 'little' elif bo == '>': byteorder = 'big' return byteorder # TODO: make a byteorder setter?
[docs] def _byteswap(self): """ Change the underlying byte order and dtype interpretation of the float, int, and (if present) data arrays. """ try: self._hf = self._hf.byteswap(True).newbyteorder('S') self._hi = self._hi.byteswap(True).newbyteorder('S') if self.data is not None: self.data = self.data.byteswap(True).newbyteorder('S') except Exception as e: # if this fails, roll it back? raise e
@property def reftime(self): """ Get or set the SAC header reference time as a UTCDateTime instance. reftime is not an attribute, but is constructed and dismantled each time directly to/from the SAC "nz"-time fields. Setting a new reftime shifts all non-null relative time headers accordingly. It accepts a UTCDateTime object, from which time shifts are calculated. ..rubric:: notes The reftime you supply will be robbed of its remainder microseconds, which are then pushed into the relative time header shifts. This means that the reftime you observe after you set it here may not exactly match the reftime you supplied; it may be `remainder microseconds` earlier. Nor will the iztype reference header value be exactly zero; it will be equal to `remainder microseconds` (as seconds). """ return _ut.get_sac_reftime(self._header) @reftime.setter def reftime(self, new_reftime): try: old_reftime = self.reftime # find the milliseconds and leftover microseconds for new reftime _, rem_microsecs = _ut.split_microseconds(new_reftime.microsecond) # snap the new reftime to the most recent milliseconds # (subtract the leftover microseconds) new_reftime.microsecond -= rem_microsecs self.nzyear = new_reftime.year self.nzjday = new_reftime.julday self.nzhour = new_reftime.hour self.nzmin = new_reftime.minute self.nzsec = new_reftime.second self.nzmsec = new_reftime.microsecond / 1000 # get the float seconds between the old and new reftimes shift = old_reftime - new_reftime # shift the relative time headers self._allt(np.float32(shift)) except AttributeError: msg = "New reference time must be an obspy.UTCDateTime instance." raise TypeError(msg) # --------------------------- I/O METHODS --------------------------------- @classmethod
[docs] def read(cls, source, headonly=False, ascii=False, byteorder=None, checksize=False): """ Construct an instance from a binary or ASCII file on disk. :param source: Full path string for File-like object from a SAC binary file on disk. If it is an open File object, open 'rb'. :type source: str or file :param headonly: If headonly is True, only read the header arrays not the data array. :type headonly: bool :param ascii: If True, file is a SAC ASCII/Alphanumeric file. :type ascii: bool :param byteorder: If omitted or None, automatic byte-order checking is done, starting with native order. If byteorder is specified and incorrect, a :class:`SacIOError` is raised. Only valid for binary files. :type byteorder: str {'little', 'big'}, optional :param checksize: If True, check that the theoretical file size from the header matches the size on disk. Only valid for binary files. :type checksize: bool :raises: :class:`SacIOError` if checksize failed, byteorder was wrong, or header arrays are wrong size. .. rubric:: Example >>> sac = SACTrace.read(filename, headonly=True) # doctest: +SKIP >>> try: # doctest: +SKIP sac.validate('data_hdrs') # doctest: +SKIP except SacInvalidContentError: # doctest: +SKIP sac = SACTrace.read(filename, headonly=False) # doctest: +SKIP sac.validate('data_hdrs') # doctest: +SKIP See also: :meth:`SACTrace.validate` """ if ascii: hf, hi, hs, data = _io.read_sac_ascii(source, headonly=headonly) else: hf, hi, hs, data = _io.read_sac(source, headonly=headonly, byteorder=byteorder, checksize=checksize) return cls._from_arrays(hf, hi, hs, data)
[docs] def write(self, dest, headonly=False, ascii=False, byteorder=None, flush_headers=True): """ Write the header and (optionally) data arrays to a SAC binary file. :param dest: Full path or File-like object to SAC binary file on disk. :type dest: str or file :param headonly: If headonly is True, only read the header arrays not the data array. :type headonly: bool :param ascii: If True, file is a SAC ASCII/Alphanumeric file. :type ascii: bool :param byteorder: If omitted or None, automatic byte-order checking is done, starting with native order. If byteorder is specified and incorrect, a :class:`SacIOError` is raised. Only valid for binary files. :type byteorder: str {'little', 'big'}, optional :param flush_headers: If True, update data headers like 'depmin' and 'depmax' with values from the data array. :type flush_headers: bool """ if headonly: data = None else: # do a check for float32 data here instead of arrayio.write_sac? data = self.data if flush_headers: self._flush_headers() if ascii: _io.write_sac_ascii(dest, self._hf, self._hi, self._hs, data) else: byteorder = byteorder or self.byteorder _io.write_sac(dest, self._hf, self._hi, self._hs, data, byteorder=byteorder)
@classmethod
[docs] def _from_arrays(cls, hf=None, hi=None, hs=None, data=None): """ Low-level array-based constructor. This constructor is good for getting a "blank" SAC object, and is used in other, perhaps more useful, alternate constructors ("See Also"). No value checking is done and header values are completely overwritten with the provided arrays, which is why this is a hidden constructor. :param hf: SAC float header array :type hf: :class:`numpy.ndarray` of floats :param hi: SAC int header array :type hi: :class:`numpy.ndarray` of ints :param hs: SAC string header array :type hs: :class:`numpy.ndarray` of str :param data: SAC data array, optional. If omitted or None, the header arrays are intialized according to :func:`arrayio.init_header_arrays`. If data is omitted, it is simply set to None on the corresponding :class:`SACTrace`. .. rubric:: Example >>> sac = SACTrace._from_arrays() # doctest: +SKIP >>> print(sac) # doctest: +SKIP Reference Time = XX/XX/XX (XXX) XX:XX:XX.XXXXXX iztype not set lcalda = False leven = False lovrok = False lpspol = False """ # TODO: handle byte order independently instead of just from "hf". # XXX: assumes hf was provided. hf0, hi0, hs0 = _io.init_header_arrays(byteorder=hf.dtype.byteorder) if hf is None: hf = hf0 if hi is None: hi = hi0 if hs is None: hs = hs0 # get the default instance, but completely replace the arrays # initializes arrays twice, but it beats converting empty arrays to a # dict and then passing it to __init__, i think...maybe... sac = cls() sac._hf = hf sac._hi = hi sac._hs = hs sac.data = data return sac # TO/FROM OBSPY TRACES
@classmethod
[docs] def from_obspy_trace(cls, trace, keep_sac_header=True): """ Construct an instance from an ObsPy Trace. :param trace: Source Trace object :type trace: :class:`~obspy.core.Trace` instance :param keep_sac_header: If True, any old stats.sac header values are kept as is, and only a minimal set of values are updated from the stats dictionary: npts, e, and data. If an old iztype and a valid reftime are present, b and e will be properly referenced to it. If False, a new SAC header is constructed from only information found in the stats dictionary, with some other default values introduced. :type keep_sac_header: bool """ try: header = _ut.obspy_to_sac_header(trace.stats, keep_sac_header) except SacError: # not enough time info in old SAC header # XXX: try to do something besides ignore the old header? header = _ut.obspy_to_sac_header(trace.stats, keep_sac_header=False) # handle the data headers data = trace.data try: if len(data) == 0: # data is a empty numpy.array data = None except TypeError: # data is None data = None try: byteorder = data.dtype.byteorder except AttributeError: # data is None byteorder = '=' hf, hi, hs = _io.dict_to_header_arrays(header, byteorder=byteorder) sac = cls._from_arrays(hf, hi, hs, data) # sac._flush_headers() return sac
[docs] def to_obspy_trace(self, debug_headers=False): """ Return an ObsPy Trace instance. Required headers: nz-time fields, npts, delta, calib, kcmpnm, kstnm, ...? :param debug_headers: Include _all_ SAC headers into the Trace.stats.sac dictionary. :type debug_headers: bool .. rubric:: Example >>> sac = SACTrace() # doctest: +SKIP >>> tr = sac.to_obspy_trace() # doctest: +SKIP """ # make the obspy test for tests/data/testxy.sac pass # ObsPy does not require a valid reftime # try: # self.validate('reftime') # except SacInvalidContentError: # if not self.nzyear: # self.nzyear = 1970 # if not self.nzjday: # self.nzjday = 1 # for hdr in ['nzhour', 'nzmin', 'nzsec', 'nzmsec']: # if not getattr(self, hdr): # setattr(self, hdr, 0) self.validate('delta') if self.data is None: # headonly is True # Make it something palatable to ObsPy data = np.array([], dtype=self._hf.dtype.byteorder + 'f4') else: data = self.data sachdr = _io.header_arrays_to_dict(self._hf, self._hi, self._hs, nulls=debug_headers) # TODO: logic to use debug_headers for real stats = _ut.sac_to_obspy_header(sachdr) return Trace(data=data, header=stats) # ---------------------- other properties/methods -------------------------
[docs] def validate(self, *tests): """ Check validity of loaded SAC file content, such as header/data consistency. :param tests: One or more of the following validity tests: 'delta' : Time step "delta" is positive. 'logicals' : Logical values are 0, 1, or null 'data_hdrs' : Length, min, mean, max of data array match header values. 'enums' : Check validity of enumerated values. 'reftime' : Reference time values in header are all set. 'reltime' : Relative time values in header are absolutely referenced. 'all' : Do all tests. :type tests: str :raises: :class:`SacInvalidContentError` if any of the specified tests fail. :class:`ValueError` if 'data_hdrs' is specified and data is None, empty array, or no tests specified. .. rubric:: Example >>> sac = SACTrace.read(filename) # doctest: +SKIP >>> try: # doctest: +SKIP sac.validate('delta') # doctest: +SKIP except SacInvalidContentError as e: # doctest: +SKIP # i'm sure this is what they meant:-) sac.delta *= -1.0 # doctest: +SKIP sac.validate('delta') # doctest: +SKIP >>> sac.data += 5.0 # doctest: +SKIP >>> try: # doctest: +SKIP sac.validate('data_hdrs') # doctest: +SKIP except SacInvalidContentError: # doctest: +SKIP sac._flush_headers() # doctest: +SKIP sac.validate('data_hdrs') # doctest: +SKIP """ _io.validate_sac_content(self._hf, self._hi, self._hs, self.data, *tests)
[docs] def _format_header_str(self, hdrlist='all'): """ Produce a print-friendly string of header values for __repr__ , .listhdr(), and .lh() """ # interpret hdrlist if hdrlist == 'all': hdrlist = sorted(self._header.keys()) elif hdrlist == 'picks': hdrlist = ('a', 'b', 'e', 'f', 'o', 't0', 't1', 't2', 't3', 't4', 't5', 't6', 't7', 't8', 't9') else: msg = "Unrecognized hdrlist '{}'".format(hdrlist) raise ValueError(msg) # start building header string # # reference time header_str = [] try: timefmt = "Reference Time = %m/%d/%Y (%j) %H:%M:%S.%f" header_str.append(self.reftime.strftime(timefmt)) except (ValueError, SacError): msg = "Reference time information incomplete." warnings.warn(msg) notime_str = "Reference Time = XX/XX/XX (XXX) XX:XX:XX.XXXXXX" header_str.append(notime_str) # # reftime type # TODO: use enumerated value dict here? iztype = self.iztype if iztype is None: header_str.append("\tiztype not set") elif iztype == 'ib': header_str.append("\tiztype IB: begin time") elif iztype == 'io': header_str.append("\tiztype IO: origin time") elif iztype == 'ia': header_str.append("\tiztype IA: first arrival time") elif iztype[1] == 't': vals = (iztype.upper(), iztype[1:]) izfmt = "\tiztype {}: user-defined time {}" header_str.append(izfmt.format(*vals)) elif iztype == 'iunkn': header_str.append("\tiztype IUNKN (Unknown)") else: header_str.append("\tunrecognized iztype: {}".format(iztype)) # # non-null headers hdrfmt = "{:10.10s} = {}" for hdr in hdrlist: # XXX: non-null header values might have no property for getattr try: header_str.append(hdrfmt.format(hdr, getattr(self, hdr))) except AttributeError: header_str.append(hdrfmt.format(hdr, self._header[hdr])) return '\n'.join(header_str)
[docs] def listhdr(self, hdrlist='all'): """ Print header values. Default is all non-null values. :param hdrlist: Which header fields to you want to list. Choose one of {'all', 'picks'} or iterable of header fields. An iterable of header fields can look like 'bea' or ('b', 'e', 'a'). 'all' (default) prints all non-null values. 'picks' prints fields which are used to define time picks. An iterable of header fields can look like 'bea' or ('b', 'e', 'a'). .. rubric:: Example >>> sac = SACTrace.read('tests/data/test.sac') # doctest: +SKIP >>> sac.lh() # doctest: +SKIP Reference Time = 07/18/1978 (199) 08:00:00.000000 unrecognized iztype: None b = 10.0 delta = 1.0 depmax = 1.0 depmin = -1.0 e = 109.0 istreg = 1 kcmpnm = Q kevnm = FUNCGEN: SINE kstnm = STA npts = 100 nvhdr = 6 nzhour = 8 nzjday = 199 nzmin = 0 nzmsec = 0 nzsec = 0 nzyear = 1978 """ # https://ds.iris.edu/files/sac-manual/commands/listhdr.html print(self._format_header_str(hdrlist))
[docs] def lh(self, *args, **kwargs): """Alias of listhdr method.""" self.listhdr(*args, **kwargs)
[docs] def __str__(self): return self._format_header_str()
[docs] def __repr__(self): # XXX: run self._flush_headers first? # TODO: make this somehow more readable. h = self._header fmt = ", {}={!r}" * len(h) argstr = fmt.format(*chain.from_iterable(h.items()))[2:] return self.__class__.__name__ + "(" + argstr + ")"
[docs] def copy(self): return deepcopy(self)
[docs] def _flush_headers(self): """ Flush to the header arrays any header property values that may not be reflected there, such as data min/max/mean, npts, e. """ # XXX: do I really care which byte order it is? # self.data = np.require(self.data, native_str('<f4')) self._hi[HD.INTHDRS.index('npts')] = self.npts self._hf[HD.FLOATHDRS.index('e')] = self.e self._hf[HD.FLOATHDRS.index('depmin')] = self.depmin self._hf[HD.FLOATHDRS.index('depmax')] = self.depmax self._hf[HD.FLOATHDRS.index('depmen')] = self.depmen
[docs] def _allt(self, shift): """ Shift all relative time headers by some value (addition). Similar to SAC's "chnhdr allt". Note ---- This method is triggered by setting an instance's iztype or changing its reference time, which is the most likely use case for this functionality. If what you're trying to do is set an origin time and make a file origin-based: SAC> CHNHDR O GMT 1982 123 13 37 10 103 SAC> LISTHDR O O 123.103 SAC> CHNHDR ALLT -123.103 IZTYPE IO ...it is recommended to just make sure your target reference header is set and correct, and set the iztype: >>> sac.o = UTCDateTime(year=1982, julday=123, # doctest: +SKIP hour=13, minute=37, # doctest: +SKIP second=10, microsecond=103) # doctest: +SKIP >>> sac.iztype = 'io' # doctest: +SKIP The iztype setter will deal with shifting the time values. """ for hdr in ['b', 'o', 'a', 'f'] + ['t'+str(i) for i in range(10)]: val = getattr(self, hdr) if val is not None: setattr(self, hdr, val + shift)
[docs] def _set_distances(self, force=False): """ Calculate dist, az, baz, gcarc. If force=True, ignore lcalda. """ if self.lcalda or force: try: m, az, baz = gps2dist_azimuth(self.evla, self.evlo, self.stla, self.stlo) dist = m / 1000.0 gcarc = kilometer2degrees(dist) self.az = az self.baz = baz self.dist = dist self.gcarc = gcarc except TypeError: # one of the geographic values is None msg = "Not enough information to calculate distance, azimuth." raise SacHeaderError(msg) else: msg = "lcalda is False or unset. To set distances, set it to True." raise SacError(msg)