235 lines
7.6 KiB
Plaintext
235 lines
7.6 KiB
Plaintext
import t, c
|
||
from stdint import *
|
||
import string
|
||
import viperlib
|
||
import stdio
|
||
import memhub
|
||
from linkedlist import GSListNode
|
||
from .__types import LLVMType, TypePrint
|
||
|
||
|
||
# ============================================================
|
||
# Value: SSA 值表示
|
||
#
|
||
# 表示一个 LLVM IR 值：SSA 临时值（%0/%result）或常量字面量（42/0）。
|
||
# 每个值有类型指针、名字、是否常量标志。
|
||
# 继承 GSListNode[Value] 获取强类型 Next: Value|CPtr（值池链表）
|
||
#
|
||
# 注意: 必须加 @t.NoVTable（理由见 __function.py 的 Line 注释）。
|
||
# ============================================================
|
||
@t.NoVTable
|
||
class Value(GSListNode[Value]):
|
||
Ty: LLVMType | t.CPtr # 值的类型
|
||
Name: t.CChar | t.CPtr # SSA 名（"%0"/"%result"）或常量文本（"42"/"1.5e+00"）
|
||
IsConst: t.CInt # 1=常量字面量, 0=SSA 临时值
|
||
IntVal: t.CInt64T # 整数常量值（当 IsConst=1 时使用）
|
||
FloatVal: t.CDouble # 浮点常量值（当 IsConst=1 且 Ty 为 Float 时使用）
|
||
|
||
|
||
# ============================================================
|
||
# 工厂函数
|
||
# ============================================================
|
||
# Value 硬编码大小: Next(8) + Ty(8) + Name(8) + IsConst(4) + IntVal(8) + FloatVal(8) = 48 字节
|
||
VALUE_SIZE: t.CDefine = 48
|
||
|
||
def new_value(pool: memhub.MemBuddy | t.CPtr) -> Value | t.CPtr:
|
||
"""从 mpool 分配一个零初始化的 Value"""
|
||
ptr: Value | t.CPtr = pool.alloc(VALUE_SIZE)
|
||
if ptr:
|
||
string.memset(ptr, 0, VALUE_SIZE)
|
||
return ptr
|
||
|
||
|
||
def ConstInt(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr,
|
||
val: t.CInt64T, name: t.CChar | t.CPtr) -> Value | t.CPtr:
|
||
"""创建整数常量值
|
||
|
||
参数:
|
||
ty: 整数类型（Int1/Int8/Int32/Int64）
|
||
val: 整数值
|
||
name: 常量文本（如 "42"），由调用方提供
|
||
"""
|
||
v: Value | t.CPtr = new_value(pool)
|
||
if v is None: return None
|
||
v.Ty = ty
|
||
v.Name = name
|
||
v.IsConst = 1
|
||
v.IntVal = val
|
||
return v
|
||
|
||
|
||
def ConstNull(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr,
|
||
name: t.CChar | t.CPtr) -> Value | t.CPtr:
|
||
"""创建 null 指针常量"""
|
||
v: Value | t.CPtr = new_value(pool)
|
||
if v is None: return None
|
||
v.Ty = ty
|
||
v.Name = name
|
||
v.IsConst = 1
|
||
v.IntVal = 0
|
||
return v
|
||
|
||
|
||
def ConstZero(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr:
|
||
"""创建 zeroinitializer 常量（零初始化聚合类型：结构体/数组）"""
|
||
v: Value | t.CPtr = new_value(pool)
|
||
if v is None: return None
|
||
v.Ty = ty
|
||
v.Name = "zeroinitializer"
|
||
v.IsConst = 1
|
||
v.IntVal = 0
|
||
return v
|
||
|
||
|
||
def ConstFloat(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr,
|
||
val: t.CDouble) -> Value | t.CPtr:
|
||
"""创建浮点常量值
|
||
|
||
参数:
|
||
ty: 浮点类型（Half/Float/Double/FP128）
|
||
val: 浮点数值
|
||
名字自动格式化为 LLVM IR 科学计数法文本（如 "1.500000e+00"）。
|
||
"""
|
||
v: Value | t.CPtr = new_value(pool)
|
||
if v is None: return None
|
||
v.Ty = ty
|
||
v.IsConst = 1
|
||
v.FloatVal = val
|
||
v.IntVal = 0
|
||
# 格式化名字: LLVM IR 浮点常量文本用十六进制位模式（0x%016llx / 0x%08x）
|
||
# 原因: %.17e 科学计数法文本经 llc 解析对部分值产生 ±1 ulp 误差
|
||
# （如 "1.10000000000000000e+01" 解析为 11.000000000000002 而非 11.0，
|
||
# 导致 add_scalar[0]==11 等浮点常量比较失败）。位模式无精度损失。
|
||
# 原 %e 默认 6 位小数还会截断精度（3.1415926535897931 -> 3.141593e+00）。
|
||
# 用 CRT stdio.snprintf: viperlib 自实现 %g 有小数位反转 bug。
|
||
f_bits: t.CInt = 64
|
||
match ty:
|
||
case llvmlite.LLVMType.Float(ftb):
|
||
f_bits = ftb
|
||
fv_bits: t.CUInt64T = 0
|
||
c.Load(c.Addr(fv_bits), c.Addr(val))
|
||
name_buf: t.CChar | t.CPtr = pool.alloc(32)
|
||
if name_buf is not None:
|
||
if f_bits == 32:
|
||
stdio.snprintf(name_buf, 32, "0x%08x", t.CUInt32T(fv_bits))
|
||
else:
|
||
stdio.snprintf(name_buf, 32, "0x%016llx", fv_bits)
|
||
v.Name = name_buf
|
||
return v
|
||
|
||
|
||
def SSAValue(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr,
|
||
name: t.CChar | t.CPtr) -> Value | t.CPtr:
|
||
"""创建 SSA 临时值
|
||
|
||
参数:
|
||
ty: 值类型
|
||
name: SSA 名（如 "%0"/"%result"），由调用方提供
|
||
"""
|
||
v: Value | t.CPtr = new_value(pool)
|
||
if v is None: return None
|
||
v.Ty = ty
|
||
v.Name = name
|
||
v.IsConst = 0
|
||
v.IntVal = 0
|
||
return v
|
||
|
||
|
||
# ============================================================
|
||
# ValuePrint: 将值序列化为 IR 文本
|
||
#
|
||
# 示例:
|
||
# ConstInt(i32, 42, "42") -> "i32 42"
|
||
# SSAValue(i32, "%0") -> "i32 %0"
|
||
# ============================================================
|
||
def ValuePrint(buf: t.CChar | t.CPtr, size: t.CSizeT,
|
||
val: Value | t.CPtr, pool: memhub.MemBuddy | t.CPtr):
|
||
"""将值 val 序列化为 IR 文本写入 buf
|
||
|
||
格式: <type> <name>
|
||
pool 用于复杂类型（Array/Struct/Func）递归打印时分配临时缓冲区。
|
||
"""
|
||
if val is None or buf is None or size == 0:
|
||
return
|
||
# 先写类型
|
||
TypePrint(buf, size, val.Ty, pool)
|
||
# 追加空格
|
||
pos: t.CSizeT = string.strlen(buf)
|
||
if pos + 1 < size:
|
||
buf[pos] = ' '
|
||
buf[pos + 1] = '\0'
|
||
# 追加名字
|
||
if val.Name is not None:
|
||
pos = string.strlen(buf)
|
||
_append_str(buf, size, val.Name)
|
||
|
||
|
||
# ============================================================
|
||
# Value 字段访问器（供其他模块绕过 stub 类型限制使用）
|
||
#
|
||
# 跨模块编译时，使用方模块只有 Value 的 stub 类型（字段不足或偏移错误），
|
||
# 导致 TransPyC 静默跳过字段访问（不生成 GEP，不报错）。
|
||
# 这些访问器在拥有完整类型定义的本模块内执行，能正确访问所有字段。
|
||
# ============================================================
|
||
def value_get_ty(val: Value | t.CPtr) -> LLVMType | t.CPtr:
|
||
"""获取值的类型"""
|
||
if val is None:
|
||
return None
|
||
return val.Ty
|
||
|
||
|
||
def value_get_next(val: Value | t.CPtr) -> Value | t.CPtr:
|
||
"""获取链表下一个值"""
|
||
if val is None:
|
||
return None
|
||
return val.Next
|
||
|
||
|
||
def value_set_ty(val: Value | t.CPtr, ty: LLVMType | t.CPtr):
|
||
"""设置值的类型"""
|
||
if val is None:
|
||
return
|
||
val.Ty = ty
|
||
|
||
|
||
def value_set_next(val: Value | t.CPtr, nxt: Value | t.CPtr):
|
||
"""设置链表下一个值"""
|
||
if val is None:
|
||
return
|
||
val.Next = nxt
|
||
|
||
|
||
def value_set_name(val: Value | t.CPtr, name: t.CChar | t.CPtr):
|
||
"""设置值的名字"""
|
||
if val is None:
|
||
return
|
||
val.Name = name
|
||
|
||
|
||
def value_set_isconst(val: Value | t.CPtr, is_const: t.CInt):
|
||
"""设置是否为常量"""
|
||
if val is None:
|
||
return
|
||
val.IsConst = is_const
|
||
|
||
|
||
# ============================================================
|
||
# 内部辅助
|
||
# ============================================================
|
||
def _append_str(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr):
|
||
"""将 src 追加到 dst 末尾"""
|
||
if dst is None or src is None: return
|
||
dlen: t.CSizeT = string.strlen(dst)
|
||
# 无符号下溢防护：dlen 超过缓冲大小时（缓冲已被破坏/悬空）直接返回，
|
||
# 否则 remain = dst_size - dlen 下溢成巨值会把 dst[dlen+i] 写穿到缓冲外
|
||
if dlen >= dst_size:
|
||
return
|
||
slen: t.CSizeT = string.strlen(src)
|
||
remain: t.CSizeT = dst_size - dlen
|
||
if remain <= 0: return
|
||
i: t.CSizeT = 0
|
||
while i < slen and i + 1 < remain:
|
||
dst[dlen + i] = src[i]
|
||
i += 1
|
||
dst[dlen + i] = '\0'
|