tsquery_gist.c File Reference

#include "postgres.h"
#include "access/gist.h"
#include "access/stratnum.h"
#include "common/int.h"
#include "tsearch/ts_utils.h"
#include "utils/fmgrprotos.h"

Include dependency graph for tsquery_gist.c:

Go to the source code of this file.

Data Structures
struct	SPLITCOST

Macros
#define	GETENTRY(vec, pos)   DatumGetTSQuerySign((vec)->vector[pos].key)
#define	WISH_F(a, b, c)   (double)( -(double)(((a)-(b))*((a)-(b))*((a)-(b)))*(c) )

Functions
Datum	gtsquery_compress (PG_FUNCTION_ARGS)
Datum	gtsquery_consistent (PG_FUNCTION_ARGS)
Datum	gtsquery_union (PG_FUNCTION_ARGS)
Datum	gtsquery_same (PG_FUNCTION_ARGS)
static int	sizebitvec (TSQuerySign sign)
static int	hemdist (TSQuerySign a, TSQuerySign b)
Datum	gtsquery_penalty (PG_FUNCTION_ARGS)
static int	comparecost (const void *a, const void *b)
Datum	gtsquery_picksplit (PG_FUNCTION_ARGS)
Datum	gtsquery_consistent_oldsig (PG_FUNCTION_ARGS)

Macro Definition Documentation

GETENTRY

#define GETENTRY	(		vec,
			pos
	)		DatumGetTSQuerySign((vec)->vector[pos].key)

Definition at line 23 of file tsquery_gist.c.

WISH_F

#define WISH_F	(		a,
			b,
			c
	)		(double)( -(double)(((a)-(b))*((a)-(b))*((a)-(b)))*(c) )

Definition at line 164 of file tsquery_gist.c.

Function Documentation

comparecost()

static int comparecost ( const void *  a, const void *  b  )

static

Definition at line 158 of file tsquery_gist.c.

{
    return pg_cmp_s32(((const SPLITCOST *) a)->cost,
                      ((const SPLITCOST *) b)->cost);
}

References a, b, and pg_cmp_s32().

Referenced by gtsquery_picksplit().

gtsquery_compress()

Datum gtsquery_compress

(

PG_FUNCTION_ARGS

)

Definition at line 27 of file tsquery_gist.c.

{
    GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
    GISTENTRY  *retval = entry;
 
    if (entry->leafkey)
    {
        TSQuerySign sign;
 
        retval = (GISTENTRY *) palloc(sizeof(GISTENTRY));
        sign = makeTSQuerySign(DatumGetTSQuery(entry->key));
 
        gistentryinit(*retval, TSQuerySignGetDatum(sign),
                      entry->rel, entry->page,
                      entry->offset, false);
    }
 
    PG_RETURN_POINTER(retval);
}

References DatumGetTSQuery(), gistentryinit, GISTENTRY::key, GISTENTRY::leafkey, makeTSQuerySign(), GISTENTRY::offset, GISTENTRY::page, palloc(), PG_GETARG_POINTER, PG_RETURN_POINTER, GISTENTRY::rel, sign, and TSQuerySignGetDatum().

gtsquery_consistent()

Datum gtsquery_consistent

(

PG_FUNCTION_ARGS

)

Definition at line 53 of file tsquery_gist.c.

{
    GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
    TSQuery     query = PG_GETARG_TSQUERY(1);
    StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
 
    /* Oid      subtype = PG_GETARG_OID(3); */
    bool       *recheck = (bool *) PG_GETARG_POINTER(4);
    TSQuerySign key = DatumGetTSQuerySign(entry->key);
    TSQuerySign sq = makeTSQuerySign(query);
    bool        retval;
 
    /* All cases served by this function are inexact */
    *recheck = true;
 
    switch (strategy)
    {
        case RTContainsStrategyNumber:
            if (GIST_LEAF(entry))
                retval = (key & sq) == sq;
            else
                retval = (key & sq) != 0;
            break;
        case RTContainedByStrategyNumber:
            if (GIST_LEAF(entry))
                retval = (key & sq) == key;
            else
                retval = (key & sq) != 0;
            break;
        default:
            retval = false;
    }
    PG_RETURN_BOOL(retval);
}

References DatumGetTSQuerySign(), GIST_LEAF, GISTENTRY::key, sort-test::key, makeTSQuerySign(), PG_GETARG_POINTER, PG_GETARG_TSQUERY, PG_GETARG_UINT16, PG_RETURN_BOOL, RTContainedByStrategyNumber, and RTContainsStrategyNumber.

Referenced by gtsquery_consistent_oldsig().

gtsquery_consistent_oldsig()

Datum gtsquery_consistent_oldsig

(

PG_FUNCTION_ARGS

)

Definition at line 273 of file tsquery_gist.c.

{
    return gtsquery_consistent(fcinfo);
}

References gtsquery_consistent().

gtsquery_penalty()

Datum gtsquery_penalty

(

PG_FUNCTION_ARGS

)

Definition at line 139 of file tsquery_gist.c.

{
    TSQuerySign origval = DatumGetTSQuerySign(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
    TSQuerySign newval = DatumGetTSQuerySign(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
    float      *penalty = (float *) PG_GETARG_POINTER(2);
 
    *penalty = hemdist(origval, newval);
 
    PG_RETURN_POINTER(penalty);
}

References DatumGetTSQuerySign(), hemdist(), sort-test::key, newval, PG_GETARG_POINTER, and PG_RETURN_POINTER.

gtsquery_picksplit()

Datum gtsquery_picksplit

(

PG_FUNCTION_ARGS

)

Definition at line 167 of file tsquery_gist.c.

{
    GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
    GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
    OffsetNumber maxoff = entryvec->n - 2;
    OffsetNumber k,
                j;
    TSQuerySign datum_l,
                datum_r;
    int32       size_alpha,
                size_beta;
    int32       size_waste,
                waste = -1;
    int32       nbytes;
    OffsetNumber seed_1 = 0,
                seed_2 = 0;
    OffsetNumber *left,
               *right;
 
    SPLITCOST  *costvector;
 
    nbytes = (maxoff + 2) * sizeof(OffsetNumber);
    left = v->spl_left = (OffsetNumber *) palloc(nbytes);
    right = v->spl_right = (OffsetNumber *) palloc(nbytes);
    v->spl_nleft = v->spl_nright = 0;
 
    for (k = FirstOffsetNumber; k < maxoff; k = OffsetNumberNext(k))
        for (j = OffsetNumberNext(k); j <= maxoff; j = OffsetNumberNext(j))
        {
            size_waste = hemdist(GETENTRY(entryvec, j), GETENTRY(entryvec, k));
            if (size_waste > waste)
            {
                waste = size_waste;
                seed_1 = k;
                seed_2 = j;
            }
        }
 
 
    if (seed_1 == 0 || seed_2 == 0)
    {
        seed_1 = 1;
        seed_2 = 2;
    }
 
    datum_l = GETENTRY(entryvec, seed_1);
    datum_r = GETENTRY(entryvec, seed_2);
 
    maxoff = OffsetNumberNext(maxoff);
    costvector = (SPLITCOST *) palloc(sizeof(SPLITCOST) * maxoff);
    for (j = FirstOffsetNumber; j <= maxoff; j = OffsetNumberNext(j))
    {
        costvector[j - 1].pos = j;
        size_alpha = hemdist(GETENTRY(entryvec, seed_1), GETENTRY(entryvec, j));
        size_beta = hemdist(GETENTRY(entryvec, seed_2), GETENTRY(entryvec, j));
        costvector[j - 1].cost = abs(size_alpha - size_beta);
    }
    qsort(costvector, maxoff, sizeof(SPLITCOST), comparecost);
 
    for (k = 0; k < maxoff; k++)
    {
        j = costvector[k].pos;
        if (j == seed_1)
        {
            *left++ = j;
            v->spl_nleft++;
            continue;
        }
        else if (j == seed_2)
        {
            *right++ = j;
            v->spl_nright++;
            continue;
        }
        size_alpha = hemdist(datum_l, GETENTRY(entryvec, j));
        size_beta = hemdist(datum_r, GETENTRY(entryvec, j));
 
        if (size_alpha < size_beta + WISH_F(v->spl_nleft, v->spl_nright, 0.05))
        {
            datum_l |= GETENTRY(entryvec, j);
            *left++ = j;
            v->spl_nleft++;
        }
        else
        {
            datum_r |= GETENTRY(entryvec, j);
            *right++ = j;
            v->spl_nright++;
        }
    }
 
    *right = *left = FirstOffsetNumber;
    v->spl_ldatum = TSQuerySignGetDatum(datum_l);
    v->spl_rdatum = TSQuerySignGetDatum(datum_r);
 
    PG_RETURN_POINTER(v);
}

References comparecost(), SPLITCOST::cost, FirstOffsetNumber, GETENTRY, hemdist(), j, GistEntryVector::n, OffsetNumberNext, palloc(), PG_GETARG_POINTER, PG_RETURN_POINTER, SPLITCOST::pos, qsort, GIST_SPLITVEC::spl_ldatum, GIST_SPLITVEC::spl_left, GIST_SPLITVEC::spl_nleft, GIST_SPLITVEC::spl_nright, GIST_SPLITVEC::spl_rdatum, GIST_SPLITVEC::spl_right, TSQuerySignGetDatum(), and WISH_F.

gtsquery_same()

Datum gtsquery_same

(

PG_FUNCTION_ARGS

)

Definition at line 107 of file tsquery_gist.c.

{
    TSQuerySign a = PG_GETARG_TSQUERYSIGN(0);
    TSQuerySign b = PG_GETARG_TSQUERYSIGN(1);
    bool       *result = (bool *) PG_GETARG_POINTER(2);
 
    *result = (a == b);
 
    PG_RETURN_POINTER(result);
}

References a, b, PG_GETARG_POINTER, PG_GETARG_TSQUERYSIGN, and PG_RETURN_POINTER.

gtsquery_union()

Datum gtsquery_union

(

PG_FUNCTION_ARGS

)

Definition at line 89 of file tsquery_gist.c.

{
    GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
    int        *size = (int *) PG_GETARG_POINTER(1);
    TSQuerySign sign;
    int         i;
 
    sign = 0;
 
    for (i = 0; i < entryvec->n; i++)
        sign |= GETENTRY(entryvec, i);
 
    *size = sizeof(TSQuerySign);
 
    PG_RETURN_TSQUERYSIGN(sign);
}

References GETENTRY, i, GistEntryVector::n, PG_GETARG_POINTER, PG_RETURN_TSQUERYSIGN, sign, and size.

hemdist()

static int hemdist ( TSQuerySign  a, TSQuerySign  b  )

static

Definition at line 131 of file tsquery_gist.c.

{
    TSQuerySign res = a ^ b;
 
    return sizebitvec(res);
}

References a, b, res, and sizebitvec().

Referenced by gtsquery_penalty(), and gtsquery_picksplit().

sizebitvec()

static int sizebitvec ( TSQuerySign  sign )

static

Definition at line 119 of file tsquery_gist.c.

{
    int         size = 0,
                i;
 
    for (i = 0; i < TSQS_SIGLEN; i++)
        size += 0x01 & (sign >> i);
 
    return size;
}

References i, sign, size, and TSQS_SIGLEN.

Referenced by hemdist().