summaryrefslogtreecommitdiff
path: root/src/passes/NameManager.cpp
blob: 13a6dd9f901642719dcaa3701a5c4ac1ab249890 (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
//
// NameManager
//

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

namespace wasm {

Name NameManager::getUnique(std::string prefix) {
  while (1) {
    Name curr = cashew::IString((prefix + std::to_string(counter++)).c_str(), false);
    if (names.find(curr) == names.end()) {
      names.insert(curr);
      return curr;
    }
  }
}

void NameManager::visitBlock(Block* curr) {
  names.insert(curr->name);
}
void NameManager::visitLoop(Loop* curr) {
  names.insert(curr->out);
  names.insert(curr->in);
}
void NameManager::visitLabel(Label* curr) {
  names.insert(curr->name);
}
void NameManager::visitBreak(Break* curr) {
  names.insert(curr->name);
}
void NameManager::visitSwitch(Switch* curr) {
  names.insert(curr->name);
  names.insert(curr->default_);
  for (auto& target : curr->targets) {
    names.insert(target);
  }
}
void NameManager::visitCall(Call* curr) {
  names.insert(curr->target);
}
void NameManager::visitCallImport(CallImport* curr) {
  names.insert(curr->target);
}
void NameManager::visitFunctionType(FunctionType* curr) {
  names.insert(curr->name);
}
void NameManager::visitFunction(Function* curr) {
  names.insert(curr->name);
  for (auto& param : curr->params) {
    names.insert(param.name);
  }
  for (auto& local : curr->locals) {
    names.insert(local.name);
  }
}
void NameManager::visitImport(Import* curr) {
  names.insert(curr->name);
}
void NameManager::visitExport(Export* curr) {
  names.insert(curr->name);
}

static RegisterPass<NameManager> registerPass;

} // namespace wasm