#include "postgres.h"
#include <stddef.h>
#include "catalog/pg_type.h"
#include "common/pg_prng.h"
#include "fmgr.h"
#include "miscadmin.h"
#include "nodes/bitmapset.h"
#include "nodes/nodes.h"
#include "nodes/pg_list.h"
#include "utils/array.h"
#include "utils/builtins.h"
#include "utils/timestamp.h"

Include dependency graph for test_bitmapset.c:

Macros
#define	EXPECT_TRUE(expr)

#define	EXPECT_NOT_NULL(expr)

#define	BITMAPSET_TO_TEXT(bms) cstring_to_text(nodeToString(bms))

#define	TEXT_TO_BITMAPSET(str) ((Bitmapset *) stringToNode(text_to_cstring(str)))

#define	PG_ARG_GETBITMAPSET(n) (PG_ARGISNULL(n) ? NULL : TEXT_TO_BITMAPSET(PG_GETARG_TEXT_PP(n)))

#define	PG_RETURN_BITMAPSET_AS_TEXT(bms) PG_RETURN_TEXT_P(BITMAPSET_TO_TEXT(bms))

Functions
	PG_FUNCTION_INFO_V1 (test_bms_make_singleton)

	PG_FUNCTION_INFO_V1 (test_bms_add_member)

	PG_FUNCTION_INFO_V1 (test_bms_del_member)

	PG_FUNCTION_INFO_V1 (test_bms_is_member)

	PG_FUNCTION_INFO_V1 (test_bms_num_members)

	PG_FUNCTION_INFO_V1 (test_bms_copy)

	PG_FUNCTION_INFO_V1 (test_bms_equal)

	PG_FUNCTION_INFO_V1 (test_bms_compare)

	PG_FUNCTION_INFO_V1 (test_bms_is_subset)

	PG_FUNCTION_INFO_V1 (test_bms_subset_compare)

	PG_FUNCTION_INFO_V1 (test_bms_union)

	PG_FUNCTION_INFO_V1 (test_bms_intersect)

	PG_FUNCTION_INFO_V1 (test_bms_difference)

	PG_FUNCTION_INFO_V1 (test_bms_offset_members)

	PG_FUNCTION_INFO_V1 (test_bms_is_empty)

	PG_FUNCTION_INFO_V1 (test_bms_membership)

	PG_FUNCTION_INFO_V1 (test_bms_singleton_member)

	PG_FUNCTION_INFO_V1 (test_bms_get_singleton_member)

	PG_FUNCTION_INFO_V1 (test_bms_next_member)

	PG_FUNCTION_INFO_V1 (test_bms_prev_member)

	PG_FUNCTION_INFO_V1 (test_bms_hash_value)

	PG_FUNCTION_INFO_V1 (test_bms_overlap)

	PG_FUNCTION_INFO_V1 (test_bms_overlap_list)

	PG_FUNCTION_INFO_V1 (test_bms_nonempty_difference)

	PG_FUNCTION_INFO_V1 (test_bms_member_index)

	PG_FUNCTION_INFO_V1 (test_bms_add_range)

	PG_FUNCTION_INFO_V1 (test_bms_add_members)

	PG_FUNCTION_INFO_V1 (test_bms_int_members)

	PG_FUNCTION_INFO_V1 (test_bms_del_members)

	PG_FUNCTION_INFO_V1 (test_bms_replace_members)

	PG_FUNCTION_INFO_V1 (test_bms_join)

	PG_FUNCTION_INFO_V1 (test_bitmap_hash)

	PG_FUNCTION_INFO_V1 (test_bitmap_match)

	PG_FUNCTION_INFO_V1 (test_random_operations)

	PG_FUNCTION_INFO_V1 (test_random_offset_operations)

Datum	test_bms_add_member (PG_FUNCTION_ARGS)

Datum	test_bms_add_members (PG_FUNCTION_ARGS)

Datum	test_bms_del_member (PG_FUNCTION_ARGS)

Datum	test_bms_is_member (PG_FUNCTION_ARGS)

Datum	test_bms_num_members (PG_FUNCTION_ARGS)

Datum	test_bms_make_singleton (PG_FUNCTION_ARGS)

Datum	test_bms_copy (PG_FUNCTION_ARGS)

Datum	test_bms_equal (PG_FUNCTION_ARGS)

Datum	test_bms_union (PG_FUNCTION_ARGS)

Datum	test_bms_membership (PG_FUNCTION_ARGS)

Datum	test_bms_next_member (PG_FUNCTION_ARGS)

Datum	test_bms_intersect (PG_FUNCTION_ARGS)

Datum	test_bms_difference (PG_FUNCTION_ARGS)

Datum	test_bms_offset_members (PG_FUNCTION_ARGS)

Datum	test_bms_compare (PG_FUNCTION_ARGS)

Datum	test_bms_is_empty (PG_FUNCTION_ARGS)

Datum	test_bms_is_subset (PG_FUNCTION_ARGS)

Datum	test_bms_subset_compare (PG_FUNCTION_ARGS)

Datum	test_bms_singleton_member (PG_FUNCTION_ARGS)

Datum	test_bms_get_singleton_member (PG_FUNCTION_ARGS)

Datum	test_bms_prev_member (PG_FUNCTION_ARGS)

Datum	test_bms_overlap (PG_FUNCTION_ARGS)

Datum	test_bms_overlap_list (PG_FUNCTION_ARGS)

Datum	test_bms_nonempty_difference (PG_FUNCTION_ARGS)

Datum	test_bms_member_index (PG_FUNCTION_ARGS)

Datum	test_bms_add_range (PG_FUNCTION_ARGS)

Datum	test_bms_int_members (PG_FUNCTION_ARGS)

Datum	test_bms_del_members (PG_FUNCTION_ARGS)

Datum	test_bms_replace_members (PG_FUNCTION_ARGS)

Datum	test_bms_join (PG_FUNCTION_ARGS)

Datum	test_bms_hash_value (PG_FUNCTION_ARGS)

Datum	test_bitmap_hash (PG_FUNCTION_ARGS)

Datum	test_bitmap_match (PG_FUNCTION_ARGS)

Datum	test_random_operations (PG_FUNCTION_ARGS)

Datum	test_random_offset_operations (PG_FUNCTION_ARGS)

Variables
	PG_MODULE_MAGIC

Macro Definition Documentation

◆ BITMAPSET_TO_TEXT

#define BITMAPSET_TO_TEXT ( bms ) cstring_to_text(nodeToString(bms))

Definition at line 90 of file test_bitmapset.c.

◆ EXPECT_NOT_NULL

#define EXPECT_NOT_NULL ( expr )

Value:

    do { \
		if ((expr) == NULL) \
			elog(ERROR, \
                 "%s was unexpectedly true in file \"%s\" line %u", \
                 #expr, __FILE__, __LINE__); \
    } while (0)

Definition at line 81 of file test_bitmapset.c.

       { \
		if ((expr) == NULL) \
			elog(ERROR, \
                 "%s was unexpectedly true in file \"%s\" line %u", \
                 #expr, __FILE__, __LINE__); \
    } while (0)

◆ EXPECT_TRUE

#define EXPECT_TRUE ( expr )

Value:

    do { \
		if (!(expr)) \
			elog(ERROR, \
                 "%s was unexpectedly false in file \"%s\" line %u", \
                 #expr, __FILE__, __LINE__); \
    } while (0)

Definition at line 73 of file test_bitmapset.c.

       { \
		if (!(expr)) \
			elog(ERROR, \
                 "%s was unexpectedly false in file \"%s\" line %u", \
                 #expr, __FILE__, __LINE__); \
    } while (0)

◆ PG_ARG_GETBITMAPSET

#define PG_ARG_GETBITMAPSET ( n ) (PG_ARGISNULL(n) ? NULL : TEXT_TO_BITMAPSET(PG_GETARG_TEXT_PP(n)))

Definition at line 97 of file test_bitmapset.c.

98 : TEXT_TO_BITMAPSET(PG_GETARG_TEXT_PP(n)))

PG_GETARG_TEXT_PP

#define PG_GETARG_TEXT_PP(n)

Definition fmgr.h:310

TEXT_TO_BITMAPSET

#define TEXT_TO_BITMAPSET(str)

Definition test_bitmapset.c:91

◆ PG_RETURN_BITMAPSET_AS_TEXT

#define PG_RETURN_BITMAPSET_AS_TEXT ( bms ) PG_RETURN_TEXT_P(BITMAPSET_TO_TEXT(bms))

Definition at line 104 of file test_bitmapset.c.

◆ TEXT_TO_BITMAPSET

#define TEXT_TO_BITMAPSET ( str ) ((Bitmapset *) stringToNode(text_to_cstring(str)))

Definition at line 91 of file test_bitmapset.c.

Function Documentation

◆ PG_FUNCTION_INFO_V1() [1/35]

PG_FUNCTION_INFO_V1 ( test_bitmap_hash )

◆ PG_FUNCTION_INFO_V1() [2/35]

PG_FUNCTION_INFO_V1 ( test_bitmap_match )

◆ PG_FUNCTION_INFO_V1() [3/35]

PG_FUNCTION_INFO_V1 ( test_bms_add_member )

◆ PG_FUNCTION_INFO_V1() [4/35]

PG_FUNCTION_INFO_V1 ( test_bms_add_members )

◆ PG_FUNCTION_INFO_V1() [5/35]

PG_FUNCTION_INFO_V1 ( test_bms_add_range )

◆ PG_FUNCTION_INFO_V1() [6/35]

PG_FUNCTION_INFO_V1 ( test_bms_compare )

◆ PG_FUNCTION_INFO_V1() [7/35]

PG_FUNCTION_INFO_V1 ( test_bms_copy )

◆ PG_FUNCTION_INFO_V1() [8/35]

PG_FUNCTION_INFO_V1 ( test_bms_del_member )

◆ PG_FUNCTION_INFO_V1() [9/35]

PG_FUNCTION_INFO_V1 ( test_bms_del_members )

◆ PG_FUNCTION_INFO_V1() [10/35]

PG_FUNCTION_INFO_V1 ( test_bms_difference )

◆ PG_FUNCTION_INFO_V1() [11/35]

PG_FUNCTION_INFO_V1 ( test_bms_equal )

◆ PG_FUNCTION_INFO_V1() [12/35]

PG_FUNCTION_INFO_V1 ( test_bms_get_singleton_member )

◆ PG_FUNCTION_INFO_V1() [13/35]

PG_FUNCTION_INFO_V1 ( test_bms_hash_value )

◆ PG_FUNCTION_INFO_V1() [14/35]

PG_FUNCTION_INFO_V1 ( test_bms_int_members )

◆ PG_FUNCTION_INFO_V1() [15/35]

PG_FUNCTION_INFO_V1 ( test_bms_intersect )

◆ PG_FUNCTION_INFO_V1() [16/35]

PG_FUNCTION_INFO_V1 ( test_bms_is_empty )

◆ PG_FUNCTION_INFO_V1() [17/35]

PG_FUNCTION_INFO_V1 ( test_bms_is_member )

◆ PG_FUNCTION_INFO_V1() [18/35]

PG_FUNCTION_INFO_V1 ( test_bms_is_subset )

◆ PG_FUNCTION_INFO_V1() [19/35]

PG_FUNCTION_INFO_V1 ( test_bms_join )

◆ PG_FUNCTION_INFO_V1() [20/35]

PG_FUNCTION_INFO_V1 ( test_bms_make_singleton )

◆ PG_FUNCTION_INFO_V1() [21/35]

PG_FUNCTION_INFO_V1 ( test_bms_member_index )

◆ PG_FUNCTION_INFO_V1() [22/35]

PG_FUNCTION_INFO_V1 ( test_bms_membership )

◆ PG_FUNCTION_INFO_V1() [23/35]

PG_FUNCTION_INFO_V1 ( test_bms_next_member )

◆ PG_FUNCTION_INFO_V1() [24/35]

PG_FUNCTION_INFO_V1 ( test_bms_nonempty_difference )

◆ PG_FUNCTION_INFO_V1() [25/35]

PG_FUNCTION_INFO_V1 ( test_bms_num_members )

◆ PG_FUNCTION_INFO_V1() [26/35]

PG_FUNCTION_INFO_V1 ( test_bms_offset_members )

◆ PG_FUNCTION_INFO_V1() [27/35]

PG_FUNCTION_INFO_V1 ( test_bms_overlap )

◆ PG_FUNCTION_INFO_V1() [28/35]

PG_FUNCTION_INFO_V1 ( test_bms_overlap_list )

◆ PG_FUNCTION_INFO_V1() [29/35]

PG_FUNCTION_INFO_V1 ( test_bms_prev_member )

◆ PG_FUNCTION_INFO_V1() [30/35]

PG_FUNCTION_INFO_V1 ( test_bms_replace_members )

◆ PG_FUNCTION_INFO_V1() [31/35]

PG_FUNCTION_INFO_V1 ( test_bms_singleton_member )

◆ PG_FUNCTION_INFO_V1() [32/35]

PG_FUNCTION_INFO_V1 ( test_bms_subset_compare )

◆ PG_FUNCTION_INFO_V1() [33/35]

PG_FUNCTION_INFO_V1 ( test_bms_union )

◆ PG_FUNCTION_INFO_V1() [34/35]

PG_FUNCTION_INFO_V1 ( test_random_offset_operations )

◆ PG_FUNCTION_INFO_V1() [35/35]

PG_FUNCTION_INFO_V1 ( test_random_operations )

◆ test_bitmap_hash()

Datum test_bitmap_hash ( PG_FUNCTION_ARGS )

Definition at line 572 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    uint32      hash_result;
 
    /* Call bitmap_hash */
    hash_result = bitmap_hash(&bms, sizeof(Bitmapset *));
 
    PG_RETURN_INT32(hash_result);
}

References bitmap_hash(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bitmap_match()

Datum test_bitmap_match ( PG_FUNCTION_ARGS )

Definition at line 584 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    int         match_result;
 
    /* Call bitmap_match with addresses of the Bitmapset pointers */
    match_result = bitmap_match(&bms1, &bms2, sizeof(Bitmapset *));
 
    PG_RETURN_INT32(match_result);
}

References bitmap_match(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_add_member()

Datum test_bms_add_member ( PG_FUNCTION_ARGS )

Definition at line 127 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int         member;
 
    if (PG_ARGISNULL(1))
        PG_RETURN_NULL();       /* invalid input */
 
    bms = PG_ARG_GETBITMAPSET(0);
    member = PG_GETARG_INT32(1);
 
    bms = bms_add_member(bms, member);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms);
}

References bms_add_member(), fb(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BITMAPSET_AS_TEXT, and PG_RETURN_NULL.

◆ test_bms_add_members()

Datum test_bms_add_members ( PG_FUNCTION_ARGS )

Definition at line 144 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
 
    /* left input is recycled */
    bms1 = bms_add_members(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms1);
}

References bms_add_members(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_add_range()

Datum test_bms_add_range ( PG_FUNCTION_ARGS )

Definition at line 493 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int32       lower,
                upper;
 
    if (PG_ARGISNULL(1) || PG_ARGISNULL(2))
        PG_RETURN_NULL();       /* invalid input */
 
    bms = PG_ARG_GETBITMAPSET(0);
    lower = PG_GETARG_INT32(1);
    upper = PG_GETARG_INT32(2);
 
    bms = bms_add_range(bms, lower, upper);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms);
}

References bms_add_range(), fb(), lower(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BITMAPSET_AS_TEXT, PG_RETURN_NULL, and upper().

◆ test_bms_compare()

Datum test_bms_compare ( PG_FUNCTION_ARGS )

Definition at line 313 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    int         result;
 
    result = bms_compare(bms1, bms2);
 
    PG_RETURN_INT32(result);
}

References bms_compare(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_INT32, and result.

◆ test_bms_copy()

Datum test_bms_copy ( PG_FUNCTION_ARGS )

Definition at line 214 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *copy_bms;
 
    copy_bms = bms_copy(bms);
 
    PG_RETURN_BITMAPSET_AS_TEXT(copy_bms);
}

References bms_copy(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_del_member()

Datum test_bms_del_member ( PG_FUNCTION_ARGS )

Definition at line 156 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int32       member;
 
    if (PG_ARGISNULL(1))
        PG_RETURN_NULL();       /* invalid input */
 
    bms = PG_ARG_GETBITMAPSET(0);
    member = PG_GETARG_INT32(1);
 
    bms = bms_del_member(bms, member);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms);
}

References bms_del_member(), fb(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BITMAPSET_AS_TEXT, and PG_RETURN_NULL.

◆ test_bms_del_members()

Datum test_bms_del_members ( PG_FUNCTION_ARGS )

Definition at line 524 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
 
    /* left input gets recycled */
    bms1 = bms_del_members(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms1);
}

References bms_del_members(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_difference()

Datum test_bms_difference ( PG_FUNCTION_ARGS )

Definition at line 290 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    Bitmapset  *result_bms;
 
    result_bms = bms_difference(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
}

References bms_difference(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_equal()

Datum test_bms_equal ( PG_FUNCTION_ARGS )

Definition at line 225 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    bool        result;
 
    result = bms_equal(bms1, bms2);
 
    PG_RETURN_BOOL(result);
}

References bms_equal(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_BOOL, and result.

◆ test_bms_get_singleton_member()

Datum test_bms_get_singleton_member ( PG_FUNCTION_ARGS )

Definition at line 371 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    int         member;
 
    /*
     * Keep this simple.  Return -1 when we detect the set is not a singleton
     * set, otherwise return the singleton member.
     */
    if (!bms_get_singleton_member(bms, &member))
        member = -1;
 
    PG_RETURN_INT32(member);
}

References bms_get_singleton_member(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_hash_value()

Datum test_bms_hash_value ( PG_FUNCTION_ARGS )

Definition at line 561 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    uint32      hash_result;
 
    hash_result = bms_hash_value(bms);
 
    PG_RETURN_INT32(hash_result);
}

References bms_hash_value(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_int_members()

Datum test_bms_int_members ( PG_FUNCTION_ARGS )

Definition at line 512 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
 
    /* left input gets recycled */
    bms1 = bms_int_members(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms1);
}

References bms_int_members(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_intersect()

Datum test_bms_intersect ( PG_FUNCTION_ARGS )

Definition at line 278 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    Bitmapset  *result_bms;
 
    result_bms = bms_intersect(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
}

References bms_intersect(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_is_empty()

Datum test_bms_is_empty ( PG_FUNCTION_ARGS )

Definition at line 325 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    bool        result;
 
    result = bms_is_empty(bms);
 
    PG_RETURN_BOOL(result);
}

References bms_is_empty, fb(), PG_ARG_GETBITMAPSET, PG_RETURN_BOOL, and result.

◆ test_bms_is_member()

Datum test_bms_is_member ( PG_FUNCTION_ARGS )

Definition at line 173 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int32       member;
    bool        result;
 
    if (PG_ARGISNULL(1))
        PG_RETURN_NULL();       /* invalid input */
 
    bms = PG_ARG_GETBITMAPSET(0);
    member = PG_GETARG_INT32(1);
 
    result = bms_is_member(member, bms);
 
    PG_RETURN_BOOL(result);
}

References bms_is_member(), fb(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BOOL, PG_RETURN_NULL, and result.

◆ test_bms_is_subset()

Datum test_bms_is_subset ( PG_FUNCTION_ARGS )

Definition at line 336 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    bool        result;
 
    result = bms_is_subset(bms1, bms2);
 
    PG_RETURN_BOOL(result);
}

References bms_is_subset(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_BOOL, and result.

◆ test_bms_join()

Datum test_bms_join ( PG_FUNCTION_ARGS )

Definition at line 548 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    Bitmapset  *result_bms;
 
    /* either input can be recycled */
    result_bms = bms_join(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
}

References bms_join(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_make_singleton()

Datum test_bms_make_singleton ( PG_FUNCTION_ARGS )

Definition at line 202 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int32       member;
 
    member = PG_GETARG_INT32(0);
    bms = bms_make_singleton(member);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms);
}

References bms_make_singleton(), fb(), PG_GETARG_INT32, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_member_index()

Datum test_bms_member_index ( PG_FUNCTION_ARGS )

Definition at line 475 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int32       member;
    int         result;
 
    if (PG_ARGISNULL(1))
        PG_RETURN_NULL();       /* invalid input */
 
    bms = PG_ARG_GETBITMAPSET(0);
    member = PG_GETARG_INT32(1);
 
    result = bms_member_index(bms, member);
 
    PG_RETURN_INT32(result);
}

References bms_member_index(), fb(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_INT32, PG_RETURN_NULL, and result.

◆ test_bms_membership()

Datum test_bms_membership ( PG_FUNCTION_ARGS )

Definition at line 249 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    BMS_Membership result;
 
    result = bms_membership(bms);
 
    PG_RETURN_INT32((int32) result);
}

References bms_membership(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_INT32, and result.

◆ test_bms_next_member()

Datum test_bms_next_member ( PG_FUNCTION_ARGS )

Definition at line 260 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int32       prevmember;
    int         result;
 
    if (PG_ARGISNULL(1))
        PG_RETURN_NULL();       /* invalid input */
 
    bms = PG_ARG_GETBITMAPSET(0);
    prevmember = PG_GETARG_INT32(1);
 
    result = bms_next_member(bms, prevmember);
 
    PG_RETURN_INT32(result);
}

References bms_next_member(), fb(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_INT32, PG_RETURN_NULL, and result.

◆ test_bms_nonempty_difference()

Datum test_bms_nonempty_difference ( PG_FUNCTION_ARGS )

Definition at line 463 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    bool        result;
 
    result = bms_nonempty_difference(bms1, bms2);
 
    PG_RETURN_BOOL(result);
}

References bms_nonempty_difference(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_BOOL, and result.

◆ test_bms_num_members()

Datum test_bms_num_members ( PG_FUNCTION_ARGS )

Definition at line 191 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    int         result;
 
    result = bms_num_members(bms);
 
    PG_RETURN_INT32(result);
}

References bms_num_members(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_INT32, and result.

◆ test_bms_offset_members()

Datum test_bms_offset_members ( PG_FUNCTION_ARGS )

Definition at line 302 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    int         offset = PG_GETARG_INT32(1);
 
    bms = bms_offset_members(bms, offset);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms);
}

References bms_offset_members(), fb(), PG_ARG_GETBITMAPSET, PG_GETARG_INT32, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_overlap()

Datum test_bms_overlap ( PG_FUNCTION_ARGS )

Definition at line 405 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    bool        result;
 
    result = bms_overlap(bms1, bms2);
 
    PG_RETURN_BOOL(result);
}

References bms_overlap(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_BOOL, and result.

◆ test_bms_overlap_list()

Datum test_bms_overlap_list ( PG_FUNCTION_ARGS )

Definition at line 417 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    ArrayType  *array;
    List       *int_list = NIL;
    bool        result;
    Datum      *elem_datums = NULL;
    bool       *elem_nulls = NULL;
    int         elem_count;
    int         i;
 
    bms = PG_ARG_GETBITMAPSET(0);
 
    if (!PG_ARGISNULL(1))
    {
        array = PG_GETARG_ARRAYTYPE_P(1);
 
        deconstruct_array(array,
                          INT4OID, sizeof(int32), true, 'i',
                          &elem_datums, &elem_nulls, &elem_count);
 
        for (i = 0; i < elem_count; i++)
        {
            if (!elem_nulls[i])
            {
                int32       member = DatumGetInt32(elem_datums[i]);
 
                int_list = lappend_int(int_list, member);
            }
        }
    }
 
    result = bms_overlap_list(bms, int_list);
 
    list_free(int_list);
 
    if (elem_datums)
        pfree(elem_datums);
 
    if (elem_nulls)
        pfree(elem_nulls);
 
    PG_RETURN_BOOL(result);
}

References bms_overlap_list(), DatumGetInt32(), deconstruct_array(), fb(), i, lappend_int(), list_free(), NIL, pfree(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_ARRAYTYPE_P, PG_RETURN_BOOL, and result.

◆ test_bms_prev_member()

Datum test_bms_prev_member ( PG_FUNCTION_ARGS )

Definition at line 387 of file test_bitmapset.c.

{
    Bitmapset  *bms;
    int32       prevmember;
    int         result;
 
    if (PG_ARGISNULL(1))
        PG_RETURN_NULL();       /* invalid input */
 
    bms = PG_ARG_GETBITMAPSET(0);
    prevmember = PG_GETARG_INT32(1);
 
    result = bms_prev_member(bms, prevmember);
 
    PG_RETURN_INT32(result);
}

References bms_prev_member(), fb(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_INT32, PG_RETURN_NULL, and result.

◆ test_bms_replace_members()

Datum test_bms_replace_members ( PG_FUNCTION_ARGS )

Definition at line 536 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
 
    /* left input gets recycled */
    bms1 = bms_replace_members(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(bms1);
}

References bms_replace_members(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_singleton_member()

Datum test_bms_singleton_member ( PG_FUNCTION_ARGS )

Definition at line 360 of file test_bitmapset.c.

{
    Bitmapset  *bms = PG_ARG_GETBITMAPSET(0);
    int         result;
 
    result = bms_singleton_member(bms);
 
    PG_RETURN_INT32(result);
}

References bms_singleton_member(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_INT32, and result.

◆ test_bms_subset_compare()

Datum test_bms_subset_compare ( PG_FUNCTION_ARGS )

Definition at line 348 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    BMS_Comparison result;
 
    result = bms_subset_compare(bms1, bms2);
 
    PG_RETURN_INT32((int32) result);
}

References bms_subset_compare(), fb(), PG_ARG_GETBITMAPSET, PG_RETURN_INT32, and result.

◆ test_bms_union()

Datum test_bms_union ( PG_FUNCTION_ARGS )

Definition at line 237 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = PG_ARG_GETBITMAPSET(0);
    Bitmapset  *bms2 = PG_ARG_GETBITMAPSET(1);
    Bitmapset  *result_bms;
 
    result_bms = bms_union(bms1, bms2);
 
    PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
}

References bms_union(), fb(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_random_offset_operations()

Datum test_random_offset_operations ( PG_FUNCTION_ARGS )

Definition at line 797 of file test_bitmapset.c.

{
    pg_prng_state state;
    int64       seed;
    int         num_ops;
    int         max_range;
    int         min_value;
    int         member;
 
    if (PG_ARGISNULL(0))
        seed = GetCurrentTimestamp();
    else
        seed = PG_GETARG_INT64(0);
 
    num_ops = PG_GETARG_INT32(1);
    max_range = PG_GETARG_INT32(2);
    min_value = PG_GETARG_INT32(3);
 
    if (PG_ARGISNULL(1) || num_ops <= 0)
        elog(ERROR, "invalid number of operations");
    if (PG_ARGISNULL(2) || max_range <= 0)
        elog(ERROR, "invalid maximum range");
    if (PG_ARGISNULL(3) || min_value < 0)
        elog(ERROR, "invalid minimum value");
 
    pg_prng_seed(&state, (uint64) seed);
 
    for (int op = 0; op < num_ops; op++)
    {
        Bitmapset  *random_bms = NULL;
        Bitmapset  *offset_bms1;
        Bitmapset  *offset_bms2 = NULL;
        int         offset;
        int         nmembers;
 
        CHECK_FOR_INTERRUPTS();
 
        /* Figure out a random offset and how many members to add */
        offset = (pg_prng_uint32(&state) % max_range) - (pg_prng_uint32(&state) % max_range);
        nmembers = pg_prng_uint32(&state) % max_range + min_value;
 
        for (int i = 0; i < nmembers; i++)
        {
            member = pg_prng_uint32(&state) % max_range + min_value;
            random_bms = bms_add_member(random_bms, member);
        }
 
        /* create a known-good set the old fashioned way */
        offset_bms2 = NULL;
        member = -1;
        while ((member = bms_next_member(random_bms, member)) >= 0)
        {
            if (member + offset >= 0)
                offset_bms2 = bms_add_member(offset_bms2, member + offset);
        }
 
        /* do the offsetting */
        offset_bms1 = bms_offset_members(random_bms, offset);
 
        /* check against the known-good set */
        if (!bms_equal(offset_bms1, offset_bms2))
            elog(ERROR, "bms_offset_members failed with offset %d seed " INT64_FORMAT, offset, seed);
 
        /* Cleanup before the next loop */
        bms_free(random_bms);
        bms_free(offset_bms1);
        bms_free(offset_bms2);
    }
 
    PG_RETURN_INT32(num_ops);
}

References bms_add_member(), bms_equal(), bms_free(), bms_next_member(), bms_offset_members(), CHECK_FOR_INTERRUPTS, elog, ERROR, fb(), GetCurrentTimestamp(), i, INT64_FORMAT, PG_ARGISNULL, PG_GETARG_INT32, PG_GETARG_INT64, pg_prng_seed(), pg_prng_uint32(), and PG_RETURN_INT32.

◆ test_random_operations()

Datum test_random_operations ( PG_FUNCTION_ARGS )

Definition at line 609 of file test_bitmapset.c.

{
    Bitmapset  *bms1 = NULL;
    Bitmapset  *bms2 = NULL;
    Bitmapset  *bms = NULL;
    Bitmapset  *result = NULL;
    pg_prng_state state;
    uint64      seed = GetCurrentTimestamp();
    int         num_ops;
    int         max_range;
    int         min_value;
    int         member;
    int        *members;
    int         num_members = 0;
    int         total_ops = 0;
 
    if (!PG_ARGISNULL(0))
        seed = PG_GETARG_INT64(0);
 
    num_ops = PG_GETARG_INT32(1);
    max_range = PG_GETARG_INT32(2);
    min_value = PG_GETARG_INT32(3);
 
    if (PG_ARGISNULL(1) || num_ops <= 0)
        elog(ERROR, "invalid number of operations");
    if (PG_ARGISNULL(2) || max_range <= 0)
        elog(ERROR, "invalid maximum range");
    if (PG_ARGISNULL(3) || min_value < 0)
        elog(ERROR, "invalid minimum value");
 
    pg_prng_seed(&state, seed);
 
    /*
     * There can be up to "num_ops" members added.  This is very unlikely,
     * still possible if all the operations hit the "0" case during phase 4
     * where multiple operation types are mixed together.
     */
    members = palloc_array(int, num_ops);
 
    /* Phase 1: Random insertions in first set */
    for (int i = 0; i < num_ops / 2; i++)
    {
        CHECK_FOR_INTERRUPTS();
 
        member = pg_prng_uint32(&state) % max_range + min_value;
 
        if (!bms_is_member(member, bms1))
            members[num_members++] = member;
        bms1 = bms_add_member(bms1, member);
    }
 
    /* Phase 2: Random insertions in second set */
    for (int i = 0; i < num_ops / 4; i++)
    {
        CHECK_FOR_INTERRUPTS();
 
        member = pg_prng_uint32(&state) % max_range + min_value;
 
        if (!bms_is_member(member, bms2))
            members[num_members++] = member;
        bms2 = bms_add_member(bms2, member);
    }
 
    /* Test union */
    result = bms_union(bms1, bms2);
    EXPECT_NOT_NULL(result);
 
    /* Verify union contains all members from first and second sets */
    for (int i = 0; i < num_members; i++)
    {
        CHECK_FOR_INTERRUPTS();
 
        if (!bms_is_member(members[i], result))
            elog(ERROR, "union missing member %d, seed " UINT64_FORMAT,
                 members[i], seed);
    }
    bms_free(result);
 
    /*
     * Test intersection, checking that all the members in the result are from
     * both the first and second sets.
     */
    result = bms_intersect(bms1, bms2);
    if (result != NULL)
    {
        member = -1;
 
        while ((member = bms_next_member(result, member)) >= 0)
        {
            CHECK_FOR_INTERRUPTS();
 
            if (!bms_is_member(member, bms1) || !bms_is_member(member, bms2))
                elog(ERROR, "intersection contains invalid member %d, seed " UINT64_FORMAT,
                     member, seed);
        }
        bms_free(result);
    }
 
    /* Phase 3: Test range operations */
    result = NULL;
    for (int i = 0; i < num_ops; i++)
    {
        int         lower = pg_prng_uint32(&state) % 100;
        int         upper = lower + (pg_prng_uint32(&state) % 20);
 
        CHECK_FOR_INTERRUPTS();
 
        result = bms_add_range(result, lower, upper);
    }
    if (result != NULL)
    {
        EXPECT_TRUE(bms_num_members(result) > 0);
        bms_free(result);
    }
 
    bms_free(bms1);
    bms_free(bms2);
 
    /*
     * Phase 4: mix of operations on a single set, cross-checking a bitmap
     * with a secondary state, "members".
     */
    num_members = 0;
 
    for (int op = 0; op < num_ops; op++)
    {
        CHECK_FOR_INTERRUPTS();
 
        switch (pg_prng_uint32(&state) % 3)
        {
            case 0:             /* add */
                member = pg_prng_uint32(&state) % max_range + min_value;
                if (!bms_is_member(member, bms))
                    members[num_members++] = member;
                bms = bms_add_member(bms, member);
                break;
            case 1:             /* delete */
                if (num_members > 0)
                {
                    int         pos = pg_prng_uint32(&state) % num_members;
 
                    member = members[pos];
                    if (!bms_is_member(member, bms))
                        elog(ERROR, "expected %d to be a valid member, seed " UINT64_FORMAT,
                             member, seed);
 
                    bms = bms_del_member(bms, member);
 
                    /*
                     * Move the final array member at the position of the
                     * member just deleted, reducing the array size by one.
                     */
                    members[pos] = members[--num_members];
                }
                break;
            case 2:             /* test membership */
                /* Verify that bitmap contains all members */
                for (int i = 0; i < num_members; i++)
                {
                    if (!bms_is_member(members[i], bms))
                        elog(ERROR, "missing member %d, seed " UINT64_FORMAT,
                             members[i], seed);
                }
                break;
        }
        total_ops++;
    }
 
    bms_free(bms);
    pfree(members);
 
    PG_RETURN_INT32(total_ops);
}

References bms_add_member(), bms_add_range(), bms_del_member(), bms_free(), bms_intersect(), bms_is_member(), bms_next_member(), bms_num_members(), bms_union(), CHECK_FOR_INTERRUPTS, elog, ERROR, EXPECT_NOT_NULL, EXPECT_TRUE, fb(), GetCurrentTimestamp(), i, lower(), palloc_array, pfree(), PG_ARGISNULL, PG_GETARG_INT32, PG_GETARG_INT64, pg_prng_seed(), pg_prng_uint32(), PG_RETURN_INT32, result, UINT64_FORMAT, and upper().

Variable Documentation

◆ PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 31 of file test_bitmapset.c.

Macros

Functions

Variables