set max_parallel_workers_per_gather=0;

select setseed(0.42);

\timing
\echo create test table tb_test_bitmaps
create temp table tb_test_bitmaps as
  select id,rb_build_agg((random()*10000000)::int) bitmap
    from generate_series(1,100)id, generate_series(1,100000)b
    group by id;

-- tb_gbs_bitmaps_gap<G>: 10 bitmaps, each with 400000 elements, with an average
-- spacing of G between consecutive elements.
\echo create test tables tb_gbs_bitmaps_gap*
create temp table tb_gbs_bitmaps_gap1 as
  select id, rb_build_agg(n::int) bitmap
    from generate_series(1,10) id, generate_series(1,400000) n
    group by id;

create temp table tb_gbs_bitmaps_gap2 as
  select id, rb_build_agg(v::int) bitmap
  from (
    select id, sum(gap) over (partition by id order by n) as v
    from (
      select id, n, 1 + (random()*(2*2-1))::int as gap
      from generate_series(1,10) id, generate_series(1,400000) n
    ) g
  ) t
  group by id;

create temp table tb_gbs_bitmaps_gap16 as
  select id, rb_build_agg(v::int) bitmap
  from (
    select id, sum(gap) over (partition by id order by n) as v
    from (
      select id, n, 1 + (random()*(2*16-1))::int as gap
      from generate_series(1,10) id, generate_series(1,400000) n
    ) g
  ) t
  group by id;

create temp table tb_gbs_bitmaps_gap64 as
  select id, rb_build_agg(v::int) bitmap
  from (
    select id, sum(gap) over (partition by id order by n) as v
    from (
      select id, n, 1 + (random()*(2*64-1))::int as gap
      from generate_series(1,10) id, generate_series(1,400000) n
    ) g
  ) t
  group by id;

create temp table tb_gbs_bitmaps_gap1024 as
  select id, rb_build_agg(v::int) bitmap
  from (
    select id, sum(gap) over (partition by id order by n) as v
    from (
      select id, n, 1 + (random()*(2*1024-1))::int as gap
      from generate_series(1,10) id, generate_series(1,400000) n
    ) g
  ) t
  group by id;
\timing

select now() time_start \gset

\echo rb_and_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_and(ra1,ra2) from t,generate_series(1,100) id;

\echo rb_and_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_and(ra1,ra2) from t,generate_series(1,1000) id;

\echo rb_or_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_or(ra1,ra2) from t,generate_series(1,100) id;

\echo rb_or_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_or(ra1,ra2) from t,generate_series(1,1000) id;

\echo rb_xor_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_xor(ra1,ra2) from t,generate_series(1,100) id;

\echo rb_xor_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_xor(ra1,ra2) from t,generate_series(1,1000) id;

\echo rb_andnot_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_andnot(ra1,ra2) from t,generate_series(1,100) id;

\echo rb_andnot_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_andnot(ra1,ra2) from t,generate_series(1,1000) id;

\echo rb_add
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_add(ra1,id) from t,generate_series(1,1000)id;

\echo rb_contains_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_contains(ra1,ra2) from t,generate_series(1,100);

\echo rb_contains_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_contains(ra1,ra2) from t,generate_series(1,1000);

\echo rb_contains_3
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_contains(ra1,id) from t,generate_series(1,1000)id;

\echo rb_intersect_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_intersect(ra1,ra2) from t,generate_series(1,100) id;

\echo rb_intersect_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_intersect(ra1,ra2) from t,generate_series(1,1000) id;

\echo rb_equals_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_equals(ra1,ra2) from t,generate_series(1,100) id;

\echo rb_equals_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_equals(ra1,ra2) from t,generate_series(1,1000) id;

\echo rb_andnot_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_andnot(ra1,ra2) from t,generate_series(1,100) id;

\echo rb_andnot_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_andnot(ra1,ra2) from t,generate_series(1,1000) id;

\echo rb_index_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_index(ra1,id) from t,generate_series(1,1000)id;

\echo rb_cardinality_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_cardinality(ra1) from t,generate_series(1,1000);

\echo rb_and_cardinality_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_and_cardinality(ra1,ra2) from t,generate_series(1,100);

\echo rb_and_cardinality_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_and_cardinality(ra1,ra2) from t,generate_series(1,1000);

\echo rb_or_cardinality_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_or_cardinality(ra1,ra2) from t,generate_series(1,100);

\echo rb_or_cardinality_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_or_cardinality(ra1,ra2) from t,generate_series(1,1000);

\echo rb_xor_cardinality_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_xor_cardinality(ra1,ra2) from t,generate_series(1,100);

\echo rb_xor_cardinality_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_xor_cardinality(ra1,ra2) from t,generate_series(1,1000);

\echo rb_andnot_cardinality_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_andnot_cardinality(ra1,ra2) from t,generate_series(1,100);

\echo rb_andnot_cardinality_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_andnot_cardinality(ra1,ra2) from t,generate_series(1,1000);

\echo rb_is_empty_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_is_empty(ra1) from t,generate_series(1,1000);

\echo rb_range_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_range(ra1,id,2000000) from t,generate_series(1,100)id;

\echo rb_range_cardinality_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_range_cardinality(ra1,id,2000000) from t,generate_series(1,1000)id;

\echo rb_min_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_min(ra1) from t,generate_series(1,1000);

\echo rb_max_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_max(ra1) from t,generate_series(1,1000);

\echo rb_rank_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_rank(ra1,100000+id) from t,generate_series(1,1000)id;

\echo rb_jaccard_dist_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,(select bitmap from tb_test_bitmaps where id=2 limit 1) ra2
)
select rb_jaccard_dist(ra1,ra2) from t,generate_series(1,100);

\echo rb_jaccard_dist_2
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1,rb_fill('{}'::roaringbitmap,5000001,15000000) ra2
)
select rb_jaccard_dist(ra1,ra2) from t,generate_series(1,1000);

\echo rb_to_array_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_to_array(ra1) from t,generate_series(1,100);

\echo rb_iterate_1
explain analyze
with t as(
  select (select bitmap from tb_test_bitmaps where id=1 limit 1) ra1
)
select rb_iterate(ra1) from t,generate_series(1,10);

\echo rb_build_agg_1
explain analyze
select rb_build_agg(id) from generate_series(1,1000000) id;

\echo rb_or_agg_1
explain analyze
select rb_or_agg(bitmap) from tb_test_bitmaps;

\echo rb_and_agg_1
explain analyze
select rb_and_agg(bitmap) from tb_test_bitmaps;

\echo rb_xor_agg_1
explain analyze
select rb_xor_agg(bitmap) from tb_test_bitmaps;

\echo rb_or_cardinality_agg_1
explain analyze
select rb_or_cardinality_agg(bitmap) from tb_test_bitmaps;

\echo rb_and_cardinality_agg_1
explain analyze
select rb_and_cardinality_agg(bitmap) from tb_test_bitmaps;

\echo rb_xor_cardinality_agg_1
explain analyze
select rb_xor_cardinality_agg(bitmap) from tb_test_bitmaps;

\echo rb_group_elements_by_source_gap1
explain analyze
select count(*) from rb_group_elements_by_source(
  (select array_agg(bitmap order by id) from tb_gbs_bitmaps_gap1)
);

\echo rb_group_elements_by_source_gap2
explain analyze
select count(*) from rb_group_elements_by_source(
  (select array_agg(bitmap order by id) from tb_gbs_bitmaps_gap2)
);

\echo rb_group_elements_by_source_gap16
explain analyze
select count(*) from rb_group_elements_by_source(
  (select array_agg(bitmap order by id) from tb_gbs_bitmaps_gap16)
);

\echo rb_group_elements_by_source_gap64
explain analyze
select count(*) from rb_group_elements_by_source(
  (select array_agg(bitmap order by id) from tb_gbs_bitmaps_gap64)
);

\echo rb_group_elements_by_source_gap1024
explain analyze
select count(*) from rb_group_elements_by_source(
  (select array_agg(bitmap order by id) from tb_gbs_bitmaps_gap1024)
);

select now() - :'time_start'::timestamp as time_escape \gset

\echo  Total time: :time_escape