97 lines
3.0 KiB
Rust
97 lines
3.0 KiB
Rust
use rustc::hir;
|
|
use rustc::mir::mono::{Linkage, Visibility};
|
|
use rustc::ty::layout::HasTyCtxt;
|
|
use crate::base;
|
|
use crate::traits::*;
|
|
|
|
use rustc::mir::mono::MonoItem;
|
|
|
|
pub trait MonoItemExt<'a, 'tcx> {
|
|
fn define<Bx: BuilderMethods<'a, 'tcx>>(&self, cx: &'a Bx::CodegenCx);
|
|
fn predefine<Bx: BuilderMethods<'a, 'tcx>>(
|
|
&self,
|
|
cx: &'a Bx::CodegenCx,
|
|
linkage: Linkage,
|
|
visibility: Visibility
|
|
);
|
|
fn to_raw_string(&self) -> String;
|
|
}
|
|
|
|
impl<'a, 'tcx: 'a> MonoItemExt<'a, 'tcx> for MonoItem<'tcx> {
|
|
fn define<Bx: BuilderMethods<'a, 'tcx>>(&self, cx: &'a Bx::CodegenCx) {
|
|
debug!("BEGIN IMPLEMENTING '{} ({})' in cgu {}",
|
|
self.to_string(cx.tcx(), true),
|
|
self.to_raw_string(),
|
|
cx.codegen_unit().name());
|
|
|
|
match *self {
|
|
MonoItem::Static(def_id) => {
|
|
cx.codegen_static(def_id, cx.tcx().is_mutable_static(def_id));
|
|
}
|
|
MonoItem::GlobalAsm(hir_id) => {
|
|
let item = cx.tcx().hir().expect_item(hir_id);
|
|
if let hir::ItemKind::GlobalAsm(ref ga) = item.kind {
|
|
cx.codegen_global_asm(ga);
|
|
} else {
|
|
span_bug!(item.span, "Mismatch between hir::Item type and MonoItem type")
|
|
}
|
|
}
|
|
MonoItem::Fn(instance) => {
|
|
base::codegen_instance::<Bx>(&cx, instance);
|
|
}
|
|
}
|
|
|
|
debug!("END IMPLEMENTING '{} ({})' in cgu {}",
|
|
self.to_string(cx.tcx(), true),
|
|
self.to_raw_string(),
|
|
cx.codegen_unit().name());
|
|
}
|
|
|
|
fn predefine<Bx: BuilderMethods<'a, 'tcx>>(
|
|
&self,
|
|
cx: &'a Bx::CodegenCx,
|
|
linkage: Linkage,
|
|
visibility: Visibility
|
|
) {
|
|
debug!("BEGIN PREDEFINING '{} ({})' in cgu {}",
|
|
self.to_string(cx.tcx(), true),
|
|
self.to_raw_string(),
|
|
cx.codegen_unit().name());
|
|
|
|
let symbol_name = self.symbol_name(cx.tcx()).name.as_str();
|
|
|
|
debug!("symbol {}", &symbol_name);
|
|
|
|
match *self {
|
|
MonoItem::Static(def_id) => {
|
|
cx.predefine_static(def_id, linkage, visibility, &symbol_name);
|
|
}
|
|
MonoItem::Fn(instance) => {
|
|
cx.predefine_fn(instance, linkage, visibility, &symbol_name);
|
|
}
|
|
MonoItem::GlobalAsm(..) => {}
|
|
}
|
|
|
|
debug!("END PREDEFINING '{} ({})' in cgu {}",
|
|
self.to_string(cx.tcx(), true),
|
|
self.to_raw_string(),
|
|
cx.codegen_unit().name());
|
|
}
|
|
|
|
fn to_raw_string(&self) -> String {
|
|
match *self {
|
|
MonoItem::Fn(instance) => {
|
|
format!("Fn({:?}, {})",
|
|
instance.def,
|
|
instance.substs.as_ptr() as usize)
|
|
}
|
|
MonoItem::Static(id) => {
|
|
format!("Static({:?})", id)
|
|
}
|
|
MonoItem::GlobalAsm(id) => {
|
|
format!("GlobalAsm({:?})", id)
|
|
}
|
|
}
|
|
}
|
|
}
|