i22006/Compilerbau2024AbsolutNichtHaskell
i22006/Compilerbau2024AbsolutNichtHaskell
2
0
Fork 0
mirror of https://github.com/JonathanFleischmann/CompilerULTIMATE.git synced 2024-12-28 10:08:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add missing visit functions

Browse Source
This commit is contained in:
Boolean-True 2024-05-01 10:49:06 +02:00
parent 12623821f7
commit 63246b815a
9 changed files with 132 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/main/java/de/maishai/ASTGenerator.java
View File

@ -33,7 +33,7 @@ public class ASTGenerator {
}
public static Variable generateVariable(DecafParser.VarContext ctx) {
return new Variable(ctx.id().getText(), getType(ctx.type()));
return new VariableGenerator().visitVar(ctx);
}
public static Parameter generateParameter(DecafParser.ParamContext ctx) {
@ -50,6 +50,11 @@ public class ASTGenerator {
return new Method(getReturnType(ctx.returntype()), methId, params, block);
}
public static MainMethod generateMainMethod(DecafParser.MainmethContext ctx) {
Block block = new BlockGenerator().visit(ctx.block());
return new MainMethod(block);
}
public static Type getType(DecafParser.TypeContext ctx){
if(ctx.INT() != null)
return Type.INT;

63
src/main/java/de/maishai/ExpressionGenerator.java
View File

@ -4,10 +4,8 @@ import de.maishai.antlr.DecafBaseVisitor;
import de.maishai.antlr.DecafParser;
import de.maishai.ast.Expression;
import de.maishai.ast.Operator;
import de.maishai.ast.records.Binary;
import de.maishai.ast.records.BoolConstant;
import de.maishai.ast.records.IntConstant;
import de.maishai.ast.records.MethodCall;
import de.maishai.ast.Type;
import de.maishai.ast.records.*;
import java.util.ArrayList;
import java.util.List;
@ -18,17 +16,6 @@ public class ExpressionGenerator extends DecafBaseVisitor<Expression> {
return generateBinary(ctx);
}
@Override
public Expression visitMethodCallExpression(DecafParser.MethodCallExpressionContext ctx) {
String functionName = ctx.methCall().id().getText();
List<Expression> args = new ArrayList<>();
for(var expr : ctx.methCall().args().expr()){
Expression astExpr = expr.accept(this);
args.add(astExpr);
}
return new MethodCall(functionName, args);
}
@Override
public Expression visitConstant(DecafParser.ConstantContext ctx) {
return generateConstant(ctx.literal());
@ -41,6 +28,22 @@ public class ExpressionGenerator extends DecafBaseVisitor<Expression> {
return this.visit(ctx.expr());
}
@Override
public Expression visitMethodCallExpression(DecafParser.MethodCallExpressionContext ctx) {
Id id = new Id(ctx.methCall().id().getText());
List<Expression> args = new ArrayList<>();
for(var expr : ctx.methCall().args().expr()){
Expression astExpr = expr.accept(this);
args.add(astExpr);
}
return new MethodCallExpression(id, args);
}
@Override
public Id visitIdentifier(DecafParser.IdentifierContext ctx){
return new Id(ctx.getText());
}
public static Expression generateConstant(DecafParser.LiteralContext ctx){
if(ctx.NUMBER() != null)
return new IntConstant(Integer.valueOf(ctx.NUMBER().getText()));
@ -62,4 +65,34 @@ public class ExpressionGenerator extends DecafBaseVisitor<Expression> {
if(ctx.MUL() != null)return Operator.MUL;
throw new RuntimeException();
}
//StatementExpression
@Override
public Expression visitAssign(DecafParser.AssignContext ctx) {
Id id = new Id(ctx.id().getText());
Expression expr = this.visit(ctx.expr());
return new Assignment(id, expr);
}
@Override
public Expression visitMethodCall(DecafParser.MethodCallContext ctx) {
Id id = new Id(ctx.methCall().id().getText());
List<Expression> args = new ArrayList<>();
for(var expr : ctx.methCall().args().expr()){
Expression astExpr = expr.accept(this);
args.add(astExpr);
}
return new MethodCall(id, args);
}
@Override
public Expression visitNew(DecafParser.NewContext ctx) {
Type type = ASTGenerator.getType(ctx.type());
List<Expression> args = new ArrayList<>();
for(var expr : ctx.args().expr()){
Expression astExpr = expr.accept(this);
args.add(astExpr);
}
return new New(type, args);
}
}

56
src/main/java/de/maishai/StatementGenerator.java
View File

@ -4,16 +4,13 @@ import de.maishai.antlr.DecafBaseVisitor;
import de.maishai.antlr.DecafParser;
import de.maishai.ast.Expression;
import de.maishai.ast.Statement;
import de.maishai.ast.Type;
import de.maishai.ast.records.*;
public class StatementGenerator extends DecafBaseVisitor<Statement> {
@Override
public Statement visitAssign(DecafParser.AssignContext ctx) {
Id id = new Id(ctx.id().getText());
Expression expr = new ExpressionGenerator().visit(ctx.expr());
return new Assignment(id, expr);
}
import java.util.ArrayList;
import java.util.List;
public class StatementGenerator extends DecafBaseVisitor<Statement> {
@Override
public Statement visitIf(DecafParser.IfContext ctx) {
Expression expr = new ExpressionGenerator().visit(ctx.expr());
@ -25,6 +22,15 @@ public class StatementGenerator extends DecafBaseVisitor<Statement> {
return new IfElse(expr, ifBlock, null);
}
@Override
public For visitFor(DecafParser.ForContext ctx) {
Expression init = new ExpressionGenerator().visit(ctx.expr(0));
Expression expr = new ExpressionGenerator().visit(ctx.expr(1));
Expression update = new ExpressionGenerator().visit(ctx.expr(2));
Block block = new BlockGenerator().visit(ctx.block());
return new For(init, expr, update, block);
}
@Override
public Statement visitWhile(DecafParser.WhileContext ctx) {
Expression expr = new ExpressionGenerator().visit(ctx.expr());
@ -32,6 +38,13 @@ public class StatementGenerator extends DecafBaseVisitor<Statement> {
return new While(expr, block);
}
@Override
public DoWhile visitDoWhile(DecafParser.DoWhileContext ctx) {
Block block = new BlockGenerator().visit(ctx.block());
Expression expr = new ExpressionGenerator().visit(ctx.expr());
return new DoWhile(block, expr);
}
@Override
public Statement visitReturn(DecafParser.ReturnContext ctx) {
return new Return(new ExpressionGenerator().visit(ctx.expr()));
@ -52,4 +65,33 @@ public class StatementGenerator extends DecafBaseVisitor<Statement> {
return new Continue();
}
//StatementExpression
@Override
public Statement visitAssign(DecafParser.AssignContext ctx) {
Id id = new Id(ctx.id().getText());
Expression expr = new ExpressionGenerator().visit(ctx.expr());
return new Assignment(id, expr);
}
@Override
public Statement visitMethodCall(DecafParser.MethodCallContext ctx) {
Id id = new Id(ctx.methCall().id().getText());
List<Expression> args = new ArrayList<>();
for(var expr : ctx.methCall().args().expr()){
Expression astExpr = expr.accept(new ExpressionGenerator());
args.add(astExpr);
}
return new MethodCall(id, args);
}
@Override
public Statement visitNew(DecafParser.NewContext ctx) {
Type type = ASTGenerator.getType(ctx.type());
List<Expression> args = new ArrayList<>();
for(var expr : ctx.args().expr()){
Expression astExpr = expr.accept(new ExpressionGenerator());
args.add(astExpr);
}
return new New(type, args);
}
}

17
src/main/java/de/maishai/VariableGenerator.java
View File

@ -4,22 +4,17 @@ package de.maishai;
import de.maishai.antlr.DecafBaseVisitor;
import de.maishai.antlr.DecafParser;
import de.maishai.ast.Type;
import de.maishai.ast.records.Id;
import de.maishai.ast.records.Variable;
public class VariableGenerator extends DecafBaseVisitor<Variable> {
@Override
public Variable visitVar(DecafParser.VarContext ctx) {
return new Variable(ctx.id().getText(), getType(ctx.type()));
}
public static Type getType(DecafParser.TypeContext ctx){
if(ctx.INT() != null)
return Type.INT;
if(ctx.BOOL() != null)
return Type.BOOL;
if(ctx.CHAR() != null)
return Type.CHAR;
throw new RuntimeException();
Id id = new Id(ctx.id().getText());
Type type = ASTGenerator.getType(ctx.type());
if(ctx.expr() != null)
return new Variable(id, type, new ExpressionGenerator().visit(ctx.expr()));
return new Variable(id, type, null);
}
}

8
src/main/java/de/maishai/ast/records/DoWhile.java Normal file
View File

@ -0,0 +1,8 @@
package de.maishai.ast.records;
import de.maishai.ast.Expression;
import de.maishai.ast.Statement;
public record DoWhile(Block block, Expression cond) implements Statement {
}

2
src/main/java/de/maishai/ast/records/MainMethod.java
View File

@ -7,5 +7,5 @@ import de.maishai.ast.ReturnType;
import java.util.List;
public record MainMethod(ReturnType type, Id id, List<Parameter> params, Block block) implements Node {
public record MainMethod(Block block) implements Node {
}

2
src/main/java/de/maishai/ast/records/MethodCall.java
View File

@ -7,5 +7,5 @@ import de.maishai.ast.Statement;
import java.util.List;
public record MethodCall(String name, List<Expression> args) implements Expression, Statement {
public record MethodCall(Id id, List<Expression> args) implements Expression, Statement {
}

11
src/main/java/de/maishai/ast/records/MethodCallExpression.java Normal file
View File

@ -0,0 +1,11 @@
package de.maishai.ast.records;
import de.maishai.ast.Expression;
import de.maishai.ast.Statement;
import java.util.List;
public record MethodCallExpression(Id id, List<Expression> args) implements Expression, Statement {
}

3
src/main/java/de/maishai/ast/records/Variable.java
View File

@ -1,8 +1,9 @@
package de.maishai.ast.records;
import de.maishai.ast.Expression;
import de.maishai.ast.Node;
import de.maishai.ast.Type;
public record Variable(String name, Type type) implements Node {
public record Variable(Id name, Type type, Expression expr) implements Node {
}