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

优化ir

Browse Source
This commit is contained in:
Luke 2025-05-06 22:15:10 +08:00
parent e6bf9ecd12
commit de2190c211
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/main/java/org/jcnc/snow/compiler/ir/builder/ProgramBuilder.java
View File

@ -31,17 +31,17 @@ public final class ProgramBuilder {
public IRProgram buildProgram(List<Node> roots) { public IRProgram buildProgram(List<Node> roots) {
IRProgram prog = new IRProgram(); IRProgram prog = new IRProgram();
for (Node n : roots) { for (Node n : roots) {
if (n instanceof ModuleNode m) { switch (n) {
// 模块节点：将模块内的所有函数加入程序 case ModuleNode m ->
m.functions().forEach(f -> prog.add(buildFunction(f))); // 模块节点：将模块内的所有函数加入程序
} else if (n instanceof FunctionNode f) { m.functions().forEach(f -> prog.add(buildFunction(f)));
// 顶层函数节点 case FunctionNode f ->
prog.add(buildFunction(f)); // 顶层函数节点
} else if (n instanceof StatementNode s) { prog.add(buildFunction(f));
// 顶层脚本语句，包装为 _start 函数 case StatementNode s ->
prog.add(buildFunction(wrapTopLevel(s))); // 顶层脚本语句，包装为 _start 函数
} else { prog.add(buildFunction(wrapTopLevel(s)));
throw new IllegalStateException("Unsupported top-level node: " + n); case null, default -> throw new IllegalStateException("Unsupported top-level node: " + n);
} }
} }
return prog; return prog;