From dc38b4e1a8fc78a8944ba631e2a671e6553703b3 Mon Sep 17 00:00:00 2001 From: Luke Date: Mon, 1 Sep 2025 17:07:09 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E4=BC=98=E5=8C=96=20CALL=20?= =?UTF-8?q?=E6=8C=87=E4=BB=A4=E4=BF=AE=E8=A1=A5=E9=80=BB=E8=BE=91=E5=B9=B6?= =?UTF-8?q?=E8=B0=83=E6=95=B4=E4=BB=A3=E7=A0=81=E7=BB=93=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 改进 patchCallFixes 方法，支持更灵活的 super 调用匹配 - 重构代码，优化记录结构和方法注释 - 增加 lastSegment 方法的注释说明 - 调整类和方法的组织结构，提高可读性 --- .../backend/builder/VMProgramBuilder.java | 119 +++++++++++------- 1 file changed, 73 insertions(+), 46 deletions(-) diff --git a/src/main/java/org/jcnc/snow/compiler/backend/builder/VMProgramBuilder.java b/src/main/java/org/jcnc/snow/compiler/backend/builder/VMProgramBuilder.java index d438366..f4e6c54 100644 --- a/src/main/java/org/jcnc/snow/compiler/backend/builder/VMProgramBuilder.java +++ b/src/main/java/org/jcnc/snow/compiler/backend/builder/VMProgramBuilder.java @@ -23,39 +23,44 @@ import java.util.*; public final class VMProgramBuilder { /** - * 未知目标的 CALL 指令修补记录(待目标地址确定后修正)。 - * @param index CALL 指令在 code 列表中的位置 - * @param target 目标函数的全名 - * @param nArgs 参数个数 + * 未解析地址的占位符，便于后期批量修补 */ - private record CallFix(int index, String target, int nArgs) {} - - /** - * 未知目标的分支指令修补记录(待目标标签确定后修正)。 - * @param index 分支指令在 code 列表中的位置 - * @param label 跳转目标标签名 - */ - private record BranchFix(int index, String label) {} - - /** 未解析地址的占位符，便于后期批量修补 */ private static final String PLACEHOLDER = "-1"; - - /** VM 指令列表 */ + /** + * VM 指令列表 + */ private final List code = new ArrayList<>(); - /** 槽位(寄存器)类型映射表(如 I/F...，用于类型检查或代码生成优化) */ + /** + * 槽位(寄存器)类型映射表(如 I/F...，用于类型检查或代码生成优化) + */ private final Map slotType = new HashMap<>(); - /** 符号(函数名/标签)到指令序号的映射表 */ + /** + * 符号(函数名/标签)到指令序号的映射表 + */ private final Map addr = new HashMap<>(); - /** 所有待修补的 CALL 指令集合 */ + /** + * 所有待修补的 CALL 指令集合 + */ private final List callFixes = new ArrayList<>(); - /** 所有待修补的分支指令集合 */ + /** + * 所有待修补的分支指令集合 + */ private final List branchFixes = new ArrayList<>(); - /** 当前代码指针(已生成指令的数量/下一个指令的位置) */ + /** + * 当前代码指针(已生成指令的数量/下一个指令的位置) + */ private int pc = 0; - // ============================== - // 槽位类型操作 - // ============================== + /** + * 获取符号名最后一段（即最后一个 '.' 之后的名字，若无 '.' 则为原名）。 + * + * @param sym 符号全名 + * @return 简名 + */ + private static String lastSegment(String sym) { + int p = sym.lastIndexOf('.'); + return (p >= 0 && p < sym.length() - 1) ? sym.substring(p + 1) : sym; + } /** * 设置槽位(局部变量/虚拟寄存器)的类型前缀。 @@ -67,6 +72,8 @@ public final class VMProgramBuilder { slotType.put(slot, prefix); } + // ============================== 槽位类型操作 ============================== + /** * 获取槽位的类型前缀，默认为 'I'(整数类型)。 * @@ -77,10 +84,9 @@ public final class VMProgramBuilder { return slotType.getOrDefault(slot, 'I'); } - // =============函数/标签声明与指令生成================= - /** * 声明一个函数/标签的起始地址，并尝试修补所有引用到此符号的 CALL/BRANCH。 + * * @param name 函数或标签全名（如 "Person.getName"、"loop.start"） */ public void beginFunction(String name) { @@ -89,11 +95,17 @@ public final class VMProgramBuilder { patchBranchFixes(name); } - /** 函数结束接口，目前无具体实现，便于将来扩展。 */ - public void endFunction() {} + // =============函数/标签声明与指令生成================= + + /** + * 函数结束接口，目前无具体实现，便于将来扩展。 + */ + public void endFunction() { + } /** * 添加一条 VM 指令或标签（末尾':'视为标签）。 + * * @param line 指令或标签 */ public void emit(String line) { @@ -109,8 +121,9 @@ public final class VMProgramBuilder { /** * 添加一条 CALL 指令。目标尚未声明时，使用占位符并登记延迟修补。 - * @param target 目标函数全名（IR 侧生成） - * @param nArgs 实参个数 + * + * @param target 目标函数全名（IR 侧生成） + * @param nArgs 实参个数 */ public void emitCall(String target, int nArgs) { Integer a = resolve(target); @@ -166,15 +179,13 @@ public final class VMProgramBuilder { } /** - * 对所有待修补的 CALL 指令进行补丁。 + * 修补所有延迟 CALL。 * 匹配规则： - * 1. 全限定名完全匹配 - * 2. 简名(最后一段)匹配 - * 3. super 调用：target 以 .super 结尾、name 以 .__init__ 结尾 - * @param name 新声明的符号名 + * 1. 全限定名完全匹配 + * 2. super 调用支持多构造：target 以 .super 结尾，name 以 .__init__N 结尾且前缀匹配 + * 3. 简名匹配（含 __init__N） */ private void patchCallFixes(String name) { - // 当前函数的简名（去掉前缀） String nameSimple = lastSegment(name); for (Iterator it = callFixes.iterator(); it.hasNext(); ) { @@ -182,13 +193,20 @@ public final class VMProgramBuilder { // 全限定名完全匹配 boolean qualifiedMatch = f.target.equals(name); - // 简名匹配 + + // super 调用绑定：target 以 .super，name 以 .__init__N 且前缀匹配 + boolean superMatch = false; + if (f.target.endsWith(".super") && name.contains(".__init__")) { + String tStruct = f.target.substring(0, f.target.length() - 6); // 去掉 ".super" + String nStruct = name.substring(0, name.indexOf(".__init__")); + superMatch = tStruct.equals(nStruct); + } + + // 简名匹配（如 getName, __init__N 等） String targetSimple = lastSegment(f.target); boolean simpleMatch = targetSimple.equals(nameSimple); - // super 调用绑定 - boolean superMatch = f.target.endsWith(".super") && name.endsWith(".__init__"); - if (qualifiedMatch || simpleMatch || superMatch) { + if (qualifiedMatch || superMatch || simpleMatch) { int targetAddr = addr.get(name); code.set(f.index, VMOpCode.CALL + " " + targetAddr + " " + f.nArgs); it.remove(); @@ -213,12 +231,21 @@ public final class VMProgramBuilder { } /** - * 获取符号名最后一段（即最后一个 '.' 之后的名字，若无 '.' 则为原名）。 - * @param sym 符号全名 - * @return 简名 + * 未知目标的 CALL 指令修补记录(待目标地址确定后修正)。 + * + * @param index CALL 指令在 code 列表中的位置 + * @param target 目标函数的全名 + * @param nArgs 参数个数 */ - private static String lastSegment(String sym) { - int p = sym.lastIndexOf('.'); - return (p >= 0 && p < sym.length() - 1) ? sym.substring(p + 1) : sym; + private record CallFix(int index, String target, int nArgs) { + } + + /** + * 未知目标的分支指令修补记录(待目标标签确定后修正)。 + * + * @param index 分支指令在 code 列表中的位置 + * @param label 跳转目标标签名 + */ + private record BranchFix(int index, String label) { } }