pkbktarr: change cb signatures to make use of pktmpln

1 files changed, 82 insertions, 0 deletions

diff --git a/test/pkbktarr.cpp b/test/pkbktarr.cpp
index 8d691eb..2069c8b 100644
--- a/test/pkbktarr.cpp
+++ b/test/pkbktarr.cpp
@@ -1,5 +1,6 @@
 
 #include "../pkmem.h"
+#include "../pktmpln.h"
 #define PK_IMPL_MEM
 #define PK_IMPL_BKTARR
 #include "../pkbktarr.h"
@@ -63,4 +64,85 @@ int main(int argc, char *argv[])
 		if (bkt_data->allocs != 0) exit(1);
 	}
 	test_teardown(&bkt_buckets, &bkt_data);
+
+	// test typed find first index
+	test_spinup(&bkt_buckets, &bkt_data);
+	{
+		assert(bkt_buckets != nullptr);
+		assert(bkt_data != nullptr);
+		struct pk_bkt_arr_handle limits;
+		limits.b = PK_BKT_ARR_HANDLE_B_MAX;
+		limits.i = PK_BKT_ARR_HANDLE_I_MAX;
+		struct pk_bkt_arr_t<int> arr(limits, bkt_buckets, bkt_data);
+
+		int val = 128;
+		struct pk_bkt_arr_handle h = pk_bkt_arr_new_handle(&arr);
+		arr[h] = val;
+		if (h.b != 0) exit(1);
+		if (h.i != 0) exit(1);
+		if (arr.head_l.b != 0) exit(1);
+		if (arr.head_l.i != 1) exit(1);
+		if (arr.head_r.b != 0) exit(1);
+		if (arr.head_r.i != 1) exit(1);
+		if (arr.idx_unused[0] != 0xFFFFFFFFFFFFFFFE) exit(1);
+		if (bkt_buckets->allocs != 2) exit(1);
+		if (bkt_data->allocs != 1) exit(1);
+		if (arr[h] != val) exit(1);
+
+		using CompCbWrapper = pk_tmpln_2<bool, const int*, const int*, const void*, const void*>;
+		CompCbWrapper cb_wrapper{};
+		cb_wrapper.func = [](const int *lhs, const int *rhs)
+		{
+			return *lhs == *rhs;
+		};
+		pk_bkt_arr_handle found_h = pk_bkt_arr_find_first_handle(&arr, &CompCbWrapper::invoke, &cb_wrapper, &val);
+		if (pk_bkt_arr_handle_compare(found_h, h) != 0) exit(1);
+
+		arr.~pk_bkt_arr_t<int>();
+		if (bkt_buckets->allocs != 0) exit(1);
+		if (bkt_data->allocs != 0) exit(1);
+	}
+	test_teardown(&bkt_buckets, &bkt_data);
+
+	// test iter
+	test_spinup(&bkt_buckets, &bkt_data);
+	{
+		assert(bkt_buckets != nullptr);
+		assert(bkt_data != nullptr);
+		struct pk_bkt_arr_handle limits;
+		limits.b = PK_BKT_ARR_HANDLE_B_MAX;
+		limits.i = PK_BKT_ARR_HANDLE_I_MAX;
+		struct pk_bkt_arr_t<int> arr(limits, bkt_buckets, bkt_data);
+
+		int val = 128;
+		struct pk_bkt_arr_handle h = pk_bkt_arr_new_handle(&arr);
+		arr[h] = val;
+		if (h.b != 0) exit(1);
+		if (h.i != 0) exit(1);
+		if (arr.head_l.b != 0) exit(1);
+		if (arr.head_l.i != 1) exit(1);
+		if (arr.head_r.b != 0) exit(1);
+		if (arr.head_r.i != 1) exit(1);
+		if (arr.idx_unused[0] != 0xFFFFFFFFFFFFFFFE) exit(1);
+		if (bkt_buckets->allocs != 2) exit(1);
+		if (bkt_data->allocs != 1) exit(1);
+		if (arr[h] != val) exit(1);
+
+		int some_counter = 0;
+		using IterCbWrapper = pk_tmpln_1<void, int*, void*>;
+		IterCbWrapper cb_wrapper{};
+		cb_wrapper.func = [&some_counter](int *lhs)
+		{
+			(void)lhs;
+			some_counter += 1;
+			return;
+		};
+		pk_bkt_arr_iterate(&arr, &IterCbWrapper::invoke, &cb_wrapper);
+		if (some_counter != 1) exit(1);
+
+		arr.~pk_bkt_arr_t<int>();
+		if (bkt_buckets->allocs != 0) exit(1);
+		if (bkt_data->allocs != 0) exit(1);
+	}
+	test_teardown(&bkt_buckets, &bkt_data);
 }