Darren Ranalli преди 18 години
родител
ревизия
6ee6a27100
променени са 1 файла, в които са добавени 8 реда и са изтрити 0 реда
  1. 8 0
      direct/src/showbase/LeakDetectors.py

+ 8 - 0
direct/src/showbase/LeakDetectors.py

@@ -1,5 +1,6 @@
 # objects that report different types of leaks to the ContainerLeakDetector
 # objects that report different types of leaks to the ContainerLeakDetector
 
 
+from pandac.PandaModules import *
 from direct.showbase.PythonUtil import gcDebugOn
 from direct.showbase.PythonUtil import gcDebugOn
 import __builtin__, gc
 import __builtin__, gc
 
 
@@ -52,3 +53,10 @@ class SceneGraphLeakDetector(LeakDetector):
     def _leakNode(self, task=None):
     def _leakNode(self, task=None):
         self._render.attachNewNode('leakNode-%s' % serialNum())
         self._render.attachNewNode('leakNode-%s' % serialNum())
         taskMgr.doMethodLater(10, self._leakNode, self._leakTaskName)
         taskMgr.doMethodLater(10, self._leakNode, self._leakTaskName)
+
+class CppMemoryUsage(LeakDetector):
+    def __len__(self):
+        if config.GetBool('track-memory-usage', 0):
+            return int(MemoryUsage.getCppSize())
+        else:
+            return 0