1 /*
2 * contrib/btree_gist/btree_enum.c
3 */
4 #include "postgres.h"
5 #include "fmgr.h"
6 #include "utils/builtins.h"
7
8 #include "btree_gist.h"
9 #include "btree_utils_num.h"
10
11 /* enums are really Oids, so we just use the same structure */
12
13 typedef struct
14 {
15 Oid lower;
16 Oid upper;
17 } oidKEY;
18
19 /*
20 ** enum ops
21 */
22 PG_FUNCTION_INFO_V1(gbt_enum_compress);
23 PG_FUNCTION_INFO_V1(gbt_enum_fetch);
24 PG_FUNCTION_INFO_V1(gbt_enum_union);
25 PG_FUNCTION_INFO_V1(gbt_enum_picksplit);
26 PG_FUNCTION_INFO_V1(gbt_enum_consistent);
27 PG_FUNCTION_INFO_V1(gbt_enum_penalty);
28 PG_FUNCTION_INFO_V1(gbt_enum_same);
29
30
31 static bool
gbt_enumgt(const void * a,const void * b,FmgrInfo * flinfo)32 gbt_enumgt(const void *a, const void *b, FmgrInfo *flinfo)
33 {
34 return DatumGetBool(
35 CallerFInfoFunctionCall2(enum_gt, flinfo, InvalidOid, ObjectIdGetDatum(*((const Oid *) a)), ObjectIdGetDatum(*((const Oid *) b)))
36 );
37 }
38 static bool
gbt_enumge(const void * a,const void * b,FmgrInfo * flinfo)39 gbt_enumge(const void *a, const void *b, FmgrInfo *flinfo)
40 {
41 return DatumGetBool(
42 CallerFInfoFunctionCall2(enum_ge, flinfo, InvalidOid, ObjectIdGetDatum(*((const Oid *) a)), ObjectIdGetDatum(*((const Oid *) b)))
43 );
44 }
45 static bool
gbt_enumeq(const void * a,const void * b,FmgrInfo * flinfo)46 gbt_enumeq(const void *a, const void *b, FmgrInfo *flinfo)
47 {
48 return (*((const Oid *) a) == *((const Oid *) b));
49 }
50 static bool
gbt_enumle(const void * a,const void * b,FmgrInfo * flinfo)51 gbt_enumle(const void *a, const void *b, FmgrInfo *flinfo)
52 {
53 return DatumGetBool(
54 CallerFInfoFunctionCall2(enum_le, flinfo, InvalidOid, ObjectIdGetDatum(*((const Oid *) a)), ObjectIdGetDatum(*((const Oid *) b)))
55 );
56 }
57 static bool
gbt_enumlt(const void * a,const void * b,FmgrInfo * flinfo)58 gbt_enumlt(const void *a, const void *b, FmgrInfo *flinfo)
59 {
60 return DatumGetBool(
61 CallerFInfoFunctionCall2(enum_lt, flinfo, InvalidOid, ObjectIdGetDatum(*((const Oid *) a)), ObjectIdGetDatum(*((const Oid *) b)))
62 );
63 }
64
65 static int
gbt_enumkey_cmp(const void * a,const void * b,FmgrInfo * flinfo)66 gbt_enumkey_cmp(const void *a, const void *b, FmgrInfo *flinfo)
67 {
68 oidKEY *ia = (oidKEY *) (((const Nsrt *) a)->t);
69 oidKEY *ib = (oidKEY *) (((const Nsrt *) b)->t);
70
71 if (ia->lower == ib->lower)
72 {
73 if (ia->upper == ib->upper)
74 return 0;
75
76 return DatumGetInt32(
77 CallerFInfoFunctionCall2(enum_cmp, flinfo, InvalidOid, ObjectIdGetDatum(ia->upper), ObjectIdGetDatum(ib->upper))
78 );
79 }
80
81 return DatumGetInt32(
82 CallerFInfoFunctionCall2(enum_cmp, flinfo, InvalidOid, ObjectIdGetDatum(ia->lower), ObjectIdGetDatum(ib->lower))
83 );
84 }
85
86 static const gbtree_ninfo tinfo =
87 {
88 gbt_t_enum,
89 sizeof(Oid),
90 8, /* sizeof(gbtreekey8) */
91 gbt_enumgt,
92 gbt_enumge,
93 gbt_enumeq,
94 gbt_enumle,
95 gbt_enumlt,
96 gbt_enumkey_cmp,
97 NULL /* no KNN support at least for now */
98 };
99
100
101 /**************************************************
102 * Enum ops
103 **************************************************/
104
105
106 Datum
gbt_enum_compress(PG_FUNCTION_ARGS)107 gbt_enum_compress(PG_FUNCTION_ARGS)
108 {
109 GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
110
111 PG_RETURN_POINTER(gbt_num_compress(entry, &tinfo));
112 }
113
114 Datum
gbt_enum_fetch(PG_FUNCTION_ARGS)115 gbt_enum_fetch(PG_FUNCTION_ARGS)
116 {
117 GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
118
119 PG_RETURN_POINTER(gbt_num_fetch(entry, &tinfo));
120 }
121
122 Datum
gbt_enum_consistent(PG_FUNCTION_ARGS)123 gbt_enum_consistent(PG_FUNCTION_ARGS)
124 {
125 GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
126 Oid query = PG_GETARG_OID(1);
127 StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
128
129 /* Oid subtype = PG_GETARG_OID(3); */
130 bool *recheck = (bool *) PG_GETARG_POINTER(4);
131 oidKEY *kkk = (oidKEY *) DatumGetPointer(entry->key);
132 GBT_NUMKEY_R key;
133
134 /* All cases served by this function are exact */
135 *recheck = false;
136
137 key.lower = (GBT_NUMKEY *) &kkk->lower;
138 key.upper = (GBT_NUMKEY *) &kkk->upper;
139
140 PG_RETURN_BOOL(
141 gbt_num_consistent(&key, (void *) &query, &strategy, GIST_LEAF(entry), &tinfo, fcinfo->flinfo)
142 );
143 }
144
145 Datum
gbt_enum_union(PG_FUNCTION_ARGS)146 gbt_enum_union(PG_FUNCTION_ARGS)
147 {
148 GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
149 void *out = palloc(sizeof(oidKEY));
150
151 *(int *) PG_GETARG_POINTER(1) = sizeof(oidKEY);
152 PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo, fcinfo->flinfo));
153 }
154
155
156 Datum
gbt_enum_penalty(PG_FUNCTION_ARGS)157 gbt_enum_penalty(PG_FUNCTION_ARGS)
158 {
159 oidKEY *origentry = (oidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
160 oidKEY *newentry = (oidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
161 float *result = (float *) PG_GETARG_POINTER(2);
162
163 penalty_num(result, origentry->lower, origentry->upper, newentry->lower, newentry->upper);
164
165 PG_RETURN_POINTER(result);
166 }
167
168 Datum
gbt_enum_picksplit(PG_FUNCTION_ARGS)169 gbt_enum_picksplit(PG_FUNCTION_ARGS)
170 {
171 PG_RETURN_POINTER(gbt_num_picksplit(
172 (GistEntryVector *) PG_GETARG_POINTER(0),
173 (GIST_SPLITVEC *) PG_GETARG_POINTER(1),
174 &tinfo, fcinfo->flinfo
175 ));
176 }
177
178 Datum
gbt_enum_same(PG_FUNCTION_ARGS)179 gbt_enum_same(PG_FUNCTION_ARGS)
180 {
181 oidKEY *b1 = (oidKEY *) PG_GETARG_POINTER(0);
182 oidKEY *b2 = (oidKEY *) PG_GETARG_POINTER(1);
183 bool *result = (bool *) PG_GETARG_POINTER(2);
184
185 *result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
186 PG_RETURN_POINTER(result);
187 }
188