bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

rustc_span: Optimize more hygiene operations using Span::map_ctxt

Browse source
This commit is contained in:
Vadim Petrochenkov 2024-06-16 00:53:00 +03:00
parent 12b33d36f3
commit c6c1c992e2
1 changed files with 25 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

40
compiler/rustc_span/src/lib.rs
View file

@ -1065,33 +1065,41 @@ impl Span {
#[inline] #[inline]
pub fn remove_mark(&mut self) -> ExpnId { pub fn remove_mark(&mut self) -> ExpnId {
let mut span = self.data(); let mut mark = ExpnId::root();
let mark = span.ctxt.remove_mark(); *self = self.map_ctxt(|mut ctxt| {
*self = Span::new(span.lo, span.hi, span.ctxt, span.parent); mark = ctxt.remove_mark();
ctxt
});
mark mark
} }
#[inline] #[inline]
pub fn adjust(&mut self, expn_id: ExpnId) -> Option<ExpnId> { pub fn adjust(&mut self, expn_id: ExpnId) -> Option<ExpnId> {
let mut span = self.data(); let mut mark = None;
let mark = span.ctxt.adjust(expn_id); *self = self.map_ctxt(|mut ctxt| {
*self = Span::new(span.lo, span.hi, span.ctxt, span.parent); mark = ctxt.adjust(expn_id);
ctxt
});
mark mark
} }
#[inline] #[inline]
pub fn normalize_to_macros_2_0_and_adjust(&mut self, expn_id: ExpnId) -> Option<ExpnId> { pub fn normalize_to_macros_2_0_and_adjust(&mut self, expn_id: ExpnId) -> Option<ExpnId> {
let mut span = self.data(); let mut mark = None;
let mark = span.ctxt.normalize_to_macros_2_0_and_adjust(expn_id); *self = self.map_ctxt(|mut ctxt| {
*self = Span::new(span.lo, span.hi, span.ctxt, span.parent); mark = ctxt.normalize_to_macros_2_0_and_adjust(expn_id);
ctxt
});
mark mark
} }
#[inline] #[inline]
pub fn glob_adjust(&mut self, expn_id: ExpnId, glob_span: Span) -> Option<Option<ExpnId>> { pub fn glob_adjust(&mut self, expn_id: ExpnId, glob_span: Span) -> Option<Option<ExpnId>> {
let mut span = self.data(); let mut mark = None;
let mark = span.ctxt.glob_adjust(expn_id, glob_span); *self = self.map_ctxt(|mut ctxt| {
*self = Span::new(span.lo, span.hi, span.ctxt, span.parent); mark = ctxt.glob_adjust(expn_id, glob_span);
ctxt
});
mark mark
} }
@ -1101,9 +1109,11 @@ impl Span {
expn_id: ExpnId, expn_id: ExpnId,
glob_span: Span, glob_span: Span,
) -> Option<Option<ExpnId>> { ) -> Option<Option<ExpnId>> {
let mut span = self.data(); let mut mark = None;
let mark = span.ctxt.reverse_glob_adjust(expn_id, glob_span); *self = self.map_ctxt(|mut ctxt| {
*self = Span::new(span.lo, span.hi, span.ctxt, span.parent); mark = ctxt.reverse_glob_adjust(expn_id, glob_span);
ctxt
});
mark mark
} }