// SPDX-License-Identifier: MIT
// Copyright (c) "2023" . The DeepCausality Authors. All Rights Reserved.

use criterion::{criterion_group, Criterion};

use deep_causality::prelude::*;

use crate::benchmarks::utils_linear_graph;

fn small_multi_layer_graph_benchmark(criterion: &mut Criterion) {
    let (g, data) = utils_linear_graph::get_small_multi_cause_graph_and_data();

    criterion.bench_function("small_multi_layer_graph_reason_all_causes", |bencher| {
        bencher.iter(|| g.reason_all_causes(&data, None).unwrap())
    });

    let index = data.len() / 2;

    criterion.bench_function(
        "small_multi_layer_graph_reason_subgraph_from_cause",
        |bencher| bencher.iter(|| g.reason_subgraph_from_cause(index, &data, None).unwrap()),
    );

    let start_index = 0;
    let stop_index = 1;

    criterion.bench_function(
        "small_multi_layer_graph_reason_shortest_path_between_causes",
        |bencher| {
            bencher.iter(|| {
                g.reason_shortest_path_between_causes(start_index, stop_index, &data, None)
                    .unwrap()
            })
        },
    );

    let obs = 0.99;

    criterion.bench_function("small_multi_layer_graph_reason_single_cause", |bencher| {
        bencher.iter(|| g.reason_single_cause(index, &[obs]).unwrap())
    });
}

criterion_group! {
    name = multi_layer_graph;
    config = Criterion::default().sample_size(100);
    targets =
    small_multi_layer_graph_benchmark,
}