-/*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 "_cv.h"
-
-/****************************************************************************************/
-
-/* lightweight convolution with 3x3 kernel */
-void icvSepConvSmall3_32f( float* src, int src_step, float* dst, int dst_step,
- CvSize src_size, const float* kx, const float* ky, float* buffer )
-{
- int dst_width, buffer_step = 0;
- int x, y;
-
- assert( src && dst && src_size.width > 2 && src_size.height > 2 &&
- (src_step & 3) == 0 && (dst_step & 3) == 0 &&
- (kx || ky) && (buffer || !kx || !ky));
-
- src_step /= sizeof(src[0]);
- dst_step /= sizeof(dst[0]);
-
- dst_width = src_size.width - 2;
-
- if( !kx )
- {
- /* set vars, so that vertical convolution
- will write results into destination ROI and
- horizontal convolution won't run */
- src_size.width = dst_width;
- buffer_step = dst_step;
- buffer = dst;
- dst_width = 0;
- }
-
- assert( src_step >= src_size.width && dst_step >= dst_width );
-
- src_size.height -= 3;
- if( !ky )
- {
- /* set vars, so that vertical convolution won't run and
- horizontal convolution will write results into destination ROI */
- src_size.height += 3;
- buffer_step = src_step;
- buffer = src;
- src_size.width = 0;
- }
-
- for( y = 0; y <= src_size.height; y++, src += src_step,
- dst += dst_step,
- buffer += buffer_step )
- {
- float* src2 = src + src_step;
- float* src3 = src + src_step*2;
- for( x = 0; x < src_size.width; x++ )
- {
- buffer[x] = (float)(ky[0]*src[x] + ky[1]*src2[x] + ky[2]*src3[x]);
- }
-
- for( x = 0; x < dst_width; x++ )
- {
- dst[x] = (float)(kx[0]*buffer[x] + kx[1]*buffer[x+1] + kx[2]*buffer[x+2]);
- }
- }
-}
-
-
-/****************************************************************************************\
- Sobel & Scharr Derivative Filters
-\****************************************************************************************/
-
-/////////////////////////////// Old IPP derivative filters ///////////////////////////////
-// still used in corner detectors (see cvcorner.cpp)
-
-icvFilterSobelVert_8u16s_C1R_t icvFilterSobelVert_8u16s_C1R_p = 0;
-icvFilterSobelHoriz_8u16s_C1R_t icvFilterSobelHoriz_8u16s_C1R_p = 0;
-icvFilterSobelVertSecond_8u16s_C1R_t icvFilterSobelVertSecond_8u16s_C1R_p = 0;
-icvFilterSobelHorizSecond_8u16s_C1R_t icvFilterSobelHorizSecond_8u16s_C1R_p = 0;
-icvFilterSobelCross_8u16s_C1R_t icvFilterSobelCross_8u16s_C1R_p = 0;
-
-icvFilterSobelVert_32f_C1R_t icvFilterSobelVert_32f_C1R_p = 0;
-icvFilterSobelHoriz_32f_C1R_t icvFilterSobelHoriz_32f_C1R_p = 0;
-icvFilterSobelVertSecond_32f_C1R_t icvFilterSobelVertSecond_32f_C1R_p = 0;
-icvFilterSobelHorizSecond_32f_C1R_t icvFilterSobelHorizSecond_32f_C1R_p = 0;
-icvFilterSobelCross_32f_C1R_t icvFilterSobelCross_32f_C1R_p = 0;
-
-icvFilterScharrVert_8u16s_C1R_t icvFilterScharrVert_8u16s_C1R_p = 0;
-icvFilterScharrHoriz_8u16s_C1R_t icvFilterScharrHoriz_8u16s_C1R_p = 0;
-icvFilterScharrVert_32f_C1R_t icvFilterScharrVert_32f_C1R_p = 0;
-icvFilterScharrHoriz_32f_C1R_t icvFilterScharrHoriz_32f_C1R_p = 0;
-
-///////////////////////////////// New IPP derivative filters /////////////////////////////
-
-#define IPCV_FILTER_PTRS( name ) \
-icvFilter##name##GetBufSize_8u16s_C1R_t \
- icvFilter##name##GetBufSize_8u16s_C1R_p = 0; \
-icvFilter##name##Border_8u16s_C1R_t \
- icvFilter##name##Border_8u16s_C1R_p = 0; \
-icvFilter##name##GetBufSize_32f_C1R_t \
- icvFilter##name##GetBufSize_32f_C1R_p = 0; \
-icvFilter##name##Border_32f_C1R_t \
- icvFilter##name##Border_32f_C1R_p = 0;
-
-IPCV_FILTER_PTRS( ScharrHoriz )
-IPCV_FILTER_PTRS( ScharrVert )
-IPCV_FILTER_PTRS( SobelHoriz )
-IPCV_FILTER_PTRS( SobelNegVert )
-IPCV_FILTER_PTRS( SobelHorizSecond )
-IPCV_FILTER_PTRS( SobelVertSecond )
-IPCV_FILTER_PTRS( SobelCross )
-IPCV_FILTER_PTRS( Laplacian )
-
-typedef CvStatus (CV_STDCALL * CvDeriv3x3GetBufSizeIPPFunc)
- ( CvSize roi, int* bufsize );
-
-typedef CvStatus (CV_STDCALL * CvDerivGetBufSizeIPPFunc)
- ( CvSize roi, int masksize, int* bufsize );
-
-typedef CvStatus (CV_STDCALL * CvDeriv3x3IPPFunc_8u )
- ( const void* src, int srcstep, void* dst, int dststep,
- CvSize size, int bordertype, uchar bordervalue, void* buffer );
-
-typedef CvStatus (CV_STDCALL * CvDeriv3x3IPPFunc_32f )
- ( const void* src, int srcstep, void* dst, int dststep,
- CvSize size, int bordertype, float bordervalue, void* buffer );
-
-typedef CvStatus (CV_STDCALL * CvDerivIPPFunc_8u )
- ( const void* src, int srcstep, void* dst, int dststep,
- CvSize size, int masksize, int bordertype,
- uchar bordervalue, void* buffer );
-
-typedef CvStatus (CV_STDCALL * CvDerivIPPFunc_32f )
- ( const void* src, int srcstep, void* dst, int dststep,
- CvSize size, int masksize, int bordertype,
- float bordervalue, void* buffer );
-
-//////////////////////////////////////////////////////////////////////////////////////////
-
-CV_IMPL void
-cvSobel( const void* srcarr, void* dstarr, int dx, int dy, int aperture_size )
-{
- CvSepFilter filter;
- void* buffer = 0;
- int local_alloc = 0;
-
- CV_FUNCNAME( "cvSobel" );
-
- __BEGIN__;
-
- int origin = 0;
- int src_type, dst_type;
- CvMat srcstub, *src = (CvMat*)srcarr;
- CvMat dststub, *dst = (CvMat*)dstarr;
-
- if( !CV_IS_MAT(src) )
- CV_CALL( src = cvGetMat( src, &srcstub ));
- if( !CV_IS_MAT(dst) )
- CV_CALL( dst = cvGetMat( dst, &dststub ));
-
- if( CV_IS_IMAGE_HDR( srcarr ))
- origin = ((IplImage*)srcarr)->origin;
-
- src_type = CV_MAT_TYPE( src->type );
- dst_type = CV_MAT_TYPE( dst->type );
-
- if( !CV_ARE_SIZES_EQ( src, dst ))
- CV_ERROR( CV_StsBadArg, "src and dst have different sizes" );
-
- if( ((aperture_size == CV_SCHARR || aperture_size == 3 || aperture_size == 5) &&
- dx <= 2 && dy <= 2 && dx + dy <= 2 && icvFilterSobelNegVertBorder_8u16s_C1R_p) &&
- (src_type == CV_8UC1 && dst_type == CV_16SC1 ||
- src_type == CV_32FC1 && dst_type == CV_32FC1) )
- {
- CvDerivGetBufSizeIPPFunc ipp_sobel_getbufsize_func = 0;
- CvDerivIPPFunc_8u ipp_sobel_func_8u = 0;
- CvDerivIPPFunc_32f ipp_sobel_func_32f = 0;
-
- CvDeriv3x3GetBufSizeIPPFunc ipp_scharr_getbufsize_func = 0;
- CvDeriv3x3IPPFunc_8u ipp_scharr_func_8u = 0;
- CvDeriv3x3IPPFunc_32f ipp_scharr_func_32f = 0;
-
- if( aperture_size == CV_SCHARR )
- {
- if( dx == 1 && dy == 0 )
- {
- if( src_type == CV_8U )
- ipp_scharr_func_8u = icvFilterScharrVertBorder_8u16s_C1R_p,
- ipp_scharr_getbufsize_func = icvFilterScharrVertGetBufSize_8u16s_C1R_p;
- else
- ipp_scharr_func_32f = icvFilterScharrVertBorder_32f_C1R_p,
- ipp_scharr_getbufsize_func = icvFilterScharrVertGetBufSize_32f_C1R_p;
- }
- else if( dx == 0 && dy == 1 )
- {
- if( src_type == CV_8U )
- ipp_scharr_func_8u = icvFilterScharrHorizBorder_8u16s_C1R_p,
- ipp_scharr_getbufsize_func = icvFilterScharrHorizGetBufSize_8u16s_C1R_p;
- else
- ipp_scharr_func_32f = icvFilterScharrHorizBorder_32f_C1R_p,
- ipp_scharr_getbufsize_func = icvFilterScharrHorizGetBufSize_32f_C1R_p;
- }
- else
- CV_ERROR( CV_StsBadArg, "Scharr filter can only be used to compute 1st image derivatives" );
- }
- else
- {
- if( dx == 1 && dy == 0 )
- {
- if( src_type == CV_8U )
- ipp_sobel_func_8u = icvFilterSobelNegVertBorder_8u16s_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelNegVertGetBufSize_8u16s_C1R_p;
- else
- ipp_sobel_func_32f = icvFilterSobelNegVertBorder_32f_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelNegVertGetBufSize_32f_C1R_p;
- }
- else if( dx == 0 && dy == 1 )
- {
- if( src_type == CV_8U )
- ipp_sobel_func_8u = icvFilterSobelHorizBorder_8u16s_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelHorizGetBufSize_8u16s_C1R_p;
- else
- ipp_sobel_func_32f = icvFilterSobelHorizBorder_32f_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelHorizGetBufSize_32f_C1R_p;
- }
- else if( dx == 2 && dy == 0 )
- {
- if( src_type == CV_8U )
- ipp_sobel_func_8u = icvFilterSobelVertSecondBorder_8u16s_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelVertSecondGetBufSize_8u16s_C1R_p;
- else
- ipp_sobel_func_32f = icvFilterSobelVertSecondBorder_32f_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelVertSecondGetBufSize_32f_C1R_p;
- }
- else if( dx == 0 && dy == 2 )
- {
- if( src_type == CV_8U )
- ipp_sobel_func_8u = icvFilterSobelHorizSecondBorder_8u16s_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelHorizSecondGetBufSize_8u16s_C1R_p;
- else
- ipp_sobel_func_32f = icvFilterSobelHorizSecondBorder_32f_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelHorizSecondGetBufSize_32f_C1R_p;
- }
- else if( dx == 1 && dy == 1 )
- {
- if( src_type == CV_8U )
- ipp_sobel_func_8u = icvFilterSobelCrossBorder_8u16s_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelCrossGetBufSize_8u16s_C1R_p;
- else
- ipp_sobel_func_32f = icvFilterSobelCrossBorder_32f_C1R_p,
- ipp_sobel_getbufsize_func = icvFilterSobelCrossGetBufSize_32f_C1R_p;
- }
- }
-
- if( (ipp_sobel_func_8u || ipp_sobel_func_32f) && ipp_sobel_getbufsize_func ||
- (ipp_scharr_func_8u || ipp_scharr_func_32f) && ipp_scharr_getbufsize_func )
- {
- int bufsize = 0, masksize = aperture_size == 3 ? 33 : 55;
- CvSize size = cvGetMatSize( src );
- uchar* src_ptr = src->data.ptr;
- uchar* dst_ptr = dst->data.ptr;
- int src_step = src->step ? src->step : CV_STUB_STEP;
- int dst_step = dst->step ? dst->step : CV_STUB_STEP;
- const int bordertype = 1; // replication border
- CvStatus status;
-
- status = ipp_sobel_getbufsize_func ?
- ipp_sobel_getbufsize_func( size, masksize, &bufsize ) :
- ipp_scharr_getbufsize_func( size, &bufsize );
-
- if( status >= 0 )
- {
- if( bufsize <= CV_MAX_LOCAL_SIZE )
- {
- buffer = cvStackAlloc( bufsize );
- local_alloc = 1;
- }
- else
- CV_CALL( buffer = cvAlloc( bufsize ));
-
- status =
- ipp_sobel_func_8u ? ipp_sobel_func_8u( src_ptr, src_step, dst_ptr, dst_step,
- size, masksize, bordertype, 0, buffer ) :
- ipp_sobel_func_32f ? ipp_sobel_func_32f( src_ptr, src_step, dst_ptr, dst_step,
- size, masksize, bordertype, 0, buffer ) :
- ipp_scharr_func_8u ? ipp_scharr_func_8u( src_ptr, src_step, dst_ptr, dst_step,
- size, bordertype, 0, buffer ) :
- ipp_scharr_func_32f ? ipp_scharr_func_32f( src_ptr, src_step, dst_ptr, dst_step,
- size, bordertype, 0, buffer ) :
- CV_NOTDEFINED_ERR;
- }
-
- if( status >= 0 &&
- (dx == 0 && dy == 1 && origin || dx == 1 && dy == 1 && !origin)) // negate the output
- cvSubRS( dst, cvScalarAll(0), dst );
-
- if( status >= 0 )
- EXIT;
- }
- }
-
- CV_CALL( filter.init_deriv( src->cols, src_type, dst_type, dx, dy,
- aperture_size, origin ? CvSepFilter::FLIP_KERNEL : 0));
- CV_CALL( filter.process( src, dst ));
-
- __END__;
-
- if( buffer && !local_alloc )
- cvFree( &buffer );
-}
-
-
-/****************************************************************************************\
- Laplacian Filter
-\****************************************************************************************/
-
-static void icvLaplaceRow_8u32s( const uchar* src, int* dst, void* params );
-static void icvLaplaceRow_8u32f( const uchar* src, float* dst, void* params );
-static void icvLaplaceRow_32f( const float* src, float* dst, void* params );
-static void icvLaplaceCol_32s16s( const int** src, short* dst, int dst_step,
- int count, void* params );
-static void icvLaplaceCol_32f( const float** src, float* dst, int dst_step,
- int count, void* params );
-
-CvLaplaceFilter::CvLaplaceFilter()
-{
- normalized = basic_laplacian = false;
-}
-
-
-CvLaplaceFilter::CvLaplaceFilter( int _max_width, int _src_type, int _dst_type, bool _normalized,
- int _ksize, int _border_mode, CvScalar _border_value )
-{
- normalized = basic_laplacian = false;
- init( _max_width, _src_type, _dst_type, _normalized, _ksize, _border_mode, _border_value );
-}
-
-
-CvLaplaceFilter::~CvLaplaceFilter()
-{
- clear();
-}
-
-
-void CvLaplaceFilter::get_work_params()
-{
- int min_rows = max_ky*2 + 3, rows = MAX(min_rows,10), row_sz;
- int width = max_width, trow_sz = 0;
- int dst_depth = CV_MAT_DEPTH(dst_type);
- int work_depth = dst_depth < CV_32F ? CV_32S : CV_32F;
- work_type = CV_MAKETYPE( work_depth, CV_MAT_CN(dst_type)*2 );
- trow_sz = cvAlign( (max_width + ksize.width - 1)*CV_ELEM_SIZE(src_type), ALIGN );
- row_sz = cvAlign( width*CV_ELEM_SIZE(work_type), ALIGN );
- buf_size = rows*row_sz;
- buf_size = MIN( buf_size, 1 << 16 );
- buf_size = MAX( buf_size, min_rows*row_sz );
- max_rows = (buf_size/row_sz)*3 + max_ky*2 + 8;
- buf_size += trow_sz;
-}
-
-
-void CvLaplaceFilter::init( int _max_width, int _src_type, int _dst_type, bool _normalized,
- int _ksize0, int _border_mode, CvScalar _border_value )
-{
- CvMat *kx = 0, *ky = 0;
-
- CV_FUNCNAME( "CvLaplaceFilter::init" );
-
- __BEGIN__;
-
- int src_depth = CV_MAT_DEPTH(_src_type), dst_depth = CV_MAT_DEPTH(_dst_type);
- int _ksize = MAX( _ksize0, 3 );
-
- normalized = _normalized;
- basic_laplacian = _ksize0 == 1;
-
- if( (src_depth != CV_8U || dst_depth != CV_16S && dst_depth != CV_32F) &&
- (src_depth != CV_32F || dst_depth != CV_32F) ||
- CV_MAT_CN(_src_type) != CV_MAT_CN(_dst_type) )
- CV_ERROR( CV_StsUnmatchedFormats,
- "Laplacian can either transform 8u->16s, or 8u->32f, or 32f->32f.\n"
- "The channel number must be the same." );
-
- if( _ksize < 1 || _ksize > CV_MAX_SOBEL_KSIZE || _ksize % 2 == 0 )
- CV_ERROR( CV_StsOutOfRange, "kernel size must be within 1..7 and odd" );
-
- CV_CALL( kx = cvCreateMat( 1, _ksize, CV_32F ));
- CV_CALL( ky = cvCreateMat( 1, _ksize, CV_32F ));
-
- CvSepFilter::init_sobel_kernel( kx, ky, 2, 0, 0 );
- CvSepFilter::init( _max_width, _src_type, _dst_type, kx, ky,
- cvPoint(-1,-1), _border_mode, _border_value );
-
- x_func = 0;
- y_func = 0;
-
- if( src_depth == CV_8U )
- {
- if( dst_depth == CV_16S )
- {
- x_func = (CvRowFilterFunc)icvLaplaceRow_8u32s;
- y_func = (CvColumnFilterFunc)icvLaplaceCol_32s16s;
- }
- else if( dst_depth == CV_32F )
- {
- x_func = (CvRowFilterFunc)icvLaplaceRow_8u32f;
- y_func = (CvColumnFilterFunc)icvLaplaceCol_32f;
- }
- }
- else if( src_depth == CV_32F )
- {
- if( dst_depth == CV_32F )
- {
- x_func = (CvRowFilterFunc)icvLaplaceRow_32f;
- y_func = (CvColumnFilterFunc)icvLaplaceCol_32f;
- }
- }
-
- if( !x_func || !y_func )
- CV_ERROR( CV_StsUnsupportedFormat, "" );
-
- __END__;
-
- cvReleaseMat( &kx );
- cvReleaseMat( &ky );
-}
-
-
-void CvLaplaceFilter::init( int _max_width, int _src_type, int _dst_type,
- bool _is_separable, CvSize _ksize,
- CvPoint _anchor, int _border_mode,
- CvScalar _border_value )
-{
- CvSepFilter::init( _max_width, _src_type, _dst_type, _is_separable,
- _ksize, _anchor, _border_mode, _border_value );
-}
-
-
-void CvLaplaceFilter::init( int _max_width, int _src_type, int _dst_type,
- const CvMat* _kx, const CvMat* _ky,
- CvPoint _anchor, int _border_mode,
- CvScalar _border_value )
-{
- CvSepFilter::init( _max_width, _src_type, _dst_type, _kx, _ky,
- _anchor, _border_mode, _border_value );
-}
-
-
-#define ICV_LAPLACE_ROW( flavor, srctype, dsttype, load_macro ) \
-static void \
-icvLaplaceRow_##flavor( const srctype* src, dsttype* dst, void* params )\
-{ \
- const CvLaplaceFilter* state = (const CvLaplaceFilter*)params; \
- const CvMat* _kx = state->get_x_kernel(); \
- const CvMat* _ky = state->get_y_kernel(); \
- const dsttype* kx = (dsttype*)_kx->data.ptr; \
- const dsttype* ky = (dsttype*)_ky->data.ptr; \
- int ksize = _kx->cols + _kx->rows - 1; \
- int i = 0, j, k, width = state->get_width(); \
- int cn = CV_MAT_CN(state->get_src_type()); \
- int ksize2 = ksize/2, ksize2n = ksize2*cn; \
- const srctype* s = src + ksize2n; \
- bool basic_laplacian = state->is_basic_laplacian(); \
- \
- kx += ksize2; \
- ky += ksize2; \
- width *= cn; \
- \
- if( basic_laplacian ) \
- for( i = 0; i < width; i++ ) \
- { \
- dsttype s0 = load_macro(s[i]); \
- dsttype s1 = (dsttype)(s[i-cn] - s0*2 + s[i+cn]); \
- dst[i] = s0; dst[i+width] = s1; \
- } \
- else if( ksize == 3 ) \
- for( i = 0; i < width; i++ ) \
- { \
- dsttype s0 = (dsttype)(s[i-cn] + s[i]*2 + s[i+cn]); \
- dsttype s1 = (dsttype)(s[i-cn] - s[i]*2 + s[i+cn]); \
- dst[i] = s0; dst[i+width] = s1; \
- } \
- else if( ksize == 5 ) \
- for( i = 0; i < width; i++ ) \
- { \
- dsttype s0 = (dsttype)(s[i-2*cn]+(s[i-cn]+s[i+cn])*4+s[i]*6+s[i+2*cn]);\
- dsttype s1 = (dsttype)(s[i-2*cn]-s[i]*2+s[i+2*cn]); \
- dst[i] = s0; dst[i+width] = s1; \
- } \
- else \
- for( i = 0; i < width; i++, s++ ) \
- { \
- dsttype s0 = ky[0]*load_macro(s[0]), s1 = kx[0]*load_macro(s[0]);\
- for( k = 1, j = cn; k <= ksize2; k++, j += cn ) \
- { \
- dsttype t = load_macro(s[j] + s[-j]); \
- s0 += ky[k]*t; s1 += kx[k]*t; \
- } \
- dst[i] = s0; dst[i+width] = s1; \
- } \
-}
-
-ICV_LAPLACE_ROW( 8u32s, uchar, int, CV_NOP )
-ICV_LAPLACE_ROW( 8u32f, uchar, float, CV_8TO32F )
-ICV_LAPLACE_ROW( 32f, float, float, CV_NOP )
-
-static void
-icvLaplaceCol_32s16s( const int** src, short* dst,
- int dst_step, int count, void* params )
-{
- const CvLaplaceFilter* state = (const CvLaplaceFilter*)params;
- const CvMat* _kx = state->get_x_kernel();
- const CvMat* _ky = state->get_y_kernel();
- const int* kx = (const int*)_kx->data.ptr;
- const int* ky = (const int*)_ky->data.ptr;
- int ksize = _kx->cols + _kx->rows - 1, ksize2 = ksize/2;
- int i = 0, k, width = state->get_width();
- int cn = CV_MAT_CN(state->get_src_type());
- bool basic_laplacian = state->is_basic_laplacian();
- bool normalized = state->is_normalized();
- int shift = ksize - 1, delta = (1 << shift) >> 1;
-
- width *= cn;
- src += ksize2;
- kx += ksize2;
- ky += ksize2;
- dst_step /= sizeof(dst[0]);
-
- if( basic_laplacian || !normalized )
- {
- normalized = false;
- shift = delta = 0;
- }
-
- for( ; count--; dst += dst_step, src++ )
- {
- if( ksize == 3 )
- {
- const int *src0 = src[-1], *src1 = src[0], *src2 = src[1];
- if( basic_laplacian )
- {
- for( i = 0; i <= width - 2; i += 2 )
- {
- int s0 = src0[i] - src1[i]*2 + src2[i] + src1[i+width];
- int s1 = src0[i+1] - src1[i+1]*2 + src2[i+1] + src1[i+width+1];
- dst[i] = (short)s0; dst[i+1] = (short)s1;
- }
-
- for( ; i < width; i++ )
- dst[i] = (short)(src0[i] - src1[i]*2 + src2[i] + src1[i+width]);
- }
- else if( !normalized )
- for( i = 0; i <= width - 2; i += 2 )
- {
- int s0 = src0[i] - src1[i]*2 + src2[i] +
- src0[i+width] + src1[i+width]*2 + src2[i+width];
- int s1 = src0[i+1] - src1[i+1]*2 + src2[i+1] +
- src0[i+width+1] + src1[i+width+1]*2 + src2[i+width+1];
- dst[i] = (short)s0; dst[i+1] = (short)s1;
- }
- else
- for( i = 0; i <= width - 2; i += 2 )
- {
- int s0 = CV_DESCALE(src0[i] - src1[i]*2 + src2[i] +
- src0[i+width] + src1[i+width]*2 + src2[i+width], 2);
- int s1 = CV_DESCALE(src0[i+1] - src1[i+1]*2 + src2[i+1] +
- src0[i+width+1] + src1[i+width+1]*2 + src2[i+width+1],2);
- dst[i] = (short)s0; dst[i+1] = (short)s1;
- }
- }
- else if( ksize == 5 )
- {
- const int *src0 = src[-2], *src1 = src[-1], *src2 = src[0], *src3 = src[1], *src4 = src[2];
-
- if( !normalized )
- for( i = 0; i <= width - 2; i += 2 )
- {
- int s0 = src0[i] - src2[i]*2 + src4[i] + src0[i+width] + src4[i+width] +
- (src1[i+width] + src3[i+width])*4 + src2[i+width]*6;
- int s1 = src0[i+1] - src2[i+1]*2 + src4[i+1] + src0[i+width+1] +
- src4[i+width+1] + (src1[i+width+1] + src3[i+width+1])*4 +
- src2[i+width+1]*6;
- dst[i] = (short)s0; dst[i+1] = (short)s1;
- }
- else
- for( i = 0; i <= width - 2; i += 2 )
- {
- int s0 = CV_DESCALE(src0[i] - src2[i]*2 + src4[i] +
- src0[i+width] + src4[i+width] +
- (src1[i+width] + src3[i+width])*4 + src2[i+width]*6, 4);
- int s1 = CV_DESCALE(src0[i+1] - src2[i+1]*2 + src4[i+1] +
- src0[i+width+1] + src4[i+width+1] +
- (src1[i+width+1] + src3[i+width+1])*4 + src2[i+width+1]*6, 4);
- dst[i] = (short)s0; dst[i+1] = (short)s1;
- }
- }
- else
- {
- if( !normalized )
- for( i = 0; i <= width - 2; i += 2 )
- {
- int s0 = kx[0]*src[0][i] + ky[0]*src[0][i+width];
- int s1 = kx[0]*src[0][i+1] + ky[0]*src[0][i+width+1];
-
- for( k = 1; k <= ksize2; k++ )
- {
- const int* src1 = src[k] + i, *src2 = src[-k] + i;
- int fx = kx[k], fy = ky[k];
- s0 += fx*(src1[0] + src2[0]) + fy*(src1[width] + src2[width]);
- s1 += fx*(src1[1] + src2[1]) + fy*(src1[width+1] + src2[width+1]);
- }
-
- dst[i] = CV_CAST_16S(s0); dst[i+1] = CV_CAST_16S(s1);
- }
- else
- for( i = 0; i <= width - 2; i += 2 )
- {
- int s0 = kx[0]*src[0][i] + ky[0]*src[0][i+width];
- int s1 = kx[0]*src[0][i+1] + ky[0]*src[0][i+width+1];
-
- for( k = 1; k <= ksize2; k++ )
- {
- const int* src1 = src[k] + i, *src2 = src[-k] + i;
- int fx = kx[k], fy = ky[k];
- s0 += fx*(src1[0] + src2[0]) + fy*(src1[width] + src2[width]);
- s1 += fx*(src1[1] + src2[1]) + fy*(src1[width+1] + src2[width+1]);
- }
-
- s0 = CV_DESCALE( s0, shift ); s1 = CV_DESCALE( s1, shift );
- dst[i] = (short)s0; dst[i+1] = (short)s1;
- }
- }
-
- for( ; i < width; i++ )
- {
- int s0 = kx[0]*src[0][i] + ky[0]*src[0][i+width];
- for( k = 1; k <= ksize2; k++ )
- {
- const int* src1 = src[k] + i, *src2 = src[-k] + i;
- s0 += kx[k]*(src1[0] + src2[0]) + ky[k]*(src1[width] + src2[width]);
- }
- s0 = (s0 + delta) >> shift;
- dst[i] = CV_CAST_16S(s0);
- }
- }
-}
-
-
-static void
-icvLaplaceCol_32f( const float** src, float* dst,
- int dst_step, int count, void* params )
-{
- const CvLaplaceFilter* state = (const CvLaplaceFilter*)params;
- const CvMat* _kx = state->get_x_kernel();
- const CvMat* _ky = state->get_y_kernel();
- const float* kx = (const float*)_kx->data.ptr;
- const float* ky = (const float*)_ky->data.ptr;
- int ksize = _kx->cols + _kx->rows - 1, ksize2 = ksize/2;
- int i = 0, k, width = state->get_width();
- int cn = CV_MAT_CN(state->get_src_type());
- bool basic_laplacian = state->is_basic_laplacian();
- bool normalized = state->is_normalized();
- float scale = 1.f/(1 << (ksize - 1));
-
- width *= cn;
- src += ksize2;
- kx += ksize2;
- ky += ksize2;
- dst_step /= sizeof(dst[0]);
-
- if( basic_laplacian || !normalized )
- {
- normalized = false;
- scale = 1.f;
- }
-
- for( ; count--; dst += dst_step, src++ )
- {
- if( ksize == 3 )
- {
- const float *src0 = src[-1], *src1 = src[0], *src2 = src[1];
- if( basic_laplacian )
- {
- for( i = 0; i <= width - 2; i += 2 )
- {
- float s0 = src0[i] - src1[i]*2 + src2[i] + src1[i+width];
- float s1 = src0[i+1] - src1[i+1]*2 + src2[i+1] + src1[i+width+1];
- dst[i] = s0; dst[i+1] = s1;
- }
-
- for( ; i < width; i++ )
- dst[i] = src0[i] - src1[i]*2 + src2[i] + src1[i+width];
- }
- else if( !normalized )
- for( i = 0; i <= width - 2; i += 2 )
- {
- float s0 = src0[i] - src1[i]*2 + src2[i] +
- src0[i+width] + src1[i+width]*2 + src2[i+width];
- float s1 = src0[i+1] - src1[i+1]*2 + src2[i+1] +
- src0[i+width+1] + src1[i+width+1]*2 + src2[i+width+1];
- dst[i] = s0; dst[i+1] = s1;
- }
- else
- for( i = 0; i <= width - 2; i += 2 )
- {
- float s0 = (src0[i] - src1[i]*2 + src2[i] +
- src0[i+width] + src1[i+width]*2 + src2[i+width])*scale;
- float s1 = (src0[i+1] - src1[i+1]*2 + src2[i+1] +
- src0[i+width+1] + src1[i+width+1]*2 + src2[i+width+1])*scale;
- dst[i] = s0; dst[i+1] = s1;
- }
- }
- else if( ksize == 5 )
- {
- const float *src0 = src[-2], *src1 = src[-1], *src2 = src[0], *src3 = src[1], *src4 = src[2];
- for( i = 0; i <= width - 2; i += 2 )
- {
- float s0 = (src0[i] - src2[i]*2 + src4[i] +
- src0[i+width] + src4[i+width] +
- (src1[i+width] + src3[i+width])*4 + src2[i+width]*6)*scale;
- float s1 = (src0[i+1] - src2[i+1]*2 + src4[i+1] +
- src0[i+width+1] + src4[i+width+1] +
- (src1[i+width+1] + src3[i+width+1])*4 + src2[i+width+1]*6)*scale;
- dst[i] = s0; dst[i+1] = s1;
- }
- }
- else
- {
- for( i = 0; i <= width - 2; i += 2 )
- {
- float s0 = kx[0]*src[0][i] + ky[0]*src[0][i+width];
- float s1 = kx[0]*src[0][i+1] + ky[0]*src[0][i+width+1];
-
- for( k = 1; k <= ksize2; k++ )
- {
- const float* src1 = src[k] + i, *src2 = src[-k] + i;
- float fx = kx[k], fy = ky[k];
- s0 += fx*(src1[0] + src2[0]) + fy*(src1[width] + src2[width]);
- s1 += fx*(src1[1] + src2[1]) + fy*(src1[width+1] + src2[width+1]);
- }
-
- s0 *= scale; s1 *= scale;
- dst[i] = s0; dst[i+1] = s1;
- }
- }
-
- for( ; i < width; i++ )
- {
- float s0 = kx[0]*src[0][i] + ky[0]*src[0][i+width];
- for( k = 1; k <= ksize2; k++ )
- {
- const float* src1 = src[k] + i, *src2 = src[-k] + i;
- s0 += kx[k]*(src1[0] + src2[0]) + ky[k]*(src1[width] + src2[width]);
- }
- dst[i] = s0*scale;
- }
- }
-}
-
-
-CV_IMPL void
-cvLaplace( const void* srcarr, void* dstarr, int aperture_size )
-{
- CvLaplaceFilter laplacian;
- void* buffer = 0;
- int local_alloc = 0;
-
- CV_FUNCNAME( "cvLaplace" );
-
- __BEGIN__;
-
- CvMat srcstub, *src = (CvMat*)srcarr;
- CvMat dststub, *dst = (CvMat*)dstarr;
- int src_type, dst_type;
-
- CV_CALL( src = cvGetMat( src, &srcstub ));
- CV_CALL( dst = cvGetMat( dst, &dststub ));
-
- src_type = CV_MAT_TYPE(src->type);
- dst_type = CV_MAT_TYPE(dst->type);
-
- if( aperture_size == 3 || aperture_size == 5 &&
- (src_type == CV_8UC1 && dst_type == CV_16SC1 ||
- src_type == CV_32FC1 && dst_type == CV_32FC1) &&
- (aperture_size == 3 || src_type < CV_32FC1) )
- {
- CvDerivGetBufSizeIPPFunc ipp_laplace_getbufsize_func = 0;
- CvDerivIPPFunc_8u ipp_laplace_func_8u = 0;
- CvDerivIPPFunc_32f ipp_laplace_func_32f = 0;
-
- if( src_type == CV_8U )
- ipp_laplace_func_8u = icvFilterLaplacianBorder_8u16s_C1R_p,
- ipp_laplace_getbufsize_func = icvFilterLaplacianGetBufSize_8u16s_C1R_p;
- else
- ipp_laplace_func_32f = icvFilterLaplacianBorder_32f_C1R_p,
- ipp_laplace_getbufsize_func = icvFilterLaplacianGetBufSize_32f_C1R_p;
-
- if( (ipp_laplace_func_8u || ipp_laplace_func_32f) && ipp_laplace_getbufsize_func )
- {
- int bufsize = 0, masksize = aperture_size == 3 ? 33 : 55;
- CvSize size = cvGetMatSize( src );
- uchar* src_ptr = src->data.ptr;
- uchar* dst_ptr = dst->data.ptr;
- int src_step = src->step ? src->step : CV_STUB_STEP;
- int dst_step = dst->step ? dst->step : CV_STUB_STEP;
- const int bordertype = 1; // replication border
- CvStatus status;
-
- status = ipp_laplace_getbufsize_func( size, masksize, &bufsize );
-
- if( status >= 0 )
- {
- if( bufsize <= CV_MAX_LOCAL_SIZE )
- {
- buffer = cvStackAlloc( bufsize );
- local_alloc = 1;
- }
- else
- CV_CALL( buffer = cvAlloc( bufsize ));
-
- status =
- ipp_laplace_func_8u ? ipp_laplace_func_8u( src_ptr, src_step, dst_ptr, dst_step,
- size, masksize, bordertype, 0, buffer ) :
- ipp_laplace_func_32f ? ipp_laplace_func_32f( src_ptr, src_step, dst_ptr, dst_step,
- size, masksize, bordertype, 0, buffer ) :
- CV_NOTDEFINED_ERR;
- }
-
- if( status >= 0 )
- EXIT;
- }
- }
-
- CV_CALL( laplacian.init( src->cols, src_type, dst_type,
- false, aperture_size ));
- CV_CALL( laplacian.process( src, dst ));
-
- __END__;
-
- if( buffer && !local_alloc )
- cvFree( &buffer );
-}
-
-/* End of file. */
projects / opencv / blobdiff