benches/helpers/mod.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

macro_rules! day {
    ($year:expr, $day:expr, $mod:ident) => {
        paste::paste! {
            #[allow(unused_must_use)]
            fn [<bench_ $year $mod _parse>](c: &mut criterion::Criterion) {
                c.bench_function(
                    &format!("{} day {} parse", $year, $day),
                    |b| {
                        b.iter_batched(
                            || advent_of_code::parse::data($year, $day).unwrap(),
                            |data| $mod::parse(criterion::black_box(data)),
                            criterion::BatchSize::PerIteration,
                        )
                    }
                );
            }
        }

        paste::paste! {
            #[allow(unused_must_use)]
            fn [<bench_ $year $mod _part1>](c: &mut criterion::Criterion) {
                c.bench_function(
                    &format!("{} day {} part 1", $year, $day),
                    |b| {
                        b.iter_batched(
                            || {
                                $mod::parse(
                                    advent_of_code::parse::data($year, $day)
                                        .unwrap(),
                                )
                                .unwrap()
                            },
                            |data| $mod::part1(criterion::black_box(data)),
                            criterion::BatchSize::PerIteration,
                        )
                    }
                );
            }
        }

        paste::paste! {
            #[allow(unused_must_use)]
            fn [<bench_ $year $mod _part2>](c: &mut criterion::Criterion) {
                c.bench_function(
                    &format!("{} day {} part 2", $year, $day),
                    |b| {
                        b.iter_batched(
                            || {
                                $mod::parse(
                                    advent_of_code::parse::data($year, $day)
                                        .unwrap(),
                                )
                                .unwrap()
                            },
                            |data| $mod::part2(criterion::black_box(data)),
                            criterion::BatchSize::PerIteration,
                        )
                    }
                );
            }
        }

        paste::paste! {
            criterion::criterion_group!(
                [<bench_ $year $mod>],
                [<bench_ $year $mod _parse>],
                [<bench_ $year $mod _part1>],
                [<bench_ $year $mod _part2>],
            );
        }
    };
}