use criterion::{criterion_group, criterion_main, Criterion};

fn build_deep_hierarchy(taffy: &mut taffy::node::Taffy) -> taffy::node::Node {
    let node111 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();
    let node112 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();

    let node121 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();
    let node122 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();

    let node11 =
        taffy.new_with_children(taffy::style::FlexboxLayout { ..Default::default() }, &[node111, node112]).unwrap();
    let node12 =
        taffy.new_with_children(taffy::style::FlexboxLayout { ..Default::default() }, &[node121, node122]).unwrap();
    let node1 =
        taffy.new_with_children(taffy::style::FlexboxLayout { ..Default::default() }, &[node11, node12]).unwrap();

    let node211 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();
    let node212 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();

    let node221 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();
    let node222 = taffy
        .new_leaf(taffy::style::FlexboxLayout {
            size: taffy::geometry::Size {
                width: taffy::style::Dimension::Points(10.0),
                height: taffy::style::Dimension::Points(10.0),
            },
            ..Default::default()
        })
        .unwrap();

    let node21 =
        taffy.new_with_children(taffy::style::FlexboxLayout { ..Default::default() }, &[node211, node212]).unwrap();
    let node22 =
        taffy.new_with_children(taffy::style::FlexboxLayout { ..Default::default() }, &[node221, node222]).unwrap();

    let node2 =
        taffy.new_with_children(taffy::style::FlexboxLayout { ..Default::default() }, &[node21, node22]).unwrap();

    taffy.new_with_children(taffy::style::FlexboxLayout { ..Default::default() }, &[node1, node2]).unwrap()
}

fn taffy_benchmarks(c: &mut Criterion) {
    // Increase sample size, because the examples are very small
    let mut group = c.benchmark_group("deep hierarchy");
    group.sample_size(200);

    group.bench_function("build", |b| {
        b.iter(|| {
            let mut taffy = taffy::node::Taffy::new();
            build_deep_hierarchy(&mut taffy);
        })
    });

    group.bench_function("single", |b| {
        b.iter(|| {
            let mut taffy = taffy::node::Taffy::new();
            let root = build_deep_hierarchy(&mut taffy);
            taffy.compute_layout(root, taffy::geometry::Size::NONE).unwrap()
        })
    });

    group.bench_function("relayout", |b| {
        let mut taffy = taffy::node::Taffy::new();
        let root = build_deep_hierarchy(&mut taffy);

        b.iter(|| {
            taffy.mark_dirty(root).unwrap();
            taffy.compute_layout(root, taffy::geometry::Size::NONE).unwrap()
        })
    });
}

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