src/librustc/macros.rs

   1 // ignore-tidy-linelength
   2
   3 macro_rules! enum_from_u32 {
   4     ($(#[$attr:meta])* pub enum $name:ident {
   5         $($variant:ident = $e:expr,)*
   6     }) => {
   7         $(#[$attr])*
   8         pub enum $name {
   9             $($variant = $e),*
  10         }
  11
  12         impl $name {
  13             pub fn from_u32(u: u32) -> Option<$name> {
  14                 $(if u == $name::$variant as u32 {
  15                     return Some($name::$variant)
  16                 })*
  17                 None
  18             }
  19         }
  20     };
  21     ($(#[$attr:meta])* pub enum $name:ident {
  22         $($variant:ident,)*
  23     }) => {
  24         $(#[$attr])*
  25         pub enum $name {
  26             $($variant,)*
  27         }
  28
  29         impl $name {
  30             pub fn from_u32(u: u32) -> Option<$name> {
  31                 $(if u == $name::$variant as u32 {
  32                     return Some($name::$variant)
  33                 })*
  34                 None
  35             }
  36         }
  37     }
  38 }
  39
  40 #[macro_export]
  41 macro_rules! bug {
  42     () => ( bug!("impossible case reached") );
  43     ($($message:tt)*) => ({
  44         $crate::util::bug::bug_fmt(file!(), line!(), format_args!($($message)*))
  45     })
  46 }
  47
  48 #[macro_export]
  49 macro_rules! span_bug {
  50     ($span:expr, $($message:tt)*) => ({
  51         $crate::util::bug::span_bug_fmt(file!(), line!(), $span, format_args!($($message)*))
  52     })
  53 }
  54
  55 #[macro_export]
  56 macro_rules! __impl_stable_hash_field {
  57     ($field:ident, $ctx:expr, $hasher:expr) => ($field.hash_stable($ctx, $hasher));
  58     ($field:ident, $ctx:expr, $hasher:expr, _) => ({ let _ = $field; });
  59     ($field:ident, $ctx:expr, $hasher:expr, $delegate:expr) => ($delegate.hash_stable($ctx, $hasher));
  60 }
  61
  62 #[macro_export]
  63 macro_rules! impl_stable_hash_for {
  64     // Enums
  65     (enum $enum_name:path {
  66         $( $variant:ident
  67            // this incorrectly allows specifying both tuple-like and struct-like fields, as in `Variant(a,b){c,d}`,
  68            // when it should be only one or the other
  69            $( ( $($field:ident $(-> $delegate:tt)?),* ) )?
  70            $( { $($named_field:ident $(-> $named_delegate:tt)?),* } )?
  71         ),* $(,)?
  72     }) => {
  73         impl_stable_hash_for!(
  74             impl<> for enum $enum_name [ $enum_name ] { $( $variant
  75                 $( ( $($field $(-> $delegate)?),* ) )?
  76                 $( { $($named_field $(-> $named_delegate)?),* } )?
  77             ),* }
  78         );
  79     };
  80     // We want to use the enum name both in the `impl ... for $enum_name` as well as for
  81     // importing all the variants. Unfortunately it seems we have to take the name
  82     // twice for this purpose
  83     (impl<$($T:ident),* $(,)?>
  84         for enum $enum_name:path
  85         [ $enum_path:path ]
  86     {
  87         $( $variant:ident
  88            // this incorrectly allows specifying both tuple-like and struct-like fields, as in `Variant(a,b){c,d}`,
  89            // when it should be only one or the other
  90            $( ( $($field:ident $(-> $delegate:tt)?),* ) )?
  91            $( { $($named_field:ident $(-> $named_delegate:tt)?),* } )?
  92         ),* $(,)?
  93     }) => {
  94         impl<$($T,)*>
  95             ::rustc_data_structures::stable_hasher::HashStable<$crate::ich::StableHashingContext<'a>>
  96             for $enum_name
  97             where $($T: ::rustc_data_structures::stable_hasher::HashStable<$crate::ich::StableHashingContext<'a>>),*
  98         {
  99             #[inline]
 100             fn hash_stable(&self,
 101                            __ctx: &mut $crate::ich::StableHashingContext<'a>,
 102                            __hasher: &mut ::rustc_data_structures::stable_hasher::StableHasher) {
 103                 use $enum_path::*;
 104                 ::std::mem::discriminant(self).hash_stable(__ctx, __hasher);
 105
 106                 match *self {
 107                     $(
 108                         $variant $( ( $(ref $field),* ) )? $( { $(ref $named_field),* } )? => {
 109                             $($( __impl_stable_hash_field!($field, __ctx, __hasher $(, $delegate)?) );*)?
 110                             $($( __impl_stable_hash_field!($named_field, __ctx, __hasher $(, $named_delegate)?) );*)?
 111                         }
 112                     )*
 113                 }
 114             }
 115         }
 116     };
 117     // Structs
 118     (struct $struct_name:path { $($field:ident $(-> $delegate:tt)?),* $(,)? }) => {
 119         impl_stable_hash_for!(
 120             impl<> for struct $struct_name { $($field $(-> $delegate)?),* }
 121         );
 122     };
 123     (impl<$($T:ident),* $(,)?> for struct $struct_name:path {
 124         $($field:ident $(-> $delegate:tt)?),* $(,)?
 125     }) => {
 126         impl<$($T,)*>
 127             ::rustc_data_structures::stable_hasher::HashStable<$crate::ich::StableHashingContext<'a>> for $struct_name
 128             where $($T: ::rustc_data_structures::stable_hasher::HashStable<$crate::ich::StableHashingContext<'a>>),*
 129         {
 130             #[inline]
 131             fn hash_stable(&self,
 132                            __ctx: &mut $crate::ich::StableHashingContext<'a>,
 133                            __hasher: &mut ::rustc_data_structures::stable_hasher::StableHasher) {
 134                 let $struct_name {
 135                     $(ref $field),*
 136                 } = *self;
 137
 138                 $( __impl_stable_hash_field!($field, __ctx, __hasher $(, $delegate)?) );*
 139             }
 140         }
 141     };
 142     // Tuple structs
 143     // We cannot use normal parentheses here, the parser won't allow it
 144     (tuple_struct $struct_name:path { $($field:ident $(-> $delegate:tt)?),*  $(,)? }) => {
 145         impl_stable_hash_for!(
 146             impl<> for tuple_struct $struct_name { $($field $(-> $delegate)?),* }
 147         );
 148     };
 149     (impl<$($T:ident),* $(,)?>
 150      for tuple_struct $struct_name:path { $($field:ident $(-> $delegate:tt)?),*  $(,)? }) => {
 151         impl<$($T,)*>
 152             ::rustc_data_structures::stable_hasher::HashStable<$crate::ich::StableHashingContext<'a>> for $struct_name
 153             where $($T: ::rustc_data_structures::stable_hasher::HashStable<$crate::ich::StableHashingContext<'a>>),*
 154         {
 155             #[inline]
 156             fn hash_stable(&self,
 157                            __ctx: &mut $crate::ich::StableHashingContext<'a>,
 158                            __hasher: &mut ::rustc_data_structures::stable_hasher::StableHasher) {
 159                 let $struct_name (
 160                     $(ref $field),*
 161                 ) = *self;
 162
 163                 $( __impl_stable_hash_field!($field, __ctx, __hasher $(, $delegate)?) );*
 164             }
 165         }
 166     };
 167 }
 168
 169 #[macro_export]
 170 macro_rules! impl_stable_hash_for_spanned {
 171     ($T:path) => (
 172
 173         impl HashStable<StableHashingContext<'a>> for ::syntax::source_map::Spanned<$T>
 174         {
 175             #[inline]
 176             fn hash_stable(&self,
 177                            hcx: &mut StableHashingContext<'a>,
 178                            hasher: &mut StableHasher) {
 179                 self.node.hash_stable(hcx, hasher);
 180                 self.span.hash_stable(hcx, hasher);
 181             }
 182         }
 183     );
 184 }
 185
 186 ///////////////////////////////////////////////////////////////////////////
 187 // Lift and TypeFoldable macros
 188 //
 189 // When possible, use one of these (relatively) convenient macros to write
 190 // the impls for you.
 191
 192 #[macro_export]
 193 macro_rules! CloneLiftImpls {
 194     (for <$tcx:lifetime> { $($ty:ty,)+ }) => {
 195         $(
 196             impl<$tcx> $crate::ty::Lift<$tcx> for $ty {
 197                 type Lifted = Self;
 198                 fn lift_to_tcx(&self, _: $crate::ty::TyCtxt<$tcx>) -> Option<Self> {
 199                     Some(Clone::clone(self))
 200                 }
 201             }
 202         )+
 203     };
 204
 205     ($($ty:ty,)+) => {
 206         CloneLiftImpls! {
 207             for <'tcx> {
 208                 $($ty,)+
 209             }
 210         }
 211     };
 212 }
 213
 214 /// Used for types that are `Copy` and which **do not care arena
 215 /// allocated data** (i.e., don't need to be folded).
 216 #[macro_export]
 217 macro_rules! CloneTypeFoldableImpls {
 218     (for <$tcx:lifetime> { $($ty:ty,)+ }) => {
 219         $(
 220             impl<$tcx> $crate::ty::fold::TypeFoldable<$tcx> for $ty {
 221                 fn super_fold_with<F: $crate::ty::fold::TypeFolder<$tcx>>(
 222                     &self,
 223                     _: &mut F
 224                 ) -> $ty {
 225                     Clone::clone(self)
 226                 }
 227
 228                 fn super_visit_with<F: $crate::ty::fold::TypeVisitor<$tcx>>(
 229                     &self,
 230                     _: &mut F)
 231                     -> bool
 232                 {
 233                     false
 234                 }
 235             }
 236         )+
 237     };
 238
 239     ($($ty:ty,)+) => {
 240         CloneTypeFoldableImpls! {
 241             for <'tcx> {
 242                 $($ty,)+
 243             }
 244         }
 245     };
 246 }
 247
 248 #[macro_export]
 249 macro_rules! CloneTypeFoldableAndLiftImpls {
 250     ($($t:tt)*) => {
 251         CloneTypeFoldableImpls! { $($t)* }
 252         CloneLiftImpls! { $($t)* }
 253     }
 254 }
 255
 256 #[macro_export]
 257 macro_rules! EnumTypeFoldableImpl {
 258     (impl<$($p:tt),*> TypeFoldable<$tcx:tt> for $s:path {
 259         $($variants:tt)*
 260     } $(where $($wc:tt)*)*) => {
 261         impl<$($p),*> $crate::ty::fold::TypeFoldable<$tcx> for $s
 262             $(where $($wc)*)*
 263         {
 264             fn super_fold_with<V: $crate::ty::fold::TypeFolder<$tcx>>(
 265                 &self,
 266                 folder: &mut V,
 267             ) -> Self {
 268                 EnumTypeFoldableImpl!(@FoldVariants(self, folder) input($($variants)*) output())
 269             }
 270
 271             fn super_visit_with<V: $crate::ty::fold::TypeVisitor<$tcx>>(
 272                 &self,
 273                 visitor: &mut V,
 274             ) -> bool {
 275                 EnumTypeFoldableImpl!(@VisitVariants(self, visitor) input($($variants)*) output())
 276             }
 277         }
 278     };
 279
 280     (@FoldVariants($this:expr, $folder:expr) input() output($($output:tt)*)) => {
 281         match $this {
 282             $($output)*
 283         }
 284     };
 285
 286     (@FoldVariants($this:expr, $folder:expr)
 287      input( ($variant:path) ( $($variant_arg:ident),* ) , $($input:tt)*)
 288      output( $($output:tt)*) ) => {
 289         EnumTypeFoldableImpl!(
 290             @FoldVariants($this, $folder)
 291                 input($($input)*)
 292                 output(
 293                     $variant ( $($variant_arg),* ) => {
 294                         $variant (
 295                             $($crate::ty::fold::TypeFoldable::fold_with($variant_arg, $folder)),*
 296                         )
 297                     }
 298                     $($output)*
 299                 )
 300         )
 301     };
 302
 303     (@FoldVariants($this:expr, $folder:expr)
 304      input( ($variant:path) { $($variant_arg:ident),* $(,)? } , $($input:tt)*)
 305      output( $($output:tt)*) ) => {
 306         EnumTypeFoldableImpl!(
 307             @FoldVariants($this, $folder)
 308                 input($($input)*)
 309                 output(
 310                     $variant { $($variant_arg),* } => {
 311                         $variant {
 312                             $($variant_arg: $crate::ty::fold::TypeFoldable::fold_with(
 313                                 $variant_arg, $folder
 314                             )),* }
 315                     }
 316                     $($output)*
 317                 )
 318         )
 319     };
 320
 321     (@FoldVariants($this:expr, $folder:expr)
 322      input( ($variant:path), $($input:tt)*)
 323      output( $($output:tt)*) ) => {
 324         EnumTypeFoldableImpl!(
 325             @FoldVariants($this, $folder)
 326                 input($($input)*)
 327                 output(
 328                     $variant => { $variant }
 329                     $($output)*
 330                 )
 331         )
 332     };
 333
 334     (@VisitVariants($this:expr, $visitor:expr) input() output($($output:tt)*)) => {
 335         match $this {
 336             $($output)*
 337         }
 338     };
 339
 340     (@VisitVariants($this:expr, $visitor:expr)
 341      input( ($variant:path) ( $($variant_arg:ident),* ) , $($input:tt)*)
 342      output( $($output:tt)*) ) => {
 343         EnumTypeFoldableImpl!(
 344             @VisitVariants($this, $visitor)
 345                 input($($input)*)
 346                 output(
 347                     $variant ( $($variant_arg),* ) => {
 348                         false $(|| $crate::ty::fold::TypeFoldable::visit_with(
 349                             $variant_arg, $visitor
 350                         ))*
 351                     }
 352                     $($output)*
 353                 )
 354         )
 355     };
 356
 357     (@VisitVariants($this:expr, $visitor:expr)
 358      input( ($variant:path) { $($variant_arg:ident),* $(,)? } , $($input:tt)*)
 359      output( $($output:tt)*) ) => {
 360         EnumTypeFoldableImpl!(
 361             @VisitVariants($this, $visitor)
 362                 input($($input)*)
 363                 output(
 364                     $variant { $($variant_arg),* } => {
 365                         false $(|| $crate::ty::fold::TypeFoldable::visit_with(
 366                             $variant_arg, $visitor
 367                         ))*
 368                     }
 369                     $($output)*
 370                 )
 371         )
 372     };
 373
 374     (@VisitVariants($this:expr, $visitor:expr)
 375      input( ($variant:path), $($input:tt)*)
 376      output( $($output:tt)*) ) => {
 377         EnumTypeFoldableImpl!(
 378             @VisitVariants($this, $visitor)
 379                 input($($input)*)
 380                 output(
 381                     $variant => { false }
 382                     $($output)*
 383                 )
 384         )
 385     };
 386 }