typst/bench/src/bench.rs

use criterion::{criterion_group, criterion_main, Criterion};
use fontdock::fs::FsIndex;

use typst::env::{Env, ResourceLoader};
use typst::eval::{eval, State};
use typst::export::pdf;
use typst::font::FsIndexExt;
use typst::layout::layout;
use typst::library;
use typst::parse::parse;
use typst::typeset;

const FONT_DIR: &str = "../fonts";
const COMA: &str = include_str!("../../tests/full/typ/coma.typ");

fn benchmarks(c: &mut Criterion) {
    macro_rules! bench {
        ($name:literal: $($tts:tt)*) => {
            c.bench_function($name, |b| b.iter(|| $($tts)*));
        };
    }

    let mut index = FsIndex::new();
    index.search_dir(FONT_DIR);

    let mut env = Env {
        fonts: index.into_dynamic_loader(),
        resources: ResourceLoader::new(),
    };

    let scope = library::new();
    let state = State::default();

    // Prepare intermediate results and run warm.
    let syntax_tree = parse(COMA).output;
    let layout_tree = eval(&syntax_tree, &mut env, &scope, state.clone()).output;
    let frames = layout(&layout_tree, &mut env);

    // Bench!
    bench!("parse-coma": parse(COMA));
    bench!("eval-coma": eval(&syntax_tree, &mut env, &scope, state.clone()));
    bench!("layout-coma": layout(&layout_tree, &mut env));
    bench!("typeset-coma": typeset(COMA, &mut env, &scope, state.clone()));
    bench!("export-pdf-coma": pdf::export(&frames, &env));
}

criterion_group!(benches, benchmarks);
criterion_main!(benches);