contrib/btree_gist/btree_float4.c

   1 /*
   2  * $PostgreSQL$
   3  */
   4 #include "btree_gist.h"
   5 #include "btree_utils_num.h"
   6
   7 typedef struct float4key
   8 {
   9         float4          lower;
  10         float4          upper;
  11 } float4KEY;
  12
  13 /*
  14 ** float4 ops
  15 */
  16 PG_FUNCTION_INFO_V1(gbt_float4_compress);
  17 PG_FUNCTION_INFO_V1(gbt_float4_union);
  18 PG_FUNCTION_INFO_V1(gbt_float4_picksplit);
  19 PG_FUNCTION_INFO_V1(gbt_float4_consistent);
  20 PG_FUNCTION_INFO_V1(gbt_float4_penalty);
  21 PG_FUNCTION_INFO_V1(gbt_float4_same);
  22
  23 Datum           gbt_float4_compress(PG_FUNCTION_ARGS);
  24 Datum           gbt_float4_union(PG_FUNCTION_ARGS);
  25 Datum           gbt_float4_picksplit(PG_FUNCTION_ARGS);
  26 Datum           gbt_float4_consistent(PG_FUNCTION_ARGS);
  27 Datum           gbt_float4_penalty(PG_FUNCTION_ARGS);
  28 Datum           gbt_float4_same(PG_FUNCTION_ARGS);
  29
  30 static bool
  31 gbt_float4gt(const void *a, const void *b)
  32 {
  33         return (*((float4 *) a) > *((float4 *) b));
  34 }
  35 static bool
  36 gbt_float4ge(const void *a, const void *b)
  37 {
  38         return (*((float4 *) a) >= *((float4 *) b));
  39 }
  40 static bool
  41 gbt_float4eq(const void *a, const void *b)
  42 {
  43         return (*((float4 *) a) == *((float4 *) b));
  44 }
  45 static bool
  46 gbt_float4le(const void *a, const void *b)
  47 {
  48         return (*((float4 *) a) <= *((float4 *) b));
  49 }
  50 static bool
  51 gbt_float4lt(const void *a, const void *b)
  52 {
  53         return (*((float4 *) a) < *((float4 *) b));
  54 }
  55
  56 static int
  57 gbt_float4key_cmp(const void *a, const void *b)
  58 {
  59
  60         if (*(float4 *) &(((Nsrt *) a)->t[0]) > *(float4 *) &(((Nsrt *) b)->t[0]))
  61                 return 1;
  62         else if (*(float4 *) &(((Nsrt *) a)->t[0]) < *(float4 *) &(((Nsrt *) b)->t[0]))
  63                 return -1;
  64         return 0;
  65
  66 }
  67
  68
  69 static const gbtree_ninfo tinfo =
  70 {
  71         gbt_t_float4,
  72         sizeof(float4),
  73         gbt_float4gt,
  74         gbt_float4ge,
  75         gbt_float4eq,
  76         gbt_float4le,
  77         gbt_float4lt,
  78         gbt_float4key_cmp
  79 };
  80
  81
  82 /**************************************************
  83  * float4 ops
  84  **************************************************/
  85
  86
  87 Datum
  88 gbt_float4_compress(PG_FUNCTION_ARGS)
  89 {
  90         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
  91         GISTENTRY  *retval = NULL;
  92
  93         PG_RETURN_POINTER(gbt_num_compress(retval, entry, &tinfo));
  94 }
  95
  96
  97 Datum
  98 gbt_float4_consistent(PG_FUNCTION_ARGS)
  99 {
 100         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 101         float4          query = PG_GETARG_FLOAT4(1);
 102         StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
 103
 104         /* Oid          subtype = PG_GETARG_OID(3); */
 105         bool       *recheck = (bool *) PG_GETARG_POINTER(4);
 106         float4KEY  *kkk = (float4KEY *) 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_float4_union(PG_FUNCTION_ARGS)
 123 {
 124         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 125         void       *out = palloc(sizeof(float4KEY));
 126
 127         *(int *) PG_GETARG_POINTER(1) = sizeof(float4KEY);
 128         PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo));
 129 }
 130
 131
 132 Datum
 133 gbt_float4_penalty(PG_FUNCTION_ARGS)
 134 {
 135         float4KEY  *origentry = (float4KEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
 136         float4KEY  *newentry = (float4KEY *) 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_float4_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_float4_same(PG_FUNCTION_ARGS)
 157 {
 158         float4KEY  *b1 = (float4KEY *) PG_GETARG_POINTER(0);
 159         float4KEY  *b2 = (float4KEY *) 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 }