Changeset 404

Timestamp:

06/09/2011 05:43:13 PM (12 years ago)

Author:

marcus

Message:

• loading multiple files
• trace order top-bottom vs. bottom-top

File:

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

SHX/trunk/sandbox/traceplotter.py

@@ -42 +42 @@
 
 class MyPanel(SP.ScrolledPanel):
-    def __init__(self, parent, fname=None):
+    def __init__(self, parent, fnames=[]):
         self.traces = Traces()
 
@@ -55 +55 @@
         self.refresh = False
         self.relativeAxis = False
-        self.traceOrder = "tb"
-        self.load(fname)
+        self.traceOrder = "bottom-top"
+        self.load(fnames)
         self.OnPaint(None)
         self.SetFocus()
@@ -64 +64 @@
 
     def getTraceUnderCursor(self, y):
+        idx = int(float(y)/self.heigth*len(self.traces))
+        if self.traceOrder == "bottom-top":
+            idx = -idx-1
+
         try:
-            return self.traces[int(float(y)/self.heigth*len(self.traces))]
+            return self.traces[idx]
         except IndexError:
             return self.traces[-1]
@@ -92 +96 @@
         evt.Skip()
 
-    def load(self, fname=None):
-        self._debug("Loading %s..." % fname)
-        try:
-            st = read(fname)
-            trace = st[0]
-            self.traces.add(trace)
-            self.resetTimeRange()
-        except Exception, e:
-            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 load(self, fnames=[]):
+        if not fnames:
+            fnames = [None]
+
+        for fname in fnames:
+            self._debug("Loading %s..." % fname)
+
+            try:
+                st = read(fname)
+                trace = st[0]
+                self.traces.add(trace)
+            except Exception, e:
+                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)
+
+        self.resetTimeRange()
 
     def resetTimeRange(self):
@@ -149 +159 @@
         canvas.SetFont(wx.Font(8, wx.SWISS, wx.NORMAL, wx.NORMAL))
 
-        for i, t in enumerate(self.traces):
+        if self.traceOrder == "top-bottom":
+            trcs = self.traces
+        elif self.traceOrder == "bottom-top":
+            trcs = self.traces[::-1]
+
+        for i, t in enumerate(trcs):
             if not self.relativeAxis:
                 start = self.start
@@ -186 +201 @@
             # 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])
+            if self.traceOrder == "top-bottom":
+                idx = i+1
+            elif self.traceOrder == "bottom-top":
+                idx = -i+len(self.traces)
+            txt = "%d: %s %s" % (idx, t.stats.station, t.stats.channel[-1])
+
             _, h, _, _ = canvas.GetFullTextExtent(txt)
             canvas.DrawText(txt, 5, i*theight+theight//2-h//2)
@@ -201 +221 @@
             self.OnPaint(evt)
 
+    def onTraceOrder(self, evt):
+        recent = self.traceOrder
+        self.traceOrder = evt.GetString()
+        if recent != self.traceOrder:
+            self.OnPaint(evt)
+
+#        print [i for i in dir(self.parent.toolbar) if "id" in i.lower()]
+#        print dir(evt), evt.GetId()
+#        self.parent.toolbar.EnableTool(evt.GetId(), False)
+
 class MyFrame(wx.Frame):
     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)
+
+        files = []
+        for fname in sys.argv[1:]:
+            files.append(fname)
+
         try:
-            fname = sys.argv[1]
-            self.dirname, self.filename = os.path.split(fname)
+            self.dirname, self.filename = os.path.split(files[-1])
         except IndexError:
             fname = None
 
-        self.canvas = MyPanel(self, fname)
+        self.canvas = MyPanel(self, files)
         self.CreateStatusBar()
         self.SetStatusText("Seimic Handler wxWidgets version")
@@ -248 +282 @@
         timeaxis = wx.Choice(tb, wx.NewId(), (-1, -1), (-1, -1), ["absolute", "relative"])
         tb.AddControl(timeaxis)
+        self.Bind(wx.EVT_CHOICE, self.canvas.onTimeAxis, timeaxis)
+        msg = ' trace order '
+        tb.AddControl(wx.StaticText(tb, -1, msg))
+        traceorder = wx.Choice(tb, wx.NewId(), (-1, -1), (-1, -1), ["bottom-top", "top-bottom"])
+        tb.AddControl(traceorder)
+        print "xxx", traceorder.GetId()
+        self.Bind(wx.EVT_CHOICE, self.canvas.onTraceOrder, traceorder)
         tb.Realize()
-        self.Bind(wx.EVT_CHOICE, self.canvas.onTimeAxis, timeaxis)
 
         self.Show()
@@ -267 +307 @@
         self.dirname = getattr(self, "dirname", "")
         self.filename = getattr(self, "filename", "")
-        fqpn_recent = os.path.join(self.dirname, self.filename)
-        
-        dlg = wx.FileDialog(self, "Choose a file", self.dirname, "", "*.*", wx.OPEN)
+       
+        dlg = wx.FileDialog(self, "Choose a file", self.dirname, "", "*.*", wx.OPEN | wx.FD_MULTIPLE)
+
+        files = []
         if dlg.ShowModal() == wx.ID_OK:
+            files = dlg.GetFilenames()
             self.filename = dlg.GetFilename()
             self.dirname = dlg.GetDirectory()
         dlg.Destroy()
 
-        fqpn = os.path.join(self.dirname, self.filename)
-
-        # only update if new file
-        if fqpn != fqpn_recent:
+        for fname in files:
+            fqpn = os.path.join(self.dirname, fname)
+
             self.SetStatusText(fqpn)
-            self.canvas.load(fqpn)
+            self.canvas.load([fqpn])
 
     def onReadDb(self, evt):