use criterion::{black_box, criterion_group, criterion_main, Criterion};
use geo_filters::config::VariableConfig;
use geo_filters::diff_count::{GeoDiffCount, GeoDiffCount13};
use geo_filters::distinct_count::GeoDistinctCount13;
use geo_filters::evaluation::hll::Hll14;
use geo_filters::Count;

fn criterion_benchmark(c: &mut Criterion) {
    let sizes = [100, 1000, 10000];

    for size in &sizes {
        let mut group = c.benchmark_group(format!("insert:{size}"));

        group.bench_function("geo_diff_count_13", |b| {
            b.iter(|| {
                let mut gc = GeoDiffCount13::default();
                for i in 0..*size {
                    gc.push(i);
                }
            })
        });
        group.bench_function("geo_diff_count_var_13", |b| {
            let c = VariableConfig::<_, u32>::new(13, 7680, 256);
            b.iter(move || {
                let mut gc = GeoDiffCount::new(c.clone());
                for i in 0..*size {
                    gc.push(i);
                }
            })
        });
        group.bench_function("geo_distinct_count_13", |b| {
            b.iter(|| {
                let mut gc = GeoDistinctCount13::default();
                for i in 0..*size {
                    gc.push(i);
                }
            })
        });
        group.bench_function("hll_14", |b| {
            b.iter(|| {
                let mut hll = Hll14::default();
                for i in 0..*size {
                    hll.push(i);
                }
            })
        });
    }

    for size in &sizes {
        let mut group = c.benchmark_group(format!("estimate:{size}"));

        group.bench_function("geo_diff_count_13", |b| {
            b.iter(|| {
                let mut gc = GeoDiffCount13::default();
                for i in 0..*size {
                    gc.push(i);
                    black_box(gc.size());
                }
            })
        });
        group.bench_function("geo_diff_count_var_13", |b| {
            let c = VariableConfig::<_, u32>::new(13, 7680, 256);
            b.iter(move || {
                let mut gc = GeoDiffCount::new(c.clone());
                for i in 0..*size {
                    gc.push(i);
                    black_box(gc.size());
                }
            })
        });
        group.bench_function("geo_distinct_count_13", |b| {
            b.iter(|| {
                let mut gc = GeoDistinctCount13::default();
                for i in 0..*size {
                    gc.push(i);
                    black_box(gc.size());
                }
            })
        });
        group.bench_function("hll_14", |b| {
            b.iter(|| {
                let mut hll = Hll14::default();
                for i in 0..*size {
                    hll.push(i);
                    black_box(hll.size());
                }
            })
        });
    }

    for size in &sizes {
        let mut group = c.benchmark_group(format!("estimate_with:{size}"));
        let size = size / 2; // so that the combined set will be the original size

        group.bench_function("geo_diff_count_13", |b| {
            b.iter(|| {
                let mut gc1 = GeoDiffCount13::default();
                let mut gc2 = GeoDiffCount13::default();
                for i in 0..size {
                    gc1.push(i * 2);
                    gc2.push(i * 2 + 1);
                    black_box(gc1.size_with_sketch(&gc2));
                }
            })
        });
        group.bench_function("geo_diff_count_var_13", |b| {
            let c = VariableConfig::<_, u32>::new(13, 7680, 256);
            b.iter(move || {
                let mut gc1 = GeoDiffCount::new(c.clone());
                let mut gc2 = GeoDiffCount::new(c.clone());
                for i in 0..size {
                    gc1.push(i * 2);
                    gc2.push(i * 2 + 1);
                    black_box(gc1.size_with_sketch(&gc2));
                }
            })
        });
        group.bench_function("geo_distinct_count_13", |b| {
            b.iter(|| {
                let mut gc1 = GeoDistinctCount13::default();
                let mut gc2 = GeoDistinctCount13::default();
                for i in 0..size {
                    gc1.push(i * 2);
                    gc2.push(i * 2 + 1);
                    black_box(gc1.size_with_sketch(&gc2));
                }
            })
        });
    }
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);