cvaccum.cpp.svn-base

来自「非结构化路识别」· SVN-BASE 代码 · 共 1,023 行 · 第 1/4 页

                            CV_PREPARE_FLT_MASK )


#define  ICV_DEF_ACC_ALL_FLT( flavor, srctype, dsttype )                               \
                                                                                       \
ICV_DEF_ACC_FUNC( icvAddSquare_##flavor##_C1IR, srctype, dsttype, CV_SQR )             \
ICV_DEF_ACCPROD_FUNC( icvAddProduct_##flavor##_C1IR, srctype, dsttype, CV_NOP )        \
ICV_DEF_ACCWEIGHT_FUNC( icvAddWeighted_##flavor##_C1IR, srctype, dsttype, CV_NOP )     \
                                                                                       \
ICV_DEF_ACCMASK_FUNC( icvAdd_##flavor##_C1IMR, 1, srctype, dsttype,                    \
                       CV_NOP, CV_MULMASK, CV_DEFINE_MASK, CV_PREPARE_FLT_MASK )       \
ICV_DEF_ACCMASK_FUNC( icvAdd_##flavor##_C3IMR, 3, srctype, dsttype,                    \
                       CV_NOP, CV_MUL, CV_DEFINE_MASK, CV_PREPARE_FLT_MASK )           \
                                                                                       \
ICV_DEF_ACCMASK_FUNC( icvAddSquare_##flavor##_C1IMR, 1, srctype, dsttype,              \
                       CV_SQR, CV_MULMASK, CV_DEFINE_MASK, CV_PREPARE_FLT_MASK )       \
ICV_DEF_ACCMASK_FUNC( icvAddSquare_##flavor##_C3IMR, 3, srctype, dsttype,              \
                       CV_SQR, CV_MUL, CV_DEFINE_MASK, CV_PREPARE_FLT_MASK )           \
                                                                                       \
ICV_DEF_ACCPRODMASK_FUNC( icvAddProduct_##flavor##_C1IMR, 1, srctype, dsttype,         \
                           CV_NOP, CV_MULMASK, CV_DEFINE_MASK, CV_PREPARE_FLT_MASK )   \
ICV_DEF_ACCPRODMASK_FUNC( icvAddProduct_##flavor##_C3IMR, 3, srctype, dsttype,         \
                           CV_NOP, CV_MUL, CV_DEFINE_MASK, CV_PREPARE_FLT_MASK )       \
                                                                                       \
ICV_DEF_ACCWEIGHTMASK_FUNC( icvAddWeighted_##flavor##_C1IMR, 1, srctype, dsttype,      \
                             CV_NOP, CV_MULMASK, CV_DEFINE_ALPHA_MASK,                 \
                             CV_PREPARE_FLT_MASK )                                     \
ICV_DEF_ACCWEIGHTMASK_FUNC( icvAddWeighted_##flavor##_C3IMR, 3, srctype, dsttype,      \
                             CV_NOP, CV_MUL, CV_DEFINE_ALPHA_MASK,                     \
                             CV_PREPARE_FLT_MASK )

#define ICV_DECLARE_ACC_FUNC( flavor, srctype, dsttype )                    \
    IPCVAPI( CvStatus, icvAdd_##flavor##_C1IR, ( const srctype*, int,       \
                                                 dsttype*, int, CvSize ))

ICV_DECLARE_ACC_FUNC( 8u64f, uchar, double )
ICV_DECLARE_ACC_FUNC( 8s64f, char, double )
ICV_DECLARE_ACC_FUNC( 16s32f, short, float )
ICV_DECLARE_ACC_FUNC( 16s64f, short, double )
ICV_DECLARE_ACC_FUNC( 32f64f, float, double )
ICV_DECLARE_ACC_FUNC( 64f, double, double )


ICV_DEF_ACC_FUNC( icvAdd_8u32f_C1IR, uchar, float, CV_8TO32F )
ICV_DEF_ACC_FUNC( icvAdd_8u64f_C1IR, uchar, double, CV_8TO32F )
ICV_DEF_ACC_FUNC( icvAdd_8s32f_C1IR, char, float, CV_8TO32F )
ICV_DEF_ACC_FUNC( icvAdd_8s64f_C1IR, char, double, CV_8TO32F )
ICV_DEF_ACC_FUNC( icvAdd_16s32f_C1IR, short, float, CV_8TO32F )
ICV_DEF_ACC_FUNC( icvAdd_16s64f_C1IR, short, double, CV_8TO32F )
ICV_DEF_ACC_FUNC( icvAdd_32f_C1IR, float, float, CV_NOP )
ICV_DEF_ACC_FUNC( icvAdd_32f64f_C1IR, float, double, CV_NOP )
ICV_DEF_ACC_FUNC( icvAdd_64f_C1IR, double, double, CV_NOP )

ICV_DEF_ACC_ALL( 8u32f, uchar, float )
ICV_DEF_ACC_ALL( 8s32f, char, float )
ICV_DEF_ACC_ALL_FLT( 32f, float, float )


void  icvInitAccTable( CvFuncTable* tabfl, CvFuncTable* tabdb,
                       CvBigFuncTable* masktab )
{
    tabfl->fn_2d[CV_8U] = (void*)icvAdd_8u32f_C1IR;
    tabfl->fn_2d[CV_8S] = (void*)icvAdd_8s32f_C1IR;
    tabfl->fn_2d[CV_16S] = (void*)icvAdd_16s32f_C1IR;
    tabfl->fn_2d[CV_32F] = (void*)icvAdd_32f_C1IR;

    tabdb->fn_2d[CV_8U] = (void*)icvAdd_8u64f_C1IR;
    tabdb->fn_2d[CV_8S] = (void*)icvAdd_8s64f_C1IR;
    tabdb->fn_2d[CV_16S] = (void*)icvAdd_16s64f_C1IR;
    tabdb->fn_2d[CV_32F] = (void*)icvAdd_32f64f_C1IR;
    tabdb->fn_2d[CV_64F] = (void*)icvAdd_64f_C1IR;

    if( masktab )
    {
        masktab->fn_2d[CV_8UC1] = (void*)icvAdd_8u32f_C1IMR;
        masktab->fn_2d[CV_8SC1] = (void*)icvAdd_8s32f_C1IMR;
        masktab->fn_2d[CV_32FC1] = (void*)icvAdd_32f_C1IMR;

        masktab->fn_2d[CV_8UC3] = (void*)icvAdd_8u32f_C3IMR;
        masktab->fn_2d[CV_8SC3] = (void*)icvAdd_8s32f_C3IMR;
        masktab->fn_2d[CV_32FC3] = (void*)icvAdd_32f_C3IMR;
    }
}


#define  ICV_DEF_INIT_ACC_TAB( FUNCNAME )                                           \
static  void  icvInit##FUNCNAME##Table( CvFuncTable* tab, CvBigFuncTable* masktab ) \
{                                                                                   \
    tab->fn_2d[CV_8U] = (void*)icv##FUNCNAME##_8u32f_C1IR;                          \
    tab->fn_2d[CV_8S] = (void*)icv##FUNCNAME##_8s32f_C1IR;                          \
    tab->fn_2d[CV_32F] = (void*)icv##FUNCNAME##_32f_C1IR;                           \
                                                                                    \
    masktab->fn_2d[CV_8UC1] = (void*)icv##FUNCNAME##_8u32f_C1IMR;                   \
    masktab->fn_2d[CV_8SC1] = (void*)icv##FUNCNAME##_8s32f_C1IMR;                   \
    masktab->fn_2d[CV_32FC1] = (void*)icv##FUNCNAME##_32f_C1IMR;                    \
                                                                                    \
    masktab->fn_2d[CV_8UC3] = (void*)icv##FUNCNAME##_8u32f_C3IMR;                   \
    masktab->fn_2d[CV_8SC3] = (void*)icv##FUNCNAME##_8s32f_C3IMR;                   \
    masktab->fn_2d[CV_32FC3] = (void*)icv##FUNCNAME##_32f_C3IMR;                    \
}


ICV_DEF_INIT_ACC_TAB( AddSquare )
ICV_DEF_INIT_ACC_TAB( AddProduct )
ICV_DEF_INIT_ACC_TAB( AddWeighted )


CV_IMPL void
cvAcc( const void* arr, void* sumarr, const void* maskarr )
{
    static CvFuncTable acc_tab[2];
    static CvBigFuncTable accmask_tab;
    static int inittab = 0;
    
    CV_FUNCNAME( "cvAcc" );

    __BEGIN__;

    int type, sumdepth;
    int mat_step, sum_step, mask_step = 0;
    CvSize size;
    CvMat stub, *mat = (CvMat*)arr;
    CvMat sumstub, *sum = (CvMat*)sumarr;
    CvMat maskstub, *mask = (CvMat*)maskarr;

    if( !inittab )
    {
        icvInitAccTable( &acc_tab[0], &acc_tab[1], &accmask_tab );
        inittab = 1;
    }

    if( !CV_IS_MAT( mat ) || !CV_IS_MAT( sum ))
    {
        int coi1 = 0, coi2 = 0;
        CV_CALL( mat = cvGetMat( mat, &stub, &coi1 ));
        CV_CALL( sum = cvGetMat( sum, &sumstub, &coi2 ));
        if( coi1 + coi2 != 0 )
            CV_ERROR( CV_BadCOI, "" );
    }

    if( !CV_ARE_CNS_EQ( mat, sum ))
        CV_ERROR( CV_StsUnmatchedFormats, "" );

    sumdepth = CV_MAT_DEPTH( sum->type );
    if( sumdepth != CV_32F && (maskarr != 0 || sumdepth != CV_64F))
        CV_ERROR( CV_BadDepth, "Bad accumulator type" );

    if( !CV_ARE_SIZES_EQ( mat, sum ))
        CV_ERROR( CV_StsUnmatchedSizes, "" );

    size = icvGetMatSize( mat );
    type = CV_MAT_TYPE( mat->type );

    mat_step = mat->step;
    sum_step = sum->step;

    if( !mask )
    {
        CvFunc2D_2A func=(CvFunc2D_2A)acc_tab[sumdepth==CV_64F].fn_2d[CV_MAT_DEPTH(type)];

        if( !func )
            CV_ERROR( CV_StsUnsupportedFormat, "Unsupported type combination" );

        size.width *= CV_MAT_CN(type);
        if( CV_IS_MAT_CONT( mat->type & sum->type ))
        {
            size.width *= size.height;
            mat_step = sum_step = CV_STUB_STEP;
            size.height = 1;
        }

        IPPI_CALL( func( mat->data.ptr, mat_step, sum->data.ptr, sum_step, size ));
    }
    else
    {
        CvFunc2D_3A func = (CvFunc2D_3A)accmask_tab.fn_2d[type];

        if( !func )
            CV_ERROR( CV_StsUnsupportedFormat, "" );

        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, "" );            

        mask_step = mask->step;

        if( CV_IS_MAT_CONT( mat->type & sum->type & mask->type ))
        {
            size.width *= size.height;
            mat_step = sum_step = mask_step = CV_STUB_STEP;
            size.height = 1;
        }

        IPPI_CALL( func( mat->data.ptr, mat_step, mask->data.ptr, mask_step,
                         sum->data.ptr, sum_step, size ));
    }

    __END__;
}


CV_IMPL void
cvSquareAcc( const void* arr, void* sq_sum, const void* maskarr )
{
    static CvFuncTable acc_tab;
    static CvBigFuncTable accmask_tab;
    static int inittab = 0;
    
    CV_FUNCNAME( "cvSquareAcc" );

    __BEGIN__;

    int coi1, coi2;
    int type;
    int mat_step, sum_step, mask_step = 0;
    CvSize size;
    CvMat stub, *mat = (CvMat*)arr;
    CvMat sumstub, *sum = (CvMat*)sq_sum;
    CvMat maskstub, *mask = (CvMat*)maskarr;

    if( !inittab )
    {
        icvInitAddSquareTable( &acc_tab, &accmask_tab );
        inittab = 1;
    }

    CV_CALL( mat = cvGetMat( mat, &stub, &coi1 ));
    CV_CALL( sum = cvGetMat( sum, &sumstub, &coi2 ));

    if( coi1 != 0 || coi2 != 0 )
        CV_ERROR( CV_BadCOI, "" );

    if( !CV_ARE_CNS_EQ( mat, sum ))
        CV_ERROR( CV_StsUnmatchedFormats, "" );

    if( CV_MAT_DEPTH( sum->type ) != CV_32F )
        CV_ERROR( CV_BadDepth, "" );

    if( !CV_ARE_SIZES_EQ( mat, sum ))
        CV_ERROR( CV_StsUnmatchedSizes, "" );

    size = icvGetMatSize( mat );
    type = CV_MAT_TYPE( mat->type );

    mat_step = mat->step;
    sum_step = sum->step;

    if( !mask )
    {
        CvFunc2D_2A func = (CvFunc2D_2A)acc_tab.fn_2d[CV_MAT_DEPTH(type)];

        if( !func )