added a modified version of erics memleak script for guis

svn path=/trunk/matplotlib/; revision=3132

Author: jdh2358

lib/matplotlib/backend_bases.py

@@ -1151,7 +1151,7 @@ class NavigationToolbar2:
 
     def __init__(self, canvas):
         self.canvas = canvas
-
+        canvas.toolbar = self
         # a dict from axes index to a list of view limits
         self._views = Stack()
         self._positions = Stack()  # stack of subplot positions

lib/matplotlib/backends/backend_tkagg.py

@@ -311,8 +311,9 @@ def destroy(self, *args):
             if self.window is not None:
                 self.window.quit()
         if self.window is not None:
-            #print 'calling window destroy'
+            #self.toolbar.destroy()
             self.window.destroy()
+
             pass
         self.window = None
 
@@ -534,7 +535,7 @@ def __init__(self, canvas, window):
         self.canvas = canvas
         self.window = window
         self._idle = True
-        Tk.Frame.__init__(self, master=self.canvas._tkcanvas)
+        #Tk.Frame.__init__(self, master=self.canvas._tkcanvas)
         NavigationToolbar2.__init__(self, canvas)
 
     def set_message(self, s):

lib/matplotlib/cbook.py

@@ -685,6 +685,19 @@ def reverse_dict(d):
     return dict([(v,k) for k,v in d.items()])
 
 
+def report_memory(i):
+    'return the memory consumed by process'
+    pid = os.getpid()
+    if sys.platform=='sunos5':
+        a2 = os.popen('ps -p %d -o osz' % pid).readlines()
+        mem = int(a2[-1].strip())
+        #print i, '  ', int(a2[-1].strip())
+    elif sys.platform.startswith('linux'):
+        a2 = os.popen('ps -p %d -o rss,sz' % pid).readlines()
+        mem = int(a2[1].split()[1])
+
+    return mem
+ 
 if __name__=='__main__':
     assert( allequal([1,1,1]) )
     assert(not  allequal([1,1,0]) )

lib/matplotlib/figure.py

@@ -153,6 +153,7 @@ def __init__(self,
         self.subplotpars = subplotpars
 
         self._axstack = Stack()  # maintain the current axes
+        self.axes = []
         self.clf()
 
         self._cachedRenderer = None
@@ -514,7 +515,13 @@ def clf(self):
         """
         Clear the figure
         """
-        self.axes = []
+        for ax in self.axes:
+            ax.cla()
+            self.delaxes(ax)
+
+        toolbar = getattr(self.canvas, 'toolbar', None)
+        if toolbar is not None:
+            toolbar.update()
         self._axstack.clear()
         self._seen = {}
         self.lines = []

unit/memleak_gui.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+'''
+This illustrates a leak that occurs with any interactive backend.
+'''
+import os, sys, time
+import gc
+import matplotlib
+
+#matplotlib.use('TkAgg') # or TkAgg or WxAgg or QtAgg or Gtk
+matplotlib.rcParams['toolbar'] = 'toolbar2'   # None, classic, toolbar2
+#matplotlib.rcParams['toolbar'] = None   # None, classic, toolbar2
+
+import pylab
+from matplotlib import _pylab_helpers as ph
+import matplotlib.cbook as cbook
+
+indStart, indEnd = 30, 50
+for i in range(indEnd):
+
+    fig = pylab.figure()
+    fig.savefig('test')
+    fig.clf()
+    pylab.close(fig)
+    val = cbook.report_memory(i)
+    print i, val 
+    gc.collect()
+    if i==indStart: start = val # wait a few cycles for memory usage to stabilize
+
+gc.collect()
+print
+print 'uncollectable list:', gc.garbage
+print
+end = val
+if i > indStart:
+    print 'Average memory consumed per loop: %1.4fk bytes\n' % ((end-start)/float(indEnd-indStart))
+