typeset 3.1.2

A DSL for defining source code pretty printers
Documentation 
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

use criterion::{Criterion, criterion_group, criterion_main};
use std::hint::black_box;
use typeset::{comp, compile, compile_safe, grp, nest, text};

fn create_deep_layout(depth: usize) -> Box<typeset::Layout> {
    let mut layout = text("base".to_string());

    for i in 0..depth {
        layout = nest(grp(comp(text(format!("level_{}", i)), layout, true, false)));
    }

    layout
}

fn create_wide_layout(width: usize) -> Box<typeset::Layout> {
    let mut layout = text("first".to_string());

    for i in 1..width {
        layout = comp(layout, text(format!("item_{}", i)), false, false);
    }

    layout
}

fn bench_compile_original(c: &mut Criterion) {
    c.bench_function("compile_original_deep_50", |b| {
        let layout = create_deep_layout(50);
        b.iter(|| compile(black_box(layout.clone())))
    });

    c.bench_function("compile_original_wide_100", |b| {
        let layout = create_wide_layout(100);
        b.iter(|| compile(black_box(layout.clone())))
    });
}

fn bench_compile_safe(c: &mut Criterion) {
    c.bench_function("compile_safe_deep_50", |b| {
        let layout = create_deep_layout(50);
        b.iter(|| compile_safe(black_box(layout.clone())).unwrap())
    });

    c.bench_function("compile_safe_wide_100", |b| {
        let layout = create_wide_layout(100);
        b.iter(|| compile_safe(black_box(layout.clone())).unwrap())
    });
}

fn bench_memory_stress(c: &mut Criterion) {
    // Test with larger layouts to stress memory allocation
    c.bench_function("compile_original_stress", |b| {
        let layout = create_deep_layout(100);
        b.iter(|| compile(black_box(layout.clone())))
    });

    c.bench_function("compile_safe_stress", |b| {
        let layout = create_deep_layout(100);
        b.iter(|| compile_safe(black_box(layout.clone())).unwrap())
    });
}

criterion_group!(
    benches,
    bench_compile_original,
    bench_compile_safe,
    bench_memory_stress
);
criterion_main!(benches);