use std::fmt::Debug;
use std::ops::Div;

use const_decimal::{Decimal, ScaledInteger};
use criterion::measurement::WallTime;
use criterion::{black_box, BatchSize, BenchmarkGroup};
use num_traits::{ConstOne, ConstZero, PrimInt};
use prop::strategy::ValueTree;
use prop::test_runner::TestRunner;
use proptest::prelude::*;

pub fn bench_all<const D: u8, I>(
    group: &mut BenchmarkGroup<'_, WallTime>,
    lo_strategy: impl Strategy<Value = I> + Clone,
    hi_strategy: impl Strategy<Value = I> + Clone,
) where
    I: ScaledInteger<D> + Debug + Div<Output = I>,
{
    primitive_div::<I>(group, lo_strategy.clone(), "lo");
    decimal_div::<D, I>(group, lo_strategy, "lo");
    primitive_div::<I>(group, hi_strategy.clone(), "hi");
    decimal_div::<D, I>(group, hi_strategy, "hi");
}

fn primitive_div<I>(
    group: &mut BenchmarkGroup<'_, WallTime>,
    strategy: impl Strategy<Value = I> + Clone,
    strategy_label: &str,
) where
    I: PrimInt + ConstZero + ConstOne,
{
    // Use proptest to generate arbitrary input values.
    let mut runner = TestRunner::deterministic();
    let input = (strategy.clone(), strategy);

    group.bench_function(&format!("primitive/div/{strategy_label}"), |bencher| {
        bencher.iter_batched(
            || {
                let (numer, denom) = input.new_tree(&mut runner).unwrap().current();
                // Avoid division by zero.
                let denom = match denom == I::ZERO {
                    true => I::ONE,
                    false => denom,
                };

                (numer, denom)
            },
            |(a, b)| black_box(black_box(a) / black_box(b)),
            BatchSize::SmallInput,
        )
    });
}

fn decimal_div<const D: u8, I>(
    group: &mut BenchmarkGroup<'_, WallTime>,
    strategy: impl Strategy<Value = I> + Clone,
    strategy_label: &str,
) where
    I: ScaledInteger<D> + Debug,
{
    // Use proptest to generate arbitrary input values.
    let mut runner = TestRunner::deterministic();
    let input =
        (strategy.clone(), strategy).prop_map(|(a, b)| (Decimal::<_, D>(a), Decimal::<_, D>(b)));

    group.bench_function(format!("decimal/div/{strategy_label}"), |bencher| {
        bencher.iter_batched(
            || {
                let (numer, denom) = input.new_tree(&mut runner).unwrap().current();
                // Avoid division by zero.
                let denom = match denom == Decimal::ZERO {
                    true => Decimal::ONE,
                    false => denom,
                };

                (numer, denom)
            },
            |(a, b)| black_box(black_box(a) / black_box(b)),
            BatchSize::SmallInput,
        )
    });
}