btree_enum.c File Reference

#include "postgres.h"
#include "btree_gist.h"
#include "btree_utils_num.h"
#include "fmgr.h"
#include "utils/fmgrprotos.h"
#include "utils/fmgroids.h"
#include "utils/rel.h"
#include "utils/sortsupport.h"

Include dependency graph for btree_enum.c:

Go to the source code of this file.

Data Structures
struct	oidKEY

Functions
	PG_FUNCTION_INFO_V1 (gbt_enum_compress)
	PG_FUNCTION_INFO_V1 (gbt_enum_fetch)
	PG_FUNCTION_INFO_V1 (gbt_enum_union)
	PG_FUNCTION_INFO_V1 (gbt_enum_picksplit)
	PG_FUNCTION_INFO_V1 (gbt_enum_consistent)
	PG_FUNCTION_INFO_V1 (gbt_enum_penalty)
	PG_FUNCTION_INFO_V1 (gbt_enum_same)
	PG_FUNCTION_INFO_V1 (gbt_enum_sortsupport)
static bool	gbt_enumgt (const void *a, const void *b, FmgrInfo *flinfo)
static bool	gbt_enumge (const void *a, const void *b, FmgrInfo *flinfo)
static bool	gbt_enumeq (const void *a, const void *b, FmgrInfo *flinfo)
static bool	gbt_enumle (const void *a, const void *b, FmgrInfo *flinfo)
static bool	gbt_enumlt (const void *a, const void *b, FmgrInfo *flinfo)
static int	gbt_enumkey_cmp (const void *a, const void *b, FmgrInfo *flinfo)
Datum	gbt_enum_compress (PG_FUNCTION_ARGS)
Datum	gbt_enum_fetch (PG_FUNCTION_ARGS)
Datum	gbt_enum_consistent (PG_FUNCTION_ARGS)
Datum	gbt_enum_union (PG_FUNCTION_ARGS)
Datum	gbt_enum_penalty (PG_FUNCTION_ARGS)
Datum	gbt_enum_picksplit (PG_FUNCTION_ARGS)
Datum	gbt_enum_same (PG_FUNCTION_ARGS)
static int	gbt_enum_ssup_cmp (Datum x, Datum y, SortSupport ssup)
Datum	gbt_enum_sortsupport (PG_FUNCTION_ARGS)

Variables
static const gbtree_ninfo	tinfo

Function Documentation

gbt_enum_compress()

Datum gbt_enum_compress

(

PG_FUNCTION_ARGS

)

Definition at line 108 of file btree_enum.c.

{
    GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 
    PG_RETURN_POINTER(gbt_num_compress(entry, &tinfo));
}

References gbt_num_compress(), PG_GETARG_POINTER, PG_RETURN_POINTER, and tinfo.

gbt_enum_consistent()

Datum gbt_enum_consistent

(

PG_FUNCTION_ARGS

)

Definition at line 124 of file btree_enum.c.

{
    GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
    Oid         query = PG_GETARG_OID(1);
    StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
#ifdef NOT_USED
    Oid         subtype = PG_GETARG_OID(3);
#endif
    bool       *recheck = (bool *) PG_GETARG_POINTER(4);
    oidKEY     *kkk = (oidKEY *) DatumGetPointer(entry->key);
    GBT_NUMKEY_R key;
 
    /* All cases served by this function are exact */
    *recheck = false;
 
    key.lower = (GBT_NUMKEY *) &kkk->lower;
    key.upper = (GBT_NUMKEY *) &kkk->upper;
 
    PG_RETURN_BOOL(gbt_num_consistent(&key, &query, &strategy,
                                      GIST_LEAF(entry), &tinfo,
                                      fcinfo->flinfo));
}

References DatumGetPointer(), fb(), gbt_num_consistent(), GIST_LEAF, GISTENTRY::key, PG_GETARG_OID, PG_GETARG_POINTER, PG_GETARG_UINT16, PG_RETURN_BOOL, and tinfo.

gbt_enum_fetch()

Datum gbt_enum_fetch

(

PG_FUNCTION_ARGS

)

Definition at line 116 of file btree_enum.c.

{
    GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 
    PG_RETURN_POINTER(gbt_num_fetch(entry, &tinfo));
}

References gbt_num_fetch(), PG_GETARG_POINTER, PG_RETURN_POINTER, and tinfo.

gbt_enum_penalty()

Datum gbt_enum_penalty

(

PG_FUNCTION_ARGS

)

Definition at line 158 of file btree_enum.c.

{
    oidKEY     *origentry = (oidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
    oidKEY     *newentry = (oidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
    float      *result = (float *) PG_GETARG_POINTER(2);
 
    penalty_num(result, origentry->lower, origentry->upper, newentry->lower, newentry->upper);
 
    PG_RETURN_POINTER(result);
}

References DatumGetPointer(), fb(), penalty_num, PG_GETARG_POINTER, and PG_RETURN_POINTER.

gbt_enum_picksplit()

Datum gbt_enum_picksplit

(

PG_FUNCTION_ARGS

)

Definition at line 170 of file btree_enum.c.

{
    PG_RETURN_POINTER(gbt_num_picksplit((GistEntryVector *) PG_GETARG_POINTER(0),
                                        (GIST_SPLITVEC *) PG_GETARG_POINTER(1),
                                        &tinfo, fcinfo->flinfo));
}

References gbt_num_picksplit(), PG_GETARG_POINTER, PG_RETURN_POINTER, and tinfo.

gbt_enum_same()

Datum gbt_enum_same

(

PG_FUNCTION_ARGS

)

Definition at line 178 of file btree_enum.c.

{
    oidKEY     *b1 = (oidKEY *) PG_GETARG_POINTER(0);
    oidKEY     *b2 = (oidKEY *) PG_GETARG_POINTER(1);
    bool       *result = (bool *) PG_GETARG_POINTER(2);
 
    *result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
    PG_RETURN_POINTER(result);
}

References fb(), gbt_num_same(), PG_GETARG_POINTER, PG_RETURN_POINTER, and tinfo.

gbt_enum_sortsupport()

Datum gbt_enum_sortsupport

(

PG_FUNCTION_ARGS

)

Definition at line 203 of file btree_enum.c.

{
    SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
    FmgrInfo   *flinfo;
 
    ssup->comparator = gbt_enum_ssup_cmp;
 
    /*
     * Since gbt_enum_ssup_cmp() uses enum_cmp() like the rest of the
     * comparison functions, it also needs to pass flinfo when calling it. The
     * caller to a SortSupport comparison function doesn't provide an FmgrInfo
     * struct, so look it up now, save it in ssup_extra and use it in
     * gbt_enum_ssup_cmp() later.
     */
    flinfo = MemoryContextAlloc(ssup->ssup_cxt, sizeof(FmgrInfo));
    fmgr_info_cxt(F_ENUM_CMP, flinfo, ssup->ssup_cxt);
    ssup->ssup_extra = flinfo;
 
    PG_RETURN_VOID();
}

References SortSupportData::comparator, fb(), fmgr_info_cxt(), gbt_enum_ssup_cmp(), MemoryContextAlloc(), PG_GETARG_POINTER, PG_RETURN_VOID, SortSupportData::ssup_cxt, and SortSupportData::ssup_extra.

gbt_enum_ssup_cmp()

static int gbt_enum_ssup_cmp ( Datum  x, Datum  y, SortSupport  ssup  )

static

Definition at line 189 of file btree_enum.c.

{
    oidKEY     *arg1 = (oidKEY *) DatumGetPointer(x);
    oidKEY     *arg2 = (oidKEY *) DatumGetPointer(y);
 
    /* for leaf items we expect lower == upper, so only compare lower */
    return DatumGetInt32(CallerFInfoFunctionCall2(enum_cmp,
                                                  ssup->ssup_extra,
                                                  InvalidOid,
                                                  ObjectIdGetDatum(arg1->lower),
                                                  ObjectIdGetDatum(arg2->lower)));
}

References CallerFInfoFunctionCall2(), DatumGetInt32(), DatumGetPointer(), enum_cmp(), fb(), InvalidOid, ObjectIdGetDatum(), SortSupportData::ssup_extra, x, and y.

Referenced by gbt_enum_sortsupport().

gbt_enum_union()

Datum gbt_enum_union

(

PG_FUNCTION_ARGS

)

Definition at line 148 of file btree_enum.c.

{
    GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
    void       *out = palloc(sizeof(oidKEY));
 
    *(int *) PG_GETARG_POINTER(1) = sizeof(oidKEY);
    PG_RETURN_POINTER(gbt_num_union(out, entryvec, &tinfo, fcinfo->flinfo));
}

References fb(), gbt_num_union(), palloc(), PG_GETARG_POINTER, PG_RETURN_POINTER, and tinfo.

gbt_enumeq()

static bool gbt_enumeq ( const void *  a, const void *  b, FmgrInfo *  flinfo  )

static

Definition at line 48 of file btree_enum.c.

{
    return (*((const Oid *) a) == *((const Oid *) b));
}

References a, and b.

gbt_enumge()

static bool gbt_enumge ( const void *  a, const void *  b, FmgrInfo *  flinfo  )

static

Definition at line 41 of file btree_enum.c.

{
    return DatumGetBool(CallerFInfoFunctionCall2(enum_ge, flinfo, InvalidOid,
                                                 ObjectIdGetDatum(*((const Oid *) a)),
                                                 ObjectIdGetDatum(*((const Oid *) b))));
}

References a, b, CallerFInfoFunctionCall2(), DatumGetBool(), enum_ge(), InvalidOid, and ObjectIdGetDatum().

gbt_enumgt()

static bool gbt_enumgt ( const void *  a, const void *  b, FmgrInfo *  flinfo  )

static

Definition at line 34 of file btree_enum.c.

{
    return DatumGetBool(CallerFInfoFunctionCall2(enum_gt, flinfo, InvalidOid,
                                                 ObjectIdGetDatum(*((const Oid *) a)),
                                                 ObjectIdGetDatum(*((const Oid *) b))));
}

References a, b, CallerFInfoFunctionCall2(), DatumGetBool(), enum_gt(), InvalidOid, and ObjectIdGetDatum().

gbt_enumkey_cmp()

static int gbt_enumkey_cmp ( const void *  a, const void *  b, FmgrInfo *  flinfo  )

static

Definition at line 68 of file btree_enum.c.

{
    oidKEY     *ia = (oidKEY *) (((const Nsrt *) a)->t);
    oidKEY     *ib = (oidKEY *) (((const Nsrt *) b)->t);
 
    if (ia->lower == ib->lower)
    {
        if (ia->upper == ib->upper)
            return 0;
 
        return DatumGetInt32(CallerFInfoFunctionCall2(enum_cmp, flinfo, InvalidOid,
                                                      ObjectIdGetDatum(ia->upper),
                                                      ObjectIdGetDatum(ib->upper)));
    }
 
    return DatumGetInt32(CallerFInfoFunctionCall2(enum_cmp, flinfo, InvalidOid,
                                                  ObjectIdGetDatum(ia->lower),
                                                  ObjectIdGetDatum(ib->lower)));
}

References a, b, CallerFInfoFunctionCall2(), DatumGetInt32(), enum_cmp(), fb(), InvalidOid, and ObjectIdGetDatum().

gbt_enumle()

static bool gbt_enumle ( const void *  a, const void *  b, FmgrInfo *  flinfo  )

static

Definition at line 53 of file btree_enum.c.

{
    return DatumGetBool(CallerFInfoFunctionCall2(enum_le, flinfo, InvalidOid,
                                                 ObjectIdGetDatum(*((const Oid *) a)),
                                                 ObjectIdGetDatum(*((const Oid *) b))));
}

References a, b, CallerFInfoFunctionCall2(), DatumGetBool(), enum_le(), InvalidOid, and ObjectIdGetDatum().

gbt_enumlt()

static bool gbt_enumlt ( const void *  a, const void *  b, FmgrInfo *  flinfo  )

static

Definition at line 60 of file btree_enum.c.

{
    return DatumGetBool(CallerFInfoFunctionCall2(enum_lt, flinfo, InvalidOid,
                                                 ObjectIdGetDatum(*((const Oid *) a)),
                                                 ObjectIdGetDatum(*((const Oid *) b))));
}

References a, b, CallerFInfoFunctionCall2(), DatumGetBool(), enum_lt(), InvalidOid, and ObjectIdGetDatum().

PG_FUNCTION_INFO_V1() [1/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_compress

)

PG_FUNCTION_INFO_V1() [2/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_consistent

)

PG_FUNCTION_INFO_V1() [3/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_fetch

)

PG_FUNCTION_INFO_V1() [4/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_penalty

)

PG_FUNCTION_INFO_V1() [5/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_picksplit

)

PG_FUNCTION_INFO_V1() [6/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_same

)

PG_FUNCTION_INFO_V1() [7/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_sortsupport

)

PG_FUNCTION_INFO_V1() [8/8]

PG_FUNCTION_INFO_V1

(

gbt_enum_union

)

Variable Documentation

tinfo

const gbtree_ninfo tinfo

static

Initial value:

=
{
    gbt_t_enum,
    sizeof(Oid),
    8,                          
    gbt_enumgt,
    gbt_enumge,
    gbt_enumeq,
    gbt_enumle,
    gbt_enumlt,
    gbt_enumkey_cmp,
    NULL                        
}

Definition at line 88 of file btree_enum.c.

{
    gbt_t_enum,
    sizeof(Oid),
    8,                          /* sizeof(gbtreekey8) */
    gbt_enumgt,
    gbt_enumge,
    gbt_enumeq,
    gbt_enumle,
    gbt_enumlt,
    gbt_enumkey_cmp,
    NULL                        /* no KNN support at least for now */
};

Referenced by gbt_enum_compress(), gbt_enum_consistent(), gbt_enum_fetch(), gbt_enum_picksplit(), gbt_enum_same(), and gbt_enum_union().