-
Notifications
You must be signed in to change notification settings - Fork 77
/
compute.rs
80 lines (67 loc) · 2.5 KB
/
compute.rs
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
#[macro_use]
extern crate criterion;
use std::fs::File;
use std::io::{Cursor, Read};
use needletail::parse_fastx_reader;
use sourmash::cmd::ComputeParameters;
use sourmash::signature::Signature;
use criterion::Criterion;
fn add_sequence(c: &mut Criterion) {
let cp = ComputeParameters::default();
let template_sig = Signature::from_params(&cp);
let mut data: Vec<u8> = vec![];
let mut f = File::open("../../tests/test-data/ecoli.genes.fna").unwrap();
let _ = f.read_to_end(&mut data);
let data = data.repeat(10);
let data_upper = data.to_ascii_uppercase();
let data_lower = data.to_ascii_lowercase();
let data_errors: Vec<u8> = data
.iter()
.enumerate()
.map(|(i, x)| if i % 89 == 1 { 'N' as u8 } else { *x })
.collect();
let mut group = c.benchmark_group("add_sequence");
group.sample_size(10);
group.bench_function("valid", |b| {
b.iter(|| {
let fasta_data = Cursor::new(data_upper.clone());
let mut sig = template_sig.clone();
let mut parser = parse_fastx_reader(fasta_data).unwrap();
while let Some(rec) = parser.next() {
sig.add_sequence(&rec.unwrap().seq(), false).unwrap();
}
});
});
group.bench_function("lowercase", |b| {
b.iter(|| {
let fasta_data = Cursor::new(data_lower.clone());
let mut sig = template_sig.clone();
let mut parser = parse_fastx_reader(fasta_data).unwrap();
while let Some(rec) = parser.next() {
sig.add_sequence(&rec.unwrap().seq(), false).unwrap();
}
});
});
group.bench_function("invalid kmers", |b| {
b.iter(|| {
let fasta_data = Cursor::new(data_errors.clone());
let mut sig = template_sig.clone();
let mut parser = parse_fastx_reader(fasta_data).unwrap();
while let Some(rec) = parser.next() {
sig.add_sequence(&rec.unwrap().seq(), true).unwrap();
}
});
});
group.bench_function("force with valid kmers", |b| {
b.iter(|| {
let fasta_data = Cursor::new(data_upper.clone());
let mut sig = template_sig.clone();
let mut parser = parse_fastx_reader(fasta_data).unwrap();
while let Some(rec) = parser.next() {
sig.add_sequence(&rec.unwrap().seq(), true).unwrap();
}
});
});
}
criterion_group!(compute, add_sequence);
criterion_main!(compute);