contrib/btree_gist/btree_int4.c

   1 /*
   2  * $PostgreSQL:$
   3  */
   4 #include "btree_gist.h"
   5 #include "btree_utils_num.h"
   6
   7 typedef struct int32key
   8 {
   9         int32           lower;
  10         int32           upper;
  11 }       int32KEY;
  12
  13 /*
  14 ** int32 ops
  15 */
  16 PG_FUNCTION_INFO_V1(gbt_int4_compress);
  17 PG_FUNCTION_INFO_V1(gbt_int4_union);
  18 PG_FUNCTION_INFO_V1(gbt_int4_picksplit);
  19 PG_FUNCTION_INFO_V1(gbt_int4_consistent);
  20 PG_FUNCTION_INFO_V1(gbt_int4_penalty);
  21 PG_FUNCTION_INFO_V1(gbt_int4_same);
  22
  23 Datum           gbt_int4_compress(PG_FUNCTION_ARGS);
  24 Datum           gbt_int4_union(PG_FUNCTION_ARGS);
  25 Datum           gbt_int4_picksplit(PG_FUNCTION_ARGS);
  26 Datum           gbt_int4_consistent(PG_FUNCTION_ARGS);
  27 Datum           gbt_int4_penalty(PG_FUNCTION_ARGS);
  28 Datum           gbt_int4_same(PG_FUNCTION_ARGS);
  29
  30
  31 static bool
  32 gbt_int4gt(const void *a, const void *b)
  33 {
  34         return (*((int32 *) a) > *((int32 *) b));
  35 }
  36 static bool
  37 gbt_int4ge(const void *a, const void *b)
  38 {
  39         return (*((int32 *) a) >= *((int32 *) b));
  40 }
  41 static bool
  42 gbt_int4eq(const void *a, const void *b)
  43 {
  44         return (*((int32 *) a) == *((int32 *) b));
  45 }
  46 static bool
  47 gbt_int4le(const void *a, const void *b)
  48 {
  49         return (*((int32 *) a) <= *((int32 *) b));
  50 }
  51 static bool
  52 gbt_int4lt(const void *a, const void *b)
  53 {
  54         return (*((int32 *) a) < *((int32 *) b));
  55 }
  56
  57 static int
  58 gbt_int4key_cmp(const void *a, const void *b)
  59 {
  60
  61         if (*(int32 *) &(((Nsrt *) a)->t[0]) > *(int32 *) &(((Nsrt *) b)->t[0]))
  62                 return 1;
  63         else if (*(int32 *) &(((Nsrt *) a)->t[0]) < *(int32 *) &(((Nsrt *) b)->t[0]))
  64                 return -1;
  65         return 0;
  66
  67 }
  68
  69
  70 static const gbtree_ninfo tinfo =
  71 {
  72         gbt_t_int4,
  73         sizeof(int32),
  74         gbt_int4gt,
  75         gbt_int4ge,
  76         gbt_int4eq,
  77         gbt_int4le,
  78         gbt_int4lt,
  79         gbt_int4key_cmp
  80 };
  81
  82
  83 /**************************************************
  84  * int32 ops
  85  **************************************************/
  86
  87
  88 Datum
  89 gbt_int4_compress(PG_FUNCTION_ARGS)
  90 {
  91         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
  92         GISTENTRY  *retval = NULL;
  93
  94         PG_RETURN_POINTER(gbt_num_compress(retval, entry, &tinfo));
  95 }
  96
  97
  98 Datum
  99 gbt_int4_consistent(PG_FUNCTION_ARGS)
 100 {
 101         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 102         int32           query = PG_GETARG_INT32(1);
 103         StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
 104         /* Oid          subtype = PG_GETARG_OID(3); */
 105         bool       *recheck = (bool *) PG_GETARG_POINTER(4);
 106         int32KEY   *kkk = (int32KEY *) DatumGetPointer(entry->key);
 107         GBT_NUMKEY_R key;
 108
 109         /* All cases served by this function are exact */
 110         *recheck = false;
 111
 112         key.lower = (GBT_NUMKEY *) & kkk->lower;
 113         key.upper = (GBT_NUMKEY *) & kkk->upper;
 114
 115         PG_RETURN_BOOL(
 116                                    gbt_num_consistent(&key, (void *) &query, &strategy, GIST_LEAF(entry), &tinfo)
 117                 );
 118 }
 119
 120
 121 Datum
 122 gbt_int4_union(PG_FUNCTION_ARGS)
 123 {
 124         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 125         void       *out = palloc(sizeof(int32KEY));
 126
 127         *(int *) PG_GETARG_POINTER(1) = sizeof(int32KEY);
 128         PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo));
 129 }
 130
 131
 132 Datum
 133 gbt_int4_penalty(PG_FUNCTION_ARGS)
 134 {
 135         int32KEY   *origentry = (int32KEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
 136         int32KEY   *newentry = (int32KEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
 137         float      *result = (float *) PG_GETARG_POINTER(2);
 138
 139         penalty_num(result, origentry->lower, origentry->upper, newentry->lower, newentry->upper);
 140
 141         PG_RETURN_POINTER(result);
 142 }
 143
 144 Datum
 145 gbt_int4_picksplit(PG_FUNCTION_ARGS)
 146 {
 147         PG_RETURN_POINTER(gbt_num_picksplit(
 148                                                                         (GistEntryVector *) PG_GETARG_POINTER(0),
 149                                                                           (GIST_SPLITVEC *) PG_GETARG_POINTER(1),
 150                                                                                 &tinfo
 151                                                                                 ));
 152 }
 153
 154 Datum
 155 gbt_int4_same(PG_FUNCTION_ARGS)
 156 {
 157         int32KEY   *b1 = (int32KEY *) PG_GETARG_POINTER(0);
 158         int32KEY   *b2 = (int32KEY *) PG_GETARG_POINTER(1);
 159         bool       *result = (bool *) PG_GETARG_POINTER(2);
 160
 161         *result = gbt_num_same((void *) b1, (void *) b2, &tinfo);
 162         PG_RETURN_POINTER(result);
 163 }