wxPython 2.1b1:

[wxWidgets.git] / utils / wxPython / tests / memleak.py

diff --git a/utils/wxPython/tests/memleak.py b/utils/wxPython/tests/memleak.py
new file mode 100644 (file)
index 0000000..2e62181
--- /dev/null
+++ b/utils/wxPython/tests/memleak.py
@@ -0,0 +1,157 @@
+
+from wxPython.wx import *
+
+#-------------------------------------------------------------------
+# class MyWindow(wxScrolledWindow):
+#----------------------------------
+# Copes with the drawing of the main scrolled window.
+#
+# Data members:
+#  num - <int> number of list entries
+#  ostart - <int> line number of the top line of the previous redraw
+#  vw - <int> width of the viewing window
+#  vh - <int> height of the viewing window
+#  smalltext - <int> 0 = size 12pt, 1 = size 9pt text
+#
+# Method members:
+#  OnPaint(evt) - basic draw handler
+#  OnDraw(dc) - called by OnPaint, redraws the screen if required
+#  update(updatelist) - called every 3 seconds if updates are needed.
+
+class MyWindow(wxScrolledWindow):
+  def __init__(self,num,parent,id,pos,size,style):
+    wxScrolledWindow.__init__(self,parent,id,pos,size,style)
+    self.SetBackgroundColour(wxWHITE)
+
+    self.num=num
+    self.ostart=0
+    self.smalltext = 0
+    self.vw,self.vh=self.GetClientSizeTuple()
+
+    # calculate font pt size needed: a bit of a kludge to get round
+    # font compatibility problems of X and Windows.
+    dc=wxClientDC(self)
+
+    dc.SetFont(wxFont(12,wxDEFAULT,wxNORMAL,wxNORMAL,FALSE))
+    if dc.GetTextExtent("XXXXXXXXXX")[0] > 100:
+      self.smalltext = 1
+
+  def OnPaint(self,evt):
+    """ overriding OnPaint to give handler. """
+    dc = wxPaintDC(self)
+    self.PrepareDC(dc)
+    self.OnDraw(dc)
+
+  def update(self,updlist):
+    """ handles line by line updating of list entries. """
+    dc = wxClientDC(self)
+    self.PrepareDC(dc)
+    dc.SetBrush(wxWHITE_BRUSH)
+    dc.SetPen(wxWHITE_PEN)
+
+    if self.smalltext == 1:
+      dc.SetFont(wxFont(9,wxDEFAULT,wxNORMAL,wxNORMAL,FALSE))
+    else:
+      dc.SetFont(wxFont(12,wxDEFAULT,wxNORMAL,wxNORMAL,FALSE))
+
+    dc.BeginDrawing()
+
+    for i in updlist:
+      if i >= self.ostart and i < self.ostart+self.vh/17+1:
+        dc.DrawRectangle(0,i*17,self.vw,17)
+        dc.DrawText("This is a simple test.Line "+str(i)+".",
+                    10,i*17+2)
+    dc.EndDrawing()
+
+  def OnDraw(self,dc):
+    """ Main redraw function. """
+
+    if self.smalltext == 1:
+      dc.SetFont(wxFont(9,wxDEFAULT,wxNORMAL,wxNORMAL,FALSE))
+    else:
+      dc.SetFont(wxFont(12,wxDEFAULT,wxNORMAL,wxNORMAL,FALSE))
+
+    vx,vstart=self.ViewStart()
+    self.vw,self.vh=self.GetClientSizeTuple()
+    vend=vstart+(self.vh/17) + 1
+    if vend > self.num: vend = self.num
+
+    dc.BeginDrawing()
+    if vstart > self.ostart: # if moving downwards...
+      for i in range(vend-(vstart-self.ostart+1),vend):
+        dc.DrawText("This is a simple test. Line "+str(i)+".",
+                    10,i*17+2)
+
+    elif vstart < self.ostart: # if moving upwards...
+      for i in range(vstart,self.ostart):
+        dc.DrawText("This is a simple test. Line "+str(i)+".",
+                    10,i*17+2)
+
+    elif vstart == self.ostart: # if not moving (redraw)...
+      #dc.Clear()
+      for i in range(vstart,vend):
+        dc.DrawText("This is a simple test. Line "+str(i)+".",
+                    10,i*17+2)
+
+    dc.EndDrawing()
+    self.ostart=vstart
+
+#--------------------------------------------------------------------
+
+class MyTimer(wxTimer):
+  def __init__(self,frame):
+    wxTimer.__init__(self)
+    self.frame_ = frame
+
+  def Notify(self):
+    self.frame_.idle()
+
+
+class MyFrame(wxFrame):
+  def __init__(self, parent, id, title):
+    wxFrame.__init__(self, parent, id, title,
+                     wxPoint(100, 100), wxSize(500, 300))
+
+    # number of entries
+    self.num = 30
+
+    # set up the scrolling window...
+    self.sw = MyWindow(self.num,self, -1,
+                       wxDefaultPosition, wxDefaultSize,
+                       wxVSCROLL|wxSUNKEN_BORDER)
+
+    self.sw.SetScrollbars(1,17,0,self.num+1)
+
+    lc = wxLayoutConstraints()
+    lc.top.SameAs(self, wxTop, 5)
+    lc.left.SameAs(self, wxLeft, 5)
+    lc.bottom.SameAs(self, wxBottom, 5)
+    lc.right.SameAs(self, wxRight,5)
+    self.sw.SetConstraints(lc)
+
+    self.timer=MyTimer(self)
+    # stupidly short interval time to accelerate memory leak problem:
+    self.timer.Start(80)
+
+  def idle(self):
+    #usually just update one or two lines; to accelerate problem,
+    #every line is updated here.
+    self.sw.update(range(self.num))
+
+
+######################################################################
+# Main procedure....
+
+if __name__ == "__main__":
+  class MyApp(wxApp):
+    def OnInit(self):
+
+      self.frame = MyFrame(NULL, -1, "Memory Leak Tester")
+      self.frame.Show(true)
+
+      self.exiting = FALSE;
+      return true
+
+  app = MyApp(0)     # Create an instance of the application class
+  app.MainLoop()     # Tell it to start processing events
+