Cache start and end point

Author: compiler-errors

compiler/rustc_ast_lowering/src/expr.rs

@@ -641,7 +641,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
             } else {
                 let try_span = this.mark_span_with_reason(
                     DesugaringKind::TryBlock,
-                    this.tcx.sess.source_map().end_point(body.span),
+                    this.tcx.end_point(body.span),
                     Some(Arc::clone(&this.allow_try_trait)),
                 );
 
@@ -1968,7 +1968,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
             span,
             Some(Arc::clone(&self.allow_try_trait)),
         );
-        let try_span = self.tcx.sess.source_map().end_point(span);
+        let try_span = self.tcx.end_point(span);
         let try_span = self.mark_span_with_reason(
             DesugaringKind::QuestionMark,
             try_span,

compiler/rustc_ast_lowering/src/lib.rs

@@ -1402,7 +1402,7 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
                 } else {
                     self.next_node_id()
                 };
-                let span = self.tcx.sess.source_map().start_point(t.span).shrink_to_hi();
+                let span = self.tcx.start_point(t.span).shrink_to_hi();
                 let region = Lifetime { ident: Ident::new(kw::UnderscoreLifetime, span), id };
                 (region, LifetimeSyntax::Hidden)
             }

compiler/rustc_middle/src/query/erase.rs

@@ -333,6 +333,7 @@ trivial! {
     rustc_span::ExpnHash,
     rustc_span::ExpnId,
     rustc_span::Span,
+    (rustc_span::Span, rustc_span::Span),
     rustc_span::Symbol,
     rustc_span::Ident,
     rustc_target::spec::PanicStrategy,

compiler/rustc_middle/src/query/keys.rs

@@ -70,6 +70,14 @@ impl Key for () {
     }
 }
 
+impl Key for Span {
+    type Cache<V> = DefaultCache<Self, V>;
+
+    fn default_span(&self, _: TyCtxt<'_>) -> Span {
+        *self
+    }
+}
+
 impl<'tcx> Key for ty::InstanceKind<'tcx> {
     type Cache<V> = DefaultCache<Self, V>;
 

compiler/rustc_middle/src/query/mod.rs

@@ -159,6 +159,10 @@ rustc_queries! {
         desc { "getting the source span" }
     }
 
+    query start_and_end_point(key: Span) -> (Span, Span) {
+        desc { "computing the start and end points for span" }
+    }
+
     /// Represents crate as a whole (as distinct from the top-level crate module).
     ///
     /// If you call `tcx.hir_crate(())` we will have to assume that any change

compiler/rustc_middle/src/ty/context.rs

@@ -2116,6 +2116,14 @@ impl<'tcx> TyCtxt<'tcx> {
         self.untracked.source_span.get(def_id).unwrap_or(DUMMY_SP)
     }
 
+    pub fn start_point(self, span: Span) -> Span {
+        self.start_and_end_point(span).0
+    }
+
+    pub fn end_point(self, span: Span) -> Span {
+        self.start_and_end_point(span).1
+    }
+
     #[inline(always)]
     pub fn with_stable_hashing_context<R>(
         self,
@@ -3430,6 +3438,10 @@ pub fn provide(providers: &mut Providers) {
         tcx.lang_items().panic_impl().is_some_and(|did| did.is_local())
     };
     providers.source_span = |tcx, def_id| tcx.untracked.source_span.get(def_id).unwrap_or(DUMMY_SP);
+    providers.start_and_end_point = |tcx, span| {
+        let sm = tcx.sess.source_map();
+        (sm.start_point(span), sm.end_point(span))
+    };
 }
 
 pub fn contains_name(attrs: &[Attribute], name: Symbol) -> bool {

compiler/rustc_mir_build/src/builder/matches/mod.rs

@@ -2489,7 +2489,7 @@ impl<'a, 'tcx> Builder<'a, 'tcx> {
                 });
 
             let source_info = self.source_info(guard_span);
-            let guard_end = self.source_info(tcx.sess.source_map().end_point(guard_span));
+            let guard_end = self.source_info(tcx.end_point(guard_span));
             let guard_frame = self.guard_context.pop().unwrap();
             debug!("Exiting guard building context with locals: {:?}", guard_frame);
 

compiler/rustc_mir_build/src/builder/scope.rs

@@ -1165,7 +1165,7 @@ impl<'a, 'tcx> Builder<'a, 'tcx> {
             if scope.region_scope == region_scope {
                 let region_scope_span = region_scope.span(self.tcx, self.region_scope_tree);
                 // Attribute scope exit drops to scope's closing brace.
-                let scope_end = self.tcx.sess.source_map().end_point(region_scope_span);
+                let scope_end = self.tcx.end_point(region_scope_span);
 
                 scope.drops.push(DropData {
                     source_info: SourceInfo { span: scope_end, scope: scope.source_scope },
@@ -1196,7 +1196,7 @@ impl<'a, 'tcx> Builder<'a, 'tcx> {
             scope.invalidate_cache();
             if scope.region_scope == region_scope {
                 let region_scope_span = region_scope.span(self.tcx, self.region_scope_tree);
-                let scope_end = self.tcx.sess.source_map().end_point(region_scope_span);
+                let scope_end = self.tcx.end_point(region_scope_span);
 
                 scope.drops.push(DropData {
                     source_info: SourceInfo { span: scope_end, scope: scope.source_scope },

compiler/rustc_resolve/src/late.rs

@@ -831,7 +831,7 @@ impl<'ra: 'ast, 'ast, 'tcx> Visitor<'ast> for LateResolutionVisitor<'_, 'ast, 'r
                 // Elided lifetime in reference: we resolve as if there was some lifetime `'_` with
                 // NodeId `ty.id`.
                 // This span will be used in case of elision failure.
-                let span = self.r.tcx.sess.source_map().start_point(ty.span);
+                let span = self.r.tcx.start_point(ty.span);
                 self.resolve_elided_lifetime(ty.id, span);
                 visit::walk_ty(self, ty);
             }

compiler/rustc_span/src/source_map.rs

@@ -886,8 +886,7 @@ impl SourceMap {
     /// Returns a new span representing just the first character of the given span.
     pub fn start_point(&self, sp: Span) -> Span {
         let width = {
-            let sp = sp.data();
-            let local_begin = self.lookup_byte_offset(sp.lo);
+            let local_begin = self.lookup_byte_offset(sp.lo());
             let start_index = local_begin.pos.to_usize();
             let src = local_begin.sf.external_src.read();