FaceRecEm/common/src/b_FDSDK/DCR.c

/*
 * Copyright (C) 2008 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/* ---- includes ----------------------------------------------------------- */

#include "DCR_Internal.h"

/* ---- related objects  --------------------------------------------------- */

/* ---- typedefs ----------------------------------------------------------- */

/* ---- constants ---------------------------------------------------------- */

/* ------------------------------------------------------------------------- */

/* ========================================================================= */
/*                                                                           */
/* ---- functions ---------------------------------------------------------- */
/*                                                                           */
/* ========================================================================= */

/* ------------------------------------------------------------------------- */

void btk_DCR_init( struct bbs_Context* cpA, struct btk_DCR* ptrA )
{
	ptrA->hsdkE = NULL;
	ptrA->hidE = btk_HID_DCR;
	bpi_DCR_init( cpA, &ptrA->dcrE );
}

/* ------------------------------------------------------------------------- */

void btk_DCR_exit( struct bbs_Context* cpA, struct btk_DCR* ptrA )
{
	ptrA->hsdkE = NULL;
	ptrA->hidE = btk_HID_DCR;
	bpi_DCR_exit( cpA, &ptrA->dcrE );
}

/* ------------------------------------------------------------------------- */

btk_DCRCreateParam btk_DCR_defaultParam()
{
	btk_DCRCreateParam paramL;
	paramL.reserved = 0;
	return paramL;
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_create( btk_HSDK hsdkA, 
						   const btk_DCRCreateParam* pCreateParamA,
						   btk_HDCR* hpdcrA )
{
	btk_HDCR hdcrL = NULL;

	if( hpdcrA == NULL )						return btk_STATUS_INVALID_HANDLE;
	if( *hpdcrA != NULL )						return btk_STATUS_INVALID_HANDLE;
	if( hsdkA == NULL )							return btk_STATUS_INVALID_HANDLE;
	if( hsdkA->hidE != btk_HID_SDK )			return btk_STATUS_INVALID_HANDLE;
	if( pCreateParamA == NULL )					return btk_STATUS_INVALID_HANDLE;
	if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_PREEXISTING_ERROR;

	hdcrL = ( btk_HDCR )bbs_MemSeg_alloc( &hsdkA->contextE, hsdkA->contextE.memTblE.espArrE[ 0 ], bbs_SIZEOF16( struct btk_DCR ) );
	if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_ERROR;

	btk_DCR_init( &hsdkA->contextE, hdcrL );
	hdcrL->hsdkE = hsdkA;

	if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_ERROR;

	bpi_DCR_create( &hsdkA->contextE,
		            &hdcrL->dcrE,
					hsdkA->maxImageWidthE,
					hsdkA->maxImageHeightE,
#ifdef btk_FRSDK
					6000 >> 1,
#else
                        0,
#endif
					&hsdkA->contextE.memTblE );

	if( bbs_Context_error( &hsdkA->contextE ) ) return btk_STATUS_ERROR;

	*hpdcrA = hdcrL;
	hsdkA->refCtrE++;

	return btk_STATUS_OK;
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_close( btk_HDCR hdcrA )
{
	btk_HSDK hsdkL = NULL;
	if( hdcrA == NULL )					return btk_STATUS_INVALID_HANDLE;
	if( hdcrA->hidE != btk_HID_DCR )	return btk_STATUS_INVALID_HANDLE;
	if( hdcrA->hsdkE == NULL )			return btk_STATUS_INVALID_HANDLE;
	hsdkL = hdcrA->hsdkE;
	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR;

	hsdkL->refCtrE--;

	btk_DCR_exit( &hsdkL->contextE, hdcrA );
	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR;

	bbs_MemSeg_free( &hsdkL->contextE, hsdkL->contextE.memTblE.espArrE[ 0 ], hdcrA );
	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR;

	return btk_STATUS_OK;
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_assignGrayByteImage( btk_HDCR hdcrA, 
									    const void* pDataA, 
										u32 widthA, 
										u32 heightA )
{
	return btk_DCR_assignImage( hdcrA, pDataA, widthA, heightA );
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_assignImage( btk_HDCR hdcrA, 
							    const void* pDataA, 
								u32 widthA, 
								u32 heightA )
{
	const char* fNameL = "btk_DCR_assignImage";

	btk_HSDK hsdkL = NULL;
	if( hdcrA == NULL )					return btk_STATUS_INVALID_HANDLE;
	if( hdcrA->hidE != btk_HID_DCR )	return btk_STATUS_INVALID_HANDLE;
	hsdkL = hdcrA->hsdkE;
	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR;

	if( pDataA == NULL )
	{
		bbs_Context_pushError( &hsdkL->contextE, 
			                   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image references inavlid memory", fNameL ) );

		return btk_STATUS_ERROR;
	}

	if( widthA == 0 || heightA == 0 )
	{
		bbs_Context_pushError( &hsdkL->contextE, 
			                   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image has size 0", fNameL ) );

		return btk_STATUS_ERROR;
	}

	bpi_DCR_assignGrayByteImage( &hsdkL->contextE, &hdcrA->dcrE, pDataA, widthA, heightA );
	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR;

	return btk_STATUS_OK;
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_assignGrayByteImageROI( btk_HDCR hdcrA, 
										   const void* pDataA, 
										   u32 widthA, 
										   u32 heightA,
										   const btk_Rect* pRectA )
{
	return btk_DCR_assignImageROI( hdcrA, pDataA, widthA, heightA, pRectA );
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_assignImageROI( btk_HDCR hdcrA, 
								   const void* pDataA, 
								   u32 widthA, 
								   u32 heightA,
								   const btk_Rect* pRectA )
{
	const char* fNameL = "btk_DCR_assignGrayByteImageROI";

	btk_HSDK hsdkL = NULL;
	if( hdcrA == NULL )					return btk_STATUS_INVALID_HANDLE;
	if( hdcrA->hidE != btk_HID_DCR )	return btk_STATUS_INVALID_HANDLE;
	hsdkL = hdcrA->hsdkE;
	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR;

	if( pDataA == NULL )
	{
		bbs_Context_pushError( &hsdkL->contextE, 
			                   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image references invalid memory", fNameL ) );
		return btk_STATUS_ERROR;
	}

	if( widthA == 0 || heightA == 0 )
	{
		bbs_Context_pushError( &hsdkL->contextE, 
			                   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned image has size 0", fNameL ) );
		return btk_STATUS_ERROR;
	}
	
	if( pRectA == NULL )
	{
		bbs_Context_pushError( &hsdkL->contextE, 
			                   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned ROI rectangle references invalid memory", fNameL ) );
		return btk_STATUS_ERROR;
	}

	if( pRectA->xMax <= pRectA->xMin || pRectA->yMax <= pRectA->yMin )
	{
		bbs_Context_pushError( &hsdkL->contextE, 
			                   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nAssigned ROI rectangle is inverted (max<min) or zero", fNameL ) );
		return btk_STATUS_ERROR;
	}

	{
		struct bts_Int16Rect rectL;
		rectL = bts_Int16Rect_create( pRectA->xMin >> 16, 
									  pRectA->yMin >> 16, 
									  pRectA->xMax >> 16, 
									  pRectA->yMax >> 16 );

		/* rect must stay within image boundaries - adjust coordinates if necessary */
		rectL.x1E = rectL.x1E < 0         ? 0 : rectL.x1E;
		rectL.y1E = rectL.y1E < 0         ? 0 : rectL.y1E;
		rectL.x2E = rectL.x2E > ( int32 )widthA    ? widthA : rectL.x2E;
		rectL.y2E = rectL.y2E > ( int32 )heightA   ? heightA : rectL.y2E;

		bpi_DCR_assignGrayByteImageROI( &hsdkL->contextE, &hdcrA->dcrE, pDataA, widthA, heightA, &rectL );
	}
	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_ERROR;

	return btk_STATUS_OK;
}

/* ------------------------------------------------------------------------- */

u32 btk_DCR_nodeCount( btk_HDCR hdcrA )
{
	if( hdcrA == NULL )					return 0;
	if( hdcrA->hidE != btk_HID_DCR )	return 0;
	return hdcrA->dcrE.sdkClusterE.clusterE.sizeE;
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_getNode( btk_HDCR hdcrA, 
						    u32 indexA, 
							btk_Node* nodePtrA )
{
	const char* fNameL = "btk_DCR_getNode";

	btk_HSDK hsdkL = NULL;
	if( hdcrA == NULL )					return btk_STATUS_INVALID_HANDLE;
	if( hdcrA->hidE != btk_HID_DCR )	return btk_STATUS_INVALID_HANDLE;
	hsdkL = hdcrA->hsdkE;
	if( nodePtrA == NULL ) return btk_STATUS_INVALID_HANDLE;

	if( bbs_Context_error( &hsdkL->contextE ) ) return btk_STATUS_PREEXISTING_ERROR;

	if( indexA >= hdcrA->dcrE.sdkClusterE.clusterE.sizeE )
	{
		bbs_Context_pushError( &hsdkL->contextE, 
			                   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nIndex is out of range", fNameL ) );
		return btk_STATUS_ERROR;
	}

	nodePtrA->id = hdcrA->dcrE.sdkClusterE.idArrE.arrPtrE[ indexA ];
	nodePtrA->x  = ( ( s16p16 )hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ indexA ].xE ) << ( 16 - hdcrA->dcrE.sdkClusterE.clusterE.bbpE );
	nodePtrA->y  = ( ( s16p16 )hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ indexA ].yE ) << ( 16 - hdcrA->dcrE.sdkClusterE.clusterE.bbpE );
	if( hdcrA->dcrE.roiRectE.x1E > 0 ) nodePtrA->x += ( int32 )hdcrA->dcrE.roiRectE.x1E << 16;
	if( hdcrA->dcrE.roiRectE.y1E > 0 ) nodePtrA->y += ( int32 )hdcrA->dcrE.roiRectE.y1E << 16;
	nodePtrA->x += ( int32 )hdcrA->dcrE.offsE.xE << 16;
	nodePtrA->y += ( int32 )hdcrA->dcrE.offsE.yE << 16;

	nodePtrA->reserved = 0;

	return btk_STATUS_OK;
}

/* ------------------------------------------------------------------------- */

btk_Status btk_DCR_getRect( btk_HDCR hdcrA, 
							btk_Rect* pRectA )
{
	const char* fNameL = "btk_DCR_getRect";

	btk_HSDK hsdkL = NULL;
	if( hdcrA == NULL )					return btk_STATUS_INVALID_HANDLE;
	if( hdcrA->hidE != btk_HID_DCR )	return btk_STATUS_INVALID_HANDLE;
	hsdkL = hdcrA->hsdkE;
	if( pRectA == NULL ) return btk_STATUS_INVALID_HANDLE;

	/* find eye nodes */
	{
		const struct bbs_Int16Arr* pIdArrL = &hdcrA->dcrE.sdkClusterE.idArrE;
		int32 lIndexL = -1;
		int32 rIndexL = -1;
		uint32 iL;
		for( iL = 0; iL < pIdArrL->sizeE; iL++ )
		{
			if( pIdArrL->arrPtrE[ iL ] == 0 )
			{
				lIndexL = iL;
			}
			else if( pIdArrL->arrPtrE[ iL ] == 1 )
			{
				rIndexL = iL;
			}
		}

		if( lIndexL == -1 || rIndexL == -1 )
		{
			bbs_Context_pushError( &hsdkL->contextE, 
								   bbs_Error_create( bbs_ERR_ERROR, 0, NULL, "%s:\nFace rectangle is not available", fNameL ) );
			return btk_STATUS_ERROR;
		}

		{
			int32 bbpL = hdcrA->dcrE.sdkClusterE.clusterE.bbpE; 
			int32 lxL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ lIndexL ].xE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL;
			int32 lyL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ lIndexL ].yE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL;
			int32 rxL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ rIndexL ].xE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL;
			int32 ryL = ( hdcrA->dcrE.sdkClusterE.clusterE.vecArrE[ rIndexL ].yE + ( 1 << ( bbpL - 1 ) ) ) >> bbpL;
			int32 doffL = ( rxL - lxL ) >> 1;

			pRectA->xMin = ( lxL - doffL ) << 16;
			pRectA->xMax = ( rxL + doffL ) << 16;
			pRectA->yMin = ( ( ( lyL + ryL + 1 ) >> 1 ) - doffL ) << 16;
			pRectA->yMax = ( pRectA->yMin + ( pRectA->xMax - pRectA->xMin ) );
			if( hdcrA->dcrE.roiRectE.x1E > 0 ) 
			{	
				pRectA->xMin += ( int32 )hdcrA->dcrE.roiRectE.x1E << 16;
				pRectA->xMax += ( int32 )hdcrA->dcrE.roiRectE.x1E << 16;
			}
			if( hdcrA->dcrE.roiRectE.y1E > 0 ) 
			{
				pRectA->yMin += ( int32 )hdcrA->dcrE.roiRectE.y1E << 16;
				pRectA->yMax += ( int32 )hdcrA->dcrE.roiRectE.y1E << 16;
			}

			pRectA->xMin += ( int32 )hdcrA->dcrE.offsE.xE << 16;
			pRectA->yMin += ( int32 )hdcrA->dcrE.offsE.yE << 16;
			pRectA->xMax += ( int32 )hdcrA->dcrE.offsE.xE << 16;
			pRectA->yMax += ( int32 )hdcrA->dcrE.offsE.yE << 16;

		}
	}

	return btk_STATUS_OK;
}


/* ------------------------------------------------------------------------- */

s8p24 btk_DCR_confidence( btk_HDCR hdcrA )
{
	btk_HSDK hsdkL = NULL;
	if( hdcrA == NULL )					return 0;
	if( hdcrA->hidE != btk_HID_DCR )	return 0;
	hsdkL = hdcrA->hsdkE;
	if( bbs_Context_error( &hsdkL->contextE ) ) return 0;

	return hdcrA->dcrE.confidenceE;
}

/* ------------------------------------------------------------------------- */

u32 btk_DCR_approved( btk_HDCR hdcrA )
{
	btk_HSDK hsdkL = NULL;
	if( hdcrA == NULL )					return 0;
	if( hdcrA->hidE != btk_HID_DCR )	return 0;
	hsdkL = hdcrA->hsdkE;
	if( bbs_Context_error( &hsdkL->contextE ) ) return 0;

	return ( u32 )hdcrA->dcrE.approvedE;
}

/* ------------------------------------------------------------------------- */

/* ========================================================================= */