add type cast for roaringbitmap and roaringbitmap64

Author: ChenHuajun

expected/roaringbitmap64.out

@@ -182,11 +182,39 @@ select roaringbitmap64('{1,9999,-88888}'::roaringbitmap64::bytea);
  {1,9999,-88888}
 (1 row)
 
+select roaringbitmap('{}')::roaringbitmap64;
+ roaringbitmap 
+---------------
+ {}
+(1 row)
+
+select roaringbitmap('{0,1,9999,2147483647,-2147483648,-2,-1}')::roaringbitmap64;
+              roaringbitmap              
+-----------------------------------------
+ {0,1,9999,2147483647,-2147483648,-2,-1}
+(1 row)
+
+select roaringbitmap64('{}')::roaringbitmap;
+ roaringbitmap64 
+-----------------
+ {}
+(1 row)
+
+select roaringbitmap64('{0,1,9999,2147483647,-2147483648,-2,-1}')::roaringbitmap;
+             roaringbitmap64             
+-----------------------------------------
+ {0,1,9999,2147483647,-2147483648,-2,-1}
+(1 row)
+
 -- Exception
 select roaringbitmap64('\x11'::bytea);
 ERROR:  bitmap format is error
 select '\x11'::bytea::roaringbitmap64;
 ERROR:  bitmap format is error
+select roaringbitmap64('{0,1,9999,2147483648,-2147483648,-2,-1}')::roaringbitmap;
+ERROR:  value "2147483648" is out of range for type integer
+select roaringbitmap64('{0,1,9999,2147483647,-2147483649,-2,-1}')::roaringbitmap;
+ERROR:  value "-2147483649" is out of range for type integer
 -- Test Opperator
 select roaringbitmap64('{}') & roaringbitmap64('{}');
  ?column? 

roaringbitmap--0.5--1.0.sql

@@ -54,6 +54,22 @@ CREATE CAST (roaringbitmap64 AS bytea) WITHOUT FUNCTION;
 
 CREATE CAST (bytea AS roaringbitmap64) WITH FUNCTION roaringbitmap64(bytea);
 
+--
+-- type cast for roaringbitmap
+--
+CREATE OR REPLACE FUNCTION rb64_to_roaringbitmap(roaringbitmap64)
+  RETURNS roaringbitmap
+  AS 'MODULE_PATHNAME', 'rb64_to_roaringbitmap'
+  LANGUAGE C STRICT IMMUTABLE PARALLEL SAFE;
+
+CREATE OR REPLACE FUNCTION rb64_from_roaringbitmap(roaringbitmap)
+  RETURNS roaringbitmap64
+  AS 'MODULE_PATHNAME', 'rb64_from_roaringbitmap'
+  LANGUAGE C STRICT IMMUTABLE PARALLEL SAFE;
+
+CREATE CAST (roaringbitmap64 AS roaringbitmap) WITH FUNCTION rb64_to_roaringbitmap(roaringbitmap64);
+CREATE CAST (roaringbitmap AS roaringbitmap64) WITH FUNCTION rb64_from_roaringbitmap(roaringbitmap);
+
 --
 -- Operator Functions
 --

roaringbitmap--1.0.sql

@@ -46,7 +46,6 @@ CREATE OR REPLACE FUNCTION roaringbitmap(bytea)
   LANGUAGE C STRICT IMMUTABLE PARALLEL SAFE;
 
 CREATE CAST (roaringbitmap AS bytea) WITHOUT FUNCTION;
-
 CREATE CAST (bytea AS roaringbitmap) WITH FUNCTION roaringbitmap(bytea);
 
 --
@@ -537,9 +536,24 @@ CREATE OR REPLACE FUNCTION roaringbitmap64(bytea)
   LANGUAGE C STRICT IMMUTABLE PARALLEL SAFE;
 
 CREATE CAST (roaringbitmap64 AS bytea) WITHOUT FUNCTION;
-
 CREATE CAST (bytea AS roaringbitmap64) WITH FUNCTION roaringbitmap64(bytea);
 
+--
+-- type cast for roaringbitmap
+--
+CREATE OR REPLACE FUNCTION rb64_to_roaringbitmap(roaringbitmap64)
+  RETURNS roaringbitmap
+  AS 'MODULE_PATHNAME', 'rb64_to_roaringbitmap'
+  LANGUAGE C STRICT IMMUTABLE PARALLEL SAFE;
+
+CREATE OR REPLACE FUNCTION rb64_from_roaringbitmap(roaringbitmap)
+  RETURNS roaringbitmap64
+  AS 'MODULE_PATHNAME', 'rb64_from_roaringbitmap'
+  LANGUAGE C STRICT IMMUTABLE PARALLEL SAFE;
+
+CREATE CAST (roaringbitmap64 AS roaringbitmap) WITH FUNCTION rb64_to_roaringbitmap(roaringbitmap64);
+CREATE CAST (roaringbitmap AS roaringbitmap64) WITH FUNCTION rb64_from_roaringbitmap(roaringbitmap);
+
 --
 -- Operator Functions
 --

roaringbitmap64.c

@@ -1528,6 +1528,88 @@ rb64_iterate(PG_FUNCTION_ARGS) {
     }
 }
 
+//convert roaringbitmap64 to roaringbitmap
+PG_FUNCTION_INFO_V1(rb64_to_roaringbitmap);
+Datum rb64_to_roaringbitmap(PG_FUNCTION_ARGS);
+
+Datum
+rb64_to_roaringbitmap(PG_FUNCTION_ARGS)
+{
+    bytea *serializedbytes = PG_GETARG_BYTEA_P(0);
+    roaring64_bitmap_t *r1;
+    roaring64_iterator_t *iterator;
+    roaring_bitmap_t *r2;
+    size_t expectedsize;
+
+    r1 = roaring64_bitmap_portable_deserialize_safe(VARDATA(serializedbytes), VARSIZE(serializedbytes) - VARHDRSZ);
+    if (!r1)
+        ereport(ERROR,
+                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
+                 errmsg("bitmap format is error")));
+
+    r2 = roaring_bitmap_create();
+    iterator = roaring64_iterator_create(r1);
+    while (roaring64_iterator_has_value(iterator))
+    {
+        int64_t value = roaring64_iterator_value(iterator);
+        if(value > INT32_MAX || value < INT32_MIN)
+            ereport(ERROR,
+                (errcode(ERRCODE_NUMERIC_VALUE_OUT_OF_RANGE),
+                errmsg("value \"%ld\" is out of range for type %s", value,
+                    "integer")));
+        roaring_bitmap_add(r2, (int32_t)value);
+        roaring64_iterator_advance(iterator);
+    }
+    roaring64_iterator_free(iterator);
+
+    expectedsize = roaring_bitmap_portable_size_in_bytes(r2);
+    serializedbytes = (bytea *) palloc(VARHDRSZ + expectedsize);
+    roaring_bitmap_portable_serialize(r2, VARDATA(serializedbytes));
+    roaring64_bitmap_free(r1);
+    roaring_bitmap_free(r2);
+
+    SET_VARSIZE(serializedbytes, VARHDRSZ + expectedsize);
+    PG_RETURN_BYTEA_P(serializedbytes);
+}
+
+//convert roaringbitmap to roaringbitmap64
+PG_FUNCTION_INFO_V1(rb64_from_roaringbitmap);
+Datum rb64_from_roaringbitmap(PG_FUNCTION_ARGS);
+
+Datum
+rb64_from_roaringbitmap(PG_FUNCTION_ARGS)
+{
+    bytea *serializedbytes = PG_GETARG_BYTEA_P(0);
+    roaring_bitmap_t *r1;
+    roaring_uint32_iterator_t *iterator;
+    roaring64_bitmap_t *r2;
+    size_t expectedsize;
+
+    r1 = roaring_bitmap_portable_deserialize_safe(VARDATA(serializedbytes), VARSIZE(serializedbytes) - VARHDRSZ);
+    if (!r1)
+        ereport(ERROR,
+                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
+                 errmsg("bitmap format is error")));
+
+    r2 = roaring64_bitmap_create();
+    iterator = roaring_iterator_create(r1);
+    while (iterator->has_value)
+    {
+        roaring64_bitmap_add(r2, (int64_t)(int32_t)iterator->current_value);
+        roaring_uint32_iterator_advance(iterator);
+    }
+    roaring_uint32_iterator_free(iterator);
+
+    expectedsize = roaring64_bitmap_portable_size_in_bytes(r2);
+    serializedbytes = (bytea *) palloc(VARHDRSZ + expectedsize);
+    roaring64_bitmap_portable_serialize(r2, VARDATA(serializedbytes));
+    roaring_bitmap_free(r1);
+    roaring64_bitmap_free(r2);
+
+    SET_VARSIZE(serializedbytes, VARHDRSZ + expectedsize);
+    PG_RETURN_BYTEA_P(serializedbytes);
+}
+
 //bitmap or trans
 PG_FUNCTION_INFO_V1(rb64_or_trans);
 Datum rb64_or_trans(PG_FUNCTION_ARGS);

sql/roaringbitmap64.sql

@@ -52,9 +52,16 @@ select '{1}'::roaringbitmap64::bytea::roaringbitmap64;
 select '{1,9999,-88888}'::roaringbitmap64::bytea::roaringbitmap64;
 select roaringbitmap64('{1,9999,-88888}'::roaringbitmap64::bytea);
 
+select roaringbitmap('{}')::roaringbitmap64;
+select roaringbitmap('{0,1,9999,2147483647,-2147483648,-2,-1}')::roaringbitmap64;
+select roaringbitmap64('{}')::roaringbitmap;
+select roaringbitmap64('{0,1,9999,2147483647,-2147483648,-2,-1}')::roaringbitmap;
+
 -- Exception
 select roaringbitmap64('\x11'::bytea);
 select '\x11'::bytea::roaringbitmap64;
+select roaringbitmap64('{0,1,9999,2147483648,-2147483648,-2,-1}')::roaringbitmap;
+select roaringbitmap64('{0,1,9999,2147483647,-2147483649,-2,-1}')::roaringbitmap;
 
 -- Test Opperator