cbook.report_memory: add Windows support via "tasklist"

efiring · efiring · commit 053b70bbfa1f · 2011-02-05T21:57:21.000Z
svn path=/trunk/matplotlib/; revision=8947
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,3 +1,6 @@
+2011-02-05 Add cbook memory monitoring for Windows, using
+           tasklist. - EF
+
 2011-02-05 Speed up Normalize and LogNorm by using in-place
            operations and by using float32 for float32 inputs
            and for ints of 2 bytes or shorter; based on
diff --git a/lib/matplotlib/cbook.py b/lib/matplotlib/cbook.py
@@ -1233,7 +1233,7 @@ def restrict_dict(d, keys):
 
 def report_memory(i=0):  # argument may go away
     'return the memory consumed by process'
-    from subprocess import Popen, PIPE
+    from subprocess import Popen, PIPE, check_output
     pid = os.getpid()
     if sys.platform=='sunos5':
         a2 = Popen('ps -p %d -o osz' % pid, shell=True,
@@ -1247,7 +1247,12 @@ def report_memory(i=0):  # argument may go away
         a2 = Popen('ps -p %d -o rss,vsz' % pid, shell=True,
             stdout=PIPE).stdout.readlines()
         mem = int(a2[1].split()[0])
-
+    elif sys.platform.startswith('win'):
+        a2 = check_output(["tasklist", "/nh", "/fi", "pid eq %d" % pid])
+        mem = int(a2.strip().split()[-2].replace(',',''))
+    else:
+        raise NotImplementedError(
+                "We don't have a memory monitor for %s" % sys.platform)
     return mem
 
 _safezip_msg = 'In safezip, len(args[0])=%d but len(args[%d])=%d'
