contrib/btree_gist/btree_int8.c

   1 /*
   2  * $PostgreSQL:$
   3  */
   4 #include "btree_gist.h"
   5 #include "btree_utils_num.h"
   6
   7 typedef struct int64key
   8 {
   9         int64           lower;
  10         int64           upper;
  11 }       int64KEY;
  12
  13 /*
  14 ** int64 ops
  15 */
  16 PG_FUNCTION_INFO_V1(gbt_int8_compress);
  17 PG_FUNCTION_INFO_V1(gbt_int8_union);
  18 PG_FUNCTION_INFO_V1(gbt_int8_picksplit);
  19 PG_FUNCTION_INFO_V1(gbt_int8_consistent);
  20 PG_FUNCTION_INFO_V1(gbt_int8_penalty);
  21 PG_FUNCTION_INFO_V1(gbt_int8_same);
  22
  23 Datum           gbt_int8_compress(PG_FUNCTION_ARGS);
  24 Datum           gbt_int8_union(PG_FUNCTION_ARGS);
  25 Datum           gbt_int8_picksplit(PG_FUNCTION_ARGS);
  26 Datum           gbt_int8_consistent(PG_FUNCTION_ARGS);
  27 Datum           gbt_int8_penalty(PG_FUNCTION_ARGS);
  28 Datum           gbt_int8_same(PG_FUNCTION_ARGS);
  29
  30
  31 static bool
  32 gbt_int8gt(const void *a, const void *b)
  33 {
  34         return (*((int64 *) a) > *((int64 *) b));
  35 }
  36 static bool
  37 gbt_int8ge(const void *a, const void *b)
  38 {
  39         return (*((int64 *) a) >= *((int64 *) b));
  40 }
  41 static bool
  42 gbt_int8eq(const void *a, const void *b)
  43 {
  44         return (*((int64 *) a) == *((int64 *) b));
  45 }
  46 static bool
  47 gbt_int8le(const void *a, const void *b)
  48 {
  49         return (*((int64 *) a) <= *((int64 *) b));
  50 }
  51 static bool
  52 gbt_int8lt(const void *a, const void *b)
  53 {
  54         return (*((int64 *) a) < *((int64 *) b));
  55 }
  56
  57 static int
  58 gbt_int8key_cmp(const void *a, const void *b)
  59 {
  60
  61         if (*(int64 *) &(((Nsrt *) a)->t[0]) > *(int64 *) &(((Nsrt *) b)->t[0]))
  62                 return 1;
  63         else if (*(int64 *) &(((Nsrt *) a)->t[0]) < *(int64 *) &(((Nsrt *) b)->t[0]))
  64                 return -1;
  65         return 0;
  66
  67 }
  68
  69
  70 static const gbtree_ninfo tinfo =
  71 {
  72         gbt_t_int8,
  73         sizeof(int64),
  74         gbt_int8gt,
  75         gbt_int8ge,
  76         gbt_int8eq,
  77         gbt_int8le,
  78         gbt_int8lt,
  79         gbt_int8key_cmp
  80 };
  81
  82
  83 /**************************************************
  84  * int64 ops
  85  **************************************************/
  86
  87
  88 Datum
  89 gbt_int8_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_int8_consistent(PG_FUNCTION_ARGS)
 100 {
 101         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 102         int64           query = PG_GETARG_INT64(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         int64KEY   *kkk = (int64KEY *) 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_int8_union(PG_FUNCTION_ARGS)
 123 {
 124         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 125         void       *out = palloc(sizeof(int64KEY));
 126
 127         *(int *) PG_GETARG_POINTER(1) = sizeof(int64KEY);
 128         PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo));
 129 }
 130
 131
 132 Datum
 133 gbt_int8_penalty(PG_FUNCTION_ARGS)
 134 {
 135         int64KEY   *origentry = (int64KEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
 136         int64KEY   *newentry = (int64KEY *) 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_int8_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_int8_same(PG_FUNCTION_ARGS)
 156 {
 157         int64KEY   *b1 = (int64KEY *) PG_GETARG_POINTER(0);
 158         int64KEY   *b2 = (int64KEY *) 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 }