compiler/rustc_monomorphize/src/lib.rs

   1 #![feature(array_windows)]
   2 #![feature(control_flow_enum)]
   3 #![recursion_limit = "256"]
   4 #![allow(rustc::potential_query_instability)]
   5 #![deny(rustc::untranslatable_diagnostic)]
   6 #![deny(rustc::diagnostic_outside_of_impl)]
   7
   8 #[macro_use]
   9 extern crate tracing;
  10 #[macro_use]
  11 extern crate rustc_middle;
  12
  13 use rustc_hir::lang_items::LangItem;
  14 use rustc_middle::traits;
  15 use rustc_middle::ty::adjustment::CustomCoerceUnsized;
  16 use rustc_middle::ty::query::Providers;
  17 use rustc_middle::ty::{self, Ty, TyCtxt};
  18
  19 mod collector;
  20 mod errors;
  21 mod partitioning;
  22 mod polymorphize;
  23 mod util;
  24
  25 fn custom_coerce_unsize_info<'tcx>(
  26     tcx: TyCtxt<'tcx>,
  27     source_ty: Ty<'tcx>,
  28     target_ty: Ty<'tcx>,
  29 ) -> CustomCoerceUnsized {
  30     let def_id = tcx.require_lang_item(LangItem::CoerceUnsized, None);
  31
  32     let trait_ref = ty::Binder::dummy(tcx.mk_trait_ref(def_id, source_ty, &[target_ty.into()]));
  33
  34     match tcx.codegen_select_candidate((ty::ParamEnv::reveal_all(), trait_ref)) {
  35         Ok(traits::ImplSource::UserDefined(traits::ImplSourceUserDefinedData {
  36             impl_def_id,
  37             ..
  38         })) => tcx.coerce_unsized_info(impl_def_id).custom_kind.unwrap(),
  39         impl_source => {
  40             bug!("invalid `CoerceUnsized` impl_source: {:?}", impl_source);
  41         }
  42     }
  43 }
  44
  45 pub fn provide(providers: &mut Providers) {
  46     partitioning::provide(providers);
  47     polymorphize::provide(providers);
  48 }