bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

implement get_filename/lines for span

Browse source
This commit is contained in:
Oğuz Ağcayazı 2023-10-11 12:44:59 +03:00
parent 6d05c430d2
commit 1d9481fdc8
3 changed files with 65 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

32
compiler/rustc_smir/src/rustc_smir/mod.rs
View file

@ -18,7 +18,7 @@ use rustc_span::def_id::{CrateNum, DefId, LOCAL_CRATE};
use rustc_target::abi::FieldIdx;
use stable_mir::mir::{CopyNonOverlapping, Statement, UserTypeProjection, VariantIdx};
use stable_mir::ty::{FloatTy, GenericParamDef, IntTy, Movability, RigidTy, Span, TyKind, UintTy};
use stable_mir::{self, opaque, Context};
use stable_mir::{self, opaque, Context, Filename};
use tracing::debug;
mod alloc;
@ -54,6 +54,36 @@ impl<'tcx> Context for Tables<'tcx> {
self.tcx.sess.source_map().span_to_diagnostic_string(self[span])
}
fn get_filename(&self, span: &Span) -> Filename {
opaque(
&self
.tcx
.sess
.source_map()
.span_to_filename(self[*span])
.display(rustc_span::FileNameDisplayPreference::Short)
.to_string(),
)
}
fn get_lines(&self, span: &Span) -> Vec<stable_mir::ty::LineInfo> {
let lines = &self
.tcx
.sess
.source_map()
.span_to_lines(self[*span])
.unwrap()
.lines
.iter()
.map(|line| stable_mir::ty::LineInfo {
line_index: line.line_index + 1,
start_col: line.start_col.0 + 1,
end_col: line.end_col.0 + 1,
})
.collect::<Vec<stable_mir::ty::LineInfo>>();
lines.to_vec()
}
fn def_kind(&mut self, def_id: stable_mir::DefId) -> stable_mir::DefKind {
self.tcx.def_kind(self[def_id]).stable(self)
}