source: SHX/trunk/sandbox/traceplotter.py @ 394

# -*- coding: utf-8 -*-

import wx
import wx.lib.scrolledpanel as SP
from SeismicHandler.core import read

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)

        st = read("/online4/2011/GR/BFO/BHZ.D/GR.BFO..BHZ.D.2011.124")
        self.tr = st[0]

    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()