use criterion::{criterion_group, criterion_main, Criterion};
use spacey::{interpreter::interpreter::InterpretError, Interpreter, InterpreterConfig};
use std::error::Error;

pub fn single_instruction_with_param_benchmark(c: &mut Criterion) {
    let config = InterpreterConfig::default_no_heap_suppressed("ws/push_stack.ws");
    let mut interpreter = Interpreter::new(config).unwrap();
    c.bench_function("single instruction with param", |b| {
        b.iter(|| -> Result<(), InterpretError> {
            interpreter.run()?;

            Ok(())
        })
    });
}

pub fn reset_vm_benchmark(c: &mut Criterion) {
    let config = InterpreterConfig::default_no_heap_suppressed("ws/hello_world.ws");
    let mut interpreter = Interpreter::new(config).unwrap();
    c.bench_function("reset vm", |b| {
        b.iter(|| -> Result<(), InterpretError> {
            interpreter.reset();

            Ok(())
        })
    });
}

pub fn hello_world_benchmark(c: &mut Criterion) {
    let config = InterpreterConfig::default_no_heap_suppressed("ws/hello_world.ws");
    let mut interpreter = Interpreter::new(config).unwrap();
    c.bench_function("hello world", |b| {
        b.iter(|| -> Result<(), InterpretError> {
            interpreter.run()?;
            interpreter.reset();

            Ok(())
        })
    });
}

pub fn hello_world_of_spaces_benchmark(c: &mut Criterion) {
    let config = InterpreterConfig::default_no_heap_suppressed("ws/hello_world_of_spaces.ws");
    let mut interpreter = Interpreter::new(config).unwrap();
    c.bench_function("hello world of spaces", |b| {
        b.iter(|| -> Result<(), InterpretError> {
            interpreter.run()?;
            interpreter.reset();

            Ok(())
        })
    });
}

pub fn sieve_benchmark(c: &mut Criterion) {
    let config = InterpreterConfig::default_no_heap_suppressed("ws/sieve.ws");
    let mut interpreter = Interpreter::new(config).unwrap();
    c.bench_function("sieve", |b| {
        b.iter(|| -> Result<(), InterpretError> {
            interpreter.run()?;
            interpreter.reset();

            Ok(())
        })
    });
}

pub fn count_benchmark(c: &mut Criterion) {
    let config = InterpreterConfig::default_no_heap_suppressed("ws/count.ws");
    let mut interpreter = Interpreter::new(config).unwrap();
    c.bench_function("count", |b| {
        b.iter(|| -> Result<(), InterpretError> {
            interpreter.run()?;
            interpreter.reset();

            Ok(())
        })
    });
}

pub fn primes_benchmark(c: &mut Criterion) {
    let config = InterpreterConfig::default_no_heap_suppressed("ws/primes.ws");
    let mut interpreter = Interpreter::new(config).unwrap();
    c.bench_function("primes", |b| {
        b.iter(|| -> Result<(), InterpretError> {
            interpreter.run()?;
            interpreter.reset();

            Ok(())
        })
    });
}

criterion_group!(
    interpreter,
    count_benchmark,
    hello_world_benchmark,
    hello_world_of_spaces_benchmark,
    sieve_benchmark,
    reset_vm_benchmark,
    single_instruction_with_param_benchmark,
    primes_benchmark
);
criterion_main!(interpreter);