use bitgrep::common::SourceFile;
use bitgrep::printers::output::SimpleOutput;
use bitgrep::printers::simple_printer::SimplePrinter;
use bitgrep::scanner::Scanner;
use bitgrep::workers::native_processor::NativeProcessor;
use bitgrep::{common::Endianness, filters::configuration::Configuration};
use std::fs::File;
use std::io::{self};
use std::path::{Path, PathBuf};

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

fn run_scanner(file_path: &PathBuf, configuration: &Configuration<f64>) {
    let filter = configuration.create_filter();
    let processor = NativeProcessor::new(Endianness::Little);
    let file = SourceFile::new(file_path.clone(), File::open(file_path).unwrap());

    // Suppress output, this does not test buffered writing but also doesn't spam output
    let io_writer = io::empty();
    let printer = SimplePrinter::new(SimpleOutput::new(), io_writer);

    let scanner = Scanner::new(file, Box::new(processor), filter.unwrap(), printer);
    scanner.scan().expect("should complete successfuly");
}

fn scanner_random_8k_minmax_benchmark(c: &mut Criterion) {
    const FILE_NAME: &str = "random.dat";

    let this_directory = Path::new(file!()).parent().unwrap();
    let path = this_directory.join(FILE_NAME);

    let configuration = Configuration {
        minimum: Some(30.1000),
        maximum: Some(35.12345),
        ..Default::default()
    };

    c.bench_function(
        format!("scanner.scan() minmax 8k random file [{FILE_NAME}]").as_str(),
        |b| {
            b.iter(|| run_scanner(&path, &configuration));
        },
    );
}

fn scanner_random_8k_literal_benchmark(c: &mut Criterion) {
    const FILE_NAME: &str = "random.dat";

    let this_directory = Path::new(file!()).parent().unwrap();
    let path = this_directory.join(FILE_NAME);

    let configuration = Configuration {
        literal: Some(33.248462071692536),
        ..Default::default()
    };

    c.bench_function(
        format!("scanner.scan() literal 8k random file [{FILE_NAME}]").as_str(),
        |b| {
            b.iter(|| run_scanner(&path, &configuration));
        },
    );
}

criterion_group!(
    benches,
    scanner_random_8k_minmax_benchmark,
    scanner_random_8k_literal_benchmark
);
criterion_main!(benches);