compiler/rustc_mir_build/src/build/misc.rs

   1 //! Miscellaneous builder routines that are not specific to building any particular
   2 //! kind of thing.
   3
   4 use crate::build::Builder;
   5
   6 use rustc_middle::mir::*;
   7 use rustc_middle::ty::{self, Ty};
   8 use rustc_span::{Span, DUMMY_SP};
   9 use rustc_trait_selection::infer::InferCtxtExt;
  10
  11 impl<'a, 'tcx> Builder<'a, 'tcx> {
  12     /// Adds a new temporary value of type `ty` storing the result of
  13     /// evaluating `expr`.
  14     ///
  15     /// N.B., **No cleanup is scheduled for this temporary.** You should
  16     /// call `schedule_drop` once the temporary is initialized.
  17     crate fn temp(&mut self, ty: Ty<'tcx>, span: Span) -> Place<'tcx> {
  18         // Mark this local as internal to avoid temporaries with types not present in the
  19         // user's code resulting in ICEs from the generator transform.
  20         let temp = self.local_decls.push(LocalDecl::new(ty, span).internal());
  21         let place = Place::from(temp);
  22         debug!("temp: created temp {:?} with type {:?}", place, self.local_decls[temp].ty);
  23         place
  24     }
  25
  26     /// Convenience function for creating a literal operand, one
  27     /// without any user type annotation.
  28     crate fn literal_operand(&mut self, span: Span, literal: ty::Const<'tcx>) -> Operand<'tcx> {
  29         let literal = literal.into();
  30         let constant = Box::new(Constant { span, user_ty: None, literal });
  31         Operand::Constant(constant)
  32     }
  33
  34     // Returns a zero literal operand for the appropriate type, works for
  35     // bool, char and integers.
  36     crate fn zero_literal(&mut self, span: Span, ty: Ty<'tcx>) -> Operand<'tcx> {
  37         let literal = ty::Const::from_bits(self.tcx, 0, ty::ParamEnv::empty().and(ty));
  38
  39         self.literal_operand(span, literal)
  40     }
  41
  42     crate fn push_usize(
  43         &mut self,
  44         block: BasicBlock,
  45         source_info: SourceInfo,
  46         value: u64,
  47     ) -> Place<'tcx> {
  48         let usize_ty = self.tcx.types.usize;
  49         let temp = self.temp(usize_ty, source_info.span);
  50         self.cfg.push_assign_constant(
  51             block,
  52             source_info,
  53             temp,
  54             Constant {
  55                 span: source_info.span,
  56                 user_ty: None,
  57                 literal: ConstantKind::from_usize(self.tcx, value),
  58             },
  59         );
  60         temp
  61     }
  62
  63     crate fn consume_by_copy_or_move(&self, place: Place<'tcx>) -> Operand<'tcx> {
  64         let tcx = self.tcx;
  65         let ty = place.ty(&self.local_decls, tcx).ty;
  66         if !self.infcx.type_is_copy_modulo_regions(self.param_env, ty, DUMMY_SP) {
  67             Operand::Move(place)
  68         } else {
  69             Operand::Copy(place)
  70         }
  71     }
  72 }