example/std_example.rs

   1 #![feature(core_intrinsics, generators, generator_trait, is_sorted)]
   2
   3 use std::arch::x86_64::*;
   4 use std::io::Write;
   5 use std::ops::Generator;
   6
   7 fn main() {
   8     println!("{:?}", std::env::args().collect::<Vec<_>>());
   9
  10     let mutex = std::sync::Mutex::new(());
  11     let _guard = mutex.lock().unwrap();
  12
  13     let _ = ::std::iter::repeat('a' as u8).take(10).collect::<Vec<_>>();
  14     let stderr = ::std::io::stderr();
  15     let mut stderr = stderr.lock();
  16
  17     std::thread::spawn(move || {
  18         println!("Hello from another thread!");
  19     });
  20
  21     writeln!(stderr, "some {} text", "<unknown>").unwrap();
  22
  23     let _ = std::process::Command::new("true").env("c", "d").spawn();
  24
  25     println!("cargo:rustc-link-lib=z");
  26
  27     static ONCE: std::sync::Once = std::sync::Once::new();
  28     ONCE.call_once(|| {});
  29
  30     let _eq = LoopState::Continue(()) == LoopState::Break(());
  31
  32     // Make sure ByValPair values with differently sized components are correctly passed
  33     map(None::<(u8, Box<Instruction>)>);
  34
  35     println!("{}", 2.3f32.exp());
  36     println!("{}", 2.3f32.exp2());
  37     println!("{}", 2.3f32.abs());
  38     println!("{}", 2.3f32.sqrt());
  39     println!("{}", 2.3f32.floor());
  40     println!("{}", 2.3f32.ceil());
  41     println!("{}", 2.3f32.min(1.0));
  42     println!("{}", 2.3f32.max(1.0));
  43     println!("{}", 2.3f32.powi(2));
  44     println!("{}", 2.3f32.log2());
  45     assert_eq!(2.3f32.copysign(-1.0), -2.3f32);
  46     println!("{}", 2.3f32.powf(2.0));
  47
  48     assert_eq!(-128i8, (-128i8).saturating_sub(1));
  49     assert_eq!(127i8, 127i8.saturating_sub(-128));
  50     assert_eq!(-128i8, (-128i8).saturating_add(-128));
  51     assert_eq!(127i8, 127i8.saturating_add(1));
  52
  53     assert_eq!(0b0000000000000000000000000010000010000000000000000000000000000000_0000000000100000000000000000000000001000000000000100000000000000u128.leading_zeros(), 26);
  54     assert_eq!(0b0000000000000000000000000010000000000000000000000000000000000000_0000000000000000000000000000000000001000000000000000000010000000u128.trailing_zeros(), 7);
  55
  56     let _d = 0i128.checked_div(2i128);
  57     let _d = 0u128.checked_div(2u128);
  58     assert_eq!(1u128 + 2, 3);
  59
  60     assert_eq!(0b100010000000000000000000000000000u128 >> 10, 0b10001000000000000000000u128);
  61     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 >> 64, 0xFEDCBA98765432u128);
  62     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 as i128 >> 64, 0xFEDCBA98765432i128);
  63
  64     let tmp = 353985398u128;
  65     assert_eq!(tmp * 932490u128, 330087843781020u128);
  66
  67     let tmp = -0x1234_5678_9ABC_DEF0i64;
  68     assert_eq!(tmp as i128, -0x1234_5678_9ABC_DEF0i128);
  69
  70     // Check that all u/i128 <-> float casts work correctly.
  71     let houndred_u128 = 100u128;
  72     let houndred_i128 = 100i128;
  73     let houndred_f32 = 100.0f32;
  74     let houndred_f64 = 100.0f64;
  75     assert_eq!(houndred_u128 as f32, 100.0);
  76     assert_eq!(houndred_u128 as f64, 100.0);
  77     assert_eq!(houndred_f32 as u128, 100);
  78     assert_eq!(houndred_f64 as u128, 100);
  79     assert_eq!(houndred_i128 as f32, 100.0);
  80     assert_eq!(houndred_i128 as f64, 100.0);
  81     assert_eq!(houndred_f32 as i128, 100);
  82     assert_eq!(houndred_f64 as i128, 100);
  83
  84     // Test signed 128bit comparing
  85     let max = usize::MAX as i128;
  86     if 100i128 < 0i128 || 100i128 > max {
  87         panic!();
  88     }
  89
  90     test_checked_mul();
  91
  92     let _a = 1u32 << 2u8;
  93
  94     let empty: [i32; 0] = [];
  95     assert!(empty.is_sorted());
  96
  97     println!("{:?}", std::intrinsics::caller_location());
  98
  99     unsafe {
 100         test_simd();
 101     }
 102
 103     Box::pin(move |mut _task_context| {
 104         yield ();
 105     }).as_mut().resume(0);
 106
 107     #[derive(Copy, Clone)]
 108     enum Nums {
 109         NegOne = -1,
 110     }
 111
 112     let kind = Nums::NegOne;
 113     assert_eq!(-1i128, kind as i128);
 114 }
 115
 116 #[target_feature(enable = "sse2")]
 117 unsafe fn test_simd() {
 118     let x = _mm_setzero_si128();
 119     let y = _mm_set1_epi16(7);
 120     let or = _mm_or_si128(x, y);
 121     let cmp_eq = _mm_cmpeq_epi8(y, y);
 122     let cmp_lt = _mm_cmplt_epi8(y, y);
 123
 124     assert_eq!(std::mem::transmute::<_, [u16; 8]>(or), [7, 7, 7, 7, 7, 7, 7, 7]);
 125     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_eq), [0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff]);
 126     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_lt), [0, 0, 0, 0, 0, 0, 0, 0]);
 127
 128     test_mm_slli_si128();
 129     test_mm_movemask_epi8();
 130     test_mm256_movemask_epi8();
 131     test_mm_add_epi8();
 132     test_mm_add_pd();
 133     test_mm_cvtepi8_epi16();
 134     test_mm_cvtsi128_si64();
 135
 136     // FIXME(#666) implement `#[rustc_arg_required_const(..)]` support
 137     //test_mm_extract_epi8();
 138
 139     let mask1 = _mm_movemask_epi8(dbg!(_mm_setr_epi8(255u8 as i8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)));
 140     assert_eq!(mask1, 1);
 141 }
 142
 143 #[target_feature(enable = "sse2")]
 144 unsafe fn test_mm_slli_si128() {
 145     #[rustfmt::skip]
 146     let a = _mm_setr_epi8(
 147         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 148     );
 149     let r = _mm_slli_si128(a, 1);
 150     let e = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
 151     assert_eq_m128i(r, e);
 152
 153     #[rustfmt::skip]
 154     let a = _mm_setr_epi8(
 155         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 156     );
 157     let r = _mm_slli_si128(a, 15);
 158     let e = _mm_setr_epi8(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1);
 159     assert_eq_m128i(r, e);
 160
 161     #[rustfmt::skip]
 162     let a = _mm_setr_epi8(
 163         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 164     );
 165     let r = _mm_slli_si128(a, 16);
 166     assert_eq_m128i(r, _mm_set1_epi8(0));
 167
 168     #[rustfmt::skip]
 169     let a = _mm_setr_epi8(
 170         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 171     );
 172     let r = _mm_slli_si128(a, -1);
 173     assert_eq_m128i(_mm_set1_epi8(0), r);
 174
 175     #[rustfmt::skip]
 176     let a = _mm_setr_epi8(
 177         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 178     );
 179     let r = _mm_slli_si128(a, -0x80000000);
 180     assert_eq_m128i(r, _mm_set1_epi8(0));
 181 }
 182
 183 #[target_feature(enable = "sse2")]
 184 unsafe fn test_mm_movemask_epi8() {
 185     #[rustfmt::skip]
 186     let a = _mm_setr_epi8(
 187         0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8, 0b01,
 188         0b0101, 0b1111_0000u8 as i8, 0, 0,
 189         0, 0, 0b1111_0000u8 as i8, 0b0101,
 190         0b01, 0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8,
 191     );
 192     let r = _mm_movemask_epi8(a);
 193     assert_eq!(r, 0b10100100_00100101);
 194 }
 195
 196 #[target_feature(enable = "avx2")]
 197 unsafe fn test_mm256_movemask_epi8() {
 198     let a = _mm256_set1_epi8(-1);
 199     let r = _mm256_movemask_epi8(a);
 200     let e = -1;
 201     assert_eq!(r, e);
 202 }
 203
 204 #[target_feature(enable = "sse2")]
 205 unsafe fn test_mm_add_epi8() {
 206     let a = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
 207     #[rustfmt::skip]
 208     let b = _mm_setr_epi8(
 209         16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31,
 210     );
 211     let r = _mm_add_epi8(a, b);
 212     #[rustfmt::skip]
 213     let e = _mm_setr_epi8(
 214         16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46,
 215     );
 216     assert_eq_m128i(r, e);
 217 }
 218
 219 #[target_feature(enable = "sse2")]
 220 unsafe fn test_mm_add_pd() {
 221     let a = _mm_setr_pd(1.0, 2.0);
 222     let b = _mm_setr_pd(5.0, 10.0);
 223     let r = _mm_add_pd(a, b);
 224     assert_eq_m128d(r, _mm_setr_pd(6.0, 12.0));
 225 }
 226
 227 fn assert_eq_m128i(x: std::arch::x86_64::__m128i, y: std::arch::x86_64::__m128i) {
 228     unsafe {
 229         assert_eq!(std::mem::transmute::<_, [u8; 16]>(x), std::mem::transmute::<_, [u8; 16]>(y));
 230     }
 231 }
 232
 233 #[target_feature(enable = "sse2")]
 234 pub unsafe fn assert_eq_m128d(a: __m128d, b: __m128d) {
 235     if _mm_movemask_pd(_mm_cmpeq_pd(a, b)) != 0b11 {
 236         panic!("{:?} != {:?}", a, b);
 237     }
 238 }
 239
 240 #[target_feature(enable = "sse2")]
 241 unsafe fn test_mm_cvtsi128_si64() {
 242     let r = _mm_cvtsi128_si64(std::mem::transmute::<[i64; 2], _>([5, 0]));
 243     assert_eq!(r, 5);
 244 }
 245
 246 #[target_feature(enable = "sse4.1")]
 247 unsafe fn test_mm_cvtepi8_epi16() {
 248     let a = _mm_set1_epi8(10);
 249     let r = _mm_cvtepi8_epi16(a);
 250     let e = _mm_set1_epi16(10);
 251     assert_eq_m128i(r, e);
 252     let a = _mm_set1_epi8(-10);
 253     let r = _mm_cvtepi8_epi16(a);
 254     let e = _mm_set1_epi16(-10);
 255     assert_eq_m128i(r, e);
 256 }
 257
 258 #[target_feature(enable = "sse4.1")]
 259 unsafe fn test_mm_extract_epi8() {
 260     #[rustfmt::skip]
 261     let a = _mm_setr_epi8(
 262         -1, 1, 2, 3, 4, 5, 6, 7,
 263         8, 9, 10, 11, 12, 13, 14, 15
 264     );
 265     let r1 = _mm_extract_epi8(a, 0);
 266     let r2 = _mm_extract_epi8(a, 19);
 267     assert_eq!(r1, 0xFF);
 268     assert_eq!(r2, 3);
 269 }
 270
 271 fn test_checked_mul() {
 272     let u: Option<u8> = u8::from_str_radix("1000", 10).ok();
 273     assert_eq!(u, None);
 274
 275     assert_eq!(1u8.checked_mul(255u8), Some(255u8));
 276     assert_eq!(255u8.checked_mul(255u8), None);
 277     assert_eq!(1i8.checked_mul(127i8), Some(127i8));
 278     assert_eq!(127i8.checked_mul(127i8), None);
 279     assert_eq!((-1i8).checked_mul(-127i8), Some(127i8));
 280     assert_eq!(1i8.checked_mul(-128i8), Some(-128i8));
 281     assert_eq!((-128i8).checked_mul(-128i8), None);
 282
 283     assert_eq!(1u64.checked_mul(u64::max_value()), Some(u64::max_value()));
 284     assert_eq!(u64::max_value().checked_mul(u64::max_value()), None);
 285     assert_eq!(1i64.checked_mul(i64::max_value()), Some(i64::max_value()));
 286     assert_eq!(i64::max_value().checked_mul(i64::max_value()), None);
 287     assert_eq!((-1i64).checked_mul(i64::min_value() + 1), Some(i64::max_value()));
 288     assert_eq!(1i64.checked_mul(i64::min_value()), Some(i64::min_value()));
 289     assert_eq!(i64::min_value().checked_mul(i64::min_value()), None);
 290 }
 291
 292 #[derive(PartialEq)]
 293 enum LoopState {
 294     Continue(()),
 295     Break(())
 296 }
 297
 298 pub enum Instruction {
 299     Increment,
 300     Loop,
 301 }
 302
 303 fn map(a: Option<(u8, Box<Instruction>)>) -> Option<Box<Instruction>> {
 304     match a {
 305         None => None,
 306         Some((_, instr)) => Some(instr),
 307     }
 308 }