Changeset 151

Timestamp:

09/21/2009 09:13:56 AM (14 years ago)

Author:

marcus

Message:

• setting trace info entry fires TRACEUPDATE event
• debug plotting routine for traces

Location:

SHX/trunk/src/SeismicHandler

Files:

• core/modules/Trace.py (modified) (2 diffs)
• core/shlib.py (modified) (1 diff)
• startup.py (modified) (2 diffs)
• tests/test_commands.py (modified) (1 diff)
• tests/test_traces.py (modified) (2 diffs)

SHX/trunk/src/SeismicHandler/core/modules/Trace.py

@@ -270 +270 @@
         return self.tracedata[self.traces[number-1]]
 
+    def plot(self, share=False, save=None):
+        """debug plot"""
+
+        if not len(self):
+            raise Exception("no traces available")
+
+        from SeismicHandler.core.modules.Graphics import pylab
+
+        fig = pylab.figure()
+        fig.canvas.set_window_title("SHX - plot of %u traces" % len(self))
+        pylab.subplots_adjust(hspace=0.1, wspace=20, left=0.1, right=0.95, bottom=0.05, top=0.95)
+
+        fig.set_facecolor('w')
+
+        for i, trc in enumerate(self):
+            # link x axis if more than one trace
+            if len(fig.axes) and share:
+                ax = pylab.subplot(len(self), 1, i+1, sharex=fig.axes[0])
+            else:
+                ax = pylab.subplot(len(self), 1, i+1)
+
+            # no frame
+            ax.set_frame_on(False)
+
+            # no y ticks
+            ax.yaxis.set_major_locator(pylab.NullLocator())
+
+            # build label from station ...
+            if trc.station:
+                label = trc.station
+            # ... and component name
+            if trc.comp:
+                label += " " + trc.comp
+
+            try:
+                ax.set_ylabel(label, rotation='horizontal')
+            except:
+                pass
+
+            # no ticks for upper traces
+            if i < len(self)-1:
+                ax.xaxis.set_major_locator(pylab.NullLocator())
+            else:
+                ax.xaxis.tick_bottom()
+
+            pylab.plot(np.arange(0, trc.delta * trc.length, trc.delta), trc.fetchData(), "k", linewidth=.5)
+
+            # add some extra space around traces
+            limit = abs(max(pylab.ylim(), key=abs)) * 1.1
+            pylab.ylim(-limit, limit)
+
+        if save:
+            pass
+        else:
+            pylab.show()
+
 if __name__ == "__main__":
     t = Traces()
     from SeismicHandler.core.commands import Run
-    from SeismicHandler.core.modules.Graphics import pylab
 
     EventManager.block(Event(EventManager.REDRAW))
@@ -283 +338 @@
 
     Run("sum all")
-
     EventManager.unblock(Event(EventManager.REDRAW))
 
-    fig = pylab.figure()
-    pylab.subplots_adjust(hspace=0.00001,wspace=20)
-
-    for i, trc in enumerate(t):
-        trc.station = "TST%u" % i
-        trc.zoom = 5.5
-        trc.comment = "Tell me what you think of me #%u" % i
-        trc.chan1 = "B"
-        trc.chan2 = "H"
-        trc.comp = "N"
-
-        EventManager.trigger(Event(EventManager.REDRAW))
-
-#        trc.listInfo()
-
-        ax = pylab.subplot(len(t),1,i+1)
-
-        # don't show axis ticks except for last one
-        if i < len(t)-1:
-            ax.xaxis.set_major_locator(pylab.NullLocator())
-
-        pylab.plot(np.arange(0, trc.delta * trc.length, trc.delta), trc.fetchData(), "k")
-#        trc.plot()
-
-    pylab.show()
+    t.plot(True)

SHX/trunk/src/SeismicHandler/core/shlib.py

@@ -165 +165 @@
             fct(trc, ientry, value, C.byref(status))
 
+        EventManager.trigger(Event(EventManager.TRACEUPDATE))
         if status.value:
             logger.error("could not set info entry '%s'" % name)

SHX/trunk/src/SeismicHandler/startup.py

@@ -38 +38 @@
 from SeismicHandler.core.modules.Variables import System, Symbol
 from SeismicHandler.core.modules.Trace import Traces as xTraces
+from SeismicHandler.core.modules import Tools
+from SeismicHandler.core.modules.Events import Event, EventManager
 
 # access to traces
@@ -54 +56 @@
         try:
             Run(options.args[0])
+            # After running script from command line, update trace information. 
+            EventManager.trigger(Event(EventManager.TRACEUPDATE))
         except IndexError:
             pass

SHX/trunk/src/SeismicHandler/tests/test_commands.py

@@ -484 +484 @@
         Run("@READF/FMT=5 %s ALL" % convertFilename("data/GRF_031102_0225.GSE"))
         self.assertEqual(int(System.tottrcs), 19)
+#        self.traces.plot()
 
     def testReads(self):

SHX/trunk/src/SeismicHandler/tests/test_traces.py

@@ -118 +118 @@
     def testInfoEntriesPy(self):
         # set values via python
+        EventManager.block(Event(EventManager.TRACEUPDATE))
         self.trace.station = "FOOBAR"
         self.trace.chan1 = "E"
@@ -124 +125 @@
         self.trace.comment = "comment with blanks"
         self.trace.zoom = 5
-
-        EventManager.trigger(Event(EventManager.REDRAW))
+        EventManager.unblock(Event(EventManager.TRACEUPDATE))
 
         self.assertEqual(self.trace.alloc, 1000)