add an option for not including stats with 0 counters (useful for when we're clearing stats)

src/core/stats.cpp

@@ -51,7 +51,7 @@ int Stats::getStatId(const std::string& name) {
     return rtn;
 }
 
-void Stats::dump() {
+void Stats::dump(bool includeZeros) {
     if (!Stats::enabled)
         return;
 
@@ -65,7 +65,8 @@ void Stats::dump() {
     std::sort(pairs.begin(), pairs.end());
 
     for (int i = 0; i < pairs.size(); i++) {
-        printf("%s: %ld\n", pairs[i].first.c_str(), (*counts)[pairs[i].second]);
+        if (includeZeros || (*counts)[pairs[i].second] > 0)
+            printf("%s: %ld\n", pairs[i].first.c_str(), (*counts)[pairs[i].second]);
     }
 }
 

src/core/stats.h

@@ -41,7 +41,7 @@ public:
     static void log(int id, int count = 1) { (*counts)[id] += count; }
 
     static void clear() { std::fill(counts->begin(), counts->end(), 0); }
-    static void dump();
+    static void dump(bool includeZeros = true);
     static void endOfInit();
 };
 

src/runtime/builtin_modules/pyston.cpp

@@ -53,8 +53,11 @@ static Box* clearStats() {
     return None;
 }
 
-static Box* dumpStats() {
-    Stats::dump();
+static Box* dumpStats(Box* includeZeros) {
+    if (includeZeros->cls != bool_cls)
+        raiseExcHelper(TypeError, "includeZeros must be a 'bool' object but received a '%s'",
+                       getTypeName(includeZeros));
+    Stats::dump(((BoxedBool*)includeZeros)->n != 0);
     return None;
 }
 
@@ -67,6 +70,7 @@ void setupPyston() {
     pyston_module->giveAttr("clearStats",
                             new BoxedBuiltinFunctionOrMethod(boxRTFunction((void*)clearStats, NONE, 0), "clearStats"));
     pyston_module->giveAttr("dumpStats",
-                            new BoxedBuiltinFunctionOrMethod(boxRTFunction((void*)dumpStats, NONE, 0), "dumpStats"));
+                            new BoxedBuiltinFunctionOrMethod(boxRTFunction((void*)dumpStats, NONE, 1, 1, false, false),
+                                                             "dumpStats", { False }));
 }
 }