Changeset 401

Timestamp:

06/08/2011 06:48:31 PM (12 years ago)

Author:

marcus

Message:

• multiple traces
• relative/absolute time-axis
• station/component information
• minor improvements

File:

• SHX/trunk/sandbox/traceplotter.py (modified) (8 diffs)

SHX/trunk/sandbox/traceplotter.py

@@ -8 +8 @@
 import os
 import wx.lib.scrolledpanel as SP
+from obspy.core import UTCDateTime
 from SeismicHandler.core import read
 from obspy.sh.core import fromUTCDateTime
 
+# width of station info
+STATION_INFO = 75
+# border on right side
+BORDER = 15
+
+class Traces(object):
+    def __init__(self):
+        self.traces = []
+        self.ids = []
+
+    def add(self, trace):
+        self.traces.append(trace)
+
+    def remove(self, trace):
+        pass
+    
+    def pop(self, trace):
+        pass
+
+    def __iter__(self):
+        for t in self.traces:
+            yield t
+
+    def __len__(self):
+        return len(self.traces)
+
+    def __getitem__(self, item):
+        return self.traces[item]
+
 class MyPanel(SP.ScrolledPanel):
     def __init__(self, parent, fname=None):
+        self.traces = Traces()
+
         SP.ScrolledPanel.__init__(self, parent, style=wx.BORDER_SIMPLE)
 
@@ -22 +54 @@
 
         self.refresh = False
+        self.relativeAxis = False
         self.load(fname)
         self.OnPaint(None)
         self.SetFocus()
 
+    def _debug(self, *args):
+        print time.time(), " ".join([str(i) for i in args])
+
+    def getTraceUnderCursor(self, y):
+        try:
+            return self.traces[int(float(y)/self.heigth*len(self.traces))]
+        except IndexError:
+            return self.traces[-1]
+
     def OnMotion(self, evt):
-        duration = self.end - self.start
-        x = evt.GetX()
-
-        t = x * duration / self.width
-
-        self.parent.SetStatusText(fromUTCDateTime(self.start + t))
-
+        if not self.relativeAxis or len(self.traces) == 1:
+            duration = self.end - self.start
+            start = self.start
+        else:
+            # get trace beneath cursor
+            y = evt.GetY()
+            t = self.getTraceUnderCursor(y)
+            
+            start = t.stats.starttime
+            duration = t.stats.endtime - t.stats.starttime
+
+        x = evt.GetX() - STATION_INFO
+        t = x * duration / (self.width - STATION_INFO - BORDER)
+        timestamp = start + t
+
+        self.parent.SetStatusText(fromUTCDateTime(timestamp))
+
+        # let the event propagate
         evt.Skip()
 
     def load(self, fname=None):
-        print time.time(), "Loading %s..." % fname
+        self._debug("Loading %s..." % fname)
         try:
             st = read(fname)
-            self.tr = st[0]
+            trace = st[0]
+            self.traces.add(trace)
+            self.resetTimeRange()
         except Exception, e:
-            print time.time(), e.message
-
-        if hasattr(self, "start"):
-            delattr(self, "start")
-            delattr(self, "end")
+            msg = e.message and e.message or "File not readable!"
+            dlg = wx.MessageDialog(self, msg, "An error occured!", wx.OK | wx.ICON_ERROR)
+            dlg.ShowModal()
+            dlg.Destroy()
+
+            self._debug(e.message)
+
+    def resetTimeRange(self):
+        min = UTCDateTime() # now
+        max = UTCDateTime(0) # 1970
+        maxlength = 0
+
+        for t in self.traces:
+            if t.stats.starttime < min:
+                min = t.stats.starttime
+            if t.stats.endtime > max:
+                max = t.stats.endtime
+            # needed for relative time axis
+            l = t.stats.endtime - t.stats.starttime
+            if l > maxlength:
+                maxlength = l
+
+        self.start = min
+        self.end = max
+        self.maxlength = maxlength
 
     def OnIdle(self, evt):
@@ -58 +133 @@
     def Paint(self):
         width, height = self.GetClientSize()
+        numTraces = len(self.traces)
+        theight = height // numTraces
         self.width = width
-
-        start = self.start = getattr(self, "start", self.tr.stats.starttime)
-        end = self.end = getattr(self, "end", self.tr.stats.endtime)
-
-        print time.time(), "Preparing plotting data..."
-        self.tr.shxPrepareImageData(width, height, (start, end), zoom=0.9)
-
-        bitmap = wx.EmptyBitmap(width, height)
-        buffer = wx.MemoryDC(bitmap)
-        # smoothed graphics, slower
-#        dc = wx.GCDC(buffer)
-        dc = buffer
-        dc.SetBrush(wx.TRANSPARENT_BRUSH)
-        dc.Clear()
-
-        dc.SetPen(wx.Pen((45,45,45), 1))
-
-        dc.BeginDrawing()
-        dc.DrawLineList(self.tr.shxImageData)
-        dc.EndDrawing()
-
-        plotStartSec = self.tr.stats.starttime - start
-        offsetX = 0
-        if plotStartSec:
-            # re-calculate offset
-            portion = plotStartSec / (end - start)
-            offsetX = portion * width
-            print time.time(), portion, offsetX
-
+        self.heigth = height
+        pltwidth = width - STATION_INFO - BORDER
+
+        # canvas holds the final figure
+        bmap = wx.EmptyBitmap(width, height)
+        canvas = wx.MemoryDC(bmap)
+        canvas.SetBrush(wx.TRANSPARENT_BRUSH)
+        canvas.Clear()
+        canvas.SetFont(wx.Font(8, wx.SWISS, wx.NORMAL, wx.NORMAL))
+
+        for i, t in enumerate(self.traces):
+            if not self.relativeAxis:
+                start = self.start
+                end = self.end
+            else:
+                start = t.stats.starttime
+                end = start + self.maxlength
+
+            self._debug("Preparing plotting data...", t.id)
+            t.shxPrepareImageData(pltwidth, theight, (start, end), zoom=0.9)
+
+            # make trace graphics sketch
+            bitmap = wx.EmptyBitmap(pltwidth, theight)
+            buffer = wx.MemoryDC(bitmap)
+            # smoothed graphics, slower
+    #        dc = wx.GCDC(buffer)
+            dc = buffer
+            dc.SetBrush(wx.TRANSPARENT_BRUSH)
+            dc.Clear()
+
+            dc.SetPen(wx.Pen((45,45,45), 1))
+
+            dc.BeginDrawing()
+            dc.DrawLineList(t.shxImageData)
+            dc.EndDrawing()
+
+            # add trace picture to canvas
+            plotStartSec = t.stats.starttime - start
+            offsetX = 0
+            if plotStartSec and not self.relativeAxis:
+                # re-calculate offset
+                portion = plotStartSec / (end - start)
+                offsetX = portion * pltwidth
+                self._debug(portion, offsetX)
+
+            # copy data
+            canvas.Blit(offsetX+STATION_INFO, i*theight, width, height, buffer, 0, 0, wx.AND)
+            txt = "%d: %s %s" % (i+1, t.stats.station, t.stats.channel[-1])
+            _, h, _, _ = canvas.GetFullTextExtent(txt)
+            canvas.DrawText(txt, 5, i*theight+theight//2-h//2)
+
+        # make it visible
         dc2 = wx.AutoBufferedPaintDCFactory(self)
         dc2.Clear()
-        dc2.Blit(offsetX, 0, width, height, buffer, 0, 0, wx.AND)
+        dc2.Blit(0, 0, width, height, canvas, 0, 0)
+
+    def onTimeAxis(self, evt):
+        recent = self.relativeAxis
+        self.relativeAxis = evt.GetString() == "relative"
+        if recent != self.relativeAxis:
+            self.OnPaint(evt)
 
 class MyFrame(wx.Frame):
-    def __init__(self, parent, title):
-        wx.Frame.__init__(self, parent, title=title, size=(640,280), style=wx.DEFAULT_FRAME_STYLE | wx.FULL_REPAINT_ON_RESIZE)
+    def __init__(self, parent, title, size=(640,280)):
+        wx.Frame.__init__(self, parent, title=title, size=size, style=wx.DEFAULT_FRAME_STYLE | wx.FULL_REPAINT_ON_RESIZE)
         try:
             fname = sys.argv[1]
@@ -100 +206 @@
         except IndexError:
             fname = None
+
         self.canvas = MyPanel(self, fname)
         self.CreateStatusBar()
@@ -106 +213 @@
         menuBar = wx.MenuBar()
         firstmenu = wx.Menu()
-        read = wx.MenuItem(firstmenu, wx.NewId(), 'Re&ad...\tr')
+        readdb = wx.MenuItem(firstmenu, wx.NewId(), 'Re&ad...\tr')
+        readdb.SetBitmap(wx.ArtProvider.GetBitmap(wx.ART_FOLDER_OPEN, wx.ART_OTHER, (16, 16)))
+        self.Bind(wx.EVT_MENU, self.onReadDb, id=readdb.GetId())
+        firstmenu.AppendItem(readdb)
+
+        read = wx.MenuItem(firstmenu, wx.NewId(), 'Read &file...\tCTRL+R')
+        read.SetBitmap(wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN, wx.ART_OTHER, (16, 16)))
         self.Bind(wx.EVT_MENU, self.onRead, id=read.GetId())
         firstmenu.AppendItem(read)
+
         quit = wx.MenuItem(firstmenu, wx.NewId(), '&Quit\tCtrl+Q')
+        quit.SetBitmap(wx.ArtProvider.GetBitmap(wx.ART_QUIT, wx.ART_OTHER, (16, 16)))
         self.Bind(wx.EVT_MENU, self.onQuit, id=quit.GetId())
         firstmenu.AppendItem(quit)
@@ -125 +240 @@
 #        self.Bind(wx.EVT_CHAR_HOOK, self.onKeyDown)
 
+        self.toolbar = tb = self.CreateToolBar(wx.TB_TEXT | wx.TB_NOICONS, -1)
+        msg = ' time-axis '
+        tb.AddControl(wx.StaticText(tb, -1, msg))
+        timeaxis = wx.Choice(tb, wx.NewId(), (-1, -1), (-1, -1), ["absolute", "relative"])
+        tb.AddControl(timeaxis)
+        tb.Realize()
+        self.Bind(wx.EVT_CHOICE, self.canvas.onTimeAxis, timeaxis)
+
         self.Show()
 
@@ -132 +255 @@
             self.onRead(evt)
         else:
-            print time.time(), "key code", kc
+            print self._debug("key code", kc)
 
     def onQuit(self, evt):
@@ -156 +279 @@
             self.canvas.load(fqpn)
 
+    def onReadDb(self, evt):
+        dlg = wx.MessageDialog(self, "not implemented yet", "An error occured!", wx.OK)
+        dlg.ShowModal()
+        dlg.Destroy()
+
 def main():
     app = wx.App(False)
-    _ = MyFrame(None, 'Trace Plotter Test')
+    _ = MyFrame(None, 'Trace Plotter Test', wx.DisplaySize())
     app.MainLoop()