#include "postgres.h"
#include <limits.h>
#include "_int.h"
#include "catalog/pg_type.h"
#include "common/int.h"
#include "lib/qunique.h"
#include "lib/sort_template.h"

Include dependency graph for _int_tool.c:

Macros
#define	ST_SORT isort

#define	ST_ELEMENT_TYPE int32

#define	ST_COMPARE(a, b, ascending) isort_cmp(a, b, ascending)

#define	ST_COMPARE_ARG_TYPE void

#define	ST_SCOPE

#define	ST_DEFINE

Functions
bool	inner_int_contains (ArrayType a, ArrayType b)

bool	inner_int_overlap (ArrayType a, ArrayType b)

ArrayType *	inner_int_union (ArrayType a, ArrayType b)

ArrayType *	inner_int_inter (ArrayType a, ArrayType b)

void	rt__int_size (ArrayType a, float size)

static int	isort_cmp (const void a, const void b, void *arg)

ArrayType *	new_intArrayType (int num)

ArrayType *	resize_intArrayType (ArrayType *a, int num)

ArrayType *	copy_intArrayType (ArrayType *a)

int	internal_size (int *a, int len)

ArrayType *	_int_unique (ArrayType *r)

void	gensign (BITVECP sign, int *a, int len, int siglen)

int32	intarray_match_first (ArrayType *a, int32 elem)

ArrayType *	intarray_add_elem (ArrayType *a, int32 elem)

ArrayType *	intarray_concat_arrays (ArrayType a, ArrayType b)

ArrayType *	int_to_intset (int32 elem)

Macro Definition Documentation

◆ ST_COMPARE

#define ST_COMPARE	(	a,
		b,
		ascending
	)	isort_cmp(a, b, ascending)

Definition at line 216 of file _int_tool.c.

◆ ST_COMPARE_ARG_TYPE

#define ST_COMPARE_ARG_TYPE void

Definition at line 217 of file _int_tool.c.

◆ ST_DEFINE

#define ST_DEFINE

Definition at line 219 of file _int_tool.c.

◆ ST_ELEMENT_TYPE

#define ST_ELEMENT_TYPE int32

Definition at line 215 of file _int_tool.c.

◆ ST_SCOPE

#define ST_SCOPE

Definition at line 218 of file _int_tool.c.

◆ ST_SORT

#define ST_SORT isort

Definition at line 214 of file _int_tool.c.

Function Documentation

◆ _int_unique()

ArrayType * _int_unique ( ArrayType * r )

Definition at line 313 of file _int_tool.c.

{
    int         num = ARRNELEMS(r);
    bool        ascending = true;
 
    num = qunique_arg(ARRPTR(r), num, sizeof(int), isort_cmp,
                      &ascending);
 
    return resize_intArrayType(r, num);
}

References ARRNELEMS, ARRPTR, isort_cmp(), qunique_arg(), and resize_intArrayType().

Referenced by g_int_union(), inner_int_union(), intset_subtract(), intset_union_elem(), and uniq().

◆ copy_intArrayType()

ArrayType * copy_intArrayType ( ArrayType * a )

Definition at line 283 of file _int_tool.c.

{
    ArrayType  *r;
    int         n = ARRNELEMS(a);
 
    r = new_intArrayType(n);
    memcpy(ARRPTR(r), ARRPTR(a), n * sizeof(int32));
    return r;
}

References a, ARRNELEMS, ARRPTR, and new_intArrayType().

Referenced by g_int_picksplit(), and inner_int_union().

◆ gensign()

void gensign	(	BITVECP	sign,
		int *	a,
		int	len,
		int	siglen
	)

Definition at line 325 of file _int_tool.c.

{
    int         i;
 
    /* we assume that the sign vector is previously zeroed */
    for (i = 0; i < len; i++)
    {
        HASH(sign, *a, siglen);
        a++;
    }
}

References a, HASH, i, len, and sign.

◆ inner_int_contains()

bool inner_int_contains	(	ArrayType *	a,
		ArrayType *	b
	)

Definition at line 15 of file _int_tool.c.

{
    int         na,
                nb;
    int         i,
                j,
                n;
    int        *da,
               *db;
 
    na = ARRNELEMS(a);
    nb = ARRNELEMS(b);
    da = ARRPTR(a);
    db = ARRPTR(b);
 
    i = j = n = 0;
    while (i < na && j < nb)
    {
        if (da[i] < db[j])
            i++;
        else if (da[i] == db[j])
        {
            n++;
            i++;
            j++;
        }
        else
            break;              /* db[j] is not in da */
    }
 
    return (n == nb);
}

References a, ARRNELEMS, ARRPTR, b, i, and j.

Referenced by _int_contains(), and g_int_consistent().

◆ inner_int_inter()

ArrayType * inner_int_inter	(	ArrayType *	a,
		ArrayType *	b
	)

Definition at line 136 of file _int_tool.c.

{
    ArrayType  *r;
    int         na,
                nb;
    int        *da,
               *db,
               *dr;
    int         i,
                j,
                k;
 
    if (ARRISEMPTY(a) || ARRISEMPTY(b))
        return new_intArrayType(0);
 
    na = ARRNELEMS(a);
    nb = ARRNELEMS(b);
    da = ARRPTR(a);
    db = ARRPTR(b);
    r = new_intArrayType(Min(na, nb));
    dr = ARRPTR(r);
 
    i = j = k = 0;
    while (i < na && j < nb)
    {
        if (da[i] < db[j])
            i++;
        else if (da[i] == db[j])
        {
            if (k == 0 || dr[k - 1] != db[j])
                dr[k++] = db[j];
            i++;
            j++;
        }
        else
            j++;
    }
 
    if (k == 0)
    {
        pfree(r);
        return new_intArrayType(0);
    }
    else
        return resize_intArrayType(r, k);
}

References a, ARRISEMPTY, ARRNELEMS, ARRPTR, b, i, j, Min, new_intArrayType(), pfree(), and resize_intArrayType().

Referenced by _int_inter(), and g_int_picksplit().

◆ inner_int_overlap()

bool inner_int_overlap	(	ArrayType *	a,
		ArrayType *	b
	)

Definition at line 50 of file _int_tool.c.

{
    int         na,
                nb;
    int         i,
                j;
    int        *da,
               *db;
 
    na = ARRNELEMS(a);
    nb = ARRNELEMS(b);
    da = ARRPTR(a);
    db = ARRPTR(b);
 
    i = j = 0;
    while (i < na && j < nb)
    {
        if (da[i] < db[j])
            i++;
        else if (da[i] == db[j])
            return true;
        else
            j++;
    }
 
    return false;
}

References a, ARRNELEMS, ARRPTR, b, i, and j.

Referenced by _int_overlap(), and g_int_consistent().

◆ inner_int_union()

ArrayType * inner_int_union	(	ArrayType *	a,
		ArrayType *	b
	)

Definition at line 79 of file _int_tool.c.

{
    ArrayType  *r = NULL;
 
    CHECKARRVALID(a);
    CHECKARRVALID(b);
 
    if (ARRISEMPTY(a) && ARRISEMPTY(b))
        return new_intArrayType(0);
    if (ARRISEMPTY(a))
        r = copy_intArrayType(b);
    if (ARRISEMPTY(b))
        r = copy_intArrayType(a);
 
    if (!r)
    {
        int         na = ARRNELEMS(a),
                    nb = ARRNELEMS(b);
        int        *da = ARRPTR(a),
                   *db = ARRPTR(b);
        int         i,
                    j,
                   *dr;
 
        r = new_intArrayType(na + nb);
        dr = ARRPTR(r);
 
        /* union */
        i = j = 0;
        while (i < na && j < nb)
        {
            if (da[i] == db[j])
            {
                *dr++ = da[i++];
                j++;
            }
            else if (da[i] < db[j])
                *dr++ = da[i++];
            else
                *dr++ = db[j++];
        }
 
        while (i < na)
            *dr++ = da[i++];
        while (j < nb)
            *dr++ = db[j++];
 
        r = resize_intArrayType(r, dr - ARRPTR(r));
    }
 
    if (ARRNELEMS(r) > 1)
        r = _int_unique(r);
 
    return r;
}

References _int_unique(), a, ARRISEMPTY, ARRNELEMS, ARRPTR, b, CHECKARRVALID, copy_intArrayType(), i, j, new_intArrayType(), and resize_intArrayType().

Referenced by _int_union(), g_int_penalty(), and g_int_picksplit().

◆ int_to_intset()

ArrayType * int_to_intset ( int32 elem )

Definition at line 388 of file _int_tool.c.

{
    ArrayType  *result;
    int32      *aa;
 
    result = new_intArrayType(1);
    aa = ARRPTR(result);
    aa[0] = elem;
    return result;
}

References ARRPTR, and new_intArrayType().

Referenced by intset().

◆ intarray_add_elem()

ArrayType * intarray_add_elem	(	ArrayType *	a,
		int32	elem
	)

Definition at line 354 of file _int_tool.c.

{
    ArrayType  *result;
    int32      *r;
    int32       c;
 
    CHECKARRVALID(a);
    c = ARRNELEMS(a);
    result = new_intArrayType(c + 1);
    r = ARRPTR(result);
    if (c > 0)
        memcpy(r, ARRPTR(a), c * sizeof(int32));
    r[c] = elem;
    return result;
}

References a, ARRNELEMS, ARRPTR, CHECKARRVALID, and new_intArrayType().

Referenced by intarray_push_elem(), and intset_union_elem().

◆ intarray_concat_arrays()

ArrayType * intarray_concat_arrays	(	ArrayType *	a,
		ArrayType *	b
	)

Definition at line 371 of file _int_tool.c.

{
    ArrayType  *result;
    int32       ac = ARRNELEMS(a);
    int32       bc = ARRNELEMS(b);
 
    CHECKARRVALID(a);
    CHECKARRVALID(b);
    result = new_intArrayType(ac + bc);
    if (ac)
        memcpy(ARRPTR(result), ARRPTR(a), ac * sizeof(int32));
    if (bc)
        memcpy(ARRPTR(result) + ac, ARRPTR(b), bc * sizeof(int32));
    return result;
}

References a, ARRNELEMS, ARRPTR, b, CHECKARRVALID, and new_intArrayType().

Referenced by intarray_push_array().

◆ intarray_match_first()

int32 intarray_match_first	(	ArrayType *	a,
		int32	elem
	)

Definition at line 338 of file _int_tool.c.

{
    int32      *aa,
                c,
                i;
 
    CHECKARRVALID(a);
    c = ARRNELEMS(a);
    aa = ARRPTR(a);
    for (i = 0; i < c; i++)
        if (aa[i] == elem)
            return (i + 1);
    return 0;
}

References a, ARRNELEMS, ARRPTR, CHECKARRVALID, and i.

Referenced by idx().

◆ internal_size()

int internal_size	(	int *	a,
		int	len
	)

Definition at line 295 of file _int_tool.c.

{
    int         i;
    int64       size = 0;
 
    for (i = 0; i < len; i += 2)
    {
        if (!i || a[i] != a[i - 1]) /* do not count repeated range */
            size += (int64) (a[i + 1]) - (int64) (a[i]) + 1;
    }
 
    if (size > (int64) INT_MAX || size < (int64) INT_MIN)
        return -1;              /* overflow */
    return (int) size;
}

References a, i, and len.

Referenced by g_int_compress(), and g_int_decompress().

◆ isort_cmp()

static int isort_cmp	(	const void *	a,
		const void *	b,
		void *	arg
	)

inlinestatic

Definition at line 191 of file _int_tool.c.

{
    int32       aval = *((const int32 *) a);
    int32       bval = *((const int32 *) b);
 
    if (*((bool *) arg))
    {
        /* compare for ascending order */
        if (aval < bval)
            return -1;
        if (aval > bval)
            return 1;
    }
    else
    {
        if (aval > bval)
            return -1;
        if (aval < bval)
            return 1;
    }
    return 0;
}

References a, arg, and b.

Referenced by _int_unique().

◆ new_intArrayType()

ArrayType * new_intArrayType ( int num )

Definition at line 224 of file _int_tool.c.

{
    ArrayType  *r;
    int         nbytes;
 
    /* if no elements, return a zero-dimensional array */
    if (num <= 0)
    {
        Assert(num == 0);
        r = construct_empty_array(INT4OID);
        return r;
    }
 
    nbytes = ARR_OVERHEAD_NONULLS(1) + sizeof(int) * num;
 
    r = (ArrayType *) palloc0(nbytes);
 
    SET_VARSIZE(r, nbytes);
    ARR_NDIM(r) = 1;
    r->dataoffset = 0;          /* marker for no null bitmap */
    ARR_ELEMTYPE(r) = INT4OID;
    ARR_DIMS(r)[0] = num;
    ARR_LBOUND(r)[0] = 1;
 
    return r;
}

References ARR_DIMS, ARR_ELEMTYPE, ARR_LBOUND, ARR_NDIM, ARR_OVERHEAD_NONULLS, Assert(), construct_empty_array(), ArrayType::dataoffset, palloc0(), and SET_VARSIZE.

Referenced by copy_intArrayType(), g_int_decompress(), g_int_union(), inner_int_inter(), inner_int_union(), int_to_intset(), intarray_add_elem(), intarray_concat_arrays(), intset_subtract(), and subarray().

◆ resize_intArrayType()

ArrayType * resize_intArrayType	(	ArrayType *	a,
		int	num
	)

Definition at line 252 of file _int_tool.c.

{
    int         nbytes;
    int         i;
 
    /* if no elements, return a zero-dimensional array */
    if (num <= 0)
    {
        Assert(num == 0);
        a = construct_empty_array(INT4OID);
        return a;
    }
 
    if (num == ARRNELEMS(a))
        return a;
 
    nbytes = ARR_DATA_OFFSET(a) + sizeof(int) * num;
 
    a = (ArrayType *) repalloc(a, nbytes);
 
    SET_VARSIZE(a, nbytes);
    /* usually the array should be 1-D already, but just in case ... */
    for (i = 0; i < ARR_NDIM(a); i++)
    {
        ARR_DIMS(a)[i] = num;
        num = 1;
    }
    return a;
}

References a, ARR_DATA_OFFSET, ARR_DIMS, ARR_NDIM, ARRNELEMS, Assert(), construct_empty_array(), i, repalloc(), and SET_VARSIZE.

Referenced by _int_unique(), g_int_compress(), inner_int_inter(), inner_int_union(), intarray_del_elem(), and intset_subtract().

◆ rt__int_size()

void rt__int_size	(	ArrayType *	a,
		float *	size
	)

Definition at line 184 of file _int_tool.c.

{
    *size = (float) ARRNELEMS(a);
}

References a, and ARRNELEMS.

Referenced by g_int_penalty(), and g_int_picksplit().

Macros

Functions