Changeset 354

Timestamp:

03/30/2011 06:18:48 PM (12 years ago)

Author:

marcus

Message:

Some experiments.

File:

• SHX/trunk/sandbox/bc_example.py (modified) (7 diffs)

SHX/trunk/sandbox/bc_example.py

@@ -2 +2 @@
 
 import wx
+import wx.lib.scrolledpanel as SP
+import sys
 from bufferedcanvas import *
 
@@ -7 +9 @@
 
 class TestCanvas(BufferedCanvas):
-    def __init__(self, parent, ID=-1):
-        self.data = -read("../SeismicHandler/tests/data/mseed/2010.284.GR.GRA1..BHN")[0].data
+    def __init__(self, parent, ID=-1, pos=wx.DefaultPosition, size=wx.DefaultSize):
+        try:
+            fname = sys.argv[1]
+        except IndexError:
+            fname = "../SeismicHandler/tests/data/mseed/2010.284.GR.GRA1..BHN"
+        self.trace = read(fname)[0]
+
         # demean
-        self.data = self.data-self.data.mean()
-        min, max = self.data.min(), self.data.max()
-        self.base = abs(max) > abs(min) and abs(max) or abs(min)
-        self.npts = len(self.data)
+        self.trace.data = self.trace.data-self.trace.data.mean()
+
+        min, max = self.trace.data.min(), self.trace.data.max()
+        self.base = abs(self.trace.max())
+        self.npts = self.trace.stats.npts
 
         self.overlay = wx.Overlay()
 
-        BufferedCanvas.__init__(self,parent,ID)
+        BufferedCanvas.__init__(self, parent, ID, size=size)
 
         self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
         self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
         self.Bind(wx.EVT_MOTION, self.OnMouseMove)
+
+        try:
+            parent.Bind(wx.EVT_SCROLLWIN_CHANGED, self.OnScrollDone)
+        except:
+            parent.Bind(wx.EVT_SCROLLWIN, self.OnScrollDone)
+
+        parent.Bind(wx.EVT_MOUSEWHEEL, self.OnMouseWheel)
+
+    def OnMouseWheel(self, evt):
+        if evt.ControlDown():
+            print "wheel CTRL event"
+        else:
+            evt.Skip()
+
+    def OnScrollDone(self, evt):
+        print "scrolling finished"
+#        self.overlay.Reset()
+        dc = wx.ClientDC(self)
+        odc = wx.DCOverlay(self.overlay, dc)
+        odc.Clear()
+        del odc
+        evt.Skip()
 
     def OnRightDown(self, evt):
@@ -77 +107 @@
 
         w, h = self.buffer.Width-20, self.buffer.Height
+        spp = self.npts // w
+        tail = self.npts % w
+
+        print "samples per pixel: %u remains: %u" % (spp, tail)
+
+        # resample
+        if spp > 2:
+            pass
+
         zoomx = float(w)/self.npts
         zoomy = float(h)/self.base/2.1
-
         
         dc.SetBackground(wx.Brush("White"))
@@ -88 +126 @@
         lines = []
         offsetx, offsety = 10, int(h / 2.)
-        print "zoom: x=%.2f y=%.2f offset: x=%i y=%i" % (zoomx, zoomy, offsetx, offsety)
+        print "x: %u y: %u zoom: x=%.2f y=%.2f offset: x=%i y=%i" % (w, h, zoomx, zoomy, offsetx, offsety)
 
-        for n, i in enumerate(self.data):
-            lines.append([n*zoomx, i*zoomy])
+        oldx, oldy = offsetx, offsety
+        for n, i in enumerate(self.trace.data):
+            newx = n*zoomx+offsetx
+            newy = -i*zoomy+offsety
+            lines.append([oldx, oldy, newx, newy])
+            oldx, oldy = newx, newy
 
         dc.SetPen(wx.Pen('Grey', 1, wx.LONG_DASH))
@@ -101 +143 @@
 #        pen.SetCap(wx.CAP_ROUND)
         dc.SetPen(pen)
-        dc.DrawLines(lines, xoffset=offsetx, yoffset=offsety)
+        dc.DrawLineList(lines)
 
 class TestFrame(wx.Frame):
@@ -113 +155 @@
 
         wx.Frame.__init__(self, parent, ID, title, pos, size, style)
-        self.canvas = TestCanvas(self)
+
+        panel = SP.ScrolledPanel(
+            parent=self,
+            pos=pos,
+            size=(1, 1),
+            style=wx.TAB_TRAVERSAL|wx.SUNKEN_BORDER,
+            name="panel"
+        )
+
+        self.canvas = TestCanvas(panel, size=[500,200])
+
+        sizer  = wx.BoxSizer()
+        sizer.Add(self.canvas, proportion=2)
+
+        panel.SetSizer(sizer)
+        panel.SetupScrolling()
+
         self.Bind(wx.EVT_CLOSE, self.onClose)
 
@@ -122 +180 @@
 def main():
     app = wx.PySimpleApp()
-    frame = TestFrame(size=[600,200])
+    frame = TestFrame(size=[700,200])
     frame.Show(True)
     app.MainLoop()