1 /*
2 * Copyright (C) 2008 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 /* ---- includes ----------------------------------------------------------- */
18
19 #include "b_TensorEm/Cluster3D.h"
20 #include "b_BasicEm/Math.h"
21 #include "b_BasicEm/Memory.h"
22 #include "b_BasicEm/Functions.h"
23
24 /* ------------------------------------------------------------------------- */
25
26 /* ========================================================================= */
27 /* */
28 /* ---- \ghd{ auxiliary functions } ---------------------------------------- */
29 /* */
30 /* ========================================================================= */
31
32 /* ------------------------------------------------------------------------- */
33
34 /* ========================================================================= */
35 /* */
36 /* ---- \ghd{ constructor / destructor } ----------------------------------- */
37 /* */
38 /* ========================================================================= */
39
40 /* ------------------------------------------------------------------------- */
41
bts_Cluster3D_init(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA)42 void bts_Cluster3D_init( struct bbs_Context* cpA,
43 struct bts_Cluster3D* ptrA )
44 {
45 ptrA->mspE = NULL;
46 ptrA->vecArrE = NULL;
47 ptrA->allocatedSizeE = 0;
48 ptrA->sizeE = 0;
49 ptrA->bbpE = 0;
50 }
51
52 /* ------------------------------------------------------------------------- */
53
bts_Cluster3D_exit(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA)54 void bts_Cluster3D_exit( struct bbs_Context* cpA,
55 struct bts_Cluster3D* ptrA )
56 {
57 bbs_MemSeg_free( cpA, ptrA->mspE, ptrA->vecArrE );
58 ptrA->vecArrE = NULL;
59 ptrA->mspE = NULL;
60 ptrA->allocatedSizeE = 0;
61 ptrA->sizeE = 0;
62 ptrA->bbpE = 0;
63 }
64
65 /* ------------------------------------------------------------------------- */
66
67 /* ========================================================================= */
68 /* */
69 /* ---- \ghd{ operators } -------------------------------------------------- */
70 /* */
71 /* ========================================================================= */
72
73 /* ------------------------------------------------------------------------- */
74
bts_Cluster3D_copy(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA,const struct bts_Cluster3D * srcPtrA)75 void bts_Cluster3D_copy( struct bbs_Context* cpA,
76 struct bts_Cluster3D* ptrA,
77 const struct bts_Cluster3D* srcPtrA )
78 {
79 #ifdef DEBUG1
80 if( ptrA->allocatedSizeE < srcPtrA->sizeE )
81 {
82 bbs_ERROR0( "void bts_Cluster3D_copy( struct bts_Cluster2D* ptrA, const struct bts_Cluster2D* srcPtrA ): allocated size too low in destination cluster" );
83 return;
84 }
85 #endif
86
87 bbs_memcpy16( ptrA->vecArrE, srcPtrA->vecArrE, bbs_SIZEOF16( struct bts_Int16Vec3D ) * srcPtrA->sizeE );
88
89 ptrA->bbpE = srcPtrA->bbpE;
90 ptrA->sizeE = srcPtrA->sizeE;
91 }
92
93 /* ------------------------------------------------------------------------- */
94
bts_Cluster3D_equal(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA,const struct bts_Cluster3D * srcPtrA)95 flag bts_Cluster3D_equal( struct bbs_Context* cpA,
96 const struct bts_Cluster3D* ptrA,
97 const struct bts_Cluster3D* srcPtrA )
98 {
99 uint32 iL;
100 const struct bts_Int16Vec3D* src1L = ptrA->vecArrE;
101 const struct bts_Int16Vec3D* src2L = srcPtrA->vecArrE;
102
103 if( ptrA->sizeE != srcPtrA->sizeE ) return FALSE;
104 if( ptrA->bbpE != srcPtrA->bbpE ) return FALSE;
105
106 for( iL = ptrA->sizeE; iL > 0; iL-- )
107 {
108 if( ( src1L->xE != src2L->xE ) ||
109 ( src1L->yE != src2L->yE ) ||
110 ( src1L->zE != src2L->zE ) ) return FALSE;
111 src1L++;
112 src2L++;
113 }
114
115 return TRUE;
116 }
117
118 /* ------------------------------------------------------------------------- */
119
120 /* ========================================================================= */
121 /* */
122 /* ---- \ghd{ query functions } -------------------------------------------- */
123 /* */
124 /* ========================================================================= */
125
126 /* ------------------------------------------------------------------------- */
127
bts_Cluster3D_center(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA)128 struct bts_Flt16Vec3D bts_Cluster3D_center( struct bbs_Context* cpA,
129 const struct bts_Cluster3D* ptrA )
130 {
131 struct bts_Int16Vec3D* vecPtrL = ptrA->vecArrE;
132 uint32 iL;
133 int32 xL = 0;
134 int32 yL = 0;
135 int32 zL = 0;
136
137 if( ptrA->sizeE == 0 ) return bts_Flt16Vec3D_create16( 0, 0, 0, 0 );
138
139 for( iL = ptrA->sizeE; iL > 0; iL-- )
140 {
141 xL += vecPtrL->xE;
142 yL += vecPtrL->yE;
143 zL += vecPtrL->zE;
144 vecPtrL++;
145 }
146
147 xL = ( ( ( xL << 1 ) / ( int32 )ptrA->sizeE ) + 1 ) >> 1;
148 yL = ( ( ( yL << 1 ) / ( int32 )ptrA->sizeE ) + 1 ) >> 1;
149 zL = ( ( ( zL << 1 ) / ( int32 )ptrA->sizeE ) + 1 ) >> 1;
150
151 return bts_Flt16Vec3D_create16( ( int16 )xL, ( int16 )yL, ( int16 )zL, ( int16 )ptrA->bbpE );
152 }
153
154 /* ------------------------------------------------------------------------- */
155
bts_Cluster3D_boundingBox(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA)156 struct bts_Int16Rect bts_Cluster3D_boundingBox( struct bbs_Context* cpA,
157 const struct bts_Cluster3D* ptrA )
158 {
159 struct bts_Int16Vec3D* vecPtrL = ptrA->vecArrE;
160 uint32 iL;
161 int32 xMinL = 65536; /*( 1 << 16 )*/
162 int32 yMinL = 65536; /*( 1 << 16 )*/
163 int32 xMaxL = 0;
164 int32 yMaxL = 0;
165
166 if( ptrA->sizeE == 0 ) return bts_Int16Rect_create( 0, 0, 0, 0 );
167
168 for( iL = ptrA->sizeE; iL > 0; iL-- )
169 {
170 xMinL = bbs_min( xMinL, vecPtrL->xE );
171 yMinL = bbs_min( yMinL, vecPtrL->yE );
172 xMaxL = bbs_max( xMaxL, vecPtrL->xE );
173 yMaxL = bbs_max( yMaxL, vecPtrL->yE );
174 vecPtrL++;
175 }
176
177 return bts_Int16Rect_create( ( int16 )xMinL, ( int16 )yMinL, ( int16 )xMaxL, ( int16 )yMaxL );
178 }
179
180 /* ------------------------------------------------------------------------- */
181
bts_Cluster3D_int32X(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA,uint32 indexA,int32 bbpA)182 int32 bts_Cluster3D_int32X( struct bbs_Context* cpA,
183 const struct bts_Cluster3D* ptrA,
184 uint32 indexA, int32 bbpA )
185 {
186 int32 shiftL = bbpA - ptrA->bbpE;
187 #ifdef DEBUG2
188 if( indexA >= ptrA->sizeE )
189 {
190 bbs_ERROR2( "int32 bts_Cluster2D_int32X( .... )\n"
191 "indexA = %i is out of range [0,%i]",
192 indexA,
193 ptrA->sizeE - 1 );
194 return 0;
195 }
196 #endif
197 if( shiftL >= 0 )
198 {
199 return ( int32 ) ptrA->vecArrE[ indexA ].xE << shiftL;
200 }
201 else
202 {
203 return ( ( ( int32 ) ptrA->vecArrE[ indexA ].xE >> ( -shiftL - 1 ) ) + 1 ) >> 1;
204 }
205 }
206
207 /* ------------------------------------------------------------------------- */
208
bts_Cluster3D_int32Y(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA,uint32 indexA,int32 bbpA)209 int32 bts_Cluster3D_int32Y( struct bbs_Context* cpA,
210 const struct bts_Cluster3D* ptrA,
211 uint32 indexA,
212 int32 bbpA )
213 {
214 int32 shiftL = bbpA - ptrA->bbpE;
215 #ifdef DEBUG2
216 if( indexA >= ptrA->sizeE )
217 {
218 bbs_ERROR2( "int32 bts_Cluster2D_int32Y( .... )\n"
219 "indexA = %i is out of range [0,%i]",
220 indexA,
221 ptrA->sizeE - 1 );
222 return 0;
223 }
224 #endif
225 if( shiftL >= 0 )
226 {
227 return ( int32 ) ptrA->vecArrE[ indexA ].yE << shiftL;
228 }
229 else
230 {
231 return ( ( ( int32 ) ptrA->vecArrE[ indexA ].yE >> ( -shiftL - 1 ) ) + 1 ) >> 1;
232 }
233 }
234
235 /* ------------------------------------------------------------------------- */
236
bts_Cluster3D_int32Z(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA,uint32 indexA,int32 bbpA)237 int32 bts_Cluster3D_int32Z( struct bbs_Context* cpA,
238 const struct bts_Cluster3D* ptrA,
239 uint32 indexA,
240 int32 bbpA )
241 {
242 int32 shiftL = bbpA - ptrA->bbpE;
243 #ifdef DEBUG2
244 if( indexA >= ptrA->sizeE )
245 {
246 bbs_ERROR2( "int32 bts_Cluster2D_int32Z( .... )\n"
247 "indexA = %i is out of range [0,%i]",
248 indexA,
249 ptrA->sizeE - 1 );
250 return 0;
251 }
252 #endif
253 if( shiftL >= 0 )
254 {
255 return ( int32 ) ptrA->vecArrE[ indexA ].zE << shiftL;
256 }
257 else
258 {
259 return ( ( ( int32 ) ptrA->vecArrE[ indexA ].zE >> ( -shiftL - 1 ) ) + 1 ) >> 1;
260 }
261 }
262
263 /* ------------------------------------------------------------------------- */
264
265 /* ========================================================================= */
266 /* */
267 /* ---- \ghd{ modify functions } ------------------------------------------- */
268 /* */
269 /* ========================================================================= */
270
271 /* ------------------------------------------------------------------------- */
272
bts_Cluster3D_create(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA,uint32 sizeA,struct bbs_MemSeg * mspA)273 void bts_Cluster3D_create( struct bbs_Context* cpA,
274 struct bts_Cluster3D* ptrA,
275 uint32 sizeA,
276 struct bbs_MemSeg* mspA )
277 {
278 if( bbs_Context_error( cpA ) ) return;
279 if( ptrA->mspE == NULL )
280 {
281 ptrA->sizeE = 0;
282 ptrA->allocatedSizeE = 0;
283 ptrA->vecArrE = NULL;
284 }
285
286 if( ptrA->sizeE == sizeA ) return;
287
288 if( ptrA->vecArrE != 0 )
289 {
290 bbs_ERROR0( "void bts_Cluster3D_create( const struct bts_Cluster3D*, uint32 ):\n"
291 "object has already been created and cannot be resized." );
292 return;
293 }
294
295 ptrA->vecArrE = bbs_MemSeg_alloc( cpA, mspA, sizeA * bbs_SIZEOF16( struct bts_Int16Vec3D ) );
296 if( bbs_Context_error( cpA ) ) return;
297 ptrA->sizeE = sizeA;
298 ptrA->allocatedSizeE = sizeA;
299 if( !mspA->sharedE ) ptrA->mspE = mspA;
300 }
301
302 /* ------------------------------------------------------------------------- */
303
bts_Cluster3D_size(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA,uint32 sizeA)304 void bts_Cluster3D_size( struct bbs_Context* cpA,
305 struct bts_Cluster3D* ptrA,
306 uint32 sizeA )
307 {
308 if( ptrA->allocatedSizeE < sizeA )
309 {
310 bbs_ERROR2( "void bts_Cluster3D_size( struct bts_Cluster3D* ptrA, uint32 sizeA ):\n"
311 "Allocated size (%i) of cluster is smaller than requested size (%i).",
312 ptrA->allocatedSizeE,
313 sizeA );
314 return;
315 }
316 ptrA->sizeE = sizeA;
317 }
318
319 /* ------------------------------------------------------------------------- */
320
bts_Cluster3D_transform(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA,struct bts_Flt16Alt3D altA)321 void bts_Cluster3D_transform( struct bbs_Context* cpA,
322 struct bts_Cluster3D* ptrA,
323 struct bts_Flt16Alt3D altA )
324 {
325 struct bts_Int16Vec3D* vecPtrL = ptrA->vecArrE;
326 uint32 iL;
327
328 int32 x0L = altA.vecE.xE;
329 int32 y0L = altA.vecE.yE;
330 int32 z0L = altA.vecE.zE;
331
332 int32 shiftL = altA.matE.bbpE + ptrA->bbpE - altA.vecE.bbpE;
333
334 if( shiftL < 0 )
335 {
336 x0L = ( ( x0L >> ( -shiftL - 1 ) ) + 1 ) >> 1;
337 y0L = ( ( y0L >> ( -shiftL - 1 ) ) + 1 ) >> 1;
338 z0L = ( ( z0L >> ( -shiftL - 1 ) ) + 1 ) >> 1;
339 }
340 else
341 {
342 x0L <<= shiftL;
343 y0L <<= shiftL;
344 z0L <<= shiftL;
345 }
346
347 if( altA.matE.bbpE > 0 )
348 {
349 x0L += (int32)1 << ( altA.matE.bbpE - 1 );
350 y0L += (int32)1 << ( altA.matE.bbpE - 1 );
351 z0L += (int32)1 << ( altA.matE.bbpE - 1 );
352 }
353
354 for( iL = ptrA->sizeE; iL > 0; iL-- )
355 {
356 int32 xL = vecPtrL->xE;
357 int32 yL = vecPtrL->yE;
358 int32 zL = vecPtrL->zE;
359 vecPtrL->xE = ( x0L + xL * altA.matE.xxE + yL * altA.matE.xyE + zL * altA.matE.xzE ) >> altA.matE.bbpE;
360 vecPtrL->yE = ( y0L + xL * altA.matE.yxE + yL * altA.matE.yyE + zL * altA.matE.yzE ) >> altA.matE.bbpE;
361 vecPtrL->zE = ( z0L + xL * altA.matE.zxE + yL * altA.matE.zyE + zL * altA.matE.zzE ) >> altA.matE.bbpE;
362 vecPtrL++;
363 }
364 }
365
366 /* ------------------------------------------------------------------------- */
367
bts_Cluster3D_centerFree(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA)368 struct bts_Flt16Vec3D bts_Cluster3D_centerFree( struct bbs_Context* cpA,
369 struct bts_Cluster3D* ptrA )
370 {
371 struct bts_Flt16Vec3D centerL = bts_Cluster3D_center( cpA, ptrA );
372 struct bts_Int16Vec3D* vecPtrL = ptrA->vecArrE;
373 uint32 iL;
374
375 for( iL = ptrA->sizeE; iL > 0; iL-- )
376 {
377 vecPtrL->xE -= centerL.xE;
378 vecPtrL->yE -= centerL.yE;
379 vecPtrL->zE -= centerL.zE;
380 vecPtrL++;
381 }
382
383 return centerL;
384 }
385
386 /* ------------------------------------------------------------------------- */
387
388 /* ========================================================================= */
389 /* */
390 /* ---- \ghd{ I/O } -------------------------------------------------------- */
391 /* */
392 /* ========================================================================= */
393
394 /* ------------------------------------------------------------------------- */
395
bts_Cluster3D_memSize(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA)396 uint32 bts_Cluster3D_memSize( struct bbs_Context* cpA,
397 const struct bts_Cluster3D *ptrA )
398 {
399 return bbs_SIZEOF16( uint32 )
400 + bbs_SIZEOF16( uint32 ) /* version */
401 + bbs_SIZEOF16( ptrA->sizeE )
402 + bbs_SIZEOF16( ptrA->bbpE )
403 + bbs_SIZEOF16( struct bts_Int16Vec3D ) * ptrA->sizeE;
404 }
405
406 /* ------------------------------------------------------------------------- */
407
bts_Cluster3D_memWrite(struct bbs_Context * cpA,const struct bts_Cluster3D * ptrA,uint16 * memPtrA)408 uint32 bts_Cluster3D_memWrite( struct bbs_Context* cpA,
409 const struct bts_Cluster3D* ptrA,
410 uint16* memPtrA )
411 {
412 uint32 memSizeL = bts_Cluster3D_memSize( cpA, ptrA );
413 memPtrA += bbs_memWrite32( &memSizeL, memPtrA );
414 memPtrA += bbs_memWriteUInt32( bts_CLUSTER3D_VERSION, memPtrA );
415 memPtrA += bbs_memWrite32( &ptrA->sizeE, memPtrA );
416 memPtrA += bbs_memWrite32( &ptrA->bbpE, memPtrA );
417 memPtrA += bbs_memWrite16Arr( cpA, ptrA->vecArrE,
418 ptrA->sizeE * bbs_SIZEOF16( struct bts_Int16Vec3D ),
419 memPtrA );
420 return memSizeL;
421 }
422
423 /* ------------------------------------------------------------------------- */
424
bts_Cluster3D_memRead(struct bbs_Context * cpA,struct bts_Cluster3D * ptrA,const uint16 * memPtrA,struct bbs_MemSeg * mspA)425 uint32 bts_Cluster3D_memRead( struct bbs_Context* cpA,
426 struct bts_Cluster3D* ptrA,
427 const uint16* memPtrA,
428 struct bbs_MemSeg* mspA )
429 {
430 uint32 memSizeL;
431 uint32 sizeL;
432 uint32 versionL;
433 if( bbs_Context_error( cpA ) ) return 0;
434 memPtrA += bbs_memRead32( &memSizeL, memPtrA );
435 memPtrA += bbs_memReadVersion32( cpA, &versionL, bts_CLUSTER3D_VERSION, memPtrA );
436 memPtrA += bbs_memRead32( &sizeL, memPtrA );
437 memPtrA += bbs_memRead32( &ptrA->bbpE, memPtrA );
438
439 if( ptrA->allocatedSizeE < sizeL )
440 {
441 bts_Cluster3D_create( cpA, ptrA, sizeL, mspA );
442 }
443 else
444 {
445 bts_Cluster3D_size( cpA, ptrA, sizeL );
446 }
447
448
449 bbs_memcpy16( ptrA->vecArrE, memPtrA, bbs_SIZEOF16( struct bts_Int16Vec3D ) * ptrA->sizeE );
450 memPtrA += bbs_memRead16Arr( cpA, ptrA->vecArrE,
451 ptrA->sizeE * bbs_SIZEOF16( struct bts_Int16Vec3D ),
452 memPtrA );
453
454 if( memSizeL != bts_Cluster3D_memSize( cpA, ptrA ) )
455 {
456 bbs_ERR0( bbs_ERR_CORRUPT_DATA, "uint32 bts_Cluster3D_memRead( const struct bts_Cluster3D* ptrA, const void* memPtrA ):\n"
457 "size mismatch" );
458 return 0;
459 }
460 return memSizeL;
461 }
462
463 /* ------------------------------------------------------------------------- */
464
465 /* ========================================================================= */
466 /* */
467 /* ---- \ghd{ exec functions } --------------------------------------------- */
468 /* */
469 /* ========================================================================= */
470
471 /* ------------------------------------------------------------------------- */
472
473 /* ========================================================================= */
474
475