blob: 9c2797819bb56ef6b8cd8427a3a64406a8efc7c9 (
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
/* Copyright (c) 2012-2017 The ANTLR Project. All rights reserved.
* Use of this file is governed by the BSD 3-clause license that
* can be found in the LICENSE.txt file in the project root.
*/
#include "atn/SingletonPredictionContext.h"
#include "support/Arrays.h"
#include "atn/ArrayPredictionContext.h"
using namespace antlr4::atn;
ArrayPredictionContext::ArrayPredictionContext(
Ref<SingletonPredictionContext> const& a)
: ArrayPredictionContext({a->parent}, {a->returnState}) {}
ArrayPredictionContext::ArrayPredictionContext(
std::vector<Ref<PredictionContext>> const& parents_,
std::vector<size_t> const& returnStates)
: PredictionContext(calculateHashCode(parents_, returnStates)),
parents(parents_),
returnStates(returnStates) {
assert(parents.size() > 0);
assert(returnStates.size() > 0);
}
ArrayPredictionContext::~ArrayPredictionContext() {}
bool ArrayPredictionContext::isEmpty() const {
// Since EMPTY_RETURN_STATE can only appear in the last position, we don't
// need to verify that size == 1.
return returnStates[0] == EMPTY_RETURN_STATE;
}
size_t ArrayPredictionContext::size() const { return returnStates.size(); }
Ref<PredictionContext> ArrayPredictionContext::getParent(size_t index) const {
return parents[index];
}
size_t ArrayPredictionContext::getReturnState(size_t index) const {
return returnStates[index];
}
bool ArrayPredictionContext::operator==(PredictionContext const& o) const {
if (this == &o) {
return true;
}
const ArrayPredictionContext* other =
dynamic_cast<const ArrayPredictionContext*>(&o);
if (other == nullptr || hashCode() != other->hashCode()) {
return false; // can't be same if hash is different
}
return antlrcpp::Arrays::equals(returnStates, other->returnStates) &&
antlrcpp::Arrays::equals(parents, other->parents);
}
std::string ArrayPredictionContext::toString() const {
if (isEmpty()) {
return "[]";
}
std::stringstream ss;
ss << "[";
for (size_t i = 0; i < returnStates.size(); i++) {
if (i > 0) {
ss << ", ";
}
if (returnStates[i] == EMPTY_RETURN_STATE) {
ss << "$";
continue;
}
ss << returnStates[i];
if (parents[i] != nullptr) {
ss << " " << parents[i]->toString();
} else {
ss << "nul";
}
}
ss << "]";
return ss.str();
}
|