58 lines
1.6 KiB
Java
Executable File
58 lines
1.6 KiB
Java
Executable File
package frontend.ast.block;
|
|
|
|
import frontend.ast.SyntaxType;
|
|
import frontend.ast.token.TokenNode;
|
|
import frontend.lexer.TokenStream;
|
|
import frontend.lexer.TokenType;
|
|
import frontend.ast.Node;
|
|
import frontend.ast.exp.Exp;
|
|
import frontend.ast.exp.LVal;
|
|
|
|
import java.util.ArrayList;
|
|
|
|
import error.Errors;
|
|
|
|
public class ForStmt extends Node {
|
|
public ForStmt(TokenStream ts) {
|
|
super(SyntaxType.FOR_STMT, ts);
|
|
}
|
|
|
|
public void parse(Errors errors) {
|
|
handleAssign(errors);
|
|
while (getCurrToken().getType() == TokenType.COMMA) {
|
|
addChild(new TokenNode(ts)); // comma
|
|
handleAssign(errors);
|
|
}
|
|
}
|
|
|
|
public void handleAssign(Errors errors) {
|
|
LVal lval = new LVal(this.ts);
|
|
lval.parse(errors);
|
|
addChild(lval);
|
|
addChild(new TokenNode(ts)); // assign
|
|
Exp exp = new Exp(this.ts);
|
|
exp.parse(errors);
|
|
addChild(exp);
|
|
}
|
|
|
|
public ArrayList<LVal> getLValList() {
|
|
ArrayList<LVal> lvalList = new ArrayList<>();
|
|
for (int i = 0; i < getChildren().size(); i++) {
|
|
if (getChild(i) instanceof LVal) {
|
|
lvalList.add((LVal) getChild(i));
|
|
}
|
|
}
|
|
return lvalList;
|
|
}
|
|
|
|
public ArrayList<Exp> getExpList() {
|
|
ArrayList<Exp> expList = new ArrayList<>();
|
|
for (int i = 0; i < getChildren().size(); i++) {
|
|
if (getChild(i) instanceof Exp) {
|
|
expList.add((Exp) getChild(i));
|
|
}
|
|
}
|
|
return expList;
|
|
}
|
|
}
|