/*M/////////////////////////////////////////////////////////////////////////////////////// // // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. // // By downloading, copying, installing or using the software you agree to this license. // If you do not agree to this license, do not download, install, // copy or use the software. // // // Intel License Agreement // For Open Source Computer Vision Library // // Copyright (C) 2000, Intel Corporation, all rights reserved. // Third party copyrights are property of their respective owners. // // Redistribution and use in source and binary forms, with or without modification, // are permitted provided that the following conditions are met: // // * Redistribution's of source code must retain the above copyright notice, // this list of conditions and the following disclaimer. // // * Redistribution's in binary form must reproduce the above copyright notice, // this list of conditions and the following disclaimer in the documentation // and/or other materials provided with the distribution. // // * The name of Intel Corporation may not be used to endorse or promote products // derived from this software without specific prior written permission. // // This software is provided by the copyright holders and contributors "as is" and // any express or implied warranties, including, but not limited to, the implied // warranties of merchantability and fitness for a particular purpose are disclaimed. // In no event shall the Intel Corporation or contributors be liable for any direct, // indirect, incidental, special, exemplary, or consequential damages // (including, but not limited to, procurement of substitute goods or services; // loss of use, data, or profits; or business interruption) however caused // and on any theory of liability, whether in contract, strict liability, // or tort (including negligence or otherwise) arising in any way out of // the use of this software, even if advised of the possibility of such damage. // //M*/ #include "_cxcore.h" #include <float.h> /****************************************************************************************\ * Mean value over the region * \****************************************************************************************/ #define ICV_MEAN_CASE_C1( len ) \ for( ; x <= (len) - 2; x += 2 ) \ { \ if( mask[x] ) \ s0 += src[x], pix++; \ if( mask[x+1] ) \ s0 += src[x+1], pix++; \ } \ \ for( ; x < (len); x++ ) \ if( mask[x] ) \ s0 += src[x], pix++ #define ICV_MEAN_CASE_C2( len ) \ for( ; x < (len); x++ ) \ if( mask[x] ) \ { \ s0 += src[x*2]; \ s1 += src[x*2+1]; \ pix++; \ } #define ICV_MEAN_CASE_C3( len ) \ for( ; x < (len); x++ ) \ if( mask[x] ) \ { \ s0 += src[x*3]; \ s1 += src[x*3+1]; \ s2 += src[x*3+2]; \ pix++; \ } #define ICV_MEAN_CASE_C4( len ) \ for( ; x < (len); x++ ) \ if( mask[x] ) \ { \ s0 += src[x*4]; \ s1 += src[x*4+1]; \ s2 += src[x*4+2]; \ s3 += src[x*4+3]; \ pix++; \ } #define ICV_MEAN_COI_CASE( len, cn ) \ for( ; x <= (len) - 2; x += 2 ) \ { \ if( mask[x] ) \ s0 += src[x*(cn)], pix++; \ if( mask[x+1] ) \ s0+=src[(x+1)*(cn)], pix++; \ } \ \ for( ; x < (len); x++ ) \ if( mask[x] ) \ s0 += src[x*(cn)], pix++; ////////////////////////////////////// entry macros ////////////////////////////////////// #define ICV_MEAN_ENTRY_COMMON() \ int pix = 0; \ step /= sizeof(src[0]) #define ICV_MEAN_ENTRY_C1( sumtype ) \ sumtype s0 = 0; \ ICV_MEAN_ENTRY_COMMON() #define ICV_MEAN_ENTRY_C2( sumtype ) \ sumtype s0 = 0, s1 = 0; \ ICV_MEAN_ENTRY_COMMON() #define ICV_MEAN_ENTRY_C3( sumtype ) \ sumtype s0 = 0, s1 = 0, s2 = 0; \ ICV_MEAN_ENTRY_COMMON() #define ICV_MEAN_ENTRY_C4( sumtype ) \ sumtype s0 = 0, s1 = 0, s2 = 0, s3 = 0; \ ICV_MEAN_ENTRY_COMMON() #define ICV_MEAN_ENTRY_BLOCK_COMMON( block_size ) \ int remaining = block_size; \ ICV_MEAN_ENTRY_COMMON() #define ICV_MEAN_ENTRY_BLOCK_C1( sumtype, worktype, block_size )\ sumtype sum0 = 0; \ worktype s0 = 0; \ ICV_MEAN_ENTRY_BLOCK_COMMON( block_size ) #define ICV_MEAN_ENTRY_BLOCK_C2( sumtype, worktype, block_size )\ sumtype sum0 = 0, sum1 = 0; \ worktype s0 = 0, s1 = 0; \ ICV_MEAN_ENTRY_BLOCK_COMMON( block_size ) #define ICV_MEAN_ENTRY_BLOCK_C3( sumtype, worktype, block_size )\ sumtype sum0 = 0, sum1 = 0, sum2 = 0; \ worktype s0 = 0, s1 = 0, s2 = 0; \ ICV_MEAN_ENTRY_BLOCK_COMMON( block_size ) #define ICV_MEAN_ENTRY_BLOCK_C4( sumtype, worktype, block_size )\ sumtype sum0 = 0, sum1 = 0, sum2 = 0, sum3 = 0; \ worktype s0 = 0, s1 = 0, s2 = 0, s3 = 0; \ ICV_MEAN_ENTRY_BLOCK_COMMON( block_size ) /////////////////////////////////////// exit macros ////////////////////////////////////// #define ICV_MEAN_EXIT_COMMON() \ double scale = pix ? 1./pix : 0 #define ICV_MEAN_EXIT_C1( tmp ) \ ICV_MEAN_EXIT_COMMON(); \ mean[0] = scale*(double)tmp##0 #define ICV_MEAN_EXIT_C2( tmp ) \ ICV_MEAN_EXIT_COMMON(); \ double t0 = scale*(double)tmp##0; \ double t1 = scale*(double)tmp##1; \ mean[0] = t0; \ mean[1] = t1 #define ICV_MEAN_EXIT_C3( tmp ) \ ICV_MEAN_EXIT_COMMON(); \ double t0 = scale*(double)tmp##0; \ double t1 = scale*(double)tmp##1; \ double t2 = scale*(double)tmp##2; \ mean[0] = t0; \ mean[1] = t1; \ mean[2] = t2 #define ICV_MEAN_EXIT_C4( tmp ) \ ICV_MEAN_EXIT_COMMON(); \ double t0 = scale*(double)tmp##0; \ double t1 = scale*(double)tmp##1; \ mean[0] = t0; \ mean[1] = t1; \ t0 = scale*(double)tmp##2; \ t1 = scale*(double)tmp##3; \ mean[2] = t0; \ mean[3] = t1 #define ICV_MEAN_EXIT_BLOCK_C1() \ sum0 += s0; \ ICV_MEAN_EXIT_C1( sum ) #define ICV_MEAN_EXIT_BLOCK_C2() \ sum0 += s0; sum1 += s1; \ ICV_MEAN_EXIT_C2( sum ) #define ICV_MEAN_EXIT_BLOCK_C3() \ sum0 += s0; sum1 += s1; \ sum2 += s2; \ ICV_MEAN_EXIT_C3( sum ) #define ICV_MEAN_EXIT_BLOCK_C4() \ sum0 += s0; sum1 += s1; \ sum2 += s2; sum3 += s3; \ ICV_MEAN_EXIT_C4( sum ) ////////////////////////////////////// update macros ///////////////////////////////////// #define ICV_MEAN_UPDATE_COMMON( block_size )\ remaining = block_size #define ICV_MEAN_UPDATE_C1( block_size ) \ ICV_MEAN_UPDATE_COMMON( block_size ); \ sum0 += s0; \ s0 = 0 #define ICV_MEAN_UPDATE_C2( block_size ) \ ICV_MEAN_UPDATE_COMMON( block_size ); \ sum0 += s0; sum1 += s1; \ s0 = s1 = 0 #define ICV_MEAN_UPDATE_C3( block_size ) \ ICV_MEAN_UPDATE_COMMON( block_size ); \ sum0 += s0; sum1 += s1; sum2 += s2; \ s0 = s1 = s2 = 0 #define ICV_MEAN_UPDATE_C4( block_size ) \ ICV_MEAN_UPDATE_COMMON( block_size ); \ sum0 += s0; sum1 += s1; \ sum2 += s2; sum3 += s3; \ s0 = s1 = s2 = s3 = 0 #define ICV_IMPL_MEAN_BLOCK_FUNC_2D( flavor, cn, \ arrtype, sumtype, worktype, block_size ) \ IPCVAPI_IMPL( CvStatus, icvMean_##flavor##_C##cn##MR, \ ( const arrtype* src, int step, \ const uchar* mask, int maskstep, \ CvSize size, double* mean ), \ (src, step, mask, maskstep, size, mean)) \ { \ ICV_MEAN_ENTRY_BLOCK_C##cn( sumtype, worktype, block_size );\ \ for( ; size.height--; src += step, mask += maskstep ) \ { \ int x = 0; \ while( x < size.width ) \ { \ int limit = MIN( remaining, size.width - x ); \ remaining -= limit; \ limit += x; \ ICV_MEAN_CASE_C##cn( limit ); \ if( remaining == 0 ) \ { \ ICV_MEAN_UPDATE_C##cn( block_size ); \ } \ } \ } \ \ { ICV_MEAN_EXIT_BLOCK_C##cn(); } \ return CV_OK; \ } #define ICV_IMPL_MEAN_FUNC_2D( flavor, cn, \ arrtype, sumtype, worktype ) \ IPCVAPI_IMPL( CvStatus, icvMean_##flavor##_C##cn##MR, \ ( const arrtype* src, int step, \ const uchar* mask, int maskstep, \ CvSize size, double* mean), \ (src, step, mask, maskstep, size, mean)) \ { \ ICV_MEAN_ENTRY_C##cn( sumtype ); \ \ for( ; size.height--; src += step, mask += maskstep ) \ { \ int x = 0; \ ICV_MEAN_CASE_C##cn( size.width ); \ } \ \ { ICV_MEAN_EXIT_C##cn( s ); } \ return CV_OK; \ } #define ICV_IMPL_MEAN_BLOCK_FUNC_2D_COI( flavor, \ arrtype, sumtype, worktype, block_size ) \ static CvStatus CV_STDCALL \ icvMean_##flavor##_CnCMR( const arrtype* src, int step, \ const uchar* mask, int maskstep, \ CvSize size, int cn, \ int coi, double* mean ) \ { \ ICV_MEAN_ENTRY_BLOCK_C1( sumtype, worktype, block_size ); \ src += coi - 1; \ \ for( ; size.height--; src += step, mask += maskstep ) \ { \ int x = 0; \ while( x < size.width ) \ { \ int limit = MIN( remaining, size.width - x ); \ remaining -= limit; \ limit += x; \ ICV_MEAN_COI_CASE( limit, cn ); \ if( remaining == 0 ) \ { \ ICV_MEAN_UPDATE_C1( block_size ); \ } \ } \ } \ \ { ICV_MEAN_EXIT_BLOCK_C1(); } \ return CV_OK; \ } #define ICV_IMPL_MEAN_FUNC_2D_COI( flavor, \ arrtype, sumtype, worktype ) \ static CvStatus CV_STDCALL \ icvMean_##flavor##_CnCMR( const arrtype* src, int step, \ const uchar* mask, int maskstep, \ CvSize size, int cn, \ int coi, double* mean ) \ { \ ICV_MEAN_ENTRY_C1( sumtype ); \ src += coi - 1; \ \ for( ; size.height--; src += step, mask += maskstep ) \ { \ int x = 0; \ ICV_MEAN_COI_CASE( size.width, cn ); \ } \ \ { ICV_MEAN_EXIT_C1( s ); } \ return CV_OK; \ } #define ICV_IMPL_MEAN_BLOCK_ALL( flavor, arrtype, sumtype, \ worktype, block_size ) \ ICV_IMPL_MEAN_BLOCK_FUNC_2D( flavor, 1, arrtype, sumtype, \ worktype, block_size ) \ ICV_IMPL_MEAN_BLOCK_FUNC_2D( flavor, 2, arrtype, sumtype, \ worktype, block_size ) \ ICV_IMPL_MEAN_BLOCK_FUNC_2D( flavor, 3, arrtype, sumtype, \ worktype, block_size ) \ ICV_IMPL_MEAN_BLOCK_FUNC_2D( flavor, 4, arrtype, sumtype, \ worktype, block_size ) \ ICV_IMPL_MEAN_BLOCK_FUNC_2D_COI( flavor, arrtype, sumtype, \ worktype, block_size ) #define ICV_IMPL_MEAN_ALL( flavor, arrtype, sumtype, worktype ) \ ICV_IMPL_MEAN_FUNC_2D( flavor, 1, arrtype, sumtype, worktype ) \ ICV_IMPL_MEAN_FUNC_2D( flavor, 2, arrtype, sumtype, worktype ) \ ICV_IMPL_MEAN_FUNC_2D( flavor, 3, arrtype, sumtype, worktype ) \ ICV_IMPL_MEAN_FUNC_2D( flavor, 4, arrtype, sumtype, worktype ) \ ICV_IMPL_MEAN_FUNC_2D_COI( flavor, arrtype, sumtype, worktype ) ICV_IMPL_MEAN_BLOCK_ALL( 8u, uchar, int64, unsigned, 1 << 24 ) ICV_IMPL_MEAN_BLOCK_ALL( 16u, ushort, int64, unsigned, 1 << 16 ) ICV_IMPL_MEAN_BLOCK_ALL( 16s, short, int64, int, 1 << 16 ) ICV_IMPL_MEAN_ALL( 32s, int, double, double ) ICV_IMPL_MEAN_ALL( 32f, float, double, double ) ICV_IMPL_MEAN_ALL( 64f, double, double, double ) #define icvMean_8s_C1MR 0 #define icvMean_8s_C2MR 0 #define icvMean_8s_C3MR 0 #define icvMean_8s_C4MR 0 #define icvMean_8s_CnCMR 0 CV_DEF_INIT_BIG_FUNC_TAB_2D( Mean, MR ) CV_DEF_INIT_FUNC_TAB_2D( Mean, CnCMR ) CV_IMPL CvScalar cvAvg( const void* img, const void* maskarr ) { CvScalar mean = {{0,0,0,0}}; static CvBigFuncTable mean_tab; static CvFuncTable meancoi_tab; static int inittab = 0; CV_FUNCNAME("cvAvg"); __BEGIN__; CvSize size; double scale; if( !maskarr ) { CV_CALL( mean = cvSum(img)); size = cvGetSize( img ); size.width *= size.height; scale = size.width ? 1./size.width : 0; mean.val[0] *= scale; mean.val[1] *= scale; mean.val[2] *= scale; mean.val[3] *= scale; } else { int type, coi = 0; int mat_step, mask_step; CvMat stub, maskstub, *mat = (CvMat*)img, *mask = (CvMat*)maskarr; if( !inittab ) { icvInitMeanMRTable( &mean_tab ); icvInitMeanCnCMRTable( &meancoi_tab ); inittab = 1; } if( !CV_IS_MAT(mat) ) CV_CALL( mat = cvGetMat( mat, &stub, &coi )); if( !CV_IS_MAT(mask) ) CV_CALL( mask = cvGetMat( mask, &maskstub )); if( !CV_IS_MASK_ARR(mask) ) CV_ERROR( CV_StsBadMask, "" ); if( !CV_ARE_SIZES_EQ( mat, mask ) ) CV_ERROR( CV_StsUnmatchedSizes, "" ); type = CV_MAT_TYPE( mat->type ); size = cvGetMatSize( mat ); mat_step = mat->step; mask_step = mask->step; if( CV_IS_MAT_CONT( mat->type & mask->type )) { size.width *= size.height; size.height = 1; mat_step = mask_step = CV_STUB_STEP; } if( CV_MAT_CN(type) == 1 || coi == 0 ) { CvFunc2D_2A1P func; if( CV_MAT_CN(type) > 4 ) CV_ERROR( CV_StsOutOfRange, "The input array must have at most 4 channels unless COI is set" ); func = (CvFunc2D_2A1P)(mean_tab.fn_2d[type]); if( !func ) CV_ERROR( CV_StsBadArg, cvUnsupportedFormat ); IPPI_CALL( func( mat->data.ptr, mat_step, mask->data.ptr, mask_step, size, mean.val )); } else { CvFunc2DnC_2A1P func = (CvFunc2DnC_2A1P)( meancoi_tab.fn_2d[CV_MAT_DEPTH(type)]); if( !func ) CV_ERROR( CV_StsBadArg, cvUnsupportedFormat ); IPPI_CALL( func( mat->data.ptr, mat_step, mask->data.ptr, mask_step, size, CV_MAT_CN(type), coi, mean.val )); } } __END__; return mean; } /* End of file */