Skip to content

Commit 7720c15

Browse files
pcwaltonpcwalton
committed
test: Implement pidigits and reverse-complement
1 parent bc0dd7f commit 7720c15

File tree

2 files changed

+328
-0
lines changed

2 files changed

+328
-0
lines changed

‎src/test/bench/shootout-pidigits.rs

Lines changed: 176 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,176 @@
1+
use core::cast::transmute;
2+
use core::from_str::FromStr;
3+
use core::libc::{STDOUT_FILENO, c_char, c_int, c_uint, c_void, fdopen, fputc};
4+
use core::libc::{fputs};
5+
use core::ptr::null;
6+
7+
struct mpz_t {
8+
_mp_alloc: c_int,
9+
_mp_size: c_int,
10+
_mp_limb_t: *c_void,
11+
}
12+
13+
impl mpz_t {
14+
fn new() -> mpz_t {
15+
mpz_t {
16+
_mp_alloc: 0,
17+
_mp_size: 0,
18+
_mp_limb_t: null(),
19+
}
20+
}
21+
}
22+
23+
#[link_args="-lgmp"]
24+
extern {
25+
#[fast_ffi]
26+
#[link_name="__gmpz_add"]
27+
fn mpz_add(x: *mpz_t, y: *mpz_t, z: *mpz_t);
28+
#[fast_ffi]
29+
#[link_name="__gmpz_cmp"]
30+
fn mpz_cmp(x: *mpz_t, y: *mpz_t) -> c_int;
31+
#[fast_ffi]
32+
#[link_name="__gmpz_fdiv_qr"]
33+
fn mpz_fdiv_qr(a: *mpz_t, b: *mpz_t, c: *mpz_t, d: *mpz_t);
34+
#[fast_ffi]
35+
#[link_name="__gmpz_get_ui"]
36+
fn mpz_get_ui(x: *mpz_t) -> c_uint;
37+
#[fast_ffi]
38+
#[link_name="__gmpz_init"]
39+
fn mpz_init(x: *mpz_t);
40+
#[fast_ffi]
41+
#[link_name="__gmpz_init_set_ui"]
42+
fn mpz_init_set_ui(x: *mpz_t, y: c_uint);
43+
#[fast_ffi]
44+
#[link_name="__gmpz_mul_2exp"]
45+
fn mpz_mul_2exp(x: *mpz_t, y: *mpz_t, z: c_uint);
46+
#[fast_ffi]
47+
#[link_name="__gmpz_mul_ui"]
48+
fn mpz_mul_ui(x: *mpz_t, y: *mpz_t, z: c_uint);
49+
#[fast_ffi]
50+
#[link_name="__gmpz_submul_ui"]
51+
fn mpz_submul_ui(x: *mpz_t, y: *mpz_t, z: c_uint);
52+
}
53+
54+
struct Context {
55+
numer: mpz_t,
56+
accum: mpz_t,
57+
denom: mpz_t,
58+
tmp1: mpz_t,
59+
tmp2: mpz_t,
60+
}
61+
62+
impl Context {
63+
fn new() -> Context {
64+
unsafe {
65+
let mut result = Context {
66+
numer: mpz_t::new(),
67+
accum: mpz_t::new(),
68+
denom: mpz_t::new(),
69+
tmp1: mpz_t::new(),
70+
tmp2: mpz_t::new(),
71+
};
72+
mpz_init(&result.tmp1);
73+
mpz_init(&result.tmp2);
74+
mpz_init_set_ui(&result.numer, 1);
75+
mpz_init_set_ui(&result.accum, 0);
76+
mpz_init_set_ui(&result.denom, 1);
77+
result
78+
}
79+
}
80+
81+
fn extract_digit(&mut self) -> i32 {
82+
unsafe {
83+
if mpz_cmp(&self.numer, &self.accum) > 0 {
84+
return -1;
85+
}
86+
87+
// Compute (numer * 3 + accum) / denom
88+
mpz_mul_2exp(&self.tmp1, &self.numer, 1);
89+
mpz_add(&self.tmp1, &self.tmp1, &self.numer);
90+
mpz_add(&self.tmp1, &self.tmp1, &self.accum);
91+
mpz_fdiv_qr(&self.tmp1, &self.tmp2, &self.tmp1, &self.denom);
92+
93+
// Now, if (numer * 4 + accum) % denom...
94+
mpz_add(&self.tmp2, &self.tmp2, &self.numer);
95+
96+
// ... is normalized, then the two divisions have the same result.
97+
if mpz_cmp(&self.tmp2, &self.denom) >= 0 {
98+
return -1;
99+
}
100+
101+
mpz_get_ui(&self.tmp1) as i32
102+
}
103+
}
104+
105+
fn next_term(&mut self, k: u32) {
106+
unsafe {
107+
let y2 = k*2 + 1;
108+
109+
mpz_mul_2exp(&self.tmp1, &self.numer, 1);
110+
mpz_add(&self.accum, &self.accum, &self.tmp1);
111+
mpz_mul_ui(&self.accum, &self.accum, y2);
112+
mpz_mul_ui(&self.numer, &self.numer, k);
113+
mpz_mul_ui(&self.denom, &self.denom, y2);
114+
}
115+
}
116+
117+
fn eliminate_digit(&mut self, d: u32) {
118+
unsafe {
119+
mpz_submul_ui(&self.accum, &self.denom, d);
120+
mpz_mul_ui(&self.accum, &self.accum, 10);
121+
mpz_mul_ui(&self.numer, &self.numer, 10);
122+
}
123+
}
124+
}
125+
126+
fn pidigits(n: u32) {
127+
unsafe {
128+
let mode = "w";
129+
let stdout = fdopen(STDOUT_FILENO as c_int, transmute(&mode[0]));
130+
131+
let mut d: i32;
132+
let mut i: u32 = 0, k: u32 = 0, m: u32;
133+
134+
let mut context = Context::new();
135+
loop {
136+
loop {
137+
k += 1;
138+
context.next_term(k);
139+
d = context.extract_digit();
140+
if d != -1 {
141+
break;
142+
}
143+
}
144+
145+
fputc((d as c_int) + ('0' as c_int), stdout);
146+
147+
i += 1;
148+
m = i % 10;
149+
if m == 0 {
150+
let res = fmt!("\t:%d\n", i as int);
151+
fputs(transmute(&res[0]), stdout);
152+
}
153+
if i >= n {
154+
break;
155+
}
156+
context.eliminate_digit(d as u32);
157+
}
158+
159+
if m != 0 {
160+
m = 10 - m;
161+
while m != 0 {
162+
m -= 1;
163+
fputc(' ' as c_int, stdout);
164+
}
165+
let res = fmt!("\t:%d\n", i as int);
166+
fputs(transmute(&res[0]), stdout);
167+
}
168+
}
169+
}
170+
171+
#[fixed_stack_segment]
172+
fn main() {
173+
let n: u32 = FromStr::from_str(os::args()[1]).get();
174+
pidigits(n);
175+
}
176+

‎src/test/bench/shootout-reverse-complement.rs

Lines changed: 152 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,152 @@
1+
use core::cast::transmute;
2+
use core::libc::{STDOUT_FILENO, c_int, fdopen, fgets, fopen, fputc, fwrite};
3+
use core::libc::{size_t};
4+
use core::ptr::null;
5+
use core::vec::{capacity, reserve, reserve_at_least};
6+
use core::vec::raw::set_len;
7+
8+
static LINE_LEN: u32 = 80;
9+
10+
static COMPLEMENTS: [u8, ..256] = [
11+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
12+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
13+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
14+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
15+
16+
0,
17+
'T' as u8,
18+
'V' as u8,
19+
'G' as u8,
20+
'H' as u8,
21+
0,
22+
0,
23+
'C' as u8,
24+
'D' as u8,
25+
0,
26+
0,
27+
'M' as u8,
28+
0,
29+
'K' as u8,
30+
'N' as u8,
31+
0,
32+
0,
33+
0,
34+
'Y' as u8,
35+
'S' as u8,
36+
'A' as u8,
37+
'A' as u8,
38+
'B' as u8,
39+
'W' as u8,
40+
0,
41+
'R' as u8,
42+
0,
43+
0,
44+
0,
45+
0,
46+
0,
47+
0,
48+
49+
0,
50+
'T' as u8,
51+
'V' as u8,
52+
'G' as u8,
53+
'H' as u8,
54+
0,
55+
0,
56+
'C' as u8,
57+
'D' as u8,
58+
0,
59+
0,
60+
'M' as u8,
61+
0,
62+
'K' as u8,
63+
'N' as u8,
64+
0,
65+
0,
66+
0,
67+
'Y' as u8,
68+
'S' as u8,
69+
'A' as u8,
70+
'A' as u8,
71+
'B' as u8,
72+
'W' as u8,
73+
0,
74+
'R' as u8,
75+
0,
76+
0,
77+
0,
78+
0,
79+
0,
80+
0,
81+
82+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
83+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
84+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
85+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
86+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
87+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
88+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
89+
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
90+
];
91+
92+
#[fixed_stack_segment]
93+
fn main() {
94+
unsafe {
95+
let mode = "r";
96+
//let stdin = fdopen(STDIN_FILENO as c_int, transmute(&mode[0]));
97+
let path = "reversecomplement-input.txt";
98+
let stdin = fopen(transmute(&path[0]), transmute(&mode[0]));
99+
let mode = "w";
100+
let stdout = fdopen(STDOUT_FILENO as c_int, transmute(&mode[0]));
101+
102+
let mut out: ~[u8] = ~[];
103+
reserve(&mut out, 12777888);
104+
let mut pos = 0;
105+
106+
loop {
107+
let needed = pos + (LINE_LEN as uint) + 1;
108+
if capacity(&out) < needed {
109+
reserve_at_least(&mut out, needed);
110+
}
111+
112+
let mut ptr = out.unsafe_mut_ref(pos);
113+
if fgets(transmute(ptr), LINE_LEN as c_int, stdin) == null() {
114+
break;
115+
}
116+
117+
// Don't change lines that begin with '>' or ';'.
118+
let first = *ptr;
119+
if first == ('>' as u8) {
120+
while *ptr != 0 {
121+
ptr = ptr.offset(1);
122+
}
123+
*ptr = '\n' as u8;
124+
125+
pos = (ptr as uint) - (out.unsafe_ref(0) as uint);
126+
fwrite(transmute(out.unsafe_ref(0)),
127+
1,
128+
pos as size_t,
129+
stdout);
130+
131+
pos = 0;
132+
loop;
133+
}
134+
135+
// Complement other lines.
136+
loop {
137+
let ch = *ptr;
138+
if ch == 0 {
139+
break;
140+
}
141+
*ptr = COMPLEMENTS.unsafe_get(ch as uint);
142+
ptr = ptr.offset(1);
143+
}
144+
*ptr = '\n' as u8;
145+
146+
pos = (ptr as uint) - (out.unsafe_ref(0) as uint);
147+
}
148+
149+
fwrite(transmute(out.unsafe_ref(0)), 1, pos as size_t, stdout);
150+
}
151+
}
152+

0 commit comments

Comments
 (0)