JCNC/SCompiler
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat:ast

Browse Source
This commit is contained in:
Luke 2025-04-22 17:37:44 +08:00
parent 98e9f6dd31
commit cc92ff577f
8 changed files with 57 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
src/main/java/org/jcnc/snow/compiler/parser/ASTPrinter.java
View File

@ -30,53 +30,59 @@ public class ASTPrinter {
private static void printStatement(ASTStatement stmt, int level) {
String indent = INDENT.repeat(level);
if (stmt instanceof ASTReturn) {
System.out.print(indent + "Return: ");
printExpression(((ASTReturn) stmt).getExpression());
System.out.println();
} else if (stmt instanceof ASTDeclare) {
ASTDeclare d = (ASTDeclare) stmt;
System.out.print(indent + "Declare " + d.getName() + ": " + d.getType());
if (d.getInitExpression() != null) {
System.out.print(" = "); printExpression(d.getInitExpression());
switch (stmt) {
case ASTReturn astReturn -> {
System.out.print(indent + "Return: ");
printExpression(astReturn.getExpression());
System.out.println();
}
System.out.println();
} else if (stmt instanceof ASTLoop) {
ASTLoop loop = (ASTLoop) stmt;
System.out.println(indent + "Loop:");
System.out.print(indent + " Init: "); printStatement(loop.getInit(), 0);
System.out.print(indent + " Condition: "); printExpression(loop.getCondition()); System.out.println();
System.out.print(indent + " Update: "); printStatement(loop.getUpdate(), 0);
System.out.println(indent + " Body:");
for (ASTStatement s : loop.getBody()) {
printStatement(s, level + 2);
case ASTDeclare d -> {
System.out.print(indent + "Declare " + d.getName() + ": " + d.getType());
if (d.getInitExpression() != null) {
System.out.print(" = ");
printExpression(d.getInitExpression());
}
System.out.println();
}
} else if (stmt instanceof ASTIf) {
ASTIf ifs = (ASTIf) stmt;
System.out.print(indent + "If condition: "); printExpression(ifs.getCondition()); System.out.println();
System.out.println(indent + " Then:");
for (ASTStatement s : ifs.getThenStatements()) {
printStatement(s, level + 2);
case ASTLoop loop -> {
System.out.println(indent + "Loop:");
System.out.print(indent + " Init: ");
printStatement(loop.getInit(), 0);
System.out.print(indent + " Condition: ");
printExpression(loop.getCondition());
System.out.println();
System.out.print(indent + " Update: ");
printStatement(loop.getUpdate(), 0);
System.out.println(indent + " Body:");
for (ASTStatement s : loop.getBody()) {
printStatement(s, level + 2);
}
}
} else {
System.out.println(indent + stmt.getClass().getSimpleName());
case ASTIf ifs -> {
System.out.print(indent + "If condition: ");
printExpression(ifs.getCondition());
System.out.println();
System.out.println(indent + " Then:");
for (ASTStatement s : ifs.getThenStatements()) {
printStatement(s, level + 2);
}
}
default -> System.out.println(indent + stmt.getClass().getSimpleName());
}
}
private static void printExpression(ASTExpression expr) {
if (expr instanceof ASTLiteral) {
System.out.print(((ASTLiteral) expr).getValue());
} else if (expr instanceof ASTIdentifier) {
System.out.print(((ASTIdentifier) expr).getName());
} else if (expr instanceof ASTBinaryOp) {
ASTBinaryOp bin = (ASTBinaryOp) expr;
System.out.print("(");
printExpression(bin.getLeft());
System.out.print(" " + bin.getOperator() + " ");
printExpression(bin.getRight());
System.out.print(")");
} else {
System.out.print(expr.getClass().getSimpleName());
switch (expr) {
case ASTLiteral astLiteral -> System.out.print(astLiteral.getValue());
case ASTIdentifier astIdentifier -> System.out.print(astIdentifier.getName());
case ASTBinaryOp bin -> {
System.out.print("(");
printExpression(bin.getLeft());
System.out.print(" " + bin.getOperator() + " ");
printExpression(bin.getRight());
System.out.print(")");
}
default -> System.out.print(expr.getClass().getSimpleName());
}
}
}

5
src/main/java/org/jcnc/snow/compiler/parser/ASTStatement.java Normal file
View File

@ -0,0 +1,5 @@
package org.jcnc.snow.compiler.parser;
import org.jcnc.snow.compiler.parser.ast.ASTNode;
public abstract class ASTStatement extends ASTNode {}

1
src/main/java/org/jcnc/snow/compiler/parser/ast/ASTDeclare.java
View File

@ -1,6 +1,7 @@
package org.jcnc.snow.compiler.parser.ast;
import org.jcnc.snow.compiler.parser.ASTExpression;
import org.jcnc.snow.compiler.parser.ASTStatement;
public class ASTDeclare extends ASTStatement {
final String name, type;

2
src/main/java/org/jcnc/snow/compiler/parser/ast/ASTFunction.java
View File

@ -1,5 +1,7 @@
package org.jcnc.snow.compiler.parser.ast;
import org.jcnc.snow.compiler.parser.ASTStatement;
import java.util.ArrayList;
import java.util.List;

1
src/main/java/org/jcnc/snow/compiler/parser/ast/ASTIf.java
View File

@ -1,6 +1,7 @@
package org.jcnc.snow.compiler.parser.ast;
import org.jcnc.snow.compiler.parser.ASTExpression;
import org.jcnc.snow.compiler.parser.ASTStatement;
import java.util.List;

1
src/main/java/org/jcnc/snow/compiler/parser/ast/ASTLoop.java
View File

@ -1,6 +1,7 @@
package org.jcnc.snow.compiler.parser.ast;
import org.jcnc.snow.compiler.parser.ASTExpression;
import org.jcnc.snow.compiler.parser.ASTStatement;
import java.util.List;

1
src/main/java/org/jcnc/snow/compiler/parser/ast/ASTReturn.java
View File

@ -1,6 +1,7 @@
package org.jcnc.snow.compiler.parser.ast;
import org.jcnc.snow.compiler.parser.ASTExpression;
import org.jcnc.snow.compiler.parser.ASTStatement;
public class ASTReturn extends ASTStatement {
final ASTExpression expr;

3
src/main/java/org/jcnc/snow/compiler/parser/ast/ASTStatement.java
View File

@ -1,3 +0,0 @@
package org.jcnc.snow.compiler.parser.ast;
public abstract class ASTStatement extends ASTNode {}