From 3b987e39ccd450e7158e5dde6155cc1e398f7fd6 Mon Sep 17 00:00:00 2001
From: "Jorge C. Leitao" <jorgecarleitao@gmail.com>
Date: Thu, 2 Sep 2021 06:08:02 +0000
Subject: [PATCH] Added benches for bitmap unary.

---
 Cargo.toml            |  4 ++++
 benches/bitmap_ops.rs | 37 +++++++++++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)
 create mode 100644 benches/bitmap_ops.rs

diff --git a/Cargo.toml b/Cargo.toml
index d9abdea1c3a..fae01b13191 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -183,3 +183,7 @@ harness = false
 [[bench]]
 name = "concat"
 harness = false
+
+[[bench]]
+name = "bitmap_ops"
+harness = false
diff --git a/benches/bitmap_ops.rs b/benches/bitmap_ops.rs
new file mode 100644
index 00000000000..a2884799923
--- /dev/null
+++ b/benches/bitmap_ops.rs
@@ -0,0 +1,37 @@
+use arrow2::bitmap::Bitmap;
+
+use criterion::{criterion_group, criterion_main, Criterion};
+
+fn bench_arrow2(lhs: &Bitmap, rhs: &Bitmap) {
+    let r = lhs | rhs;
+    assert!(r.null_count() > 0);
+}
+
+fn add_benchmark(c: &mut Criterion) {
+    (10..=20).step_by(2).for_each(|log2_size| {
+        let size = 2usize.pow(log2_size);
+
+        let bitmap: Bitmap = (0..size).into_iter().map(|x| x % 3 == 0).collect();
+        c.bench_function(&format!("bitmap aligned not 2^{}", log2_size), |b| {
+            b.iter(|| {
+                let r = !&bitmap;
+                assert!(r.null_count() > 0);
+            })
+        });
+        let bitmap1 = bitmap.clone().slice(1, size - 1);
+        c.bench_function(&format!("bitmap not 2^{}", log2_size), |b| {
+            b.iter(|| {
+                let r = !&bitmap1;
+                assert!(r.null_count() > 0);
+            })
+        });
+
+        let bitmap1: Bitmap = (0..size).into_iter().map(|x| x % 4 == 0).collect();
+        c.bench_function(&format!("bitmap aligned or 2^{}", log2_size), |b| {
+            b.iter(|| bench_arrow2(&bitmap, &bitmap1))
+        });
+    });
+}
+
+criterion_group!(benches, add_benchmark);
+criterion_main!(benches);