1 #![deny(rustc::untranslatable_diagnostic)]
2 #![deny(rustc::diagnostic_outside_of_impl)]
5 use crate::{nll::ToRegionVid, region_infer::RegionInferenceContext};
6 use rustc_index::vec::{Idx, IndexVec};
7 use rustc_middle::mir::{Body, Local};
8 use rustc_middle::ty::{RegionVid, TyCtxt};
9 use rustc_span::source_map::Span;
10 use rustc_span::symbol::Symbol;
12 impl<'tcx> RegionInferenceContext<'tcx> {
13 pub(crate) fn get_var_name_and_span_for_region(
17 local_names: &IndexVec<Local, Option<Symbol>>,
18 upvars: &[Upvar<'tcx>],
20 ) -> Option<(Option<Symbol>, Span)> {
21 debug!("get_var_name_and_span_for_region(fr={fr:?})");
22 assert!(self.universal_regions().is_universal_region(fr));
24 debug!("get_var_name_and_span_for_region: attempting upvar");
25 self.get_upvar_index_for_region(tcx, fr)
27 // FIXME(project-rfc-2229#8): Use place span for diagnostics
28 let (name, span) = self.get_upvar_name_and_span_for_region(tcx, upvars, index);
32 debug!("get_var_name_and_span_for_region: attempting argument");
33 self.get_argument_index_for_region(tcx, fr).map(|index| {
34 self.get_argument_name_and_span_for_region(body, local_names, index)
39 /// Search the upvars (if any) to find one that references fr. Return its index.
40 pub(crate) fn get_upvar_index_for_region(
46 self.universal_regions().defining_ty.upvar_tys().position(|upvar_ty| {
47 debug!("get_upvar_index_for_region: upvar_ty={upvar_ty:?}");
48 tcx.any_free_region_meets(&upvar_ty, |r| {
49 let r = r.to_region_vid();
50 debug!("get_upvar_index_for_region: r={r:?} fr={fr:?}");
55 let upvar_ty = self.universal_regions().defining_ty.upvar_tys().nth(upvar_index);
58 "get_upvar_index_for_region: found {fr:?} in upvar {upvar_index} which has type {upvar_ty:?}",
64 /// Given the index of an upvar, finds its name and the span from where it was
66 pub(crate) fn get_upvar_name_and_span_for_region(
69 upvars: &[Upvar<'tcx>],
72 let upvar_hir_id = upvars[upvar_index].place.get_root_variable();
73 debug!("get_upvar_name_and_span_for_region: upvar_hir_id={upvar_hir_id:?}");
75 let upvar_name = tcx.hir().name(upvar_hir_id);
76 let upvar_span = tcx.hir().span(upvar_hir_id);
78 "get_upvar_name_and_span_for_region: upvar_name={upvar_name:?} upvar_span={upvar_span:?}",
81 (upvar_name, upvar_span)
84 /// Search the argument types for one that references fr (which should be a free region).
85 /// Returns Some(_) with the index of the input if one is found.
87 /// N.B., in the case of a closure, the index is indexing into the signature as seen by the
88 /// user - in particular, index 0 is not the implicit self parameter.
89 pub(crate) fn get_argument_index_for_region(
94 let implicit_inputs = self.universal_regions().defining_ty.implicit_inputs();
96 self.universal_regions().unnormalized_input_tys.iter().skip(implicit_inputs).position(
98 debug!("get_argument_index_for_region: arg_ty = {arg_ty:?}");
99 tcx.any_free_region_meets(arg_ty, |r| r.to_region_vid() == fr)
104 "get_argument_index_for_region: found {fr:?} in argument {argument_index} which has type {:?}",
105 self.universal_regions().unnormalized_input_tys[argument_index],
111 /// Given the index of an argument, finds its name (if any) and the span from where it was
113 pub(crate) fn get_argument_name_and_span_for_region(
116 local_names: &IndexVec<Local, Option<Symbol>>,
117 argument_index: usize,
118 ) -> (Option<Symbol>, Span) {
119 let implicit_inputs = self.universal_regions().defining_ty.implicit_inputs();
120 let argument_local = Local::new(implicit_inputs + argument_index + 1);
121 debug!("get_argument_name_and_span_for_region: argument_local={argument_local:?}");
123 let argument_name = local_names[argument_local];
124 let argument_span = body.local_decls[argument_local].source_info.span;
126 "get_argument_name_and_span_for_region: argument_name={argument_name:?} argument_span={argument_span:?}",
129 (argument_name, argument_span)