chipmunk 5.3.4.5 → 6.1.3.0.rc1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (73) hide show
  1. data/ext/chipmunk/chipmunk.c +199 -28
  2. data/ext/chipmunk/chipmunk.h +123 -68
  3. data/ext/chipmunk/chipmunk_ffi.h +129 -11
  4. data/ext/chipmunk/chipmunk_private.h +232 -16
  5. data/ext/chipmunk/chipmunk_types.h +94 -30
  6. data/ext/chipmunk/chipmunk_unsafe.h +12 -3
  7. data/ext/chipmunk/constraints/cpConstraint.h +90 -34
  8. data/ext/chipmunk/{cpDampedRotarySpring.h → constraints/cpDampedRotarySpring.h} +18 -8
  9. data/ext/chipmunk/{cpDampedSpring.h → constraints/cpDampedSpring.h} +27 -16
  10. data/ext/chipmunk/constraints/cpGearJoint.h +17 -7
  11. data/ext/chipmunk/constraints/cpGrooveJoint.h +19 -10
  12. data/ext/chipmunk/constraints/cpPinJoint.h +17 -8
  13. data/ext/chipmunk/constraints/cpPivotJoint.h +18 -9
  14. data/ext/chipmunk/constraints/cpRatchetJoint.h +17 -8
  15. data/ext/chipmunk/constraints/cpRotaryLimitJoint.h +16 -7
  16. data/ext/chipmunk/{cpSimpleMotor.h → constraints/cpSimpleMotor.h} +15 -6
  17. data/ext/chipmunk/constraints/cpSlideJoint.h +18 -9
  18. data/ext/chipmunk/constraints/util.h +36 -44
  19. data/ext/chipmunk/cpArbiter.c +159 -94
  20. data/ext/chipmunk/cpArbiter.h +135 -129
  21. data/ext/chipmunk/cpArray.c +37 -56
  22. data/ext/chipmunk/cpBB.c +1 -12
  23. data/ext/chipmunk/cpBB.h +80 -18
  24. data/ext/chipmunk/cpBBTree.c +891 -0
  25. data/ext/chipmunk/cpBody.c +185 -47
  26. data/ext/chipmunk/cpBody.h +156 -124
  27. data/ext/chipmunk/cpCollision.c +126 -115
  28. data/ext/chipmunk/cpConstraint.c +10 -6
  29. data/ext/chipmunk/cpDampedRotarySpring.c +26 -17
  30. data/ext/chipmunk/cpDampedSpring.c +25 -18
  31. data/ext/chipmunk/cpGearJoint.c +23 -17
  32. data/ext/chipmunk/cpGrooveJoint.c +26 -22
  33. data/ext/chipmunk/cpHashSet.c +51 -51
  34. data/ext/chipmunk/cpPinJoint.c +26 -19
  35. data/ext/chipmunk/cpPivotJoint.c +23 -19
  36. data/ext/chipmunk/cpPolyShape.c +93 -69
  37. data/ext/chipmunk/cpPolyShape.h +33 -69
  38. data/ext/chipmunk/cpRatchetJoint.c +26 -21
  39. data/ext/chipmunk/cpRotaryLimitJoint.c +28 -22
  40. data/ext/chipmunk/cpShape.c +122 -133
  41. data/ext/chipmunk/cpShape.h +146 -95
  42. data/ext/chipmunk/cpSimpleMotor.c +24 -17
  43. data/ext/chipmunk/cpSlideJoint.c +28 -26
  44. data/ext/chipmunk/cpSpace.c +251 -196
  45. data/ext/chipmunk/cpSpace.h +173 -103
  46. data/ext/chipmunk/cpSpaceComponent.c +236 -159
  47. data/ext/chipmunk/cpSpaceHash.c +259 -159
  48. data/ext/chipmunk/cpSpaceQuery.c +127 -59
  49. data/ext/chipmunk/cpSpaceStep.c +235 -197
  50. data/ext/chipmunk/cpSpatialIndex.c +69 -0
  51. data/ext/chipmunk/cpSpatialIndex.h +227 -0
  52. data/ext/chipmunk/cpSweep1D.c +254 -0
  53. data/ext/chipmunk/cpVect.c +11 -26
  54. data/ext/chipmunk/cpVect.h +76 -71
  55. data/ext/chipmunk/extconf.rb +4 -31
  56. data/ext/chipmunk/prime.h +1 -1
  57. data/ext/chipmunk/rb_chipmunk.c +36 -45
  58. data/ext/chipmunk/rb_chipmunk.h +6 -3
  59. data/ext/chipmunk/rb_cpArbiter.c +2 -2
  60. data/ext/chipmunk/rb_cpBB.c +116 -35
  61. data/ext/chipmunk/rb_cpBody.c +5 -12
  62. data/ext/chipmunk/rb_cpConstraint.c +144 -9
  63. data/ext/chipmunk/rb_cpShape.c +69 -78
  64. data/ext/chipmunk/rb_cpSpace.c +81 -76
  65. metadata +61 -61
  66. data/LICENSE +0 -22
  67. data/README +0 -110
  68. data/Rakefile +0 -102
  69. data/ext/chipmunk/cpArray.h +0 -49
  70. data/ext/chipmunk/cpCollision.h +0 -28
  71. data/ext/chipmunk/cpHashSet.h +0 -82
  72. data/ext/chipmunk/cpSpaceHash.h +0 -110
  73. data/lib/chipmunk.rb +0 -194
@@ -18,39 +18,49 @@
18
18
  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
19
19
  * SOFTWARE.
20
20
  */
21
-
21
+
22
+ /// @defgroup cpBBB cpBB
23
+ /// Chipmunk's axis-aligned 2D bounding box type along with a few handy routines.
24
+ /// @{
25
+
26
+ /// Chipmunk's axis-aligned 2D bounding box type. (left, bottom, right, top)
22
27
  typedef struct cpBB{
23
28
  cpFloat l, b, r ,t;
24
29
  } cpBB;
25
30
 
26
- static inline cpBB
27
- cpBBNew(const cpFloat l, const cpFloat b,
28
- const cpFloat r, const cpFloat t)
31
+ /// Convenience constructor for cpBB structs.
32
+ static inline cpBB cpBBNew(const cpFloat l, const cpFloat b, const cpFloat r, const cpFloat t)
29
33
  {
30
34
  cpBB bb = {l, b, r, t};
31
35
  return bb;
32
36
  }
33
37
 
34
- static inline cpBool
35
- cpBBintersects(const cpBB a, const cpBB b)
38
+ /// Constructs a cpBB for a circle with the given position and radius.
39
+ static inline cpBB cpBBNewForCircle(const cpVect p, const cpFloat r)
40
+ {
41
+ return cpBBNew(p.x - r, p.y - r, p.x + r, p.y + r);
42
+ }
43
+
44
+ /// Returns true if @c a and @c b intersect.
45
+ static inline cpBool cpBBIntersects(const cpBB a, const cpBB b)
36
46
  {
37
- return (a.l<=b.r && b.l<=a.r && a.b<=b.t && b.b<=a.t);
47
+ return (a.l <= b.r && b.l <= a.r && a.b <= b.t && b.b <= a.t);
38
48
  }
39
49
 
40
- static inline cpBool
41
- cpBBcontainsBB(const cpBB bb, const cpBB other)
50
+ /// Returns true if @c other lies completely within @c bb.
51
+ static inline cpBool cpBBContainsBB(const cpBB bb, const cpBB other)
42
52
  {
43
- return (bb.l < other.l && bb.r > other.r && bb.b < other.b && bb.t > other.t);
53
+ return (bb.l <= other.l && bb.r >= other.r && bb.b <= other.b && bb.t >= other.t);
44
54
  }
45
55
 
46
- static inline cpBool
47
- cpBBcontainsVect(const cpBB bb, const cpVect v)
56
+ /// Returns true if @c bb contains @c v.
57
+ static inline cpBool cpBBContainsVect(const cpBB bb, const cpVect v)
48
58
  {
49
- return (bb.l < v.x && bb.r > v.x && bb.b < v.y && bb.t > v.y);
59
+ return (bb.l <= v.x && bb.r >= v.x && bb.b <= v.y && bb.t >= v.y);
50
60
  }
51
61
 
52
- static inline cpBB
53
- cpBBmerge(const cpBB a, const cpBB b){
62
+ /// Returns a bounding box that holds both bounding boxes.
63
+ static inline cpBB cpBBMerge(const cpBB a, const cpBB b){
54
64
  return cpBBNew(
55
65
  cpfmin(a.l, b.l),
56
66
  cpfmin(a.b, b.b),
@@ -59,8 +69,8 @@ cpBBmerge(const cpBB a, const cpBB b){
59
69
  );
60
70
  }
61
71
 
62
- static inline cpBB
63
- cpBBexpand(const cpBB bb, const cpVect v){
72
+ /// Returns a bounding box that holds both @c bb and @c v.
73
+ static inline cpBB cpBBExpand(const cpBB bb, const cpVect v){
64
74
  return cpBBNew(
65
75
  cpfmin(bb.l, v.x),
66
76
  cpfmin(bb.b, v.y),
@@ -69,6 +79,58 @@ cpBBexpand(const cpBB bb, const cpVect v){
69
79
  );
70
80
  }
71
81
 
72
- cpVect cpBBClampVect(const cpBB bb, const cpVect v); // clamps the vector to lie within the bbox
82
+ /// Returns the area of the bounding box.
83
+ static inline cpFloat cpBBArea(cpBB bb)
84
+ {
85
+ return (bb.r - bb.l)*(bb.t - bb.b);
86
+ }
87
+
88
+ /// Merges @c a and @c b and returns the area of the merged bounding box.
89
+ static inline cpFloat cpBBMergedArea(cpBB a, cpBB b)
90
+ {
91
+ return (cpfmax(a.r, b.r) - cpfmin(a.l, b.l))*(cpfmax(a.t, b.t) - cpfmin(a.b, b.b));
92
+ }
93
+
94
+ /// Returns the fraction along the segment query the cpBB is hit. Returns INFINITY if it doesn't hit.
95
+ static inline cpFloat cpBBSegmentQuery(cpBB bb, cpVect a, cpVect b)
96
+ {
97
+ cpFloat idx = 1.0f/(b.x - a.x);
98
+ cpFloat tx1 = (bb.l == a.x ? -INFINITY : (bb.l - a.x)*idx);
99
+ cpFloat tx2 = (bb.r == a.x ? INFINITY : (bb.r - a.x)*idx);
100
+ cpFloat txmin = cpfmin(tx1, tx2);
101
+ cpFloat txmax = cpfmax(tx1, tx2);
102
+
103
+ cpFloat idy = 1.0f/(b.y - a.y);
104
+ cpFloat ty1 = (bb.b == a.y ? -INFINITY : (bb.b - a.y)*idy);
105
+ cpFloat ty2 = (bb.t == a.y ? INFINITY : (bb.t - a.y)*idy);
106
+ cpFloat tymin = cpfmin(ty1, ty2);
107
+ cpFloat tymax = cpfmax(ty1, ty2);
108
+
109
+ if(tymin <= txmax && txmin <= tymax){
110
+ cpFloat min = cpfmax(txmin, tymin);
111
+ cpFloat max = cpfmin(txmax, tymax);
112
+
113
+ if(0.0 <= max && min <= 1.0) return cpfmax(min, 0.0);
114
+ }
115
+
116
+ return INFINITY;
117
+ }
118
+
119
+ /// Return true if the bounding box intersects the line segment with ends @c a and @c b.
120
+ static inline cpBool cpBBIntersectsSegment(cpBB bb, cpVect a, cpVect b)
121
+ {
122
+ return (cpBBSegmentQuery(bb, a, b) != INFINITY);
123
+ }
124
+
125
+ /// Clamp a vector to a bounding box.
126
+ static inline cpVect
127
+ cpBBClampVect(const cpBB bb, const cpVect v)
128
+ {
129
+ return cpv(cpfclamp(v.x, bb.l, bb.r), cpfclamp(v.y, bb.b, bb.t));
130
+ }
131
+
73
132
  // TODO edge case issue
133
+ /// Wrap a vector to a bounding box.
74
134
  cpVect cpBBWrapVect(const cpBB bb, const cpVect v); // wrap a vector to a bbox
135
+
136
+ ///@}
@@ -0,0 +1,891 @@
1
+ /* Copyright (c) 2009 Scott Lembcke
2
+ *
3
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
4
+ * of this software and associated documentation files (the "Software"), to deal
5
+ * in the Software without restriction, including without limitation the rights
6
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7
+ * copies of the Software, and to permit persons to whom the Software is
8
+ * furnished to do so, subject to the following conditions:
9
+ *
10
+ * The above copyright notice and this permission notice shall be included in
11
+ * all copies or substantial portions of the Software.
12
+ *
13
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
19
+ * SOFTWARE.
20
+ */
21
+
22
+ #include "stdlib.h"
23
+ #include "stdio.h"
24
+
25
+ #include "chipmunk_private.h"
26
+
27
+ static inline cpSpatialIndexClass *Klass();
28
+
29
+ typedef struct Node Node;
30
+ typedef struct Pair Pair;
31
+
32
+ struct cpBBTree {
33
+ cpSpatialIndex spatialIndex;
34
+ cpBBTreeVelocityFunc velocityFunc;
35
+
36
+ cpHashSet *leaves;
37
+ Node *root;
38
+
39
+ Node *pooledNodes;
40
+ Pair *pooledPairs;
41
+ cpArray *allocatedBuffers;
42
+
43
+ cpTimestamp stamp;
44
+ };
45
+
46
+ struct Node {
47
+ void *obj;
48
+ cpBB bb;
49
+ Node *parent;
50
+
51
+ union {
52
+ // Internal nodes
53
+ struct { Node *a, *b; } children;
54
+
55
+ // Leaves
56
+ struct {
57
+ cpTimestamp stamp;
58
+ Pair *pairs;
59
+ } leaf;
60
+ } node;
61
+ };
62
+
63
+ // Can't use anonymous unions and still get good x-compiler compatability
64
+ #define A node.children.a
65
+ #define B node.children.b
66
+ #define STAMP node.leaf.stamp
67
+ #define PAIRS node.leaf.pairs
68
+
69
+ typedef struct Thread {
70
+ Pair *prev;
71
+ Node *leaf;
72
+ Pair *next;
73
+ } Thread;
74
+
75
+ struct Pair { Thread a, b; };
76
+
77
+ //MARK: Misc Functions
78
+
79
+ static inline cpBB
80
+ GetBB(cpBBTree *tree, void *obj)
81
+ {
82
+ cpBB bb = tree->spatialIndex.bbfunc(obj);
83
+
84
+ cpBBTreeVelocityFunc velocityFunc = tree->velocityFunc;
85
+ if(velocityFunc){
86
+ cpFloat coef = 0.1f;
87
+ cpFloat x = (bb.r - bb.l)*coef;
88
+ cpFloat y = (bb.t - bb.b)*coef;
89
+
90
+ cpVect v = cpvmult(velocityFunc(obj), 0.1f);
91
+ return cpBBNew(bb.l + cpfmin(-x, v.x), bb.b + cpfmin(-y, v.y), bb.r + cpfmax(x, v.x), bb.t + cpfmax(y, v.y));
92
+ } else {
93
+ return bb;
94
+ }
95
+ }
96
+
97
+ static inline cpBBTree *
98
+ GetTree(cpSpatialIndex *index)
99
+ {
100
+ return (index && index->klass == Klass() ? (cpBBTree *)index : NULL);
101
+ }
102
+
103
+ static inline Node *
104
+ GetRootIfTree(cpSpatialIndex *index){
105
+ return (index && index->klass == Klass() ? ((cpBBTree *)index)->root : NULL);
106
+ }
107
+
108
+ static inline cpBBTree *
109
+ GetMasterTree(cpBBTree *tree)
110
+ {
111
+ cpBBTree *dynamicTree = GetTree(tree->spatialIndex.dynamicIndex);
112
+ return (dynamicTree ? dynamicTree : tree);
113
+ }
114
+
115
+ static inline void
116
+ IncrementStamp(cpBBTree *tree)
117
+ {
118
+ cpBBTree *dynamicTree = GetTree(tree->spatialIndex.dynamicIndex);
119
+ if(dynamicTree){
120
+ dynamicTree->stamp++;
121
+ } else {
122
+ tree->stamp++;
123
+ }
124
+ }
125
+
126
+ //MARK: Pair/Thread Functions
127
+
128
+ static void
129
+ PairRecycle(cpBBTree *tree, Pair *pair)
130
+ {
131
+ // Share the pool of the master tree.
132
+ // TODO would be lovely to move the pairs stuff into an external data structure.
133
+ tree = GetMasterTree(tree);
134
+
135
+ pair->a.next = tree->pooledPairs;
136
+ tree->pooledPairs = pair;
137
+ }
138
+
139
+ static Pair *
140
+ PairFromPool(cpBBTree *tree)
141
+ {
142
+ // Share the pool of the master tree.
143
+ // TODO would be lovely to move the pairs stuff into an external data structure.
144
+ tree = GetMasterTree(tree);
145
+
146
+ Pair *pair = tree->pooledPairs;
147
+
148
+ if(pair){
149
+ tree->pooledPairs = pair->a.next;
150
+ return pair;
151
+ } else {
152
+ // Pool is exhausted, make more
153
+ int count = CP_BUFFER_BYTES/sizeof(Pair);
154
+ cpAssertHard(count, "Internal Error: Buffer size is too small.");
155
+
156
+ Pair *buffer = (Pair *)cpcalloc(1, CP_BUFFER_BYTES);
157
+ cpArrayPush(tree->allocatedBuffers, buffer);
158
+
159
+ // push all but the first one, return the first instead
160
+ for(int i=1; i<count; i++) PairRecycle(tree, buffer + i);
161
+ return buffer;
162
+ }
163
+ }
164
+
165
+ static inline void
166
+ ThreadUnlink(Thread thread)
167
+ {
168
+ Pair *next = thread.next;
169
+ Pair *prev = thread.prev;
170
+
171
+ if(next){
172
+ if(next->a.leaf == thread.leaf) next->a.prev = prev; else next->b.prev = prev;
173
+ }
174
+
175
+ if(prev){
176
+ if(prev->a.leaf == thread.leaf) prev->a.next = next; else prev->b.next = next;
177
+ } else {
178
+ thread.leaf->PAIRS = next;
179
+ }
180
+ }
181
+
182
+ static void
183
+ PairsClear(Node *leaf, cpBBTree *tree)
184
+ {
185
+ Pair *pair = leaf->PAIRS;
186
+ leaf->PAIRS = NULL;
187
+
188
+ while(pair){
189
+ if(pair->a.leaf == leaf){
190
+ Pair *next = pair->a.next;
191
+ ThreadUnlink(pair->b);
192
+ PairRecycle(tree, pair);
193
+ pair = next;
194
+ } else {
195
+ Pair *next = pair->b.next;
196
+ ThreadUnlink(pair->a);
197
+ PairRecycle(tree, pair);
198
+ pair = next;
199
+ }
200
+ }
201
+ }
202
+
203
+ static void
204
+ PairInsert(Node *a, Node *b, cpBBTree *tree)
205
+ {
206
+ Pair *nextA = a->PAIRS, *nextB = b->PAIRS;
207
+ Pair *pair = PairFromPool(tree);
208
+ Pair temp = {{NULL, a, nextA},{NULL, b, nextB}};
209
+
210
+ a->PAIRS = b->PAIRS = pair;
211
+ *pair = temp;
212
+
213
+ if(nextA){
214
+ if(nextA->a.leaf == a) nextA->a.prev = pair; else nextA->b.prev = pair;
215
+ }
216
+
217
+ if(nextB){
218
+ if(nextB->a.leaf == b) nextB->a.prev = pair; else nextB->b.prev = pair;
219
+ }
220
+ }
221
+
222
+
223
+ //MARK: Node Functions
224
+
225
+ static void
226
+ NodeRecycle(cpBBTree *tree, Node *node)
227
+ {
228
+ node->parent = tree->pooledNodes;
229
+ tree->pooledNodes = node;
230
+ }
231
+
232
+ static Node *
233
+ NodeFromPool(cpBBTree *tree)
234
+ {
235
+ Node *node = tree->pooledNodes;
236
+
237
+ if(node){
238
+ tree->pooledNodes = node->parent;
239
+ return node;
240
+ } else {
241
+ // Pool is exhausted, make more
242
+ int count = CP_BUFFER_BYTES/sizeof(Node);
243
+ cpAssertHard(count, "Internal Error: Buffer size is too small.");
244
+
245
+ Node *buffer = (Node *)cpcalloc(1, CP_BUFFER_BYTES);
246
+ cpArrayPush(tree->allocatedBuffers, buffer);
247
+
248
+ // push all but the first one, return the first instead
249
+ for(int i=1; i<count; i++) NodeRecycle(tree, buffer + i);
250
+ return buffer;
251
+ }
252
+ }
253
+
254
+ static inline void
255
+ NodeSetA(Node *node, Node *value)
256
+ {
257
+ node->A = value;
258
+ value->parent = node;
259
+ }
260
+
261
+ static inline void
262
+ NodeSetB(Node *node, Node *value)
263
+ {
264
+ node->B = value;
265
+ value->parent = node;
266
+ }
267
+
268
+ static Node *
269
+ NodeNew(cpBBTree *tree, Node *a, Node *b)
270
+ {
271
+ Node *node = NodeFromPool(tree);
272
+
273
+ node->obj = NULL;
274
+ node->bb = cpBBMerge(a->bb, b->bb);
275
+ node->parent = NULL;
276
+
277
+ NodeSetA(node, a);
278
+ NodeSetB(node, b);
279
+
280
+ return node;
281
+ }
282
+
283
+ static inline cpBool
284
+ NodeIsLeaf(Node *node)
285
+ {
286
+ return (node->obj != NULL);
287
+ }
288
+
289
+ static inline Node *
290
+ NodeOther(Node *node, Node *child)
291
+ {
292
+ return (node->A == child ? node->B : node->A);
293
+ }
294
+
295
+ static inline void
296
+ NodeReplaceChild(Node *parent, Node *child, Node *value, cpBBTree *tree)
297
+ {
298
+ cpAssertSoft(!NodeIsLeaf(parent), "Internal Error: Cannot replace child of a leaf.");
299
+ cpAssertSoft(child == parent->A || child == parent->B, "Internal Error: Node is not a child of parent.");
300
+
301
+ if(parent->A == child){
302
+ NodeRecycle(tree, parent->A);
303
+ NodeSetA(parent, value);
304
+ } else {
305
+ NodeRecycle(tree, parent->B);
306
+ NodeSetB(parent, value);
307
+ }
308
+
309
+ for(Node *node=parent; node; node = node->parent){
310
+ node->bb = cpBBMerge(node->A->bb, node->B->bb);
311
+ }
312
+ }
313
+
314
+ //MARK: Subtree Functions
315
+
316
+ static inline cpFloat
317
+ cpBBProximity(cpBB a, cpBB b)
318
+ {
319
+ return cpfabs(a.l + a.r - b.l - b.r) + cpfabs(a.b + a.t - b.b - b.t);
320
+ }
321
+
322
+ static Node *
323
+ SubtreeInsert(Node *subtree, Node *leaf, cpBBTree *tree)
324
+ {
325
+ if(subtree == NULL){
326
+ return leaf;
327
+ } else if(NodeIsLeaf(subtree)){
328
+ return NodeNew(tree, leaf, subtree);
329
+ } else {
330
+ cpFloat cost_a = cpBBArea(subtree->B->bb) + cpBBMergedArea(subtree->A->bb, leaf->bb);
331
+ cpFloat cost_b = cpBBArea(subtree->A->bb) + cpBBMergedArea(subtree->B->bb, leaf->bb);
332
+
333
+ if(cost_a == cost_b){
334
+ cost_a = cpBBProximity(subtree->A->bb, leaf->bb);
335
+ cost_b = cpBBProximity(subtree->B->bb, leaf->bb);
336
+ }
337
+
338
+ if(cost_b < cost_a){
339
+ NodeSetB(subtree, SubtreeInsert(subtree->B, leaf, tree));
340
+ } else {
341
+ NodeSetA(subtree, SubtreeInsert(subtree->A, leaf, tree));
342
+ }
343
+
344
+ subtree->bb = cpBBMerge(subtree->bb, leaf->bb);
345
+ return subtree;
346
+ }
347
+ }
348
+
349
+ static void
350
+ SubtreeQuery(Node *subtree, void *obj, cpBB bb, cpSpatialIndexQueryFunc func, void *data)
351
+ {
352
+ if(cpBBIntersects(subtree->bb, bb)){
353
+ if(NodeIsLeaf(subtree)){
354
+ func(obj, subtree->obj, data);
355
+ } else {
356
+ SubtreeQuery(subtree->A, obj, bb, func, data);
357
+ SubtreeQuery(subtree->B, obj, bb, func, data);
358
+ }
359
+ }
360
+ }
361
+
362
+
363
+ static cpFloat
364
+ SubtreeSegmentQuery(Node *subtree, void *obj, cpVect a, cpVect b, cpFloat t_exit, cpSpatialIndexSegmentQueryFunc func, void *data)
365
+ {
366
+ if(NodeIsLeaf(subtree)){
367
+ return func(obj, subtree->obj, data);
368
+ } else {
369
+ cpFloat t_a = cpBBSegmentQuery(subtree->A->bb, a, b);
370
+ cpFloat t_b = cpBBSegmentQuery(subtree->B->bb, a, b);
371
+
372
+ if(t_a < t_b){
373
+ if(t_a < t_exit) t_exit = cpfmin(t_exit, SubtreeSegmentQuery(subtree->A, obj, a, b, t_exit, func, data));
374
+ if(t_b < t_exit) t_exit = cpfmin(t_exit, SubtreeSegmentQuery(subtree->B, obj, a, b, t_exit, func, data));
375
+ } else {
376
+ if(t_b < t_exit) t_exit = cpfmin(t_exit, SubtreeSegmentQuery(subtree->B, obj, a, b, t_exit, func, data));
377
+ if(t_a < t_exit) t_exit = cpfmin(t_exit, SubtreeSegmentQuery(subtree->A, obj, a, b, t_exit, func, data));
378
+ }
379
+
380
+ return t_exit;
381
+ }
382
+ }
383
+
384
+ static void
385
+ SubtreeRecycle(cpBBTree *tree, Node *node)
386
+ {
387
+ if(!NodeIsLeaf(node)){
388
+ SubtreeRecycle(tree, node->A);
389
+ SubtreeRecycle(tree, node->B);
390
+ NodeRecycle(tree, node);
391
+ }
392
+ }
393
+
394
+ static inline Node *
395
+ SubtreeRemove(Node *subtree, Node *leaf, cpBBTree *tree)
396
+ {
397
+ if(leaf == subtree){
398
+ return NULL;
399
+ } else {
400
+ Node *parent = leaf->parent;
401
+ if(parent == subtree){
402
+ Node *other = NodeOther(subtree, leaf);
403
+ other->parent = subtree->parent;
404
+ NodeRecycle(tree, subtree);
405
+ return other;
406
+ } else {
407
+ NodeReplaceChild(parent->parent, parent, NodeOther(parent, leaf), tree);
408
+ return subtree;
409
+ }
410
+ }
411
+ }
412
+
413
+ //MARK: Marking Functions
414
+
415
+ typedef struct MarkContext {
416
+ cpBBTree *tree;
417
+ Node *staticRoot;
418
+ cpSpatialIndexQueryFunc func;
419
+ void *data;
420
+ } MarkContext;
421
+
422
+ static void
423
+ MarkLeafQuery(Node *subtree, Node *leaf, cpBool left, MarkContext *context)
424
+ {
425
+ if(cpBBIntersects(leaf->bb, subtree->bb)){
426
+ if(NodeIsLeaf(subtree)){
427
+ if(left){
428
+ PairInsert(leaf, subtree, context->tree);
429
+ } else {
430
+ if(subtree->STAMP < leaf->STAMP) PairInsert(subtree, leaf, context->tree);
431
+ context->func(leaf->obj, subtree->obj, context->data);
432
+ }
433
+ } else {
434
+ MarkLeafQuery(subtree->A, leaf, left, context);
435
+ MarkLeafQuery(subtree->B, leaf, left, context);
436
+ }
437
+ }
438
+ }
439
+
440
+ static void
441
+ MarkLeaf(Node *leaf, MarkContext *context)
442
+ {
443
+ cpBBTree *tree = context->tree;
444
+ if(leaf->STAMP == GetMasterTree(tree)->stamp){
445
+ Node *staticRoot = context->staticRoot;
446
+ if(staticRoot) MarkLeafQuery(staticRoot, leaf, cpFalse, context);
447
+
448
+ for(Node *node = leaf; node->parent; node = node->parent){
449
+ if(node == node->parent->A){
450
+ MarkLeafQuery(node->parent->B, leaf, cpTrue, context);
451
+ } else {
452
+ MarkLeafQuery(node->parent->A, leaf, cpFalse, context);
453
+ }
454
+ }
455
+ } else {
456
+ Pair *pair = leaf->PAIRS;
457
+ while(pair){
458
+ if(leaf == pair->b.leaf){
459
+ context->func(pair->a.leaf->obj, leaf->obj, context->data);
460
+ pair = pair->b.next;
461
+ } else {
462
+ pair = pair->a.next;
463
+ }
464
+ }
465
+ }
466
+ }
467
+
468
+ static void
469
+ MarkSubtree(Node *subtree, MarkContext *context)
470
+ {
471
+ if(NodeIsLeaf(subtree)){
472
+ MarkLeaf(subtree, context);
473
+ } else {
474
+ MarkSubtree(subtree->A, context);
475
+ MarkSubtree(subtree->B, context);
476
+ }
477
+ }
478
+
479
+ //MARK: Leaf Functions
480
+
481
+ static Node *
482
+ LeafNew(cpBBTree *tree, void *obj, cpBB bb)
483
+ {
484
+ Node *node = NodeFromPool(tree);
485
+ node->obj = obj;
486
+ node->bb = GetBB(tree, obj);
487
+
488
+ node->parent = NULL;
489
+ node->STAMP = 0;
490
+ node->PAIRS = NULL;
491
+
492
+ return node;
493
+ }
494
+
495
+ static cpBool
496
+ LeafUpdate(Node *leaf, cpBBTree *tree)
497
+ {
498
+ Node *root = tree->root;
499
+ cpBB bb = tree->spatialIndex.bbfunc(leaf->obj);
500
+
501
+ if(!cpBBContainsBB(leaf->bb, bb)){
502
+ leaf->bb = GetBB(tree, leaf->obj);
503
+
504
+ root = SubtreeRemove(root, leaf, tree);
505
+ tree->root = SubtreeInsert(root, leaf, tree);
506
+
507
+ PairsClear(leaf, tree);
508
+ leaf->STAMP = GetMasterTree(tree)->stamp;
509
+
510
+ return cpTrue;
511
+ }
512
+
513
+ return cpFalse;
514
+ }
515
+
516
+ static void VoidQueryFunc(void *obj1, void *obj2, void *data){}
517
+
518
+ static void
519
+ LeafAddPairs(Node *leaf, cpBBTree *tree)
520
+ {
521
+ cpSpatialIndex *dynamicIndex = tree->spatialIndex.dynamicIndex;
522
+ if(dynamicIndex){
523
+ Node *dynamicRoot = GetRootIfTree(dynamicIndex);
524
+ if(dynamicRoot){
525
+ cpBBTree *dynamicTree = GetTree(dynamicIndex);
526
+ MarkContext context = {dynamicTree, NULL, NULL, NULL};
527
+ MarkLeafQuery(dynamicRoot, leaf, cpTrue, &context);
528
+ }
529
+ } else {
530
+ Node *staticRoot = GetRootIfTree(tree->spatialIndex.staticIndex);
531
+ MarkContext context = {tree, staticRoot, VoidQueryFunc, NULL};
532
+ MarkLeaf(leaf, &context);
533
+ }
534
+ }
535
+
536
+ //MARK: Memory Management Functions
537
+
538
+ cpBBTree *
539
+ cpBBTreeAlloc(void)
540
+ {
541
+ return (cpBBTree *)cpcalloc(1, sizeof(cpBBTree));
542
+ }
543
+
544
+ static int
545
+ leafSetEql(void *obj, Node *node)
546
+ {
547
+ return (obj == node->obj);
548
+ }
549
+
550
+ static void *
551
+ leafSetTrans(void *obj, cpBBTree *tree)
552
+ {
553
+ return LeafNew(tree, obj, tree->spatialIndex.bbfunc(obj));
554
+ }
555
+
556
+ cpSpatialIndex *
557
+ cpBBTreeInit(cpBBTree *tree, cpSpatialIndexBBFunc bbfunc, cpSpatialIndex *staticIndex)
558
+ {
559
+ cpSpatialIndexInit((cpSpatialIndex *)tree, Klass(), bbfunc, staticIndex);
560
+
561
+ tree->velocityFunc = NULL;
562
+
563
+ tree->leaves = cpHashSetNew(0, (cpHashSetEqlFunc)leafSetEql);
564
+ tree->root = NULL;
565
+
566
+ tree->pooledNodes = NULL;
567
+ tree->allocatedBuffers = cpArrayNew(0);
568
+
569
+ tree->stamp = 0;
570
+
571
+ return (cpSpatialIndex *)tree;
572
+ }
573
+
574
+ void
575
+ cpBBTreeSetVelocityFunc(cpSpatialIndex *index, cpBBTreeVelocityFunc func)
576
+ {
577
+ if(index->klass != Klass()){
578
+ cpAssertWarn(cpFalse, "Ignoring cpBBTreeSetVelocityFunc() call to non-tree spatial index.");
579
+ return;
580
+ }
581
+
582
+ ((cpBBTree *)index)->velocityFunc = func;
583
+ }
584
+
585
+ cpSpatialIndex *
586
+ cpBBTreeNew(cpSpatialIndexBBFunc bbfunc, cpSpatialIndex *staticIndex)
587
+ {
588
+ return cpBBTreeInit(cpBBTreeAlloc(), bbfunc, staticIndex);
589
+ }
590
+
591
+ static void
592
+ cpBBTreeDestroy(cpBBTree *tree)
593
+ {
594
+ cpHashSetFree(tree->leaves);
595
+
596
+ if(tree->allocatedBuffers) cpArrayFreeEach(tree->allocatedBuffers, cpfree);
597
+ cpArrayFree(tree->allocatedBuffers);
598
+ }
599
+
600
+ //MARK: Insert/Remove
601
+
602
+ static void
603
+ cpBBTreeInsert(cpBBTree *tree, void *obj, cpHashValue hashid)
604
+ {
605
+ Node *leaf = (Node *)cpHashSetInsert(tree->leaves, hashid, obj, tree, (cpHashSetTransFunc)leafSetTrans);
606
+
607
+ Node *root = tree->root;
608
+ tree->root = SubtreeInsert(root, leaf, tree);
609
+
610
+ leaf->STAMP = GetMasterTree(tree)->stamp;
611
+ LeafAddPairs(leaf, tree);
612
+ IncrementStamp(tree);
613
+ }
614
+
615
+ static void
616
+ cpBBTreeRemove(cpBBTree *tree, void *obj, cpHashValue hashid)
617
+ {
618
+ Node *leaf = (Node *)cpHashSetRemove(tree->leaves, hashid, obj);
619
+
620
+ tree->root = SubtreeRemove(tree->root, leaf, tree);
621
+ PairsClear(leaf, tree);
622
+ NodeRecycle(tree, leaf);
623
+ }
624
+
625
+ static cpBool
626
+ cpBBTreeContains(cpBBTree *tree, void *obj, cpHashValue hashid)
627
+ {
628
+ return (cpHashSetFind(tree->leaves, hashid, obj) != NULL);
629
+ }
630
+
631
+ //MARK: Reindex
632
+
633
+ static void
634
+ cpBBTreeReindexQuery(cpBBTree *tree, cpSpatialIndexQueryFunc func, void *data)
635
+ {
636
+ if(!tree->root) return;
637
+
638
+ // LeafUpdate() may modify tree->root. Don't cache it.
639
+ cpHashSetEach(tree->leaves, (cpHashSetIteratorFunc)LeafUpdate, tree);
640
+
641
+ cpSpatialIndex *staticIndex = tree->spatialIndex.staticIndex;
642
+ Node *staticRoot = (staticIndex && staticIndex->klass == Klass() ? ((cpBBTree *)staticIndex)->root : NULL);
643
+
644
+ MarkContext context = {tree, staticRoot, func, data};
645
+ MarkSubtree(tree->root, &context);
646
+ if(staticIndex && !staticRoot) cpSpatialIndexCollideStatic((cpSpatialIndex *)tree, staticIndex, func, data);
647
+
648
+ IncrementStamp(tree);
649
+ }
650
+
651
+ static void
652
+ cpBBTreeReindex(cpBBTree *tree)
653
+ {
654
+ cpBBTreeReindexQuery(tree, VoidQueryFunc, NULL);
655
+ }
656
+
657
+ static void
658
+ cpBBTreeReindexObject(cpBBTree *tree, void *obj, cpHashValue hashid)
659
+ {
660
+ Node *leaf = (Node *)cpHashSetFind(tree->leaves, hashid, obj);
661
+ if(leaf){
662
+ if(LeafUpdate(leaf, tree)) LeafAddPairs(leaf, tree);
663
+ IncrementStamp(tree);
664
+ }
665
+ }
666
+
667
+ //MARK: Query
668
+
669
+ static void
670
+ cpBBTreeSegmentQuery(cpBBTree *tree, void *obj, cpVect a, cpVect b, cpFloat t_exit, cpSpatialIndexSegmentQueryFunc func, void *data)
671
+ {
672
+ Node *root = tree->root;
673
+ if(root) SubtreeSegmentQuery(root, obj, a, b, t_exit, func, data);
674
+ }
675
+
676
+ static void
677
+ cpBBTreeQuery(cpBBTree *tree, void *obj, cpBB bb, cpSpatialIndexQueryFunc func, void *data)
678
+ {
679
+ if(tree->root) SubtreeQuery(tree->root, obj, bb, func, data);
680
+ }
681
+
682
+ //MARK: Misc
683
+
684
+ static int
685
+ cpBBTreeCount(cpBBTree *tree)
686
+ {
687
+ return cpHashSetCount(tree->leaves);
688
+ }
689
+
690
+ typedef struct eachContext {
691
+ cpSpatialIndexIteratorFunc func;
692
+ void *data;
693
+ } eachContext;
694
+
695
+ static void each_helper(Node *node, eachContext *context){context->func(node->obj, context->data);}
696
+
697
+ static void
698
+ cpBBTreeEach(cpBBTree *tree, cpSpatialIndexIteratorFunc func, void *data)
699
+ {
700
+ eachContext context = {func, data};
701
+ cpHashSetEach(tree->leaves, (cpHashSetIteratorFunc)each_helper, &context);
702
+ }
703
+
704
+ static cpSpatialIndexClass klass = {
705
+ (cpSpatialIndexDestroyImpl)cpBBTreeDestroy,
706
+
707
+ (cpSpatialIndexCountImpl)cpBBTreeCount,
708
+ (cpSpatialIndexEachImpl)cpBBTreeEach,
709
+
710
+ (cpSpatialIndexContainsImpl)cpBBTreeContains,
711
+ (cpSpatialIndexInsertImpl)cpBBTreeInsert,
712
+ (cpSpatialIndexRemoveImpl)cpBBTreeRemove,
713
+
714
+ (cpSpatialIndexReindexImpl)cpBBTreeReindex,
715
+ (cpSpatialIndexReindexObjectImpl)cpBBTreeReindexObject,
716
+ (cpSpatialIndexReindexQueryImpl)cpBBTreeReindexQuery,
717
+
718
+ (cpSpatialIndexQueryImpl)cpBBTreeQuery,
719
+ (cpSpatialIndexSegmentQueryImpl)cpBBTreeSegmentQuery,
720
+ };
721
+
722
+ static inline cpSpatialIndexClass *Klass(){return &klass;}
723
+
724
+
725
+ //MARK: Tree Optimization
726
+
727
+ static int
728
+ cpfcompare(const cpFloat *a, const cpFloat *b){
729
+ return (*a < *b ? -1 : (*b < *a ? 1 : 0));
730
+ }
731
+
732
+ static void
733
+ fillNodeArray(Node *node, Node ***cursor){
734
+ (**cursor) = node;
735
+ (*cursor)++;
736
+ }
737
+
738
+ static Node *
739
+ partitionNodes(cpBBTree *tree, Node **nodes, int count)
740
+ {
741
+ if(count == 1){
742
+ return nodes[0];
743
+ } else if(count == 2) {
744
+ return NodeNew(tree, nodes[0], nodes[1]);
745
+ }
746
+
747
+ // Find the AABB for these nodes
748
+ cpBB bb = nodes[0]->bb;
749
+ for(int i=1; i<count; i++) bb = cpBBMerge(bb, nodes[i]->bb);
750
+
751
+ // Split it on it's longest axis
752
+ cpBool splitWidth = (bb.r - bb.l > bb.t - bb.b);
753
+
754
+ // Sort the bounds and use the median as the splitting point
755
+ cpFloat *bounds = (cpFloat *)cpcalloc(count*2, sizeof(cpFloat));
756
+ if(splitWidth){
757
+ for(int i=0; i<count; i++){
758
+ bounds[2*i + 0] = nodes[i]->bb.l;
759
+ bounds[2*i + 1] = nodes[i]->bb.r;
760
+ }
761
+ } else {
762
+ for(int i=0; i<count; i++){
763
+ bounds[2*i + 0] = nodes[i]->bb.b;
764
+ bounds[2*i + 1] = nodes[i]->bb.t;
765
+ }
766
+ }
767
+
768
+ qsort(bounds, count*2, sizeof(cpFloat), (int (*)(const void *, const void *))cpfcompare);
769
+ cpFloat split = (bounds[count - 1] + bounds[count])*0.5f; // use the medain as the split
770
+ cpfree(bounds);
771
+
772
+ // Generate the child BBs
773
+ cpBB a = bb, b = bb;
774
+ if(splitWidth) a.r = b.l = split; else a.t = b.b = split;
775
+
776
+ // Partition the nodes
777
+ int right = count;
778
+ for(int left=0; left < right;){
779
+ Node *node = nodes[left];
780
+ if(cpBBMergedArea(node->bb, b) < cpBBMergedArea(node->bb, a)){
781
+ // if(cpBBProximity(node->bb, b) < cpBBProximity(node->bb, a)){
782
+ right--;
783
+ nodes[left] = nodes[right];
784
+ nodes[right] = node;
785
+ } else {
786
+ left++;
787
+ }
788
+ }
789
+
790
+ if(right == count){
791
+ Node *node = NULL;
792
+ for(int i=0; i<count; i++) node = SubtreeInsert(node, nodes[i], tree);
793
+ return node;
794
+ }
795
+
796
+ // Recurse and build the node!
797
+ return NodeNew(tree,
798
+ partitionNodes(tree, nodes, right),
799
+ partitionNodes(tree, nodes + right, count - right)
800
+ );
801
+ }
802
+
803
+ //static void
804
+ //cpBBTreeOptimizeIncremental(cpBBTree *tree, int passes)
805
+ //{
806
+ // for(int i=0; i<passes; i++){
807
+ // Node *root = tree->root;
808
+ // Node *node = root;
809
+ // int bit = 0;
810
+ // unsigned int path = tree->opath;
811
+ //
812
+ // while(!NodeIsLeaf(node)){
813
+ // node = (path&(1<<bit) ? node->a : node->b);
814
+ // bit = (bit + 1)&(sizeof(unsigned int)*8 - 1);
815
+ // }
816
+ //
817
+ // root = subtreeRemove(root, node, tree);
818
+ // tree->root = subtreeInsert(root, node, tree);
819
+ // }
820
+ //}
821
+
822
+ void
823
+ cpBBTreeOptimize(cpSpatialIndex *index)
824
+ {
825
+ if(index->klass != &klass){
826
+ cpAssertWarn(cpFalse, "Ignoring cpBBTreeOptimize() call to non-tree spatial index.");
827
+ return;
828
+ }
829
+
830
+ cpBBTree *tree = (cpBBTree *)index;
831
+ Node *root = tree->root;
832
+ if(!root) return;
833
+
834
+ int count = cpBBTreeCount(tree);
835
+ Node **nodes = (Node **)cpcalloc(count, sizeof(Node *));
836
+ Node **cursor = nodes;
837
+
838
+ cpHashSetEach(tree->leaves, (cpHashSetIteratorFunc)fillNodeArray, &cursor);
839
+
840
+ SubtreeRecycle(tree, root);
841
+ tree->root = partitionNodes(tree, nodes, count);
842
+ cpfree(nodes);
843
+ }
844
+
845
+ //MARK: Debug Draw
846
+
847
+ //#define CP_BBTREE_DEBUG_DRAW
848
+ #ifdef CP_BBTREE_DEBUG_DRAW
849
+ #include "OpenGL/gl.h"
850
+ #include "OpenGL/glu.h"
851
+ #include <GLUT/glut.h>
852
+
853
+ static void
854
+ NodeRender(Node *node, int depth)
855
+ {
856
+ if(!NodeIsLeaf(node) && depth <= 10){
857
+ NodeRender(node->a, depth + 1);
858
+ NodeRender(node->b, depth + 1);
859
+ }
860
+
861
+ cpBB bb = node->bb;
862
+
863
+ // GLfloat v = depth/2.0f;
864
+ // glColor3f(1.0f - v, v, 0.0f);
865
+ glLineWidth(cpfmax(5.0f - depth, 1.0f));
866
+ glBegin(GL_LINES); {
867
+ glVertex2f(bb.l, bb.b);
868
+ glVertex2f(bb.l, bb.t);
869
+
870
+ glVertex2f(bb.l, bb.t);
871
+ glVertex2f(bb.r, bb.t);
872
+
873
+ glVertex2f(bb.r, bb.t);
874
+ glVertex2f(bb.r, bb.b);
875
+
876
+ glVertex2f(bb.r, bb.b);
877
+ glVertex2f(bb.l, bb.b);
878
+ }; glEnd();
879
+ }
880
+
881
+ void
882
+ cpBBTreeRenderDebug(cpSpatialIndex *index){
883
+ if(index->klass != &klass){
884
+ cpAssertWarn(cpFalse, "Ignoring cpBBTreeRenderDebug() call to non-tree spatial index.");
885
+ return;
886
+ }
887
+
888
+ cpBBTree *tree = (cpBBTree *)index;
889
+ if(tree->root) NodeRender(tree->root, 0);
890
+ }
891
+ #endif