print out total nodes in --metrics

1 files changed, 4 insertions, 2 deletions

diff --git a/src/passes/Metrics.cpp b/src/passes/Metrics.cpp
index 03f8bc447..9673b8873 100644
--- a/src/passes/Metrics.cpp
+++ b/src/passes/Metrics.cpp
@@ -30,8 +30,7 @@ struct Metrics : public WalkerPass<WasmWalker<Metrics>> {
   map<const char *, int> counts;
   void walk(Expression *&curr) override {
     WalkerPass::walk(curr);
-    if (!curr)
-      return;
+    if (!curr) return;
     auto name = getExpressionName(curr);
     counts[name]++;
   }
@@ -40,8 +39,10 @@ struct Metrics : public WalkerPass<WasmWalker<Metrics>> {
     o << "Counts"
       << "\n";
     vector<const char*> keys;
+    int total = 0;
     for (auto i : counts) {
       keys.push_back(i.first);
+      total += i.second;
     }
     sort(keys.begin(), keys.end(), [](const char* a, const char* b) -> bool {
       return strcmp(b, a) > 0;
@@ -68,6 +69,7 @@ struct Metrics : public WalkerPass<WasmWalker<Metrics>> {
       }
       o << "\n";
     }
+    o << left << setw(16) << "Total" << ": " << setw(8) << total << '\n';
     lastMetricsPass = this;
   }
 };