summaryrefslogtreecommitdiff
path: root/src/passes/Metrics.cpp
blob: 3624786a49f3d1a70edb5643eeb6b8c1a4131b8b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/*
 * Copyright 2016 WebAssembly Community Group participants
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <iomanip>
#include <pass.h>
#include <wasm.h>

namespace wasm {

using namespace std;

// Prints metrics between optimization passes.
struct Metrics : public WalkerPass<WasmWalker<Metrics>> {
  static Metrics *lastMetricsPass;

  map<const char *, int> counts;
  void walk(Expression *&curr) override {
    WalkerPass::walk(curr);
    if (!curr)
      return;
    auto name = getExpressionName(curr);
    counts[name]++;
  }
  void finalize(PassRunner *runner, Module *module) override {
    ostream &o = cout;
    o << "Counts"
      << "\n";
    for (auto i = counts.cbegin(); i != counts.cend(); i++) {
      o << " " << left << setw(15) << i->first << ": " << setw(8)
        << i->second;
      if (lastMetricsPass) {
        if (lastMetricsPass->counts.count(i->first)) {
          int before = lastMetricsPass->counts[i->first];
          int after = i->second;
          if (after - before) {
            if (after > before) {
              Colors::red(o);
            } else {
              Colors::green(o);
            }
            o << right << setw(8);
            o << showpos << after - before << noshowpos;
            Colors::normal(o);
          }
        }
      }
      o << "\n";
    }
    lastMetricsPass = this;
  }
};

Metrics *Metrics::lastMetricsPass;
static RegisterPass<Metrics> registerPass("metrics", "reports metrics");

} // namespace wasm