zzy
2c13ac54df
- 在ABI类型计算中添加FLOAT和DOUBLE类型的映射 - 修复AST操作符注释中的歧义描述 - 为ast2ir上下文添加类型缓存以解决递归结构体定义问题 - 实现复合初始化表达式的支持，包括数组和结构体初始化 - 添加前置和后置自增/自减操作符的IR转换 - 实现三元条件表达式的IR生成 - 添加类型转换(cast)和sizeof操作符的支持 - 重构数组长度推断逻辑并添加类型大小计算函数 - 实现结构体和联合体的递归类型解析 - 添加函数指针调用相关的IR节点类型定义 fix(ast): 修正间接操作符的注释说明 refactor(ast2ir): 优化代码结构并添加必要的断言验证
287 lines
10 KiB
C
287 lines
10 KiB
C
#ifndef __SCC_HIR_BUILDER_H__
|
||
#define __SCC_HIR_BUILDER_H__
|
||
|
||
#include "scc_hir.h"
|
||
#include "scc_hir_prog.h"
|
||
|
||
typedef struct scc_hir_builder scc_hir_builder_t;
|
||
|
||
/**
|
||
* @brief IR 构建器上下文
|
||
*
|
||
* 负责管理 IR 构建过程中的所有状态：
|
||
* - 类型统一化（type uniquing）
|
||
* - 符号命名分配
|
||
* - 内存管理
|
||
* - 当前构建位置（函数、基本块）
|
||
*/
|
||
struct scc_hir_builder {
|
||
scc_hir_cprog_t *cprog;
|
||
scc_hir_func_ref_t current_func; ///< 当前正在构建的函数
|
||
scc_hir_bblock_ref_t current_bblock; ///< 当前基本块
|
||
|
||
scc_hashtable_t type_uniquing; // 类型哈希表：hash -> type_ref
|
||
scc_hashtable_t const_pool; // 常量哈希表：hash -> node_ref
|
||
scc_hashtable_t func_decl_set; // 函数声明集合：name -> func_ref
|
||
#ifndef SCC_NO_DEBUG
|
||
int borrow_depth;
|
||
const char *dbg_file;
|
||
int dbg_line;
|
||
#endif
|
||
};
|
||
|
||
#ifndef SCC_NO_DEBUG
|
||
#define SCC_HIR_BUILDER_BEGIN_BORROW(builder, ptr_var, ptr_expr) \
|
||
do { \
|
||
(builder)->borrow_depth++; \
|
||
(builder)->dbg_file = __FILE__; \
|
||
(builder)->dbg_line = __LINE__; \
|
||
ptr_var = (ptr_expr); \
|
||
} while (0)
|
||
|
||
#define SCC_HIR_BUILDER_END_BORROW(builder) \
|
||
do { \
|
||
(builder)->borrow_depth--; \
|
||
} while (0)
|
||
|
||
#define SCC_HIR_BUILDER_CHECK_NO_BORROW(builder) \
|
||
do { \
|
||
if ((builder)->borrow_depth != 0) { \
|
||
Panic("IR Builder: attempt to reallocate while borrowed at %s:%d", \
|
||
(builder)->dbg_file, (builder)->dbg_line); \
|
||
} \
|
||
} while (0)
|
||
#else
|
||
#define SCC_HIR_BUILDER_BEGIN_BORROW(builder, ptr_var, ptr_expr) \
|
||
ptr_var = (ptr_expr)
|
||
#define SCC_HIR_BUILDER_END_BORROW(builder) ((void)0)
|
||
#define SCC_HIR_BUILDER_CHECK_NO_BORROW(builder) ((void)0)
|
||
#endif
|
||
|
||
/**
|
||
* @brief 初始化 IR 构建器
|
||
*/
|
||
void scc_hir_builder_init(scc_hir_builder_t *builder, scc_hir_cprog_t *cprog);
|
||
|
||
/**
|
||
* @brief 销毁 IR 构建器及其所有资源
|
||
*/
|
||
void scc_hir_builder_drop(scc_hir_builder_t *builder);
|
||
|
||
static inline scc_hir_module_t *
|
||
scc_hir_builder_get_module(scc_hir_builder_t *builder) {
|
||
return &builder->cprog->module;
|
||
}
|
||
|
||
scc_hir_func_ref_t scc_hir_builder_func(scc_hir_builder_t *builder,
|
||
scc_hir_type_ref_t type_ref,
|
||
const char *name);
|
||
|
||
scc_hir_type_ref_t scc_hir_builder_type(scc_hir_builder_t *builder,
|
||
const scc_hir_type_t *type_desc);
|
||
|
||
void scc_hir_builder_add_instr(scc_hir_builder_t *builder,
|
||
scc_hir_value_ref_t instr);
|
||
|
||
scc_hir_value_ref_t scc_hir_builder_global_alloca(scc_hir_builder_t *builder,
|
||
scc_hir_type_ref_t type,
|
||
scc_hir_value_ref_t value);
|
||
|
||
/**
|
||
* @brief 创建alloca指令（在当前基本块中）
|
||
* @param type 分配的类型
|
||
* @param name 变量名（可为nullptr）
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_alloca(scc_hir_builder_t *builder,
|
||
scc_hir_type_ref_t type,
|
||
const char *name);
|
||
|
||
#define SCC_HIR_BUILDER_TYPE_FUNC(scc_type) \
|
||
SCC_MAYBE_UNUSED static inline scc_hir_type_ref_t \
|
||
scc_hir_builder_type_##scc_type(scc_hir_builder_t *builder) { \
|
||
scc_hir_type_t type_desc; \
|
||
scc_hir_type_init(&type_desc, SCC_HIR_TYPE_##scc_type); \
|
||
return scc_hir_builder_type(builder, &type_desc); \
|
||
}
|
||
|
||
SCC_HIR_BUILDER_TYPE_FUNC(unknown)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(void)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(i8)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(i16)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(i32)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(i64)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(i128)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(u8)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(u16)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(u32)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(u64)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(u128)
|
||
// SCC_HIR_BUILDER_TYPE_FUNC(f8)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(f16)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(f32)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(f64)
|
||
SCC_HIR_BUILDER_TYPE_FUNC(f128)
|
||
|
||
scc_hir_value_ref_t scc_hir_builder_builtin_memcpy(scc_hir_builder_t *builder,
|
||
scc_hir_value_ref_t dest,
|
||
scc_hir_value_ref_t src,
|
||
scc_hir_value_ref_t len);
|
||
|
||
// TODO
|
||
static inline scc_hir_value_ref_t
|
||
scc_hir_builder_integer(scc_hir_builder_t *builder, scc_hir_type_ref_t type,
|
||
scc_ap_t *val) {
|
||
scc_hir_value_t value;
|
||
scc_hir_value_init(&value, nullptr, SCC_HIR_VALUE_TAG_INTEGER);
|
||
// FIXME assign
|
||
value.data.integer = *val;
|
||
value.type = type;
|
||
return scc_hir_module_add_value(&builder->cprog->module, &value);
|
||
}
|
||
|
||
scc_hir_value_ref_t scc_hir_builder_const_string(scc_hir_builder_t *builder,
|
||
const char *str, usize len);
|
||
|
||
/**
|
||
* @brief 开始构建函数
|
||
* @param func_ref 函数引用
|
||
* @param param_names 参数名列表（可为nullptr）
|
||
* @return void
|
||
*/
|
||
void scc_hir_builder_begin_func(scc_hir_builder_t *builder,
|
||
scc_hir_func_ref_t func_ref);
|
||
|
||
/**
|
||
* @brief 结束当前函数的构建
|
||
*/
|
||
void scc_hir_builder_end_func(scc_hir_builder_t *builder);
|
||
|
||
/**
|
||
* @brief 获取当前正在构建的函数
|
||
*/
|
||
scc_hir_func_ref_t scc_hir_builder_current_func(scc_hir_builder_t *builder);
|
||
|
||
/**
|
||
* @brief 创建一个新的基本块，并自动添加到当前函数中，但不改变当前块。
|
||
* @param builder IR构建器
|
||
* @param label 基本块标签（可为 nullptr，自动生成）
|
||
* @return 新基本块的引用
|
||
*/
|
||
scc_hir_bblock_ref_t scc_hir_builder_bblock(scc_hir_builder_t *builder,
|
||
const char *label);
|
||
|
||
/**
|
||
* @brief 开始构建新的基本块
|
||
* @param label 基本块标签（可为nullptr，自动生成）
|
||
* @return 基本块引用
|
||
*/
|
||
scc_hir_bblock_ref_t scc_hir_builder_begin_bblock(scc_hir_builder_t *builder,
|
||
const char *label);
|
||
|
||
/**
|
||
* @brief 结束当前基本块的构建
|
||
*/
|
||
void scc_hir_builder_end_bblock(scc_hir_builder_t *builder);
|
||
|
||
scc_hir_func_ref_t scc_hir_builder_current_bblock(scc_hir_builder_t *builder);
|
||
/**
|
||
* @brief 设置当前基本块
|
||
*/
|
||
void scc_hir_builder_set_current_bblock(scc_hir_builder_t *builder,
|
||
scc_hir_bblock_ref_t bblock);
|
||
|
||
scc_hir_value_ref_t scc_hir_builder_func_arg_ref(scc_hir_builder_t *builder,
|
||
scc_hir_type_ref_t type,
|
||
const char *name,
|
||
usize arg_idx);
|
||
|
||
/**
|
||
* @brief 创建load指令
|
||
* @param ptr 指针操作数
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_load(scc_hir_builder_t *builder,
|
||
scc_hir_value_ref_t ptr);
|
||
|
||
/**
|
||
* @brief 创建store指令
|
||
* @param ptr 目标指针
|
||
* @param value 要存储的值
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_store(scc_hir_builder_t *builder,
|
||
scc_hir_value_ref_t ptr,
|
||
scc_hir_value_ref_t value);
|
||
|
||
/**
|
||
* @brief 创建getptr指令（指针运算）
|
||
* @param ptr 基础指针
|
||
* @param index 索引值
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_get_elem_ptr(scc_hir_builder_t *builder,
|
||
scc_hir_value_ref_t ptr,
|
||
scc_hir_value_ref_t index);
|
||
|
||
/**
|
||
* @brief 创建二元运算指令
|
||
* @param op 操作符
|
||
* @param lhs 左操作数
|
||
* @param rhs 右操作数
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_binop(scc_hir_builder_t *builder,
|
||
scc_hir_op_type_t op,
|
||
scc_hir_value_ref_t lhs,
|
||
scc_hir_value_ref_t rhs);
|
||
|
||
/**
|
||
* @brief 创建比较指令
|
||
* @param op 比较操作符
|
||
* @param lhs 左操作数
|
||
* @param rhs 右操作数
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_cmp(scc_hir_builder_t *builder,
|
||
scc_hir_op_type_t op,
|
||
scc_hir_value_ref_t lhs,
|
||
scc_hir_value_ref_t rhs);
|
||
|
||
/**
|
||
* @brief 创建跳转指令（无条件）
|
||
* @param target 目标基本块
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_jump(scc_hir_builder_t *builder,
|
||
scc_hir_bblock_ref_t target);
|
||
|
||
/**
|
||
* @brief 创建条件分支指令
|
||
* @param cond 条件值
|
||
* @param true_target 条件为真时的目标
|
||
* @param false_target 条件为假时的目标
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_branch(scc_hir_builder_t *builder,
|
||
scc_hir_value_ref_t cond,
|
||
scc_hir_bblock_ref_t true_target,
|
||
scc_hir_bblock_ref_t false_target);
|
||
|
||
/**
|
||
* @brief 创建函数调用指令
|
||
* @param callee 被调用函数
|
||
* @param args 参数列表
|
||
* @param arg_count 参数数量
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_call(scc_hir_builder_t *builder,
|
||
scc_hir_func_ref_t callee,
|
||
const scc_hir_value_ref_t *args,
|
||
usize arg_count);
|
||
|
||
/**
|
||
* @brief 创建返回指令（带返回值）
|
||
* @param value 返回值
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_ret(scc_hir_builder_t *builder,
|
||
scc_hir_value_ref_t value);
|
||
|
||
/**
|
||
* @brief 创建返回指令（void返回）
|
||
*/
|
||
scc_hir_value_ref_t scc_hir_builder_ret_void(scc_hir_builder_t *builder);
|
||
|
||
#endif /* __SCC_HIR_BUILDER_H__ */
|