1 | // pgen2.cc
|
2 |
|
3 | #include "pgen2.h"
|
4 |
|
5 | #include <vector>
|
6 |
|
7 | namespace pnode {
|
8 |
|
9 | PNode::PNode(int typ, syntax_asdl::Token* tok, List<PNode*>*)
|
10 | : typ(typ), tok(tok), children() {
|
11 | }
|
12 |
|
13 | void PNode::AddChild(PNode* node) {
|
14 | children.push_back(node);
|
15 | }
|
16 |
|
17 | PNode* PNode::GetChild(int i) {
|
18 | int j = i;
|
19 | if (j < 0) {
|
20 | j += NumChildren();
|
21 | }
|
22 | return children[j];
|
23 | }
|
24 |
|
25 | int PNode::NumChildren() {
|
26 | return children.size();
|
27 | }
|
28 |
|
29 | PNodeAllocator::PNodeAllocator() : arena_(new std::vector<PNode>()) {
|
30 | arena_->reserve(512);
|
31 | }
|
32 |
|
33 | PNode* PNodeAllocator::NewPNode(int typ, syntax_asdl::Token* tok) {
|
34 | CHECK(arena_->size() < arena_->capacity());
|
35 | arena_->emplace_back(typ, tok, nullptr);
|
36 | return arena_->data() + (arena_->size() - 1);
|
37 | }
|
38 |
|
39 | void PNodeAllocator::Clear() {
|
40 | delete arena_;
|
41 | arena_ = nullptr;
|
42 | }
|
43 |
|
44 | } // namespace pnode
|