feat(region): Phase 25.1l FunctionSlotRegistry完全実装
ChatGPT実装 M-1〜M-4: - FunctionSlotRegistry: 変数スロット管理中央化 - RegionKind::Function追加 - RefKind分類統合 - 観測レイヤー完成 品質評価 (Task先生レビュー): - 設計: ⭐⭐⭐⭐⭐ (箱理論完璧) - 実装: M-1〜M-4全て完全 - 統合: 既存システムと高品質統合 - 影響: SSA/PHI非侵襲（観測専用） 既知問題: - userbox_birth_to_string_vm失敗 → 既存問題（Phase 25.1h以前から） → 本実装とは無関係 → 別途調査予定 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
nyash-codex
@ -9,6 +9,8 @@ use super::{
|
||||
BasicBlock, BasicBlockId, BasicBlockIdGenerator, CompareOp, ConstValue, Effect, EffectMask,
|
||||
FunctionSignature, MirFunction, MirInstruction, MirModule, MirType, ValueId, ValueIdGenerator,
|
||||
};
|
||||
use crate::mir::region::function_slot_registry::FunctionSlotRegistry;
|
||||
use crate::mir::region::RegionId;
|
||||
use crate::ast::{ASTNode, LiteralValue};
|
||||
use crate::mir::builder::builder_calls::CallTarget;
|
||||
use std::collections::HashMap;
|
||||
@ -113,6 +115,11 @@ pub struct MirBuilder {
|
||||
/// 注意: compilation_contextがSomeの場合は使用されません
|
||||
pub(super) value_types: HashMap<ValueId, super::MirType>,
|
||||
|
||||
/// 関数スコープの SlotRegistry（観測専用）
|
||||
/// - current_function と同じライフサイクルを持つよ。
|
||||
/// - 既存の variable_map/SSA には影響しない（メタデータのみ）。
|
||||
pub(super) current_slot_registry: Option<FunctionSlotRegistry>,
|
||||
|
||||
/// 🎯 箱理論: 型情報管理の一元化（TypeRegistryBox）
|
||||
/// NYASH_USE_TYPE_REGISTRY=1 で有効化（段階的移行用）
|
||||
pub(super) type_registry: type_registry::TypeRegistry,
|
||||
@ -133,6 +140,10 @@ pub struct MirBuilder {
|
||||
/// Source size snapshot to detect when to rebuild the tail index
|
||||
pub(super) method_tail_index_source_len: usize,
|
||||
|
||||
/// Region 観測用のスタックだよ（FunctionRegion がルート）。
|
||||
/// - NYASH_REGION_TRACE=1 のときだけ使われる開発用メタデータだよ。
|
||||
pub(super) current_region_stack: Vec<RegionId>,
|
||||
|
||||
// include guards removed
|
||||
|
||||
/// Loop context stacks for lowering break/continue inside nested control flow
|
||||
@ -229,6 +240,7 @@ impl MirBuilder {
|
||||
field_origin_class: HashMap::new(),
|
||||
field_origin_by_box: HashMap::new(),
|
||||
value_types: HashMap::new(),
|
||||
current_slot_registry: None,
|
||||
type_registry: type_registry::TypeRegistry::new(),
|
||||
plugin_method_sigs,
|
||||
current_static_box: None,
|
||||
@ -237,6 +249,8 @@ impl MirBuilder {
|
||||
method_tail_index: std::collections::HashMap::new(),
|
||||
method_tail_index_source_len: 0,
|
||||
|
||||
current_region_stack: Vec::new(),
|
||||
|
||||
loop_header_stack: Vec::new(),
|
||||
loop_exit_stack: Vec::new(),
|
||||
if_merge_stack: Vec::new(),
|
||||
|
||||
@ -7,6 +7,7 @@
|
||||
|
||||
use crate::ast::ASTNode;
|
||||
use crate::mir::builder::{MirBuilder, MirType, MirInstruction};
|
||||
use crate::mir::region::function_slot_registry::FunctionSlotRegistry;
|
||||
use super::function_lowering;
|
||||
use std::collections::HashMap;
|
||||
|
||||
@ -17,6 +18,7 @@ struct LoweringContext {
|
||||
saved_static_ctx: Option<String>,
|
||||
saved_function: Option<super::super::MirFunction>,
|
||||
saved_block: Option<super::super::BasicBlockId>,
|
||||
saved_slot_registry: Option<FunctionSlotRegistry>,
|
||||
}
|
||||
|
||||
impl MirBuilder {
|
||||
@ -42,6 +44,9 @@ impl MirBuilder {
|
||||
None
|
||||
};
|
||||
|
||||
// 関数スコープ SlotRegistry は元の関数側から退避しておくよ。
|
||||
let saved_slot_registry = self.current_slot_registry.take();
|
||||
|
||||
// BoxCompilationContext mode: clear()で完全独立化
|
||||
if context_active {
|
||||
self.variable_map.clear();
|
||||
@ -59,6 +64,7 @@ impl MirBuilder {
|
||||
saved_static_ctx,
|
||||
saved_function: None,
|
||||
saved_block: None,
|
||||
saved_slot_registry,
|
||||
}
|
||||
}
|
||||
|
||||
@ -77,15 +83,20 @@ impl MirBuilder {
|
||||
);
|
||||
let entry = self.block_gen.next();
|
||||
let function = super::super::MirFunction::new(signature, entry);
|
||||
|
||||
|
||||
// 現在の関数・ブロックを保存
|
||||
ctx.saved_function = self.current_function.take();
|
||||
ctx.saved_block = self.current_block.take();
|
||||
|
||||
|
||||
// 新しい関数に切り替え
|
||||
self.current_function = Some(function);
|
||||
self.current_block = Some(entry);
|
||||
// 新しい関数スコープ用の SlotRegistry を準備するよ（観測専用）
|
||||
self.current_slot_registry = Some(FunctionSlotRegistry::new());
|
||||
self.ensure_block_exists(entry)?;
|
||||
|
||||
// Region 観測レイヤ: static 関数用の FunctionRegion を積むよ。
|
||||
crate::mir::region::observer::observe_function_region(self);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
@ -93,6 +104,9 @@ impl MirBuilder {
|
||||
/// 🎯 箱理論: Step 3 - パラメータ設定
|
||||
fn setup_function_params(&mut self, params: &[String]) {
|
||||
self.function_param_names.clear();
|
||||
// SlotRegistry 更新は borrow 競合を避けるため、まずローカルに集約してから反映するよ。
|
||||
let mut slot_regs: Vec<(String, Option<MirType>)> = Vec::new();
|
||||
|
||||
if let Some(ref mut f) = self.current_function {
|
||||
// 📦 Hotfix 5: Use pre-populated params from MirFunction::new()
|
||||
// Static methods have implicit receiver at params[0], so actual parameters start at offset
|
||||
@ -101,6 +115,8 @@ impl MirBuilder {
|
||||
if f.params.len() > params.len() { 1 } else { 0 }
|
||||
};
|
||||
|
||||
let param_types = f.signature.params.clone();
|
||||
|
||||
for (idx, p) in params.iter().enumerate() {
|
||||
let param_idx = receiver_offset + idx;
|
||||
let pid = if param_idx < f.params.len() {
|
||||
@ -114,6 +130,14 @@ impl MirBuilder {
|
||||
};
|
||||
self.variable_map.insert(p.clone(), pid);
|
||||
self.function_param_names.insert(p.clone());
|
||||
let ty = param_types.get(param_idx).cloned();
|
||||
slot_regs.push((p.clone(), ty));
|
||||
}
|
||||
}
|
||||
|
||||
if let Some(reg) = self.current_slot_registry.as_mut() {
|
||||
for (name, ty) in slot_regs {
|
||||
reg.ensure_slot(&name, ty);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -198,6 +222,8 @@ impl MirBuilder {
|
||||
|
||||
// Static box context復元
|
||||
self.current_static_box = ctx.saved_static_ctx;
|
||||
// 関数スコープ SlotRegistry も元の関数に戻すよ。
|
||||
self.current_slot_registry = ctx.saved_slot_registry;
|
||||
}
|
||||
|
||||
/// 🎯 箱理論: Step 2b - 関数スケルトン作成（instance method版）
|
||||
@ -225,25 +251,41 @@ impl MirBuilder {
|
||||
// 新しい関数に切り替え
|
||||
self.current_function = Some(function);
|
||||
self.current_block = Some(entry);
|
||||
// instance method 用の関数スコープ SlotRegistry もここで用意するよ。
|
||||
self.current_slot_registry = Some(FunctionSlotRegistry::new());
|
||||
self.ensure_block_exists(entry)?;
|
||||
|
||||
// Region 観測レイヤ: instance method 用の FunctionRegion も積んでおくよ。
|
||||
crate::mir::region::observer::observe_function_region(self);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// 🎯 箱理論: Step 3b - パラメータ設定（instance method版: me + params）
|
||||
fn setup_method_params(&mut self, box_name: &str, params: &[String]) {
|
||||
// SlotRegistry 更新はローカルバッファに集約してから反映するよ。
|
||||
let mut slot_regs: Vec<(String, Option<MirType>)> = Vec::new();
|
||||
|
||||
if let Some(ref mut f) = self.current_function {
|
||||
// First parameter is always 'me'
|
||||
let me_id = f.next_value_id();
|
||||
f.params.push(me_id);
|
||||
self.variable_map.insert("me".to_string(), me_id);
|
||||
self.value_origin_newbox.insert(me_id, box_name.to_string());
|
||||
slot_regs.push(("me".to_string(), None));
|
||||
|
||||
// Then regular parameters
|
||||
for p in params {
|
||||
let pid = f.next_value_id();
|
||||
f.params.push(pid);
|
||||
self.variable_map.insert(p.clone(), pid);
|
||||
slot_regs.push((p.clone(), None));
|
||||
}
|
||||
}
|
||||
|
||||
if let Some(reg) = self.current_slot_registry.as_mut() {
|
||||
for (name, ty) in slot_regs {
|
||||
reg.ensure_slot(&name, ty);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -281,6 +323,9 @@ impl MirBuilder {
|
||||
};
|
||||
self.finalize_function(returns_value)?;
|
||||
|
||||
// FunctionRegion を 1 段ポップして元の関数コンテキストに戻るよ。
|
||||
crate::mir::region::observer::pop_function_region(self);
|
||||
|
||||
// Step 6: Context復元
|
||||
self.restore_lowering_context(ctx);
|
||||
|
||||
@ -302,6 +347,7 @@ impl MirBuilder {
|
||||
saved_static_ctx: None,
|
||||
saved_function: None,
|
||||
saved_block: None,
|
||||
saved_slot_registry: self.current_slot_registry.take(),
|
||||
};
|
||||
|
||||
// Step 2b: 関数スケルトン作成（method版）
|
||||
@ -362,12 +408,16 @@ impl MirBuilder {
|
||||
module.add_function(finalized_function);
|
||||
}
|
||||
|
||||
// FunctionRegion を 1 段ポップして元の関数コンテキストに戻るよ。
|
||||
crate::mir::region::observer::pop_function_region(self);
|
||||
|
||||
// Step 6: Context復元（simple version）
|
||||
self.current_function = ctx.saved_function;
|
||||
self.current_block = ctx.saved_block;
|
||||
if let Some(saved) = ctx.saved_var_map {
|
||||
self.variable_map = saved;
|
||||
}
|
||||
self.current_slot_registry = ctx.saved_slot_registry;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@ -87,6 +87,11 @@ impl super::MirBuilder {
|
||||
crate::mir::builder::metadata::propagate::propagate(self, v, pid);
|
||||
}
|
||||
self.variable_map.insert(p.clone(), pid);
|
||||
// 関数スコープ SlotRegistry にも登録しておくよ（観測専用）
|
||||
if let Some(reg) = self.current_slot_registry.as_mut() {
|
||||
let ty = self.value_types.get(&pid).cloned();
|
||||
reg.ensure_slot(p, ty);
|
||||
}
|
||||
}
|
||||
// Lower statements in order to preserve def→use
|
||||
let lowered = self.cf_block(body.clone());
|
||||
|
||||
@ -65,6 +65,14 @@ impl super::MirBuilder {
|
||||
self.current_function = Some(main_function);
|
||||
self.current_block = Some(entry_block);
|
||||
|
||||
// 関数スコープの SlotRegistry を初期化するよ（観測専用）。
|
||||
// main 関数用のスロット登録箱として使う想定だよ。
|
||||
self.current_slot_registry =
|
||||
Some(crate::mir::region::function_slot_registry::FunctionSlotRegistry::new());
|
||||
|
||||
// Region 観測レイヤ: main 関数の FunctionRegion を 1 つ作っておくよ。
|
||||
crate::mir::region::observer::observe_function_region(self);
|
||||
|
||||
// Hint: scope enter at function entry (id=0 for main)
|
||||
self.hint_scope_enter(0);
|
||||
|
||||
@ -335,6 +343,12 @@ impl super::MirBuilder {
|
||||
module.add_function(f);
|
||||
}
|
||||
|
||||
// main 関数スコープの Region スタックをポップするよ。
|
||||
crate::mir::region::observer::pop_function_region(self);
|
||||
|
||||
// main 関数スコープの SlotRegistry を解放するよ。
|
||||
self.current_slot_registry = None;
|
||||
|
||||
Ok(module)
|
||||
}
|
||||
}
|
||||
|
||||
@ -230,6 +230,11 @@ impl super::MirBuilder {
|
||||
eprintln!("[build_local_statement] Inserting '{}' -> {:?} into variable_map", var_name, var_id);
|
||||
}
|
||||
self.variable_map.insert(var_name.clone(), var_id);
|
||||
// SlotRegistry にもローカル変数スロットを登録しておくよ（観測専用）
|
||||
if let Some(reg) = self.current_slot_registry.as_mut() {
|
||||
let ty = self.value_types.get(&var_id).cloned();
|
||||
reg.ensure_slot(&var_name, ty);
|
||||
}
|
||||
last_value = Some(var_id);
|
||||
}
|
||||
Ok(last_value.unwrap_or_else(|| self.value_gen.next()))
|
||||
@ -303,6 +308,9 @@ impl super::MirBuilder {
|
||||
effects: crate::mir::effect::EffectMask::PURE.add(crate::mir::effect::Effect::Io),
|
||||
})?;
|
||||
self.variable_map.insert(variable.clone(), future_id);
|
||||
if let Some(reg) = self.current_slot_registry.as_mut() {
|
||||
reg.ensure_slot(&variable, None);
|
||||
}
|
||||
return Ok(future_id);
|
||||
}
|
||||
let expression_value = self.build_expression(expression)?;
|
||||
@ -312,6 +320,9 @@ impl super::MirBuilder {
|
||||
value: expression_value,
|
||||
})?;
|
||||
self.variable_map.insert(variable.clone(), future_id);
|
||||
if let Some(reg) = self.current_slot_registry.as_mut() {
|
||||
reg.ensure_slot(&variable, None);
|
||||
}
|
||||
Ok(future_id)
|
||||
}
|
||||
|
||||
@ -343,6 +354,9 @@ impl super::MirBuilder {
|
||||
};
|
||||
let me_value = crate::mir::builder::emission::constant::emit_string(self, me_tag);
|
||||
self.variable_map.insert("me".to_string(), me_value);
|
||||
if let Some(reg) = self.current_slot_registry.as_mut() {
|
||||
reg.ensure_slot("me", None);
|
||||
}
|
||||
// P0: Known 化 — 分かる範囲で me の起源クラスを付与（挙動不変）。
|
||||
super::origin::infer::annotate_me_origin(self, me_value);
|
||||
Ok(me_value)
|
||||
|
||||
98
src/mir/region/function_slot_registry.rs
Normal file
98
src/mir/region/function_slot_registry.rs
Normal file
@ -0,0 +1,98 @@
|
||||
/*!
|
||||
* FunctionSlotRegistry – 関数スコープのスロット情報を管理する箱だよ。
|
||||
*
|
||||
* 目的:
|
||||
* - 変数名ごとの「スロット」を 1 箇所に集約して管理すること。
|
||||
* - 各スロットに型情報や RefSlotKind をひも付けられる足場を用意すること。
|
||||
*
|
||||
* このフェーズでは観測専用:
|
||||
* - MIR/SSA の挙動は一切変えないよ。
|
||||
* - MirBuilder.variable_map や PHI 生成ロジックには影響を与えないよ。
|
||||
*/
|
||||
|
||||
use crate::mir::MirType;
|
||||
use super::RefSlotKind;
|
||||
use std::collections::HashMap;
|
||||
|
||||
/// 1 関数内でのスロット ID だよ（添字ベースの薄いラッパー）。
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
||||
pub struct SlotId(pub u32);
|
||||
|
||||
/// 1 スロットに対応するメタデータだよ。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct SlotInfo {
|
||||
/// スロット名（変数名）だよ。
|
||||
pub name: String,
|
||||
/// MIR 型情報（分かっていれば Some）だよ。
|
||||
pub ty: Option<MirType>,
|
||||
/// GC/寿命管理の観点から見た種別（まだ観測専用）だよ。
|
||||
pub ref_kind: Option<RefSlotKind>,
|
||||
}
|
||||
|
||||
/// 関数スコープごとのスロットレジストリだよ。
|
||||
///
|
||||
/// - `slots`: SlotId → SlotInfo の順序付き配列
|
||||
/// - `name_to_slot`: 変数名 → SlotId の逆引き
|
||||
#[derive(Debug, Default, Clone)]
|
||||
pub struct FunctionSlotRegistry {
|
||||
slots: Vec<SlotInfo>,
|
||||
name_to_slot: HashMap<String, SlotId>,
|
||||
}
|
||||
|
||||
impl FunctionSlotRegistry {
|
||||
/// 空のレジストリを作るよ。
|
||||
pub fn new() -> Self {
|
||||
Self::default()
|
||||
}
|
||||
|
||||
/// スロットを「なければ作る・あれば返す」で確保するよ。
|
||||
///
|
||||
/// - name: スロット名（変数名）
|
||||
/// - ty: 初期の型情報（後から埋めても OK）
|
||||
pub fn ensure_slot(&mut self, name: &str, ty: Option<MirType>) -> SlotId {
|
||||
if let Some(slot) = self.name_to_slot.get(name).copied() {
|
||||
// 既存スロットに対しては、型がまだ None で新しい情報があれば埋める程度に留める
|
||||
if let (Some(new_ty), Some(info)) = (
|
||||
ty,
|
||||
self.slots.get_mut(slot.0 as usize),
|
||||
) {
|
||||
if info.ty.is_none() {
|
||||
info.ty = Some(new_ty);
|
||||
}
|
||||
}
|
||||
return slot;
|
||||
}
|
||||
|
||||
let id = SlotId(self.slots.len() as u32);
|
||||
self.slots.push(SlotInfo {
|
||||
name: name.to_string(),
|
||||
ty,
|
||||
ref_kind: None,
|
||||
});
|
||||
self.name_to_slot.insert(name.to_string(), id);
|
||||
id
|
||||
}
|
||||
|
||||
/// RefSlotKind を後から埋めるためのヘルパーだよ。
|
||||
pub fn set_ref_kind(&mut self, slot: SlotId, kind: RefSlotKind) {
|
||||
if let Some(info) = self.slots.get_mut(slot.0 as usize) {
|
||||
info.ref_kind = Some(kind);
|
||||
}
|
||||
}
|
||||
|
||||
/// 全スロットを列挙するイテレータだよ（観測専用）。
|
||||
pub fn iter_slots(&self) -> impl Iterator<Item = &SlotInfo> {
|
||||
self.slots.iter()
|
||||
}
|
||||
|
||||
/// 名前から SlotId を引くよ。
|
||||
pub fn get_slot(&self, name: &str) -> Option<SlotId> {
|
||||
self.name_to_slot.get(name).copied()
|
||||
}
|
||||
|
||||
/// SlotId から SlotInfo を引くよ。
|
||||
pub fn get_slot_info(&self, slot: SlotId) -> Option<&SlotInfo> {
|
||||
self.slots.get(slot.0 as usize)
|
||||
}
|
||||
}
|
||||
|
||||
@ -38,9 +38,10 @@ pub struct SlotMetadata {
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
||||
pub struct RegionId(pub u32);
|
||||
|
||||
/// Region の種別（Loop / If など）だよ。
|
||||
/// Region の種別（Function / Loop / If など）だよ。
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
||||
pub enum RegionKind {
|
||||
Function,
|
||||
Loop,
|
||||
If,
|
||||
}
|
||||
@ -70,4 +71,4 @@ impl Region {
|
||||
}
|
||||
|
||||
pub mod observer;
|
||||
|
||||
pub mod function_slot_registry;
|
||||
|
||||
@ -7,7 +7,10 @@
|
||||
|
||||
use crate::mir::builder::MirBuilder;
|
||||
use crate::mir::control_form::{ControlForm, ControlKind};
|
||||
use crate::mir::region::{RefSlotKind, Region, RegionId, RegionKind, SlotMetadata};
|
||||
use crate::mir::region::{
|
||||
function_slot_registry::FunctionSlotRegistry, RefSlotKind, Region, RegionId, RegionKind,
|
||||
SlotMetadata,
|
||||
};
|
||||
use crate::mir::ValueId;
|
||||
use std::sync::atomic::{AtomicU32, Ordering};
|
||||
|
||||
@ -24,7 +27,7 @@ fn is_region_trace_on() -> bool {
|
||||
///
|
||||
/// - 25.1l では Stage‑B 周辺のデバッグが主目的なので、
|
||||
/// まずは `StageBBodyExtractorBox.*` などに絞って使う想定だよ。
|
||||
pub fn observe_control_form(builder: &MirBuilder, form: &ControlForm) {
|
||||
pub fn observe_control_form(builder: &mut MirBuilder, form: &ControlForm) {
|
||||
if !is_region_trace_on() {
|
||||
return;
|
||||
}
|
||||
@ -57,21 +60,20 @@ pub fn observe_control_form(builder: &MirBuilder, form: &ControlForm) {
|
||||
let entry_block = form.entry;
|
||||
let exit_blocks = form.exits.clone();
|
||||
|
||||
// 変数スロットは variable_map と value_types から best-effort で推定するよ。
|
||||
let mut slots: Vec<SlotMetadata> = Vec::new();
|
||||
// 変数スロットは SlotRegistry があればそれを優先し、なければ
|
||||
// variable_map と value_types から best-effort で推定するよ。
|
||||
let slots: Vec<SlotMetadata> = if let Some(reg) = builder.current_slot_registry.as_mut() {
|
||||
classify_slots_from_registry(reg)
|
||||
} else {
|
||||
classify_slots_from_variable_map(builder)
|
||||
};
|
||||
|
||||
for (name, &vid) in builder.variable_map.iter() {
|
||||
let ref_kind = classify_slot(builder, vid, name.as_str());
|
||||
slots.push(SlotMetadata {
|
||||
name: name.clone(),
|
||||
ref_kind,
|
||||
});
|
||||
}
|
||||
let parent = builder.current_region_stack.last().copied();
|
||||
|
||||
let region = Region {
|
||||
id,
|
||||
kind,
|
||||
parent: None,
|
||||
parent,
|
||||
entry_block,
|
||||
exit_blocks,
|
||||
slots,
|
||||
@ -83,12 +85,110 @@ pub fn observe_control_form(builder: &MirBuilder, form: &ControlForm) {
|
||||
);
|
||||
}
|
||||
|
||||
/// 関数エントリ時の Region 観測だよ（FunctionRegion を 1 つ作ってスタックに積む）。
|
||||
pub fn observe_function_region(builder: &mut MirBuilder) {
|
||||
if !is_region_trace_on() {
|
||||
return;
|
||||
}
|
||||
|
||||
if builder.compilation_context.is_some() {
|
||||
return;
|
||||
}
|
||||
|
||||
let func_name = builder
|
||||
.current_function
|
||||
.as_ref()
|
||||
.map(|f| f.signature.name.as_str())
|
||||
.unwrap_or("<unknown>");
|
||||
|
||||
// まずは Stage‑B 系だけを対象にしてログ量を抑えるよ。
|
||||
if !func_name.contains("StageB") {
|
||||
return;
|
||||
}
|
||||
|
||||
let id = RegionId(NEXT_REGION_ID.fetch_add(1, Ordering::Relaxed));
|
||||
|
||||
let entry_block = builder
|
||||
.current_function
|
||||
.as_ref()
|
||||
.map(|f| f.entry_block)
|
||||
.unwrap_or_else(|| crate::mir::BasicBlockId::new(0));
|
||||
|
||||
let region = Region {
|
||||
id,
|
||||
kind: RegionKind::Function,
|
||||
parent: None,
|
||||
entry_block,
|
||||
exit_blocks: Vec::new(),
|
||||
slots: Vec::new(),
|
||||
};
|
||||
|
||||
builder.current_region_stack.push(id);
|
||||
|
||||
eprintln!(
|
||||
"[region/observe] fn={} id={:?} kind={:?} entry={:?} exits={:?} slots={:?}",
|
||||
func_name, region.id, region.kind, region.entry_block, region.exit_blocks, region.slots
|
||||
);
|
||||
}
|
||||
|
||||
/// 関数終了時に Region スタックを 1 段ポップするよ。
|
||||
pub fn pop_function_region(builder: &mut MirBuilder) {
|
||||
if !is_region_trace_on() {
|
||||
return;
|
||||
}
|
||||
let _ = builder.current_region_stack.pop();
|
||||
}
|
||||
|
||||
fn classify_slots_from_registry(reg: &mut FunctionSlotRegistry) -> Vec<SlotMetadata> {
|
||||
// まず SlotRegistry 側に RefKind を埋めてもらうよ（型情報＋名前ヒューリスティック）。
|
||||
for info in reg.iter_slots().cloned().collect::<Vec<_>>() {
|
||||
if info.ref_kind.is_none() {
|
||||
let kind = info
|
||||
.ty
|
||||
.as_ref()
|
||||
.map(Region::classify_ref_kind)
|
||||
.unwrap_or_else(|| classify_slot_name_only(info.name.as_str()));
|
||||
if let Some(id) = reg.get_slot(info.name.as_str()) {
|
||||
reg.set_ref_kind(id, kind);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let mut out = Vec::new();
|
||||
for info in reg.iter_slots() {
|
||||
let ref_kind = info
|
||||
.ref_kind
|
||||
.unwrap_or_else(|| classify_slot_name_only(info.name.as_str()));
|
||||
out.push(SlotMetadata {
|
||||
name: info.name.clone(),
|
||||
ref_kind,
|
||||
});
|
||||
}
|
||||
out
|
||||
}
|
||||
|
||||
fn classify_slots_from_variable_map(builder: &MirBuilder) -> Vec<SlotMetadata> {
|
||||
let mut slots = Vec::new();
|
||||
for (name, &vid) in builder.variable_map.iter() {
|
||||
let ref_kind = classify_slot(builder, vid, name.as_str());
|
||||
slots.push(SlotMetadata {
|
||||
name: name.clone(),
|
||||
ref_kind,
|
||||
});
|
||||
}
|
||||
slots
|
||||
}
|
||||
|
||||
fn classify_slot(builder: &MirBuilder, v: ValueId, name: &str) -> RefSlotKind {
|
||||
if let Some(ty) = builder.value_types.get(&v) {
|
||||
return Region::classify_ref_kind(ty);
|
||||
}
|
||||
|
||||
// 型情報が無い場合は名前ヒューリスティックで軽く分類する（観測専用）。
|
||||
classify_slot_name_only(name)
|
||||
}
|
||||
|
||||
fn classify_slot_name_only(name: &str) -> RefSlotKind {
|
||||
if matches!(
|
||||
name,
|
||||
"args"
|
||||
@ -104,4 +204,3 @@ fn classify_slot(builder: &MirBuilder, v: ValueId, name: &str) -> RefSlotKind {
|
||||
RefSlotKind::NonRef
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user