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         /* Oid          subtype = PG_GETARG_OID(3); */
 104         bool       *recheck = (bool *) PG_GETARG_POINTER(4);
 105         float4KEY  *kkk = (float4KEY *) DatumGetPointer(entry->key);
 106         GBT_NUMKEY_R key;
 107
 108         /* All cases served by this function are exact */
 109         *recheck = false;
 110
 111         key.lower = (GBT_NUMKEY *) & kkk->lower;
 112         key.upper = (GBT_NUMKEY *) & kkk->upper;
 113
 114         PG_RETURN_BOOL(
 115                                    gbt_num_consistent(&key, (void *) &query, &strategy, GIST_LEAF(entry), &tinfo)
 116                 );
 117 }
 118
 119
 120 Datum
 121 gbt_float4_union(PG_FUNCTION_ARGS)
 122 {
 123         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 124         void       *out = palloc(sizeof(float4KEY));
 125
 126         *(int *) PG_GETARG_POINTER(1) = sizeof(float4KEY);
 127         PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo));
 128 }
 129
 130
 131 Datum
 132 gbt_float4_penalty(PG_FUNCTION_ARGS)
 133 {
 134         float4KEY  *origentry = (float4KEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
 135         float4KEY  *newentry = (float4KEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
 136         float      *result = (float *) PG_GETARG_POINTER(2);
 137
 138         penalty_num(result, origentry->lower, origentry->upper, newentry->lower, newentry->upper);
 139
 140         PG_RETURN_POINTER(result);
 141
 142 }
 143
 144 Datum
 145 gbt_float4_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_float4_same(PG_FUNCTION_ARGS)
 156 {
 157         float4KEY  *b1 = (float4KEY *) PG_GETARG_POINTER(0);
 158         float4KEY  *b2 = (float4KEY *) 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 }