Altered SkInstCnt to print out atexit

http://codereview.appspot.com/6348043/

git-svn-id: http://skia.googlecode.com/svn/trunk@4377 2bbb7eff-a529-9590-31e7-b0007b416f81

diff --git a/gm/gmmain.cpp b/gm/gmmain.cpp
index b34a442..8483dae 100644 (file)
--- a/gm/gmmain.cpp
+++ b/gm/gmmain.cpp
@@ -1091,10 +1091,5 @@ int main(int argc, char * const argv[]) {
     delete grFactory;
     SkGraphics::Term();
 
-#ifdef SK_DEBUG
-    SkRefCnt::CheckInstanceCount();
-    SkWStream::CheckInstanceCount();
-#endif
-
     return (0 == testsFailed) ? 0 : -1;
 }

diff --git a/include/core/SkInstCnt.h b/include/core/SkInstCnt.h
index 18ea9bb..9ec18d5 100644 (file)
--- a/include/core/SkInstCnt.h
+++ b/include/core/SkInstCnt.h
@@ -19,16 +19,20 @@
  * CheckInstanceCount methods should be made
  */
 #ifdef SK_ENABLE_INST_COUNT
+#include <stdlib.h>
 #include "SkTArray.h"
 
 extern bool gPrintInstCount;
 
+// The non-root classes just register themselves with their parent
 #define SK_DECLARE_INST_COUNT(className)                                    \
     SK_DECLARE_INST_COUNT_INTERNAL(className,                               \
                                 INHERITED::AddInstChild(CheckInstanceCount);)
 
+// The root classes registers a function to print out the memory stats when
+// the app ends
 #define SK_DECLARE_INST_COUNT_ROOT(className)                               \
-    SK_DECLARE_INST_COUNT_INTERNAL(className, ;)
+    SK_DECLARE_INST_COUNT_INTERNAL(className, atexit(exitPrint);)
 
 #define SK_DECLARE_INST_COUNT_INTERNAL(className, initStep)                 \
     class SkInstanceCountHelper {                                           \
@@ -59,6 +63,10 @@ extern bool gPrintInstCount;
         return SkInstanceCountHelper::gInstanceCount;                       \
     }                                                                       \
                                                                             \
+    static void exitPrint() {                                               \
+        CheckInstanceCount();                                               \
+    }                                                                       \
+                                                                            \
     static int CheckInstanceCount(int level = 0) {                          \
         if (gPrintInstCount && 0 != SkInstanceCountHelper::gInstanceCount) {\
             SkDebugf("%*c Leaked %s: %d\n",                                 \