RegionCtxt merged body_id with body_owner

This commit is contained in:
lcnr 2022-06-27 17:38:01 +02:00
parent 24799e3720
commit edd45f9d41
2 changed files with 22 additions and 30 deletions

View File

@ -493,7 +493,7 @@ fn typeck_with_fallback<'tcx>(
if fn_sig.is_some() { if fn_sig.is_some() {
fcx.regionck_fn(id, body, span, wf_tys); fcx.regionck_fn(id, body, span, wf_tys);
} else { } else {
fcx.regionck_expr(body); fcx.regionck_body(body);
} }
fcx.resolve_type_vars_in_body(body) fcx.resolve_type_vars_in_body(body)

View File

@ -76,9 +76,9 @@ use crate::check::dropck;
use crate::check::FnCtxt; use crate::check::FnCtxt;
use crate::mem_categorization as mc; use crate::mem_categorization as mc;
use crate::outlives::outlives_bounds::InferCtxtExt as _; use crate::outlives::outlives_bounds::InferCtxtExt as _;
use hir::def_id::LocalDefId;
use rustc_data_structures::stable_set::FxHashSet; use rustc_data_structures::stable_set::FxHashSet;
use rustc_hir as hir; use rustc_hir as hir;
use rustc_hir::def_id::LocalDefId;
use rustc_hir::intravisit::{self, Visitor}; use rustc_hir::intravisit::{self, Visitor};
use rustc_hir::PatKind; use rustc_hir::PatKind;
use rustc_infer::infer::outlives::env::OutlivesEnvironment; use rustc_infer::infer::outlives::env::OutlivesEnvironment;
@ -149,19 +149,18 @@ impl<'tcx> OutlivesEnvironmentExt<'tcx> for OutlivesEnvironment<'tcx> {
// PUBLIC ENTRY POINTS // PUBLIC ENTRY POINTS
impl<'a, 'tcx> FnCtxt<'a, 'tcx> { impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
pub fn regionck_expr(&self, body: &'tcx hir::Body<'tcx>) { pub fn regionck_body(&self, body: &'tcx hir::Body<'tcx>) {
let subject = self.tcx.hir().body_owner_def_id(body.id()); let body_owner = self.tcx.hir().body_owner_def_id(body.id());
let id = body.value.hir_id; let mut rcx = RegionCtxt::new(self, body_owner, self.param_env);
let mut rcx = RegionCtxt::new(self, id, Subject(subject), self.param_env);
// There are no add'l implied bounds when checking a // There are no add'l implied bounds when checking a
// standalone expr (e.g., the `E` in a type like `[u32; E]`). // standalone body (e.g., the `E` in a type like `[u32; E]`).
rcx.outlives_environment.save_implied_bounds(id); rcx.outlives_environment.save_implied_bounds(rcx.body_id());
if !self.errors_reported_since_creation() { if !self.errors_reported_since_creation() {
// regionck assumes typeck succeeded // regionck assumes typeck succeeded
rcx.visit_body(body); rcx.visit_body(body);
rcx.visit_region_obligations(id); rcx.visit_region_obligations(rcx.body_id());
} }
// Checked by NLL // Checked by NLL
rcx.fcx.skip_region_resolution(); rcx.fcx.skip_region_resolution();
@ -171,10 +170,10 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
/// types from which we should derive implied bounds, if any. /// types from which we should derive implied bounds, if any.
#[instrument(level = "debug", skip(self))] #[instrument(level = "debug", skip(self))]
pub fn regionck_item(&self, item_id: hir::HirId, span: Span, wf_tys: FxHashSet<Ty<'tcx>>) { pub fn regionck_item(&self, item_id: hir::HirId, span: Span, wf_tys: FxHashSet<Ty<'tcx>>) {
let subject = self.tcx.hir().local_def_id(item_id); let body_owner = self.tcx.hir().local_def_id(item_id);
let mut rcx = RegionCtxt::new(self, item_id, Subject(subject), self.param_env); let mut rcx = RegionCtxt::new(self, body_owner, self.param_env);
rcx.outlives_environment.add_implied_bounds(self, wf_tys, item_id, span); rcx.outlives_environment.add_implied_bounds(self, wf_tys, item_id, span);
rcx.outlives_environment.save_implied_bounds(item_id); rcx.outlives_environment.save_implied_bounds(rcx.body_id());
rcx.visit_region_obligations(item_id); rcx.visit_region_obligations(item_id);
rcx.resolve_regions_and_report_errors(); rcx.resolve_regions_and_report_errors();
} }
@ -195,9 +194,8 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
wf_tys: FxHashSet<Ty<'tcx>>, wf_tys: FxHashSet<Ty<'tcx>>,
) { ) {
debug!("regionck_fn(id={})", fn_id); debug!("regionck_fn(id={})", fn_id);
let subject = self.tcx.hir().body_owner_def_id(body.id()); let body_owner = self.tcx.hir().body_owner_def_id(body.id());
let hir_id = body.value.hir_id; let mut rcx = RegionCtxt::new(self, body_owner, self.param_env);
let mut rcx = RegionCtxt::new(self, hir_id, Subject(subject), self.param_env);
// We need to add the implied bounds from the function signature // We need to add the implied bounds from the function signature
rcx.outlives_environment.add_implied_bounds(self, wf_tys, fn_id, span); rcx.outlives_environment.add_implied_bounds(self, wf_tys, fn_id, span);
rcx.outlives_environment.save_implied_bounds(fn_id); rcx.outlives_environment.save_implied_bounds(fn_id);
@ -220,9 +218,6 @@ pub struct RegionCtxt<'a, 'tcx> {
outlives_environment: OutlivesEnvironment<'tcx>, outlives_environment: OutlivesEnvironment<'tcx>,
// id of innermost fn body id
body_id: hir::HirId,
// TODO: is this always equal to `body_id`?
body_owner: LocalDefId, body_owner: LocalDefId,
} }
@ -233,17 +228,21 @@ impl<'a, 'tcx> Deref for RegionCtxt<'a, 'tcx> {
} }
} }
pub struct Subject(LocalDefId);
impl<'a, 'tcx> RegionCtxt<'a, 'tcx> { impl<'a, 'tcx> RegionCtxt<'a, 'tcx> {
pub fn new( pub fn new(
fcx: &'a FnCtxt<'a, 'tcx>, fcx: &'a FnCtxt<'a, 'tcx>,
initial_body_id: hir::HirId, body_owner: LocalDefId,
Subject(subject): Subject,
param_env: ty::ParamEnv<'tcx>, param_env: ty::ParamEnv<'tcx>,
) -> RegionCtxt<'a, 'tcx> { ) -> RegionCtxt<'a, 'tcx> {
let outlives_environment = OutlivesEnvironment::new(param_env); let outlives_environment = OutlivesEnvironment::new(param_env);
RegionCtxt { fcx, body_id: initial_body_id, body_owner: subject, outlives_environment } RegionCtxt { fcx, body_owner, outlives_environment }
}
/// FIXME: Ideally all the callers would deal with
/// `LocalDefId`s as well. Ah well, this code is going
/// to be removed soon anyways 🤷
pub fn body_id(&self) -> hir::HirId {
self.tcx.hir().local_def_id_to_hir_id(self.body_owner)
} }
/// Try to resolve the type for the given node, returning `t_err` if an error results. Note that /// Try to resolve the type for the given node, returning `t_err` if an error results. Note that
@ -299,9 +298,7 @@ impl<'a, 'tcx> RegionCtxt<'a, 'tcx> {
) { ) {
// When we enter a function, we can derive // When we enter a function, we can derive
debug!("visit_fn_body(id={:?})", id); debug!("visit_fn_body(id={:?})", id);
let body_id = body.id(); let body_id = body.id();
self.body_id = body_id.hir_id;
self.body_owner = self.tcx.hir().body_owner_def_id(body_id); self.body_owner = self.tcx.hir().body_owner_def_id(body_id);
let Some(fn_sig) = self.typeck_results.borrow().liberated_fn_sigs().get(id) else { let Some(fn_sig) = self.typeck_results.borrow().liberated_fn_sigs().get(id) else {
@ -327,12 +324,10 @@ impl<'a, 'tcx> RegionCtxt<'a, 'tcx> {
debug!("visit_inline_const(id={:?})", id); debug!("visit_inline_const(id={:?})", id);
// Save state of current function. We will restore afterwards. // Save state of current function. We will restore afterwards.
let old_body_id = self.body_id;
let old_body_owner = self.body_owner; let old_body_owner = self.body_owner;
let env_snapshot = self.outlives_environment.push_snapshot_pre_typeck_child(); let env_snapshot = self.outlives_environment.push_snapshot_pre_typeck_child();
let body_id = body.id(); let body_id = body.id();
self.body_id = body_id.hir_id;
self.body_owner = self.tcx.hir().body_owner_def_id(body_id); self.body_owner = self.tcx.hir().body_owner_def_id(body_id);
self.outlives_environment.save_implied_bounds(body_id.hir_id); self.outlives_environment.save_implied_bounds(body_id.hir_id);
@ -342,7 +337,6 @@ impl<'a, 'tcx> RegionCtxt<'a, 'tcx> {
// Restore state from previous function. // Restore state from previous function.
self.outlives_environment.pop_snapshot_post_typeck_child(env_snapshot); self.outlives_environment.pop_snapshot_post_typeck_child(env_snapshot);
self.body_id = old_body_id;
self.body_owner = old_body_owner; self.body_owner = old_body_owner;
} }
@ -398,7 +392,6 @@ impl<'a, 'tcx> Visitor<'tcx> for RegionCtxt<'a, 'tcx> {
// Save state of current function before invoking // Save state of current function before invoking
// `visit_fn_body`. We will restore afterwards. // `visit_fn_body`. We will restore afterwards.
let old_body_id = self.body_id;
let old_body_owner = self.body_owner; let old_body_owner = self.body_owner;
let env_snapshot = self.outlives_environment.push_snapshot_pre_typeck_child(); let env_snapshot = self.outlives_environment.push_snapshot_pre_typeck_child();
@ -407,7 +400,6 @@ impl<'a, 'tcx> Visitor<'tcx> for RegionCtxt<'a, 'tcx> {
// Restore state from previous function. // Restore state from previous function.
self.outlives_environment.pop_snapshot_post_typeck_child(env_snapshot); self.outlives_environment.pop_snapshot_post_typeck_child(env_snapshot);
self.body_id = old_body_id;
self.body_owner = old_body_owner; self.body_owner = old_body_owner;
} }