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