src/librustc_mir/hair/util.rs

   1 use rustc::hir;
   2 use rustc::ty::{self, CanonicalUserType, TyCtxt, UserType};
   3
   4 crate trait UserAnnotatedTyHelpers<'gcx: 'tcx, 'tcx> {
   5     fn tcx(&self) -> TyCtxt<'_, 'gcx, 'tcx>;
   6
   7     fn tables(&self) -> &ty::TypeckTables<'tcx>;
   8
   9     /// Looks up the type associated with this hir-id and applies the
  10     /// user-given substitutions; the hir-id must map to a suitable
  11     /// type.
  12     fn user_substs_applied_to_ty_of_hir_id(
  13         &self,
  14         hir_id: hir::HirId,
  15     ) -> Option<CanonicalUserType<'tcx>> {
  16         let user_provided_types = self.tables().user_provided_types();
  17         let mut user_ty = *user_provided_types.get(hir_id)?;
  18         debug!("user_subts_applied_to_ty_of_hir_id: user_ty={:?}", user_ty);
  19         let ty = self.tables().node_type(hir_id);
  20         match ty.sty {
  21             ty::Adt(adt_def, ..) => {
  22                 if let UserType::TypeOf(ref mut did, _) = &mut user_ty.value {
  23                     *did = adt_def.did;
  24                 }
  25                 Some(user_ty)
  26             }
  27             ty::FnDef(..) => Some(user_ty),
  28             _ => bug!(
  29                 "ty: {:?} should not have user provided type {:?} recorded ",
  30                 ty,
  31                 user_ty
  32             ),
  33         }
  34     }
  35 }