annotate trunk/chipmunkd/cpSpaceHash.d @ 17:131331ebb599

cpVect toString method and some cleanup
author Extrawurst
date Thu, 09 Dec 2010 01:51:30 +0100
parents df4ebc8add66
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
1
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
2 // written in the D programming language
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
3
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
4 module chipmunkd.cpSpaceHash;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
5
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
6 import chipmunkd.chipmunk;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
7 import chipmunkd.prime;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
8
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
9 // The spatial hash is Chipmunk's default (and currently only) spatial index type.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
10 // Based on a chained hash table.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
11
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
12 // Used internally to track objects added to the hash
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
13 struct cpHandle{
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
14 // Pointer to the object
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
15 void *obj;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
16 // Retain count
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
17 int retain;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
18 // Query stamp. Used to make sure two objects
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
19 // aren't identified twice in the same query.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
20 cpTimestamp stamp;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
21 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
22
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
23 // Linked list element for in the chains.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
24 struct cpSpaceHashBin{
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
25 cpHandle *handle;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
26 cpSpaceHashBin *next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
27 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
28
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
29 // BBox callback. Called whenever the hash needs a bounding box from an object.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
30 alias cpBB function(void *obj)cpSpaceHashBBFunc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
31
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
32 struct cpSpaceHash{
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
33 // Number of cells in the table.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
34 int numcells;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
35 // Dimentions of the cells.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
36 cpFloat celldim;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
37
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
38 // BBox callback.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
39 cpSpaceHashBBFunc bbfunc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
40
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
41 // Hashset of the handles and the recycled ones.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
42 cpHashSet *handleSet;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
43 cpArray *pooledHandles;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
44
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
45 // The table and the recycled bins.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
46 cpSpaceHashBin **table;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
47 cpSpaceHashBin *pooledBins;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
48
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
49 // list of buffers to free on destruction.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
50 cpArray *allocatedBuffers;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
51
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
52 // Incremented on each query. See cpHandle.stamp.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
53 cpTimestamp stamp;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
54 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
55
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
56 ////Basic allocation/destruction functions.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
57 //cpSpaceHash *cpSpaceHashAlloc(void);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
58 //cpSpaceHash *cpSpaceHashInit(cpSpaceHash *hash, cpFloat celldim, int cells, cpSpaceHashBBFunc bbfunc);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
59 //cpSpaceHash *cpSpaceHashNew(cpFloat celldim, int cells, cpSpaceHashBBFunc bbfunc);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
60 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
61 //void cpSpaceHashDestroy(cpSpaceHash *hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
62 //void cpSpaceHashFree(cpSpaceHash *hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
63 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
64 //// Resize the hashtable. (Does not rehash! You must call cpSpaceHashRehash() if needed.)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
65 //void cpSpaceHashResize(cpSpaceHash *hash, cpFloat celldim, int numcells);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
66 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
67 //// Add an object to the hash.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
68 //void cpSpaceHashInsert(cpSpaceHash *hash, void *obj, cpHashValue id, cpBB _deprecated_ignored);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
69 //// Remove an object from the hash.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
70 //void cpSpaceHashRemove(cpSpaceHash *hash, void *obj, cpHashValue id);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
71 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
72 //// Iterator function
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
73 alias void function(void *obj, void *data)cpSpaceHashIterator;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
74 //// Iterate over the objects in the hash.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
75 //void cpSpaceHashEach(cpSpaceHash *hash, cpSpaceHashIterator func, void *data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
76 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
77 //// Rehash the contents of the hash.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
78 //void cpSpaceHashRehash(cpSpaceHash *hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
79 //// Rehash only a specific object.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
80 //void cpSpaceHashRehashObject(cpSpaceHash *hash, void *obj, cpHashValue id);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
81 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
82 //// Query callback.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
83 alias void function(void *obj1, void *obj2, void *data)cpSpaceHashQueryFunc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
84 //// Point query the hash. A reference to the query point is passed as obj1 to the query callback.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
85 //void cpSpaceHashPointQuery(cpSpaceHash *hash, cpVect point, cpSpaceHashQueryFunc func, void *data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
86 //// Query the hash for a given BBox.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
87 //void cpSpaceHashQuery(cpSpaceHash *hash, void *obj, cpBB bb, cpSpaceHashQueryFunc func, void *data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
88 //// Run a query for the object, then insert it. (Optimized case)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
89 //void cpSpaceHashQueryInsert(cpSpaceHash *hash, void *obj, cpBB bb, cpSpaceHashQueryFunc func, void *data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
90 //// Rehashes while querying for each object. (Optimized case)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
91 //void cpSpaceHashQueryRehash(cpSpaceHash *hash, cpSpaceHashQueryFunc func, void *data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
92 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
93 //// Segment Query callback.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
94 //// Return value is uesd for early exits of the query.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
95 //// If while traversing the grid, the raytrace function detects that an entire grid cell is beyond the hit point, it will stop the trace.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
96 alias cpFloat function(void *obj1, void *obj2, void *data)cpSpaceHashSegmentQueryFunc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
97 //void cpSpaceHashSegmentQuery(cpSpaceHash *hash, void *obj, cpVect a, cpVect b, cpFloat t_exit, cpSpaceHashSegmentQueryFunc func, void *data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
98 //
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
99
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
100 static cpHandle*
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
101 cpHandleInit(cpHandle *hand, void *obj)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
102 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
103 hand.obj = obj;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
104 hand.retain = 0;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
105 hand.stamp = 0;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
106
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
107 return hand;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
108 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
109
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
110 static void cpHandleRetain(cpHandle *hand){hand.retain++;}
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
111
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
112 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
113 cpHandleRelease(cpHandle *hand, cpArray *pooledHandles)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
114 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
115 hand.retain--;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
116 if(hand.retain == 0) cpArrayPush(pooledHandles, hand);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
117 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
118
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
119 cpSpaceHash*
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
120 cpSpaceHashAlloc()
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
121 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
122 return cast(cpSpaceHash *)cpcalloc(1, cpSpaceHash.sizeof);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
123 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
124
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
125 // Frees the old table, and allocate a new one.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
126 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
127 cpSpaceHashAllocTable(cpSpaceHash *hash, int numcells)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
128 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
129 cpfree(hash.table);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
130
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
131 hash.numcells = numcells;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
132 hash.table = cast(cpSpaceHashBin **)cpcalloc(numcells, (cpSpaceHashBin *).sizeof);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
133 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
134
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
135 // Equality function for the handleset.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
136 static int handleSetEql(void *obj, cpHandle* hand){return (obj == hand.obj);}
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
137
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
138 // Transformation function for the handleset.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
139 static void *
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
140 handleSetTrans(void *obj, cpSpaceHash *hash)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
141 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
142 if(hash.pooledHandles.num == 0){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
143 // handle pool is exhausted, make more
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
144 int count = CP_BUFFER_BYTES/cpHandle.sizeof;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
145 assert(count, "Buffer size is too small.");
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
146
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
147 cpHandle *buffer = cast(cpHandle *)cpmalloc(CP_BUFFER_BYTES);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
148 cpArrayPush(hash.allocatedBuffers, buffer);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
149
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
150 for(int i=0; i<count; i++) cpArrayPush(hash.pooledHandles, buffer + i);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
151 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
152
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
153 cpHandle *hand = cpHandleInit(cast(cpHandle *) cpArrayPop(hash.pooledHandles), obj);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
154 cpHandleRetain(hand);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
155
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
156 return hand;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
157 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
158
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
159 cpSpaceHash*
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
160 cpSpaceHashInit(cpSpaceHash *hash, cpFloat celldim, int numcells, cpSpaceHashBBFunc bbfunc)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
161 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
162 cpSpaceHashAllocTable(hash, next_prime(numcells));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
163 hash.celldim = celldim;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
164 hash.bbfunc = bbfunc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
165
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
166 hash.handleSet = cpHashSetNew(0, cast(cpHashSetEqlFunc)&handleSetEql, cast(cpHashSetTransFunc)&handleSetTrans);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
167 hash.pooledHandles = cpArrayNew(0);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
168
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
169 hash.pooledBins = null;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
170 hash.allocatedBuffers = cpArrayNew(0);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
171
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
172 hash.stamp = 1;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
173
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
174 return hash;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
175 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
176
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
177 cpSpaceHash*
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
178 cpSpaceHashNew(cpFloat celldim, int cells, cpSpaceHashBBFunc bbfunc)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
179 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
180 return cpSpaceHashInit(cpSpaceHashAlloc(), celldim, cells, bbfunc);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
181 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
182
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
183 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
184 recycleBin(cpSpaceHash *hash, cpSpaceHashBin *bin)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
185 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
186 bin.next = hash.pooledBins;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
187 hash.pooledBins = bin;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
188 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
189
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
190 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
191 clearHashCell(cpSpaceHash *hash, int idx)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
192 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
193 cpSpaceHashBin *bin = hash.table[idx];
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
194 while(bin){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
195 cpSpaceHashBin *next = bin.next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
196
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
197 cpHandleRelease(bin.handle, hash.pooledHandles);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
198 recycleBin(hash, bin);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
199
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
200 bin = next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
201 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
202
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
203 hash.table[idx] = null;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
204 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
205
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
206 // Clear all cells in the hashtable.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
207 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
208 clearHash(cpSpaceHash *hash)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
209 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
210 for(int i=0; i<hash.numcells; i++)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
211 clearHashCell(hash, i);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
212 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
213
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
214 static void freeWrap(void *ptr, void *unused){cpfree(ptr);}
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
215
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
216 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
217 cpSpaceHashDestroy(cpSpaceHash *hash)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
218 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
219 clearHash(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
220
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
221 cpHashSetFree(hash.handleSet);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
222
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
223 cpArrayEach(hash.allocatedBuffers, &freeWrap, null);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
224 cpArrayFree(hash.allocatedBuffers);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
225 cpArrayFree(hash.pooledHandles);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
226
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
227 cpfree(hash.table);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
228 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
229
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
230 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
231 cpSpaceHashFree(cpSpaceHash *hash)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
232 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
233 if(hash){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
234 cpSpaceHashDestroy(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
235 cpfree(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
236 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
237 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
238
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
239 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
240 cpSpaceHashResize(cpSpaceHash *hash, cpFloat celldim, int numcells)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
241 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
242 // Clear the hash to release the old handle locks.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
243 clearHash(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
244
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
245 hash.celldim = celldim;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
246 cpSpaceHashAllocTable(hash, next_prime(numcells));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
247 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
248
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
249 // Return true if the chain contains the handle.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
250 static cpBool
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
251 containsHandle(cpSpaceHashBin *bin, cpHandle *hand)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
252 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
253 while(bin){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
254 if(bin.handle == hand) return cpTrue;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
255 bin = bin.next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
256 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
257
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
258 return cpFalse;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
259 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
260
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
261 // Get a recycled or new bin.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
262 static cpSpaceHashBin *
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
263 getEmptyBin(cpSpaceHash *hash)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
264 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
265 cpSpaceHashBin *bin = hash.pooledBins;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
266
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
267 if(bin){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
268 hash.pooledBins = bin.next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
269 return bin;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
270 } else {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
271 // Pool is exhausted, make more
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
272 int count = CP_BUFFER_BYTES/cpSpaceHashBin.sizeof;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
273 assert(count, "Buffer size is too small.");
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
274
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
275 cpSpaceHashBin *buffer = cast(cpSpaceHashBin *)cpmalloc(CP_BUFFER_BYTES);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
276 cpArrayPush(hash.allocatedBuffers, buffer);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
277
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
278 // push all but the first one, return the first instead
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
279 for(int i=1; i<count; i++) recycleBin(hash, buffer + i);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
280 return buffer;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
281 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
282 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
283
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
284 // The hash function itself.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
285 static cpHashValue
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
286 hash_func(cpHashValue x, cpHashValue y, cpHashValue n)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
287 {
17
131331ebb599 cpVect toString method and some cleanup
Extrawurst
parents: 15
diff changeset
288 return cast(cpHashValue)(( (x*1640531513uL) ^ (y*2654435789uL) ) % n);
4
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
289 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
290
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
291 // Much faster than (int)floor(f)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
292 // Profiling showed floor() to be a sizable performance hog
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
293 static int
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
294 floor_int(cpFloat f)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
295 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
296 int i = cast(int)f;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
297 return (f < 0.0f && f != i ? i - 1 : i);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
298 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
299
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
300 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
301 hashHandle(cpSpaceHash *hash, cpHandle *hand, cpBB bb)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
302 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
303 // Find the dimensions in cell coordinates.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
304 cpFloat dim = hash.celldim;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
305 int l = floor_int(bb.l/dim); // Fix by ShiftZ
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
306 int r = floor_int(bb.r/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
307 int b = floor_int(bb.b/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
308 int t = floor_int(bb.t/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
309
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
310 int n = hash.numcells;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
311 for(int i=l; i<=r; i++){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
312 for(int j=b; j<=t; j++){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
313 int idx = hash_func(i,j,n);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
314 cpSpaceHashBin *bin = hash.table[idx];
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
315
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
316 // Don't add an object twice to the same cell.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
317 if(containsHandle(bin, hand)) continue;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
318
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
319 cpHandleRetain(hand);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
320 // Insert a new bin for the handle in this cell.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
321 cpSpaceHashBin *newBin = getEmptyBin(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
322 newBin.handle = hand;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
323 newBin.next = bin;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
324 hash.table[idx] = newBin;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
325 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
326 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
327 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
328
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
329 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
330 cpSpaceHashInsert(cpSpaceHash *hash, void *obj, cpHashValue hashid, cpBB _deprecated_unused)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
331 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
332 cpHandle *hand = cast(cpHandle *)cpHashSetInsert(hash.handleSet, hashid, obj, hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
333 hashHandle(hash, hand, hash.bbfunc(obj));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
334 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
335
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
336 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
337 cpSpaceHashRehashObject(cpSpaceHash *hash, void *obj, cpHashValue hashid)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
338 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
339 cpHandle *hand = cast(cpHandle *)cpHashSetRemove(hash.handleSet, hashid, obj);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
340
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
341 if(hand){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
342 hand.obj = null;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
343 cpHandleRelease(hand, hash.pooledHandles);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
344
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
345 cpSpaceHashInsert(hash, obj, hashid, cpBBNew(0.0f, 0.0f, 0.0f, 0.0f));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
346 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
347 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
348
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
349 static void handleRehashHelper(cpHandle *hand, cpSpaceHash *hash){hashHandle(hash, hand, hash.bbfunc(hand.obj));}
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
350
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
351 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
352 cpSpaceHashRehash(cpSpaceHash *hash)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
353 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
354 clearHash(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
355 cpHashSetEach(hash.handleSet, cast(cpHashSetIterFunc)&handleRehashHelper, hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
356 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
357
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
358 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
359 cpSpaceHashRemove(cpSpaceHash *hash, void *obj, cpHashValue hashid)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
360 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
361 cpHandle *hand = cast(cpHandle *)cpHashSetRemove(hash.handleSet, hashid, obj);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
362
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
363 if(hand){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
364 hand.obj = null;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
365 cpHandleRelease(hand, hash.pooledHandles);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
366 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
367 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
368
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
369 struct eachPair {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
370 cpSpaceHashIterator func;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
371 void *data;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
372 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
373
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
374 static void eachHelper(cpHandle *hand, eachPair *pair){pair.func(hand.obj, pair.data);}
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
375
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
376 // Iterate over the objects in the spatial hash.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
377 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
378 cpSpaceHashEach(cpSpaceHash *hash, cpSpaceHashIterator func, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
379 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
380 eachPair pair = {func, data};
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
381 cpHashSetEach(hash.handleSet, cast(cpHashSetIterFunc)&eachHelper, &pair);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
382 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
383
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
384 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
385 removeOrphanedHandles(cpSpaceHash *hash, cpSpaceHashBin **bin_ptr)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
386 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
387 cpSpaceHashBin *bin = *bin_ptr;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
388 while(bin){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
389 cpHandle *hand = bin.handle;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
390 cpSpaceHashBin *next = bin.next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
391
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
392 if(!hand.obj){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
393 // orphaned handle, unlink and recycle the bin
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
394 (*bin_ptr) = bin.next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
395 recycleBin(hash, bin);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
396
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
397 cpHandleRelease(hand, hash.pooledHandles);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
398 } else {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
399 bin_ptr = &bin.next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
400 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
401
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
402 bin = next;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
403 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
404 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
405
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
406 // Calls the callback function for the objects in a given chain.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
407 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
408 query(cpSpaceHash *hash, cpSpaceHashBin **bin_ptr, void *obj, cpSpaceHashQueryFunc func, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
409 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
410 restart:
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
411 for(cpSpaceHashBin *bin = *bin_ptr; bin; bin = bin.next){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
412 cpHandle *hand = bin.handle;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
413 void *other = hand.obj;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
414
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
415 if(hand.stamp == hash.stamp || obj == other){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
416 continue;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
417 } else if(other){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
418 func(obj, other, data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
419 hand.stamp = hash.stamp;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
420 } else {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
421 // The object for this handle has been removed
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
422 // cleanup this cell and restart the query
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
423 removeOrphanedHandles(hash, bin_ptr);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
424 goto restart; // GCC not smart enough/able to tail call an inlined function.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
425 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
426 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
427 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
428
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
429 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
430 cpSpaceHashPointQuery(cpSpaceHash *hash, cpVect point, cpSpaceHashQueryFunc func, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
431 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
432 cpFloat dim = hash.celldim;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
433 int idx = hash_func(floor_int(point.x/dim), floor_int(point.y/dim), hash.numcells); // Fix by ShiftZ
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
434
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
435 query(hash, &hash.table[idx], &point, func, data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
436 hash.stamp++;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
437 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
438
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
439 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
440 cpSpaceHashQuery(cpSpaceHash *hash, void *obj, cpBB bb, cpSpaceHashQueryFunc func, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
441 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
442 // Get the dimensions in cell coordinates.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
443 cpFloat dim = hash.celldim;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
444 int l = floor_int(bb.l/dim); // Fix by ShiftZ
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
445 int r = floor_int(bb.r/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
446 int b = floor_int(bb.b/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
447 int t = floor_int(bb.t/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
448
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
449 int n = hash.numcells;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
450 cpSpaceHashBin **table = hash.table;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
451
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
452 // Iterate over the cells and query them.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
453 for(int i=l; i<=r; i++){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
454 for(int j=b; j<=t; j++){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
455 query(hash, &table[hash_func(i,j,n)], obj, func, data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
456 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
457 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
458
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
459 hash.stamp++;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
460 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
461
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
462 // Similar to struct eachPair above.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
463 struct queryRehashPair {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
464 cpSpaceHash *hash;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
465 cpSpaceHashQueryFunc func;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
466 void *data;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
467 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
468
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
469 // Hashset iterator func used with cpSpaceHashQueryRehash().
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
470 static void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
471 handleQueryRehashHelper(void *elt, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
472 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
473 cpHandle *hand = cast(cpHandle *)elt;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
474
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
475 // Unpack the user callback data.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
476 queryRehashPair *pair = cast(queryRehashPair *)data;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
477 cpSpaceHash *hash = pair.hash;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
478 cpSpaceHashQueryFunc func = pair.func;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
479
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
480 cpFloat dim = hash.celldim;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
481 int n = hash.numcells;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
482
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
483 void *obj = hand.obj;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
484 cpBB bb = hash.bbfunc(obj);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
485
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
486 int l = floor_int(bb.l/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
487 int r = floor_int(bb.r/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
488 int b = floor_int(bb.b/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
489 int t = floor_int(bb.t/dim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
490
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
491 cpSpaceHashBin **table = hash.table;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
492
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
493 for(int i=l; i<=r; i++){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
494 for(int j=b; j<=t; j++){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
495 int idx = hash_func(i,j,n);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
496 cpSpaceHashBin *bin = table[idx];
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
497
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
498 if(containsHandle(bin, hand)) continue;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
499
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
500 cpHandleRetain(hand); // this MUST be done first in case the object is removed in func()
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
501 query(hash, &bin, obj, func, pair.data);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
502
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
503 cpSpaceHashBin *newBin = getEmptyBin(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
504 newBin.handle = hand;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
505 newBin.next = bin;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
506 table[idx] = newBin;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
507 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
508 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
509
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
510 // Increment the stamp for each object hashed.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
511 hash.stamp++;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
512 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
513
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
514 void
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
515 cpSpaceHashQueryRehash(cpSpaceHash *hash, cpSpaceHashQueryFunc func, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
516 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
517 clearHash(hash);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
518
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
519 queryRehashPair pair = {hash, func, data};
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
520 cpHashSetEach(hash.handleSet, &handleQueryRehashHelper, &pair);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
521 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
522
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
523 static cpFloat
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
524 segmentQuery(cpSpaceHash *hash, cpSpaceHashBin **bin_ptr, void *obj, cpSpaceHashSegmentQueryFunc func, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
525 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
526 cpFloat t = 1.0f;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
527
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
528 restart:
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
529 for(cpSpaceHashBin *bin = *bin_ptr; bin; bin = bin.next){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
530 cpHandle *hand = bin.handle;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
531 void *other = hand.obj;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
532
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
533 // Skip over certain conditions
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
534 if(hand.stamp == hash.stamp){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
535 continue;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
536 } else if(other){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
537 t = cpfmin(t, func(obj, other, data));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
538 hand.stamp = hash.stamp;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
539 } else {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
540 // The object for this handle has been removed
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
541 // cleanup this cell and restart the query
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
542 removeOrphanedHandles(hash, bin_ptr);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
543 goto restart; // GCC not smart enough/able to tail call an inlined function.
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
544 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
545 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
546
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
547 return t;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
548 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
549
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
550 // modified from http://playtechs.blogspot.com/2007/03/raytracing-on-grid.html
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
551 void cpSpaceHashSegmentQuery(cpSpaceHash *hash, void *obj, cpVect a, cpVect b, cpFloat t_exit, cpSpaceHashSegmentQueryFunc func, void *data)
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
552 {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
553 a = cpvmult(a, 1.0f/hash.celldim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
554 b = cpvmult(b, 1.0f/hash.celldim);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
555
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
556 int cell_x = floor_int(a.x), cell_y = floor_int(a.y);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
557
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
558 cpFloat t = 0;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
559
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
560 int x_inc, y_inc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
561 cpFloat temp_v, temp_h;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
562
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
563 if (b.x > a.x){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
564 x_inc = 1;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
565 temp_h = (cpffloor(a.x + 1.0f) - a.x);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
566 } else {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
567 x_inc = -1;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
568 temp_h = (a.x - cpffloor(a.x));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
569 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
570
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
571 if (b.y > a.y){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
572 y_inc = 1;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
573 temp_v = (cpffloor(a.y + 1.0f) - a.y);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
574 } else {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
575 y_inc = -1;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
576 temp_v = (a.y - cpffloor(a.y));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
577 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
578
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
579 // Division by zero is *very* slow on ARM
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
580 cpFloat dx = cpfabs(b.x - a.x), dy = cpfabs(b.y - a.y);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
581 cpFloat dt_dx = (dx ? 1.0f/dx : INFINITY), dt_dy = (dy ? 1.0f/dy : INFINITY);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
582
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
583 // fix NANs in horizontal directions
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
584 cpFloat next_h = (temp_h ? temp_h*dt_dx : dt_dx);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
585 cpFloat next_v = (temp_v ? temp_v*dt_dy : dt_dy);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
586
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
587 cpSpaceHashBin **table = hash.table;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
588
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
589 int n = hash.numcells;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
590 while(t < t_exit){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
591 int idx = hash_func(cell_x, cell_y, n);
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
592 t_exit = cpfmin(t_exit, segmentQuery(hash, &table[idx], obj, func, data));
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
593
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
594 if (next_v < next_h){
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
595 cell_y += y_inc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
596 t = next_v;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
597 next_v += dt_dy;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
598 } else {
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
599 cell_x += x_inc;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
600 t = next_h;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
601 next_h += dt_dx;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
602 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
603 }
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
604
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
605 hash.stamp++;
7ebbd4d05553 initial commit
Extrawurst
parents:
diff changeset
606 }