use std::hint::black_box;

use criterion::{Criterion, criterion_group, criterion_main};
use math_parser::{ast, context::EvalContext};

macro_rules! generate_benchmarks {
	($( $input:expr_2021 ),* $(,)?) => {
		fn parsing_bench(c: &mut Criterion) {
			$(
				c.bench_function(concat!("parse ", $input), |b| {
					b.iter(|| {
						let _ = black_box(ast::Node::try_parse_from_str($input));
					});
				});
			)*
		}

		fn evaluation_bench(c: &mut Criterion) {
			$(
				let expr = match ast::Node::try_parse_from_str($input) {
					Ok(expr) => expr,
					Err(err) => panic!("failed to parse `{}`: {err}", $input),
				};
				let context = EvalContext::default();

				c.bench_function(concat!("eval ", $input), |b| {
					b.iter(|| {
						let _ = black_box(expr.eval(&context));
					});
				});
			)*
		}

		criterion_group!(benches, parsing_bench, evaluation_bench);
		criterion_main!(benches);
	};
}

generate_benchmarks! {
	"(3 * (4 + sqrt(25)) - cos(pi/3) * (2^3)) + 5 * e", // Mixed nested functions, constants, and operations
	"((5 + 2 * (3 - sqrt(49)))^2) / (1 + sqrt(16)) + tau / 2", // Complex nested expression with constants
	"log(100, 10) + (5 * sin(pi/4) + sqrt(81)) / (2 * phi)", // Logarithmic and trigonometric functions
	"(sqrt(144) * 2 + 5) / (3 * (4 - sin(pi / 6))) + e^2", // Combined square root, trigonometric, and exponential operations
	"cos(2 * pi) + tan(pi / 3) * log(32, 2) - sqrt(256)", // Multiple trigonometric and logarithmic functions
	"(10 * (3 + 2) - 8 / 2)^2 + 7 * (2^4) - sqrt(225) + phi", // Mixed arithmetic with constants
	"(5^2 + 3^3) * (sqrt(81) + sqrt(64)) - tau * log(1000, 10)", // Power and square root with constants
	"((8 * sqrt(49) - 2 * e) + log(256, 2) / (2 + cos(pi))) * 1.5", // Nested functions and constants
	"(tan(pi / 4) + 5) * (3 + sqrt(36)) / (log(1024, 2) - 4)", // Nested functions with trigonometry and logarithm
	"((3 * e + 2 * sqrt(100)) - cos(tau / 4)) * log(27, 3) + phi", // Mixed constant usage and functions
	"(sqrt(100) + 5 * sin(pi / 6) - 8 / log(64, 2)) + e^(1.5)", // Complex mix of square root, division, and exponentiation
	"((sin(pi/2) + cos(0)) * (e^2 - 2 * sqrt(16))) / (log(100, 10) + pi)", // Nested trigonometric, exponential, and logarithmic functions
	"(5 * (7 + sqrt(121)) - (log(243, 3) * phi)) + 3^5 / tau", //
}
