src/test/codegen/enum-match.rs

   1 // compile-flags: -Copt-level=1
   2 // only-x86_64
   3
   4 #![crate_type = "lib"]
   5
   6 // Check each of the 3 cases for `codegen_get_discr`.
   7
   8 // Case 0: One tagged variant.
   9 pub enum Enum0 {
  10     A(bool),
  11     B,
  12 }
  13
  14 // CHECK: define i8 @match0{{.*}}
  15 // CHECK-NEXT: start:
  16 // CHECK-NEXT: %1 = icmp eq i8 %0, 2
  17 // CHECK-NEXT: %2 = and i8 %0, 1
  18 // CHECK-NEXT: %.0 = select i1 %1, i8 13, i8 %2
  19 #[no_mangle]
  20 pub fn match0(e: Enum0) -> u8 {
  21     use Enum0::*;
  22     match e {
  23         A(b) => b as u8,
  24         B => 13,
  25     }
  26 }
  27
  28 // Case 1: Niche values are on a boundary for `range`.
  29 pub enum Enum1 {
  30     A(bool),
  31     B,
  32     C,
  33 }
  34
  35 // CHECK: define i8 @match1{{.*}}
  36 // CHECK-NEXT: start:
  37 // CHECK-NEXT: %1 = icmp ugt i8 %0, 1
  38 // CHECK-NEXT: %2 = zext i8 %0 to i64
  39 // CHECK-NEXT: %3 = add nsw i64 %2, -1
  40 // CHECK-NEXT: %_2 = select i1 %1, i64 %3, i64 0
  41 // CHECK-NEXT: switch i64 %_2, label {{.*}} [
  42 #[no_mangle]
  43 pub fn match1(e: Enum1) -> u8 {
  44     use Enum1::*;
  45     match e {
  46         A(b) => b as u8,
  47         B => 13,
  48         C => 100,
  49     }
  50 }
  51
  52 // Case 2: Special cases don't apply.
  53 pub enum X {
  54     _2=2, _3, _4, _5, _6, _7, _8, _9, _10, _11,
  55     _12, _13, _14, _15, _16, _17, _18, _19, _20,
  56     _21, _22, _23, _24, _25, _26, _27, _28, _29,
  57     _30, _31, _32, _33, _34, _35, _36, _37, _38,
  58     _39, _40, _41, _42, _43, _44, _45, _46, _47,
  59     _48, _49, _50, _51, _52, _53, _54, _55, _56,
  60     _57, _58, _59, _60, _61, _62, _63, _64, _65,
  61     _66, _67, _68, _69, _70, _71, _72, _73, _74,
  62     _75, _76, _77, _78, _79, _80, _81, _82, _83,
  63     _84, _85, _86, _87, _88, _89, _90, _91, _92,
  64     _93, _94, _95, _96, _97, _98, _99, _100, _101,
  65     _102, _103, _104, _105, _106, _107, _108, _109,
  66     _110, _111, _112, _113, _114, _115, _116, _117,
  67     _118, _119, _120, _121, _122, _123, _124, _125,
  68     _126, _127, _128, _129, _130, _131, _132, _133,
  69     _134, _135, _136, _137, _138, _139, _140, _141,
  70     _142, _143, _144, _145, _146, _147, _148, _149,
  71     _150, _151, _152, _153, _154, _155, _156, _157,
  72     _158, _159, _160, _161, _162, _163, _164, _165,
  73     _166, _167, _168, _169, _170, _171, _172, _173,
  74     _174, _175, _176, _177, _178, _179, _180, _181,
  75     _182, _183, _184, _185, _186, _187, _188, _189,
  76     _190, _191, _192, _193, _194, _195, _196, _197,
  77     _198, _199, _200, _201, _202, _203, _204, _205,
  78     _206, _207, _208, _209, _210, _211, _212, _213,
  79     _214, _215, _216, _217, _218, _219, _220, _221,
  80     _222, _223, _224, _225, _226, _227, _228, _229,
  81     _230, _231, _232, _233, _234, _235, _236, _237,
  82     _238, _239, _240, _241, _242, _243, _244, _245,
  83     _246, _247, _248, _249, _250, _251, _252, _253,
  84 }
  85
  86 pub enum Enum2 {
  87     A(X),
  88     B,
  89     C,
  90     D,
  91     E,
  92 }
  93
  94 // CHECK: define i8 @match2{{.*}}
  95 // CHECK-NEXT: start:
  96 // CHECK-NEXT: %1 = add i8 %0, 2
  97 // CHECK-NEXT: %2 = zext i8 %1 to i64
  98 // CHECK-NEXT: %3 = icmp ult i8 %1, 4
  99 // CHECK-NEXT: %4 = add nuw nsw i64 %2, 1
 100 // CHECK-NEXT: %_2 = select i1 %3, i64 %4, i64 0
 101 // CHECK-NEXT: switch i64 %_2, label {{.*}} [
 102 #[no_mangle]
 103 pub fn match2(e: Enum2) -> u8 {
 104     use Enum2::*;
 105     match e {
 106         A(b) => b as u8,
 107         B => 13,
 108         C => 100,
 109         D => 200,
 110         E => 250,
 111     }
 112 }