29 lines
849 B
Rust
Raw Normal View History

use rustc::traits;
use rustc::ty::adjustment::CustomCoerceUnsized;
use rustc::ty::{self, Ty, TyCtxt};
2017-10-25 15:39:54 +02:00
pub mod collector;
2017-11-23 16:02:02 +01:00
pub mod partitioning;
pub fn custom_coerce_unsize_info<'tcx>(
2019-06-14 00:48:52 +03:00
tcx: TyCtxt<'tcx>,
source_ty: Ty<'tcx>,
target_ty: Ty<'tcx>,
) -> CustomCoerceUnsized {
let def_id = tcx.lang_items().coerce_unsized_trait().unwrap();
let trait_ref = ty::Binder::bind(ty::TraitRef {
def_id,
2019-12-22 17:42:04 -05:00
substs: tcx.mk_substs_trait(source_ty, &[target_ty.into()]),
});
2019-12-22 17:42:04 -05:00
match tcx.codegen_fulfill_obligation((ty::ParamEnv::reveal_all(), trait_ref)) {
Some(traits::VtableImpl(traits::VtableImplData { impl_def_id, .. })) => {
tcx.coerce_unsized_info(impl_def_id).custom_kind.unwrap()
}
vtable => {
bug!("invalid `CoerceUnsized` vtable: {:?}", vtable);
}
}
}