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

修改标识符

Browse Source
This commit is contained in:
Luke 2025-05-07 20:05:20 +08:00
parent de21808c2c
commit d4c8aea05f
2 changed files with 46 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
src/main/java/org/jcnc/snow/compiler/backend/VMCodeGenerator.java
View File

@ -2,7 +2,10 @@ package org.jcnc.snow.compiler.backend;
import org.jcnc.snow.compiler.ir.core.IRFunction; import org.jcnc.snow.compiler.ir.core.IRFunction;
import org.jcnc.snow.compiler.ir.core.IRInstruction; import org.jcnc.snow.compiler.ir.core.IRInstruction;
import org.jcnc.snow.compiler.ir.instruction.*; import org.jcnc.snow.compiler.ir.instruction.BinaryOperationInstruction;
import org.jcnc.snow.compiler.ir.instruction.LoadConstInstruction;
import org.jcnc.snow.compiler.ir.instruction.ReturnInstruction;
import org.jcnc.snow.compiler.ir.instruction.UnaryOperationInstruction;
import org.jcnc.snow.compiler.ir.value.IRConstant; import org.jcnc.snow.compiler.ir.value.IRConstant;
import org.jcnc.snow.compiler.ir.value.IRVirtualRegister; import org.jcnc.snow.compiler.ir.value.IRVirtualRegister;
import org.jcnc.snow.vm.engine.VMOpCode; import org.jcnc.snow.vm.engine.VMOpCode;
@ -18,26 +21,28 @@ import java.util.Map;
*/ */
public final class VMCodeGenerator { public final class VMCodeGenerator {
private final Map<IRVirtualRegister,Integer> slotMap; private final Map<IRVirtualRegister, Integer> slotMap;
private final List<String> code = new ArrayList<>(); private final List<String> code = new ArrayList<>();
private String currentFnName; // ← 当前正在生成的函数名 private String currentFnName; // ← 当前正在生成的函数名
public VMCodeGenerator(Map<IRVirtualRegister,Integer> slotMap){ public VMCodeGenerator(Map<IRVirtualRegister, Integer> slotMap) {
this.slotMap = slotMap; this.slotMap = slotMap;
} }
/** 主入口：IR → 指令序列 */ /**
public List<String> generate(IRFunction fn){ * 主入口：IR → 指令序列
*/
public List<String> generate(IRFunction fn) {
currentFnName = fn.name(); // ← 保存函数名 currentFnName = fn.name(); // ← 保存函数名
for (IRInstruction inst : fn.body()) { for (IRInstruction inst : fn.body()) {
switch (inst) { switch (inst) {
case LoadConstInstruction c -> genLoadConst(c); case LoadConstInstruction c -> genLoadConst(c);
case BinaryOperationInstruction b -> genBinOp(b); case BinaryOperationInstruction b -> genBinOp(b);
case UnaryOperationInstruction u -> genUnary(u); case UnaryOperationInstruction u -> genUnary(u);
case ReturnInstruction r -> genRet(r); case ReturnInstruction r -> genRet(r);
default -> throw new IllegalStateException("Unsupported IR: "+inst); default -> throw new IllegalStateException("Unsupported IR: " + inst);
} }
} }
/* 注意：不再额外添加 HALT，结束行为由 genRet 决定 */ /* 注意：不再额外添加 HALT，结束行为由 genRet 决定 */
@ -46,43 +51,43 @@ public final class VMCodeGenerator {
/* ───────────── 指令生成 ───────────── */ /* ───────────── 指令生成 ───────────── */
private void genLoadConst(LoadConstInstruction c){ private void genLoadConst(LoadConstInstruction c) {
IRConstant k = (IRConstant)c.operands().get(0); IRConstant k = (IRConstant) c.operands().get(0);
emit(op("I_PUSH"), k.value().toString()); emit(op("I_PUSH"), k.value().toString());
emit(op("I_STORE"), slot(c.dest())+""); emit(op("I_STORE"), slot(c.dest()) + "");
} }
private void genBinOp(BinaryOperationInstruction b){ private void genBinOp(BinaryOperationInstruction b) {
emit(op("I_LOAD"), slot((IRVirtualRegister)b.operands().get(0))+""); emit(op("I_LOAD"), slot((IRVirtualRegister) b.operands().get(0)) + "");
emit(op("I_LOAD"), slot((IRVirtualRegister)b.operands().get(1))+""); emit(op("I_LOAD"), slot((IRVirtualRegister) b.operands().get(1)) + "");
String opcode = switch (b.op()){ String opcode = switch (b.op()) {
case ADD_I32 -> "I_ADD"; case ADD_I32 -> "I_ADD";
case SUB_I32 -> "I_SUB"; case SUB_I32 -> "I_SUB";
case MUL_I32 -> "I_MUL"; case MUL_I32 -> "I_MUL";
case DIV_I32 -> "I_DIV"; case DIV_I32 -> "I_DIV";
default -> throw new IllegalStateException("Unsupported bin op "+b.op()); default -> throw new IllegalStateException("Unsupported bin op " + b.op());
}; };
emit(op(opcode)); emit(op(opcode));
emit(op("I_STORE"), slot(b.dest())+""); emit(op("I_STORE"), slot(b.dest()) + "");
} }
private void genUnary(UnaryOperationInstruction u){ private void genUnary(UnaryOperationInstruction u) {
emit(op("I_LOAD"), slot((IRVirtualRegister)u.operands().get(0))+""); emit(op("I_LOAD"), slot((IRVirtualRegister) u.operands().get(0)) + "");
String opcode = switch (u.op()){ String opcode = switch (u.op()) {
case NEG_I32 -> "I_NEG"; case NEG_I32 -> "I_NEG";
default -> throw new IllegalStateException("Unsupported unary op "+u.op()); default -> throw new IllegalStateException("Unsupported unary op " + u.op());
}; };
emit(op(opcode)); emit(op(opcode));
emit(op("I_STORE"), slot(u.dest())+""); emit(op("I_STORE"), slot(u.dest()) + "");
} }
private void genRet(ReturnInstruction r){ private void genRet(ReturnInstruction r) {
/* — 主函数：加载返回值(若有) → HALT — */ /* — 主函数：加载返回值(若有) → HALT — */
if ("main".equals(currentFnName)) { if ("main".equals(currentFnName)) {
if (r.value() != null) { if (r.value() != null) {
emit(op("I_LOAD"), slot(r.value())+""); emit(op("I_LOAD"), slot(r.value()) + "");
} }
emit(op("HALT")); // 结束整个程序 emit(op("HALT")); // 结束整个程序
return; return;
@ -90,29 +95,29 @@ public final class VMCodeGenerator {
/* — 普通函数：LOAD (可选) → RET — */ /* — 普通函数：LOAD (可选) → RET — */
if (r.value() != null) { if (r.value() != null) {
emit(op("I_LOAD"), slot(r.value())+""); emit(op("I_LOAD"), slot(r.value()) + "");
} }
emit(op("RET")); emit(op("RET"));
} }
/* ───────────── 工具函数 ───────────── */ /* ───────────── 工具函数 ───────────── */
private int slot(IRVirtualRegister r){ private int slot(IRVirtualRegister r) {
Integer s = slotMap.get(r); Integer s = slotMap.get(r);
if (s == null) throw new IllegalStateException("Register "+r+" 未映射槽位"); if (s == null) throw new IllegalStateException("Register " + r + " 未映射槽位");
return s; return s;
} }
private String op(String name){ private String op(String name) {
try { try {
Field f = VMOpCode.class.getField(name); Field f = VMOpCode.class.getField(name);
return f.get(null).toString(); return f.get(null).toString();
} catch (Exception e){ } catch (Exception e) {
throw new RuntimeException("Unknown opcode "+name, e); throw new RuntimeException("Unknown opcode " + name, e);
} }
} }
private void emit(String opcode, String... args){ private void emit(String opcode, String... args) {
StringBuilder sb = new StringBuilder(opcode); StringBuilder sb = new StringBuilder(opcode);
for (String a : args) sb.append(' ').append(a); for (String a : args) sb.append(' ').append(a);
code.add(sb.toString()); code.add(sb.toString());

10
src/main/java/org/jcnc/snow/compiler/ir/instruction/LoadConstInstruction.java
View File

@ -14,10 +14,14 @@ import java.util.List;
* 将字面量常量 k 加载到目标虚拟寄存器 dest 中，以便后续指令使用该常量值。 * 将字面量常量 k 加载到目标虚拟寄存器 dest 中，以便后续指令使用该常量值。
*/ */
public final class LoadConstInstruction extends IRInstruction { public final class LoadConstInstruction extends IRInstruction {
/** 要加载的常量值 */ /**
* 要加载的常量值
*/
private final IRConstant k; private final IRConstant k;
/** 存放常量的目标虚拟寄存器 */ /**
* 存放常量的目标虚拟寄存器
*/
private final IRVirtualRegister dest; private final IRVirtualRegister dest;
/** /**
@ -28,7 +32,7 @@ public final class LoadConstInstruction extends IRInstruction {
*/ */
public LoadConstInstruction(IRVirtualRegister dest, IRConstant k) { public LoadConstInstruction(IRVirtualRegister dest, IRConstant k) {
this.dest = dest; this.dest = dest;
this.k = k; this.k = k;
} }
/** /**