crates/wavelet/bitdict/benches/my_benchmark.rs

View this file on GitHub
Last update: 2024-10-09 22:07:44+09:00

Depends on

crates/wavelet/bitdict/src/lib.rs

Code

use bitdict::BitDict;
use criterion::{black_box, criterion_group, criterion_main, Criterion};
use rand::prelude::*;
use rand_pcg::Pcg32;

pub fn rank1(c: &mut Criterion) {
    let mut rng = Pcg32::seed_from_u64(0);
    const SIZE: usize = 100000;
    let bool_vec: Vec<bool> = (0..SIZE).map(|_| rng.gen()).collect();
    let bit_vec = BitDict::from(&bool_vec[..]);
    c.bench_function("rank1_all", |b| {
        b.iter(|| {
            for i in 0..SIZE {
                black_box(bit_vec.rank_1(i));
            }
        });
    });
}

pub fn rank0(c: &mut Criterion) {
    let mut rng = Pcg32::seed_from_u64(0);
    const SIZE: usize = 100000;
    let bool_vec: Vec<bool> = (0..SIZE).map(|_| rng.gen()).collect();
    let bit_vec = BitDict::from(&bool_vec[..]);
    c.bench_function("rank0_all", |b| {
        b.iter(|| {
            for i in 0..SIZE {
                black_box(bit_vec.rank_0(i));
            }
        });
    });
}

pub fn select1(c: &mut Criterion) {
    let mut rng = Pcg32::seed_from_u64(0);
    const SIZE: usize = 100000;
    let bool_vec: Vec<bool> = (0..SIZE).map(|_| rng.gen()).collect();
    let bit_vec = BitDict::from(&bool_vec[..]);
    c.bench_function("select1_all", |b| {
        b.iter(|| {
            for i in 0..SIZE {
                black_box(bit_vec.select_1(i));
            }
        });
    });
}

pub fn select0(c: &mut Criterion) {
    let mut rng = Pcg32::seed_from_u64(0);
    const SIZE: usize = 100000;
    let bool_vec: Vec<bool> = (0..SIZE).map(|_| rng.gen()).collect();
    let bit_vec = BitDict::from(&bool_vec[..]);
    c.bench_function("select0_all", |b| {
        b.iter(|| {
            for i in 0..SIZE {
                black_box(bit_vec.select_0(i));
            }
        });
    });
}

criterion_group! {
    name = benches;
    config = Criterion::default();
    targets = rank1, rank0, select1, select0
}
criterion_main!(benches);

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.13.9/x64/lib/python3.13/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.13.9/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/rust.py", line 288, in bundle
    raise NotImplementedError
NotImplementedError