PostgreSQL Source Code  git master
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros
geo_spgist.c File Reference
#include "postgres.h"
#include "access/spgist.h"
#include "access/stratnum.h"
#include "catalog/pg_type.h"
#include "utils/builtins.h"
#include "utils/geo_decls.h"
Include dependency graph for geo_spgist.c:

Go to the source code of this file.

Data Structures

struct  Range
 
struct  RangeBox
 
struct  RectBox
 

Functions

static int compareDoubles (const void *a, const void *b)
 
static uint8 getQuadrant (BOX *centroid, BOX *inBox)
 
static RangeBoxgetRangeBox (BOX *box)
 
static RectBoxinitRectBox (void)
 
static RectBoxnextRectBox (RectBox *rect_box, RangeBox *centroid, uint8 quadrant)
 
static bool overlap2D (RangeBox *range_box, Range *query)
 
static bool overlap4D (RectBox *rect_box, RangeBox *query)
 
static bool contain2D (RangeBox *range_box, Range *query)
 
static bool contain4D (RectBox *rect_box, RangeBox *query)
 
static bool contained2D (RangeBox *range_box, Range *query)
 
static bool contained4D (RectBox *rect_box, RangeBox *query)
 
static bool lower2D (RangeBox *range_box, Range *query)
 
static bool higher2D (RangeBox *range_box, Range *query)
 
static bool left4D (RectBox *rect_box, RangeBox *query)
 
static bool overLeft4D (RectBox *rect_box, RangeBox *query)
 
static bool right4D (RectBox *rect_box, RangeBox *query)
 
static bool overRight4D (RectBox *rect_box, RangeBox *query)
 
static bool below4D (RectBox *rect_box, RangeBox *query)
 
static bool overBelow4D (RectBox *rect_box, RangeBox *query)
 
static bool above4D (RectBox *rect_box, RangeBox *query)
 
static bool overAbove4D (RectBox *rect_box, RangeBox *query)
 
Datum spg_box_quad_config (PG_FUNCTION_ARGS)
 
Datum spg_box_quad_choose (PG_FUNCTION_ARGS)
 
Datum spg_box_quad_picksplit (PG_FUNCTION_ARGS)
 
Datum spg_box_quad_inner_consistent (PG_FUNCTION_ARGS)
 
Datum spg_box_quad_leaf_consistent (PG_FUNCTION_ARGS)
 

Function Documentation

static bool above4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 341 of file geo_spgist.c.

References higher2D(), RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

342 {
343  return higher2D(&rect_box->range_box_y, &query->right);
344 }
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
static bool higher2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:291
static bool below4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 327 of file geo_spgist.c.

References lower2D(), RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

328 {
329  return lower2D(&rect_box->range_box_y, &query->right);
330 }
static bool lower2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:283
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
static int compareDoubles ( const void *  a,
const void *  b 
)
static

Definition at line 90 of file geo_spgist.c.

Referenced by spg_box_quad_picksplit().

91 {
92  double x = *(double *) a;
93  double y = *(double *) b;
94 
95  if (x == y)
96  return 0;
97  return (x > y) ? 1 : -1;
98 }
static bool contain2D ( RangeBox range_box,
Range query 
)
static

Definition at line 249 of file geo_spgist.c.

References FPge, FPle, Range::high, RangeBox::left, Range::low, and RangeBox::right.

Referenced by contain4D().

250 {
251  return FPge(range_box->right.high, query->high) &&
252  FPle(range_box->left.low, query->low);
253 }
Range right
Definition: geo_spgist.c:109
double low
Definition: geo_spgist.c:102
double high
Definition: geo_spgist.c:103
#define FPge(A, B)
Definition: geo_decls.h:42
#define FPle(A, B)
Definition: geo_decls.h:40
Range left
Definition: geo_spgist.c:108
static bool contain4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 257 of file geo_spgist.c.

References contain2D(), RangeBox::left, RectBox::range_box_x, RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

258 {
259  return contain2D(&rect_box->range_box_x, &query->left) &&
260  contain2D(&rect_box->range_box_y, &query->right);
261 }
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
RangeBox range_box_x
Definition: geo_spgist.c:114
static bool contain2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:249
Range left
Definition: geo_spgist.c:108
static bool contained2D ( RangeBox range_box,
Range query 
)
static

Definition at line 265 of file geo_spgist.c.

References FPge, FPle, Range::high, RangeBox::left, Range::low, and RangeBox::right.

Referenced by contained4D().

266 {
267  return FPle(range_box->left.low, query->high) &&
268  FPge(range_box->left.high, query->low) &&
269  FPle(range_box->right.low, query->high) &&
270  FPge(range_box->right.high, query->low);
271 }
Range right
Definition: geo_spgist.c:109
double low
Definition: geo_spgist.c:102
double high
Definition: geo_spgist.c:103
#define FPge(A, B)
Definition: geo_decls.h:42
#define FPle(A, B)
Definition: geo_decls.h:40
Range left
Definition: geo_spgist.c:108
static bool contained4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 275 of file geo_spgist.c.

References contained2D(), RangeBox::left, RectBox::range_box_x, RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

276 {
277  return contained2D(&rect_box->range_box_x, &query->left) &&
278  contained2D(&rect_box->range_box_y, &query->right);
279 }
static bool contained2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:265
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
RangeBox range_box_x
Definition: geo_spgist.c:114
Range left
Definition: geo_spgist.c:108
static uint8 getQuadrant ( BOX centroid,
BOX inBox 
)
static

Definition at line 127 of file geo_spgist.c.

References BOX::high, BOX::low, Point::x, and Point::y.

Referenced by spg_box_quad_choose(), and spg_box_quad_picksplit().

128 {
129  uint8 quadrant = 0;
130 
131  if (inBox->low.x > centroid->low.x)
132  quadrant |= 0x8;
133 
134  if (inBox->high.x > centroid->high.x)
135  quadrant |= 0x4;
136 
137  if (inBox->low.y > centroid->low.y)
138  quadrant |= 0x2;
139 
140  if (inBox->high.y > centroid->high.y)
141  quadrant |= 0x1;
142 
143  return quadrant;
144 }
double y
Definition: geo_decls.h:60
unsigned char uint8
Definition: c.h:263
double x
Definition: geo_decls.h:60
Point low
Definition: geo_decls.h:104
Point high
Definition: geo_decls.h:104
static RangeBox* getRangeBox ( BOX box)
static

Definition at line 154 of file geo_spgist.c.

References Range::high, BOX::high, RangeBox::left, Range::low, BOX::low, palloc(), RangeBox::right, Point::x, and Point::y.

Referenced by spg_box_quad_inner_consistent().

155 {
156  RangeBox *range_box = (RangeBox *) palloc(sizeof(RangeBox));
157 
158  range_box->left.low = box->low.x;
159  range_box->left.high = box->high.x;
160 
161  range_box->right.low = box->low.y;
162  range_box->right.high = box->high.y;
163 
164  return range_box;
165 }
double y
Definition: geo_decls.h:60
Range right
Definition: geo_spgist.c:109
double x
Definition: geo_decls.h:60
double low
Definition: geo_spgist.c:102
Point low
Definition: geo_decls.h:104
double high
Definition: geo_spgist.c:103
Point high
Definition: geo_decls.h:104
void * palloc(Size size)
Definition: mcxt.c:891
Range left
Definition: geo_spgist.c:108
static bool higher2D ( RangeBox range_box,
Range query 
)
static

Definition at line 291 of file geo_spgist.c.

References FPgt, Range::high, RangeBox::left, and RangeBox::right.

Referenced by above4D(), overAbove4D(), overRight4D(), and right4D().

292 {
293  return FPgt(range_box->left.high, query->high) &&
294  FPgt(range_box->right.high, query->high);
295 }
Range right
Definition: geo_spgist.c:109
#define FPgt(A, B)
Definition: geo_decls.h:41
double high
Definition: geo_spgist.c:103
Range left
Definition: geo_spgist.c:108
static RectBox* initRectBox ( void  )
static

Definition at line 174 of file geo_spgist.c.

References get_float8_infinity(), Range::high, RangeBox::left, Range::low, palloc(), RectBox::range_box_x, RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

175 {
176  RectBox *rect_box = (RectBox *) palloc(sizeof(RectBox));
177  double infinity = get_float8_infinity();
178 
179  rect_box->range_box_x.left.low = -infinity;
180  rect_box->range_box_x.left.high = infinity;
181 
182  rect_box->range_box_x.right.low = -infinity;
183  rect_box->range_box_x.right.high = infinity;
184 
185  rect_box->range_box_y.left.low = -infinity;
186  rect_box->range_box_y.left.high = infinity;
187 
188  rect_box->range_box_y.right.low = -infinity;
189  rect_box->range_box_y.right.high = infinity;
190 
191  return rect_box;
192 }
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
RangeBox range_box_x
Definition: geo_spgist.c:114
double low
Definition: geo_spgist.c:102
double get_float8_infinity(void)
Definition: float.c:121
double high
Definition: geo_spgist.c:103
void * palloc(Size size)
Definition: mcxt.c:891
Range left
Definition: geo_spgist.c:108
static bool left4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 299 of file geo_spgist.c.

References RangeBox::left, lower2D(), and RectBox::range_box_x.

Referenced by spg_box_quad_inner_consistent().

300 {
301  return lower2D(&rect_box->range_box_x, &query->left);
302 }
static bool lower2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:283
RangeBox range_box_x
Definition: geo_spgist.c:114
Range left
Definition: geo_spgist.c:108
static bool lower2D ( RangeBox range_box,
Range query 
)
static

Definition at line 283 of file geo_spgist.c.

References FPlt, RangeBox::left, Range::low, and RangeBox::right.

Referenced by below4D(), left4D(), overBelow4D(), and overLeft4D().

284 {
285  return FPlt(range_box->left.low, query->low) &&
286  FPlt(range_box->right.low, query->low);
287 }
Range right
Definition: geo_spgist.c:109
double low
Definition: geo_spgist.c:102
#define FPlt(A, B)
Definition: geo_decls.h:39
Range left
Definition: geo_spgist.c:108
static RectBox* nextRectBox ( RectBox rect_box,
RangeBox centroid,
uint8  quadrant 
)
static

Definition at line 202 of file geo_spgist.c.

References Range::high, RangeBox::left, Range::low, palloc(), RectBox::range_box_x, RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

203 {
204  RectBox *next_rect_box = (RectBox *) palloc(sizeof(RectBox));
205 
206  memcpy(next_rect_box, rect_box, sizeof(RectBox));
207 
208  if (quadrant & 0x8)
209  next_rect_box->range_box_x.left.low = centroid->left.low;
210  else
211  next_rect_box->range_box_x.left.high = centroid->left.low;
212 
213  if (quadrant & 0x4)
214  next_rect_box->range_box_x.right.low = centroid->left.high;
215  else
216  next_rect_box->range_box_x.right.high = centroid->left.high;
217 
218  if (quadrant & 0x2)
219  next_rect_box->range_box_y.left.low = centroid->right.low;
220  else
221  next_rect_box->range_box_y.left.high = centroid->right.low;
222 
223  if (quadrant & 0x1)
224  next_rect_box->range_box_y.right.low = centroid->right.high;
225  else
226  next_rect_box->range_box_y.right.high = centroid->right.high;
227 
228  return next_rect_box;
229 }
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
RangeBox range_box_x
Definition: geo_spgist.c:114
double low
Definition: geo_spgist.c:102
double high
Definition: geo_spgist.c:103
void * palloc(Size size)
Definition: mcxt.c:891
Range left
Definition: geo_spgist.c:108
static bool overAbove4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 348 of file geo_spgist.c.

References higher2D(), RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

349 {
350  return higher2D(&rect_box->range_box_y, &query->right);
351 }
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
static bool higher2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:291
static bool overBelow4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 334 of file geo_spgist.c.

References RangeBox::left, lower2D(), and RectBox::range_box_y.

Referenced by spg_box_quad_inner_consistent().

335 {
336  return lower2D(&rect_box->range_box_y, &query->left);
337 }
static bool lower2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:283
RangeBox range_box_y
Definition: geo_spgist.c:115
Range left
Definition: geo_spgist.c:108
static bool overlap2D ( RangeBox range_box,
Range query 
)
static

Definition at line 233 of file geo_spgist.c.

References FPge, FPle, Range::high, RangeBox::left, Range::low, and RangeBox::right.

Referenced by overlap4D().

234 {
235  return FPge(range_box->right.high, query->low) &&
236  FPle(range_box->left.low, query->high);
237 }
Range right
Definition: geo_spgist.c:109
double low
Definition: geo_spgist.c:102
double high
Definition: geo_spgist.c:103
#define FPge(A, B)
Definition: geo_decls.h:42
#define FPle(A, B)
Definition: geo_decls.h:40
Range left
Definition: geo_spgist.c:108
static bool overlap4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 241 of file geo_spgist.c.

References RangeBox::left, overlap2D(), RectBox::range_box_x, RectBox::range_box_y, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

242 {
243  return overlap2D(&rect_box->range_box_x, &query->left) &&
244  overlap2D(&rect_box->range_box_y, &query->right);
245 }
RangeBox range_box_y
Definition: geo_spgist.c:115
Range right
Definition: geo_spgist.c:109
RangeBox range_box_x
Definition: geo_spgist.c:114
static bool overlap2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:233
Range left
Definition: geo_spgist.c:108
static bool overLeft4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 306 of file geo_spgist.c.

References lower2D(), RectBox::range_box_x, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

307 {
308  return lower2D(&rect_box->range_box_x, &query->right);
309 }
static bool lower2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:283
Range right
Definition: geo_spgist.c:109
RangeBox range_box_x
Definition: geo_spgist.c:114
static bool overRight4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 320 of file geo_spgist.c.

References higher2D(), RectBox::range_box_x, and RangeBox::right.

Referenced by spg_box_quad_inner_consistent().

321 {
322  return higher2D(&rect_box->range_box_x, &query->right);
323 }
Range right
Definition: geo_spgist.c:109
RangeBox range_box_x
Definition: geo_spgist.c:114
static bool higher2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:291
static bool right4D ( RectBox rect_box,
RangeBox query 
)
static

Definition at line 313 of file geo_spgist.c.

References higher2D(), RangeBox::left, and RectBox::range_box_x.

Referenced by spg_box_quad_inner_consistent().

314 {
315  return higher2D(&rect_box->range_box_x, &query->left);
316 }
RangeBox range_box_x
Definition: geo_spgist.c:114
Range left
Definition: geo_spgist.c:108
static bool higher2D(RangeBox *range_box, Range *query)
Definition: geo_spgist.c:291
Datum spg_box_quad_choose ( PG_FUNCTION_ARGS  )

Definition at line 373 of file geo_spgist.c.

References spgChooseIn::allTheSame, BoxPGetDatum, spgChooseIn::datum, DatumGetBoxP, getQuadrant(), spgChooseOut::matchNode, PG_GETARG_POINTER, PG_RETURN_VOID, spgChooseIn::prefixDatum, spgChooseOut::result, spgChooseOut::resultType, and spgMatchNode.

374 {
377  BOX *centroid = DatumGetBoxP(in->prefixDatum),
378  *box = DatumGetBoxP(in->datum);
379 
380  out->resultType = spgMatchNode;
381  out->result.matchNode.restDatum = BoxPGetDatum(box);
382 
383  /* nodeN will be set by core, when allTheSame. */
384  if (!in->allTheSame)
385  out->result.matchNode.nodeN = getQuadrant(centroid, box);
386 
387  PG_RETURN_VOID();
388 }
Definition: geo_decls.h:102
Datum datum
Definition: spgist.h:56
struct spgChooseOut::@41::@42 matchNode
Datum prefixDatum
Definition: spgist.h:63
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:232
spgChooseResultType resultType
Definition: spgist.h:77
#define BoxPGetDatum(X)
Definition: geo_decls.h:160
#define PG_RETURN_VOID()
Definition: fmgr.h:293
#define DatumGetBoxP(X)
Definition: geo_decls.h:159
static uint8 getQuadrant(BOX *centroid, BOX *inBox)
Definition: geo_spgist.c:127
union spgChooseOut::@41 result
bool allTheSame
Definition: spgist.h:61
Datum spg_box_quad_config ( PG_FUNCTION_ARGS  )

Definition at line 357 of file geo_spgist.c.

References BOXOID, spgConfigOut::canReturnData, spgConfigOut::labelType, spgConfigOut::longValuesOK, PG_GETARG_POINTER, PG_RETURN_VOID, spgConfigOut::prefixType, and VOIDOID.

358 {
360 
361  cfg->prefixType = BOXOID;
362  cfg->labelType = VOIDOID; /* We don't need node labels. */
363  cfg->canReturnData = true;
364  cfg->longValuesOK = false;
365 
366  PG_RETURN_VOID();
367 }
bool canReturnData
Definition: spgist.h:47
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:232
#define VOIDOID
Definition: pg_type.h:678
#define BOXOID
Definition: pg_type.h:394
bool longValuesOK
Definition: spgist.h:48
Oid prefixType
Definition: spgist.h:45
#define PG_RETURN_VOID()
Definition: fmgr.h:293
Oid labelType
Definition: spgist.h:46
Datum spg_box_quad_inner_consistent ( PG_FUNCTION_ARGS  )

Definition at line 464 of file geo_spgist.c.

References above4D(), spgInnerConsistentIn::allTheSame, below4D(), contain4D(), contained4D(), DatumGetBoxP, elog, ERROR, flag(), getRangeBox(), i, initRectBox(), left4D(), MemoryContextSwitchTo(), nextRectBox(), spgInnerConsistentIn::nkeys, spgInnerConsistentIn::nNodes, spgInnerConsistentOut::nNodes, spgInnerConsistentOut::nodeNumbers, overAbove4D(), overBelow4D(), overlap4D(), overLeft4D(), overRight4D(), palloc(), pfree(), PG_GETARG_POINTER, PG_RETURN_VOID, spgInnerConsistentIn::prefixDatum, right4D(), RTAboveStrategyNumber, RTBelowStrategyNumber, RTContainedByStrategyNumber, RTContainsStrategyNumber, RTLeftStrategyNumber, RTOverAboveStrategyNumber, RTOverBelowStrategyNumber, RTOverlapStrategyNumber, RTOverLeftStrategyNumber, RTOverRightStrategyNumber, RTRightStrategyNumber, RTSameStrategyNumber, spgInnerConsistentIn::scankeys, ScanKeyData::sk_argument, ScanKeyData::sk_strategy, spgInnerConsistentIn::traversalMemoryContext, spgInnerConsistentIn::traversalValue, and spgInnerConsistentOut::traversalValues.

465 {
468  int i;
469  MemoryContext old_ctx;
470  RectBox *rect_box;
471  uint8 quadrant;
472  RangeBox *centroid,
473  **queries;
474 
475  if (in->allTheSame)
476  {
477  /* Report that all nodes should be visited */
478  out->nNodes = in->nNodes;
479  out->nodeNumbers = (int *) palloc(sizeof(int) * in->nNodes);
480  for (i = 0; i < in->nNodes; i++)
481  out->nodeNumbers[i] = i;
482 
483  PG_RETURN_VOID();
484  }
485 
486  /*
487  * We are saving the traversal value or initialize it an unbounded one, if
488  * we have just begun to walk the tree.
489  */
490  if (in->traversalValue)
491  rect_box = in->traversalValue;
492  else
493  rect_box = initRectBox();
494 
495  /*
496  * We are casting the prefix and queries to RangeBoxes for ease of the
497  * following operations.
498  */
499  centroid = getRangeBox(DatumGetBoxP(in->prefixDatum));
500  queries = (RangeBox **) palloc(in->nkeys * sizeof(RangeBox *));
501  for (i = 0; i < in->nkeys; i++)
502  queries[i] = getRangeBox(DatumGetBoxP(in->scankeys[i].sk_argument));
503 
504  /* Allocate enough memory for nodes */
505  out->nNodes = 0;
506  out->nodeNumbers = (int *) palloc(sizeof(int) * in->nNodes);
507  out->traversalValues = (void **) palloc(sizeof(void *) * in->nNodes);
508 
509  /*
510  * We switch memory context, because we want to allocate memory for new
511  * traversal values (next_rect_box) and pass these pieces of memory to
512  * further call of this function.
513  */
515 
516  for (quadrant = 0; quadrant < in->nNodes; quadrant++)
517  {
518  RectBox *next_rect_box = nextRectBox(rect_box, centroid, quadrant);
519  bool flag = true;
520 
521  for (i = 0; i < in->nkeys; i++)
522  {
523  StrategyNumber strategy = in->scankeys[i].sk_strategy;
524 
525  switch (strategy)
526  {
528  flag = overlap4D(next_rect_box, queries[i]);
529  break;
530 
532  flag = contain4D(next_rect_box, queries[i]);
533  break;
534 
537  flag = contained4D(next_rect_box, queries[i]);
538  break;
539 
541  flag = left4D(next_rect_box, queries[i]);
542  break;
543 
545  flag = overLeft4D(next_rect_box, queries[i]);
546  break;
547 
549  flag = right4D(next_rect_box, queries[i]);
550  break;
551 
553  flag = overRight4D(next_rect_box, queries[i]);
554  break;
555 
557  flag = above4D(next_rect_box, queries[i]);
558  break;
559 
561  flag = overAbove4D(next_rect_box, queries[i]);
562  break;
563 
565  flag = below4D(next_rect_box, queries[i]);
566  break;
567 
569  flag = overBelow4D(next_rect_box, queries[i]);
570  break;
571 
572  default:
573  elog(ERROR, "unrecognized strategy: %d", strategy);
574  }
575 
576  /* If any check is failed, we have found our answer. */
577  if (!flag)
578  break;
579  }
580 
581  if (flag)
582  {
583  out->traversalValues[out->nNodes] = next_rect_box;
584  out->nodeNumbers[out->nNodes] = quadrant;
585  out->nNodes++;
586  }
587  else
588  {
589  /*
590  * If this node is not selected, we don't need to keep the next
591  * traversal value in the memory context.
592  */
593  pfree(next_rect_box);
594  }
595  }
596 
597  /* Switch back */
598  MemoryContextSwitchTo(old_ctx);
599 
600  PG_RETURN_VOID();
601 }
static bool overAbove4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:348
static bool overBelow4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:334
static bool contain4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:257
#define RTLeftStrategyNumber
Definition: stratnum.h:44
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:109
unsigned char uint8
Definition: c.h:263
void * traversalValue
Definition: spgist.h:139
#define RTOverBelowStrategyNumber
Definition: stratnum.h:52
static bool overRight4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:320
uint16 StrategyNumber
Definition: stratnum.h:22
#define RTContainedByStrategyNumber
Definition: stratnum.h:51
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:232
static bool left4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:299
#define RTBelowStrategyNumber
Definition: stratnum.h:53
#define RTOverAboveStrategyNumber
Definition: stratnum.h:55
static bool below4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:327
void pfree(void *pointer)
Definition: mcxt.c:992
static bool overlap4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:241
MemoryContext traversalMemoryContext
Definition: spgist.h:140
#define ERROR
Definition: elog.h:43
StrategyNumber sk_strategy
Definition: skey.h:68
char * flag(int b)
Definition: test-ctype.c:33
#define RTSameStrategyNumber
Definition: stratnum.h:49
#define RTOverRightStrategyNumber
Definition: stratnum.h:47
void ** traversalValues
Definition: spgist.h:159
static bool above4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:341
#define RTOverLeftStrategyNumber
Definition: stratnum.h:45
static RectBox * initRectBox(void)
Definition: geo_spgist.c:174
#define PG_RETURN_VOID()
Definition: fmgr.h:293
#define RTAboveStrategyNumber
Definition: stratnum.h:54
#define RTRightStrategyNumber
Definition: stratnum.h:48
static bool contained4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:275
#define RTContainsStrategyNumber
Definition: stratnum.h:50
ScanKey scankeys
Definition: spgist.h:135
#define DatumGetBoxP(X)
Definition: geo_decls.h:159
void * palloc(Size size)
Definition: mcxt.c:891
static bool overLeft4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:306
int i
#define RTOverlapStrategyNumber
Definition: stratnum.h:46
#define elog
Definition: elog.h:219
static bool right4D(RectBox *rect_box, RangeBox *query)
Definition: geo_spgist.c:313
static RangeBox * getRangeBox(BOX *box)
Definition: geo_spgist.c:154
Datum sk_argument
Definition: skey.h:72
static RectBox * nextRectBox(RectBox *rect_box, RangeBox *centroid, uint8 quadrant)
Definition: geo_spgist.c:202
Datum spg_box_quad_leaf_consistent ( PG_FUNCTION_ARGS  )

Definition at line 607 of file geo_spgist.c.

References box_above(), box_below(), box_contain(), box_contained(), box_left(), box_overabove(), box_overbelow(), box_overlap(), box_overleft(), box_overright(), box_right(), box_same(), DatumGetBool, DirectFunctionCall2, elog, ERROR, flag(), i, spgLeafConsistentIn::leafDatum, spgLeafConsistentOut::leafValue, spgLeafConsistentIn::nkeys, PG_GETARG_POINTER, PG_RETURN_BOOL, spgLeafConsistentOut::recheck, RTAboveStrategyNumber, RTBelowStrategyNumber, RTContainedByStrategyNumber, RTContainsStrategyNumber, RTLeftStrategyNumber, RTOverAboveStrategyNumber, RTOverBelowStrategyNumber, RTOverlapStrategyNumber, RTOverLeftStrategyNumber, RTOverRightStrategyNumber, RTRightStrategyNumber, RTSameStrategyNumber, spgLeafConsistentIn::scankeys, ScanKeyData::sk_argument, and ScanKeyData::sk_strategy.

608 {
611  Datum leaf = in->leafDatum;
612  bool flag = true;
613  int i;
614 
615  /* All tests are exact. */
616  out->recheck = false;
617 
618  /* leafDatum is what it is... */
619  out->leafValue = in->leafDatum;
620 
621  /* Perform the required comparison(s) */
622  for (i = 0; i < in->nkeys; i++)
623  {
624  StrategyNumber strategy = in->scankeys[i].sk_strategy;
625  Datum query = in->scankeys[i].sk_argument;
626 
627  switch (strategy)
628  {
631  query));
632  break;
633 
636  query));
637  break;
638 
641  query));
642  break;
643 
646  query));
647  break;
648 
651  query));
652  break;
653 
656  query));
657  break;
658 
661  query));
662  break;
663 
666  query));
667  break;
668 
671  query));
672  break;
673 
676  query));
677  break;
678 
681  query));
682  break;
683 
686  query));
687  break;
688 
689  default:
690  elog(ERROR, "unrecognized strategy: %d", strategy);
691  }
692 
693  /* If any check is failed, we have found our answer. */
694  if (!flag)
695  break;
696  }
697 
698  PG_RETURN_BOOL(flag);
699 }
Datum box_right(PG_FUNCTION_ARGS)
Definition: geo_ops.c:564
Datum box_contained(PG_FUNCTION_ARGS)
Definition: geo_ops.c:636
Datum box_same(PG_FUNCTION_ARGS)
Definition: geo_ops.c:504
Datum box_left(PG_FUNCTION_ARGS)
Definition: geo_ops.c:538
#define RTLeftStrategyNumber
Definition: stratnum.h:44
Datum box_overright(PG_FUNCTION_ARGS)
Definition: geo_ops.c:579
Datum box_overlap(PG_FUNCTION_ARGS)
Definition: geo_ops.c:518
#define RTOverBelowStrategyNumber
Definition: stratnum.h:52
uint16 StrategyNumber
Definition: stratnum.h:22
#define RTContainedByStrategyNumber
Definition: stratnum.h:51
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:232
#define RTBelowStrategyNumber
Definition: stratnum.h:53
#define RTOverAboveStrategyNumber
Definition: stratnum.h:55
#define ERROR
Definition: elog.h:43
StrategyNumber sk_strategy
Definition: skey.h:68
ScanKey scankeys
Definition: spgist.h:167
Datum box_contain(PG_FUNCTION_ARGS)
Definition: geo_ops.c:650
Datum box_above(PG_FUNCTION_ARGS)
Definition: geo_ops.c:613
char * flag(int b)
Definition: test-ctype.c:33
#define DatumGetBool(X)
Definition: postgres.h:401
#define RTSameStrategyNumber
Definition: stratnum.h:49
#define RTOverRightStrategyNumber
Definition: stratnum.h:47
#define RTOverLeftStrategyNumber
Definition: stratnum.h:45
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:303
uintptr_t Datum
Definition: postgres.h:374
Datum box_overbelow(PG_FUNCTION_ARGS)
Definition: geo_ops.c:602
Datum box_below(PG_FUNCTION_ARGS)
Definition: geo_ops.c:590
#define RTAboveStrategyNumber
Definition: stratnum.h:54
#define RTRightStrategyNumber
Definition: stratnum.h:48
Datum box_overleft(PG_FUNCTION_ARGS)
Definition: geo_ops.c:553
#define RTContainsStrategyNumber
Definition: stratnum.h:50
int i
#define RTOverlapStrategyNumber
Definition: stratnum.h:46
Datum box_overabove(PG_FUNCTION_ARGS)
Definition: geo_ops.c:625
#define elog
Definition: elog.h:219
Datum sk_argument
Definition: skey.h:72
#define DirectFunctionCall2(func, arg1, arg2)
Definition: fmgr.h:557
Datum spg_box_quad_picksplit ( PG_FUNCTION_ARGS  )

Definition at line 397 of file geo_spgist.c.

References BoxPGetDatum, compareDoubles(), DatumGetBoxP, spgPickSplitIn::datums, getQuadrant(), spgPickSplitOut::hasPrefix, BOX::high, i, spgPickSplitOut::leafTupleDatums, BOX::low, spgPickSplitOut::mapTuplesToNodes, spgPickSplitOut::nNodes, spgPickSplitOut::nodeLabels, spgPickSplitIn::nTuples, NULL, palloc(), PG_GETARG_POINTER, PG_RETURN_VOID, spgPickSplitOut::prefixDatum, qsort, Point::x, and Point::y.

398 {
401  BOX *centroid;
402  int median,
403  i;
404  double *lowXs = palloc(sizeof(double) * in->nTuples);
405  double *highXs = palloc(sizeof(double) * in->nTuples);
406  double *lowYs = palloc(sizeof(double) * in->nTuples);
407  double *highYs = palloc(sizeof(double) * in->nTuples);
408 
409  /* Calculate median of all 4D coordinates */
410  for (i = 0; i < in->nTuples; i++)
411  {
412  BOX *box = DatumGetBoxP(in->datums[i]);
413 
414  lowXs[i] = box->low.x;
415  highXs[i] = box->high.x;
416  lowYs[i] = box->low.y;
417  highYs[i] = box->high.y;
418  }
419 
420  qsort(lowXs, in->nTuples, sizeof(double), compareDoubles);
421  qsort(highXs, in->nTuples, sizeof(double), compareDoubles);
422  qsort(lowYs, in->nTuples, sizeof(double), compareDoubles);
423  qsort(highYs, in->nTuples, sizeof(double), compareDoubles);
424 
425  median = in->nTuples / 2;
426 
427  centroid = palloc(sizeof(BOX));
428 
429  centroid->low.x = lowXs[median];
430  centroid->high.x = highXs[median];
431  centroid->low.y = lowYs[median];
432  centroid->high.y = highYs[median];
433 
434  /* Fill the output */
435  out->hasPrefix = true;
436  out->prefixDatum = BoxPGetDatum(centroid);
437 
438  out->nNodes = 16;
439  out->nodeLabels = NULL; /* We don't need node labels. */
440 
441  out->mapTuplesToNodes = palloc(sizeof(int) * in->nTuples);
442  out->leafTupleDatums = palloc(sizeof(Datum) * in->nTuples);
443 
444  /*
445  * Assign ranges to corresponding nodes according to quadrants relative to
446  * the "centroid" range
447  */
448  for (i = 0; i < in->nTuples; i++)
449  {
450  BOX *box = DatumGetBoxP(in->datums[i]);
451  uint8 quadrant = getQuadrant(centroid, box);
452 
453  out->leafTupleDatums[i] = BoxPGetDatum(box);
454  out->mapTuplesToNodes[i] = quadrant;
455  }
456 
457  PG_RETURN_VOID();
458 }
Definition: geo_decls.h:102
Datum * leafTupleDatums
Definition: spgist.h:127
Datum * datums
Definition: spgist.h:114
double y
Definition: geo_decls.h:60
unsigned char uint8
Definition: c.h:263
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:232
double x
Definition: geo_decls.h:60
Datum * nodeLabels
Definition: spgist.h:124
Point low
Definition: geo_decls.h:104
uintptr_t Datum
Definition: postgres.h:374
#define BoxPGetDatum(X)
Definition: geo_decls.h:160
#define PG_RETURN_VOID()
Definition: fmgr.h:293
#define NULL
Definition: c.h:226
bool hasPrefix
Definition: spgist.h:120
#define DatumGetBoxP(X)
Definition: geo_decls.h:159
Point high
Definition: geo_decls.h:104
static uint8 getQuadrant(BOX *centroid, BOX *inBox)
Definition: geo_spgist.c:127
void * palloc(Size size)
Definition: mcxt.c:891
int i
int * mapTuplesToNodes
Definition: spgist.h:126
static int compareDoubles(const void *a, const void *b)
Definition: geo_spgist.c:90
Datum prefixDatum
Definition: spgist.h:121
#define qsort(a, b, c, d)
Definition: port.h:440