src/test/bench/core-std.rs

   1 // Copyright 2012 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 // Microbenchmarks for various functions in std and extra
  12
  13 #[feature(macro_rules)];
  14
  15 extern crate rand;
  16 extern crate time;
  17
  18 use time::precise_time_s;
  19 use rand::Rng;
  20 use std::mem::swap;
  21 use std::os;
  22 use std::str;
  23 use std::slice;
  24 use std::vec;
  25 use std::io::File;
  26
  27 macro_rules! bench (
  28     ($argv:expr, $id:ident) => (maybe_run_test($argv, stringify!($id).to_owned(), $id))
  29 )
  30
  31 fn main() {
  32     let argv = os::args();
  33     let _tests = argv.slice(1, argv.len());
  34
  35     bench!(argv, shift_push);
  36     bench!(argv, read_line);
  37     bench!(argv, vec_plus);
  38     bench!(argv, vec_append);
  39     bench!(argv, vec_push_all);
  40     bench!(argv, is_utf8_ascii);
  41     bench!(argv, is_utf8_multibyte);
  42 }
  43
  44 fn maybe_run_test(argv: &[~str], name: ~str, test: ||) {
  45     let mut run_test = false;
  46
  47     if os::getenv("RUST_BENCH").is_some() {
  48         run_test = true
  49     } else if argv.len() > 0 {
  50         run_test = argv.iter().any(|x| x == &~"all") || argv.iter().any(|x| x == &name)
  51     }
  52
  53     if !run_test {
  54         return
  55     }
  56
  57     let start = precise_time_s();
  58     test();
  59     let stop = precise_time_s();
  60
  61     println!("{}:\t\t{} ms", name, (stop - start) * 1000.0);
  62 }
  63
  64 fn shift_push() {
  65     let mut v1 = Vec::from_elem(30000, 1);
  66     let mut v2 = Vec::new();
  67
  68     while v1.len() > 0 {
  69         v2.push(v1.shift().unwrap());
  70     }
  71 }
  72
  73 fn read_line() {
  74     use std::io::BufferedReader;
  75
  76     let mut path = Path::new(env!("CFG_SRC_DIR"));
  77     path.push("src/test/bench/shootout-k-nucleotide.data");
  78
  79     for _ in range(0, 3) {
  80         let mut reader = BufferedReader::new(File::open(&path).unwrap());
  81         for _line in reader.lines() {
  82         }
  83     }
  84 }
  85
  86 fn vec_plus() {
  87     let mut r = rand::task_rng();
  88
  89     let mut v = Vec::new();
  90     let mut i = 0;
  91     while i < 1500 {
  92         let rv = slice::from_elem(r.gen_range(0u, i + 1), i);
  93         if r.gen() {
  94             v.push_all_move(rv);
  95         } else {
  96             v = rv + v;
  97         }
  98         i += 1;
  99     }
 100 }
 101
 102 fn vec_append() {
 103     let mut r = rand::task_rng();
 104
 105     let mut v = Vec::new();
 106     let mut i = 0;
 107     while i < 1500 {
 108         let rv = slice::from_elem(r.gen_range(0u, i + 1), i);
 109         if r.gen() {
 110             v = vec::append(v, rv);
 111         }
 112         else {
 113             v = vec::append(rv, v);
 114         }
 115         i += 1;
 116     }
 117 }
 118
 119 fn vec_push_all() {
 120     let mut r = rand::task_rng();
 121
 122     let mut v = Vec::new();
 123     for i in range(0u, 1500) {
 124         let mut rv = slice::from_elem(r.gen_range(0u, i + 1), i);
 125         if r.gen() {
 126             v.push_all(rv);
 127         }
 128         else {
 129             swap(&mut v, &mut rv);
 130             v.push_all(rv);
 131         }
 132     }
 133 }
 134
 135 fn is_utf8_ascii() {
 136     let mut v : Vec<u8> = Vec::new();
 137     for _ in range(0u, 20000) {
 138         v.push('b' as u8);
 139         if !str::is_utf8(v) {
 140             fail!("is_utf8 failed");
 141         }
 142     }
 143 }
 144
 145 fn is_utf8_multibyte() {
 146     let s = "b¢€𤭢";
 147     let mut v : Vec<u8> = Vec::new();
 148     for _ in range(0u, 5000) {
 149         v.push_all(s.as_bytes());
 150         if !str::is_utf8(v) {
 151             fail!("is_utf8 failed");
 152         }
 153     }
 154 }