nalgebra/benches/common/macros.rs

#![macro_use]

macro_rules! bench_binop(
    ($name: ident, $t1: ty, $t2: ty, $binop: ident) => {
        fn $name(bh: &mut criterion::Criterion) {
            use rand::SeedableRng;
            let mut rng = IsaacRng::seed_from_u64(0);
            let a = rng.gen::<$t1>();
            let b = rng.gen::<$t2>();

            bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
                a.$binop(b)
            }));
        }
    }
);

macro_rules! bench_binop_ref(
    ($name: ident, $t1: ty, $t2: ty, $binop: ident) => {
        fn $name(bh: &mut criterion::Criterion) {
            use rand::SeedableRng;
            let mut rng = IsaacRng::seed_from_u64(0);
            let a = rng.gen::<$t1>();
            let b = rng.gen::<$t2>();

            bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
                a.$binop(&b)
            }));
        }
    }
);

macro_rules! bench_binop_fn(
    ($name: ident, $t1: ty, $t2: ty, $binop: path) => {
        fn $name(bh: &mut criterion::Criterion) {
            use rand::SeedableRng;
            let mut rng = IsaacRng::seed_from_u64(0);
            let a = rng.gen::<$t1>();
            let b = rng.gen::<$t2>();

            bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
                $binop(&a, &b)
            }));
        }
    }
);

macro_rules! bench_unop_na(
    ($name: ident, $t: ty, $unop: ident) => {
        fn $name(bh: &mut criterion::Criterion) {
            const LEN: usize = 1 << 13;

            use rand::SeedableRng;
            let mut rng = IsaacRng::seed_from_u64(0);

            let elems: Vec<$t> =  (0usize .. LEN).map(|_| rng.gen::<$t>()).collect();
            let mut i = 0;

            bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
                i = (i + 1) & (LEN - 1);

                unsafe {
                    std::hint::black_box(na::$unop(elems.get_unchecked(i)))
                }
            }));
        }
    }
);

macro_rules! bench_unop(
    ($name: ident, $t: ty, $unop: ident) => {
        fn $name(bh: &mut criterion::Criterion) {
            const LEN: usize = 1 << 13;

            use rand::SeedableRng;
            let mut rng = IsaacRng::seed_from_u64(0);

            let mut elems: Vec<$t> =  (0usize .. LEN).map(|_| rng.gen::<$t>()).collect();
            let mut i = 0;

            bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
                i = (i + 1) & (LEN - 1);

                unsafe {
                    std::hint::black_box(elems.get_unchecked_mut(i).$unop())
                }
            }));
        }
    }
);

macro_rules! bench_construction(
    ($name: ident, $constructor: path, $( $args: ident: $types: ty),*) => {
        fn $name(bh: &mut criterion::Criterion) {
            const LEN: usize = 1 << 13;

            use rand::SeedableRng;
            let mut rng = IsaacRng::seed_from_u64(0);

            $(let $args: Vec<$types> = (0usize .. LEN).map(|_| rng.gen::<$types>()).collect();)*
            let mut i = 0;

            bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
                i = (i + 1) & (LEN - 1);

                unsafe {
                    let res = $constructor($(*$args.get_unchecked(i),)*);
                    std::hint::black_box(res)
                }
            }));
        }
    }
);