The Android Open Source Project | 7f81d9b | 2009-03-03 19:30:08 -0800 | [diff] [blame] | 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 "DCR_Internal.h" |
| 20 | |
| 21 | /* ---- related objects --------------------------------------------------- */ |
| 22 | |
| 23 | /* ---- typedefs ----------------------------------------------------------- */ |
| 24 | |
| 25 | /* ---- constants ---------------------------------------------------------- */ |
| 26 | |
| 27 | /* ------------------------------------------------------------------------- */ |
| 28 | |
| 29 | /* ========================================================================= */ |
| 30 | /* */ |
| 31 | /* ---- functions ---------------------------------------------------------- */ |
| 32 | /* */ |
| 33 | /* ========================================================================= */ |
| 34 | |
| 35 | /* ------------------------------------------------------------------------- */ |
| 36 | |
| 37 | void btk_DCR_init( struct bbs_Context* cpA, struct btk_DCR* ptrA ) |
| 38 | { |
| 39 | ptrA->hsdkE = NULL; |
| 40 | ptrA->hidE = btk_HID_DCR; |
| 41 | bpi_DCR_init( cpA, &ptrA->dcrE ); |
| 42 | } |
| 43 | |
| 44 | /* ------------------------------------------------------------------------- */ |
| 45 | |
| 46 | void btk_DCR_exit( struct bbs_Context* cpA, struct btk_DCR* ptrA ) |
| 47 | { |
| 48 | ptrA->hsdkE = NULL; |
| 49 | ptrA->hidE = btk_HID_DCR; |
| 50 | bpi_DCR_exit( cpA, &ptrA->dcrE ); |
| 51 | } |
| 52 | |
| 53 | /* ------------------------------------------------------------------------- */ |
| 54 | |
| 55 | btk_DCRCreateParam btk_DCR_defaultParam() |
| 56 | { |
| 57 | btk_DCRCreateParam paramL; |
| 58 | paramL.reserved = 0; |
| 59 | return paramL; |
| 60 | } |
| 61 | |
| 62 | /* ------------------------------------------------------------------------- */ |
| 63 | |
| 64 | btk_Status btk_DCR_create( btk_HSDK hsdkA, |
| 65 | const btk_DCRCreateParam* pCreateParamA, |
| 66 | btk_HDCR* hpdcrA ) |
| 67 | { |
| 68 | btk_HDCR hdcrL = NULL; |
| 69 | |
| 70 | if( hpdcrA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 71 | if( *hpdcrA != NULL ) return btk_STATUS_INVALID_HANDLE; |
| 72 | if( hsdkA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 73 | if( hsdkA->hidE != btk_HID_SDK ) return btk_STATUS_INVALID_HANDLE; |
| 74 | if( pCreateParamA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 75 | if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_PREEXISTING_ERROR; |
| 76 | |
| 77 | hdcrL = ( btk_HDCR )bbs_MemSeg_alloc( &hsdkA->contextE, hsdkA->contextE.memTblE.espArrE[ 0 ], bbs_SIZEOF16( struct btk_DCR ) ); |
| 78 | if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_ERROR; |
| 79 | |
| 80 | btk_DCR_init( &hsdkA->contextE, hdcrL ); |
| 81 | hdcrL->hsdkE = hsdkA; |
| 82 | |
| 83 | if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_ERROR; |
| 84 | |
| 85 | bpi_DCR_create( &hsdkA->contextE, |
| 86 | &hdcrL->dcrE, |
| 87 | hsdkA->maxImageWidthE, |
| 88 | hsdkA->maxImageHeightE, |
| 89 | #ifdef btk_FRSDK |
| 90 | 6000 >> 1, |
| 91 | #else |
| 92 | 0, |
| 93 | #endif |
| 94 | &hsdkA->contextE.memTblE ); |
| 95 | |
| 96 | if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_ERROR; |
| 97 | |
| 98 | *hpdcrA = hdcrL; |
| 99 | hsdkA->refCtrE++; |
| 100 | |
| 101 | return btk_STATUS_OK; |
| 102 | } |
| 103 | |
| 104 | /* ------------------------------------------------------------------------- */ |
| 105 | |
| 106 | btk_Status btk_DCR_close( btk_HDCR hdcrA ) |
| 107 | { |
| 108 | btk_HSDK hsdkL = NULL; |
| 109 | if( hdcrA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 110 | if( hdcrA->hidE != btk_HID_DCR ) return btk_STATUS_INVALID_HANDLE; |
| 111 | if( hdcrA->hsdkE == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 112 | hsdkL = hdcrA->hsdkE; |
| 113 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR; |
| 114 | |
| 115 | hsdkL->refCtrE--; |
| 116 | |
| 117 | btk_DCR_exit( &hsdkL->contextE, hdcrA ); |
| 118 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR; |
| 119 | |
| 120 | bbs_MemSeg_free( &hsdkL->contextE, hsdkL->contextE.memTblE.espArrE[ 0 ], hdcrA ); |
| 121 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR; |
| 122 | |
| 123 | return btk_STATUS_OK; |
| 124 | } |
| 125 | |
| 126 | /* ------------------------------------------------------------------------- */ |
| 127 | |
| 128 | btk_Status btk_DCR_assignGrayByteImage( btk_HDCR hdcrA, |
| 129 | const void* pDataA, |
| 130 | u32 widthA, |
| 131 | u32 heightA ) |
| 132 | { |
| 133 | return btk_DCR_assignImage( hdcrA, pDataA, widthA, heightA ); |
| 134 | } |
| 135 | |
| 136 | /* ------------------------------------------------------------------------- */ |
| 137 | |
| 138 | btk_Status btk_DCR_assignImage( btk_HDCR hdcrA, |
| 139 | const void* pDataA, |
| 140 | u32 widthA, |
| 141 | u32 heightA ) |
| 142 | { |
| 143 | const char* fNameL = "btk_DCR_assignImage"; |
| 144 | |
| 145 | btk_HSDK hsdkL = NULL; |
| 146 | if( hdcrA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 147 | if( hdcrA->hidE != btk_HID_DCR ) return btk_STATUS_INVALID_HANDLE; |
| 148 | hsdkL = hdcrA->hsdkE; |
| 149 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR; |
| 150 | |
| 151 | if( pDataA == NULL ) |
| 152 | { |
| 153 | bbs_Context_pushError( &hsdkL->contextE, |
| 154 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image references inavlid memory", fNameL ) ); |
| 155 | |
| 156 | return btk_STATUS_ERROR; |
| 157 | } |
| 158 | |
| 159 | if( widthA == 0 || heightA == 0 ) |
| 160 | { |
| 161 | bbs_Context_pushError( &hsdkL->contextE, |
| 162 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image has size 0", fNameL ) ); |
| 163 | |
| 164 | return btk_STATUS_ERROR; |
| 165 | } |
| 166 | |
| 167 | bpi_DCR_assignGrayByteImage( &hsdkL->contextE, &hdcrA->dcrE, pDataA, widthA, heightA ); |
| 168 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR; |
| 169 | |
| 170 | return btk_STATUS_OK; |
| 171 | } |
| 172 | |
| 173 | /* ------------------------------------------------------------------------- */ |
| 174 | |
| 175 | btk_Status btk_DCR_assignGrayByteImageROI( btk_HDCR hdcrA, |
| 176 | const void* pDataA, |
| 177 | u32 widthA, |
| 178 | u32 heightA, |
| 179 | const btk_Rect* pRectA ) |
| 180 | { |
| 181 | return btk_DCR_assignImageROI( hdcrA, pDataA, widthA, heightA, pRectA ); |
| 182 | } |
| 183 | |
| 184 | /* ------------------------------------------------------------------------- */ |
| 185 | |
| 186 | btk_Status btk_DCR_assignImageROI( btk_HDCR hdcrA, |
| 187 | const void* pDataA, |
| 188 | u32 widthA, |
| 189 | u32 heightA, |
| 190 | const btk_Rect* pRectA ) |
| 191 | { |
| 192 | const char* fNameL = "btk_DCR_assignGrayByteImageROI"; |
| 193 | |
| 194 | btk_HSDK hsdkL = NULL; |
| 195 | if( hdcrA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 196 | if( hdcrA->hidE != btk_HID_DCR ) return btk_STATUS_INVALID_HANDLE; |
| 197 | hsdkL = hdcrA->hsdkE; |
| 198 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR; |
| 199 | |
| 200 | if( pDataA == NULL ) |
| 201 | { |
| 202 | bbs_Context_pushError( &hsdkL->contextE, |
| 203 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image references invalid memory", fNameL ) ); |
| 204 | return btk_STATUS_ERROR; |
| 205 | } |
| 206 | |
| 207 | if( widthA == 0 || heightA == 0 ) |
| 208 | { |
| 209 | bbs_Context_pushError( &hsdkL->contextE, |
| 210 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image has size 0", fNameL ) ); |
| 211 | return btk_STATUS_ERROR; |
| 212 | } |
| 213 | |
| 214 | if( pRectA == NULL ) |
| 215 | { |
| 216 | bbs_Context_pushError( &hsdkL->contextE, |
| 217 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned ROI rectangle references invalid memory", fNameL ) ); |
| 218 | return btk_STATUS_ERROR; |
| 219 | } |
| 220 | |
| 221 | if( pRectA->xMax <= pRectA->xMin || pRectA->yMax <= pRectA->yMin ) |
| 222 | { |
| 223 | bbs_Context_pushError( &hsdkL->contextE, |
| 224 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned ROI rectangle is inverted (max<min) or zero", fNameL ) ); |
| 225 | return btk_STATUS_ERROR; |
| 226 | } |
| 227 | |
| 228 | { |
| 229 | struct bts_Int16Rect rectL; |
| 230 | rectL = bts_Int16Rect_create( pRectA->xMin >> 16, |
| 231 | pRectA->yMin >> 16, |
| 232 | pRectA->xMax >> 16, |
| 233 | pRectA->yMax >> 16 ); |
| 234 | |
| 235 | /* rect must stay within image boundaries - adjust coordinates if necessary */ |
| 236 | rectL.x1E = rectL.x1E < 0 ? 0 : rectL.x1E; |
| 237 | rectL.y1E = rectL.y1E < 0 ? 0 : rectL.y1E; |
| 238 | rectL.x2E = rectL.x2E > ( int32 )widthA ? widthA : rectL.x2E; |
| 239 | rectL.y2E = rectL.y2E > ( int32 )heightA ? heightA : rectL.y2E; |
| 240 | |
| 241 | bpi_DCR_assignGrayByteImageROI( &hsdkL->contextE, &hdcrA->dcrE, pDataA, widthA, heightA, &rectL ); |
| 242 | } |
| 243 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR; |
| 244 | |
| 245 | return btk_STATUS_OK; |
| 246 | } |
| 247 | |
| 248 | /* ------------------------------------------------------------------------- */ |
| 249 | |
| 250 | u32 btk_DCR_nodeCount( btk_HDCR hdcrA ) |
| 251 | { |
| 252 | if( hdcrA == NULL ) return 0; |
| 253 | if( hdcrA->hidE != btk_HID_DCR ) return 0; |
| 254 | return hdcrA->dcrE.sdkClusterE.clusterE.sizeE; |
| 255 | } |
| 256 | |
| 257 | /* ------------------------------------------------------------------------- */ |
| 258 | |
| 259 | btk_Status btk_DCR_getNode( btk_HDCR hdcrA, |
| 260 | u32 indexA, |
| 261 | btk_Node* nodePtrA ) |
| 262 | { |
| 263 | const char* fNameL = "btk_DCR_getNode"; |
| 264 | |
| 265 | btk_HSDK hsdkL = NULL; |
| 266 | if( hdcrA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 267 | if( hdcrA->hidE != btk_HID_DCR ) return btk_STATUS_INVALID_HANDLE; |
| 268 | hsdkL = hdcrA->hsdkE; |
| 269 | if( nodePtrA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 270 | |
| 271 | if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR; |
| 272 | |
| 273 | if( indexA >= hdcrA->dcrE.sdkClusterE.clusterE.sizeE ) |
| 274 | { |
| 275 | bbs_Context_pushError( &hsdkL->contextE, |
| 276 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nIndex is out of range", fNameL ) ); |
| 277 | return btk_STATUS_ERROR; |
| 278 | } |
| 279 | |
| 280 | nodePtrA->id = hdcrA->dcrE.sdkClusterE.idArrE.arrPtrE[ indexA ]; |
| 281 | nodePtrA->x = ( ( s16p16 )hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ indexA ].xE ) << ( 16 - hdcrA->dcrE.sdkClusterE.clusterE.bbpE ); |
| 282 | nodePtrA->y = ( ( s16p16 )hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ indexA ].yE ) << ( 16 - hdcrA->dcrE.sdkClusterE.clusterE.bbpE ); |
| 283 | if( hdcrA->dcrE.roiRectE.x1E > 0 ) nodePtrA->x += ( int32 )hdcrA->dcrE.roiRectE.x1E << 16; |
| 284 | if( hdcrA->dcrE.roiRectE.y1E > 0 ) nodePtrA->y += ( int32 )hdcrA->dcrE.roiRectE.y1E << 16; |
| 285 | nodePtrA->x += ( int32 )hdcrA->dcrE.offsE.xE << 16; |
| 286 | nodePtrA->y += ( int32 )hdcrA->dcrE.offsE.yE << 16; |
| 287 | |
| 288 | nodePtrA->reserved = 0; |
| 289 | |
| 290 | return btk_STATUS_OK; |
| 291 | } |
| 292 | |
| 293 | /* ------------------------------------------------------------------------- */ |
| 294 | |
| 295 | btk_Status btk_DCR_getRect( btk_HDCR hdcrA, |
| 296 | btk_Rect* pRectA ) |
| 297 | { |
| 298 | const char* fNameL = "btk_DCR_getRect"; |
| 299 | |
| 300 | btk_HSDK hsdkL = NULL; |
| 301 | if( hdcrA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 302 | if( hdcrA->hidE != btk_HID_DCR ) return btk_STATUS_INVALID_HANDLE; |
| 303 | hsdkL = hdcrA->hsdkE; |
| 304 | if( pRectA == NULL ) return btk_STATUS_INVALID_HANDLE; |
| 305 | |
| 306 | /* find eye nodes */ |
| 307 | { |
| 308 | const struct bbs_Int16Arr* pIdArrL = &hdcrA->dcrE.sdkClusterE.idArrE; |
| 309 | int32 lIndexL = -1; |
| 310 | int32 rIndexL = -1; |
| 311 | uint32 iL; |
| 312 | for( iL = 0; iL < pIdArrL->sizeE; iL++ ) |
| 313 | { |
| 314 | if( pIdArrL->arrPtrE[ iL ] == 0 ) |
| 315 | { |
| 316 | lIndexL = iL; |
| 317 | } |
| 318 | else if( pIdArrL->arrPtrE[ iL ] == 1 ) |
| 319 | { |
| 320 | rIndexL = iL; |
| 321 | } |
| 322 | } |
| 323 | |
| 324 | if( lIndexL == -1 || rIndexL == -1 ) |
| 325 | { |
| 326 | bbs_Context_pushError( &hsdkL->contextE, |
| 327 | bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nFace rectangle is not available", fNameL ) ); |
| 328 | return btk_STATUS_ERROR; |
| 329 | } |
| 330 | |
| 331 | { |
| 332 | int32 bbpL = hdcrA->dcrE.sdkClusterE.clusterE.bbpE; |
| 333 | int32 lxL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ lIndexL ].xE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL; |
| 334 | int32 lyL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ lIndexL ].yE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL; |
| 335 | int32 rxL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ rIndexL ].xE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL; |
| 336 | int32 ryL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ rIndexL ].yE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL; |
| 337 | int32 doffL = ( rxL - lxL ) >> 1; |
| 338 | |
| 339 | pRectA->xMin = ( lxL - doffL ) << 16; |
| 340 | pRectA->xMax = ( rxL + doffL ) << 16; |
| 341 | pRectA->yMin = ( ( ( lyL + ryL + 1 ) >> 1 ) - doffL ) << 16; |
| 342 | pRectA->yMax = ( pRectA->yMin + ( pRectA->xMax - pRectA->xMin ) ); |
| 343 | if( hdcrA->dcrE.roiRectE.x1E > 0 ) |
| 344 | { |
| 345 | pRectA->xMin += ( int32 )hdcrA->dcrE.roiRectE.x1E << 16; |
| 346 | pRectA->xMax += ( int32 )hdcrA->dcrE.roiRectE.x1E << 16; |
| 347 | } |
| 348 | if( hdcrA->dcrE.roiRectE.y1E > 0 ) |
| 349 | { |
| 350 | pRectA->yMin += ( int32 )hdcrA->dcrE.roiRectE.y1E << 16; |
| 351 | pRectA->yMax += ( int32 )hdcrA->dcrE.roiRectE.y1E << 16; |
| 352 | } |
| 353 | |
| 354 | pRectA->xMin += ( int32 )hdcrA->dcrE.offsE.xE << 16; |
| 355 | pRectA->yMin += ( int32 )hdcrA->dcrE.offsE.yE << 16; |
| 356 | pRectA->xMax += ( int32 )hdcrA->dcrE.offsE.xE << 16; |
| 357 | pRectA->yMax += ( int32 )hdcrA->dcrE.offsE.yE << 16; |
| 358 | |
| 359 | } |
| 360 | } |
| 361 | |
| 362 | return btk_STATUS_OK; |
| 363 | } |
| 364 | |
| 365 | |
| 366 | /* ------------------------------------------------------------------------- */ |
| 367 | |
| 368 | s8p24 btk_DCR_confidence( btk_HDCR hdcrA ) |
| 369 | { |
| 370 | btk_HSDK hsdkL = NULL; |
| 371 | if( hdcrA == NULL ) return 0; |
| 372 | if( hdcrA->hidE != btk_HID_DCR ) return 0; |
| 373 | hsdkL = hdcrA->hsdkE; |
| 374 | if( bbs_Context_error( &hsdkL->contextE ) ) return 0; |
| 375 | |
| 376 | return hdcrA->dcrE.confidenceE; |
| 377 | } |
| 378 | |
| 379 | /* ------------------------------------------------------------------------- */ |
| 380 | |
| 381 | u32 btk_DCR_approved( btk_HDCR hdcrA ) |
| 382 | { |
| 383 | btk_HSDK hsdkL = NULL; |
| 384 | if( hdcrA == NULL ) return 0; |
| 385 | if( hdcrA->hidE != btk_HID_DCR ) return 0; |
| 386 | hsdkL = hdcrA->hsdkE; |
| 387 | if( bbs_Context_error( &hsdkL->contextE ) ) return 0; |
| 388 | |
| 389 | return ( u32 )hdcrA->dcrE.approvedE; |
| 390 | } |
| 391 | |
| 392 | /* ------------------------------------------------------------------------- */ |
| 393 | |
| 394 | /* ========================================================================= */ |