Rev 186: Corbin Simpson added a custom hook for numpy arrays. in http://bazaar.launchpad.net/~meliae-dev/meliae/trunk

[John Arbash Meinel]

At http://bazaar.launchpad.net/~meliae-dev/meliae/trunk

------------------------------------------------------------
revno: 186 [merge]
revision-id: john at arbash-meinel.com-20110225164245-i098e8rw752xq84j
parent: john at arbash-meinel.com-20100810162258-ihwwv2vfctbh418q
parent: mostawesomedude at gmail.com-20110219011820-y67gghrn77gbd9h3
committer: John Arbash Meinel <john at arbash-meinel.com>
branch nick: trunk
timestamp: Fri 2011-02-25 10:42:45 -0600
message:
  Corbin Simpson added a custom hook for numpy arrays.
modified:
  meliae/scanner.py              scanner.py-20090402040945-idv8d14z8re05gvg-1
-------------- next part --------------
=== modified file 'meliae/scanner.py'
--- a/meliae/scanner.py	2010-08-10 16:20:09 +0000
+++ b/meliae/scanner.py	2011-02-25 16:42:45 +0000
@@ -1,4 +1,4 @@
-# Copyright (C) 2009, 2010 Canonical Ltd
+# Copyright (C) 2009, 2010, 2011 Canonical Ltd
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
@@ -27,6 +27,17 @@
 get_referents = _scanner.get_referents
 add_special_size = _scanner.add_special_size
 
+def _size_of_ndarray(ndarray_obj):
+    """
+    Return the size of a Numpy ndarray's internal storage.
+
+    Doesn't yet handle views into other arrays.
+    """
+
+    return ndarray_obj.nbytes
+
+add_special_size("numpy.ndarray", _size_of_ndarray, _size_of_ndarray)
+
 
 def dump_all_referenced(outf, obj, is_pending=False):
     """Recursively dump everything that is referenced from obj."""