source: SHX/trunk/SeismicHandler/patches/ObsPy.py @ 388

# -*- coding: utf-8 -*-
#
# Copyright (C) 2008-2011 Marcus Walther (walther@szgrf.bgr.de)
#
# This file is part of Seismic Handler eXtended (SHX)
# Full details can be found at project website http://www.seismic-handler.org/
#
# SHX is free software; you can redistribute it and/or modify
# it under the terms of the GNU LESSER GENERAL PUBLIC LICENSE as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# SHX is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with SHX (see license.txt); if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA

"""
This modules defines extensions to obspy. They are patched into the
original obspy modules at a very basic level in SeismicHandler/__init__.py

This module is named ObsPy.py to prevent import loops.
"""

import numpy as np
from obspy.core import Stream, Trace
#from obspy.core import UTCDateTime

__all__ = [
    "patched",
]

def _streamAutoMerge(self, *args, **kwargs):
    """
    Call obspy's Stream merge method to fill up gaps using the latest pre-gap
    sample value. This destroys the spectra of the data but it's common.
    """
    if self.traces:
        self.merge(method=1, fill_value='latest')

def streamAutoMergeInit(self, *args, **kwargs):
    self.__shx_init__(*args, **kwargs)
    _streamAutoMerge(self)

def streamAutoMergeAppend(self, *args, **kwargs):
    self.__shx_append(*args, **kwargs)
    _streamAutoMerge(self, *args, **kwargs)

def streamAutoMergeExtend(self, *args, **kwargs):
    self.__shx_extend(*args, **kwargs)
    _streamAutoMerge(self, *args, **kwargs)

def streamAutoMergeInsert(self, *args, **kwargs):
    self.__shx_insert(*args, **kwargs)
    _streamAutoMerge(self, *args, **kwargs)

def streamAutoMergeAdd(self, *args, **kwargs):
    self.__shx_add__(*args, **kwargs)
    _streamAutoMerge(self, *args, **kwargs)

def tracePrepareDataForImage(self, width, height=None, timescale=None, zoom=1, norm="window"):
    """
    Preparing trace data for fast plotting. Using numpy's minmax feature - this
    will not just downsample the data!

    width : total width of plotting window in pixels
    height : hight of plotting window in pixels
    zoom : global zoom factor
    timescale : extension of current timescale (min and max)
    norm : norm method: "window", "total" or value
    """
    if not timescale:
        # use trace's dimensions
        windowstart, windowend = (self.stats.starttime, self.stats.endtime)
    else:
        windowstart, windowend = timescale

    # XXX
    if not height:
        height = 1000

    # get slice for time window
    pt = self.slice(windowstart, windowend)

    duration_total = windowend - windowstart
    duration_trace = pt.stats.endtime - pt.stats.starttime

    # width of the total screen that is covered by trace data (pixels)
    portion = duration_trace * width / duration_total
    
    # data points per pixel
    npts = pt.stats.npts
    dpp = int(npts / portion)

    # get copy of data
    self.shx_plotdata = pt.data.copy()

    # use minmax approach if more than 4 data points per pixel
    if dpp >= 4:
        dimy = npts // dpp
        shape = dpp * dimy
        print "minmax", npts, dpp, dimy, shape
        

    print width, height, norm, duration_trace, duration_total, portion, dpp, type(self.shx_plotdata)

    if hasattr(norm, "lower"):
        norm = norm.lower()
        if norm == "window":
            norm = abs(pt.max())
        elif norm == "total":
            norm = self.max()
        else:
            raise ValueError("Invalid input for normation!")
    else:
        norm = norm

# traces in a stream are merge automatically
Stream.__shx_init__, Stream.__init__ = Stream.__init__, streamAutoMergeInit
Stream.__shx_append, Stream.append = Stream.append, streamAutoMergeAppend
Stream.__shx_extend, Stream.extend = Stream.extend, streamAutoMergeExtend
Stream.__shx_insert, Stream.insert = Stream.insert, streamAutoMergeInsert
Stream.__shx_add__, Stream.__add__ = Stream.__add__, streamAutoMergeAdd

# add method for faster plotting
Trace.shx_prepareImageData = tracePrepareDataForImage

patched = True