llvm/codegen: extract wrapper/object emission into object.rs; dedupe mod.rs. runner/json_v0_bridge: introduce BridgeEnv + VarScope to unify lowering paths (lower_expr/args) and cache env flags; thread env through stmt lowering; minor HashMap type cleanups. Build + PyVM stage-2 smokes passed.

This commit is contained in:
Selfhosting Dev
2025-09-17 02:50:39 +09:00
parent f2ffa30645
commit 4bd49884ca
32 changed files with 3348 additions and 2304 deletions

View File

@ -24,6 +24,15 @@ What Changed (today)
- dev プロファイル `tools/dev_env.sh phi_off` を追加。ルート清掃ユーティリティ `tools/clean_root_artifacts.sh` を追加。 - dev プロファイル `tools/dev_env.sh phi_off` を追加。ルート清掃ユーティリティ `tools/clean_root_artifacts.sh` を追加。
- CIGH Actionsを curated LLVMPHIon/off実行に刷新。旧JITジョブは停止。 - CIGH Actionsを curated LLVMPHIon/off実行に刷新。旧JITジョブは停止。
SelfHosting plumbing (20250916, later in day)
- Runner: 自己ホスト経路で子プログラム(`apps/selfhost-compiler/compiler.nyash`)を優先実行し、`--read-tmp` 常時付与で安定運用に変更。
- PyVM 優先の統一(`NYASH_VM_USE_PY=1`)は EXE/inline/child の全分岐で尊重。
- CLI: `--stage3` を追加(`NYASH_NY_COMPILER_STAGE3=1` を設定)。
- Script args パススルーB案実装: `nyash ... FILE -- arg1 arg2``NYASH_SCRIPT_ARGS_JSON=["arg1","arg2"]` として `Main.main(args)` に ArrayBox で注入。
- Interpreter: `ArrayBox.of(...)` を追加して初期配列を簡潔に構築。
- Runner: Python MVP は `NYASH_NY_COMPILER_SKIP_PY=1` でスキップ可能(自己ホスト子を優先)。
Decision (Phase15 wrapup) Decision (Phase15 wrapup)
- MIR13 移行PHI 非生成): Phase15 の締めとして、MIR 生成層Bridge/Builderは PHI を生成しない方針に切替。PHI 合成は LLVM 層llvmlite/Resolverに集約。 - MIR13 移行PHI 非生成): Phase15 の締めとして、MIR 生成層Bridge/Builderは PHI を生成しない方針に切替。PHI 合成は LLVM 層llvmlite/Resolverに集約。
- LoopForm は次フェーズMIR18で導入: まずは MIR14 を維持し、次フェーズで `LoopHeader/Enter/Latch` 等の占位命令を追加。現行 Phase15 は CFG パターン検知でループ搬送値を合成。 - LoopForm は次フェーズMIR18で導入: まずは MIR14 を維持し、次フェーズで `LoopHeader/Enter/Latch` 等の占位命令を追加。現行 Phase15 は CFG パターン検知でループ搬送値を合成。
@ -36,6 +45,22 @@ Next Focus (Throw/Try — LLVM first)
- LLVM 実装方針: Throw/Try の MIR 命令を LLVM 側がどう扱うかpanic扱い or fallbackを設計し、smoke 更新案を作る(現状 Throw は trap/unreachable 最小降ろし完了)。 - LLVM 実装方針: Throw/Try の MIR 命令を LLVM 側がどう扱うかpanic扱い or fallbackを設計し、smoke 更新案を作る(現状 Throw は trap/unreachable 最小降ろし完了)。
- テスト計画: JSON フィクスチャと `tools/llvm_smoke.sh` を中心に Stage-3 例外用のスモーク/単体テストを整備。 - テスト計画: JSON フィクスチャと `tools/llvm_smoke.sh` を中心に Stage-3 例外用のスモーク/単体テストを整備。
Open Items (handoff)
- Selfhost Stage3 E2E smokeRunner→子→JSON→Bridgeの最終緑化
- 現状: 子へ `--read-tmp` は付与済み。`--stage3` は env→子へ透過済み。
- 直近 TODO: tools/selfhost_stage3_accept_smoke.sh を Runner 優先経路に再調整し、`NYASH_NY_COMPILER_SKIP_PY=1` 併用で安定化。
- 期待値: try/finally/break/continue/throw(accept) で exit=0degrade 経路)
- CLI argv 前処理の挙動監視
- `--` なしの通常実行での Positional FILE 受理を再検証Clapの警告再現時は build_command の宣言順/Arg要件を再点検
- LLVM: Throw 実投げの終了コード方針を決定し、trap on/off の期待値を固定化。
Next Steps (suggested order)
1) Selfhost Stage3 E2E 緑化smoke 修正 → 確認)
2) CLI FILE positional の回帰があれば即修正Clap設定の微調整
3) LLVM Throw 実投げの設計(終了コード/シグナル)+スモーク
4) Runner 実装の集約modes/common → selfhost.rsと微イズ削減
5) CI に Selfhost Stage2/E2E軽量をオプションで追加
※ Cranelift/JIT 系は当面対象外。ビルド時も LLVM のみを有効化JIT 関連 feature/CI は無視)。 ※ Cranelift/JIT 系は当面対象外。ビルド時も LLVM のみを有効化JIT 関連 feature/CI は無視)。
Runner updates (20250916) Runner updates (20250916)

View File

@ -10,7 +10,8 @@ pub(super) fn lower_one_function<'ctx>(
) -> Result<(), String> { ) -> Result<(), String> {
// Create basic blocks (prefix names with function label to avoid any ambiguity) // Create basic blocks (prefix names with function label to avoid any ambiguity)
let fn_label = sanitize_symbol(name); let fn_label = sanitize_symbol(name);
let (mut bb_map, entry_bb) = instructions::create_basic_blocks(codegen, llvm_func, func, &fn_label); let (mut bb_map, entry_bb) =
instructions::create_basic_blocks(codegen, llvm_func, func, &fn_label);
let mut cursor = instructions::builder_cursor::BuilderCursor::new(&codegen.builder); let mut cursor = instructions::builder_cursor::BuilderCursor::new(&codegen.builder);
cursor.at_end(func.entry_block, entry_bb); cursor.at_end(func.entry_block, entry_bb);
let mut vmap: HashMap<ValueId, BasicValueEnum> = HashMap::new(); let mut vmap: HashMap<ValueId, BasicValueEnum> = HashMap::new();
@ -23,19 +24,25 @@ pub(super) fn lower_one_function<'ctx>(
Vec<(ValueId, PhiValue, Vec<(crate::mir::BasicBlockId, ValueId)>)>, Vec<(ValueId, PhiValue, Vec<(crate::mir::BasicBlockId, ValueId)>)>,
> = HashMap::new(); > = HashMap::new();
// Snapshot of values at the end of each basic block (for sealed-SSA PHI wiring) // Snapshot of values at the end of each basic block (for sealed-SSA PHI wiring)
let mut block_end_values: HashMap<crate::mir::BasicBlockId, HashMap<ValueId, BasicValueEnum>> = HashMap::new(); let mut block_end_values: HashMap<crate::mir::BasicBlockId, HashMap<ValueId, BasicValueEnum>> =
HashMap::new();
// Build successors and predecessors map (for optional sealed-SSA PHI wiring) // Build successors and predecessors map (for optional sealed-SSA PHI wiring)
let mut succs: HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>> = HashMap::new(); let mut succs: HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>> =
HashMap::new();
for (bid, block) in &func.blocks { for (bid, block) in &func.blocks {
let v: Vec<crate::mir::BasicBlockId> = block.successors.iter().copied().collect(); let v: Vec<crate::mir::BasicBlockId> = block.successors.iter().copied().collect();
succs.insert(*bid, v); succs.insert(*bid, v);
} }
let mut preds: HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>> = HashMap::new(); let mut preds: HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>> =
HashMap::new();
for (b, ss) in &succs { for (b, ss) in &succs {
for s in ss { preds.entry(*s).or_default().push(*b); } for s in ss {
preds.entry(*s).or_default().push(*b);
}
} }
// Track sealed blocks to know when all preds of a successor are sealed // Track sealed blocks to know when all preds of a successor are sealed
let mut sealed_blocks: std::collections::HashSet<crate::mir::BasicBlockId> = std::collections::HashSet::new(); let mut sealed_blocks: std::collections::HashSet<crate::mir::BasicBlockId> =
std::collections::HashSet::new();
// Bind parameters // Bind parameters
for (i, pid) in func.params.iter().enumerate() { for (i, pid) in func.params.iter().enumerate() {
if let Some(av) = llvm_func.get_nth_param(i as u32) { if let Some(av) = llvm_func.get_nth_param(i as u32) {
@ -105,8 +112,17 @@ pub(super) fn lower_one_function<'ctx>(
// Default sealed-SSA ON unless explicitly disabled with NYASH_LLVM_PHI_SEALED=0 // Default sealed-SSA ON unless explicitly disabled with NYASH_LLVM_PHI_SEALED=0
let sealed_mode = std::env::var("NYASH_LLVM_PHI_SEALED").ok().as_deref() != Some("0"); let sealed_mode = std::env::var("NYASH_LLVM_PHI_SEALED").ok().as_deref() != Some("0");
// LoopForm registry (per-function lowering; gated) // LoopForm registry (per-function lowering; gated)
let mut loopform_registry: HashMap<crate::mir::BasicBlockId, (inkwell::basic_block::BasicBlock, PhiValue, PhiValue, inkwell::basic_block::BasicBlock)> = HashMap::new(); let mut loopform_registry: HashMap<
let mut loopform_body_to_header: HashMap<crate::mir::BasicBlockId, crate::mir::BasicBlockId> = HashMap::new(); crate::mir::BasicBlockId,
(
inkwell::basic_block::BasicBlock,
PhiValue,
PhiValue,
inkwell::basic_block::BasicBlock,
),
> = HashMap::new();
let mut loopform_body_to_header: HashMap<crate::mir::BasicBlockId, crate::mir::BasicBlockId> =
HashMap::new();
// Per-function Resolver for dominance-safe value access (i64 minimal) // Per-function Resolver for dominance-safe value access (i64 minimal)
let mut resolver = instructions::Resolver::new(); let mut resolver = instructions::Resolver::new();
for (bi, bid) in block_ids.iter().enumerate() { for (bi, bid) in block_ids.iter().enumerate() {
@ -117,10 +133,15 @@ pub(super) fn lower_one_function<'ctx>(
eprintln!("[LLVM] lowering bb={}", bid.as_u32()); eprintln!("[LLVM] lowering bb={}", bid.as_u32());
} }
let block = func.blocks.get(bid).unwrap(); let block = func.blocks.get(bid).unwrap();
let mut defined_in_block: std::collections::HashSet<ValueId> = std::collections::HashSet::new(); let mut defined_in_block: std::collections::HashSet<ValueId> =
std::collections::HashSet::new();
for inst in &block.instructions { for inst in &block.instructions {
match inst { match inst {
MirInstruction::NewBox { dst, box_type, args } => { MirInstruction::NewBox {
dst,
box_type,
args,
} => {
instructions::lower_newbox( instructions::lower_newbox(
codegen, codegen,
&mut cursor, &mut cursor,
@ -136,42 +157,74 @@ pub(super) fn lower_one_function<'ctx>(
&block_end_values, &block_end_values,
)?; )?;
defined_in_block.insert(*dst); defined_in_block.insert(*dst);
}, }
MirInstruction::Const { dst, value } => { MirInstruction::Const { dst, value } => {
let bval = match value { let bval = match value {
ConstValue::Integer(i) => codegen.context.i64_type().const_int(*i as u64, true).into(), ConstValue::Integer(i) => {
codegen.context.i64_type().const_int(*i as u64, true).into()
}
ConstValue::Float(f) => codegen.context.f64_type().const_float(*f).into(), ConstValue::Float(f) => codegen.context.f64_type().const_float(*f).into(),
ConstValue::Bool(b) => codegen.context.bool_type().const_int(*b as u64, false).into(), ConstValue::Bool(b) => codegen
.context
.bool_type()
.const_int(*b as u64, false)
.into(),
ConstValue::String(s) => { ConstValue::String(s) => {
// Hoist string creation to entry block to dominate all uses. // Hoist string creation to entry block to dominate all uses.
let entry_term = unsafe { entry_bb.get_terminator() }; let entry_term = unsafe { entry_bb.get_terminator() };
if let Some(t) = entry_term { entry_builder.position_before(&t); } if let Some(t) = entry_term {
else { entry_builder.position_at_end(entry_bb); } entry_builder.position_before(&t);
} else {
entry_builder.position_at_end(entry_bb);
}
let gv = entry_builder let gv = entry_builder
.build_global_string_ptr(s, "str") .build_global_string_ptr(s, "str")
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let len = codegen.context.i32_type().const_int(s.len() as u64, false); let len = codegen.context.i32_type().const_int(s.len() as u64, false);
let rt = codegen.context.ptr_type(inkwell::AddressSpace::from(0)); let rt = codegen.context.ptr_type(inkwell::AddressSpace::from(0));
let fn_ty = rt.fn_type(&[ let fn_ty = rt.fn_type(
codegen.context.ptr_type(inkwell::AddressSpace::from(0)).into(), &[
codegen
.context
.ptr_type(inkwell::AddressSpace::from(0))
.into(),
codegen.context.i32_type().into(), codegen.context.i32_type().into(),
], false); ],
false,
);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash_string_new") .get_function("nyash_string_new")
.unwrap_or_else(|| codegen.module.add_function("nyash_string_new", fn_ty, None)); .unwrap_or_else(|| {
codegen.module.add_function("nyash_string_new", fn_ty, None)
});
let call = entry_builder let call = entry_builder
.build_call(callee, &[gv.as_pointer_value().into(), len.into()], "strnew") .build_call(
callee,
&[gv.as_pointer_value().into(), len.into()],
"strnew",
)
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("nyash_string_new returned void".to_string())? call.try_as_basic_value()
.left()
.ok_or("nyash_string_new returned void".to_string())?
} }
ConstValue::Null => codegen.context.ptr_type(inkwell::AddressSpace::from(0)).const_zero().into(), ConstValue::Null => codegen
.context
.ptr_type(inkwell::AddressSpace::from(0))
.const_zero()
.into(),
ConstValue::Void => codegen.context.i64_type().const_zero().into(), ConstValue::Void => codegen.context.i64_type().const_zero().into(),
}; };
vmap.insert(*dst, bval); vmap.insert(*dst, bval);
defined_in_block.insert(*dst); defined_in_block.insert(*dst);
}, }
MirInstruction::Call { dst, func: callee, args, .. } => { MirInstruction::Call {
dst,
func: callee,
args,
..
} => {
instructions::lower_call( instructions::lower_call(
codegen, codegen,
&mut cursor, &mut cursor,
@ -188,9 +241,18 @@ pub(super) fn lower_one_function<'ctx>(
&preds, &preds,
&block_end_values, &block_end_values,
)?; )?;
if let Some(d) = dst { defined_in_block.insert(*d); } if let Some(d) = dst {
}, defined_in_block.insert(*d);
MirInstruction::BoxCall { dst, box_val, method, method_id, args, effects: _ } => { }
}
MirInstruction::BoxCall {
dst,
box_val,
method,
method_id,
args,
effects: _,
} => {
instructions::lower_boxcall( instructions::lower_boxcall(
codegen, codegen,
&mut cursor, &mut cursor,
@ -209,9 +271,17 @@ pub(super) fn lower_one_function<'ctx>(
&preds, &preds,
&block_end_values, &block_end_values,
)?; )?;
if let Some(d) = dst { defined_in_block.insert(*d); } if let Some(d) = dst {
}, defined_in_block.insert(*d);
MirInstruction::ExternCall { dst, iface_name, method_name, args, effects: _ } => { }
}
MirInstruction::ExternCall {
dst,
iface_name,
method_name,
args,
effects: _,
} => {
instructions::lower_externcall( instructions::lower_externcall(
codegen, codegen,
&mut cursor, &mut cursor,
@ -227,8 +297,10 @@ pub(super) fn lower_one_function<'ctx>(
&preds, &preds,
&block_end_values, &block_end_values,
)?; )?;
if let Some(d) = dst { defined_in_block.insert(*d); } if let Some(d) = dst {
}, defined_in_block.insert(*d);
}
}
MirInstruction::UnaryOp { dst, op, operand } => { MirInstruction::UnaryOp { dst, op, operand } => {
instructions::lower_unary( instructions::lower_unary(
codegen, codegen,
@ -245,16 +317,43 @@ pub(super) fn lower_one_function<'ctx>(
&block_end_values, &block_end_values,
)?; )?;
defined_in_block.insert(*dst); defined_in_block.insert(*dst);
}, }
MirInstruction::BinOp { dst, op, lhs, rhs } => { MirInstruction::BinOp { dst, op, lhs, rhs } => {
instructions::lower_binop(codegen, &mut cursor, &mut resolver, *bid, func, &mut vmap, *dst, op, lhs, rhs, &bb_map, &preds, &block_end_values)?; instructions::lower_binop(
codegen,
&mut cursor,
&mut resolver,
*bid,
func,
&mut vmap,
*dst,
op,
lhs,
rhs,
&bb_map,
&preds,
&block_end_values,
)?;
defined_in_block.insert(*dst); defined_in_block.insert(*dst);
}, }
MirInstruction::Compare { dst, op, lhs, rhs } => { MirInstruction::Compare { dst, op, lhs, rhs } => {
let out = instructions::lower_compare(codegen, &mut cursor, &mut resolver, *bid, func, &vmap, op, lhs, rhs, &bb_map, &preds, &block_end_values)?; let out = instructions::lower_compare(
codegen,
&mut cursor,
&mut resolver,
*bid,
func,
&vmap,
op,
lhs,
rhs,
&bb_map,
&preds,
&block_end_values,
)?;
vmap.insert(*dst, out); vmap.insert(*dst, out);
defined_in_block.insert(*dst); defined_in_block.insert(*dst);
}, }
MirInstruction::Store { value, ptr } => { MirInstruction::Store { value, ptr } => {
instructions::lower_store( instructions::lower_store(
codegen, codegen,
@ -270,17 +369,30 @@ pub(super) fn lower_one_function<'ctx>(
&preds, &preds,
&block_end_values, &block_end_values,
)?; )?;
}, }
MirInstruction::Load { dst, ptr } => { MirInstruction::Load { dst, ptr } => {
instructions::lower_load(codegen, &mut cursor, *bid, &mut vmap, &mut allocas, &mut alloca_elem_types, dst, ptr)?; instructions::lower_load(
codegen,
&mut cursor,
*bid,
&mut vmap,
&mut allocas,
&mut alloca_elem_types,
dst,
ptr,
)?;
defined_in_block.insert(*dst); defined_in_block.insert(*dst);
}, }
MirInstruction::Phi { .. } => { /* precreated */ } MirInstruction::Phi { .. } => { /* precreated */ }
_ => { /* ignore others */ } _ => { /* ignore others */ }
} }
// Snapshot end-of-block values // Snapshot end-of-block values
let mut snap: HashMap<ValueId, BasicValueEnum> = HashMap::new(); let mut snap: HashMap<ValueId, BasicValueEnum> = HashMap::new();
for vid in &defined_in_block { if let Some(v) = vmap.get(vid).copied() { snap.insert(*vid, v); } } for vid in &defined_in_block {
if let Some(v) = vmap.get(vid).copied() {
snap.insert(*vid, v);
}
}
block_end_values.insert(*bid, snap); block_end_values.insert(*bid, snap);
} }
// Terminator handling // Terminator handling
@ -288,42 +400,85 @@ pub(super) fn lower_one_function<'ctx>(
cursor.at_end(*bid, bb); cursor.at_end(*bid, bb);
match term { match term {
MirInstruction::Return { value } => { MirInstruction::Return { value } => {
instructions::emit_return(codegen, &mut cursor, &mut resolver, *bid, func, &vmap, value, &bb_map, &preds, &block_end_values)?; instructions::emit_return(
codegen,
&mut cursor,
&mut resolver,
*bid,
func,
&vmap,
value,
&bb_map,
&preds,
&block_end_values,
)?;
} }
MirInstruction::Jump { target } => { MirInstruction::Jump { target } => {
let mut handled = false; let mut handled = false;
if std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1") && if std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1")
std::env::var("NYASH_LOOPFORM_BODY2DISPATCH").ok().as_deref() == Some("1") { && std::env::var("NYASH_LOOPFORM_BODY2DISPATCH")
.ok()
.as_deref()
== Some("1")
{
if let Some(hdr) = loopform_body_to_header.get(bid) { if let Some(hdr) = loopform_body_to_header.get(bid) {
if hdr == target { if hdr == target {
if let Some((dispatch_bb, tag_phi, payload_phi, _latch_bb)) = loopform_registry.get(hdr) { if let Some((dispatch_bb, tag_phi, payload_phi, _latch_bb)) =
loopform_registry.get(hdr)
{
let i8t = codegen.context.i8_type(); let i8t = codegen.context.i8_type();
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let pred_llbb = *bb_map.get(bid).ok_or("loopform: body llbb missing")?; let pred_llbb =
*bb_map.get(bid).ok_or("loopform: body llbb missing")?;
let z = i8t.const_zero(); let z = i8t.const_zero();
let pz = i64t.const_zero(); let pz = i64t.const_zero();
tag_phi.add_incoming(&[(&z, pred_llbb)]); tag_phi.add_incoming(&[(&z, pred_llbb)]);
payload_phi.add_incoming(&[(&pz, pred_llbb)]); payload_phi.add_incoming(&[(&pz, pred_llbb)]);
cursor.emit_term(*bid, |b| { b.build_unconditional_branch(*dispatch_bb).unwrap(); }); cursor.emit_term(*bid, |b| {
b.build_unconditional_branch(*dispatch_bb).unwrap();
});
handled = true; handled = true;
} }
} }
} }
} }
if !handled { if !handled {
instructions::emit_jump(codegen, &mut cursor, *bid, target, &bb_map, &phis_by_block)?; instructions::emit_jump(
codegen,
&mut cursor,
*bid,
target,
&bb_map,
&phis_by_block,
)?;
} }
} }
MirInstruction::Branch { condition, then_bb, else_bb } => { MirInstruction::Branch {
condition,
then_bb,
else_bb,
} => {
let mut handled_by_loopform = false; let mut handled_by_loopform = false;
if std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1") { if std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1") {
let mut is_back = |start: crate::mir::BasicBlockId| -> u8 { let mut is_back = |start: crate::mir::BasicBlockId| -> u8 {
if let Some(b) = func.blocks.get(&start) { if let Some(b) = func.blocks.get(&start) {
if let Some(crate::mir::instruction::MirInstruction::Jump { target }) = &b.terminator { if let Some(crate::mir::instruction::MirInstruction::Jump {
if target == bid { return 1; } target,
}) = &b.terminator
{
if target == bid {
return 1;
}
if let Some(b2) = func.blocks.get(target) { if let Some(b2) = func.blocks.get(target) {
if let Some(crate::mir::instruction::MirInstruction::Jump { target: t2 }) = &b2.terminator { if let Some(
if t2 == bid { return 2; } crate::mir::instruction::MirInstruction::Jump {
target: t2,
},
) = &b2.terminator
{
if t2 == bid {
return 2;
}
} }
} }
} }
@ -332,73 +487,163 @@ pub(super) fn lower_one_function<'ctx>(
}; };
let d_then = is_back(*then_bb); let d_then = is_back(*then_bb);
let d_else = is_back(*else_bb); let d_else = is_back(*else_bb);
let choose_body = if d_then > 0 && d_else == 0 { Some((*then_bb, *else_bb)) } let choose_body = if d_then > 0 && d_else == 0 {
else if d_else > 0 && d_then == 0 { Some((*else_bb, *then_bb)) } Some((*then_bb, *else_bb))
else if d_then > 0 && d_else > 0 { if d_then <= d_else { Some((*then_bb, *else_bb)) } else { Some((*else_bb, *then_bb)) } } } else if d_else > 0 && d_then == 0 {
else { None }; Some((*else_bb, *then_bb))
} else if d_then > 0 && d_else > 0 {
if d_then <= d_else {
Some((*then_bb, *else_bb))
} else {
Some((*else_bb, *then_bb))
}
} else {
None
};
if let Some((body_sel, after_sel)) = choose_body { if let Some((body_sel, after_sel)) = choose_body {
let body_block = func.blocks.get(&body_sel).unwrap(); let body_block = func.blocks.get(&body_sel).unwrap();
handled_by_loopform = instructions::lower_while_loopform( handled_by_loopform = instructions::lower_while_loopform(
codegen, &mut cursor, &mut resolver, func, llvm_func, condition, &body_block.instructions, codegen,
loopform_loop_id, &fn_label, *bid, body_sel, after_sel, &bb_map, &vmap, &preds, &block_end_values, &mut cursor,
&mut loopform_registry, &mut loopform_body_to_header, &mut resolver,
func,
llvm_func,
condition,
&body_block.instructions,
loopform_loop_id,
&fn_label,
*bid,
body_sel,
after_sel,
&bb_map,
&vmap,
&preds,
&block_end_values,
&mut loopform_registry,
&mut loopform_body_to_header,
)?; )?;
loopform_loop_id = loopform_loop_id.wrapping_add(1); loopform_loop_id = loopform_loop_id.wrapping_add(1);
} }
} }
if !handled_by_loopform { if !handled_by_loopform {
instructions::emit_branch(codegen, &mut cursor, &mut resolver, *bid, condition, then_bb, else_bb, &bb_map, &phis_by_block, &vmap, &preds, &block_end_values)?; instructions::emit_branch(
codegen,
&mut cursor,
&mut resolver,
*bid,
condition,
then_bb,
else_bb,
&bb_map,
&phis_by_block,
&vmap,
&preds,
&block_end_values,
)?;
} }
} }
_ => { _ => {
cursor.at_end(*bid, bb); cursor.at_end(*bid, bb);
if let Some(next_bid) = block_ids.get(bi + 1) { if let Some(next_bid) = block_ids.get(bi + 1) {
instructions::emit_jump(codegen, &mut cursor, *bid, next_bid, &bb_map, &phis_by_block)?; instructions::emit_jump(
codegen,
&mut cursor,
*bid,
next_bid,
&bb_map,
&phis_by_block,
)?;
} else { } else {
let entry_first = func.entry_block; let entry_first = func.entry_block;
instructions::emit_jump(codegen, &mut cursor, *bid, &entry_first, &bb_map, &phis_by_block)?; instructions::emit_jump(
codegen,
&mut cursor,
*bid,
&entry_first,
&bb_map,
&phis_by_block,
)?;
} }
} }
} }
} else { } else {
if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") { if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") {
eprintln!("[LLVM] no terminator in MIR for bb={} (fallback)", bid.as_u32()); eprintln!(
"[LLVM] no terminator in MIR for bb={} (fallback)",
bid.as_u32()
);
} }
cursor.at_end(*bid, bb); cursor.at_end(*bid, bb);
if let Some(next_bid) = block_ids.get(bi + 1) { if let Some(next_bid) = block_ids.get(bi + 1) {
instructions::emit_jump(codegen, &mut cursor, *bid, next_bid, &bb_map, &phis_by_block)?; instructions::emit_jump(
codegen,
&mut cursor,
*bid,
next_bid,
&bb_map,
&phis_by_block,
)?;
} else { } else {
let entry_first = func.entry_block; let entry_first = func.entry_block;
instructions::emit_jump(codegen, &mut cursor, *bid, &entry_first, &bb_map, &phis_by_block)?; instructions::emit_jump(
codegen,
&mut cursor,
*bid,
&entry_first,
&bb_map,
&phis_by_block,
)?;
} }
} }
if unsafe { bb.get_terminator() }.is_none() { if unsafe { bb.get_terminator() }.is_none() {
if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") { if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") {
eprintln!("[LLVM] extra guard inserting fallback for bb={}", bid.as_u32()); eprintln!(
"[LLVM] extra guard inserting fallback for bb={}",
bid.as_u32()
);
} }
cursor.at_end(*bid, bb); cursor.at_end(*bid, bb);
if let Some(next_bid) = block_ids.get(bi + 1) { if let Some(next_bid) = block_ids.get(bi + 1) {
instructions::emit_jump(codegen, &mut cursor, *bid, next_bid, &bb_map, &phis_by_block)?; instructions::emit_jump(
codegen,
&mut cursor,
*bid,
next_bid,
&bb_map,
&phis_by_block,
)?;
} else { } else {
let entry_first = func.entry_block; let entry_first = func.entry_block;
instructions::emit_jump(codegen, &mut cursor, *bid, &entry_first, &bb_map, &phis_by_block)?; instructions::emit_jump(
codegen,
&mut cursor,
*bid,
&entry_first,
&bb_map,
&phis_by_block,
)?;
} }
} }
if sealed_mode { if sealed_mode {
instructions::flow::seal_block(codegen, &mut cursor, func, *bid, &succs, &bb_map, &phis_by_block, &block_end_values)?; instructions::flow::seal_block(
codegen,
&mut cursor,
func,
*bid,
&succs,
&bb_map,
&phis_by_block,
&block_end_values,
)?;
sealed_blocks.insert(*bid); sealed_blocks.insert(*bid);
} }
} }
if std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1") && if std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1")
std::env::var("NYASH_LOOPFORM_LATCH2HEADER").ok().as_deref() == Some("1") { && std::env::var("NYASH_LOOPFORM_LATCH2HEADER").ok().as_deref() == Some("1")
{
for (hdr_bid, (_dispatch_bb, _tag_phi, _payload_phi, latch_bb)) in &loopform_registry { for (hdr_bid, (_dispatch_bb, _tag_phi, _payload_phi, latch_bb)) in &loopform_registry {
if let Some(phis) = phis_by_block.get(hdr_bid) { if let Some(phis) = phis_by_block.get(hdr_bid) {
instructions::normalize_header_phis_for_latch( instructions::normalize_header_phis_for_latch(codegen, *hdr_bid, *latch_bb, phis)?;
codegen,
*hdr_bid,
*latch_bb,
phis,
)?;
} }
} }
instructions::dev_check_dispatch_only_phi(&phis_by_block, &loopform_registry); instructions::dev_check_dispatch_only_phi(&phis_by_block, &loopform_registry);

View File

@ -2,9 +2,9 @@ use std::collections::HashMap;
use inkwell::values::BasicValueEnum; use inkwell::values::BasicValueEnum;
use super::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, CompareOp, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, CompareOp, ValueId};
use super::builder_cursor::BuilderCursor;
/// Compare lowering: return the resulting BasicValueEnum (i1) /// Compare lowering: return the resulting BasicValueEnum (i1)
pub(in super::super) fn lower_compare<'ctx, 'b>( pub(in super::super) fn lower_compare<'ctx, 'b>(
@ -17,21 +17,93 @@ pub(in super::super) fn lower_compare<'ctx, 'b>(
op: &CompareOp, op: &CompareOp,
lhs: &ValueId, lhs: &ValueId,
rhs: &ValueId, rhs: &ValueId,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
>,
) -> Result<BasicValueEnum<'ctx>, String> { ) -> Result<BasicValueEnum<'ctx>, String> {
use crate::backend::llvm::compiler::helpers::{as_float, as_int}; use crate::backend::llvm::compiler::helpers::{as_float, as_int};
// Synthesize proxy values via Resolver according to metadata // Synthesize proxy values via Resolver according to metadata
let lv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(lhs) { let lv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(lhs) {
Some(crate::mir::MirType::Float) => resolver.resolve_f64(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap)?.into(), Some(crate::mir::MirType::Float) => resolver
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver.resolve_ptr(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap)?.into(), .resolve_f64(
_ => resolver.resolve_i64(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap)?.into(), codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver
.resolve_ptr(
codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
_ => resolver
.resolve_i64(
codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
}; };
let rv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(rhs) { let rv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(rhs) {
Some(crate::mir::MirType::Float) => resolver.resolve_f64(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap)?.into(), Some(crate::mir::MirType::Float) => resolver
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver.resolve_ptr(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap)?.into(), .resolve_f64(
_ => resolver.resolve_i64(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap)?.into(), codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver
.resolve_ptr(
codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
_ => resolver
.resolve_i64(
codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
}; };
// String equality/inequality by content when annotated as String/StringBox // String equality/inequality by content when annotated as String/StringBox
if matches!(op, CompareOp::Eq | CompareOp::Ne) { if matches!(op, CompareOp::Eq | CompareOp::Ne) {
@ -48,19 +120,42 @@ pub(in super::super) fn lower_compare<'ctx, 'b>(
if l_is_str && r_is_str { if l_is_str && r_is_str {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
// Convert both sides to handles if needed // Convert both sides to handles if needed
let mut to_handle = |v: BasicValueEnum<'ctx>| -> Result<inkwell::values::IntValue<'ctx>, String> { let mut to_handle =
|v: BasicValueEnum<'ctx>| -> Result<inkwell::values::IntValue<'ctx>, String> {
match v { match v {
BasicValueEnum::IntValue(iv) => { BasicValueEnum::IntValue(iv) => {
if iv.get_type() == i64t { Ok(iv) } else { cursor.emit_instr(cur_bid, |b| b.build_int_s_extend(iv, i64t, "i2i64")).map_err(|e| e.to_string()) } if iv.get_type() == i64t {
Ok(iv)
} else {
cursor
.emit_instr(cur_bid, |b| {
b.build_int_s_extend(iv, i64t, "i2i64")
})
.map_err(|e| e.to_string())
}
} }
BasicValueEnum::PointerValue(pv) => { BasicValueEnum::PointerValue(pv) => {
let fnty = i64t.fn_type(&[codegen.context.ptr_type(inkwell::AddressSpace::from(0)).into()], false); let fnty = i64t.fn_type(
&[codegen
.context
.ptr_type(inkwell::AddressSpace::from(0))
.into()],
false,
);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty, None)); .unwrap_or_else(|| {
codegen.module.add_function(
"nyash.box.from_i8_string",
fnty,
None,
)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[pv.into()], "str_ptr_to_handle_cmp")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[pv.into()], "str_ptr_to_handle_cmp")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
@ -77,9 +172,15 @@ pub(in super::super) fn lower_compare<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.eq_hh") .get_function("nyash.string.eq_hh")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.eq_hh", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.eq_hh", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[lh.into(), rh.into()], "str_eq_hh")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[lh.into(), rh.into()], "str_eq_hh")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let iv = call let iv = call
.try_as_basic_value() .try_as_basic_value()
@ -93,7 +194,9 @@ pub(in super::super) fn lower_compare<'ctx, 'b>(
inkwell::IntPredicate::EQ inkwell::IntPredicate::EQ
}; };
let b = cursor let b = cursor
.emit_instr(cur_bid, |bd| bd.build_int_compare(pred, iv, zero, "str_eq_to_bool")) .emit_instr(cur_bid, |bd| {
bd.build_int_compare(pred, iv, zero, "str_eq_to_bool")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
return Ok(b.into()); return Ok(b.into());
} }
@ -101,10 +204,28 @@ pub(in super::super) fn lower_compare<'ctx, 'b>(
let out = if let (Some(_li0), Some(_ri0)) = (as_int(lv), as_int(rv)) { let out = if let (Some(_li0), Some(_ri0)) = (as_int(lv), as_int(rv)) {
// Localize integer operands into current block to satisfy dominance // Localize integer operands into current block to satisfy dominance
let mut li = resolver let mut li = resolver
.resolve_i64(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap) .resolve_i64(
codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)
.unwrap_or_else(|_| as_int(lv).unwrap()); .unwrap_or_else(|_| as_int(lv).unwrap());
let mut ri = resolver let mut ri = resolver
.resolve_i64(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap) .resolve_i64(
codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)
.unwrap_or_else(|_| as_int(rv).unwrap()); .unwrap_or_else(|_| as_int(rv).unwrap());
// Normalize integer widths: extend the narrower to match the wider to satisfy LLVM // Normalize integer widths: extend the narrower to match the wider to satisfy LLVM
let lw = li.get_type().get_bit_width(); let lw = li.get_type().get_bit_width();
@ -112,11 +233,15 @@ pub(in super::super) fn lower_compare<'ctx, 'b>(
if lw != rw { if lw != rw {
if lw < rw { if lw < rw {
li = cursor li = cursor
.emit_instr(cur_bid, |b| b.build_int_z_extend(li, ri.get_type(), "icmp_zext_l")) .emit_instr(cur_bid, |b| {
b.build_int_z_extend(li, ri.get_type(), "icmp_zext_l")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
} else { } else {
ri = cursor ri = cursor
.emit_instr(cur_bid, |b| b.build_int_z_extend(ri, li.get_type(), "icmp_zext_r")) .emit_instr(cur_bid, |b| {
b.build_int_z_extend(ri, li.get_type(), "icmp_zext_r")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
} }
} }
@ -213,14 +338,21 @@ pub(in super::super) fn lower_compare<'ctx, 'b>(
Ok(out) Ok(out)
} }
fn guessed_zero<'ctx>(codegen: &CodegenContext<'ctx>, func: &MirFunction, vid: &crate::mir::ValueId) -> BasicValueEnum<'ctx> { fn guessed_zero<'ctx>(
codegen: &CodegenContext<'ctx>,
func: &MirFunction,
vid: &crate::mir::ValueId,
) -> BasicValueEnum<'ctx> {
use crate::mir::MirType as MT; use crate::mir::MirType as MT;
match func.metadata.value_types.get(vid) { match func.metadata.value_types.get(vid) {
Some(MT::Bool) => codegen.context.bool_type().const_zero().into(), Some(MT::Bool) => codegen.context.bool_type().const_zero().into(),
Some(MT::Integer) => codegen.context.i64_type().const_zero().into(), Some(MT::Integer) => codegen.context.i64_type().const_zero().into(),
Some(MT::Float) => codegen.context.f64_type().const_zero().into(), Some(MT::Float) => codegen.context.f64_type().const_zero().into(),
Some(MT::String) | Some(MT::Box(_)) | Some(MT::Array(_)) | Some(MT::Future(_)) | Some(MT::Unknown) | Some(MT::Void) | None => { Some(MT::String) | Some(MT::Box(_)) | Some(MT::Array(_)) | Some(MT::Future(_))
codegen.context.ptr_type(inkwell::AddressSpace::from(0)).const_zero().into() | Some(MT::Unknown) | Some(MT::Void) | None => codegen
} .context
.ptr_type(inkwell::AddressSpace::from(0))
.const_zero()
.into(),
} }
} }

View File

@ -1,11 +1,11 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::{values::BasicValueEnum, AddressSpace};
use crate::backend::llvm::compiler::codegen::types; use crate::backend::llvm::compiler::codegen::types;
use inkwell::{values::BasicValueEnum, AddressSpace};
use super::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, instruction::UnaryOp, BasicBlockId, BinaryOp, ValueId}; use crate::mir::{function::MirFunction, instruction::UnaryOp, BasicBlockId, BinaryOp, ValueId};
use super::builder_cursor::BuilderCursor;
/// Lower UnaryOp and store into vmap (0-diff) /// Lower UnaryOp and store into vmap (0-diff)
pub(in super::super) fn lower_unary<'ctx, 'b>( pub(in super::super) fn lower_unary<'ctx, 'b>(
@ -18,27 +18,67 @@ pub(in super::super) fn lower_unary<'ctx, 'b>(
dst: ValueId, dst: ValueId,
op: &UnaryOp, op: &UnaryOp,
operand: &ValueId, operand: &ValueId,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
use crate::mir::MirType as MT; use crate::mir::MirType as MT;
let out = match op { let out = match op {
UnaryOp::Neg => { UnaryOp::Neg => match func.metadata.value_types.get(operand) {
match func.metadata.value_types.get(operand) {
Some(MT::Float) => { Some(MT::Float) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, *operand, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
cursor.emit_instr(cur_bid, |b| b.build_float_neg(fv, "fneg")).map_err(|e| e.to_string())?.into() codegen,
cursor,
cur_bid,
*operand,
bb_map,
preds,
block_end_values,
vmap,
)?;
cursor
.emit_instr(cur_bid, |b| b.build_float_neg(fv, "fneg"))
.map_err(|e| e.to_string())?
.into()
} }
_ => { _ => {
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *operand, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
cursor.emit_instr(cur_bid, |b| b.build_int_neg(iv, "ineg")).map_err(|e| e.to_string())?.into() codegen,
} cursor,
} cur_bid,
*operand,
bb_map,
preds,
block_end_values,
vmap,
)?;
cursor
.emit_instr(cur_bid, |b| b.build_int_neg(iv, "ineg"))
.map_err(|e| e.to_string())?
.into()
} }
},
UnaryOp::Not | UnaryOp::BitNot => { UnaryOp::Not | UnaryOp::BitNot => {
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *operand, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
cursor.emit_instr(cur_bid, |b| b.build_not(iv, "inot")).map_err(|e| e.to_string())?.into() codegen,
cursor,
cur_bid,
*operand,
bb_map,
preds,
block_end_values,
vmap,
)?;
cursor
.emit_instr(cur_bid, |b| b.build_not(iv, "inot"))
.map_err(|e| e.to_string())?
.into()
} }
}; };
vmap.insert(dst, out); vmap.insert(dst, out);
@ -57,23 +97,95 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
op: &BinaryOp, op: &BinaryOp,
lhs: &ValueId, lhs: &ValueId,
rhs: &ValueId, rhs: &ValueId,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
use crate::backend::llvm::compiler::helpers::{as_float, as_int}; use crate::backend::llvm::compiler::helpers::{as_float, as_int};
use inkwell::values::BasicValueEnum as BVE; use inkwell::values::BasicValueEnum as BVE;
use inkwell::IntPredicate; use inkwell::IntPredicate;
// Construct lhs/rhs proxy values via Resolver according to metadata (no vmap direct access) // Construct lhs/rhs proxy values via Resolver according to metadata (no vmap direct access)
let lv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(lhs) { let lv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(lhs) {
Some(crate::mir::MirType::Float) => resolver.resolve_f64(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap)?.into(), Some(crate::mir::MirType::Float) => resolver
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver.resolve_ptr(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap)?.into(), .resolve_f64(
_ => resolver.resolve_i64(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap)?.into(), codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver
.resolve_ptr(
codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
_ => resolver
.resolve_i64(
codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
}; };
let rv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(rhs) { let rv: BasicValueEnum<'ctx> = match func.metadata.value_types.get(rhs) {
Some(crate::mir::MirType::Float) => resolver.resolve_f64(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap)?.into(), Some(crate::mir::MirType::Float) => resolver
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver.resolve_ptr(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap)?.into(), .resolve_f64(
_ => resolver.resolve_i64(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap)?.into(), codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) => resolver
.resolve_ptr(
codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
_ => resolver
.resolve_i64(
codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)?
.into(),
}; };
let mut handled_concat = false; let mut handled_concat = false;
if let BinaryOp::Add = op { if let BinaryOp::Add = op {
@ -91,10 +203,15 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_ss") .get_function("nyash.string.concat_ss")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_ss", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_ss", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[lp.into(), rp.into()], "concat_ss")) b.build_call(callee, &[lp.into(), rp.into()], "concat_ss")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
@ -103,7 +220,9 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
// store as handle (i64) across blocks // store as handle (i64) across blocks
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let h = cursor let h = cursor
.emit_instr(cur_bid, |b| b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")) .emit_instr(cur_bid, |b| {
b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(dst, h.into()); vmap.insert(dst, h.into());
handled_concat = true; handled_concat = true;
@ -115,10 +234,15 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let conv = codegen let conv = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty_conv, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty_conv, None)
});
let call_c = cursor let call_c = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(conv, &[lp.into()], "lhs_i8_to_handle")) b.build_call(conv, &[lp.into()], "lhs_i8_to_handle")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let lh = call_c let lh = call_c
.try_as_basic_value() .try_as_basic_value()
@ -129,10 +253,15 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_hh") .get_function("nyash.string.concat_hh")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_hh", fnty_hh, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_hh", fnty_hh, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[lh.into(), ri.into()], "concat_hh")) b.build_call(callee, &[lh.into(), ri.into()], "concat_hh")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
@ -146,7 +275,11 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_si") .get_function("nyash.string.concat_si")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_si", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_si", fnty, None)
});
let call = codegen let call = codegen
.builder .builder
.build_call(callee, &[lp.into(), ri.into()], "concat_si") .build_call(callee, &[lp.into(), ri.into()], "concat_si")
@ -157,7 +290,9 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
.ok_or("concat_si returned void".to_string())?; .ok_or("concat_si returned void".to_string())?;
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let h = cursor let h = cursor
.emit_instr(cur_bid, |b| b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")) .emit_instr(cur_bid, |b| {
b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(dst, h.into()); vmap.insert(dst, h.into());
handled_concat = true; handled_concat = true;
@ -170,7 +305,11 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let conv = codegen let conv = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty_conv, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty_conv, None)
});
let call_c = codegen let call_c = codegen
.builder .builder
.build_call(conv, &[rp.into()], "rhs_i8_to_handle") .build_call(conv, &[rp.into()], "rhs_i8_to_handle")
@ -184,10 +323,15 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_hh") .get_function("nyash.string.concat_hh")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_hh", fnty_hh, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_hh", fnty_hh, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[li.into(), rh.into()], "concat_hh")) b.build_call(callee, &[li.into(), rh.into()], "concat_hh")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
@ -201,10 +345,15 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_is") .get_function("nyash.string.concat_is")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_is", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_is", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[li.into(), rp.into()], "concat_is")) b.build_call(callee, &[li.into(), rp.into()], "concat_is")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
@ -212,7 +361,9 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
.ok_or("concat_is returned void".to_string())?; .ok_or("concat_is returned void".to_string())?;
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let h = cursor let h = cursor
.emit_instr(cur_bid, |b| b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")) .emit_instr(cur_bid, |b| {
b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(dst, h.into()); vmap.insert(dst, h.into());
handled_concat = true; handled_concat = true;
@ -227,39 +378,107 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
let out = if let (Some(_li0), Some(_ri0)) = (as_int(lv), as_int(rv)) { let out = if let (Some(_li0), Some(_ri0)) = (as_int(lv), as_int(rv)) {
// Localize integer operands into current block to satisfy dominance (normalize to i64) // Localize integer operands into current block to satisfy dominance (normalize to i64)
let li = resolver.resolve_i64(codegen, cursor, cur_bid, *lhs, bb_map, preds, block_end_values, vmap) let li = resolver
.resolve_i64(
codegen,
cursor,
cur_bid,
*lhs,
bb_map,
preds,
block_end_values,
vmap,
)
.unwrap_or_else(|_| codegen.context.i64_type().const_zero()); .unwrap_or_else(|_| codegen.context.i64_type().const_zero());
let ri = resolver.resolve_i64(codegen, cursor, cur_bid, *rhs, bb_map, preds, block_end_values, vmap) let ri = resolver
.resolve_i64(
codegen,
cursor,
cur_bid,
*rhs,
bb_map,
preds,
block_end_values,
vmap,
)
.unwrap_or_else(|_| codegen.context.i64_type().const_zero()); .unwrap_or_else(|_| codegen.context.i64_type().const_zero());
use BinaryOp as B; use BinaryOp as B;
match op { match op {
B::Add => cursor.emit_instr(cur_bid, |b| b.build_int_add(li, ri, "iadd")).map_err(|e| e.to_string())?.into(), B::Add => cursor
B::Sub => cursor.emit_instr(cur_bid, |b| b.build_int_sub(li, ri, "isub")).map_err(|e| e.to_string())?.into(), .emit_instr(cur_bid, |b| b.build_int_add(li, ri, "iadd"))
B::Mul => cursor.emit_instr(cur_bid, |b| b.build_int_mul(li, ri, "imul")).map_err(|e| e.to_string())?.into(), .map_err(|e| e.to_string())?
B::Div => cursor.emit_instr(cur_bid, |b| b.build_int_signed_div(li, ri, "idiv")).map_err(|e| e.to_string())?.into(), .into(),
B::Mod => cursor.emit_instr(cur_bid, |b| b.build_int_signed_rem(li, ri, "imod")).map_err(|e| e.to_string())?.into(), B::Sub => cursor
B::BitAnd => cursor.emit_instr(cur_bid, |b| b.build_and(li, ri, "iand")).map_err(|e| e.to_string())?.into(), .emit_instr(cur_bid, |b| b.build_int_sub(li, ri, "isub"))
B::BitOr => cursor.emit_instr(cur_bid, |b| b.build_or(li, ri, "ior")).map_err(|e| e.to_string())?.into(), .map_err(|e| e.to_string())?
B::BitXor => cursor.emit_instr(cur_bid, |b| b.build_xor(li, ri, "ixor")).map_err(|e| e.to_string())?.into(), .into(),
B::Shl => cursor.emit_instr(cur_bid, |b| b.build_left_shift(li, ri, "ishl")).map_err(|e| e.to_string())?.into(), B::Mul => cursor
B::Shr => cursor.emit_instr(cur_bid, |b| b.build_right_shift(li, ri, false, "ishr")).map_err(|e| e.to_string())?.into(), .emit_instr(cur_bid, |b| b.build_int_mul(li, ri, "imul"))
.map_err(|e| e.to_string())?
.into(),
B::Div => cursor
.emit_instr(cur_bid, |b| b.build_int_signed_div(li, ri, "idiv"))
.map_err(|e| e.to_string())?
.into(),
B::Mod => cursor
.emit_instr(cur_bid, |b| b.build_int_signed_rem(li, ri, "imod"))
.map_err(|e| e.to_string())?
.into(),
B::BitAnd => cursor
.emit_instr(cur_bid, |b| b.build_and(li, ri, "iand"))
.map_err(|e| e.to_string())?
.into(),
B::BitOr => cursor
.emit_instr(cur_bid, |b| b.build_or(li, ri, "ior"))
.map_err(|e| e.to_string())?
.into(),
B::BitXor => cursor
.emit_instr(cur_bid, |b| b.build_xor(li, ri, "ixor"))
.map_err(|e| e.to_string())?
.into(),
B::Shl => cursor
.emit_instr(cur_bid, |b| b.build_left_shift(li, ri, "ishl"))
.map_err(|e| e.to_string())?
.into(),
B::Shr => cursor
.emit_instr(cur_bid, |b| b.build_right_shift(li, ri, false, "ishr"))
.map_err(|e| e.to_string())?
.into(),
B::And | B::Or => { B::And | B::Or => {
// Treat as logical on integers: convert to i1 and and/or // Treat as logical on integers: convert to i1 and and/or
let lb = types::to_bool(codegen.context, li.into(), &codegen.builder)?; let lb = types::to_bool(codegen.context, li.into(), &codegen.builder)?;
let rb = types::to_bool(codegen.context, ri.into(), &codegen.builder)?; let rb = types::to_bool(codegen.context, ri.into(), &codegen.builder)?;
match op { match op {
B::And => cursor.emit_instr(cur_bid, |b| b.build_and(lb, rb, "land")).map_err(|e| e.to_string())?.into(), B::And => cursor
_ => cursor.emit_instr(cur_bid, |b| b.build_or(lb, rb, "lor")).map_err(|e| e.to_string())?.into(), .emit_instr(cur_bid, |b| b.build_and(lb, rb, "land"))
.map_err(|e| e.to_string())?
.into(),
_ => cursor
.emit_instr(cur_bid, |b| b.build_or(lb, rb, "lor"))
.map_err(|e| e.to_string())?
.into(),
} }
} }
} }
} else if let (Some(lf), Some(rf)) = (as_float(lv), as_float(rv)) { } else if let (Some(lf), Some(rf)) = (as_float(lv), as_float(rv)) {
use BinaryOp as B; use BinaryOp as B;
match op { match op {
B::Add => cursor.emit_instr(cur_bid, |b| b.build_float_add(lf, rf, "fadd")).map_err(|e| e.to_string())?.into(), B::Add => cursor
B::Sub => cursor.emit_instr(cur_bid, |b| b.build_float_sub(lf, rf, "fsub")).map_err(|e| e.to_string())?.into(), .emit_instr(cur_bid, |b| b.build_float_add(lf, rf, "fadd"))
B::Mul => cursor.emit_instr(cur_bid, |b| b.build_float_mul(lf, rf, "fmul")).map_err(|e| e.to_string())?.into(), .map_err(|e| e.to_string())?
B::Div => cursor.emit_instr(cur_bid, |b| b.build_float_div(lf, rf, "fdiv")).map_err(|e| e.to_string())?.into(), .into(),
B::Sub => cursor
.emit_instr(cur_bid, |b| b.build_float_sub(lf, rf, "fsub"))
.map_err(|e| e.to_string())?
.into(),
B::Mul => cursor
.emit_instr(cur_bid, |b| b.build_float_mul(lf, rf, "fmul"))
.map_err(|e| e.to_string())?
.into(),
B::Div => cursor
.emit_instr(cur_bid, |b| b.build_float_div(lf, rf, "fdiv"))
.map_err(|e| e.to_string())?
.into(),
B::Mod => return Err("fmod not supported yet".to_string()), B::Mod => return Err("fmod not supported yet".to_string()),
_ => return Err("bit/logic ops on float".to_string()), _ => return Err("bit/logic ops on float".to_string()),
} }
@ -273,14 +492,21 @@ pub(in super::super) fn lower_binop<'ctx, 'b>(
Ok(()) Ok(())
} }
fn guessed_zero<'ctx>(codegen: &CodegenContext<'ctx>, func: &MirFunction, vid: &ValueId) -> BasicValueEnum<'ctx> { fn guessed_zero<'ctx>(
codegen: &CodegenContext<'ctx>,
func: &MirFunction,
vid: &ValueId,
) -> BasicValueEnum<'ctx> {
use crate::mir::MirType as MT; use crate::mir::MirType as MT;
match func.metadata.value_types.get(vid) { match func.metadata.value_types.get(vid) {
Some(MT::Bool) => codegen.context.bool_type().const_zero().into(), Some(MT::Bool) => codegen.context.bool_type().const_zero().into(),
Some(MT::Integer) => codegen.context.i64_type().const_zero().into(), Some(MT::Integer) => codegen.context.i64_type().const_zero().into(),
Some(MT::Float) => codegen.context.f64_type().const_zero().into(), Some(MT::Float) => codegen.context.f64_type().const_zero().into(),
Some(MT::String) | Some(MT::Box(_)) | Some(MT::Array(_)) | Some(MT::Future(_)) | Some(MT::Unknown) | Some(MT::Void) | None => { Some(MT::String) | Some(MT::Box(_)) | Some(MT::Array(_)) | Some(MT::Future(_))
codegen.context.ptr_type(AddressSpace::from(0)).const_zero().into() | Some(MT::Unknown) | Some(MT::Void) | None => codegen
} .context
.ptr_type(AddressSpace::from(0))
.const_zero()
.into(),
} }
} }

View File

@ -2,9 +2,9 @@ use std::collections::HashMap;
use inkwell::values::BasicValueEnum as BVE; use inkwell::values::BasicValueEnum as BVE;
use super::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
use super::builder_cursor::BuilderCursor;
/// Handle ArrayBox fast-paths. Returns true if handled. /// Handle ArrayBox fast-paths. Returns true if handled.
pub(super) fn try_handle_array_method<'ctx, 'b>( pub(super) fn try_handle_array_method<'ctx, 'b>(
@ -19,9 +19,15 @@ pub(super) fn try_handle_array_method<'ctx, 'b>(
method: &str, method: &str,
args: &[ValueId], args: &[ValueId],
recv_h: inkwell::values::IntValue<'ctx>, recv_h: inkwell::values::IntValue<'ctx>,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<bool, String> { ) -> Result<bool, String> {
// Only when receiver is ArrayBox // Only when receiver is ArrayBox
let is_array = matches!(func.metadata.value_types.get(box_val), Some(crate::mir::MirType::Box(b)) if b == "ArrayBox") let is_array = matches!(func.metadata.value_types.get(box_val), Some(crate::mir::MirType::Box(b)) if b == "ArrayBox")
@ -38,14 +44,25 @@ pub(super) fn try_handle_array_method<'ctx, 'b>(
if args.len() != 1 { if args.len() != 1 {
return Err("ArrayBox.get expects 1 arg".to_string()); return Err("ArrayBox.get expects 1 arg".to_string());
} }
let idx_i = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let idx_i = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash_array_get_h") .get_function("nyash_array_get_h")
.unwrap_or_else(|| codegen.module.add_function("nyash_array_get_h", fnty, None)); .unwrap_or_else(|| codegen.module.add_function("nyash_array_get_h", fnty, None));
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), idx_i.into()], "aget")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), idx_i.into()], "aget")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -63,15 +80,35 @@ pub(super) fn try_handle_array_method<'ctx, 'b>(
if args.len() != 2 { if args.len() != 2 {
return Err("ArrayBox.set expects 2 arg".to_string()); return Err("ArrayBox.set expects 2 arg".to_string());
} }
let idx_i = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let idx_i = resolver.resolve_i64(
let val_i = resolver.resolve_i64(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?; codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let val_i = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i64t.into(), i64t.into(), i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i64t.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash_array_set_h") .get_function("nyash_array_set_h")
.unwrap_or_else(|| codegen.module.add_function("nyash_array_set_h", fnty, None)); .unwrap_or_else(|| codegen.module.add_function("nyash_array_set_h", fnty, None));
let _ = cursor let _ = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), idx_i.into(), val_i.into()], "aset")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), idx_i.into(), val_i.into()], "aset")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
Ok(true) Ok(true)
} }
@ -82,14 +119,29 @@ pub(super) fn try_handle_array_method<'ctx, 'b>(
if args.len() != 1 { if args.len() != 1 {
return Err("ArrayBox.push expects 1 arg".to_string()); return Err("ArrayBox.push expects 1 arg".to_string());
} }
let val_i = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let val_i = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash_array_push_h") .get_function("nyash_array_push_h")
.unwrap_or_else(|| codegen.module.add_function("nyash_array_push_h", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash_array_push_h", fnty, None)
});
let _ = cursor let _ = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), val_i.into()], "apush")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), val_i.into()], "apush")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
Ok(true) Ok(true)
} }
@ -104,7 +156,11 @@ pub(super) fn try_handle_array_method<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash_array_length_h") .get_function("nyash_array_length_h")
.unwrap_or_else(|| codegen.module.add_function("nyash_array_length_h", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash_array_length_h", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into()], "alen")) .emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into()], "alen"))
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;

View File

@ -2,8 +2,8 @@ use inkwell::basic_block::BasicBlock;
use inkwell::values::{BasicValueEnum, FunctionValue, PhiValue}; use inkwell::values::{BasicValueEnum, FunctionValue, PhiValue};
use std::collections::HashMap; use std::collections::HashMap;
use crate::backend::llvm::context::CodegenContext;
use super::super::types::map_mirtype_to_basic; use super::super::types::map_mirtype_to_basic;
use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
// Small, safe extraction: create LLVM basic blocks for a MIR function and // Small, safe extraction: create LLVM basic blocks for a MIR function and
@ -16,9 +16,10 @@ pub(in super::super) fn create_basic_blocks<'ctx>(
) -> (HashMap<BasicBlockId, BasicBlock<'ctx>>, BasicBlock<'ctx>) { ) -> (HashMap<BasicBlockId, BasicBlock<'ctx>>, BasicBlock<'ctx>) {
let mut bb_map: HashMap<BasicBlockId, BasicBlock> = HashMap::new(); let mut bb_map: HashMap<BasicBlockId, BasicBlock> = HashMap::new();
let entry_first = func.entry_block; let entry_first = func.entry_block;
let entry_bb = codegen let entry_bb = codegen.context.append_basic_block(
.context llvm_func,
.append_basic_block(llvm_func, &format!("{}_bb{}", fn_label, entry_first.as_u32())); &format!("{}_bb{}", fn_label, entry_first.as_u32()),
);
bb_map.insert(entry_first, entry_bb); bb_map.insert(entry_first, entry_bb);
for bid in func.block_ids() { for bid in func.block_ids() {
if bid == entry_first { if bid == entry_first {
@ -38,10 +39,7 @@ pub(in super::super) fn precreate_phis<'ctx>(
bb_map: &HashMap<BasicBlockId, BasicBlock<'ctx>>, bb_map: &HashMap<BasicBlockId, BasicBlock<'ctx>>,
vmap: &mut HashMap<ValueId, BasicValueEnum<'ctx>>, vmap: &mut HashMap<ValueId, BasicValueEnum<'ctx>>,
) -> Result< ) -> Result<
HashMap< HashMap<BasicBlockId, Vec<(ValueId, PhiValue<'ctx>, Vec<(BasicBlockId, ValueId)>)>>,
BasicBlockId,
Vec<(ValueId, PhiValue<'ctx>, Vec<(BasicBlockId, ValueId)>)>,
>,
String, String,
> { > {
use super::super::types::map_mirtype_to_basic; use super::super::types::map_mirtype_to_basic;
@ -63,15 +61,34 @@ pub(in super::super) fn precreate_phis<'ctx>(
// Prefer pointer when any input (or dst) is String/Box/Array/Future/Unknown // Prefer pointer when any input (or dst) is String/Box/Array/Future/Unknown
let mut wants_ptr = false; let mut wants_ptr = false;
if let Some(mt) = func.metadata.value_types.get(dst) { if let Some(mt) = func.metadata.value_types.get(dst) {
wants_ptr |= matches!(mt, crate::mir::MirType::String | crate::mir::MirType::Box(_) | crate::mir::MirType::Array(_) | crate::mir::MirType::Future(_) | crate::mir::MirType::Unknown); wants_ptr |= matches!(
mt,
crate::mir::MirType::String
| crate::mir::MirType::Box(_)
| crate::mir::MirType::Array(_)
| crate::mir::MirType::Future(_)
| crate::mir::MirType::Unknown
);
} }
for (_, iv) in inputs.iter() { for (_, iv) in inputs.iter() {
if let Some(mt) = func.metadata.value_types.get(iv) { if let Some(mt) = func.metadata.value_types.get(iv) {
wants_ptr |= matches!(mt, crate::mir::MirType::String | crate::mir::MirType::Box(_) | crate::mir::MirType::Array(_) | crate::mir::MirType::Future(_) | crate::mir::MirType::Unknown); wants_ptr |= matches!(
mt,
crate::mir::MirType::String
| crate::mir::MirType::Box(_)
| crate::mir::MirType::Array(_)
| crate::mir::MirType::Future(_)
| crate::mir::MirType::Unknown
);
} }
} }
if wants_ptr { if wants_ptr {
phi_ty = Some(codegen.context.ptr_type(inkwell::AddressSpace::from(0)).into()); phi_ty = Some(
codegen
.context
.ptr_type(inkwell::AddressSpace::from(0))
.into(),
);
} else if let Some(mt) = func.metadata.value_types.get(dst) { } else if let Some(mt) = func.metadata.value_types.get(dst) {
phi_ty = Some(map_mirtype_to_basic(codegen.context, mt)); phi_ty = Some(map_mirtype_to_basic(codegen.context, mt));
} else if let Some((_, iv)) = inputs.first() { } else if let Some((_, iv)) = inputs.first() {

View File

@ -1,17 +1,17 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::AddressSpace;
use inkwell::values::BasicValueEnum as BVE; use inkwell::values::BasicValueEnum as BVE;
use inkwell::AddressSpace;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
mod fields; mod fields;
pub(crate) mod invoke; pub(crate) mod invoke;
mod marshal; mod marshal;
use self::marshal as marshal_mod;
use self::invoke as invoke_mod; use self::invoke as invoke_mod;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use self::marshal as marshal_mod;
use super::builder_cursor::BuilderCursor; use super::builder_cursor::BuilderCursor;
use super::ctx::{LowerFnCtx, BlockCtx}; use super::ctx::{BlockCtx, LowerFnCtx};
use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
// BoxCall lowering (large): mirrors existing logic; kept in one function for now // BoxCall lowering (large): mirrors existing logic; kept in one function for now
pub(in super::super) fn lower_boxcall<'ctx, 'b>( pub(in super::super) fn lower_boxcall<'ctx, 'b>(
@ -28,24 +28,45 @@ pub(in super::super) fn lower_boxcall<'ctx, 'b>(
args: &[ValueId], args: &[ValueId],
box_type_ids: &HashMap<String, i64>, box_type_ids: &HashMap<String, i64>,
entry_builder: &inkwell::builder::Builder<'ctx>, entry_builder: &inkwell::builder::Builder<'ctx>,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
use crate::backend::llvm::compiler::helpers::{as_float, as_int};
use super::super::types::classify_tag; use super::super::types::classify_tag;
use crate::backend::llvm::compiler::helpers::{as_float, as_int};
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
// Resolve receiver as handle and pointer (i8*) via Resolver to ensure dominance safety // Resolve receiver as handle and pointer (i8*) via Resolver to ensure dominance safety
let recv_h = resolver.resolve_i64( let recv_h = resolver.resolve_i64(
codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap, codegen,
cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?; )?;
let recv_p = resolver.resolve_ptr( let recv_p = resolver.resolve_ptr(
codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap, codegen,
cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?; )?;
let recv_v: BVE = recv_p.into(); let recv_v: BVE = recv_p.into();
// Resolve type_id // Resolve type_id
let type_id: i64 = if let Some(crate::mir::MirType::Box(bname)) = func.metadata.value_types.get(box_val) { let type_id: i64 =
if let Some(crate::mir::MirType::Box(bname)) = func.metadata.value_types.get(box_val) {
*box_type_ids.get(bname).unwrap_or(&0) *box_type_ids.get(bname).unwrap_or(&0)
} else if let Some(crate::mir::MirType::String) = func.metadata.value_types.get(box_val) { } else if let Some(crate::mir::MirType::String) = func.metadata.value_types.get(box_val) {
*box_type_ids.get("StringBox").unwrap_or(&0) *box_type_ids.get("StringBox").unwrap_or(&0)
@ -55,19 +76,47 @@ pub(in super::super) fn lower_boxcall<'ctx, 'b>(
// Delegate String methods // Delegate String methods
if super::strings::try_handle_string_method( if super::strings::try_handle_string_method(
codegen, cursor, resolver, cur_bid, func, vmap, dst, box_val, method, args, codegen,
bb_map, preds, block_end_values, cursor,
resolver,
cur_bid,
func,
vmap,
dst,
box_val,
method,
args,
bb_map,
preds,
block_end_values,
)? { )? {
return Ok(()); return Ok(());
} }
// Delegate Map methods first (to avoid Array fallback catching get/set ambiguously) // Delegate Map methods first (to avoid Array fallback catching get/set ambiguously)
if super::maps::try_handle_map_method(codegen, cursor, resolver, cur_bid, func, vmap, dst, box_val, method, args, recv_h)? { if super::maps::try_handle_map_method(
codegen, cursor, resolver, cur_bid, func, vmap, dst, box_val, method, args, recv_h,
)? {
return Ok(()); return Ok(());
} }
// Delegate Array methods // Delegate Array methods
if super::arrays::try_handle_array_method(codegen, cursor, resolver, cur_bid, func, vmap, dst, box_val, method, args, recv_h, bb_map, preds, block_end_values)? { if super::arrays::try_handle_array_method(
codegen,
cursor,
resolver,
cur_bid,
func,
vmap,
dst,
box_val,
method,
args,
recv_h,
bb_map,
preds,
block_end_values,
)? {
return Ok(()); return Ok(());
} }
@ -114,9 +163,15 @@ pub(in super::super) fn lower_boxcall<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash_array_length_h") .get_function("nyash_array_length_h")
.unwrap_or_else(|| codegen.module.add_function("nyash_array_length_h", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash_array_length_h", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into()], "alen_fallback")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into()], "alen_fallback")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -163,42 +218,116 @@ pub(in super::super) fn lower_boxcall<'ctx, 'b>(
// Sanitize symbol the same way as codegen/mod.rs does // Sanitize symbol the same way as codegen/mod.rs does
let sym: String = { let sym: String = {
let mut s = String::from("ny_f_"); let mut s = String::from("ny_f_");
s.push_str(&candidate.replace('.', "_").replace('/', "_").replace('-', "_")); s.push_str(
&candidate
.replace('.', "_")
.replace('/', "_")
.replace('-', "_"),
);
s s
}; };
if let Some(callee) = codegen.module.get_function(&sym) { if let Some(callee) = codegen.module.get_function(&sym) {
// Coerce arguments to callee parameter types // Coerce arguments to callee parameter types
let exp_tys = callee.get_type().get_param_types(); let exp_tys = callee.get_type().get_param_types();
if exp_tys.len() != args.len() { return Err("boxcall direct-call: arg count mismatch".to_string()); } if exp_tys.len() != args.len() {
let mut call_args: Vec<inkwell::values::BasicMetadataValueEnum> = Vec::with_capacity(args.len()); return Err("boxcall direct-call: arg count mismatch".to_string());
}
let mut call_args: Vec<inkwell::values::BasicMetadataValueEnum> =
Vec::with_capacity(args.len());
for (i, a) in args.iter().enumerate() { for (i, a) in args.iter().enumerate() {
use inkwell::types::BasicMetadataTypeEnum as BMTy; use inkwell::types::BasicMetadataTypeEnum as BMTy;
let coerced: BVE<'ctx> = match exp_tys[i] { let coerced: BVE<'ctx> = match exp_tys[i] {
BMTy::IntType(it) => { BMTy::IntType(it) => {
// Use Resolver via our surrounding lowering // Use Resolver via our surrounding lowering
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *a, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*a,
bb_map,
preds,
block_end_values,
vmap,
)?;
let bw_dst = it.get_bit_width(); let bw_dst = it.get_bit_width();
let bw_src = iv.get_type().get_bit_width(); let bw_src = iv.get_type().get_bit_width();
if bw_src == bw_dst { iv.into() } if bw_src == bw_dst {
else if bw_src < bw_dst { cursor.emit_instr(cur_bid, |b| b.build_int_z_extend(iv, it, "boxcall_arg_zext")).map_err(|e| e.to_string())?.into() } iv.into()
else if bw_dst == 1 { super::super::types::to_bool(codegen.context, iv.into(), &codegen.builder)?.into() } } else if bw_src < bw_dst {
else { cursor.emit_instr(cur_bid, |b| b.build_int_truncate(iv, it, "boxcall_arg_trunc")).map_err(|e| e.to_string())?.into() } cursor
.emit_instr(cur_bid, |b| {
b.build_int_z_extend(iv, it, "boxcall_arg_zext")
})
.map_err(|e| e.to_string())?
.into()
} else if bw_dst == 1 {
super::super::types::to_bool(
codegen.context,
iv.into(),
&codegen.builder,
)?
.into()
} else {
cursor
.emit_instr(cur_bid, |b| {
b.build_int_truncate(iv, it, "boxcall_arg_trunc")
})
.map_err(|e| e.to_string())?
.into()
}
} }
BMTy::PointerType(pt) => { BMTy::PointerType(pt) => {
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *a, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
let p = cursor.emit_instr(cur_bid, |b| b.build_int_to_ptr(iv, pt, "boxcall_arg_i2p")).map_err(|e| e.to_string())?; codegen,
cursor,
cur_bid,
*a,
bb_map,
preds,
block_end_values,
vmap,
)?;
let p = cursor
.emit_instr(cur_bid, |b| {
b.build_int_to_ptr(iv, pt, "boxcall_arg_i2p")
})
.map_err(|e| e.to_string())?;
p.into() p.into()
} }
BMTy::FloatType(ft) => { BMTy::FloatType(ft) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, *a, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
if fv.get_type() == ft { fv.into() } else { cursor.emit_instr(cur_bid, |b| b.build_float_cast(fv, ft, "boxcall_arg_fcast")).map_err(|e| e.to_string())?.into() } codegen,
cursor,
cur_bid,
*a,
bb_map,
preds,
block_end_values,
vmap,
)?;
if fv.get_type() == ft {
fv.into()
} else {
cursor
.emit_instr(cur_bid, |b| {
b.build_float_cast(fv, ft, "boxcall_arg_fcast")
})
.map_err(|e| e.to_string())?
.into()
}
}
_ => {
return Err(
"boxcall direct-call: unsupported parameter type".to_string()
)
} }
_ => return Err("boxcall direct-call: unsupported parameter type".to_string()),
}; };
call_args.push(coerced.into()); call_args.push(coerced.into());
} }
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &call_args, "user_meth_call")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &call_args, "user_meth_call")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
if let Some(rv) = call.try_as_basic_value().left() { if let Some(rv) = call.try_as_basic_value().left() {
@ -216,22 +345,66 @@ pub(in super::super) fn lower_boxcall<'ctx, 'b>(
.emit_instr(cur_bid, |b| b.build_global_string_ptr(method, "meth_name")) .emit_instr(cur_bid, |b| b.build_global_string_ptr(method, "meth_name"))
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
// up to 2 args for this minimal path // up to 2 args for this minimal path
let a1 = if let Some(v0) = args.get(0) { resolver.resolve_i64(codegen, cursor, cur_bid, *v0, bb_map, preds, block_end_values, vmap)? } else { i64t.const_zero() }; let a1 = if let Some(v0) = args.get(0) {
let a2 = if let Some(v1) = args.get(1) { resolver.resolve_i64(codegen, cursor, cur_bid, *v1, bb_map, preds, block_end_values, vmap)? } else { i64t.const_zero() }; resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*v0,
bb_map,
preds,
block_end_values,
vmap,
)?
} else {
i64t.const_zero()
};
let a2 = if let Some(v1) = args.get(1) {
resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*v1,
bb_map,
preds,
block_end_values,
vmap,
)?
} else {
i64t.const_zero()
};
let fnty = i64t.fn_type( let fnty = i64t.fn_type(
&[ &[
i64t.into(), // recv handle i64t.into(), // recv handle
codegen.context.ptr_type(AddressSpace::from(0)).into(), // method cstr codegen.context.ptr_type(AddressSpace::from(0)).into(), // method cstr
i64t.into(), i64t.into(), i64t.into(), // argc, a1, a2 i64t.into(),
i64t.into(),
i64t.into(), // argc, a1, a2
], ],
false, false,
); );
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.plugin.invoke_by_name_i64") .get_function("nyash.plugin.invoke_by_name_i64")
.unwrap_or_else(|| codegen.module.add_function("nyash.plugin.invoke_by_name_i64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.plugin.invoke_by_name_i64", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), mname.as_pointer_value().into(), argc.into(), a1.into(), a2.into()], "pinvoke_by_name")) .emit_instr(cur_bid, |b| {
b.build_call(
callee,
&[
recv_h.into(),
mname.as_pointer_value().into(),
argc.into(),
a1.into(),
a2.into(),
],
"pinvoke_by_name",
)
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -241,28 +414,66 @@ pub(in super::super) fn lower_boxcall<'ctx, 'b>(
// Inline minimal return normalization similar to store_invoke_return() // Inline minimal return normalization similar to store_invoke_return()
if let Some(mt) = func.metadata.value_types.get(d) { if let Some(mt) = func.metadata.value_types.get(d) {
match mt { match mt {
crate::mir::MirType::Integer => { vmap.insert(*d, rv); } crate::mir::MirType::Integer => {
vmap.insert(*d, rv);
}
crate::mir::MirType::Bool => { crate::mir::MirType::Bool => {
if let BVE::IntValue(iv) = rv { if let BVE::IntValue(iv) = rv {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let zero = i64t.const_zero(); let zero = i64t.const_zero();
let b1 = cursor.emit_instr(cur_bid, |bd| bd.build_int_compare(inkwell::IntPredicate::NE, iv, zero, "bool_i64_to_i1")).map_err(|e| e.to_string())?; let b1 = cursor
.emit_instr(cur_bid, |bd| {
bd.build_int_compare(
inkwell::IntPredicate::NE,
iv,
zero,
"bool_i64_to_i1",
)
})
.map_err(|e| e.to_string())?;
vmap.insert(*d, b1.into()); vmap.insert(*d, b1.into());
} else { vmap.insert(*d, rv); } } else {
vmap.insert(*d, rv);
}
} }
crate::mir::MirType::String => { crate::mir::MirType::String => {
if let BVE::IntValue(iv) = rv { if let BVE::IntValue(iv) = rv {
let p = cursor.emit_instr(cur_bid, |bd| bd.build_int_to_ptr(iv, codegen.context.ptr_type(AddressSpace::from(0)), "str_h2p_ret")).map_err(|e| e.to_string())?; let p = cursor
.emit_instr(cur_bid, |bd| {
bd.build_int_to_ptr(
iv,
codegen.context.ptr_type(AddressSpace::from(0)),
"str_h2p_ret",
)
})
.map_err(|e| e.to_string())?;
vmap.insert(*d, p.into()); vmap.insert(*d, p.into());
} else { vmap.insert(*d, rv); } } else {
vmap.insert(*d, rv);
} }
crate::mir::MirType::Box(_) | crate::mir::MirType::Array(_) | crate::mir::MirType::Future(_) | crate::mir::MirType::Unknown => { }
crate::mir::MirType::Box(_)
| crate::mir::MirType::Array(_)
| crate::mir::MirType::Future(_)
| crate::mir::MirType::Unknown => {
if let BVE::IntValue(iv) = rv { if let BVE::IntValue(iv) = rv {
let p = cursor.emit_instr(cur_bid, |bd| bd.build_int_to_ptr(iv, codegen.context.ptr_type(AddressSpace::from(0)), "h2p_ret")).map_err(|e| e.to_string())?; let p = cursor
.emit_instr(cur_bid, |bd| {
bd.build_int_to_ptr(
iv,
codegen.context.ptr_type(AddressSpace::from(0)),
"h2p_ret",
)
})
.map_err(|e| e.to_string())?;
vmap.insert(*d, p.into()); vmap.insert(*d, p.into());
} else { vmap.insert(*d, rv); } } else {
vmap.insert(*d, rv);
}
}
_ => {
vmap.insert(*d, rv);
} }
_ => { vmap.insert(*d, rv); }
} }
} else { } else {
vmap.insert(*d, rv); vmap.insert(*d, rv);
@ -286,7 +497,9 @@ pub(in super::super) fn lower_boxcall_boxed<'ctx, 'b>(
entry_builder: &inkwell::builder::Builder<'ctx>, entry_builder: &inkwell::builder::Builder<'ctx>,
) -> Result<(), String> { ) -> Result<(), String> {
// Optional dev check: ensure block is open for insertion // Optional dev check: ensure block is open for insertion
if ctx.dev_checks { ctx.cursor.assert_open(blk.cur_bid); } if ctx.dev_checks {
ctx.cursor.assert_open(blk.cur_bid);
}
lower_boxcall( lower_boxcall(
ctx.codegen, ctx.codegen,
ctx.cursor, ctx.cursor,
@ -299,7 +512,8 @@ pub(in super::super) fn lower_boxcall_boxed<'ctx, 'b>(
method, method,
method_id, method_id,
args, args,
ctx.box_type_ids.ok_or_else(|| "LowerFnCtx.box_type_ids missing".to_string())?, ctx.box_type_ids
.ok_or_else(|| "LowerFnCtx.box_type_ids missing".to_string())?,
entry_builder, entry_builder,
ctx.bb_map, ctx.bb_map,
ctx.preds, ctx.preds,
@ -322,9 +536,15 @@ pub(in super::super) fn lower_boxcall_via_ctx<'ctx, 'b>(
args: &[ValueId], args: &[ValueId],
box_type_ids: &'b HashMap<String, i64>, box_type_ids: &'b HashMap<String, i64>,
entry_builder: &inkwell::builder::Builder<'ctx>, entry_builder: &inkwell::builder::Builder<'ctx>,
bb_map: &'b std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &'b std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &'b std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &'b std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &'b std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &'b std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
let llbb = *bb_map.get(&cur_bid).ok_or("missing cur bb")?; let llbb = *bb_map.get(&cur_bid).ok_or("missing cur bb")?;
let blkctx = BlockCtx::new(cur_bid, llbb); let blkctx = BlockCtx::new(cur_bid, llbb);
@ -364,18 +584,45 @@ fn coerce_to_type<'ctx>(
if bw_src == bw_dst { if bw_src == bw_dst {
Ok(iv.into()) Ok(iv.into())
} else if bw_src < bw_dst { } else if bw_src < bw_dst {
Ok(codegen.builder.build_int_z_extend(iv, it, "bc_zext").map_err(|e| e.to_string())?.into()) Ok(codegen
.builder
.build_int_z_extend(iv, it, "bc_zext")
.map_err(|e| e.to_string())?
.into())
} else if bw_dst == 1 { } else if bw_dst == 1 {
Ok(super::super::types::to_bool(codegen.context, iv.into(), &codegen.builder)?.into()) Ok(
super::super::types::to_bool(codegen.context, iv.into(), &codegen.builder)?
.into(),
)
} else { } else {
Ok(codegen.builder.build_int_truncate(iv, it, "bc_trunc").map_err(|e| e.to_string())?.into()) Ok(codegen
.builder
.build_int_truncate(iv, it, "bc_trunc")
.map_err(|e| e.to_string())?
.into())
} }
} }
(inkwell::values::BasicValueEnum::PointerValue(pv), BT::IntType(it)) => Ok(codegen.builder.build_ptr_to_int(pv, it, "bc_p2i").map_err(|e| e.to_string())?.into()), (inkwell::values::BasicValueEnum::PointerValue(pv), BT::IntType(it)) => Ok(codegen
(inkwell::values::BasicValueEnum::FloatValue(fv), BT::IntType(it)) => Ok(codegen.builder.build_float_to_signed_int(fv, it, "bc_f2i").map_err(|e| e.to_string())?.into()), .builder
(inkwell::values::BasicValueEnum::IntValue(iv), BT::PointerType(pt)) => Ok(codegen.builder.build_int_to_ptr(iv, pt, "bc_i2p").map_err(|e| e.to_string())?.into()), .build_ptr_to_int(pv, it, "bc_p2i")
.map_err(|e| e.to_string())?
.into()),
(inkwell::values::BasicValueEnum::FloatValue(fv), BT::IntType(it)) => Ok(codegen
.builder
.build_float_to_signed_int(fv, it, "bc_f2i")
.map_err(|e| e.to_string())?
.into()),
(inkwell::values::BasicValueEnum::IntValue(iv), BT::PointerType(pt)) => Ok(codegen
.builder
.build_int_to_ptr(iv, pt, "bc_i2p")
.map_err(|e| e.to_string())?
.into()),
(inkwell::values::BasicValueEnum::PointerValue(pv), BT::PointerType(_)) => Ok(pv.into()), (inkwell::values::BasicValueEnum::PointerValue(pv), BT::PointerType(_)) => Ok(pv.into()),
(inkwell::values::BasicValueEnum::IntValue(iv), BT::FloatType(ft)) => Ok(codegen.builder.build_signed_int_to_float(iv, ft, "bc_i2f").map_err(|e| e.to_string())?.into()), (inkwell::values::BasicValueEnum::IntValue(iv), BT::FloatType(ft)) => Ok(codegen
.builder
.build_signed_int_to_float(iv, ft, "bc_i2f")
.map_err(|e| e.to_string())?
.into()),
(inkwell::values::BasicValueEnum::FloatValue(fv), BT::FloatType(_)) => Ok(fv.into()), (inkwell::values::BasicValueEnum::FloatValue(fv), BT::FloatType(_)) => Ok(fv.into()),
(v, _) => Ok(v), (v, _) => Ok(v),
} }

View File

@ -1,9 +1,9 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::{values::BasicValueEnum as BVE, AddressSpace}; use super::super::ctx::{BlockCtx, LowerFnCtx};
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::ValueId; use crate::mir::ValueId;
use super::super::ctx::{LowerFnCtx, BlockCtx}; use inkwell::{values::BasicValueEnum as BVE, AddressSpace};
/// Handle getField/setField; returns true if handled. /// Handle getField/setField; returns true if handled.
use super::super::builder_cursor::BuilderCursor; use super::super::builder_cursor::BuilderCursor;
@ -18,9 +18,15 @@ pub(super) fn try_handle_field_method<'ctx, 'b>(
args: &[ValueId], args: &[ValueId],
recv_h: inkwell::values::IntValue<'ctx>, recv_h: inkwell::values::IntValue<'ctx>,
resolver: &mut super::super::Resolver<'ctx>, resolver: &mut super::super::Resolver<'ctx>,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<bool, String> { ) -> Result<bool, String> {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
match method { match method {
@ -28,16 +34,30 @@ pub(super) fn try_handle_field_method<'ctx, 'b>(
if args.len() != 1 { if args.len() != 1 {
return Err("getField expects 1 arg (name)".to_string()); return Err("getField expects 1 arg (name)".to_string());
} }
let name_p = resolver let name_p = resolver.resolve_ptr(
.resolve_ptr(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let i8p = codegen.context.ptr_type(AddressSpace::from(0)); let i8p = codegen.context.ptr_type(AddressSpace::from(0));
let fnty = i64t.fn_type(&[i64t.into(), i8p.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.instance.get_field_h") .get_function("nyash.instance.get_field_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.instance.get_field_h", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.instance.get_field_h", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), name_p.into()], "getField")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), name_p.into()], "getField")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -62,18 +82,44 @@ pub(super) fn try_handle_field_method<'ctx, 'b>(
if args.len() != 2 { if args.len() != 2 {
return Err("setField expects 2 args (name, value)".to_string()); return Err("setField expects 2 args (name, value)".to_string());
} }
let name_p = resolver let name_p = resolver.resolve_ptr(
.resolve_ptr(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; codegen,
let val_h = resolver cursor,
.resolve_i64(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?; cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let val_h = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?;
let i8p = codegen.context.ptr_type(AddressSpace::from(0)); let i8p = codegen.context.ptr_type(AddressSpace::from(0));
let fnty = i64t.fn_type(&[i64t.into(), i8p.into(), i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i8p.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.instance.set_field_h") .get_function("nyash.instance.set_field_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.instance.set_field_h", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.instance.set_field_h", fnty, None)
});
let _ = cursor let _ = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), name_p.into(), val_h.into()], "setField")) .emit_instr(cur_bid, |b| {
b.build_call(
callee,
&[recv_h.into(), name_p.into(), val_h.into()],
"setField",
)
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
Ok(true) Ok(true)
} }

View File

@ -2,9 +2,9 @@ use std::collections::HashMap;
use inkwell::{values::BasicValueEnum as BVE, AddressSpace}; use inkwell::{values::BasicValueEnum as BVE, AddressSpace};
use super::super::ctx::{BlockCtx, LowerFnCtx};
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, ValueId}; use crate::mir::{function::MirFunction, ValueId};
use super::super::ctx::{LowerFnCtx, BlockCtx};
// use super::marshal::{get_i64, get_tag_const}; // use super::marshal::{get_i64, get_tag_const};
@ -22,9 +22,15 @@ pub(super) fn try_handle_tagged_invoke<'ctx, 'b>(
args: &[ValueId], args: &[ValueId],
entry_builder: &inkwell::builder::Builder<'ctx>, entry_builder: &inkwell::builder::Builder<'ctx>,
cur_bid: crate::mir::BasicBlockId, cur_bid: crate::mir::BasicBlockId,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let argc_val = i64t.const_int(args.len() as u64, false); let argc_val = i64t.const_int(args.len() as u64, false);
@ -35,13 +41,44 @@ pub(super) fn try_handle_tagged_invoke<'ctx, 'b>(
for (i, vid) in args.iter().enumerate() { for (i, vid) in args.iter().enumerate() {
let iv = match func.metadata.value_types.get(vid) { let iv = match func.metadata.value_types.get(vid) {
Some(crate::mir::MirType::Float) => { Some(crate::mir::MirType::Float) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, *vid, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
*vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false); let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false);
let callee = codegen.module.get_function("nyash.box.from_f64").unwrap_or_else(|| codegen.module.add_function("nyash.box.from_f64", fnty, None)); let callee = codegen
let call = codegen.builder.build_call(callee, &[fv.into()], "arg_f2h").map_err(|e| e.to_string())?; .module
call.try_as_basic_value().left().ok_or("from_f64 returned void".to_string())?.into_int_value() .get_function("nyash.box.from_f64")
.unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_f64", fnty, None)
});
let call = codegen
.builder
.build_call(callee, &[fv.into()], "arg_f2h")
.map_err(|e| e.to_string())?;
call.try_as_basic_value()
.left()
.ok_or("from_f64 returned void".to_string())?
.into_int_value()
} }
_ => resolver.resolve_i64(codegen, cursor, cur_bid, *vid, bb_map, preds, block_end_values, vmap)?, _ => resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*vid,
bb_map,
preds,
block_end_values,
vmap,
)?,
}; };
a[i] = iv; a[i] = iv;
} }
@ -49,23 +86,40 @@ pub(super) fn try_handle_tagged_invoke<'ctx, 'b>(
for (i, vid) in args.iter().enumerate() { for (i, vid) in args.iter().enumerate() {
let tag = match func.metadata.value_types.get(vid) { let tag = match func.metadata.value_types.get(vid) {
Some(crate::mir::MirType::Float) => 5, Some(crate::mir::MirType::Float) => 5,
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) | Some(crate::mir::MirType::Array(_)) | Some(crate::mir::MirType::Future(_)) | Some(crate::mir::MirType::Unknown) => 8, Some(crate::mir::MirType::String)
| Some(crate::mir::MirType::Box(_))
| Some(crate::mir::MirType::Array(_))
| Some(crate::mir::MirType::Future(_))
| Some(crate::mir::MirType::Unknown) => 8,
_ => 3, _ => 3,
}; };
tags[i] = i64t.const_int(tag as u64, false); tags[i] = i64t.const_int(tag as u64, false);
} }
let fnty = i64t.fn_type( let fnty = i64t.fn_type(
&[ &[
i64t.into(), i64t.into(), i64t.into(), i64t.into(), i64t.into(),
i64t.into(), i64t.into(), i64t.into(), i64t.into(), i64t.into(),
i64t.into(), i64t.into(), i64t.into(), i64t.into(), i64t.into(),
i64t.into(),
i64t.into(),
i64t.into(),
i64t.into(),
i64t.into(),
i64t.into(),
i64t.into(),
i64t.into(),
i64t.into(),
], ],
false, false,
); );
let callee = codegen let callee = codegen
.module .module
.get_function("nyash_plugin_invoke3_tagged_i64") .get_function("nyash_plugin_invoke3_tagged_i64")
.unwrap_or_else(|| codegen.module.add_function("nyash_plugin_invoke3_tagged_i64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash_plugin_invoke3_tagged_i64", fnty, None)
});
let tid = i64t.const_int(type_id as u64, true); let tid = i64t.const_int(type_id as u64, true);
let midv = i64t.const_int(mid as u64, false); let midv = i64t.const_int(mid as u64, false);
let call = codegen let call = codegen
@ -73,9 +127,18 @@ pub(super) fn try_handle_tagged_invoke<'ctx, 'b>(
.build_call( .build_call(
callee, callee,
&[ &[
tid.into(), midv.into(), argc_val.into(), recv_h.into(), tid.into(),
a[0].into(), tags[0].into(), a[1].into(), tags[1].into(), midv.into(),
a[2].into(), tags[2].into(), a[3].into(), tags[3].into(), argc_val.into(),
recv_h.into(),
a[0].into(),
tags[0].into(),
a[1].into(),
tags[1].into(),
a[2].into(),
tags[2].into(),
a[3].into(),
tags[3].into(),
], ],
"pinvoke_tagged", "pinvoke_tagged",
) )
@ -118,21 +181,62 @@ pub(super) fn try_handle_tagged_invoke<'ctx, 'b>(
}; };
let vi = match func.metadata.value_types.get(vid) { let vi = match func.metadata.value_types.get(vid) {
Some(crate::mir::MirType::Float) => { Some(crate::mir::MirType::Float) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, *vid, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
*vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false); let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false);
let callee = codegen.module.get_function("nyash.box.from_f64").unwrap_or_else(|| codegen.module.add_function("nyash.box.from_f64", fnty, None)); let callee = codegen
let call = codegen.builder.build_call(callee, &[fv.into()], "arg_f2h").map_err(|e| e.to_string())?; .module
call.try_as_basic_value().left().ok_or("from_f64 returned void".to_string())?.into_int_value() .get_function("nyash.box.from_f64")
.unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_f64", fnty, None)
});
let call = codegen
.builder
.build_call(callee, &[fv.into()], "arg_f2h")
.map_err(|e| e.to_string())?;
call.try_as_basic_value()
.left()
.ok_or("from_f64 returned void".to_string())?
.into_int_value()
} }
_ => resolver.resolve_i64(codegen, cursor, cur_bid, *vid, bb_map, preds, block_end_values, vmap)?, _ => resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*vid,
bb_map,
preds,
block_end_values,
vmap,
)?,
}; };
let ti = match func.metadata.value_types.get(vid) { let ti = match func.metadata.value_types.get(vid) {
Some(crate::mir::MirType::Float) => i64t.const_int(5, false), Some(crate::mir::MirType::Float) => i64t.const_int(5, false),
Some(crate::mir::MirType::String) | Some(crate::mir::MirType::Box(_)) | Some(crate::mir::MirType::Array(_)) | Some(crate::mir::MirType::Future(_)) | Some(crate::mir::MirType::Unknown) => i64t.const_int(8, false), Some(crate::mir::MirType::String)
| Some(crate::mir::MirType::Box(_))
| Some(crate::mir::MirType::Array(_))
| Some(crate::mir::MirType::Future(_))
| Some(crate::mir::MirType::Unknown) => i64t.const_int(8, false),
_ => i64t.const_int(3, false), _ => i64t.const_int(3, false),
}; };
codegen.builder.build_store(gep_v, vi).map_err(|e| e.to_string())?; codegen
codegen.builder.build_store(gep_t, ti).map_err(|e| e.to_string())?; .builder
.build_store(gep_v, vi)
.map_err(|e| e.to_string())?;
codegen
.builder
.build_store(gep_t, ti)
.map_err(|e| e.to_string())?;
} }
let vals_ptr = codegen let vals_ptr = codegen
.builder .builder
@ -164,14 +268,25 @@ pub(super) fn try_handle_tagged_invoke<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.plugin.invoke_tagged_v_i64") .get_function("nyash.plugin.invoke_tagged_v_i64")
.unwrap_or_else(|| codegen.module.add_function("nyash.plugin.invoke_tagged_v_i64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.plugin.invoke_tagged_v_i64", fnty, None)
});
let tid = i64t.const_int(type_id as u64, true); let tid = i64t.const_int(type_id as u64, true);
let midv = i64t.const_int(mid as u64, false); let midv = i64t.const_int(mid as u64, false);
let call = codegen let call = codegen
.builder .builder
.build_call( .build_call(
callee, callee,
&[tid.into(), midv.into(), argc_val.into(), recv_h.into(), vals_ptr.into(), tags_ptr.into()], &[
tid.into(),
midv.into(),
argc_val.into(),
recv_h.into(),
vals_ptr.into(),
tags_ptr.into(),
],
"pinvoke_tagged_v", "pinvoke_tagged_v",
) )
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
@ -213,8 +328,11 @@ fn store_invoke_return<'ctx>(
} }
crate::mir::MirType::String => { crate::mir::MirType::String => {
// Keep as i64 handle across blocks (pointer is produced on demand via Resolver) // Keep as i64 handle across blocks (pointer is produced on demand via Resolver)
if let BVE::IntValue(iv) = rv { vmap.insert(dst, iv.into()); } if let BVE::IntValue(iv) = rv {
else { return Err("invoke ret expected i64 for String".to_string()); } vmap.insert(dst, iv.into());
} else {
return Err("invoke ret expected i64 for String".to_string());
}
} }
crate::mir::MirType::Box(_) crate::mir::MirType::Box(_)
| crate::mir::MirType::Array(_) | crate::mir::MirType::Array(_)

View File

@ -1,10 +1,10 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::{values::BasicValueEnum as BVE, AddressSpace};
use crate::backend::llvm::compiler::codegen::types; use crate::backend::llvm::compiler::codegen::types;
use inkwell::{values::BasicValueEnum as BVE, AddressSpace};
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{ValueId, function::MirFunction}; use crate::mir::{function::MirFunction, ValueId};
/// Convert a value to i64 handle/int for plugin invoke (ptr->i64, f64->box->i64) /// Convert a value to i64 handle/int for plugin invoke (ptr->i64, f64->box->i64)
pub(super) fn get_i64<'ctx, 'b>( pub(super) fn get_i64<'ctx, 'b>(
@ -15,30 +15,64 @@ pub(super) fn get_i64<'ctx, 'b>(
func: &MirFunction, func: &MirFunction,
vmap: &HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>, vmap: &HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
vid: ValueId, vid: ValueId,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<inkwell::values::IntValue<'ctx>, String> { ) -> Result<inkwell::values::IntValue<'ctx>, String> {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
match func.metadata.value_types.get(&vid) { match func.metadata.value_types.get(&vid) {
Some(crate::mir::MirType::Float) => { Some(crate::mir::MirType::Float) => {
// Box f64 then use its handle // Box f64 then use its handle
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, vid, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false); let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_f64") .get_function("nyash.box.from_f64")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_f64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_f64", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[fv.into()], "arg_f64_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[fv.into()], "arg_f64_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
.left() .left()
.ok_or("from_f64 returned void".to_string())?; .ok_or("from_f64 returned void".to_string())?;
if let BVE::IntValue(h) = rv { Ok(h) } else { Err("from_f64 ret expected i64".to_string()) } if let BVE::IntValue(h) = rv {
Ok(h)
} else {
Err("from_f64 ret expected i64".to_string())
} }
_ => resolver.resolve_i64(codegen, cursor, cur_bid, vid, bb_map, preds, block_end_values, vmap), }
_ => resolver.resolve_i64(
codegen,
cursor,
cur_bid,
vid,
bb_map,
preds,
block_end_values,
vmap,
),
} }
} }

View File

@ -1,9 +1,6 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::{ use inkwell::{basic_block::BasicBlock, builder::Builder};
basic_block::BasicBlock,
builder::Builder,
};
use crate::mir::BasicBlockId; use crate::mir::BasicBlockId;
@ -17,11 +14,21 @@ pub struct BuilderCursor<'ctx, 'b> {
impl<'ctx, 'b> BuilderCursor<'ctx, 'b> { impl<'ctx, 'b> BuilderCursor<'ctx, 'b> {
pub fn new(builder: &'b Builder<'ctx>) -> Self { pub fn new(builder: &'b Builder<'ctx>) -> Self {
Self { builder, closed_by_bid: HashMap::new(), cur_bid: None, cur_llbb: None } Self {
builder,
closed_by_bid: HashMap::new(),
cur_bid: None,
cur_llbb: None,
}
} }
/// Temporarily switch to another block, run body, then restore previous position/state. /// Temporarily switch to another block, run body, then restore previous position/state.
pub fn with_block<R>(&mut self, bid: BasicBlockId, bb: BasicBlock<'ctx>, body: impl FnOnce(&mut BuilderCursor<'ctx, 'b>) -> R) -> R { pub fn with_block<R>(
&mut self,
bid: BasicBlockId,
bb: BasicBlock<'ctx>,
body: impl FnOnce(&mut BuilderCursor<'ctx, 'b>) -> R,
) -> R {
let prev_bid = self.cur_bid; let prev_bid = self.cur_bid;
let prev_bb = self.cur_llbb; let prev_bb = self.cur_llbb;
// Preserve previous closed state // Preserve previous closed state
@ -66,7 +73,11 @@ impl<'ctx, 'b> BuilderCursor<'ctx, 'b> {
pub fn assert_open(&self, bid: BasicBlockId) { pub fn assert_open(&self, bid: BasicBlockId) {
if let Some(closed) = self.closed_by_bid.get(&bid) { if let Some(closed) = self.closed_by_bid.get(&bid) {
assert!(!closed, "attempt to insert into closed block {}", bid.as_u32()); assert!(
!closed,
"attempt to insert into closed block {}",
bid.as_u32()
);
} }
} }
@ -86,7 +97,11 @@ impl<'ctx, 'b> BuilderCursor<'ctx, 'b> {
f(self.builder); f(self.builder);
// After emitting a terminator, assert the current basic block now has one // After emitting a terminator, assert the current basic block now has one
if let Some(bb) = self.cur_llbb { if let Some(bb) = self.cur_llbb {
assert!(unsafe { bb.get_terminator() }.is_some(), "expected terminator in bb {}", bid.as_u32()); assert!(
unsafe { bb.get_terminator() }.is_some(),
"expected terminator in bb {}",
bid.as_u32()
);
} }
self.closed_by_bid.insert(bid, true); self.closed_by_bid.insert(bid, true);
} }

View File

@ -1,10 +1,13 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::{types::BasicMetadataTypeEnum as BMT, values::{BasicMetadataValueEnum, BasicValueEnum as BVE, FunctionValue}}; use inkwell::{
types::BasicMetadataTypeEnum as BMT,
values::{BasicMetadataValueEnum, BasicValueEnum as BVE, FunctionValue},
};
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
/// Lower a direct Call where callee is provided as a const string ValueId in MIR14. /// Lower a direct Call where callee is provided as a const string ValueId in MIR14.
/// ///
@ -23,9 +26,15 @@ pub(in super::super) fn lower_call<'ctx, 'b>(
args: &[ValueId], args: &[ValueId],
const_strs: &HashMap<ValueId, String>, const_strs: &HashMap<ValueId, String>,
llvm_funcs: &HashMap<String, FunctionValue<'ctx>>, llvm_funcs: &HashMap<String, FunctionValue<'ctx>>,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
let name_s = const_strs let name_s = const_strs
.get(callee) .get(callee)
@ -51,17 +60,47 @@ pub(in super::super) fn lower_call<'ctx, 'b>(
let coerced: BVE<'ctx> = match exp_tys[i] { let coerced: BVE<'ctx> = match exp_tys[i] {
BMTy::IntType(it) => { BMTy::IntType(it) => {
// Localize as i64, then adjust width to callee expectation // Localize as i64, then adjust width to callee expectation
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *a, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*a,
bb_map,
preds,
block_end_values,
vmap,
)?;
let bw_dst = it.get_bit_width(); let bw_dst = it.get_bit_width();
let bw_src = iv.get_type().get_bit_width(); let bw_src = iv.get_type().get_bit_width();
if bw_src == bw_dst { iv.into() } if bw_src == bw_dst {
else if bw_src < bw_dst { cursor.emit_instr(cur_bid, |b| b.build_int_z_extend(iv, it, "call_arg_zext")).map_err(|e| e.to_string())?.into() } iv.into()
else if bw_dst == 1 { super::super::types::to_bool(codegen.context, iv.into(), &codegen.builder)?.into() } } else if bw_src < bw_dst {
else { cursor.emit_instr(cur_bid, |b| b.build_int_truncate(iv, it, "call_arg_trunc")).map_err(|e| e.to_string())?.into() } cursor
.emit_instr(cur_bid, |b| b.build_int_z_extend(iv, it, "call_arg_zext"))
.map_err(|e| e.to_string())?
.into()
} else if bw_dst == 1 {
super::super::types::to_bool(codegen.context, iv.into(), &codegen.builder)?
.into()
} else {
cursor
.emit_instr(cur_bid, |b| b.build_int_truncate(iv, it, "call_arg_trunc"))
.map_err(|e| e.to_string())?
.into()
}
} }
BMTy::PointerType(pt) => { BMTy::PointerType(pt) => {
// Localize as i64 handle and convert to expected pointer type // Localize as i64 handle and convert to expected pointer type
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *a, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*a,
bb_map,
preds,
block_end_values,
vmap,
)?;
let p = cursor let p = cursor
.emit_instr(cur_bid, |b| b.build_int_to_ptr(iv, pt, "call_arg_i2p")) .emit_instr(cur_bid, |b| b.build_int_to_ptr(iv, pt, "call_arg_i2p"))
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
@ -69,9 +108,19 @@ pub(in super::super) fn lower_call<'ctx, 'b>(
} }
BMTy::FloatType(ft) => { BMTy::FloatType(ft) => {
// Localize as f64, then adjust to callee expectation width if needed // Localize as f64, then adjust to callee expectation width if needed
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, *a, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
if fv.get_type() == ft { fv.into() } codegen,
else { cursor,
cur_bid,
*a,
bb_map,
preds,
block_end_values,
vmap,
)?;
if fv.get_type() == ft {
fv.into()
} else {
// Cast f64<->f32 as needed // Cast f64<->f32 as needed
cursor cursor
.emit_instr(cur_bid, |b| b.build_float_cast(fv, ft, "call_arg_fcast")) .emit_instr(cur_bid, |b| b.build_float_cast(fv, ft, "call_arg_fcast"))

View File

@ -12,11 +12,7 @@ pub(in super::super) fn lower_const<'ctx>(
value: &ConstValue, value: &ConstValue,
) -> Result<(), String> { ) -> Result<(), String> {
let bval = match value { let bval = match value {
ConstValue::Integer(i) => codegen ConstValue::Integer(i) => codegen.context.i64_type().const_int(*i as u64, true).into(),
.context
.i64_type()
.const_int(*i as u64, true)
.into(),
ConstValue::Float(f) => codegen.context.f64_type().const_float(*f).into(), ConstValue::Float(f) => codegen.context.f64_type().const_float(*f).into(),
ConstValue::Bool(b) => codegen ConstValue::Bool(b) => codegen
.context .context
@ -28,10 +24,7 @@ pub(in super::super) fn lower_const<'ctx>(
.builder .builder
.build_global_string_ptr(s, "str") .build_global_string_ptr(s, "str")
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let len = codegen let len = codegen.context.i32_type().const_int(s.len() as u64, false);
.context
.i32_type()
.const_int(s.len() as u64, false);
// declare i8* @nyash_string_new(i8*, i32) // declare i8* @nyash_string_new(i8*, i32)
let rt = codegen.context.ptr_type(inkwell::AddressSpace::from(0)); let rt = codegen.context.ptr_type(inkwell::AddressSpace::from(0));
let fn_ty = rt.fn_type( let fn_ty = rt.fn_type(
@ -50,7 +43,11 @@ pub(in super::super) fn lower_const<'ctx>(
.unwrap_or_else(|| codegen.module.add_function("nyash_string_new", fn_ty, None)); .unwrap_or_else(|| codegen.module.add_function("nyash_string_new", fn_ty, None));
let call = codegen let call = codegen
.builder .builder
.build_call(callee, &[gv.as_pointer_value().into(), len.into()], "strnew") .build_call(
callee,
&[gv.as_pointer_value().into(), len.into()],
"strnew",
)
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value() call.try_as_basic_value()
.left() .left()

View File

@ -2,7 +2,7 @@ use std::collections::HashMap;
use inkwell::{ use inkwell::{
basic_block::BasicBlock, basic_block::BasicBlock,
values::{BasicValueEnum as BVE, IntValue, PointerValue, FloatValue}, values::{BasicValueEnum as BVE, FloatValue, IntValue, PointerValue},
}; };
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
@ -70,8 +70,7 @@ impl<'ctx, 'b> LowerFnCtx<'ctx, 'b> {
#[inline] #[inline]
pub fn ensure_i64(&mut self, blk: &BlockCtx<'ctx>, v: ValueId) -> LlResult<IntValue<'ctx>> { pub fn ensure_i64(&mut self, blk: &BlockCtx<'ctx>, v: ValueId) -> LlResult<IntValue<'ctx>> {
self.cursor.assert_open(blk.cur_bid); self.cursor.assert_open(blk.cur_bid);
self.resolver self.resolver.resolve_i64(
.resolve_i64(
self.codegen, self.codegen,
self.cursor, self.cursor,
blk.cur_bid, blk.cur_bid,
@ -86,8 +85,7 @@ impl<'ctx, 'b> LowerFnCtx<'ctx, 'b> {
#[inline] #[inline]
pub fn ensure_ptr(&mut self, blk: &BlockCtx<'ctx>, v: ValueId) -> LlResult<PointerValue<'ctx>> { pub fn ensure_ptr(&mut self, blk: &BlockCtx<'ctx>, v: ValueId) -> LlResult<PointerValue<'ctx>> {
self.cursor.assert_open(blk.cur_bid); self.cursor.assert_open(blk.cur_bid);
self.resolver self.resolver.resolve_ptr(
.resolve_ptr(
self.codegen, self.codegen,
self.cursor, self.cursor,
blk.cur_bid, blk.cur_bid,
@ -102,8 +100,7 @@ impl<'ctx, 'b> LowerFnCtx<'ctx, 'b> {
#[inline] #[inline]
pub fn ensure_f64(&mut self, blk: &BlockCtx<'ctx>, v: ValueId) -> LlResult<FloatValue<'ctx>> { pub fn ensure_f64(&mut self, blk: &BlockCtx<'ctx>, v: ValueId) -> LlResult<FloatValue<'ctx>> {
self.cursor.assert_open(blk.cur_bid); self.cursor.assert_open(blk.cur_bid);
self.resolver self.resolver.resolve_f64(
.resolve_f64(
self.codegen, self.codegen,
self.cursor, self.cursor,
blk.cur_bid, blk.cur_bid,

View File

@ -3,9 +3,9 @@ use std::collections::HashMap;
use inkwell::values::BasicValueEnum as BVE; use inkwell::values::BasicValueEnum as BVE;
use inkwell::AddressSpace; use inkwell::AddressSpace;
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{BasicBlockId, ValueId}; use crate::mir::{BasicBlockId, ValueId};
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
pub(super) fn lower_log_or_trace<'ctx, 'b>( pub(super) fn lower_log_or_trace<'ctx, 'b>(
codegen: &CodegenContext<'ctx>, codegen: &CodegenContext<'ctx>,
@ -17,15 +17,30 @@ pub(super) fn lower_log_or_trace<'ctx, 'b>(
iface_name: &str, iface_name: &str,
method_name: &str, method_name: &str,
args: &[ValueId], args: &[ValueId],
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
if args.len() != 1 { if args.len() != 1 {
return Err(format!("{}.{} expects 1 arg", iface_name, method_name)); return Err(format!("{}.{} expects 1 arg", iface_name, method_name));
} }
// Localize to i64 (handle path) to avoid vmap shape inspection // Localize to i64 (handle path) to avoid vmap shape inspection
let arg_val = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let arg_val = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let fnty = i64t.fn_type(&[i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into()], false);
let fname = if iface_name == "env.console" { let fname = if iface_name == "env.console" {
@ -42,7 +57,9 @@ pub(super) fn lower_log_or_trace<'ctx, 'b>(
.get_function(fname) .get_function(fname)
.unwrap_or_else(|| codegen.module.add_function(fname, fnty, None)); .unwrap_or_else(|| codegen.module.add_function(fname, fnty, None));
let _ = cursor let _ = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[arg_val.into()], "console_log_h")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[arg_val.into()], "console_log_h")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
vmap.insert(*d, codegen.context.i64_type().const_zero().into()); vmap.insert(*d, codegen.context.i64_type().const_zero().into());
@ -66,7 +83,11 @@ pub(super) fn lower_readline<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.console.readline") .get_function("nyash.console.readline")
.unwrap_or_else(|| codegen.module.add_function("nyash.console.readline", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.console.readline", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[], "readline")) .emit_instr(cur_bid, |b| b.build_call(callee, &[], "readline"))
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;

View File

@ -3,9 +3,9 @@ use std::collections::HashMap;
use inkwell::values::BasicValueEnum as BVE; use inkwell::values::BasicValueEnum as BVE;
use inkwell::AddressSpace; use inkwell::AddressSpace;
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
pub(super) fn lower_future_spawn_instance<'ctx, 'b>( pub(super) fn lower_future_spawn_instance<'ctx, 'b>(
codegen: &CodegenContext<'ctx>, codegen: &CodegenContext<'ctx>,
@ -15,24 +15,54 @@ pub(super) fn lower_future_spawn_instance<'ctx, 'b>(
vmap: &mut HashMap<ValueId, BVE<'ctx>>, vmap: &mut HashMap<ValueId, BVE<'ctx>>,
dst: &Option<ValueId>, dst: &Option<ValueId>,
args: &[ValueId], args: &[ValueId],
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
if args.len() < 2 { if args.len() < 2 {
return Err("env.future.spawn_instance expects at least (recv, method_name)".to_string()); return Err("env.future.spawn_instance expects at least (recv, method_name)".to_string());
} }
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let i8p = codegen.context.ptr_type(AddressSpace::from(0)); let i8p = codegen.context.ptr_type(AddressSpace::from(0));
let recv_h = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let recv_h = resolver.resolve_i64(
let name_p = resolver.resolve_ptr(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?; codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let name_p = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i64t.into(), i8p.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.future.spawn_instance") .get_function("nyash.future.spawn_instance")
.unwrap_or_else(|| codegen.module.add_function("nyash.future.spawn_instance", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.future.spawn_instance", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), name_p.into()], "spawn_instance")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), name_p.into()], "spawn_instance")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -53,24 +83,44 @@ pub(super) fn lower_local_get<'ctx, 'b>(
vmap: &mut HashMap<ValueId, BVE<'ctx>>, vmap: &mut HashMap<ValueId, BVE<'ctx>>,
dst: &Option<ValueId>, dst: &Option<ValueId>,
args: &[ValueId], args: &[ValueId],
_bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, _bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
_preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, _preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
_block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, _block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
if args.len() != 1 { if args.len() != 1 {
return Err("env.local.get expects 1 arg".to_string()); return Err("env.local.get expects 1 arg".to_string());
} }
let name_p = _resolver let name_p = _resolver.resolve_ptr(
.resolve_ptr(codegen, cursor, cur_bid, args[0], _bb_map, _preds, _block_end_values, vmap)?; codegen,
cursor,
cur_bid,
args[0],
_bb_map,
_preds,
_block_end_values,
vmap,
)?;
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let i8p = codegen.context.ptr_type(AddressSpace::from(0)); let i8p = codegen.context.ptr_type(AddressSpace::from(0));
let fnty = i64t.fn_type(&[i8p.into()], false); let fnty = i64t.fn_type(&[i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.env.local.get_h") .get_function("nyash.env.local.get_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.env.local.get_h", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.env.local.get_h", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[name_p.into()], "local_get_h")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[name_p.into()], "local_get_h")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
@ -94,7 +144,9 @@ pub(super) fn lower_local_get<'ctx, 'b>(
let h = rv.into_int_value(); let h = rv.into_int_value();
let pty = codegen.context.ptr_type(AddressSpace::from(0)); let pty = codegen.context.ptr_type(AddressSpace::from(0));
let ptr = cursor let ptr = cursor
.emit_instr(cur_bid, |b| b.build_int_to_ptr(h, pty, "local_get_handle_to_ptr")) .emit_instr(cur_bid, |b| {
b.build_int_to_ptr(h, pty, "local_get_handle_to_ptr")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(*d, ptr.into()); vmap.insert(*d, ptr.into());
} }
@ -118,24 +170,40 @@ pub(super) fn lower_box_new<'ctx, 'b>(
vmap: &mut HashMap<ValueId, BVE<'ctx>>, vmap: &mut HashMap<ValueId, BVE<'ctx>>,
dst: &Option<ValueId>, dst: &Option<ValueId>,
args: &[ValueId], args: &[ValueId],
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
// Two variants: (name) and (argc, arg1, arg2, arg3, arg4) with optional ptr conversion // Two variants: (name) and (argc, arg1, arg2, arg3, arg4) with optional ptr conversion
// Prefer the i64 birth when possible; else call env.box.new(name) // Prefer the i64 birth when possible; else call env.box.new(name)
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let i8p = codegen.context.ptr_type(AddressSpace::from(0)); let i8p = codegen.context.ptr_type(AddressSpace::from(0));
if args.len() == 1 { if args.len() == 1 {
let name_p = resolver let name_p = resolver.resolve_ptr(
.resolve_ptr(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i8p.into()], false); let fnty = i64t.fn_type(&[i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.env.box.new") .get_function("nyash.env.box.new")
.unwrap_or_else(|| codegen.module.add_function("nyash.env.box.new", fnty, None)); .unwrap_or_else(|| codegen.module.add_function("nyash.env.box.new", fnty, None));
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[name_p.into()], "env_box_new")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[name_p.into()], "env_box_new")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let h = call let h = call
.try_as_basic_value() .try_as_basic_value()
@ -167,102 +235,248 @@ pub(super) fn lower_box_new<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.env.box.new_i64") .get_function("nyash.env.box.new_i64")
.unwrap_or_else(|| codegen.module.add_function("nyash.env.box.new_i64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.env.box.new_i64", fnty, None)
});
// arg0: type name string pointer // arg0: type name string pointer
if args.is_empty() { if args.is_empty() {
return Err("env.box.new_i64 requires at least type name".to_string()); return Err("env.box.new_i64 requires at least type name".to_string());
} }
let ty_ptr = resolver let ty_ptr = resolver.resolve_ptr(
.resolve_ptr(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let mut a1 = i64t.const_zero(); let mut a1 = i64t.const_zero();
if args.len() >= 2 { if args.len() >= 2 {
a1 = match func.metadata.value_types.get(&args[1]) { a1 = match func.metadata.value_types.get(&args[1]) {
Some(crate::mir::MirType::Float) => { Some(crate::mir::MirType::Float) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false); let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_f64") .get_function("nyash.box.from_f64")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_f64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_f64", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[fv.into()], "arg1_f64_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[fv.into()], "arg1_f64_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_f64 returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_f64 returned void".to_string())?
.into_int_value()
} }
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?; let pv = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i8p.into()], false); let fnty = i64t.fn_type(&[i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[pv.into()], "arg1_i8_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[pv.into()], "arg1_i8_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_i8_string returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_i8_string returned void".to_string())?
.into_int_value()
} }
_ => resolver.resolve_i64(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?, _ => resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?,
}; };
} }
let mut a2 = i64t.const_zero(); let mut a2 = i64t.const_zero();
if args.len() >= 3 { if args.len() >= 3 {
a2 = match func.metadata.value_types.get(&args[2]) { a2 = match func.metadata.value_types.get(&args[2]) {
Some(crate::mir::MirType::Float) => { Some(crate::mir::MirType::Float) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, args[2], bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
args[2],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false); let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_f64") .get_function("nyash.box.from_f64")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_f64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_f64", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[fv.into()], "arg2_f64_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[fv.into()], "arg2_f64_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_f64 returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_f64 returned void".to_string())?
.into_int_value()
} }
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, args[2], bb_map, preds, block_end_values, vmap)?; let pv = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[2],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i8p.into()], false); let fnty = i64t.fn_type(&[i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[pv.into()], "arg2_i8_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[pv.into()], "arg2_i8_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_i8_string returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_i8_string returned void".to_string())?
.into_int_value()
} }
_ => resolver.resolve_i64(codegen, cursor, cur_bid, args[2], bb_map, preds, block_end_values, vmap)?, _ => resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[2],
bb_map,
preds,
block_end_values,
vmap,
)?,
}; };
} }
let mut a3 = i64t.const_zero(); let mut a3 = i64t.const_zero();
if args.len() >= 4 { if args.len() >= 4 {
a3 = match func.metadata.value_types.get(&args[3]) { a3 = match func.metadata.value_types.get(&args[3]) {
Some(crate::mir::MirType::Integer) | Some(crate::mir::MirType::Bool) => { Some(crate::mir::MirType::Integer) | Some(crate::mir::MirType::Bool) => resolver
resolver.resolve_i64(codegen, cursor, cur_bid, args[3], bb_map, preds, block_end_values, vmap)? .resolve_i64(
} codegen,
cursor,
cur_bid,
args[3],
bb_map,
preds,
block_end_values,
vmap,
)?,
Some(crate::mir::MirType::Float) => { Some(crate::mir::MirType::Float) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, args[3], bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
args[3],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false); let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_f64") .get_function("nyash.box.from_f64")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_f64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_f64", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[fv.into()], "arg3_f64_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[fv.into()], "arg3_f64_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_f64 returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_f64 returned void".to_string())?
.into_int_value()
} }
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, args[3], bb_map, preds, block_end_values, vmap)?; let pv = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[3],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i8p.into()], false); let fnty = i64t.fn_type(&[i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[pv.into()], "arg3_i8_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[pv.into()], "arg3_i8_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_i8_string returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_i8_string returned void".to_string())?
.into_int_value()
} }
_ => return Err("unsupported arg value for env.box.new".to_string()), _ => return Err("unsupported arg value for env.box.new".to_string()),
}; };
@ -270,50 +484,109 @@ pub(super) fn lower_box_new<'ctx, 'b>(
let mut a4 = i64t.const_zero(); let mut a4 = i64t.const_zero();
if args.len() >= 5 { if args.len() >= 5 {
a4 = match func.metadata.value_types.get(&args[4]) { a4 = match func.metadata.value_types.get(&args[4]) {
Some(crate::mir::MirType::Integer) | Some(crate::mir::MirType::Bool) => { Some(crate::mir::MirType::Integer) | Some(crate::mir::MirType::Bool) => resolver
resolver.resolve_i64(codegen, cursor, cur_bid, args[4], bb_map, preds, block_end_values, vmap)? .resolve_i64(
} codegen,
cursor,
cur_bid,
args[4],
bb_map,
preds,
block_end_values,
vmap,
)?,
Some(crate::mir::MirType::Float) => { Some(crate::mir::MirType::Float) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, args[4], bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
args[4],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false); let fnty = i64t.fn_type(&[codegen.context.f64_type().into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_f64") .get_function("nyash.box.from_f64")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_f64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_f64", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[fv.into()], "arg4_f64_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[fv.into()], "arg4_f64_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_f64 returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_f64 returned void".to_string())?
.into_int_value()
} }
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, args[4], bb_map, preds, block_end_values, vmap)?; let pv = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[4],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i8p.into()], false); let fnty = i64t.fn_type(&[i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[pv.into()], "arg4_i8_to_box")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[pv.into()], "arg4_i8_to_box")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
call.try_as_basic_value().left().ok_or("from_i8_string returned void".to_string())?.into_int_value() call.try_as_basic_value()
.left()
.ok_or("from_i8_string returned void".to_string())?
.into_int_value()
} }
_ => return Err("unsupported arg value for env.box.new".to_string()), _ => return Err("unsupported arg value for env.box.new".to_string()),
}; };
} }
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call( .emit_instr(cur_bid, |b| {
b.build_call(
callee, callee,
&[ty_ptr.into(), argc_val.into(), a1.into(), a2.into(), a3.into(), a4.into()], &[
ty_ptr.into(),
argc_val.into(),
a1.into(),
a2.into(),
a3.into(),
a4.into(),
],
"env_box_new_i64x", "env_box_new_i64x",
)) )
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
.left() .left()
.ok_or("env.box.new_i64 returned void".to_string())?; .ok_or("env.box.new_i64 returned void".to_string())?;
let i64v = if let BVE::IntValue(iv) = rv { iv } else { return Err("env.box.new_i64 ret expected i64".to_string()); }; let i64v = if let BVE::IntValue(iv) = rv {
iv
} else {
return Err("env.box.new_i64 ret expected i64".to_string());
};
let out_ptr = cursor let out_ptr = cursor
.emit_instr(cur_bid, |b| b.build_int_to_ptr(i64v, i8p, "box_handle_to_ptr")) .emit_instr(cur_bid, |b| {
b.build_int_to_ptr(i64v, i8p, "box_handle_to_ptr")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
vmap.insert(*d, out_ptr.into()); vmap.insert(*d, out_ptr.into());

View File

@ -3,10 +3,10 @@ mod env;
use std::collections::HashMap; use std::collections::HashMap;
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
use inkwell::values::BasicValueEnum as BVE; use inkwell::values::BasicValueEnum as BVE;
use crate::backend::llvm::compiler::codegen::instructions::builder_cursor::BuilderCursor;
/// Full ExternCall lowering dispatcher (console/debug/env.*) /// Full ExternCall lowering dispatcher (console/debug/env.*)
pub(in super::super) fn lower_externcall<'ctx, 'b>( pub(in super::super) fn lower_externcall<'ctx, 'b>(
@ -20,16 +20,34 @@ pub(in super::super) fn lower_externcall<'ctx, 'b>(
iface_name: &str, iface_name: &str,
method_name: &str, method_name: &str,
args: &[ValueId], args: &[ValueId],
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
// console/debug // console/debug
if (iface_name == "env.console" if (iface_name == "env.console" && matches!(method_name, "log" | "warn" | "error"))
&& matches!(method_name, "log" | "warn" | "error"))
|| (iface_name == "env.debug" && method_name == "trace") || (iface_name == "env.debug" && method_name == "trace")
{ {
return console::lower_log_or_trace(codegen, cursor, resolver, cur_bid, vmap, dst, iface_name, method_name, args, bb_map, preds, block_end_values); return console::lower_log_or_trace(
codegen,
cursor,
resolver,
cur_bid,
vmap,
dst,
iface_name,
method_name,
args,
bb_map,
preds,
block_end_values,
);
} }
if iface_name == "env.console" && method_name == "readLine" { if iface_name == "env.console" && method_name == "readLine" {
return console::lower_readline(codegen, cursor, cur_bid, vmap, dst, args); return console::lower_readline(codegen, cursor, cur_bid, vmap, dst, args);
@ -37,13 +55,48 @@ pub(in super::super) fn lower_externcall<'ctx, 'b>(
// env.* // env.*
if iface_name == "env.future" && method_name == "spawn_instance" { if iface_name == "env.future" && method_name == "spawn_instance" {
return env::lower_future_spawn_instance(codegen, cursor, resolver, cur_bid, vmap, dst, args, bb_map, preds, block_end_values); return env::lower_future_spawn_instance(
codegen,
cursor,
resolver,
cur_bid,
vmap,
dst,
args,
bb_map,
preds,
block_end_values,
);
} }
if iface_name == "env.local" && method_name == "get" { if iface_name == "env.local" && method_name == "get" {
return env::lower_local_get(codegen, cursor, resolver, cur_bid, func, vmap, dst, args, bb_map, preds, block_end_values); return env::lower_local_get(
codegen,
cursor,
resolver,
cur_bid,
func,
vmap,
dst,
args,
bb_map,
preds,
block_end_values,
);
} }
if iface_name == "env.box" && method_name == "new" { if iface_name == "env.box" && method_name == "new" {
return env::lower_box_new(codegen, cursor, resolver, cur_bid, func, vmap, dst, args, bb_map, preds, block_end_values); return env::lower_box_new(
codegen,
cursor,
resolver,
cur_bid,
func,
vmap,
dst,
args,
bb_map,
preds,
block_end_values,
);
} }
Err(format!( Err(format!(

View File

@ -5,11 +5,13 @@ use std::collections::HashMap;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
use super::super::types::{to_bool, map_mirtype_to_basic}; use super::super::types::{map_mirtype_to_basic, to_bool};
use super::builder_cursor::BuilderCursor; use super::builder_cursor::BuilderCursor;
use super::Resolver; use super::Resolver;
fn phi_trace_on() -> bool { std::env::var("NYASH_LLVM_TRACE_PHI").ok().as_deref() == Some("1") } fn phi_trace_on() -> bool {
std::env::var("NYASH_LLVM_TRACE_PHI").ok().as_deref() == Some("1")
}
pub(in super::super) fn emit_return<'ctx, 'b>( pub(in super::super) fn emit_return<'ctx, 'b>(
codegen: &CodegenContext<'ctx>, codegen: &CodegenContext<'ctx>,
@ -25,7 +27,9 @@ pub(in super::super) fn emit_return<'ctx, 'b>(
) -> Result<(), String> { ) -> Result<(), String> {
match (&func.signature.return_type, value) { match (&func.signature.return_type, value) {
(crate::mir::MirType::Void, _) => { (crate::mir::MirType::Void, _) => {
cursor.emit_term(cur_bid, |b| { b.build_return(None).unwrap(); }); cursor.emit_term(cur_bid, |b| {
b.build_return(None).unwrap();
});
Ok(()) Ok(())
} }
(_t, Some(vid)) => { (_t, Some(vid)) => {
@ -34,30 +38,83 @@ pub(in super::super) fn emit_return<'ctx, 'b>(
use inkwell::types::BasicTypeEnum as BT; use inkwell::types::BasicTypeEnum as BT;
let v_adj: BasicValueEnum<'ctx> = match expected { let v_adj: BasicValueEnum<'ctx> = match expected {
BT::IntType(it) => { BT::IntType(it) => {
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *vid, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
// Cast to expected width // Cast to expected width
let bw_src = iv.get_type().get_bit_width(); let bw_src = iv.get_type().get_bit_width();
let bw_dst = it.get_bit_width(); let bw_dst = it.get_bit_width();
if bw_src == bw_dst { iv.into() } if bw_src == bw_dst {
else if bw_src < bw_dst { cursor.emit_instr(cur_bid, |b| b.build_int_z_extend(iv, it, "ret_zext")).map_err(|e| e.to_string())?.into() } iv.into()
else if bw_dst == 1 { to_bool(codegen.context, iv.into(), &codegen.builder)?.into() } } else if bw_src < bw_dst {
else { cursor.emit_instr(cur_bid, |b| b.build_int_truncate(iv, it, "ret_trunc")).map_err(|e| e.to_string())?.into() } cursor
.emit_instr(cur_bid, |b| b.build_int_z_extend(iv, it, "ret_zext"))
.map_err(|e| e.to_string())?
.into()
} else if bw_dst == 1 {
to_bool(codegen.context, iv.into(), &codegen.builder)?.into()
} else {
cursor
.emit_instr(cur_bid, |b| b.build_int_truncate(iv, it, "ret_trunc"))
.map_err(|e| e.to_string())?
.into()
}
} }
BT::PointerType(pt) => { BT::PointerType(pt) => {
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, *vid, bb_map, preds, block_end_values, vmap)?; let pv = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
*vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
// If expected pointer type differs (e.g., typed ptr vs i8*), bitcast // If expected pointer type differs (e.g., typed ptr vs i8*), bitcast
if pv.get_type() == pt { pv.into() } if pv.get_type() == pt {
else { codegen.builder.build_pointer_cast(pv, pt, "ret_bitcast").map_err(|e| e.to_string())?.into() } pv.into()
} else {
codegen
.builder
.build_pointer_cast(pv, pt, "ret_bitcast")
.map_err(|e| e.to_string())?
.into()
}
} }
BT::FloatType(ft) => { BT::FloatType(ft) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, *vid, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
if fv.get_type() == ft { fv.into() } codegen,
else { cursor.emit_instr(cur_bid, |b| b.build_float_cast(fv, ft, "ret_fcast")).map_err(|e| e.to_string())?.into() } cursor,
cur_bid,
*vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
if fv.get_type() == ft {
fv.into()
} else {
cursor
.emit_instr(cur_bid, |b| b.build_float_cast(fv, ft, "ret_fcast"))
.map_err(|e| e.to_string())?
.into()
}
} }
_ => return Err("unsupported return basic type".to_string()), _ => return Err("unsupported return basic type".to_string()),
}; };
cursor.emit_term(cur_bid, |b| { cursor.emit_term(cur_bid, |b| {
b.build_return(Some(&v_adj)).map_err(|e| e.to_string()).unwrap(); b.build_return(Some(&v_adj))
.map_err(|e| e.to_string())
.unwrap();
}); });
Ok(()) Ok(())
} }
@ -82,10 +139,16 @@ pub(in super::super) fn emit_jump<'ctx, 'b>(
eprintln!("[LLVM] emit_jump: {} -> {}", bid.as_u32(), target.as_u32()); eprintln!("[LLVM] emit_jump: {} -> {}", bid.as_u32(), target.as_u32());
} }
cursor.emit_term(bid, |b| { cursor.emit_term(bid, |b| {
b.build_unconditional_branch(tbb).map_err(|e| e.to_string()).unwrap(); b.build_unconditional_branch(tbb)
.map_err(|e| e.to_string())
.unwrap();
}); });
if phi_trace_on() { if phi_trace_on() {
eprintln!("[PHI:jump] pred={} -> succ={}", bid.as_u32(), target.as_u32()); eprintln!(
"[PHI:jump] pred={} -> succ={}",
bid.as_u32(),
target.as_u32()
);
} }
Ok(()) Ok(())
} }
@ -108,7 +171,16 @@ pub(in super::super) fn emit_branch<'ctx, 'b>(
block_end_values: &HashMap<BasicBlockId, HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &HashMap<BasicBlockId, HashMap<ValueId, BasicValueEnum<'ctx>>>,
) -> Result<(), String> { ) -> Result<(), String> {
// Localize condition as i64 and convert to i1 via != 0Resolver 経由のみ) // Localize condition as i64 and convert to i1 via != 0Resolver 経由のみ)
let ci = resolver.resolve_i64(codegen, cursor, bid, *condition, bb_map, preds, block_end_values, vmap)?; let ci = resolver.resolve_i64(
codegen,
cursor,
bid,
*condition,
bb_map,
preds,
block_end_values,
vmap,
)?;
let zero = codegen.context.i64_type().const_zero(); let zero = codegen.context.i64_type().const_zero();
let b = codegen let b = codegen
.builder .builder
@ -118,10 +190,17 @@ pub(in super::super) fn emit_branch<'ctx, 'b>(
let tbb = *bb_map.get(then_bb).ok_or("then bb missing")?; let tbb = *bb_map.get(then_bb).ok_or("then bb missing")?;
let ebb = *bb_map.get(else_bb).ok_or("else bb missing")?; let ebb = *bb_map.get(else_bb).ok_or("else bb missing")?;
if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") { if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") {
eprintln!("[LLVM] emit_branch: {} -> then {} / else {}", bid.as_u32(), then_bb.as_u32(), else_bb.as_u32()); eprintln!(
"[LLVM] emit_branch: {} -> then {} / else {}",
bid.as_u32(),
then_bb.as_u32(),
else_bb.as_u32()
);
} }
cursor.emit_term(bid, |bd| { cursor.emit_term(bid, |bd| {
bd.build_conditional_branch(b, tbb, ebb).map_err(|e| e.to_string()).unwrap(); bd.build_conditional_branch(b, tbb, ebb)
.map_err(|e| e.to_string())
.unwrap();
}); });
Ok(()) Ok(())
} }
@ -147,7 +226,10 @@ fn coerce_to_type<'ctx>(
.into()) .into())
} else if bw_dst == 1 { } else if bw_dst == 1 {
// Narrow to i1 via != 0 // Narrow to i1 via != 0
Ok(super::super::types::to_bool(codegen.context, iv.into(), &codegen.builder)?.into()) Ok(
super::super::types::to_bool(codegen.context, iv.into(), &codegen.builder)?
.into(),
)
} else { } else {
Ok(codegen Ok(codegen
.builder .builder
@ -228,9 +310,13 @@ pub(in super::super) fn seal_block<'ctx, 'b>(
if let Some(pred_llbb) = bb_map.get(&bid) { if let Some(pred_llbb) = bb_map.get(&bid) {
cursor.with_block(bid, *pred_llbb, |c| { cursor.with_block(bid, *pred_llbb, |c| {
let term = unsafe { pred_llbb.get_terminator() }; let term = unsafe { pred_llbb.get_terminator() };
if let Some(t) = term { codegen.builder.position_before(&t); } if let Some(t) = term {
else { c.position_at_end(*pred_llbb); } codegen.builder.position_before(&t);
val = coerce_to_type(codegen, phi, val).expect("coerce_to_type in seal_block"); } else {
c.position_at_end(*pred_llbb);
}
val = coerce_to_type(codegen, phi, val)
.expect("coerce_to_type in seal_block");
}); });
} }
let pred_bb = *bb_map.get(&bid).ok_or("pred bb missing")?; let pred_bb = *bb_map.get(&bid).ok_or("pred bb missing")?;
@ -245,7 +331,11 @@ pub(in super::super) fn seal_block<'ctx, 'b>(
bid.as_u32(), bid.as_u32(),
in_vid.as_u32(), in_vid.as_u32(),
tys, tys,
if snap_opt.is_some() { " (snapshot)" } else { " (synth)" } if snap_opt.is_some() {
" (snapshot)"
} else {
" (synth)"
}
); );
} }
match val { match val {
@ -263,7 +353,9 @@ pub(in super::super) fn seal_block<'ctx, 'b>(
BT::IntType(it) => it.const_zero().into(), BT::IntType(it) => it.const_zero().into(),
BT::FloatType(ft) => ft.const_zero().into(), BT::FloatType(ft) => ft.const_zero().into(),
BT::PointerType(pt) => pt.const_zero().into(), BT::PointerType(pt) => pt.const_zero().into(),
_ => return Err("unsupported phi type for zero synth (seal)".to_string()), _ => {
return Err("unsupported phi type for zero synth (seal)".to_string())
}
}; };
if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") { if std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1") {
eprintln!( eprintln!(
@ -303,7 +395,9 @@ pub(in super::super) fn finalize_phis<'ctx, 'b>(
vmap: &HashMap<ValueId, BasicValueEnum<'ctx>>, vmap: &HashMap<ValueId, BasicValueEnum<'ctx>>,
) -> Result<(), String> { ) -> Result<(), String> {
let pred_list = preds.get(&succ_bb).cloned().unwrap_or_default(); let pred_list = preds.get(&succ_bb).cloned().unwrap_or_default();
if pred_list.is_empty() { return Ok(()); } if pred_list.is_empty() {
return Ok(());
}
if let Some(phis) = phis_by_block.get(&succ_bb) { if let Some(phis) = phis_by_block.get(&succ_bb) {
for (_dst, phi, inputs) in phis { for (_dst, phi, inputs) in phis {
for pred in &pred_list { for pred in &pred_list {
@ -314,7 +408,9 @@ pub(in super::super) fn finalize_phis<'ctx, 'b>(
// we add at most once per pred in seal_block. If duplicates occurred earlier, // we add at most once per pred in seal_block. If duplicates occurred earlier,
// adding again is harmlessly ignored by verifier if identical; otherwise rely on our new regime. // adding again is harmlessly ignored by verifier if identical; otherwise rely on our new regime.
// Fetch value snapshot at end of pred; fallback per our policy // Fetch value snapshot at end of pred; fallback per our policy
let snap_opt = block_end_values.get(pred).and_then(|m| m.get(in_vid).copied()); let snap_opt = block_end_values
.get(pred)
.and_then(|m| m.get(in_vid).copied());
let mut val = if let Some(sv) = snap_opt { let mut val = if let Some(sv) = snap_opt {
sv sv
} else { } else {
@ -334,17 +430,26 @@ pub(in super::super) fn finalize_phis<'ctx, 'b>(
if let Some(pred_llbb) = bb_map.get(pred) { if let Some(pred_llbb) = bb_map.get(pred) {
cursor.with_block(*pred, *pred_llbb, |c| { cursor.with_block(*pred, *pred_llbb, |c| {
let term = unsafe { pred_llbb.get_terminator() }; let term = unsafe { pred_llbb.get_terminator() };
if let Some(t) = term { codegen.builder.position_before(&t); } if let Some(t) = term {
else { c.position_at_end(*pred_llbb); } codegen.builder.position_before(&t);
val = coerce_to_type(codegen, phi, val).expect("coerce_to_type finalize_phis"); } else {
c.position_at_end(*pred_llbb);
}
val = coerce_to_type(codegen, phi, val)
.expect("coerce_to_type finalize_phis");
}); });
} }
let pred_bb = *bb_map.get(pred).ok_or("pred bb missing")?; let pred_bb = *bb_map.get(pred).ok_or("pred bb missing")?;
if phi_trace_on() { if phi_trace_on() {
eprintln!( eprintln!(
"[PHI:finalize] succ={} pred={} vid={} ty={}", "[PHI:finalize] succ={} pred={} vid={} ty={}",
succ_bb.as_u32(), pred.as_u32(), in_vid.as_u32(), succ_bb.as_u32(),
phi.as_basic_value().get_type().print_to_string().to_string() pred.as_u32(),
in_vid.as_u32(),
phi.as_basic_value()
.get_type()
.print_to_string()
.to_string()
); );
} }
match val { match val {
@ -362,12 +467,16 @@ pub(in super::super) fn finalize_phis<'ctx, 'b>(
BT::IntType(it) => it.const_zero().into(), BT::IntType(it) => it.const_zero().into(),
BT::FloatType(ft) => ft.const_zero().into(), BT::FloatType(ft) => ft.const_zero().into(),
BT::PointerType(pt) => pt.const_zero().into(), BT::PointerType(pt) => pt.const_zero().into(),
_ => return Err("unsupported phi type for zero synth (finalize)".to_string()), _ => {
return Err("unsupported phi type for zero synth (finalize)".to_string())
}
}; };
if phi_trace_on() { if phi_trace_on() {
eprintln!( eprintln!(
"[PHI:finalize] succ={} pred={} vid=? ty={} src=synth_zero", "[PHI:finalize] succ={} pred={} vid=? ty={} src=synth_zero",
succ_bb.as_u32(), pred.as_u32(), bt.print_to_string().to_string() succ_bb.as_u32(),
pred.as_u32(),
bt.print_to_string().to_string()
); );
} }
match z { match z {
@ -393,7 +502,10 @@ pub(in super::super) fn localize_to_i64<'ctx, 'b>(
vid: ValueId, vid: ValueId,
bb_map: &std::collections::HashMap<BasicBlockId, BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<BasicBlockId, BasicBlock<'ctx>>,
preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>, preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>,
block_end_values: &std::collections::HashMap<BasicBlockId, std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
BasicBlockId,
std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
>,
vmap: &std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>, vmap: &std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
) -> Result<IntValue<'ctx>, String> { ) -> Result<IntValue<'ctx>, String> {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
@ -413,7 +525,10 @@ pub(in super::super) fn localize_to_i64<'ctx, 'b>(
} else { } else {
codegen.builder.position_at_end(cur_llbb); codegen.builder.position_at_end(cur_llbb);
} }
let phi = codegen.builder.build_phi(i64t, &format!("loc_i64_{}", vid.as_u32())).map_err(|e| e.to_string())?; let phi = codegen
.builder
.build_phi(i64t, &format!("loc_i64_{}", vid.as_u32()))
.map_err(|e| e.to_string())?;
for p in &pred_list { for p in &pred_list {
let pred_bb = *bb_map.get(p).ok_or("pred bb missing")?; let pred_bb = *bb_map.get(p).ok_or("pred bb missing")?;
// Fetch snapshot at end of pred; if missing, synthesize zero // Fetch snapshot at end of pred; if missing, synthesize zero
@ -425,13 +540,33 @@ pub(in super::super) fn localize_to_i64<'ctx, 'b>(
let mut iv_out = i64t.const_zero(); let mut iv_out = i64t.const_zero();
cursor.with_block(*p, pred_bb, |c| { cursor.with_block(*p, pred_bb, |c| {
let term = unsafe { pred_bb.get_terminator() }; let term = unsafe { pred_bb.get_terminator() };
if let Some(t) = term { codegen.builder.position_before(&t); } else { c.position_at_end(pred_bb); } if let Some(t) = term {
codegen.builder.position_before(&t);
} else {
c.position_at_end(pred_bb);
}
iv_out = match base { iv_out = match base {
BasicValueEnum::IntValue(iv) => { BasicValueEnum::IntValue(iv) => {
if iv.get_type() == i64t { iv } else { codegen.builder.build_int_z_extend(iv, i64t, "loc_zext_p").map_err(|e| e.to_string()).unwrap() } if iv.get_type() == i64t {
iv
} else {
codegen
.builder
.build_int_z_extend(iv, i64t, "loc_zext_p")
.map_err(|e| e.to_string())
.unwrap()
} }
BasicValueEnum::PointerValue(pv) => codegen.builder.build_ptr_to_int(pv, i64t, "loc_p2i_p").map_err(|e| e.to_string()).unwrap(), }
BasicValueEnum::FloatValue(fv) => codegen.builder.build_float_to_signed_int(fv, i64t, "loc_f2i_p").map_err(|e| e.to_string()).unwrap(), BasicValueEnum::PointerValue(pv) => codegen
.builder
.build_ptr_to_int(pv, i64t, "loc_p2i_p")
.map_err(|e| e.to_string())
.unwrap(),
BasicValueEnum::FloatValue(fv) => codegen
.builder
.build_float_to_signed_int(fv, i64t, "loc_f2i_p")
.map_err(|e| e.to_string())
.unwrap(),
_ => i64t.const_zero(), _ => i64t.const_zero(),
}; };
}); });
@ -439,11 +574,15 @@ pub(in super::super) fn localize_to_i64<'ctx, 'b>(
if phi_trace_on() { if phi_trace_on() {
eprintln!( eprintln!(
"[PHI:resolve] cur={} pred={} vid={} ty=i64", "[PHI:resolve] cur={} pred={} vid={} ty=i64",
cur_bid.as_u32(), p.as_u32(), vid.as_u32() cur_bid.as_u32(),
p.as_u32(),
vid.as_u32()
); );
} }
} }
// Restore insertion point // Restore insertion point
if let Some(bb) = saved_ip { codegen.builder.position_at_end(bb); } if let Some(bb) = saved_ip {
codegen.builder.position_at_end(bb);
}
Ok(phi.as_basic_value().into_int_value()) Ok(phi.as_basic_value().into_int_value())
} }

View File

@ -4,16 +4,11 @@ use inkwell::{
}; };
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{ use crate::mir::{function::MirFunction, instruction::MirInstruction, BasicBlockId, ValueId};
function::MirFunction,
instruction::MirInstruction,
BasicBlockId,
ValueId,
};
use super::super::types::to_bool;
use super::builder_cursor::BuilderCursor; use super::builder_cursor::BuilderCursor;
use super::Resolver; use super::Resolver;
use super::super::types::to_bool;
/// LoopForm scaffolding — fixed block layout for while/loop normalization /// LoopForm scaffolding — fixed block layout for while/loop normalization
pub struct LoopFormContext<'ctx> { pub struct LoopFormContext<'ctx> {
@ -52,7 +47,15 @@ impl<'ctx> LoopFormContext<'ctx> {
let exit = codegen let exit = codegen
.context .context
.append_basic_block(function, &format!("{}_lf{}_exit", prefix, loop_id)); .append_basic_block(function, &format!("{}_lf{}_exit", prefix, loop_id));
Self { preheader, header, body, dispatch, latch, exit, loop_id } Self {
preheader,
header,
body,
dispatch,
latch,
exit,
loop_id,
}
} }
} }
@ -76,13 +79,26 @@ pub fn lower_while_loopform<'ctx, 'b>(
bb_map: &std::collections::HashMap<BasicBlockId, BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<BasicBlockId, BasicBlock<'ctx>>,
vmap: &std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>, vmap: &std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>, preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>,
block_end_values: &std::collections::HashMap<BasicBlockId, std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
BasicBlockId,
std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
>,
// Registry to allow later body→dispatch wiring (simple bodies) // Registry to allow later body→dispatch wiring (simple bodies)
registry: &mut std::collections::HashMap<BasicBlockId, (BasicBlock<'ctx>, PhiValue<'ctx>, PhiValue<'ctx>, BasicBlock<'ctx>)>, registry: &mut std::collections::HashMap<
BasicBlockId,
(
BasicBlock<'ctx>,
PhiValue<'ctx>,
PhiValue<'ctx>,
BasicBlock<'ctx>,
),
>,
body_to_header: &mut std::collections::HashMap<BasicBlockId, BasicBlockId>, body_to_header: &mut std::collections::HashMap<BasicBlockId, BasicBlockId>,
) -> Result<bool, String> { ) -> Result<bool, String> {
let enabled = std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1"); let enabled = std::env::var("NYASH_ENABLE_LOOPFORM").ok().as_deref() == Some("1");
if !enabled { return Ok(false); } if !enabled {
return Ok(false);
}
// Create LoopForm fixed blocks under the same function // Create LoopForm fixed blocks under the same function
let lf = LoopFormContext::new(codegen, llvm_func, loop_id, prefix); let lf = LoopFormContext::new(codegen, llvm_func, loop_id, prefix);
@ -95,7 +111,16 @@ pub fn lower_while_loopform<'ctx, 'b>(
.unwrap(); .unwrap();
// Header: evaluate condition via Resolver and branch to body (for true) or dispatch (for false) // Header: evaluate condition via Resolver and branch to body (for true) or dispatch (for false)
let ci = resolver.resolve_i64(codegen, cursor, header_bid, *condition, bb_map, preds, block_end_values, vmap)?; let ci = resolver.resolve_i64(
codegen,
cursor,
header_bid,
*condition,
bb_map,
preds,
block_end_values,
vmap,
)?;
let cond_i1 = codegen let cond_i1 = codegen
.builder .builder
.build_int_compare( .build_int_compare(
@ -124,7 +149,9 @@ pub fn lower_while_loopform<'ctx, 'b>(
// Dispatch: create PHIs (tag i8, payload i64) and switch(tag) // Dispatch: create PHIs (tag i8, payload i64) and switch(tag)
// For now, only header(false) contributes (Break=1); body path does not reach dispatch in Phase 1 wiring. // For now, only header(false) contributes (Break=1); body path does not reach dispatch in Phase 1 wiring.
let orig_after = *bb_map.get(&after_bb).ok_or("loopform: after bb missing")?; let orig_after = *bb_map.get(&after_bb).ok_or("loopform: after bb missing")?;
let header_llbb = *bb_map.get(&header_bid).ok_or("loopform: header bb missing")?; let header_llbb = *bb_map
.get(&header_bid)
.ok_or("loopform: header bb missing")?;
let (tag_phi, payload_phi) = cursor.with_block(after_bb, lf.dispatch, |c| { let (tag_phi, payload_phi) = cursor.with_block(after_bb, lf.dispatch, |c| {
let i8t = codegen.context.i8_type(); let i8t = codegen.context.i8_type();
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
@ -219,20 +246,35 @@ pub fn normalize_header_phis_for_latch<'ctx>(
pub(in super::super) fn dev_check_dispatch_only_phi<'ctx>( pub(in super::super) fn dev_check_dispatch_only_phi<'ctx>(
phis_by_block: &std::collections::HashMap< phis_by_block: &std::collections::HashMap<
crate::mir::BasicBlockId, crate::mir::BasicBlockId,
Vec<(crate::mir::ValueId, inkwell::values::PhiValue<'ctx>, Vec<(crate::mir::BasicBlockId, crate::mir::ValueId)>)>, Vec<(
crate::mir::ValueId,
inkwell::values::PhiValue<'ctx>,
Vec<(crate::mir::BasicBlockId, crate::mir::ValueId)>,
)>,
>, >,
loopform_registry: &std::collections::HashMap< loopform_registry: &std::collections::HashMap<
crate::mir::BasicBlockId, crate::mir::BasicBlockId,
(inkwell::basic_block::BasicBlock<'ctx>, inkwell::values::PhiValue<'ctx>, inkwell::values::PhiValue<'ctx>, inkwell::basic_block::BasicBlock<'ctx>) (
inkwell::basic_block::BasicBlock<'ctx>,
inkwell::values::PhiValue<'ctx>,
inkwell::values::PhiValue<'ctx>,
inkwell::basic_block::BasicBlock<'ctx>,
),
>, >,
) { ) {
if std::env::var("NYASH_DEV_CHECK_DISPATCH_ONLY_PHI").ok().as_deref() != Some("1") { if std::env::var("NYASH_DEV_CHECK_DISPATCH_ONLY_PHI")
.ok()
.as_deref()
!= Some("1")
{
return; return;
} }
// Best-effort: Just report PHI presence per block when LoopForm registry is non-empty. // Best-effort: Just report PHI presence per block when LoopForm registry is non-empty.
if !loopform_registry.is_empty() { if !loopform_registry.is_empty() {
for (bid, phis) in phis_by_block.iter() { for (bid, phis) in phis_by_block.iter() {
if phis.is_empty() { continue; } if phis.is_empty() {
continue;
}
eprintln!("[DEV][PHI] bb={} has {} PHI(s)", bid.as_u32(), phis.len()); eprintln!("[DEV][PHI] bb={} has {} PHI(s)", bid.as_u32(), phis.len());
} }
} }

View File

@ -2,9 +2,9 @@ use std::collections::HashMap;
use inkwell::{values::BasicValueEnum as BVE, AddressSpace}; use inkwell::{values::BasicValueEnum as BVE, AddressSpace};
use super::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId}; use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
use super::builder_cursor::BuilderCursor;
/// Handle MapBox fast-paths (core-first). Returns true if handled. /// Handle MapBox fast-paths (core-first). Returns true if handled.
pub(super) fn try_handle_map_method<'ctx, 'b>( pub(super) fn try_handle_map_method<'ctx, 'b>(
@ -61,18 +61,49 @@ pub(super) fn try_handle_map_method<'ctx, 'b>(
let key_i = match func.metadata.value_types.get(&args[0]) { let key_i = match func.metadata.value_types.get(&args[0]) {
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
// string key: i8* -> handle // string key: i8* -> handle
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, args[0], &std::collections::HashMap::new(), &std::collections::HashMap::new(), &std::collections::HashMap::new(), vmap)?; let pv = resolver.resolve_ptr(
let fnty_conv = i64t.fn_type(&[codegen.context.ptr_type(AddressSpace::from(0)).into()], false); codegen,
cursor,
cur_bid,
args[0],
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
vmap,
)?;
let fnty_conv = i64t.fn_type(
&[codegen.context.ptr_type(AddressSpace::from(0)).into()],
false,
);
let conv = codegen let conv = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty_conv, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty_conv, None)
});
let kcall = cursor let kcall = cursor
.emit_instr(cur_bid, |b| b.build_call(conv, &[pv.into()], "key_i8_to_handle")) .emit_instr(cur_bid, |b| {
b.build_call(conv, &[pv.into()], "key_i8_to_handle")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
kcall.try_as_basic_value().left().ok_or("from_i8_string returned void".to_string())?.into_int_value() kcall
.try_as_basic_value()
.left()
.ok_or("from_i8_string returned void".to_string())?
.into_int_value()
} }
_ => resolver.resolve_i64(codegen, cursor, cur_bid, args[0], &std::collections::HashMap::new(), &std::collections::HashMap::new(), &std::collections::HashMap::new(), vmap)? _ => resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
vmap,
)?,
}; };
let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false);
let callee = codegen let callee = codegen
@ -80,7 +111,9 @@ pub(super) fn try_handle_map_method<'ctx, 'b>(
.get_function("nyash.map.has_h") .get_function("nyash.map.has_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.map.has_h", fnty, None)); .unwrap_or_else(|| codegen.module.add_function("nyash.map.has_h", fnty, None));
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), key_i.into()], "mhas")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), key_i.into()], "mhas")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -101,15 +134,32 @@ pub(super) fn try_handle_map_method<'ctx, 'b>(
let call = match func.metadata.value_types.get(&args[0]) { let call = match func.metadata.value_types.get(&args[0]) {
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
// key: i8* -> i64 handle via from_i8_string (string key) // key: i8* -> i64 handle via from_i8_string (string key)
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, args[0], &std::collections::HashMap::new(), &std::collections::HashMap::new(), &std::collections::HashMap::new(), vmap)?; let pv = resolver.resolve_ptr(
let fnty_conv = i64t codegen,
.fn_type(&[codegen.context.ptr_type(AddressSpace::from(0)).into()], false); cursor,
cur_bid,
args[0],
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
vmap,
)?;
let fnty_conv = i64t.fn_type(
&[codegen.context.ptr_type(AddressSpace::from(0)).into()],
false,
);
let conv = codegen let conv = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty_conv, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty_conv, None)
});
let kcall = cursor let kcall = cursor
.emit_instr(cur_bid, |b| b.build_call(conv, &[pv.into()], "key_i8_to_handle")) .emit_instr(cur_bid, |b| {
b.build_call(conv, &[pv.into()], "key_i8_to_handle")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let kh = kcall let kh = kcall
.try_as_basic_value() .try_as_basic_value()
@ -120,20 +170,37 @@ pub(super) fn try_handle_map_method<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.map.get_hh") .get_function("nyash.map.get_hh")
.unwrap_or_else(|| codegen.module.add_function("nyash.map.get_hh", fnty, None)); .unwrap_or_else(|| {
codegen.module.add_function("nyash.map.get_hh", fnty, None)
});
cursor cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), kh.into()], "mget_hh")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), kh.into()], "mget_hh")
})
.map_err(|e| e.to_string())? .map_err(|e| e.to_string())?
} }
_ => { _ => {
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], &std::collections::HashMap::new(), &std::collections::HashMap::new(), &std::collections::HashMap::new(), vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
vmap,
)?;
let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.map.get_h") .get_function("nyash.map.get_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.map.get_h", fnty, None)); .unwrap_or_else(|| {
codegen.module.add_function("nyash.map.get_h", fnty, None)
});
cursor cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), iv.into()], "mget")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), iv.into()], "mget")
})
.map_err(|e| e.to_string())? .map_err(|e| e.to_string())?
} }
}; };
@ -155,27 +222,69 @@ pub(super) fn try_handle_map_method<'ctx, 'b>(
} }
let key_i = match func.metadata.value_types.get(&args[0]) { let key_i = match func.metadata.value_types.get(&args[0]) {
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, args[0], &std::collections::HashMap::new(), &std::collections::HashMap::new(), &std::collections::HashMap::new(), vmap)?; let pv = resolver.resolve_ptr(
let fnty_conv = i64t.fn_type(&[codegen.context.ptr_type(AddressSpace::from(0)).into()], false); codegen,
cursor,
cur_bid,
args[0],
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
vmap,
)?;
let fnty_conv = i64t.fn_type(
&[codegen.context.ptr_type(AddressSpace::from(0)).into()],
false,
);
let conv = codegen let conv = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty_conv, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty_conv, None)
});
let kcall = cursor let kcall = cursor
.emit_instr(cur_bid, |b| b.build_call(conv, &[pv.into()], "key_i8_to_handle")) .emit_instr(cur_bid, |b| {
b.build_call(conv, &[pv.into()], "key_i8_to_handle")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
kcall.try_as_basic_value().left().ok_or("from_i8_string returned void".to_string())?.into_int_value() kcall
.try_as_basic_value()
.left()
.ok_or("from_i8_string returned void".to_string())?
.into_int_value()
} }
_ => resolver.resolve_i64(codegen, cursor, cur_bid, args[0], &std::collections::HashMap::new(), &std::collections::HashMap::new(), &std::collections::HashMap::new(), vmap)? _ => resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
vmap,
)?,
}; };
let val_i = resolver.resolve_i64(codegen, cursor, cur_bid, args[1], &std::collections::HashMap::new(), &std::collections::HashMap::new(), &std::collections::HashMap::new(), vmap)?; let val_i = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[1],
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
&std::collections::HashMap::new(),
vmap,
)?;
let fnty = i64t.fn_type(&[i64t.into(), i64t.into(), i64t.into()], false); let fnty = i64t.fn_type(&[i64t.into(), i64t.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.map.set_h") .get_function("nyash.map.set_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.map.set_h", fnty, None)); .unwrap_or_else(|| codegen.module.add_function("nyash.map.set_h", fnty, None));
let _ = cursor let _ = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[recv_h.into(), key_i.into(), val_i.into()], "mset")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[recv_h.into(), key_i.into(), val_i.into()], "mset")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
Ok(true) Ok(true)
} }

View File

@ -2,9 +2,9 @@ use std::collections::HashMap;
use inkwell::values::BasicValueEnum; use inkwell::values::BasicValueEnum;
use super::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{BasicBlockId, ValueId}; use crate::mir::{BasicBlockId, ValueId};
use super::builder_cursor::BuilderCursor;
// Lower Store: handle allocas with element type tracking and integer width adjust // Lower Store: handle allocas with element type tracking and integer width adjust
pub(in super::super) fn lower_store<'ctx, 'b>( pub(in super::super) fn lower_store<'ctx, 'b>(
@ -17,18 +17,51 @@ pub(in super::super) fn lower_store<'ctx, 'b>(
alloca_elem_types: &mut HashMap<ValueId, inkwell::types::BasicTypeEnum<'ctx>>, alloca_elem_types: &mut HashMap<ValueId, inkwell::types::BasicTypeEnum<'ctx>>,
value: &ValueId, value: &ValueId,
ptr: &ValueId, ptr: &ValueId,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
use inkwell::types::BasicTypeEnum; use inkwell::types::BasicTypeEnum;
// Resolve value preferring native kind; try i64, then f64, else pointer // Resolve value preferring native kind; try i64, then f64, else pointer
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let val: BasicValueEnum = if let Ok(iv) = resolver.resolve_i64(codegen, cursor, cur_bid, *value, bb_map, preds, block_end_values, vmap) { let val: BasicValueEnum = if let Ok(iv) = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*value,
bb_map,
preds,
block_end_values,
vmap,
) {
iv.into() iv.into()
} else if let Ok(fv) = resolver.resolve_f64(codegen, cursor, cur_bid, *value, bb_map, preds, block_end_values, vmap) { } else if let Ok(fv) = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
*value,
bb_map,
preds,
block_end_values,
vmap,
) {
fv.into() fv.into()
} else if let Ok(pv) = resolver.resolve_ptr(codegen, cursor, cur_bid, *value, bb_map, preds, block_end_values, vmap) { } else if let Ok(pv) = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
*value,
bb_map,
preds,
block_end_values,
vmap,
) {
pv.into() pv.into()
} else { } else {
// Fallback: zero i64 // Fallback: zero i64
@ -41,7 +74,9 @@ pub(in super::super) fn lower_store<'ctx, 'b>(
_ => return Err("unsupported store value type".to_string()), _ => return Err("unsupported store value type".to_string()),
}; };
if let Some(existing) = allocas.get(ptr).copied() { if let Some(existing) = allocas.get(ptr).copied() {
let existing_elem = *alloca_elem_types.get(ptr).ok_or("alloca elem type missing")?; let existing_elem = *alloca_elem_types
.get(ptr)
.ok_or("alloca elem type missing")?;
if existing_elem != elem_ty { if existing_elem != elem_ty {
match (val, existing_elem) { match (val, existing_elem) {
(BasicValueEnum::IntValue(iv), BasicTypeEnum::IntType(t)) => { (BasicValueEnum::IntValue(iv), BasicTypeEnum::IntType(t)) => {
@ -93,7 +128,9 @@ pub(in super::super) fn lower_store<'ctx, 'b>(
} }
} else { } else {
let slot = cursor let slot = cursor
.emit_instr(cur_bid, |b| b.build_alloca(elem_ty, &format!("slot_{}", ptr.as_u32()))) .emit_instr(cur_bid, |b| {
b.build_alloca(elem_ty, &format!("slot_{}", ptr.as_u32()))
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
cursor cursor
.emit_instr(cur_bid, |b| b.build_store(slot, val)) .emit_instr(cur_bid, |b| b.build_store(slot, val))
@ -116,20 +153,26 @@ pub(in super::super) fn lower_load<'ctx, 'b>(
) -> Result<(), String> { ) -> Result<(), String> {
use inkwell::types::BasicTypeEnum; use inkwell::types::BasicTypeEnum;
let (slot, elem_ty) = if let Some(s) = allocas.get(ptr).copied() { let (slot, elem_ty) = if let Some(s) = allocas.get(ptr).copied() {
let et = *alloca_elem_types.get(ptr).ok_or("alloca elem type missing")?; let et = *alloca_elem_types
.get(ptr)
.ok_or("alloca elem type missing")?;
(s, et) (s, et)
} else { } else {
// Default new slot as i64 for uninitialized loads // Default new slot as i64 for uninitialized loads
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let slot = cursor let slot = cursor
.emit_instr(cur_bid, |b| b.build_alloca(i64t, &format!("slot_{}", ptr.as_u32()))) .emit_instr(cur_bid, |b| {
b.build_alloca(i64t, &format!("slot_{}", ptr.as_u32()))
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
allocas.insert(*ptr, slot); allocas.insert(*ptr, slot);
alloca_elem_types.insert(*ptr, i64t.into()); alloca_elem_types.insert(*ptr, i64t.into());
(slot, i64t.into()) (slot, i64t.into())
}; };
let lv = cursor let lv = cursor
.emit_instr(cur_bid, |b| b.build_load(elem_ty, slot, &format!("load_{}", dst.as_u32()))) .emit_instr(cur_bid, |b| {
b.build_load(elem_ty, slot, &format!("load_{}", dst.as_u32()))
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(*dst, lv); vmap.insert(*dst, lv);
Ok(()) Ok(())
@ -145,9 +188,15 @@ pub(in super::super) fn lower_copy<'ctx, 'b>(
vmap: &mut HashMap<ValueId, BasicValueEnum<'ctx>>, vmap: &mut HashMap<ValueId, BasicValueEnum<'ctx>>,
dst: &ValueId, dst: &ValueId,
src: &ValueId, src: &ValueId,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
// Choose resolution kind based on metadata type preference // Choose resolution kind based on metadata type preference
use inkwell::types::BasicTypeEnum as BT; use inkwell::types::BasicTypeEnum as BT;
@ -160,20 +209,56 @@ pub(in super::super) fn lower_copy<'ctx, 'b>(
let out: BasicValueEnum<'ctx> = match expected_bt { let out: BasicValueEnum<'ctx> = match expected_bt {
Some(BT::IntType(_)) | None => { Some(BT::IntType(_)) | None => {
// Prefer i64 for unknown // Prefer i64 for unknown
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *src, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*src,
bb_map,
preds,
block_end_values,
vmap,
)?;
iv.into() iv.into()
} }
Some(BT::PointerType(_)) => { Some(BT::PointerType(_)) => {
let pv = resolver.resolve_ptr(codegen, cursor, cur_bid, *src, bb_map, preds, block_end_values, vmap)?; let pv = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
*src,
bb_map,
preds,
block_end_values,
vmap,
)?;
pv.into() pv.into()
} }
Some(BT::FloatType(_)) => { Some(BT::FloatType(_)) => {
let fv = resolver.resolve_f64(codegen, cursor, cur_bid, *src, bb_map, preds, block_end_values, vmap)?; let fv = resolver.resolve_f64(
codegen,
cursor,
cur_bid,
*src,
bb_map,
preds,
block_end_values,
vmap,
)?;
fv.into() fv.into()
} }
_ => { _ => {
// Fallback i64 // Fallback i64
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, *src, bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*src,
bb_map,
preds,
block_end_values,
vmap,
)?;
iv.into() iv.into()
} }
}; };

View File

@ -1,34 +1,34 @@
mod blocks;
pub mod builder_cursor;
pub mod flow;
mod externcall;
mod newbox;
mod boxcall;
pub mod ctx;
pub mod string_ops;
mod arith; mod arith;
mod mem;
mod consts;
mod strings;
mod arrays;
mod maps;
mod arith_ops; mod arith_ops;
mod arrays;
mod blocks;
mod boxcall;
pub mod builder_cursor;
mod call; mod call;
mod consts;
pub mod ctx;
mod externcall;
pub mod flow;
mod loopform; mod loopform;
mod maps;
mod mem;
mod newbox;
mod resolver; mod resolver;
pub mod string_ops;
mod strings;
pub(super) use blocks::{create_basic_blocks, precreate_phis};
pub(super) use flow::{emit_branch, emit_jump, emit_return};
pub(super) use externcall::lower_externcall;
pub(super) use newbox::lower_newbox;
pub(super) use boxcall::{lower_boxcall, lower_boxcall_boxed, lower_boxcall_via_ctx};
pub(super) use arith::lower_compare; pub(super) use arith::lower_compare;
pub(super) use mem::{lower_load, lower_store};
pub(super) use mem::lower_copy;
pub(super) use consts::lower_const;
pub(super) use arith_ops::{lower_binop, lower_unary}; pub(super) use arith_ops::{lower_binop, lower_unary};
pub(super) use blocks::{create_basic_blocks, precreate_phis};
pub(super) use boxcall::{lower_boxcall, lower_boxcall_boxed, lower_boxcall_via_ctx};
pub(super) use call::lower_call; pub(super) use call::lower_call;
pub(super) use loopform::{LoopFormContext, lower_while_loopform}; pub(super) use consts::lower_const;
pub(super) use loopform::normalize_header_phis_for_latch; pub(super) use externcall::lower_externcall;
pub(super) use flow::{emit_branch, emit_jump, emit_return};
pub(super) use loopform::dev_check_dispatch_only_phi; pub(super) use loopform::dev_check_dispatch_only_phi;
pub(super) use loopform::normalize_header_phis_for_latch;
pub(super) use loopform::{lower_while_loopform, LoopFormContext};
pub(super) use mem::lower_copy;
pub(super) use mem::{lower_load, lower_store};
pub(super) use newbox::lower_newbox;
pub(super) use resolver::Resolver; pub(super) use resolver::Resolver;

View File

@ -1,11 +1,11 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::AddressSpace;
use inkwell::values::BasicValueEnum as BVE; use inkwell::values::BasicValueEnum as BVE;
use inkwell::AddressSpace;
use super::builder_cursor::BuilderCursor;
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{BasicBlockId, ValueId}; use crate::mir::{BasicBlockId, ValueId};
use super::builder_cursor::BuilderCursor;
// NewBox lowering (subset consistent with existing code) // NewBox lowering (subset consistent with existing code)
pub(in super::super) fn lower_newbox<'ctx, 'b>( pub(in super::super) fn lower_newbox<'ctx, 'b>(
@ -18,14 +18,29 @@ pub(in super::super) fn lower_newbox<'ctx, 'b>(
box_type: &str, box_type: &str,
args: &[ValueId], args: &[ValueId],
box_type_ids: &HashMap<String, i64>, box_type_ids: &HashMap<String, i64>,
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<(), String> { ) -> Result<(), String> {
match (box_type, args.len()) { match (box_type, args.len()) {
("StringBox", 1) => { ("StringBox", 1) => {
// Resolve as i8* string pointer (AOT string fast-path) // Resolve as i8* string pointer (AOT string fast-path)
let p = resolver.resolve_ptr(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let p = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
vmap.insert(dst, p.into()); vmap.insert(dst, p.into());
Ok(()) Ok(())
} }
@ -36,19 +51,47 @@ pub(in super::super) fn lower_newbox<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.birth_i64") .get_function("nyash.box.birth_i64")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.birth_i64", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.birth_i64", fnty, None)
});
let argc = i64t.const_int(args.len() as u64, false); let argc = i64t.const_int(args.len() as u64, false);
let mut a1 = i64t.const_zero(); let mut a1 = i64t.const_zero();
let mut a2 = i64t.const_zero(); let mut a2 = i64t.const_zero();
if args.len() >= 1 { if args.len() >= 1 {
a1 = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; a1 = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
} }
if args.len() >= 2 { if args.len() >= 2 {
a2 = resolver.resolve_i64(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?; a2 = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?;
} }
let tid = i64t.const_int(type_id as u64, true); let tid = i64t.const_int(type_id as u64, true);
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[tid.into(), argc.into(), a1.into(), a2.into()], "birth_i64")) .emit_instr(cur_bid, |b| {
b.build_call(
callee,
&[tid.into(), argc.into(), a1.into(), a2.into()],
"birth_i64",
)
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let h = call let h = call
.try_as_basic_value() .try_as_basic_value()
@ -70,10 +113,8 @@ pub(in super::super) fn lower_newbox<'ctx, 'b>(
} }
let type_id = *box_type_ids.get(box_type).unwrap_or(&0); let type_id = *box_type_ids.get(box_type).unwrap_or(&0);
// Temporary gate: allow forcing MapBox to plugin path explicitly // Temporary gate: allow forcing MapBox to plugin path explicitly
let force_plugin_map = std::env::var("NYASH_LLVM_FORCE_PLUGIN_MAP") let force_plugin_map =
.ok() std::env::var("NYASH_LLVM_FORCE_PLUGIN_MAP").ok().as_deref() == Some("1");
.as_deref()
== Some("1");
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
// Core-first: avoid birth_h for built-ins we provide directly (MapBox/ArrayBox) // Core-first: avoid birth_h for built-ins we provide directly (MapBox/ArrayBox)
let is_core_builtin = box_type == "MapBox" || box_type == "ArrayBox"; let is_core_builtin = box_type == "MapBox" || box_type == "ArrayBox";
@ -83,7 +124,11 @@ pub(in super::super) fn lower_newbox<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.birth_h") .get_function("nyash.box.birth_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.birth_h", fn_ty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.birth_h", fn_ty, None)
});
let tid = i64t.const_int(type_id as u64, true); let tid = i64t.const_int(type_id as u64, true);
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[tid.into()], "birth")) .emit_instr(cur_bid, |b| b.build_call(callee, &[tid.into()], "birth"))
@ -106,12 +151,20 @@ pub(in super::super) fn lower_newbox<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.env.box.new") .get_function("nyash.env.box.new")
.unwrap_or_else(|| codegen.module.add_function("nyash.env.box.new", fn_ty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.env.box.new", fn_ty, None)
});
let tn = cursor let tn = cursor
.emit_instr(cur_bid, |b| b.build_global_string_ptr(box_type, "box_type_name")) .emit_instr(cur_bid, |b| {
b.build_global_string_ptr(box_type, "box_type_name")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b.build_call(callee, &[tn.as_pointer_value().into()], "env_box_new")) .emit_instr(cur_bid, |b| {
b.build_call(callee, &[tn.as_pointer_value().into()], "env_box_new")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let h_i64 = call let h_i64 = call
.try_as_basic_value() .try_as_basic_value()

View File

@ -1,7 +1,7 @@
use std::collections::HashMap; use std::collections::HashMap;
use inkwell::values::{BasicValueEnum as BVE, IntValue};
use inkwell::values::PointerValue; use inkwell::values::PointerValue;
use inkwell::values::{BasicValueEnum as BVE, IntValue};
use crate::backend::llvm::context::CodegenContext; use crate::backend::llvm::context::CodegenContext;
use crate::mir::{BasicBlockId, ValueId}; use crate::mir::{BasicBlockId, ValueId};
@ -19,7 +19,11 @@ pub struct Resolver<'ctx> {
impl<'ctx> Resolver<'ctx> { impl<'ctx> Resolver<'ctx> {
pub fn new() -> Self { pub fn new() -> Self {
Self { i64_locals: HashMap::new(), ptr_locals: HashMap::new(), f64_locals: HashMap::new() } Self {
i64_locals: HashMap::new(),
ptr_locals: HashMap::new(),
f64_locals: HashMap::new(),
}
} }
/// Resolve a MIR value as an i64 dominating the current block. /// Resolve a MIR value as an i64 dominating the current block.
@ -32,13 +36,25 @@ impl<'ctx> Resolver<'ctx> {
vid: ValueId, vid: ValueId,
bb_map: &std::collections::HashMap<BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>,
preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>, preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>,
block_end_values: &std::collections::HashMap<BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
vmap: &std::collections::HashMap<ValueId, BVE<'ctx>>, vmap: &std::collections::HashMap<ValueId, BVE<'ctx>>,
) -> Result<IntValue<'ctx>, String> { ) -> Result<IntValue<'ctx>, String> {
if let Some(iv) = self.i64_locals.get(&(cur_bid, vid)).copied() { if let Some(iv) = self.i64_locals.get(&(cur_bid, vid)).copied() {
return Ok(iv); return Ok(iv);
} }
let iv = localize_to_i64(codegen, cursor, cur_bid, vid, bb_map, preds, block_end_values, vmap)?; let iv = localize_to_i64(
codegen,
cursor,
cur_bid,
vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
self.i64_locals.insert((cur_bid, vid), iv); self.i64_locals.insert((cur_bid, vid), iv);
Ok(iv) Ok(iv)
} }
@ -52,7 +68,10 @@ impl<'ctx> Resolver<'ctx> {
vid: ValueId, vid: ValueId,
bb_map: &std::collections::HashMap<BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>,
preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>, preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>,
block_end_values: &std::collections::HashMap<BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
vmap: &std::collections::HashMap<ValueId, BVE<'ctx>>, vmap: &std::collections::HashMap<ValueId, BVE<'ctx>>,
) -> Result<PointerValue<'ctx>, String> { ) -> Result<PointerValue<'ctx>, String> {
if let Some(pv) = self.ptr_locals.get(&(cur_bid, vid)).copied() { if let Some(pv) = self.ptr_locals.get(&(cur_bid, vid)).copied() {
@ -61,7 +80,16 @@ impl<'ctx> Resolver<'ctx> {
// Avoid using current vmap directly to keep dominance safe under multiple predecessors. // Avoid using current vmap directly to keep dominance safe under multiple predecessors.
// Strategy: localize as i64 (dominance-safe PHI), then convert to i8* in current block. // Strategy: localize as i64 (dominance-safe PHI), then convert to i8* in current block.
let i8p = codegen.context.ptr_type(inkwell::AddressSpace::from(0)); let i8p = codegen.context.ptr_type(inkwell::AddressSpace::from(0));
let iv = localize_to_i64(codegen, cursor, cur_bid, vid, bb_map, preds, block_end_values, vmap)?; let iv = localize_to_i64(
codegen,
cursor,
cur_bid,
vid,
bb_map,
preds,
block_end_values,
vmap,
)?;
let pv = cursor let pv = cursor
.emit_instr(cur_bid, |b| b.build_int_to_ptr(iv, i8p, "loc_i2p_dom")) .emit_instr(cur_bid, |b| b.build_int_to_ptr(iv, i8p, "loc_i2p_dom"))
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
@ -78,7 +106,10 @@ impl<'ctx> Resolver<'ctx> {
vid: ValueId, vid: ValueId,
bb_map: &std::collections::HashMap<BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>,
preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>, preds: &std::collections::HashMap<BasicBlockId, Vec<BasicBlockId>>,
block_end_values: &std::collections::HashMap<BasicBlockId, std::collections::HashMap<ValueId, BVE<'ctx>>>, block_end_values: &std::collections::HashMap<
BasicBlockId,
std::collections::HashMap<ValueId, BVE<'ctx>>,
>,
vmap: &std::collections::HashMap<ValueId, BVE<'ctx>>, vmap: &std::collections::HashMap<ValueId, BVE<'ctx>>,
) -> Result<inkwell::values::FloatValue<'ctx>, String> { ) -> Result<inkwell::values::FloatValue<'ctx>, String> {
if let Some(fv) = self.f64_locals.get(&(cur_bid, vid)).copied() { if let Some(fv) = self.f64_locals.get(&(cur_bid, vid)).copied() {
@ -89,9 +120,15 @@ impl<'ctx> Resolver<'ctx> {
let cur_llbb = *bb_map.get(&cur_bid).ok_or("cur bb missing")?; let cur_llbb = *bb_map.get(&cur_bid).ok_or("cur bb missing")?;
let pred_list = preds.get(&cur_bid).cloned().unwrap_or_default(); let pred_list = preds.get(&cur_bid).cloned().unwrap_or_default();
let saved_ip = codegen.builder.get_insert_block(); let saved_ip = codegen.builder.get_insert_block();
if let Some(first) = cur_llbb.get_first_instruction() { codegen.builder.position_before(&first); } if let Some(first) = cur_llbb.get_first_instruction() {
else { codegen.builder.position_at_end(cur_llbb); } codegen.builder.position_before(&first);
let phi = codegen.builder.build_phi(f64t, &format!("loc_f64_{}", vid.as_u32())).map_err(|e| e.to_string())?; } else {
codegen.builder.position_at_end(cur_llbb);
}
let phi = codegen
.builder
.build_phi(f64t, &format!("loc_f64_{}", vid.as_u32()))
.map_err(|e| e.to_string())?;
if pred_list.is_empty() { if pred_list.is_empty() {
// No predecessor: conservatively zerovmap には依存しない) // No predecessor: conservatively zerovmap には依存しない)
let z = f64t.const_zero(); let z = f64t.const_zero();
@ -106,10 +143,18 @@ impl<'ctx> Resolver<'ctx> {
let mut coerced = f64t.const_zero(); let mut coerced = f64t.const_zero();
cursor.with_block(*p, pred_bb, |c| { cursor.with_block(*p, pred_bb, |c| {
let term = unsafe { pred_bb.get_terminator() }; let term = unsafe { pred_bb.get_terminator() };
if let Some(t) = term { codegen.builder.position_before(&t); } else { c.position_at_end(pred_bb); } if let Some(t) = term {
codegen.builder.position_before(&t);
} else {
c.position_at_end(pred_bb);
}
coerced = match base { coerced = match base {
BVE::FloatValue(fv) => fv, BVE::FloatValue(fv) => fv,
BVE::IntValue(iv) => codegen.builder.build_signed_int_to_float(iv, f64t, "loc_i2f_p").map_err(|e| e.to_string()).unwrap(), BVE::IntValue(iv) => codegen
.builder
.build_signed_int_to_float(iv, f64t, "loc_i2f_p")
.map_err(|e| e.to_string())
.unwrap(),
BVE::PointerValue(_) => f64t.const_zero(), BVE::PointerValue(_) => f64t.const_zero(),
_ => f64t.const_zero(), _ => f64t.const_zero(),
}; };
@ -117,7 +162,9 @@ impl<'ctx> Resolver<'ctx> {
phi.add_incoming(&[(&coerced, pred_bb)]); phi.add_incoming(&[(&coerced, pred_bb)]);
} }
} }
if let Some(bb) = saved_ip { codegen.builder.position_at_end(bb); } if let Some(bb) = saved_ip {
codegen.builder.position_at_end(bb);
}
let out = phi.as_basic_value().into_float_value(); let out = phi.as_basic_value().into_float_value();
self.f64_locals.insert((cur_bid, vid), out); self.f64_locals.insert((cur_bid, vid), out);
Ok(out) Ok(out)

View File

@ -7,10 +7,13 @@ pub struct StrPtr<'ctx>(pub PointerValue<'ctx>);
impl<'ctx> StrHandle<'ctx> { impl<'ctx> StrHandle<'ctx> {
#[inline] #[inline]
pub fn as_i64(&self) -> IntValue<'ctx> { self.0 } pub fn as_i64(&self) -> IntValue<'ctx> {
self.0
}
} }
impl<'ctx> From<PointerValue<'ctx>> for StrPtr<'ctx> { impl<'ctx> From<PointerValue<'ctx>> for StrPtr<'ctx> {
fn from(p: PointerValue<'ctx>) -> Self { Self(p) } fn from(p: PointerValue<'ctx>) -> Self {
Self(p)
}
} }

View File

@ -2,10 +2,10 @@ use std::collections::HashMap;
use inkwell::{values::BasicValueEnum as BVE, AddressSpace}; use inkwell::{values::BasicValueEnum as BVE, AddressSpace};
use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
use super::builder_cursor::BuilderCursor; use super::builder_cursor::BuilderCursor;
use super::Resolver; use super::Resolver;
use crate::backend::llvm::context::CodegenContext;
use crate::mir::{function::MirFunction, BasicBlockId, ValueId};
/// Handle String-specific methods. Returns true if handled, false to let caller continue. /// Handle String-specific methods. Returns true if handled, false to let caller continue.
pub(super) fn try_handle_string_method<'ctx, 'b>( pub(super) fn try_handle_string_method<'ctx, 'b>(
@ -19,9 +19,15 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
box_val: &ValueId, box_val: &ValueId,
method: &str, method: &str,
args: &[ValueId], args: &[ValueId],
bb_map: &std::collections::HashMap<crate::mir::BasicBlockId, inkwell::basic_block::BasicBlock<'ctx>>, bb_map: &std::collections::HashMap<
crate::mir::BasicBlockId,
inkwell::basic_block::BasicBlock<'ctx>,
>,
preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>, preds: &std::collections::HashMap<crate::mir::BasicBlockId, Vec<crate::mir::BasicBlockId>>,
block_end_values: &std::collections::HashMap<crate::mir::BasicBlockId, std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>>, block_end_values: &std::collections::HashMap<
crate::mir::BasicBlockId,
std::collections::HashMap<ValueId, inkwell::values::BasicValueEnum<'ctx>>,
>,
) -> Result<bool, String> { ) -> Result<bool, String> {
// Receiver annotation check (kept for future diagnostics) // Receiver annotation check (kept for future diagnostics)
let _is_string_recv = match func.metadata.value_types.get(box_val) { let _is_string_recv = match func.metadata.value_types.get(box_val) {
@ -40,26 +46,58 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
// Resolve rhs as either pointer (string) or i64 (handle/int) // Resolve rhs as either pointer (string) or i64 (handle/int)
let rhs_val = match func.metadata.value_types.get(&args[0]) { let rhs_val = match func.metadata.value_types.get(&args[0]) {
Some(crate::mir::MirType::String) => { Some(crate::mir::MirType::String) => {
let p = resolver.resolve_ptr(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let p = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
BVE::PointerValue(p) BVE::PointerValue(p)
} }
_ => { _ => {
// Default to integer form for non-String metadata // Default to integer form for non-String metadata
let iv = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let iv = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
BVE::IntValue(iv) BVE::IntValue(iv)
} }
}; };
let lp = resolver.resolve_ptr(codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap)?; let lp = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?;
match (BVE::PointerValue(lp), rhs_val) { match (BVE::PointerValue(lp), rhs_val) {
(BVE::PointerValue(lp), BVE::PointerValue(rp)) => { (BVE::PointerValue(lp), BVE::PointerValue(rp)) => {
let fnty = i8p.fn_type(&[i8p.into(), i8p.into()], false); let fnty = i8p.fn_type(&[i8p.into(), i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_ss") .get_function("nyash.string.concat_ss")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_ss", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_ss", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[lp.into(), rp.into()], "concat_ss_call")) b.build_call(callee, &[lp.into(), rp.into()], "concat_ss_call")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -69,7 +107,9 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
// return as handle (i64) across blocks // return as handle (i64) across blocks
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let h = cursor let h = cursor
.emit_instr(cur_bid, |b| b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")) .emit_instr(cur_bid, |b| {
b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(*d, h.into()); vmap.insert(*d, h.into());
} }
@ -78,15 +118,29 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
(BVE::PointerValue(lp), BVE::IntValue(_ri)) => { (BVE::PointerValue(lp), BVE::IntValue(_ri)) => {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
// Localize rhs integer in current block via Resolver // Localize rhs integer in current block via Resolver
let ri = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let ri = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i8p.fn_type(&[i8p.into(), i64t.into()], false); let fnty = i8p.fn_type(&[i8p.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_si") .get_function("nyash.string.concat_si")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_si", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_si", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[lp.into(), ri.into()], "concat_si_call")) b.build_call(callee, &[lp.into(), ri.into()], "concat_si_call")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -95,7 +149,9 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
.ok_or("concat_si returned void".to_string())?; .ok_or("concat_si returned void".to_string())?;
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let h = cursor let h = cursor
.emit_instr(cur_bid, |b| b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")) .emit_instr(cur_bid, |b| {
b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(*d, h.into()); vmap.insert(*d, h.into());
} }
@ -104,15 +160,29 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
(BVE::PointerValue(_li_as_p), BVE::PointerValue(rp)) => { (BVE::PointerValue(_li_as_p), BVE::PointerValue(rp)) => {
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
// Localize receiver integer in current block (box_val) // Localize receiver integer in current block (box_val)
let li = resolver.resolve_i64(codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap)?; let li = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i8p.fn_type(&[i64t.into(), i8p.into()], false); let fnty = i8p.fn_type(&[i64t.into(), i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.concat_is") .get_function("nyash.string.concat_is")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.concat_is", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.concat_is", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[li.into(), rp.into()], "concat_is_call")) b.build_call(callee, &[li.into(), rp.into()], "concat_is_call")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -121,7 +191,9 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
.ok_or("concat_is returned void".to_string())?; .ok_or("concat_is returned void".to_string())?;
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let h = cursor let h = cursor
.emit_instr(cur_bid, |b| b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")) .emit_instr(cur_bid, |b| {
b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(*d, h.into()); vmap.insert(*d, h.into());
} }
@ -139,23 +211,53 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
// Prefer i64 handle from resolver; if metadata says String but actual is i8*, box it // Prefer i64 handle from resolver; if metadata says String but actual is i8*, box it
if let Some(crate::mir::MirType::String) = func.metadata.value_types.get(box_val) { if let Some(crate::mir::MirType::String) = func.metadata.value_types.get(box_val) {
// Receiver is a String: resolve pointer then box to i64 // Receiver is a String: resolve pointer then box to i64
let p = resolver.resolve_ptr(codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap)?; let p = resolver.resolve_ptr(
let fnty = i64t.fn_type(&[codegen.context.ptr_type(AddressSpace::from(0)).into()], false); codegen,
cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(
&[codegen.context.ptr_type(AddressSpace::from(0)).into()],
false,
);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.box.from_i8_string") .get_function("nyash.box.from_i8_string")
.unwrap_or_else(|| codegen.module.add_function("nyash.box.from_i8_string", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.box.from_i8_string", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[p.into()], "str_ptr_to_handle")) b.build_call(callee, &[p.into()], "str_ptr_to_handle")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
let rv = call let rv = call
.try_as_basic_value() .try_as_basic_value()
.left() .left()
.ok_or("from_i8_string returned void".to_string())?; .ok_or("from_i8_string returned void".to_string())?;
if let BVE::IntValue(iv) = rv { iv } else { return Err("from_i8_string ret expected i64".to_string()); } if let BVE::IntValue(iv) = rv {
iv
} else { } else {
resolver.resolve_i64(codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap)? return Err("from_i8_string ret expected i64".to_string());
}
} else {
resolver.resolve_i64(
codegen,
cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?
} }
}; };
// call i64 @nyash.string.len_h(i64) // call i64 @nyash.string.len_h(i64)
@ -163,10 +265,15 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.len_h") .get_function("nyash.string.len_h")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.len_h", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.len_h", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[recv_h.into()], "strlen_h")) b.build_call(callee, &[recv_h.into()], "strlen_h")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -186,18 +293,54 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let i8p = codegen.context.ptr_type(AddressSpace::from(0)); let i8p = codegen.context.ptr_type(AddressSpace::from(0));
// receiver pointer via Resolver // receiver pointer via Resolver
let recv_p = resolver.resolve_ptr(codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap)?; let recv_p = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?;
// Localize start/end indices to current block via sealed snapshots (i64) // Localize start/end indices to current block via sealed snapshots (i64)
let s = resolver.resolve_i64(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; let s = resolver.resolve_i64(
let e = resolver.resolve_i64(codegen, cursor, cur_bid, args[1], bb_map, preds, block_end_values, vmap)?; codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let e = resolver.resolve_i64(
codegen,
cursor,
cur_bid,
args[1],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i8p.fn_type(&[i8p.into(), i64t.into(), i64t.into()], false); let fnty = i8p.fn_type(&[i8p.into(), i64t.into(), i64t.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.substring_sii") .get_function("nyash.string.substring_sii")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.substring_sii", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.substring_sii", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[recv_p.into(), s.into(), e.into()], "substring_call")) b.build_call(
callee,
&[recv_p.into(), s.into(), e.into()],
"substring_call",
)
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call
@ -206,7 +349,9 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
.ok_or("substring returned void".to_string())?; .ok_or("substring returned void".to_string())?;
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let h = cursor let h = cursor
.emit_instr(cur_bid, |b| b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i_sub")) .emit_instr(cur_bid, |b| {
b.build_ptr_to_int(rv.into_pointer_value(), i64t, "str_ptr2i_sub")
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
vmap.insert(*d, h.into()); vmap.insert(*d, h.into());
} }
@ -221,17 +366,43 @@ pub(super) fn try_handle_string_method<'ctx, 'b>(
let i64t = codegen.context.i64_type(); let i64t = codegen.context.i64_type();
let i8p = codegen.context.ptr_type(AddressSpace::from(0)); let i8p = codegen.context.ptr_type(AddressSpace::from(0));
// receiver pointer via Resolver (String fast path) // receiver pointer via Resolver (String fast path)
let recv_p = resolver.resolve_ptr(codegen, cursor, cur_bid, *box_val, bb_map, preds, block_end_values, vmap)?; let recv_p = resolver.resolve_ptr(
let needle_p = resolver codegen,
.resolve_ptr(codegen, cursor, cur_bid, args[0], bb_map, preds, block_end_values, vmap)?; cursor,
cur_bid,
*box_val,
bb_map,
preds,
block_end_values,
vmap,
)?;
let needle_p = resolver.resolve_ptr(
codegen,
cursor,
cur_bid,
args[0],
bb_map,
preds,
block_end_values,
vmap,
)?;
let fnty = i64t.fn_type(&[i8p.into(), i8p.into()], false); let fnty = i64t.fn_type(&[i8p.into(), i8p.into()], false);
let callee = codegen let callee = codegen
.module .module
.get_function("nyash.string.lastIndexOf_ss") .get_function("nyash.string.lastIndexOf_ss")
.unwrap_or_else(|| codegen.module.add_function("nyash.string.lastIndexOf_ss", fnty, None)); .unwrap_or_else(|| {
codegen
.module
.add_function("nyash.string.lastIndexOf_ss", fnty, None)
});
let call = cursor let call = cursor
.emit_instr(cur_bid, |b| b .emit_instr(cur_bid, |b| {
.build_call(callee, &[recv_p.into(), needle_p.into()], "lastindexof_call")) b.build_call(
callee,
&[recv_p.into(), needle_p.into()],
"lastindexof_call",
)
})
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
if let Some(d) = dst { if let Some(d) = dst {
let rv = call let rv = call

File diff suppressed because it is too large Load Diff

View File

@ -0,0 +1,125 @@
use super::sanitize_symbol;
use crate::backend::llvm::context::CodegenContext;
use inkwell::values::BasicValueEnum;
pub(super) fn emit_wrapper_and_object<'ctx>(
codegen: &CodegenContext<'ctx>,
entry_name: &str,
output_path: &str,
) -> Result<(), String> {
let i64t = codegen.context.i64_type();
let ny_main_ty = i64t.fn_type(&[], false);
let ny_main = codegen.module.add_function("ny_main", ny_main_ty, None);
let entry_bb = codegen.context.append_basic_block(ny_main, "entry");
codegen.builder.position_at_end(entry_bb);
let entry_sym = format!("ny_f_{}", sanitize_symbol(entry_name));
let entry_fn = codegen
.module
.get_function(&entry_sym)
.ok_or_else(|| format!("entry function symbol not found: {}", entry_sym))?;
let call = codegen
.builder
.build_call(entry_fn, &[], "call_main")
.map_err(|e| e.to_string())?;
let rv = call.try_as_basic_value().left();
let ret_v = if let Some(v) = rv {
match v {
BasicValueEnum::IntValue(iv) => {
if iv.get_type().get_bit_width() == 64 {
iv
} else {
codegen
.builder
.build_int_z_extend(iv, i64t, "ret_zext")
.map_err(|e| e.to_string())?
}
}
BasicValueEnum::PointerValue(pv) => codegen
.builder
.build_ptr_to_int(pv, i64t, "ret_p2i")
.map_err(|e| e.to_string())?,
BasicValueEnum::FloatValue(fv) => codegen
.builder
.build_float_to_signed_int(fv, i64t, "ret_f2i")
.map_err(|e| e.to_string())?,
_ => i64t.const_zero(),
}
} else {
i64t.const_zero()
};
codegen
.builder
.build_return(Some(&ret_v))
.map_err(|e| e.to_string())?;
if !ny_main.verify(true) {
return Err("ny_main verification failed".to_string());
}
let verbose = std::env::var("NYASH_CLI_VERBOSE").ok().as_deref() == Some("1");
if verbose {
eprintln!("[LLVM] emitting object to {} (begin)", output_path);
}
match codegen.target_machine.write_to_file(
&codegen.module,
inkwell::targets::FileType::Object,
std::path::Path::new(output_path),
) {
Ok(_) => {
if std::fs::metadata(output_path).is_err() {
let buf = codegen
.target_machine
.write_to_memory_buffer(&codegen.module, inkwell::targets::FileType::Object)
.map_err(|e| format!("Failed to get object buffer: {}", e))?;
std::fs::write(output_path, buf.as_slice())
.map_err(|e| format!("Failed to write object to '{}': {}", output_path, e))?;
if verbose {
eprintln!(
"[LLVM] wrote object via memory buffer fallback: {} ({} bytes)",
output_path,
buf.get_size()
);
}
} else if verbose {
if let Ok(meta) = std::fs::metadata(output_path) {
eprintln!(
"[LLVM] wrote object via file API: {} ({} bytes)",
output_path,
meta.len()
);
}
}
if verbose {
eprintln!("[LLVM] emit complete (Ok branch) for {}", output_path);
}
Ok(())
}
Err(e) => {
let buf = codegen
.target_machine
.write_to_memory_buffer(&codegen.module, inkwell::targets::FileType::Object)
.map_err(|ee| {
format!(
"Failed to write object ({}); and memory buffer failed: {}",
e, ee
)
})?;
std::fs::write(output_path, buf.as_slice()).map_err(|ee| {
format!(
"Failed to write object to '{}': {} (original error: {})",
output_path, ee, e
)
})?;
if verbose {
eprintln!(
"[LLVM] wrote object via error fallback: {} ({} bytes)",
output_path,
buf.get_size()
);
eprintln!(
"[LLVM] emit complete (Err branch handled) for {}",
output_path
);
}
Ok(())
}
}
}

View File

@ -54,9 +54,7 @@ pub(super) fn to_i64_any<'ctx>(
let castp = builder let castp = builder
.build_pointer_cast(tmp, fptr_ty, "i64p_to_f64p") .build_pointer_cast(tmp, fptr_ty, "i64p_to_f64p")
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
builder builder.build_store(castp, fv).map_err(|e| e.to_string())?;
.build_store(castp, fv)
.map_err(|e| e.to_string())?;
builder builder
.build_load(i64t, tmp, "ld_f2i") .build_load(i64t, tmp, "ld_f2i")
.map_err(|e| e.to_string())? .map_err(|e| e.to_string())?

View File

@ -1,27 +1,39 @@
use std::collections::HashMap; use crate::mir::instruction::{ConstValue, MirInstruction};
use crate::mir::ValueId; use crate::mir::ValueId;
use crate::mir::instruction::{MirInstruction, ConstValue}; use std::collections::HashMap;
pub(super) fn sanitize_symbol(name: &str) -> String { pub(super) fn sanitize_symbol(name: &str) -> String {
name.chars() name.chars()
.map(|c| match c { '.' | '/' | '-' => '_', other => other }) .map(|c| match c {
'.' | '/' | '-' => '_',
other => other,
})
.collect() .collect()
} }
pub(super) fn build_const_str_map(f: &crate::mir::function::MirFunction) -> HashMap<ValueId, String> { pub(super) fn build_const_str_map(
f: &crate::mir::function::MirFunction,
) -> HashMap<ValueId, String> {
let mut m = HashMap::new(); let mut m = HashMap::new();
for bid in f.block_ids() { for bid in f.block_ids() {
if let Some(b) = f.blocks.get(&bid) { if let Some(b) = f.blocks.get(&bid) {
for inst in &b.instructions { for inst in &b.instructions {
if let MirInstruction::Const { dst, value: ConstValue::String(s) } = inst { if let MirInstruction::Const {
dst,
value: ConstValue::String(s),
} = inst
{
m.insert(*dst, s.clone()); m.insert(*dst, s.clone());
} }
} }
if let Some(MirInstruction::Const { dst, value: ConstValue::String(s) }) = &b.terminator { if let Some(MirInstruction::Const {
dst,
value: ConstValue::String(s),
}) = &b.terminator
{
m.insert(*dst, s.clone()); m.insert(*dst, s.clone());
} }
} }
} }
m m
} }

View File

@ -3,6 +3,7 @@ use crate::mir::{
MirModule, MirPrinter, MirType, ValueId, MirModule, MirPrinter, MirType, ValueId,
}; };
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::collections::HashMap;
#[derive(Debug, Deserialize, Serialize)] #[derive(Debug, Deserialize, Serialize)]
struct ProgramV0 { struct ProgramV0 {
@ -125,12 +126,97 @@ struct LoopContext {
exit_bb: BasicBlockId, exit_bb: BasicBlockId,
} }
struct BridgeEnv {
throw_enabled: bool,
mir_no_phi: bool,
allow_me_dummy: bool,
me_class: String,
}
impl BridgeEnv {
fn load() -> Self {
Self {
throw_enabled: std::env::var("NYASH_BRIDGE_THROW_ENABLE").ok().as_deref() == Some("1"),
mir_no_phi: crate::config::env::mir_no_phi(),
allow_me_dummy: std::env::var("NYASH_BRIDGE_ME_DUMMY").ok().as_deref() == Some("1"),
me_class: std::env::var("NYASH_BRIDGE_ME_CLASS").unwrap_or_else(|_| "Main".to_string()),
}
}
}
trait VarScope {
fn resolve(
&mut self,
env: &BridgeEnv,
f: &mut MirFunction,
cur_bb: BasicBlockId,
name: &str,
) -> Result<Option<ValueId>, String>;
}
struct NoVars;
impl VarScope for NoVars {
fn resolve(
&mut self,
_env: &BridgeEnv,
_f: &mut MirFunction,
_cur_bb: BasicBlockId,
name: &str,
) -> Result<Option<ValueId>, String> {
Err(format!("undefined variable in this context: {}", name))
}
}
struct MapVars<'a> {
vars: &'a mut HashMap<String, ValueId>,
}
impl<'a> MapVars<'a> {
fn new(vars: &'a mut HashMap<String, ValueId>) -> Self {
Self { vars }
}
}
impl<'a> VarScope for MapVars<'a> {
fn resolve(
&mut self,
env: &BridgeEnv,
f: &mut MirFunction,
cur_bb: BasicBlockId,
name: &str,
) -> Result<Option<ValueId>, String> {
if let Some(&vid) = self.vars.get(name) {
return Ok(Some(vid));
}
if name == "me" {
if env.allow_me_dummy {
let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur_bb) {
bb.add_instruction(MirInstruction::NewBox {
dst,
box_type: env.me_class.clone(),
args: vec![],
});
}
self.vars.insert(name.to_string(), dst);
Ok(Some(dst))
} else {
Err("undefined 'me' outside box context (set NYASH_BRIDGE_ME_DUMMY=1 to inject placeholder)".into())
}
} else {
Ok(None)
}
}
}
fn lower_throw( fn lower_throw(
env: &BridgeEnv,
f: &mut MirFunction, f: &mut MirFunction,
cur_bb: BasicBlockId, cur_bb: BasicBlockId,
exception_value: ValueId, exception_value: ValueId,
) -> (ValueId, BasicBlockId) { ) -> (ValueId, BasicBlockId) {
if std::env::var("NYASH_BRIDGE_THROW_ENABLE").ok().as_deref() == Some("1") { if env.throw_enabled {
if let Some(bb) = f.get_block_mut(cur_bb) { if let Some(bb) = f.get_block_mut(cur_bb) {
bb.set_terminator(MirInstruction::Throw { bb.set_terminator(MirInstruction::Throw {
exception: exception_value, exception: exception_value,
@ -158,6 +244,7 @@ pub fn parse_json_v0_to_module(json: &str) -> Result<MirModule, String> {
} }
// Create module and main function // Create module and main function
let mut module = MirModule::new("ny_json_v0".into()); let mut module = MirModule::new("ny_json_v0".into());
let env = BridgeEnv::load();
let sig = FunctionSignature { let sig = FunctionSignature {
name: "main".into(), name: "main".into(),
params: vec![], params: vec![],
@ -172,12 +259,17 @@ pub fn parse_json_v0_to_module(json: &str) -> Result<MirModule, String> {
} }
// Variable map for simple locals (Stage-2; currently minimal) // Variable map for simple locals (Stage-2; currently minimal)
let mut var_map: std::collections::HashMap<String, crate::mir::ValueId> = let mut var_map: HashMap<String, ValueId> = HashMap::new();
std::collections::HashMap::new();
let mut loop_stack: Vec<LoopContext> = Vec::new(); let mut loop_stack: Vec<LoopContext> = Vec::new();
let start_bb = f.entry_block; let start_bb = f.entry_block;
let end_bb = let end_bb = lower_stmt_list_with_vars(
lower_stmt_list_with_vars(&mut f, start_bb, &prog.body, &mut var_map, &mut loop_stack)?; &mut f,
start_bb,
&prog.body,
&mut var_map,
&mut loop_stack,
&env,
)?;
// Ensure function terminates: add `ret 0` to last un-terminated block (prefer end_bb else entry) // Ensure function terminates: add `ret 0` to last un-terminated block (prefer end_bb else entry)
let need_default_ret = f.blocks.iter().any(|(_k, b)| !b.is_terminated()); let need_default_ret = f.blocks.iter().any(|(_k, b)| !b.is_terminated());
if need_default_ret { if need_default_ret {
@ -211,11 +303,13 @@ fn next_block_id(f: &MirFunction) -> BasicBlockId {
BasicBlockId::new(mx) BasicBlockId::new(mx)
} }
fn lower_expr( fn lower_expr_with_scope<S: VarScope>(
env: &BridgeEnv,
f: &mut MirFunction, f: &mut MirFunction,
cur_bb: BasicBlockId, cur_bb: BasicBlockId,
e: &ExprV0, e: &ExprV0,
) -> Result<(crate::mir::ValueId, BasicBlockId), String> { vars: &mut S,
) -> Result<(ValueId, BasicBlockId), String> {
match e { match e {
ExprV0::Int { value } => { ExprV0::Int { value } => {
// Accept number or stringified digits // Accept number or stringified digits
@ -256,8 +350,8 @@ fn lower_expr(
Ok((dst, cur_bb)) Ok((dst, cur_bb))
} }
ExprV0::Binary { op, lhs, rhs } => { ExprV0::Binary { op, lhs, rhs } => {
let (l, cur_after_l) = lower_expr(f, cur_bb, lhs)?; let (l, cur_after_l) = lower_expr_with_scope(env, f, cur_bb, lhs, vars)?;
let (r, cur_after_r) = lower_expr(f, cur_after_l, rhs)?; let (r, cur_after_r) = lower_expr_with_scope(env, f, cur_after_l, rhs, vars)?;
let bop = match op.as_str() { let bop = match op.as_str() {
"+" => BinaryOp::Add, "+" => BinaryOp::Add,
"-" => BinaryOp::Sub, "-" => BinaryOp::Sub,
@ -281,7 +375,7 @@ fn lower_expr(
method, method,
args, args,
} => { } => {
let (arg_ids, cur2) = lower_args(f, cur_bb, args)?; let (arg_ids, cur2) = lower_args_with_scope(env, f, cur_bb, args, vars)?;
let dst = f.next_value_id(); let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur2) { if let Some(bb) = f.get_block_mut(cur2) {
bb.add_instruction(MirInstruction::ExternCall { bb.add_instruction(MirInstruction::ExternCall {
@ -295,8 +389,8 @@ fn lower_expr(
Ok((dst, cur2)) Ok((dst, cur2))
} }
ExprV0::Compare { op, lhs, rhs } => { ExprV0::Compare { op, lhs, rhs } => {
let (l, cur_after_l) = lower_expr(f, cur_bb, lhs)?; let (l, cur_after_l) = lower_expr_with_scope(env, f, cur_bb, lhs, vars)?;
let (r, cur_after_r) = lower_expr(f, cur_after_l, rhs)?; let (r, cur_after_r) = lower_expr_with_scope(env, f, cur_after_l, rhs, vars)?;
let cop = match op.as_str() { let cop = match op.as_str() {
"==" => crate::mir::CompareOp::Eq, "==" => crate::mir::CompareOp::Eq,
"!=" => crate::mir::CompareOp::Ne, "!=" => crate::mir::CompareOp::Ne,
@ -319,7 +413,7 @@ fn lower_expr(
} }
ExprV0::Logical { op, lhs, rhs } => { ExprV0::Logical { op, lhs, rhs } => {
// Short-circuit boolean logic with branches (+phi or edge-copy) // Short-circuit boolean logic with branches (+phi or edge-copy)
let (l, cur_after_l) = lower_expr(f, cur_bb, lhs)?; let (l, cur_after_l) = lower_expr_with_scope(env, f, cur_bb, lhs, vars)?;
let rhs_bb = next_block_id(f); let rhs_bb = next_block_id(f);
let fall_bb = BasicBlockId::new(rhs_bb.0 + 1); let fall_bb = BasicBlockId::new(rhs_bb.0 + 1);
let merge_bb = BasicBlockId::new(rhs_bb.0 + 2); let merge_bb = BasicBlockId::new(rhs_bb.0 + 2);
@ -380,7 +474,7 @@ fn lower_expr(
bb.set_terminator(MirInstruction::Jump { target: merge_bb }); bb.set_terminator(MirInstruction::Jump { target: merge_bb });
} }
// evaluate rhs starting at rhs_bb and ensure the terminal block jumps to merge // evaluate rhs starting at rhs_bb and ensure the terminal block jumps to merge
let (rval, rhs_end) = lower_expr(f, rhs_bb, rhs)?; let (rval, rhs_end) = lower_expr_with_scope(env, f, rhs_bb, rhs, vars)?;
if let Some(bb) = f.get_block_mut(rhs_end) { if let Some(bb) = f.get_block_mut(rhs_end) {
if !bb.is_terminated() { if !bb.is_terminated() {
bb.set_terminator(MirInstruction::Jump { target: merge_bb }); bb.set_terminator(MirInstruction::Jump { target: merge_bb });
@ -393,15 +487,21 @@ fn lower_expr(
); );
} }
// Merge: PHI または edge-copy で合流値を定義 // Merge: PHI または edge-copy で合流値を定義
let no_phi = crate::config::env::mir_no_phi(); let no_phi = env.mir_no_phi;
let out = f.next_value_id(); let out = f.next_value_id();
if no_phi { if no_phi {
// Edge copies in predecessors // Edge copies in predecessors
if let Some(bb) = f.get_block_mut(fall_bb) { if let Some(bb) = f.get_block_mut(fall_bb) {
bb.add_instruction(MirInstruction::Copy { dst: out, src: cdst }); bb.add_instruction(MirInstruction::Copy {
dst: out,
src: cdst,
});
} }
if let Some(bb) = f.get_block_mut(rhs_end) { if let Some(bb) = f.get_block_mut(rhs_end) {
bb.add_instruction(MirInstruction::Copy { dst: out, src: rval }); bb.add_instruction(MirInstruction::Copy {
dst: out,
src: rval,
});
} }
} else if let Some(bb) = f.get_block_mut(merge_bb) { } else if let Some(bb) = f.get_block_mut(merge_bb) {
let mut inputs: Vec<(BasicBlockId, ValueId)> = vec![(fall_bb, cdst)]; let mut inputs: Vec<(BasicBlockId, ValueId)> = vec![(fall_bb, cdst)];
@ -432,7 +532,7 @@ fn lower_expr(
// For each element: eval then push // For each element: eval then push
let mut cur = cur_bb; let mut cur = cur_bb;
for e in args { for e in args {
let (v, c) = lower_expr(f, cur, e)?; let (v, c) = lower_expr_with_scope(env, f, cur, e, vars)?;
cur = c; cur = c;
let tmp = f.next_value_id(); let tmp = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) { if let Some(bb) = f.get_block_mut(cur) {
@ -462,9 +562,9 @@ fn lower_expr(
let mut it = args.iter(); let mut it = args.iter();
while let Some(k) = it.next() { while let Some(k) = it.next() {
if let Some(v) = it.next() { if let Some(v) = it.next() {
let (kv, cur2) = lower_expr(f, cur, k)?; let (kv, cur2) = lower_expr_with_scope(env, f, cur, k, vars)?;
cur = cur2; cur = cur2;
let (vv, cur3) = lower_expr(f, cur, v)?; let (vv, cur3) = lower_expr_with_scope(env, f, cur, v, vars)?;
cur = cur3; cur = cur3;
let tmp = f.next_value_id(); let tmp = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) { if let Some(bb) = f.get_block_mut(cur) {
@ -484,7 +584,7 @@ fn lower_expr(
return Ok((mapv, cur)); return Ok((mapv, cur));
} }
// Fallback: treat as normal dynamic call // Fallback: treat as normal dynamic call
let (arg_ids, cur) = lower_args(f, cur_bb, args)?; let (arg_ids, cur) = lower_args_with_scope(env, f, cur_bb, args, vars)?;
let fun_val = f.next_value_id(); let fun_val = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) { if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::Const { bb.add_instruction(MirInstruction::Const {
@ -509,7 +609,7 @@ fn lower_expr(
matches!(&**recv, ExprV0::New { class, .. } if class == "ConsoleBox"); matches!(&**recv, ExprV0::New { class, .. } if class == "ConsoleBox");
if recv_is_console_new && (method == "println" || method == "print" || method == "log") if recv_is_console_new && (method == "println" || method == "print" || method == "log")
{ {
let (arg_ids, cur2) = lower_args(f, cur_bb, args)?; let (arg_ids, cur2) = lower_args_with_scope(env, f, cur_bb, args, vars)?;
let dst = f.next_value_id(); let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur2) { if let Some(bb) = f.get_block_mut(cur2) {
bb.add_instruction(MirInstruction::ExternCall { bb.add_instruction(MirInstruction::ExternCall {
@ -522,8 +622,8 @@ fn lower_expr(
} }
return Ok((dst, cur2)); return Ok((dst, cur2));
} }
let (recv_v, cur) = lower_expr(f, cur_bb, recv)?; let (recv_v, cur) = lower_expr_with_scope(env, f, cur_bb, recv, vars)?;
let (arg_ids, cur2) = lower_args(f, cur, args)?; let (arg_ids, cur2) = lower_args_with_scope(env, f, cur, args, vars)?;
let dst = f.next_value_id(); let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur2) { if let Some(bb) = f.get_block_mut(cur2) {
bb.add_instruction(MirInstruction::BoxCall { bb.add_instruction(MirInstruction::BoxCall {
@ -538,7 +638,7 @@ fn lower_expr(
Ok((dst, cur2)) Ok((dst, cur2))
} }
ExprV0::New { class, args } => { ExprV0::New { class, args } => {
let (arg_ids, cur) = lower_args(f, cur_bb, args)?; let (arg_ids, cur) = lower_args_with_scope(env, f, cur_bb, args, vars)?;
let dst = f.next_value_id(); let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) { if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::NewBox { bb.add_instruction(MirInstruction::NewBox {
@ -549,292 +649,50 @@ fn lower_expr(
} }
Ok((dst, cur)) Ok((dst, cur))
} }
ExprV0::Var { name } => Err(format!("undefined variable in this context: {}", name)), ExprV0::Var { name } => match vars.resolve(env, f, cur_bb, name)? {
Some(v) => Ok((v, cur_bb)),
None => Err(format!("undefined variable: {}", name)),
},
ExprV0::Throw { expr } => { ExprV0::Throw { expr } => {
let (exc, cur) = lower_expr(f, cur_bb, expr)?; let (exc, cur) = lower_expr_with_scope(env, f, cur_bb, expr, vars)?;
let (dst, cur) = lower_throw(f, cur, exc); let (dst, cur) = lower_throw(env, f, cur, exc);
Ok((dst, cur)) Ok((dst, cur))
} }
} }
} }
fn lower_expr_with_vars( fn lower_expr(
env: &BridgeEnv,
f: &mut MirFunction, f: &mut MirFunction,
cur_bb: BasicBlockId, cur_bb: BasicBlockId,
e: &ExprV0, e: &ExprV0,
vars: &mut std::collections::HashMap<String, crate::mir::ValueId>, ) -> Result<(ValueId, BasicBlockId), String> {
) -> Result<(crate::mir::ValueId, BasicBlockId), String> { let mut scope = NoVars;
match e { lower_expr_with_scope(env, f, cur_bb, e, &mut scope)
ExprV0::Var { name } => { }
if let Some(&vid) = vars.get(name) {
return Ok((vid, cur_bb)); fn lower_expr_with_vars(
} env: &BridgeEnv,
if name == "me" { f: &mut MirFunction,
// Optional gate: allow a dummy 'me' instance for Stage-2 JSON smoke cur_bb: BasicBlockId,
if std::env::var("NYASH_BRIDGE_ME_DUMMY").ok().as_deref() == Some("1") { e: &ExprV0,
let class = std::env::var("NYASH_BRIDGE_ME_CLASS") vars: &mut HashMap<String, ValueId>,
.unwrap_or_else(|_| "Main".to_string()); ) -> Result<(ValueId, BasicBlockId), String> {
let dst = f.next_value_id(); let mut scope = MapVars::new(vars);
if let Some(bb) = f.get_block_mut(cur_bb) { lower_expr_with_scope(env, f, cur_bb, e, &mut scope)
bb.add_instruction(MirInstruction::NewBox {
dst,
box_type: class,
args: vec![],
});
}
vars.insert("me".to_string(), dst);
return Ok((dst, cur_bb));
} else {
return Err("undefined 'me' outside box context (set NYASH_BRIDGE_ME_DUMMY=1 to inject placeholder)".into());
}
}
Err(format!("undefined variable: {}", name))
}
ExprV0::Throw { expr } => {
let (exc, cur) = lower_expr_with_vars(f, cur_bb, expr, vars)?;
let (dst, cur) = lower_throw(f, cur, exc);
Ok((dst, cur))
}
ExprV0::Call { name, args } => {
// Special: array literal lowering in vars context
if name == "array.of" {
let arr = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur_bb) {
bb.add_instruction(MirInstruction::NewBox {
dst: arr,
box_type: "ArrayBox".into(),
args: vec![],
});
}
let mut cur = cur_bb;
for e in args {
let (v, c) = lower_expr_with_vars(f, cur, e, vars)?;
cur = c;
let tmp = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::BoxCall {
dst: Some(tmp),
box_val: arr,
method: "push".into(),
method_id: None,
args: vec![v],
effects: EffectMask::READ,
});
}
}
return Ok((arr, cur));
}
// Special: map literal lowering in vars context
if name == "map.of" {
let mapv = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur_bb) {
bb.add_instruction(MirInstruction::NewBox {
dst: mapv,
box_type: "MapBox".into(),
args: vec![],
});
}
let mut cur = cur_bb;
let mut it = args.iter();
while let Some(k) = it.next() {
if let Some(v) = it.next() {
let (kv, cur2) = lower_expr_with_vars(f, cur, k, vars)?;
cur = cur2;
let (vv, cur3) = lower_expr_with_vars(f, cur, v, vars)?;
cur = cur3;
let tmp = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::BoxCall {
dst: Some(tmp),
box_val: mapv,
method: "set".into(),
method_id: None,
args: vec![kv, vv],
effects: EffectMask::READ,
});
}
} else {
break;
}
}
return Ok((mapv, cur));
}
// Lower args
let (arg_ids, cur) = lower_args_with_vars(f, cur_bb, args, vars)?;
// Encode as: const fun_name; call
let fun_val = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::Const {
dst: fun_val,
value: ConstValue::String(name.clone()),
});
}
let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::Call {
dst: Some(dst),
func: fun_val,
args: arg_ids,
effects: EffectMask::READ,
});
}
Ok((dst, cur))
}
ExprV0::Method { recv, method, args } => {
let recv_is_console_new =
matches!(&**recv, ExprV0::New { class, .. } if class == "ConsoleBox");
if recv_is_console_new && (method == "println" || method == "print" || method == "log")
{
let (arg_ids, cur2) = lower_args_with_vars(f, cur_bb, args, vars)?;
let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur2) {
bb.add_instruction(MirInstruction::ExternCall {
dst: Some(dst),
iface_name: "env.console".into(),
method_name: "log".into(),
args: arg_ids,
effects: EffectMask::READ,
});
}
return Ok((dst, cur2));
}
let (recv_v, cur) = lower_expr_with_vars(f, cur_bb, recv, vars)?;
let (arg_ids, cur2) = lower_args_with_vars(f, cur, args, vars)?;
let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur2) {
bb.add_instruction(MirInstruction::BoxCall {
dst: Some(dst),
box_val: recv_v,
method: method.clone(),
method_id: None,
args: arg_ids,
effects: EffectMask::READ,
});
}
Ok((dst, cur2))
}
ExprV0::New { class, args } => {
let (arg_ids, cur) = lower_args_with_vars(f, cur_bb, args, vars)?;
let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::NewBox {
dst,
box_type: class.clone(),
args: arg_ids,
});
}
Ok((dst, cur))
}
ExprV0::Binary { op, lhs, rhs } => {
let (l, cur_after_l) = lower_expr_with_vars(f, cur_bb, lhs, vars)?;
let (r, cur_after_r) = lower_expr_with_vars(f, cur_after_l, rhs, vars)?;
let bop = match op.as_str() {
"+" => BinaryOp::Add,
"-" => BinaryOp::Sub,
"*" => BinaryOp::Mul,
"/" => BinaryOp::Div,
_ => return Err("unsupported op".into()),
};
let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur_after_r) {
bb.add_instruction(MirInstruction::BinOp {
dst,
op: bop,
lhs: l,
rhs: r,
});
}
Ok((dst, cur_after_r))
}
ExprV0::Compare { op, lhs, rhs } => {
let (l, cur_after_l) = lower_expr_with_vars(f, cur_bb, lhs, vars)?;
let (r, cur_after_r) = lower_expr_with_vars(f, cur_after_l, rhs, vars)?;
let cop = match op.as_str() {
"==" => crate::mir::CompareOp::Eq,
"!=" => crate::mir::CompareOp::Ne,
"<" => crate::mir::CompareOp::Lt,
"<=" => crate::mir::CompareOp::Le,
">" => crate::mir::CompareOp::Gt,
">=" => crate::mir::CompareOp::Ge,
_ => return Err("unsupported compare op".into()),
};
let dst = f.next_value_id();
if let Some(bb) = f.get_block_mut(cur_after_r) {
bb.add_instruction(MirInstruction::Compare {
dst,
op: cop,
lhs: l,
rhs: r,
});
}
Ok((dst, cur_after_r))
}
ExprV0::Logical { op, lhs, rhs } => {
let (l, cur_after_l) = lower_expr_with_vars(f, cur_bb, lhs, vars)?;
let rhs_bb = next_block_id(f);
let fall_bb = BasicBlockId::new(rhs_bb.0 + 1);
let merge_bb = BasicBlockId::new(rhs_bb.0 + 2);
f.add_block(crate::mir::BasicBlock::new(rhs_bb));
f.add_block(crate::mir::BasicBlock::new(fall_bb));
f.add_block(crate::mir::BasicBlock::new(merge_bb));
let is_and = matches!(op.as_str(), "&&" | "and");
if let Some(bb) = f.get_block_mut(cur_after_l) {
if is_and {
bb.set_terminator(MirInstruction::Branch {
condition: l,
then_bb: rhs_bb,
else_bb: fall_bb,
});
} else {
bb.set_terminator(MirInstruction::Branch {
condition: l,
then_bb: fall_bb,
else_bb: rhs_bb,
});
}
}
let cdst = f.next_value_id();
if let Some(bb) = f.get_block_mut(fall_bb) {
let cval = if is_and {
ConstValue::Bool(false)
} else {
ConstValue::Bool(true)
};
bb.add_instruction(MirInstruction::Const {
dst: cdst,
value: cval,
});
bb.set_terminator(MirInstruction::Jump { target: merge_bb });
}
let (rval, rhs_end) = lower_expr_with_vars(f, rhs_bb, rhs, vars)?;
if let Some(bb) = f.get_block_mut(rhs_end) {
if !bb.is_terminated() {
bb.set_terminator(MirInstruction::Jump { target: merge_bb });
}
}
let out = f.next_value_id();
if let Some(bb) = f.get_block_mut(merge_bb) {
bb.insert_instruction_after_phis(MirInstruction::Phi {
dst: out,
inputs: vec![(rhs_end, rval), (fall_bb, cdst)],
});
}
Ok((out, merge_bb))
}
_ => lower_expr(f, cur_bb, e),
}
} }
fn lower_stmt_with_vars( fn lower_stmt_with_vars(
f: &mut MirFunction, f: &mut MirFunction,
cur_bb: BasicBlockId, cur_bb: BasicBlockId,
s: &StmtV0, s: &StmtV0,
vars: &mut std::collections::HashMap<String, crate::mir::ValueId>, vars: &mut HashMap<String, ValueId>,
loop_stack: &mut Vec<LoopContext>, loop_stack: &mut Vec<LoopContext>,
env: &BridgeEnv,
) -> Result<BasicBlockId, String> { ) -> Result<BasicBlockId, String> {
match s { match s {
StmtV0::Return { expr } => { StmtV0::Return { expr } => {
let (v, cur) = lower_expr_with_vars(f, cur_bb, expr, vars)?; let (v, cur) = lower_expr_with_vars(env, f, cur_bb, expr, vars)?;
if let Some(bb) = f.get_block_mut(cur) { if let Some(bb) = f.get_block_mut(cur) {
bb.set_terminator(MirInstruction::Return { value: Some(v) }); bb.set_terminator(MirInstruction::Return { value: Some(v) });
} }
@ -845,7 +703,7 @@ fn lower_stmt_with_vars(
method, method,
args, args,
} => { } => {
let (arg_ids, cur) = lower_args_with_vars(f, cur_bb, args, vars)?; let (arg_ids, cur) = lower_args_with_vars(env, f, cur_bb, args, vars)?;
if let Some(bb) = f.get_block_mut(cur) { if let Some(bb) = f.get_block_mut(cur) {
bb.add_instruction(MirInstruction::ExternCall { bb.add_instruction(MirInstruction::ExternCall {
dst: None, dst: None,
@ -858,11 +716,11 @@ fn lower_stmt_with_vars(
Ok(cur) Ok(cur)
} }
StmtV0::Expr { expr } => { StmtV0::Expr { expr } => {
let (_v, cur) = lower_expr_with_vars(f, cur_bb, expr, vars)?; let (_v, cur) = lower_expr_with_vars(env, f, cur_bb, expr, vars)?;
Ok(cur) Ok(cur)
} }
StmtV0::Local { name, expr } => { StmtV0::Local { name, expr } => {
let (v, cur) = lower_expr_with_vars(f, cur_bb, expr, vars)?; let (v, cur) = lower_expr_with_vars(env, f, cur_bb, expr, vars)?;
vars.insert(name.clone(), v); vars.insert(name.clone(), v);
Ok(cur) Ok(cur)
} }
@ -918,10 +776,16 @@ fn lower_stmt_with_vars(
if !try_enabled || catches.is_empty() || catches.len() > 1 { if !try_enabled || catches.is_empty() || catches.len() > 1 {
let mut tmp_vars = vars.clone(); let mut tmp_vars = vars.clone();
let mut next_bb = let mut next_bb =
lower_stmt_list_with_vars(f, cur_bb, try_body, &mut tmp_vars, loop_stack)?; lower_stmt_list_with_vars(f, cur_bb, try_body, &mut tmp_vars, loop_stack, env)?;
if !finally.is_empty() { if !finally.is_empty() {
next_bb = next_bb = lower_stmt_list_with_vars(
lower_stmt_list_with_vars(f, next_bb, finally, &mut tmp_vars, loop_stack)?; f,
next_bb,
finally,
&mut tmp_vars,
loop_stack,
env,
)?;
} }
*vars = tmp_vars; *vars = tmp_vars;
return Ok(next_bb); return Ok(next_bb);
@ -960,7 +824,7 @@ fn lower_stmt_with_vars(
let mut try_vars = vars.clone(); let mut try_vars = vars.clone();
let try_end = let try_end =
lower_stmt_list_with_vars(f, try_bb, try_body, &mut try_vars, loop_stack)?; lower_stmt_list_with_vars(f, try_bb, try_body, &mut try_vars, loop_stack, env)?;
if let Some(bb) = f.get_block_mut(try_end) { if let Some(bb) = f.get_block_mut(try_end) {
if !bb.is_terminated() { if !bb.is_terminated() {
bb.set_terminator(MirInstruction::Jump { bb.set_terminator(MirInstruction::Jump {
@ -980,6 +844,7 @@ fn lower_stmt_with_vars(
&catch_clause.body, &catch_clause.body,
&mut catch_vars, &mut catch_vars,
loop_stack, loop_stack,
env,
)?; )?;
if let Some(param) = &catch_clause.param { if let Some(param) = &catch_clause.param {
catch_vars.remove(param); catch_vars.remove(param);
@ -1045,6 +910,7 @@ fn lower_stmt_with_vars(
finally, finally,
&mut finally_vars, &mut finally_vars,
loop_stack, loop_stack,
env,
)?; )?;
if let Some(bb) = f.get_block_mut(final_end) { if let Some(bb) = f.get_block_mut(final_end) {
if !bb.is_terminated() { if !bb.is_terminated() {
@ -1098,7 +964,7 @@ fn lower_stmt_with_vars(
} }
StmtV0::If { cond, then, r#else } => { StmtV0::If { cond, then, r#else } => {
// Lower condition first // Lower condition first
let (cval, cur) = lower_expr_with_vars(f, cur_bb, cond, vars)?; let (cval, cur) = lower_expr_with_vars(env, f, cur_bb, cond, vars)?;
// Create then/else/merge blocks // Create then/else/merge blocks
let then_bb = next_block_id(f); let then_bb = next_block_id(f);
let else_bb = BasicBlockId::new(then_bb.0 + 1); let else_bb = BasicBlockId::new(then_bb.0 + 1);
@ -1118,7 +984,8 @@ fn lower_stmt_with_vars(
// Clone current vars as branch-local maps // Clone current vars as branch-local maps
let base_vars = vars.clone(); let base_vars = vars.clone();
let mut then_vars = base_vars.clone(); let mut then_vars = base_vars.clone();
let tend = lower_stmt_list_with_vars(f, then_bb, then, &mut then_vars, loop_stack)?; let tend =
lower_stmt_list_with_vars(f, then_bb, then, &mut then_vars, loop_stack, env)?;
if let Some(bb) = f.get_block_mut(tend) { if let Some(bb) = f.get_block_mut(tend) {
if !bb.is_terminated() { if !bb.is_terminated() {
bb.set_terminator(MirInstruction::Jump { target: merge_bb }); bb.set_terminator(MirInstruction::Jump { target: merge_bb });
@ -1127,7 +994,7 @@ fn lower_stmt_with_vars(
let (else_end_pred, else_vars) = if let Some(elses) = r#else { let (else_end_pred, else_vars) = if let Some(elses) = r#else {
let mut ev = base_vars.clone(); let mut ev = base_vars.clone();
let eend = lower_stmt_list_with_vars(f, else_bb, elses, &mut ev, loop_stack)?; let eend = lower_stmt_list_with_vars(f, else_bb, elses, &mut ev, loop_stack, env)?;
if let Some(bb) = f.get_block_mut(eend) { if let Some(bb) = f.get_block_mut(eend) {
if !bb.is_terminated() { if !bb.is_terminated() {
bb.set_terminator(MirInstruction::Jump { target: merge_bb }); bb.set_terminator(MirInstruction::Jump { target: merge_bb });
@ -1144,10 +1011,14 @@ fn lower_stmt_with_vars(
// Merge at then/else predecessorsPHI or edge-copy // Merge at then/else predecessorsPHI or edge-copy
use std::collections::HashSet; use std::collections::HashSet;
let no_phi = crate::config::env::mir_no_phi(); let no_phi = env.mir_no_phi;
let mut names: HashSet<String> = base_vars.keys().cloned().collect(); let mut names: HashSet<String> = base_vars.keys().cloned().collect();
for k in then_vars.keys() { names.insert(k.clone()); } for k in then_vars.keys() {
for k in else_vars.keys() { names.insert(k.clone()); } names.insert(k.clone());
}
for k in else_vars.keys() {
names.insert(k.clone());
}
for name in names { for name in names {
let tv = then_vars.get(&name).copied(); let tv = then_vars.get(&name).copied();
@ -1155,13 +1026,22 @@ fn lower_stmt_with_vars(
let exists_base = base_vars.contains_key(&name); let exists_base = base_vars.contains_key(&name);
match (tv, ev, exists_base) { match (tv, ev, exists_base) {
(Some(tval), Some(eval), _) => { (Some(tval), Some(eval), _) => {
let merged = if tval == eval { tval } else { let merged = if tval == eval {
tval
} else {
let dst = f.next_value_id(); let dst = f.next_value_id();
if no_phi { if no_phi {
if let Some(bb) = f.get_block_mut(tend) { bb.add_instruction(MirInstruction::Copy { dst, src: tval }); } if let Some(bb) = f.get_block_mut(tend) {
if let Some(bb) = f.get_block_mut(else_end_pred) { bb.add_instruction(MirInstruction::Copy { dst, src: eval }); } bb.add_instruction(MirInstruction::Copy { dst, src: tval });
}
if let Some(bb) = f.get_block_mut(else_end_pred) {
bb.add_instruction(MirInstruction::Copy { dst, src: eval });
}
} else if let Some(bb) = f.get_block_mut(merge_bb) { } else if let Some(bb) = f.get_block_mut(merge_bb) {
bb.insert_instruction_after_phis(MirInstruction::Phi { dst, inputs: vec![(tend, tval), (else_end_pred, eval)] }); bb.insert_instruction_after_phis(MirInstruction::Phi {
dst,
inputs: vec![(tend, tval), (else_end_pred, eval)],
});
} }
dst dst
}; };
@ -1169,13 +1049,22 @@ fn lower_stmt_with_vars(
} }
(Some(tval), None, true) => { (Some(tval), None, true) => {
if let Some(&bval) = base_vars.get(&name) { if let Some(&bval) = base_vars.get(&name) {
let merged = if tval == bval { tval } else { let merged = if tval == bval {
tval
} else {
let dst = f.next_value_id(); let dst = f.next_value_id();
if no_phi { if no_phi {
if let Some(bb) = f.get_block_mut(tend) { bb.add_instruction(MirInstruction::Copy { dst, src: tval }); } if let Some(bb) = f.get_block_mut(tend) {
if let Some(bb) = f.get_block_mut(else_end_pred) { bb.add_instruction(MirInstruction::Copy { dst, src: bval }); } bb.add_instruction(MirInstruction::Copy { dst, src: tval });
}
if let Some(bb) = f.get_block_mut(else_end_pred) {
bb.add_instruction(MirInstruction::Copy { dst, src: bval });
}
} else if let Some(bb) = f.get_block_mut(merge_bb) { } else if let Some(bb) = f.get_block_mut(merge_bb) {
bb.insert_instruction_after_phis(MirInstruction::Phi { dst, inputs: vec![(tend, tval), (else_end_pred, bval)] }); bb.insert_instruction_after_phis(MirInstruction::Phi {
dst,
inputs: vec![(tend, tval), (else_end_pred, bval)],
});
} }
dst dst
}; };
@ -1184,13 +1073,22 @@ fn lower_stmt_with_vars(
} }
(None, Some(eval), true) => { (None, Some(eval), true) => {
if let Some(&bval) = base_vars.get(&name) { if let Some(&bval) = base_vars.get(&name) {
let merged = if eval == bval { eval } else { let merged = if eval == bval {
eval
} else {
let dst = f.next_value_id(); let dst = f.next_value_id();
if no_phi { if no_phi {
if let Some(bb) = f.get_block_mut(tend) { bb.add_instruction(MirInstruction::Copy { dst, src: bval }); } if let Some(bb) = f.get_block_mut(tend) {
if let Some(bb) = f.get_block_mut(else_end_pred) { bb.add_instruction(MirInstruction::Copy { dst, src: eval }); } bb.add_instruction(MirInstruction::Copy { dst, src: bval });
}
if let Some(bb) = f.get_block_mut(else_end_pred) {
bb.add_instruction(MirInstruction::Copy { dst, src: eval });
}
} else if let Some(bb) = f.get_block_mut(merge_bb) { } else if let Some(bb) = f.get_block_mut(merge_bb) {
bb.insert_instruction_after_phis(MirInstruction::Phi { dst, inputs: vec![(tend, bval), (else_end_pred, eval)] }); bb.insert_instruction_after_phis(MirInstruction::Phi {
dst,
inputs: vec![(tend, bval), (else_end_pred, eval)],
});
} }
dst dst
}; };
@ -1220,11 +1118,10 @@ fn lower_stmt_with_vars(
} }
// Snapshot base vars and set up merged ids for loop-carried vars // Snapshot base vars and set up merged ids for loop-carried vars
let no_phi = crate::config::env::mir_no_phi(); let no_phi = env.mir_no_phi;
let base_vars = vars.clone(); let base_vars = vars.clone();
let orig_names: Vec<String> = base_vars.keys().cloned().collect(); let orig_names: Vec<String> = base_vars.keys().cloned().collect();
let mut phi_map: std::collections::HashMap<String, crate::mir::ValueId> = let mut phi_map: HashMap<String, ValueId> = HashMap::new();
std::collections::HashMap::new();
for name in &orig_names { for name in &orig_names {
if let Some(&bval) = base_vars.get(name) { if let Some(&bval) = base_vars.get(name) {
let dst = f.next_value_id(); let dst = f.next_value_id();
@ -1235,7 +1132,10 @@ fn lower_stmt_with_vars(
} }
} else if let Some(bb) = f.get_block_mut(cond_bb) { } else if let Some(bb) = f.get_block_mut(cond_bb) {
// Initial incoming from preheader via PHI // Initial incoming from preheader via PHI
bb.insert_instruction_after_phis(MirInstruction::Phi { dst, inputs: vec![(cur_bb, bval)] }); bb.insert_instruction_after_phis(MirInstruction::Phi {
dst,
inputs: vec![(cur_bb, bval)],
});
} }
phi_map.insert(name.clone(), dst); phi_map.insert(name.clone(), dst);
} }
@ -1246,7 +1146,7 @@ fn lower_stmt_with_vars(
} }
// Lower condition using phi-backed vars // Lower condition using phi-backed vars
let (cval, _cend) = lower_expr_with_vars(f, cond_bb, cond, vars)?; let (cval, _cend) = lower_expr_with_vars(env, f, cond_bb, cond, vars)?;
if let Some(bb) = f.get_block_mut(cond_bb) { if let Some(bb) = f.get_block_mut(cond_bb) {
bb.set_terminator(MirInstruction::Branch { bb.set_terminator(MirInstruction::Branch {
condition: cval, condition: cval,
@ -1258,7 +1158,8 @@ fn lower_stmt_with_vars(
// Lower body; record end block and body-out vars // Lower body; record end block and body-out vars
let mut body_vars = vars.clone(); let mut body_vars = vars.clone();
loop_stack.push(LoopContext { cond_bb, exit_bb }); loop_stack.push(LoopContext { cond_bb, exit_bb });
let bend_res = lower_stmt_list_with_vars(f, body_bb, body, &mut body_vars, loop_stack); let bend_res =
lower_stmt_list_with_vars(f, body_bb, body, &mut body_vars, loop_stack, env);
loop_stack.pop(); loop_stack.pop();
let bend = bend_res?; let bend = bend_res?;
if let Some(bb) = f.get_block_mut(bend) { if let Some(bb) = f.get_block_mut(bend) {
@ -1278,7 +1179,10 @@ fn lower_stmt_with_vars(
for (name, &phi_dst) in &phi_map { for (name, &phi_dst) in &phi_map {
if let Some(&latch_val) = body_vars.get(name) { if let Some(&latch_val) = body_vars.get(name) {
if let Some(bb) = f.get_block_mut(bend) { if let Some(bb) = f.get_block_mut(bend) {
bb.add_instruction(MirInstruction::Copy { dst: phi_dst, src: latch_val }); bb.add_instruction(MirInstruction::Copy {
dst: phi_dst,
src: latch_val,
});
} }
} }
} }
@ -1317,12 +1221,13 @@ fn lower_stmt_list_with_vars(
f: &mut MirFunction, f: &mut MirFunction,
start_bb: BasicBlockId, start_bb: BasicBlockId,
stmts: &[StmtV0], stmts: &[StmtV0],
vars: &mut std::collections::HashMap<String, crate::mir::ValueId>, vars: &mut HashMap<String, ValueId>,
loop_stack: &mut Vec<LoopContext>, loop_stack: &mut Vec<LoopContext>,
env: &BridgeEnv,
) -> Result<BasicBlockId, String> { ) -> Result<BasicBlockId, String> {
let mut cur = start_bb; let mut cur = start_bb;
for s in stmts { for s in stmts {
cur = lower_stmt_with_vars(f, cur, s, vars, loop_stack)?; cur = lower_stmt_with_vars(f, cur, s, vars, loop_stack, env)?;
if let Some(bb) = f.blocks.get(&cur) { if let Some(bb) = f.blocks.get(&cur) {
if bb.is_terminated() { if bb.is_terminated() {
break; break;
@ -1331,16 +1236,17 @@ fn lower_stmt_list_with_vars(
} }
Ok(cur) Ok(cur)
} }
fn lower_args_with_vars( fn lower_args_with_scope<S: VarScope>(
env: &BridgeEnv,
f: &mut MirFunction, f: &mut MirFunction,
cur_bb: BasicBlockId, cur_bb: BasicBlockId,
args: &[ExprV0], args: &[ExprV0],
vars: &mut std::collections::HashMap<String, crate::mir::ValueId>, scope: &mut S,
) -> Result<(Vec<crate::mir::ValueId>, BasicBlockId), String> { ) -> Result<(Vec<ValueId>, BasicBlockId), String> {
let mut out = Vec::with_capacity(args.len()); let mut out = Vec::with_capacity(args.len());
let mut cur = cur_bb; let mut cur = cur_bb;
for a in args { for a in args {
let (v, c) = lower_expr_with_vars(f, cur, a, vars)?; let (v, c) = lower_expr_with_scope(env, f, cur, a, scope)?;
out.push(v); out.push(v);
cur = c; cur = c;
} }
@ -1348,18 +1254,24 @@ fn lower_args_with_vars(
} }
fn lower_args( fn lower_args(
env: &BridgeEnv,
f: &mut MirFunction, f: &mut MirFunction,
cur_bb: BasicBlockId, cur_bb: BasicBlockId,
args: &[ExprV0], args: &[ExprV0],
) -> Result<(Vec<crate::mir::ValueId>, BasicBlockId), String> { ) -> Result<(Vec<ValueId>, BasicBlockId), String> {
let mut out = Vec::with_capacity(args.len()); let mut scope = NoVars;
let mut cur = cur_bb; lower_args_with_scope(env, f, cur_bb, args, &mut scope)
for a in args { }
let (v, c) = lower_expr(f, cur, a)?;
out.push(v); fn lower_args_with_vars(
cur = c; env: &BridgeEnv,
} f: &mut MirFunction,
Ok((out, cur)) cur_bb: BasicBlockId,
args: &[ExprV0],
vars: &mut HashMap<String, ValueId>,
) -> Result<(Vec<ValueId>, BasicBlockId), String> {
let mut scope = MapVars::new(vars);
lower_args_with_scope(env, f, cur_bb, args, &mut scope)
} }
pub fn maybe_dump_mir(module: &MirModule) { pub fn maybe_dump_mir(module: &MirModule) {

View File

@ -44,4 +44,7 @@ run "$ROOT_DIR/apps/tests/loop_if_phi.nyash"
# Peek expression # Peek expression
run "$ROOT_DIR/apps/tests/peek_expr_block.nyash" run "$ROOT_DIR/apps/tests/peek_expr_block.nyash"
# Try/finally control-flow without actual throw
run "$ROOT_DIR/apps/tests/try_finally_break_inner_loop.nyash"
echo "[curated-llvm] OK" echo "[curated-llvm] OK"