diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/ast_utils.h | 6 | ||||
-rw-r--r-- | src/passes/CMakeLists.txt | 1 | ||||
-rw-r--r-- | src/passes/NameList.cpp | 38 |
3 files changed, 42 insertions, 3 deletions
diff --git a/src/ast_utils.h b/src/ast_utils.h index ee5c76b69..600abd1b6 100644 --- a/src/ast_utils.h +++ b/src/ast_utils.h @@ -169,16 +169,16 @@ struct EffectAnalyzer : public PostWalker<EffectAnalyzer, Visitor<EffectAnalyzer // Meausure the size of an AST struct Measurer : public PostWalker<Measurer, UnifiedExpressionVisitor<Measurer>> { - size_t size = 0; + Index size = 0; void visitExpression(Expression* curr) { size++; } - static bool measure(Expression* tree) { + static Index measure(Expression* tree) { Measurer measurer; measurer.walk(tree); - return !!measurer.size; + return measurer.size; } }; diff --git a/src/passes/CMakeLists.txt b/src/passes/CMakeLists.txt index ab55dff85..88d513407 100644 --- a/src/passes/CMakeLists.txt +++ b/src/passes/CMakeLists.txt @@ -8,6 +8,7 @@ SET(passes_SOURCES MergeBlocks.cpp Metrics.cpp NameManager.cpp + NameList.cpp OptimizeInstructions.cpp PostEmscripten.cpp Print.cpp diff --git a/src/passes/NameList.cpp b/src/passes/NameList.cpp new file mode 100644 index 000000000..ed0d1835a --- /dev/null +++ b/src/passes/NameList.cpp @@ -0,0 +1,38 @@ +/* + * 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. + */ + +// +// Write out the name list of the module, similar to `nm`. +// + +#include "wasm.h" +#include "pass.h" +#include "ast_utils.h" + +namespace wasm { + +struct NameList : public Pass { + void run(PassRunner* runner, Module* module) override { + for (auto& func : module->functions) { + std::cout << " " << func->name << " : " << Measurer::measure(func->body) << '\n'; + } + } +}; + +static RegisterPass<NameList> registerPass("nm", "name list"); + +} // namespace wasm + |