Age Owner Branch data TLA Line data Source code
1 : : /*
2 : : * contrib/btree_gist/btree_bit.c
3 : : */
4 : : #include "postgres.h"
5 : :
6 : : #include "btree_gist.h"
7 : : #include "btree_utils_var.h"
8 : : #include "utils/fmgrprotos.h"
9 : : #include "utils/sortsupport.h"
10 : : #include "utils/varbit.h"
11 : : #include "varatt.h"
12 : :
13 : : /* GiST support functions */
7961 teodor@sigaev.ru 14 :CBC 5 : PG_FUNCTION_INFO_V1(gbt_bit_compress);
15 : 5 : PG_FUNCTION_INFO_V1(gbt_bit_union);
16 : 5 : PG_FUNCTION_INFO_V1(gbt_bit_picksplit);
17 : 5 : PG_FUNCTION_INFO_V1(gbt_bit_consistent);
18 : 5 : PG_FUNCTION_INFO_V1(gbt_bit_penalty);
19 : 5 : PG_FUNCTION_INFO_V1(gbt_bit_same);
346 heikki.linnakangas@i 20 : 4 : PG_FUNCTION_INFO_V1(gbt_bit_sortsupport);
21 : 4 : PG_FUNCTION_INFO_V1(gbt_varbit_sortsupport);
22 : :
23 : :
24 : : /* define for comparison */
25 : :
26 : : static bool
3281 andrew@dunslane.net 27 : 900 : gbt_bitgt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
28 : : {
5441 tgl@sss.pgh.pa.us 29 : 900 : return DatumGetBool(DirectFunctionCall2(bitgt,
30 : : PointerGetDatum(a),
31 : : PointerGetDatum(b)));
32 : : }
33 : :
34 : : static bool
3281 andrew@dunslane.net 35 : 900 : gbt_bitge(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
36 : : {
5441 tgl@sss.pgh.pa.us 37 : 900 : return DatumGetBool(DirectFunctionCall2(bitge,
38 : : PointerGetDatum(a),
39 : : PointerGetDatum(b)));
40 : : }
41 : :
42 : : static bool
3281 andrew@dunslane.net 43 : 300 : gbt_biteq(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
44 : : {
5441 tgl@sss.pgh.pa.us 45 : 300 : return DatumGetBool(DirectFunctionCall2(biteq,
46 : : PointerGetDatum(a),
47 : : PointerGetDatum(b)));
48 : : }
49 : :
50 : : static bool
3281 andrew@dunslane.net 51 : 600 : gbt_bitle(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
52 : : {
5441 tgl@sss.pgh.pa.us 53 : 600 : return DatumGetBool(DirectFunctionCall2(bitle,
54 : : PointerGetDatum(a),
55 : : PointerGetDatum(b)));
56 : : }
57 : :
58 : : static bool
3281 andrew@dunslane.net 59 : 600 : gbt_bitlt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
60 : : {
5441 tgl@sss.pgh.pa.us 61 : 600 : return DatumGetBool(DirectFunctionCall2(bitlt,
62 : : PointerGetDatum(a),
63 : : PointerGetDatum(b)));
64 : : }
65 : :
66 : : static int32
3281 andrew@dunslane.net 67 : 14707 : gbt_bitcmp(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
68 : : {
5441 tgl@sss.pgh.pa.us 69 : 14707 : return DatumGetInt32(DirectFunctionCall2(byteacmp,
70 : : PointerGetDatum(a),
71 : : PointerGetDatum(b)));
72 : : }
73 : :
74 : :
75 : : static bytea *
7868 bruce@momjian.us 76 : 3638 : gbt_bit_xfrm(bytea *leaf)
77 : : {
78 : 3638 : bytea *out = leaf;
4324 heikki.linnakangas@i 79 : 3638 : int sz = VARBITBYTES(leaf) + VARHDRSZ;
80 : 3638 : int padded_sz = INTALIGN(sz);
81 : :
82 : 3638 : out = (bytea *) palloc(padded_sz);
83 : : /* initialize the padding bytes to zero */
84 [ + + ]: 11799 : while (sz < padded_sz)
85 : 8161 : ((char *) out)[sz++] = 0;
86 : 3638 : SET_VARSIZE(out, padded_sz);
1132 peter@eisentraut.org 87 : 3638 : memcpy(VARDATA(out), VARBITS(leaf), VARBITBYTES(leaf));
7868 bruce@momjian.us 88 : 3638 : return out;
89 : : }
90 : :
91 : :
92 : :
93 : :
94 : : static GBT_VARKEY *
3281 andrew@dunslane.net 95 : 3598 : gbt_bit_l2n(GBT_VARKEY *leaf, FmgrInfo *flinfo)
96 : : {
7868 bruce@momjian.us 97 : 3598 : GBT_VARKEY *out = leaf;
98 : 3598 : GBT_VARKEY_R r = gbt_var_key_readable(leaf);
99 : : bytea *o;
100 : :
101 : 3598 : o = gbt_bit_xfrm(r.lower);
102 : 3598 : r.upper = r.lower = o;
4007 heikki.linnakangas@i 103 : 3598 : out = gbt_var_key_copy(&r);
7868 bruce@momjian.us 104 : 3598 : pfree(o);
105 : :
106 : 3598 : return out;
107 : : }
108 : :
109 : : static const gbtree_vinfo tinfo =
110 : : {
111 : : gbt_t_bit,
112 : : 0,
113 : : true,
114 : : gbt_bitgt,
115 : : gbt_bitge,
116 : : gbt_biteq,
117 : : gbt_bitle,
118 : : gbt_bitlt,
119 : : gbt_bitcmp,
120 : : gbt_bit_l2n
121 : : };
122 : :
123 : :
124 : : /**************************************************
125 : : * GiST support functions
126 : : **************************************************/
127 : :
128 : : Datum
129 : 1208 : gbt_bit_compress(PG_FUNCTION_ARGS)
130 : : {
131 : 1208 : GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
132 : :
133 : 1208 : PG_RETURN_POINTER(gbt_var_compress(entry, &tinfo));
134 : : }
135 : :
136 : : Datum
7961 teodor@sigaev.ru 137 : 3340 : gbt_bit_consistent(PG_FUNCTION_ARGS)
138 : : {
7868 bruce@momjian.us 139 : 3340 : GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
472 peter@eisentraut.org 140 : 3340 : void *query = DatumGetByteaP(PG_GETARG_DATUM(1));
7868 bruce@momjian.us 141 : 3340 : StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
142 : : #ifdef NOT_USED
143 : : Oid subtype = PG_GETARG_OID(3);
144 : : #endif
6544 tgl@sss.pgh.pa.us 145 : 3340 : bool *recheck = (bool *) PG_GETARG_POINTER(4);
146 : : bool retval;
147 : 3340 : GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
7868 bruce@momjian.us 148 : 3340 : GBT_VARKEY_R r = gbt_var_key_readable(key);
149 : :
150 : : /* All cases served by this function are exact */
6544 tgl@sss.pgh.pa.us 151 : 3340 : *recheck = false;
152 : :
7868 bruce@momjian.us 153 [ + + ]: 3340 : if (GIST_LEAF(entry))
5441 tgl@sss.pgh.pa.us 154 : 3300 : retval = gbt_var_consistent(&r, query, strategy, PG_GET_COLLATION(),
155 : : true, &tinfo, fcinfo->flinfo);
156 : : else
157 : : {
7868 bruce@momjian.us 158 : 40 : bytea *q = gbt_bit_xfrm((bytea *) query);
159 : :
5441 tgl@sss.pgh.pa.us 160 : 40 : retval = gbt_var_consistent(&r, q, strategy, PG_GET_COLLATION(),
161 : : false, &tinfo, fcinfo->flinfo);
162 : : }
7868 bruce@momjian.us 163 : 3340 : PG_RETURN_BOOL(retval);
164 : : }
165 : :
166 : : Datum
7961 teodor@sigaev.ru 167 : 2 : gbt_bit_union(PG_FUNCTION_ARGS)
168 : : {
7868 bruce@momjian.us 169 : 2 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
170 : 2 : int32 *size = (int *) PG_GETARG_POINTER(1);
171 : :
5441 tgl@sss.pgh.pa.us 172 : 2 : PG_RETURN_POINTER(gbt_var_union(entryvec, size, PG_GET_COLLATION(),
173 : : &tinfo, fcinfo->flinfo));
174 : : }
175 : :
176 : : Datum
7961 teodor@sigaev.ru 177 : 6 : gbt_bit_picksplit(PG_FUNCTION_ARGS)
178 : : {
7868 bruce@momjian.us 179 : 6 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
180 : 6 : GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
181 : :
5441 tgl@sss.pgh.pa.us 182 : 6 : gbt_var_picksplit(entryvec, v, PG_GET_COLLATION(),
183 : : &tinfo, fcinfo->flinfo);
7868 bruce@momjian.us 184 : 6 : PG_RETURN_POINTER(v);
185 : : }
186 : :
187 : : Datum
7961 teodor@sigaev.ru 188 :UBC 0 : gbt_bit_same(PG_FUNCTION_ARGS)
189 : : {
7868 bruce@momjian.us 190 : 0 : Datum d1 = PG_GETARG_DATUM(0);
191 : 0 : Datum d2 = PG_GETARG_DATUM(1);
192 : 0 : bool *result = (bool *) PG_GETARG_POINTER(2);
193 : :
3281 andrew@dunslane.net 194 : 0 : *result = gbt_var_same(d1, d2, PG_GET_COLLATION(), &tinfo, fcinfo->flinfo);
5441 tgl@sss.pgh.pa.us 195 : 0 : PG_RETURN_POINTER(result);
196 : : }
197 : :
198 : : Datum
7961 teodor@sigaev.ru 199 : 0 : gbt_bit_penalty(PG_FUNCTION_ARGS)
200 : : {
7868 bruce@momjian.us 201 : 0 : GISTENTRY *o = (GISTENTRY *) PG_GETARG_POINTER(0);
202 : 0 : GISTENTRY *n = (GISTENTRY *) PG_GETARG_POINTER(1);
7612 neilc@samurai.com 203 : 0 : float *result = (float *) PG_GETARG_POINTER(2);
204 : :
5441 tgl@sss.pgh.pa.us 205 : 0 : PG_RETURN_POINTER(gbt_var_penalty(result, o, n, PG_GET_COLLATION(),
206 : : &tinfo, fcinfo->flinfo));
207 : : }
208 : :
209 : : static int
346 heikki.linnakangas@i 210 :CBC 11338 : gbt_bit_ssup_cmp(Datum x, Datum y, SortSupport ssup)
211 : : {
212 : 11338 : GBT_VARKEY *key1 = PG_DETOAST_DATUM(x);
213 : 11338 : GBT_VARKEY *key2 = PG_DETOAST_DATUM(y);
214 : :
215 : 11338 : GBT_VARKEY_R arg1 = gbt_var_key_readable(key1);
216 : 11338 : GBT_VARKEY_R arg2 = gbt_var_key_readable(key2);
217 : : Datum result;
218 : :
219 : : /* for leaf items we expect lower == upper, so only compare lower */
220 : 11338 : result = DirectFunctionCall2(byteacmp,
221 : : PointerGetDatum(arg1.lower),
222 : : PointerGetDatum(arg2.lower));
223 : :
224 [ + - ]: 11338 : GBT_FREE_IF_COPY(key1, x);
225 [ + - ]: 11338 : GBT_FREE_IF_COPY(key2, y);
226 : :
227 : 11338 : return DatumGetInt32(result);
228 : : }
229 : :
230 : : Datum
231 : 1 : gbt_bit_sortsupport(PG_FUNCTION_ARGS)
232 : : {
233 : 1 : SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
234 : :
235 : 1 : ssup->comparator = gbt_bit_ssup_cmp;
236 : 1 : ssup->ssup_extra = NULL;
237 : :
238 : 1 : PG_RETURN_VOID();
239 : : }
240 : :
241 : : Datum
242 : 1 : gbt_varbit_sortsupport(PG_FUNCTION_ARGS)
243 : : {
244 : 1 : SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
245 : :
246 : 1 : ssup->comparator = gbt_bit_ssup_cmp;
247 : 1 : ssup->ssup_extra = NULL;
248 : :
249 : 1 : PG_RETURN_VOID();
250 : : }
|
