| /* |
| * 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 <algorithm> |
| #include <iomanip> |
| #include <pass.h> |
| #include <support/colors.h> |
| #include <wasm.h> |
| |
| namespace wasm { |
| |
| using namespace std; |
| |
| // Prints metrics between optimization passes. |
| struct Metrics : public WalkerPass<PostWalker<Metrics, UnifiedExpressionVisitor<Metrics>>> { |
| static Metrics *lastMetricsPass; |
| |
| map<const char *, int> counts; |
| |
| void visitExpression(Expression* curr) { |
| auto name = getExpressionName(curr); |
| counts[name]++; |
| } |
| |
| void visitModule(Module* module) { |
| ostream &o = cout; |
| o << "Counts" |
| << "\n"; |
| vector<const char*> keys; |
| int total = 0; |
| for (auto i : counts) { |
| keys.push_back(i.first); |
| total += i.second; |
| } |
| // add total |
| keys.push_back("[total]"); |
| counts["[total]"] = total; |
| // add vars |
| size_t vars = 0; |
| for (auto& func : module->functions) { |
| vars += func->getNumVars(); |
| } |
| keys.push_back("[vars]"); |
| counts["[vars]"] = vars; |
| // add functions |
| keys.push_back("[funcs]"); |
| counts["[funcs]"] = module->functions.size(); |
| // add memory and table |
| if (module->memory.exists) { |
| Index size = 0; |
| for (auto& segment: module->memory.segments) { |
| size += segment.data.size(); |
| } |
| keys.push_back("[memory-data]"); |
| counts["[memory-data]"] = size; |
| } |
| if (module->table.exists) { |
| Index size = 0; |
| for (auto& segment: module->table.segments) { |
| size += segment.data.size(); |
| } |
| keys.push_back("[table-data]"); |
| counts["[table-data]"] = size; |
| } |
| // sort |
| sort(keys.begin(), keys.end(), [](const char* a, const char* b) -> bool { |
| return strcmp(b, a) > 0; |
| }); |
| for (auto* key : keys) { |
| auto value = counts[key]; |
| o << " " << left << setw(15) << key << ": " << setw(8) |
| << value; |
| if (lastMetricsPass) { |
| if (lastMetricsPass->counts.count(key)) { |
| int before = lastMetricsPass->counts[key]; |
| int after = value; |
| 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; |
| } |
| }; |
| |
| Pass *createMetricsPass() { |
| return new Metrics(); |
| } |
| |
| Metrics *Metrics::lastMetricsPass; |
| |
| } // namespace wasm |