Changeset 394

Timestamp:

05/25/2011 06:12:40 PM (12 years ago)

Author:

marcus

Message:

more plotting tests

Location:

SHX/trunk

Files:

• SeismicHandler/patches/ObsPy.py (modified) (5 diffs)
• sandbox/bc_example.py (modified) (1 diff)
• sandbox/traceplotter.py (modified) (1 diff)

SHX/trunk/SeismicHandler/patches/ObsPy.py

@@ -28 +28 @@
 
 import numpy as np
+import math
+import wx
 from obspy.core import Stream, Trace
 #from obspy.core import UTCDateTime
@@ -70 +72 @@
     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
+    zoom : global zoom factor (y extension)
+    timescale : extension of current timescale (tuple of min and max)
+    norm : norm method: "window", "total" or value (effects y extension)
     """
     if not timescale:
-        # use trace's dimensions
+        # use trace's total dimensions
         windowstart, windowend = (self.stats.starttime, self.stats.endtime)
     else:
@@ -91 +93 @@
 
     # width of the total screen that is covered by trace data (pixels)
-    portion = duration_trace * width / duration_total
+    pixel_width = duration_trace * width / duration_total
+    self.shxPixelWidth = pixel_width
     
     # data points per pixel
     npts = pt.stats.npts
-    dpp = int(npts / portion)
+    dpp = int(npts / pixel_width)
 
     # get copy of data
-    self.shx_plotdata = pt.data.copy()
+    self.shxPlotData = pt.data.copy()
 
     # use minmax approach if more than 4 data points per pixel
+    # self.shx_plotdata will be changed
     if dpp >= 4:
-        dimy = npts // dpp
-        shape = dpp * dimy
-        print "minmax", npts, dpp, dimy, shape
-        
+        # do integer division since array shape values cannot be floats
+        dimx = npts // dpp
+        covered = dpp * dimx
 
-    print width, height, norm, duration_trace, duration_total, portion, dpp, type(self.shx_plotdata)
+        # get data for reshape operation
+        data = self.shxPlotData[:covered]
 
+        # extra treatment for remaining values
+        remaining = self.shxPlotData[covered:]
+
+        # reshape data and get min/max values
+        data.shape = dimx, dpp
+        _min = data.min(axis=1)
+        _max = data.max(axis=1)
+        # combine data
+        joined = np.append([_min], [_max], axis=0).T.flatten()
+        # handle remaining
+        if len(remaining):
+            joined = np.append(joined, [remaining.min(), remaining.max()])
+
+        print "pixel_width", pixel_width, "minmax", "npts", npts, "dpp", dpp, "dimx", dimx, \
+              "covered", covered, "len_rest", len(remaining), "data_len", \
+              len(data), "joined", len(joined)
+
+        self.shxPlotData = joined
+
+    # At this stage the x-transformed data can be cached! If pixel_width doesn't
+    # change, the data can be reused to save time.
+
+    print width, height, norm, duration_trace, duration_total, pixel_width, dpp, type(self.shxPlotData)
+
+    # get basis for normation
     if hasattr(norm, "lower"):
         norm = norm.lower()
@@ -117 +146 @@
         else:
             raise ValueError("Invalid input for normation!")
-    else:
-        norm = norm
 
-# traces in a stream are merge automatically
+    # Calculate y extension, leads to raw "image" data.
+    # Use a copy to save time if only y changes occur (e.g. zooming).
+    # Apply normation, calibration and total height factor
+    y = self.shxPlotData.copy()
+    y *= self.stats.calib / norm * height / 2 * zoom
+
+    print self.shxPlotData.min(), self.shxPlotData.max(), y.min(), y.max()
+
+    # factor for x stretching (trade-off between pixel_width and array shape)
+    stretch = pixel_width / len(y)
+    width = int(math.ceil(pixel_width))
+
+    # calculate line data
+    # do not rely on framework axis' management -> shift/flip y-axis
+    offset = height // 2
+    lines = []
+    oldx, oldy = 0, -y[0] + offset
+
+    for i in xrange(1, len(y)):
+        newx = i * stretch
+        newy = -y[i] + offset
+        lines.append([oldx, oldy, newx, newy])
+        oldx, oldy = newx, newy
+
+    self.shxImageData = lines
+
+def __traceX(parameters):
+    """
+    Only the x axis is adopted here. Zoom, norm etc. regarding the y extension
+    is done ...
+    """
+    pass
+
+# monkey patching obspy stream class: traces are merge automatically if stream
+# get's altered.
 Stream.__shx_init__, Stream.__init__ = Stream.__init__, streamAutoMergeInit
 Stream.__shx_append, Stream.append = Stream.append, streamAutoMergeAppend
@@ -128 +189 @@
 
 # add method for faster plotting
-Trace.shx_prepareImageData = tracePrepareDataForImage
+Trace.shxPrepareImageData = tracePrepareDataForImage
 
+# just an indicator
 patched = True

SHX/trunk/sandbox/bc_example.py

@@ -152 +152 @@
                  pos=wx.DefaultPosition,
                  size=wx.DefaultSize,
-                 style=wx.DEFAULT_FRAME_STYLE):
+                 style=wx.DEFAULT_FRAME_STYLE | wx.FULL_REPAINT_ON_RESIZE):
 
         wx.Frame.__init__(self, parent, ID, title, pos, size, style)

SHX/trunk/sandbox/traceplotter.py

@@ -1 +1 @@
 # -*- coding: utf-8 -*-
 
+import wx
+import wx.lib.scrolledpanel as SP
 from SeismicHandler.core import read
-#from obspy.core import UTCDateTime
 
-st = read()
-tr = st[0]
+class MyPanel(SP.ScrolledPanel):
+    def __init__(self, parent):
+        SP.ScrolledPanel.__init__(self, parent, style=wx.BORDER_SIMPLE)
+        self.Bind(wx.EVT_PAINT, self.OnPaint)
+#        self.Bind(wx.EVT_RESIZE, self.OnPaint)
+        self.Bind(wx.EVT_IDLE, self.OnIdle)
+        self.refresh = False
+        self.OnPaint(None)
 
-start = tr.stats.starttime - 4
-end = tr.stats.endtime + 4
+        st = read("/online4/2011/GR/BFO/BHZ.D/GR.BFO..BHZ.D.2011.124")
+        self.tr = st[0]
 
-tr.shx_prepareImageData(700, 100, (start, end))
+    def OnIdle(self, evt):
+        if self.refresh:
+            self.Paint()
+            self.refresh = False
+
+    def OnPaint(self, evt):
+        self.refresh = True
+
+    def Paint(self):
+        width, height = self.GetClientSize()
+
+        start = self.tr.stats.starttime - 4
+        end = self.tr.stats.endtime + 4
+
+        self.tr.shxPrepareImageData(width, height, (start, end))
+
+        bitmap = wx.EmptyBitmap(width, height)
+        buffer = wx.MemoryDC(bitmap)
+        buffer.Clear()
+#        dc = wx.GCDC(buffer)
+        dc = buffer
+        dc.BeginDrawing()
+        dc.DrawLineList(self.tr.shxImageData)
+        dc.EndDrawing()
+
+        dc2 = wx.AutoBufferedPaintDCFactory(self)
+        dc2.Blit(0, 0, width, height, buffer, 0, 0, wx.COPY)
+
+class MyFrame(wx.Frame):
+    def __init__(self, parent, title):
+        wx.Frame.__init__(self, parent, title=title, size=(640,480), style=wx.DEFAULT_FRAME_STYLE | wx.FULL_REPAINT_ON_RESIZE)
+        self.canvas = MyPanel(self)
+        self.Show()
+
+def main():
+    app = wx.App(False)
+    frame = MyFrame(None, 'Trace Plotter Test')
+    app.MainLoop()
+
+if __name__ == '__main__':
+    main()