summaryrefslogtreecommitdiff
path: root/src/ast/nodes/CompForNode.java
blob: 405a400ecae6c64ff00bd648e18295618eee7a07 (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
package ast.nodes;

import ast.types.*;
import java.util.ArrayList;
import semanticanalysis.SemanticError;
import semanticanalysis.SymbolTable;

/**
 * Node for the `comp_for` statement of the grammar. 'for' exprlist 'in' expr
 * comp_iter?
 */
public class CompForNode implements Node {

    protected ExprListNode exprlist;
    protected ExprNode single_expr;
    protected CompIterNode comp_iter;

    public CompForNode(Node exprlist, Node single_expr, Node comp_iter) {
        this.exprlist = (ExprListNode) exprlist;
        this.single_expr = (ExprNode) single_expr;
        this.comp_iter = (CompIterNode) comp_iter;
    }

    @Override
    public ArrayList<SemanticError> checkSemantics(SymbolTable ST, int _nesting) {
        ArrayList<SemanticError> errors = new ArrayList<>();

        errors.addAll(exprlist.checkSemantics(ST, _nesting));
        errors.addAll(single_expr.checkSemantics(ST, _nesting));
        if (comp_iter != null) {
            errors.addAll(comp_iter.checkSemantics(ST, _nesting));
        }
        return errors;
    }

    @Override
    public Type typeCheck() {
        return new VoidType();
    }

    // TODO: add code generation for arglist node
    @Override
    public String codeGeneration() {
        return "";
    }

    @Override
    public String toPrint(String prefix) {
        String str = prefix + "CompForNode\n";

        prefix += "  ";
        str += exprlist.toPrint(prefix);
        str += single_expr.toPrint(prefix);
        str += comp_iter.toPrint(prefix);
        return str;
    }

}