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/builtins.h"
9 : : #include "utils/bytea.h"
10 : : #include "utils/varbit.h"
11 : :
12 : :
13 : : /*
14 : : ** Bit ops
15 : : */
7261 teodor@sigaev.ru 16 :CBC 3 : PG_FUNCTION_INFO_V1(gbt_bit_compress);
17 : 3 : PG_FUNCTION_INFO_V1(gbt_bit_union);
18 : 3 : PG_FUNCTION_INFO_V1(gbt_bit_picksplit);
19 : 3 : PG_FUNCTION_INFO_V1(gbt_bit_consistent);
20 : 3 : PG_FUNCTION_INFO_V1(gbt_bit_penalty);
21 : 3 : PG_FUNCTION_INFO_V1(gbt_bit_same);
22 : :
23 : :
24 : : /* define for comparison */
25 : :
26 : : static bool
2581 andrew@dunslane.net 27 : 880 : gbt_bitgt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
28 : : {
4741 tgl@sss.pgh.pa.us 29 : 880 : return DatumGetBool(DirectFunctionCall2(bitgt,
30 : : PointerGetDatum(a),
31 : : PointerGetDatum(b)));
32 : : }
33 : :
34 : : static bool
2581 andrew@dunslane.net 35 : 880 : gbt_bitge(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
36 : : {
4741 tgl@sss.pgh.pa.us 37 : 880 : return DatumGetBool(DirectFunctionCall2(bitge,
38 : : PointerGetDatum(a),
39 : : PointerGetDatum(b)));
40 : : }
41 : :
42 : : static bool
2581 andrew@dunslane.net 43 : 298 : gbt_biteq(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
44 : : {
4741 tgl@sss.pgh.pa.us 45 : 298 : return DatumGetBool(DirectFunctionCall2(biteq,
46 : : PointerGetDatum(a),
47 : : PointerGetDatum(b)));
48 : : }
49 : :
50 : : static bool
2581 andrew@dunslane.net 51 : 618 : gbt_bitle(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
52 : : {
4741 tgl@sss.pgh.pa.us 53 : 618 : return DatumGetBool(DirectFunctionCall2(bitle,
54 : : PointerGetDatum(a),
55 : : PointerGetDatum(b)));
56 : : }
57 : :
58 : : static bool
2581 andrew@dunslane.net 59 : 618 : gbt_bitlt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
60 : : {
4741 tgl@sss.pgh.pa.us 61 : 618 : return DatumGetBool(DirectFunctionCall2(bitlt,
62 : : PointerGetDatum(a),
63 : : PointerGetDatum(b)));
64 : : }
65 : :
66 : : static int32
2581 andrew@dunslane.net 67 : 24018 : gbt_bitcmp(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
68 : : {
4741 tgl@sss.pgh.pa.us 69 : 24018 : return DatumGetInt32(DirectFunctionCall2(byteacmp,
70 : : PointerGetDatum(a),
71 : : PointerGetDatum(b)));
72 : : }
73 : :
74 : :
75 : : static bytea *
7168 bruce@momjian.us 76 : 5100 : gbt_bit_xfrm(bytea *leaf)
77 : : {
78 : 5100 : bytea *out = leaf;
3624 heikki.linnakangas@i 79 : 5100 : int sz = VARBITBYTES(leaf) + VARHDRSZ;
80 : 5100 : int padded_sz = INTALIGN(sz);
81 : :
82 : 5100 : out = (bytea *) palloc(padded_sz);
83 : : /* initialize the padding bytes to zero */
84 [ + + ]: 16368 : while (sz < padded_sz)
85 : 11268 : ((char *) out)[sz++] = 0;
86 : 5100 : SET_VARSIZE(out, padded_sz);
432 peter@eisentraut.org 87 : 5100 : memcpy(VARDATA(out), VARBITS(leaf), VARBITBYTES(leaf));
7168 bruce@momjian.us 88 : 5100 : return out;
89 : : }
90 : :
91 : :
92 : :
93 : :
94 : : static GBT_VARKEY *
2581 andrew@dunslane.net 95 : 5060 : gbt_bit_l2n(GBT_VARKEY *leaf, FmgrInfo *flinfo)
96 : : {
7168 bruce@momjian.us 97 : 5060 : GBT_VARKEY *out = leaf;
98 : 5060 : GBT_VARKEY_R r = gbt_var_key_readable(leaf);
99 : : bytea *o;
100 : :
101 : 5060 : o = gbt_bit_xfrm(r.lower);
102 : 5060 : r.upper = r.lower = o;
3307 heikki.linnakangas@i 103 : 5060 : out = gbt_var_key_copy(&r);
7168 bruce@momjian.us 104 : 5060 : pfree(o);
105 : :
106 : 5060 : 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 : : * Bit ops
126 : : **************************************************/
127 : :
128 : : Datum
129 : 1220 : gbt_bit_compress(PG_FUNCTION_ARGS)
130 : : {
131 : 1220 : GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
132 : :
133 : 1220 : PG_RETURN_POINTER(gbt_var_compress(entry, &tinfo));
134 : : }
135 : :
136 : : Datum
7261 teodor@sigaev.ru 137 : 3334 : gbt_bit_consistent(PG_FUNCTION_ARGS)
138 : : {
7168 bruce@momjian.us 139 : 3334 : GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
140 : 3334 : void *query = (void *) DatumGetByteaP(PG_GETARG_DATUM(1));
141 : 3334 : StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
142 : :
143 : : /* Oid subtype = PG_GETARG_OID(3); */
5844 tgl@sss.pgh.pa.us 144 : 3334 : bool *recheck = (bool *) PG_GETARG_POINTER(4);
145 : : bool retval;
146 : 3334 : GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
7168 bruce@momjian.us 147 : 3334 : GBT_VARKEY_R r = gbt_var_key_readable(key);
148 : :
149 : : /* All cases served by this function are exact */
5844 tgl@sss.pgh.pa.us 150 : 3334 : *recheck = false;
151 : :
7168 bruce@momjian.us 152 [ + + ]: 3334 : if (GIST_LEAF(entry))
4741 tgl@sss.pgh.pa.us 153 : 3294 : retval = gbt_var_consistent(&r, query, strategy, PG_GET_COLLATION(),
154 : : true, &tinfo, fcinfo->flinfo);
155 : : else
156 : : {
7168 bruce@momjian.us 157 : 40 : bytea *q = gbt_bit_xfrm((bytea *) query);
158 : :
4741 tgl@sss.pgh.pa.us 159 : 40 : retval = gbt_var_consistent(&r, q, strategy, PG_GET_COLLATION(),
160 : : false, &tinfo, fcinfo->flinfo);
161 : : }
7168 bruce@momjian.us 162 : 3334 : PG_RETURN_BOOL(retval);
163 : : }
164 : :
165 : :
166 : :
167 : : Datum
7261 teodor@sigaev.ru 168 : 717 : gbt_bit_union(PG_FUNCTION_ARGS)
169 : : {
7168 bruce@momjian.us 170 : 717 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
171 : 717 : int32 *size = (int *) PG_GETARG_POINTER(1);
172 : :
4741 tgl@sss.pgh.pa.us 173 : 717 : PG_RETURN_POINTER(gbt_var_union(entryvec, size, PG_GET_COLLATION(),
174 : : &tinfo, fcinfo->flinfo));
175 : : }
176 : :
177 : :
178 : : Datum
7261 teodor@sigaev.ru 179 : 6 : gbt_bit_picksplit(PG_FUNCTION_ARGS)
180 : : {
7168 bruce@momjian.us 181 : 6 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
182 : 6 : GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
183 : :
4741 tgl@sss.pgh.pa.us 184 : 6 : gbt_var_picksplit(entryvec, v, PG_GET_COLLATION(),
185 : : &tinfo, fcinfo->flinfo);
7168 bruce@momjian.us 186 : 6 : PG_RETURN_POINTER(v);
187 : : }
188 : :
189 : : Datum
7261 teodor@sigaev.ru 190 : 715 : gbt_bit_same(PG_FUNCTION_ARGS)
191 : : {
7168 bruce@momjian.us 192 : 715 : Datum d1 = PG_GETARG_DATUM(0);
193 : 715 : Datum d2 = PG_GETARG_DATUM(1);
194 : 715 : bool *result = (bool *) PG_GETARG_POINTER(2);
195 : :
2581 andrew@dunslane.net 196 : 715 : *result = gbt_var_same(d1, d2, PG_GET_COLLATION(), &tinfo, fcinfo->flinfo);
4741 tgl@sss.pgh.pa.us 197 : 715 : PG_RETURN_POINTER(result);
198 : : }
199 : :
200 : :
201 : : Datum
7261 teodor@sigaev.ru 202 : 1542 : gbt_bit_penalty(PG_FUNCTION_ARGS)
203 : : {
7168 bruce@momjian.us 204 : 1542 : GISTENTRY *o = (GISTENTRY *) PG_GETARG_POINTER(0);
205 : 1542 : GISTENTRY *n = (GISTENTRY *) PG_GETARG_POINTER(1);
6912 neilc@samurai.com 206 : 1542 : float *result = (float *) PG_GETARG_POINTER(2);
207 : :
4741 tgl@sss.pgh.pa.us 208 : 1542 : PG_RETURN_POINTER(gbt_var_penalty(result, o, n, PG_GET_COLLATION(),
209 : : &tinfo, fcinfo->flinfo));
210 : : }
|
