contrib/btree_gist/btree_cash.c

   1 /*
   2  * $PostgreSQL:$
   3  */
   4 #include "btree_gist.h"
   5 #include "btree_utils_num.h"
   6 #include "utils/cash.h"
   7
   8 typedef struct
   9 {
  10         Cash            lower;
  11         Cash            upper;
  12 }       cashKEY;
  13
  14 /*
  15 ** Cash ops
  16 */
  17 PG_FUNCTION_INFO_V1(gbt_cash_compress);
  18 PG_FUNCTION_INFO_V1(gbt_cash_union);
  19 PG_FUNCTION_INFO_V1(gbt_cash_picksplit);
  20 PG_FUNCTION_INFO_V1(gbt_cash_consistent);
  21 PG_FUNCTION_INFO_V1(gbt_cash_penalty);
  22 PG_FUNCTION_INFO_V1(gbt_cash_same);
  23
  24 Datum           gbt_cash_compress(PG_FUNCTION_ARGS);
  25 Datum           gbt_cash_union(PG_FUNCTION_ARGS);
  26 Datum           gbt_cash_picksplit(PG_FUNCTION_ARGS);
  27 Datum           gbt_cash_consistent(PG_FUNCTION_ARGS);
  28 Datum           gbt_cash_penalty(PG_FUNCTION_ARGS);
  29 Datum           gbt_cash_same(PG_FUNCTION_ARGS);
  30
  31 static bool
  32 gbt_cashgt(const void *a, const void *b)
  33 {
  34         return (*((Cash *) a) > *((Cash *) b));
  35 }
  36 static bool
  37 gbt_cashge(const void *a, const void *b)
  38 {
  39         return (*((Cash *) a) >= *((Cash *) b));
  40 }
  41 static bool
  42 gbt_casheq(const void *a, const void *b)
  43 {
  44         return (*((Cash *) a) == *((Cash *) b));
  45 }
  46 static bool
  47 gbt_cashle(const void *a, const void *b)
  48 {
  49         return (*((Cash *) a) <= *((Cash *) b));
  50 }
  51 static bool
  52 gbt_cashlt(const void *a, const void *b)
  53 {
  54         return (*((Cash *) a) < *((Cash *) b));
  55 }
  56
  57 static int
  58 gbt_cashkey_cmp(const void *a, const void *b)
  59 {
  60
  61         if (*(Cash *) &(((Nsrt *) a)->t[0]) > *(Cash *) &(((Nsrt *) b)->t[0]))
  62                 return 1;
  63         else if (*(Cash *) &(((Nsrt *) a)->t[0]) < *(Cash *) &(((Nsrt *) b)->t[0]))
  64                 return -1;
  65         return 0;
  66
  67 }
  68
  69
  70 static const gbtree_ninfo tinfo =
  71 {
  72         gbt_t_cash,
  73         sizeof(Cash),
  74         gbt_cashgt,
  75         gbt_cashge,
  76         gbt_casheq,
  77         gbt_cashle,
  78         gbt_cashlt,
  79         gbt_cashkey_cmp
  80 };
  81
  82
  83 /**************************************************
  84  * Cash ops
  85  **************************************************/
  86
  87
  88 Datum
  89 gbt_cash_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_cash_consistent(PG_FUNCTION_ARGS)
 100 {
 101         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 102         Cash            query = PG_GETARG_CASH(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         cashKEY    *kkk = (cashKEY *) 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_cash_union(PG_FUNCTION_ARGS)
 123 {
 124         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 125         void       *out = palloc(sizeof(cashKEY));
 126
 127         *(int *) PG_GETARG_POINTER(1) = sizeof(cashKEY);
 128         PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo));
 129 }
 130
 131
 132 Datum
 133 gbt_cash_penalty(PG_FUNCTION_ARGS)
 134 {
 135         cashKEY    *origentry = (cashKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
 136         cashKEY    *newentry = (cashKEY *) 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
 145 Datum
 146 gbt_cash_picksplit(PG_FUNCTION_ARGS)
 147 {
 148         PG_RETURN_POINTER(gbt_num_picksplit(
 149                                                                         (GistEntryVector *) PG_GETARG_POINTER(0),
 150                                                                           (GIST_SPLITVEC *) PG_GETARG_POINTER(1),
 151                                                                                 &tinfo
 152                                                                                 ));
 153 }
 154
 155 Datum
 156 gbt_cash_same(PG_FUNCTION_ARGS)
 157 {
 158         cashKEY    *b1 = (cashKEY *) PG_GETARG_POINTER(0);
 159         cashKEY    *b2 = (cashKEY *) PG_GETARG_POINTER(1);
 160         bool       *result = (bool *) PG_GETARG_POINTER(2);
 161
 162         *result = gbt_num_same((void *) b1, (void *) b2, &tinfo);
 163         PG_RETURN_POINTER(result);
 164 }