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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#include <functional>
#include "wasm.h"
#include "mixed_arena.h"
namespace wasm {
class Pass;
//
// Global registry of all passes in /passes/
//
struct PassRegistry {
static PassRegistry* get();
typedef std::function<Pass* ()> Creator;
void registerPass(const char* name, const char *description, Creator create);
Pass* createPass(std::string name);
std::vector<std::string> getRegisteredNames();
std::string getPassDescription(std::string name);
private:
struct PassInfo {
std::string description;
Creator create;
PassInfo() {}
PassInfo(std::string description, Creator create) : description(description), create(create) {}
};
std::map<std::string, PassInfo> passInfos;
};
//
// Utility class to register a pass. See pass files for usage.
//
template<class P>
struct RegisterPass {
RegisterPass(const char* name, const char *description) {
PassRegistry::get()->registerPass(name, description, []() {
return new P();
});
}
};
//
// Runs a set of passes, in order
//
struct PassRunner {
MixedArena* allocator;
std::vector<Pass*> passes;
Pass* currPass;
PassRunner(MixedArena* allocator) : allocator(allocator) {}
void add(std::string passName);
template<class P>
void add();
void run(Module* module);
// Get the last pass that was already executed of a certain type.
template<class P>
P* getLast();
~PassRunner();
};
//
// Core pass class
//
struct Pass : public WasmWalker {
// Override this to perform preparation work before the pass runs
virtual void prepare(PassRunner* runner, Module *module) {}
void run(PassRunner* runner, Module *module) {
prepare(runner, module);
startWalk(module);
}
};
// Standard passes. All passes in /passes/ are runnable from the shell,
// but registering them here in addition allows them to communicate
// e.g. through PassRunner::getLast
// Handles names in a module, in particular adding names without duplicates
struct NameManager : public Pass {
Name getUnique(std::string prefix);
// TODO: getUniqueInFunction
// visitors
void visitBlock(Block* curr) override;
void visitLoop(Loop* curr) override;
void visitLabel(Label* curr) override;
void visitBreak(Break* curr) override;
void visitSwitch(Switch* curr) override;
void visitCall(Call* curr) override;
void visitCallImport(CallImport* curr) override;
void visitFunctionType(FunctionType* curr) override;
void visitFunction(Function* curr) override;
void visitImport(Import* curr) override;
void visitExport(Export* curr) override;
private:
std::set<Name> names;
size_t counter = 0;
};
} // namespace wasm
|
