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
|
package ast.nodes;
import ast.types.*;
import java.util.ArrayList;
import semanticanalysis.SemanticError;
import semanticanalysis.SymbolTable;
/**
* Node for the `testlist_comp` statement of the grammar.
*/
public class TestlistCompNode implements Node {
private final ArrayList<Node> exprs;
private final CompForNode comp;
public TestlistCompNode(ArrayList<Node> exprs, Node comp) {
this.exprs = exprs;
this.comp = (CompForNode) comp;
}
@Override
public ArrayList<SemanticError> checkSemantics(SymbolTable ST, int _nesting, FunctionType ft) {
ArrayList<SemanticError> errors = new ArrayList<>();
if (comp != null) {
// If comp is set, then we save the atom in the ST (we assume the first expr is
// an atom). We ignore the `comp.checkSemantics()`.
String id = ((ExprNode) exprs.get(0)).getId();
Type t = ((ExprNode) exprs.get(0)).typeCheck();
ST.insert(id, t, _nesting, "");
} else {
// If comp is not set, then exprs is a list of 1 or more element
for (var param : exprs) {
var exp = (ExprNode) param;
if (exp.getId() != null && !exp.isFunctionCall()) {
ST.insert(exp.getId(), exp.typeCheck(), _nesting, "");
}
errors.addAll(param.checkSemantics(ST, _nesting, ft));
}
}
return errors;
}
public int getSize() {
return exprs.size();
}
/**
* Returns the i-th expressions of `exprs` field. If the index is greater or
* equals than the size return `null`.
*/
public Node getElem(int i) {
if (i >= this.exprs.size()) {
return null;
}
return exprs.get(i);
}
@Override
public Type typeCheck() {
return new VoidType();
}
@Override
public String codeGeneration() {
String str = "";
for (var param : exprs) {
str += param.codeGeneration();
}
return str;
}
@Override
public String printAST(String prefix) {
String str = prefix + "Testlist_comp\n";
prefix += " ";
for (var param : exprs) {
str += param.printAST(prefix);
}
return str;
}
@Override
public String toPrint(String prefix) {
String str = prefix;
str += exprs.get(0).toPrint("");
if (comp != null) {
str += comp.toPrint("");
} else {
for (int i = 1; i < exprs.size(); i++) {
str += ", " + exprs.get(i).toPrint("");
}
}
return str;
}
}
|
