1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
// Microbenchmarks for various functions in core and std
extern mod std;
use std::time::precise_time_s;
use core::rand::RngUtil;
use core::util;
macro_rules! bench (
($id:ident) => (maybe_run_test(argv, stringify!($id).to_owned(), $id))
)
fn main() {
let argv = os::args();
let tests = vec::slice(argv, 1, argv.len());
bench!(shift_push);
bench!(read_line);
bench!(vec_plus);
bench!(vec_append);
bench!(vec_push_all);
}
fn maybe_run_test(argv: &[~str], name: ~str, test: &fn()) {
let mut run_test = false;
if os::getenv(~"RUST_BENCH").is_some() {
run_test = true
} else if argv.len() > 0 {
run_test = argv.contains(&~"all") || argv.contains(&name)
}
if !run_test {
return
}
let start = precise_time_s();
test();
let stop = precise_time_s();
io::println(fmt!("%s:\t\t%f ms", name, (stop - start) * 1000f));
}
fn shift_push() {
let mut v1 = vec::from_elem(30000, 1);
let mut v2 = ~[];
while v1.len() > 0 {
v2.push(v1.shift());
}
}
fn read_line() {
let path = Path(env!("CFG_SRC_DIR"))
.push_rel(&Path("src/test/bench/shootout-k-nucleotide.data"));
for int::range(0, 3) |_i| {
let reader = result::get(&io::file_reader(&path));
while !reader.eof() {
reader.read_line();
}
}
}
fn vec_plus() {
let mut r = rand::rng();
let mut v = ~[];
let mut i = 0;
while i < 1500 {
let rv = vec::from_elem(r.gen_uint_range(0, i + 1), i);
if r.gen() {
v += rv;
}
else {
v = rv + v;
}
i += 1;
}
}
fn vec_append() {
let mut r = rand::rng();
let mut v = ~[];
let mut i = 0;
while i < 1500 {
let rv = vec::from_elem(r.gen_uint_range(0, i + 1), i);
if r.gen() {
v = vec::append(v, rv);
}
else {
v = vec::append(rv, v);
}
i += 1;
}
}
fn vec_push_all() {
let mut r = rand::rng();
let mut v = ~[];
for uint::range(0, 1500) |i| {
let mut rv = vec::from_elem(r.gen_uint_range(0, i + 1), i);
if r.gen() {
v.push_all(rv);
}
else {
util::swap(&mut v, &mut rv);
v.push_all(rv);
}
}
}
|
