mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-12-21 08:29:39 +08:00
8436489c81
This extends GiST's support for nearest-neighbor searches to many of the standard data types. Teodor Sigaev
143 lines
3.8 KiB
C
143 lines
3.8 KiB
C
/*
|
|
* contrib/btree_gist/btree_utils_num.h
|
|
*/
|
|
#ifndef __BTREE_UTILS_NUM_H__
|
|
#define __BTREE_UTILS_NUM_H__
|
|
|
|
#include "btree_gist.h"
|
|
#include "utils/rel.h"
|
|
|
|
#include <math.h>
|
|
#include <float.h>
|
|
|
|
typedef char GBT_NUMKEY;
|
|
|
|
/* Better readable key */
|
|
typedef struct
|
|
{
|
|
const GBT_NUMKEY *lower,
|
|
*upper;
|
|
} GBT_NUMKEY_R;
|
|
|
|
|
|
/* for sorting */
|
|
typedef struct
|
|
{
|
|
int i;
|
|
GBT_NUMKEY *t;
|
|
} Nsrt;
|
|
|
|
|
|
/* type description */
|
|
|
|
typedef struct
|
|
{
|
|
|
|
/* Attribs */
|
|
|
|
enum gbtree_type t; /* data type */
|
|
int32 size; /* size of type , 0 means variable */
|
|
|
|
/* Methods */
|
|
|
|
bool (*f_gt) (const void *, const void *); /* greater than */
|
|
bool (*f_ge) (const void *, const void *); /* greater or equal */
|
|
bool (*f_eq) (const void *, const void *); /* equal */
|
|
bool (*f_le) (const void *, const void *); /* less or equal */
|
|
bool (*f_lt) (const void *, const void *); /* less than */
|
|
int (*f_cmp) (const void *, const void *); /* key compare function */
|
|
float8 (*f_dist) (const void *, const void *); /* key distance function */
|
|
} gbtree_ninfo;
|
|
|
|
|
|
/*
|
|
* Numeric btree functions
|
|
*/
|
|
|
|
|
|
|
|
/*
|
|
* Note: The factor 0.49 in following macro avoids floating point overflows
|
|
*/
|
|
#define penalty_num(result,olower,oupper,nlower,nupper) do { \
|
|
double tmp = 0.0F; \
|
|
(*(result)) = 0.0F; \
|
|
if ( (nupper) > (oupper) ) \
|
|
tmp += ( ((double)nupper)*0.49F - ((double)oupper)*0.49F ); \
|
|
if ( (olower) > (nlower) ) \
|
|
tmp += ( ((double)olower)*0.49F - ((double)nlower)*0.49F ); \
|
|
if (tmp > 0.0F) \
|
|
{ \
|
|
(*(result)) += FLT_MIN; \
|
|
(*(result)) += (float) ( ((double)(tmp)) / ( (double)(tmp) + ( ((double)(oupper))*0.49F - ((double)(olower))*0.49F ) ) ); \
|
|
(*(result)) *= (FLT_MAX / (((GISTENTRY *) PG_GETARG_POINTER(0))->rel->rd_att->natts + 1)); \
|
|
} \
|
|
} while (0);
|
|
|
|
|
|
/*
|
|
* Convert an Interval to an approximate equivalent number of seconds
|
|
* (as a double). Here because we need it for time/timetz as well as
|
|
* interval. See interval_cmp_internal for comparison.
|
|
*/
|
|
#ifdef HAVE_INT64_TIMESTAMP
|
|
#define INTERVAL_TO_SEC(ivp) \
|
|
(((double) (ivp)->time) / ((double) USECS_PER_SEC) + \
|
|
(ivp)->day * (24.0 * SECS_PER_HOUR) + \
|
|
(ivp)->month * (30.0 * SECS_PER_DAY))
|
|
#else
|
|
#define INTERVAL_TO_SEC(ivp) \
|
|
((ivp)->time + \
|
|
(ivp)->day * (24.0 * SECS_PER_HOUR) + \
|
|
(ivp)->month * (30.0 * SECS_PER_DAY))
|
|
#endif
|
|
|
|
#define GET_FLOAT_DISTANCE(t, arg1, arg2) Abs( ((float8) *((const t *) (arg1))) - ((float8) *((const t *) (arg2))) )
|
|
|
|
#define SAMESIGN(a,b) (((a) < 0) == ((b) < 0))
|
|
|
|
/*
|
|
* check to see if a float4/8 val has underflowed or overflowed
|
|
* borrowed from src/backend/utils/adt/float.c
|
|
*/
|
|
#define CHECKFLOATVAL(val, inf_is_valid, zero_is_valid) \
|
|
do { \
|
|
if (isinf(val) && !(inf_is_valid)) \
|
|
ereport(ERROR, \
|
|
(errcode(ERRCODE_NUMERIC_VALUE_OUT_OF_RANGE), \
|
|
errmsg("value out of range: overflow"))); \
|
|
\
|
|
if ((val) == 0.0 && !(zero_is_valid)) \
|
|
ereport(ERROR, \
|
|
(errcode(ERRCODE_NUMERIC_VALUE_OUT_OF_RANGE), \
|
|
errmsg("value out of range: underflow"))); \
|
|
} while(0)
|
|
|
|
|
|
extern Interval *abs_interval(Interval *a);
|
|
|
|
extern bool gbt_num_consistent(const GBT_NUMKEY_R *key, const void *query,
|
|
const StrategyNumber *strategy, bool is_leaf,
|
|
const gbtree_ninfo *tinfo);
|
|
|
|
extern float8 gbt_num_distance(const GBT_NUMKEY_R *key, const void *query,
|
|
bool is_leaf, const gbtree_ninfo *tinfo);
|
|
|
|
extern GIST_SPLITVEC *gbt_num_picksplit(const GistEntryVector *entryvec, GIST_SPLITVEC *v,
|
|
const gbtree_ninfo *tinfo);
|
|
|
|
extern GISTENTRY *gbt_num_compress(GISTENTRY *retval, GISTENTRY *entry,
|
|
const gbtree_ninfo *tinfo);
|
|
|
|
|
|
extern void *gbt_num_union(GBT_NUMKEY *out, const GistEntryVector *entryvec,
|
|
const gbtree_ninfo *tinfo);
|
|
|
|
extern bool gbt_num_same(const GBT_NUMKEY *a, const GBT_NUMKEY *b,
|
|
const gbtree_ninfo *tinfo);
|
|
|
|
extern void gbt_num_bin_union(Datum *u, GBT_NUMKEY *e,
|
|
const gbtree_ninfo *tinfo);
|
|
|
|
#endif
|