+++ /dev/null
-/*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"
-#include "_cvlist.h"
-
-#define halfPi ((float)(CV_PI*0.5))
-#define Pi ((float)CV_PI)
-#define a0 0 /*-4.172325e-7f*/ /*(-(float)0x7)/((float)0x1000000); */
-#define a1 1.000025f /*((float)0x1922253)/((float)0x1000000)*2/Pi; */
-#define a2 -2.652905e-4f /*(-(float)0x2ae6)/((float)0x1000000)*4/(Pi*Pi); */
-#define a3 -0.165624f /*(-(float)0xa45511)/((float)0x1000000)*8/(Pi*Pi*Pi); */
-#define a4 -1.964532e-3f /*(-(float)0x30fd3)/((float)0x1000000)*16/(Pi*Pi*Pi*Pi); */
-#define a5 1.02575e-2f /*((float)0x191cac)/((float)0x1000000)*32/(Pi*Pi*Pi*Pi*Pi); */
-#define a6 -9.580378e-4f /*(-(float)0x3af27)/((float)0x1000000)*64/(Pi*Pi*Pi*Pi*Pi*Pi); */
-
-#define _sin(x) ((((((a6*(x) + a5)*(x) + a4)*(x) + a3)*(x) + a2)*(x) + a1)*(x) + a0)
-#define _cos(x) _sin(halfPi - (x))
-
-/****************************************************************************************\
-* Classical Hough Transform *
-\****************************************************************************************/
-
-typedef struct CvLinePolar
-{
- float rho;
- float angle;
-}
-CvLinePolar;
-
-/*=====================================================================================*/
-
-#define hough_cmp_gt(l1,l2) (aux[l1] > aux[l2])
-
-static CV_IMPLEMENT_QSORT_EX( icvHoughSortDescent32s, int, hough_cmp_gt, const int* )
-
-/*
-Here image is an input raster;
-step is it's step; size characterizes it's ROI;
-rho and theta are discretization steps (in pixels and radians correspondingly).
-threshold is the minimum number of pixels in the feature for it
-to be a candidate for line. lines is the output
-array of (rho, theta) pairs. linesMax is the buffer size (number of pairs).
-Functions return the actual number of found lines.
-*/
-static void
-icvHoughLinesStandard( const CvMat* img, float rho, float theta,
- int threshold, CvSeq *lines, int linesMax )
-{
- int *accum = 0;
- int *sort_buf=0;
- float *tabSin = 0;
- float *tabCos = 0;
-
- CV_FUNCNAME( "icvHoughLinesStandard" );
-
- __BEGIN__;
-
- const uchar* image;
- int step, width, height;
- int numangle, numrho;
- int total = 0;
- float ang;
- int r, n;
- int i, j;
- float irho = 1 / rho;
- double scale;
-
- CV_ASSERT( CV_IS_MAT(img) && CV_MAT_TYPE(img->type) == CV_8UC1 );
-
- image = img->data.ptr;
- step = img->step;
- width = img->cols;
- height = img->rows;
-
- numangle = cvRound(CV_PI / theta);
- numrho = cvRound(((width + height) * 2 + 1) / rho);
-
- CV_CALL( accum = (int*)cvAlloc( sizeof(accum[0]) * (numangle+2) * (numrho+2) ));
- CV_CALL( sort_buf = (int*)cvAlloc( sizeof(accum[0]) * numangle * numrho ));
- CV_CALL( tabSin = (float*)cvAlloc( sizeof(tabSin[0]) * numangle ));
- CV_CALL( tabCos = (float*)cvAlloc( sizeof(tabCos[0]) * numangle ));
- memset( accum, 0, sizeof(accum[0]) * (numangle+2) * (numrho+2) );
-
- for( ang = 0, n = 0; n < numangle; ang += theta, n++ )
- {
- tabSin[n] = (float)(sin(ang) * irho);
- tabCos[n] = (float)(cos(ang) * irho);
- }
-
- // stage 1. fill accumulator
- for( i = 0; i < height; i++ )
- for( j = 0; j < width; j++ )
- {
- if( image[i * step + j] != 0 )
- for( n = 0; n < numangle; n++ )
- {
- r = cvRound( j * tabCos[n] + i * tabSin[n] );
- r += (numrho - 1) / 2;
- accum[(n+1) * (numrho+2) + r+1]++;
- }
- }
-
- // stage 2. find local maximums
- for( r = 0; r < numrho; r++ )
- for( n = 0; n < numangle; n++ )
- {
- int base = (n+1) * (numrho+2) + r+1;
- if( accum[base] > threshold &&
- accum[base] > accum[base - 1] && accum[base] >= accum[base + 1] &&
- accum[base] > accum[base - numrho - 2] && accum[base] >= accum[base + numrho + 2] )
- sort_buf[total++] = base;
- }
-
- // stage 3. sort the detected lines by accumulator value
- icvHoughSortDescent32s( sort_buf, total, accum );
-
- // stage 4. store the first min(total,linesMax) lines to the output buffer
- linesMax = MIN(linesMax, total);
- scale = 1./(numrho+2);
- for( i = 0; i < linesMax; i++ )
- {
- CvLinePolar line;
- int idx = sort_buf[i];
- int n = cvFloor(idx*scale) - 1;
- int r = idx - (n+1)*(numrho+2) - 1;
- line.rho = (r - (numrho - 1)*0.5f) * rho;
- line.angle = n * theta;
- cvSeqPush( lines, &line );
- }
-
- __END__;
-
- cvFree( &sort_buf );
- cvFree( &tabSin );
- cvFree( &tabCos );
- cvFree( &accum );
-}
-
-
-/****************************************************************************************\
-* Multi-Scale variant of Classical Hough Transform *
-\****************************************************************************************/
-
-#if defined _MSC_VER && _MSC_VER >= 1200
-#pragma warning( disable: 4714 )
-#endif
-
-//DECLARE_AND_IMPLEMENT_LIST( _index, h_ );
-IMPLEMENT_LIST( _index, h_ )
-
-static void
-icvHoughLinesSDiv( const CvMat* img,
- float rho, float theta, int threshold,
- int srn, int stn,
- CvSeq* lines, int linesMax )
-{
- uchar *caccum = 0;
- uchar *buffer = 0;
- float *sinTable = 0;
- int *x = 0;
- int *y = 0;
- _CVLIST *list = 0;
-
- CV_FUNCNAME( "icvHoughLinesSDiv" );
-
- __BEGIN__;
-
-#define _POINT(row, column)\
- (image_src[(row)*step+(column)])
-
- uchar *mcaccum = 0;
- int rn, tn; /* number of rho and theta discrete values */
- int index, i;
- int ri, ti, ti1, ti0;
- int row, col;
- float r, t; /* Current rho and theta */
- float rv; /* Some temporary rho value */
- float irho;
- float itheta;
- float srho, stheta;
- float isrho, istheta;
-
- const uchar* image_src;
- int w, h, step;
- int fn = 0;
- float xc, yc;
-
- const float d2r = (float)(Pi / 180);
- int sfn = srn * stn;
- int fi;
- int count;
- int cmax = 0;
-
- CVPOS pos;
- _index *pindex;
- _index vi;
-
- CV_ASSERT( CV_IS_MAT(img) && CV_MAT_TYPE(img->type) == CV_8UC1 );
- CV_ASSERT( linesMax > 0 && rho > 0 && theta > 0 );
-
- threshold = MIN( threshold, 255 );
-
- image_src = img->data.ptr;
- step = img->step;
- w = img->cols;
- h = img->rows;
-
- irho = 1 / rho;
- itheta = 1 / theta;
- srho = rho / srn;
- stheta = theta / stn;
- isrho = 1 / srho;
- istheta = 1 / stheta;
-
- rn = cvFloor( sqrt( (double)w * w + (double)h * h ) * irho );
- tn = cvFloor( 2 * Pi * itheta );
-
- list = h_create_list__index( linesMax < 1000 ? linesMax : 1000 );
- vi.value = threshold;
- vi.rho = -1;
- h_add_head__index( list, &vi );
-
- /* Precalculating sin */
- CV_CALL( sinTable = (float*)cvAlloc( 5 * tn * stn * sizeof( float )));
-
- for( index = 0; index < 5 * tn * stn; index++ )
- {
- sinTable[index] = (float)cos( stheta * index * 0.2f );
- }
-
- CV_CALL( caccum = (uchar*)cvAlloc( rn * tn * sizeof( caccum[0] )));
- memset( caccum, 0, rn * tn * sizeof( caccum[0] ));
-
- /* Counting all feature pixels */
- for( row = 0; row < h; row++ )
- for( col = 0; col < w; col++ )
- fn += _POINT( row, col ) != 0;
-
- CV_CALL( x = (int*)cvAlloc( fn * sizeof(x[0])));
- CV_CALL( y = (int*)cvAlloc( fn * sizeof(y[0])));
-
- /* Full Hough Transform (it's accumulator update part) */
- fi = 0;
- for( row = 0; row < h; row++ )
- {
- for( col = 0; col < w; col++ )
- {
- if( _POINT( row, col ))
- {
- int halftn;
- float r0;
- float scale_factor;
- int iprev = -1;
- float phi, phi1;
- float theta_it; /* Value of theta for iterating */
-
- /* Remember the feature point */
- x[fi] = col;
- y[fi] = row;
- fi++;
-
- yc = (float) row + 0.5f;
- xc = (float) col + 0.5f;
-
- /* Update the accumulator */
- t = (float) fabs( cvFastArctan( yc, xc ) * d2r );
- r = (float) sqrt( (double)xc * xc + (double)yc * yc );
- r0 = r * irho;
- ti0 = cvFloor( (t + Pi / 2) * itheta );
-
- caccum[ti0]++;
-
- theta_it = rho / r;
- theta_it = theta_it < theta ? theta_it : theta;
- scale_factor = theta_it * itheta;
- halftn = cvFloor( Pi / theta_it );
- for( ti1 = 1, phi = theta_it - halfPi, phi1 = (theta_it + t) * itheta;
- ti1 < halftn; ti1++, phi += theta_it, phi1 += scale_factor )
- {
- rv = r0 * _cos( phi );
- i = cvFloor( rv ) * tn;
- i += cvFloor( phi1 );
- assert( i >= 0 );
- assert( i < rn * tn );
- caccum[i] = (uchar) (caccum[i] + ((i ^ iprev) != 0));
- iprev = i;
- if( cmax < caccum[i] )
- cmax = caccum[i];
- }
- }
- }
- }
-
- /* Starting additional analysis */
- count = 0;
- for( ri = 0; ri < rn; ri++ )
- {
- for( ti = 0; ti < tn; ti++ )
- {
- if( caccum[ri * tn + ti > threshold] )
- {
- count++;
- }
- }
- }
-
- if( count * 100 > rn * tn )
- {
- icvHoughLinesStandard( img, rho, theta, threshold, lines, linesMax );
- EXIT;
- }
-
- CV_CALL( buffer = (uchar *) cvAlloc(srn * stn + 2));
- mcaccum = buffer + 1;
-
- count = 0;
- for( ri = 0; ri < rn; ri++ )
- {
- for( ti = 0; ti < tn; ti++ )
- {
- if( caccum[ri * tn + ti] > threshold )
- {
- count++;
- memset( mcaccum, 0, sfn * sizeof( uchar ));
-
- for( index = 0; index < fn; index++ )
- {
- int ti2;
- float r0;
-
- yc = (float) y[index] + 0.5f;
- xc = (float) x[index] + 0.5f;
-
- /* Update the accumulator */
- t = (float) fabs( cvFastArctan( yc, xc ) * d2r );
- r = (float) sqrt( (double)xc * xc + (double)yc * yc ) * isrho;
- ti0 = cvFloor( (t + Pi * 0.5f) * istheta );
- ti2 = (ti * stn - ti0) * 5;
- r0 = (float) ri *srn;
-
- for( ti1 = 0 /*, phi = ti*theta - Pi/2 - t */ ; ti1 < stn; ti1++, ti2 += 5
- /*phi += stheta */ )
- {
- /*rv = r*_cos(phi) - r0; */
- rv = r * sinTable[(int) (abs( ti2 ))] - r0;
- i = cvFloor( rv ) * stn + ti1;
-
- i = CV_IMAX( i, -1 );
- i = CV_IMIN( i, sfn );
- mcaccum[i]++;
- assert( i >= -1 );
- assert( i <= sfn );
- }
- }
-
- /* Find peaks in maccum... */
- for( index = 0; index < sfn; index++ )
- {
- i = 0;
- pos = h_get_tail_pos__index( list );
- if( h_get_prev__index( &pos )->value < mcaccum[index] )
- {
- vi.value = mcaccum[index];
- vi.rho = index / stn * srho + ri * rho;
- vi.theta = index % stn * stheta + ti * theta - halfPi;
- while( h_is_pos__index( pos ))
- {
- if( h_get__index( pos )->value > mcaccum[index] )
- {
- h_insert_after__index( list, pos, &vi );
- if( h_get_count__index( list ) > linesMax )
- {
- h_remove_tail__index( list );
- }
- break;
- }
- h_get_prev__index( &pos );
- }
- if( !h_is_pos__index( pos ))
- {
- h_add_head__index( list, &vi );
- if( h_get_count__index( list ) > linesMax )
- {
- h_remove_tail__index( list );
- }
- }
- }
- }
- }
- }
- }
-
- pos = h_get_head_pos__index( list );
- if( h_get_count__index( list ) == 1 )
- {
- if( h_get__index( pos )->rho < 0 )
- {
- h_clear_list__index( list );
- }
- }
- else
- {
- while( h_is_pos__index( pos ))
- {
- CvLinePolar line;
- pindex = h_get__index( pos );
- if( pindex->rho < 0 )
- {
- /* This should be the last element... */
- h_get_next__index( &pos );
- assert( !h_is_pos__index( pos ));
- break;
- }
- line.rho = pindex->rho;
- line.angle = pindex->theta;
- cvSeqPush( lines, &line );
-
- if( lines->total >= linesMax )
- EXIT;
- h_get_next__index( &pos );
- }
- }
-
- __END__;
-
- h_destroy_list__index( list );
- cvFree( &sinTable );
- cvFree( &x );
- cvFree( &y );
- cvFree( &caccum );
- cvFree( &buffer );
-}
-
-
-/****************************************************************************************\
-* Probabilistic Hough Transform *
-\****************************************************************************************/
-
-#if defined WIN64 && defined EM64T && _MSC_VER == 1400 && !defined CV_ICC
-#pragma optimize("",off)
-#endif
-
-static void
-icvHoughLinesProbabalistic( CvMat* image,
- float rho, float theta, int threshold,
- int lineLength, int lineGap,
- CvSeq *lines, int linesMax )
-{
- CvMat* accum = 0;
- CvMat* mask = 0;
- CvMat* trigtab = 0;
- CvMemStorage* storage = 0;
-
- CV_FUNCNAME( "icvHoughLinesProbalistic" );
-
- __BEGIN__;
-
- CvSeq* seq;
- CvSeqWriter writer;
- int width, height;
- int numangle, numrho;
- float ang;
- int r, n, count;
- CvPoint pt;
- float irho = 1 / rho;
- CvRNG rng = cvRNG(-1);
- const float* ttab;
- uchar* mdata0;
-
- CV_ASSERT( CV_IS_MAT(image) && CV_MAT_TYPE(image->type) == CV_8UC1 );
-
- width = image->cols;
- height = image->rows;
-
- numangle = cvRound(CV_PI / theta);
- numrho = cvRound(((width + height) * 2 + 1) / rho);
-
- CV_CALL( accum = cvCreateMat( numangle, numrho, CV_32SC1 ));
- CV_CALL( mask = cvCreateMat( height, width, CV_8UC1 ));
- CV_CALL( trigtab = cvCreateMat( 1, numangle, CV_32FC2 ));
- cvZero( accum );
-
- CV_CALL( storage = cvCreateMemStorage(0) );
-
- for( ang = 0, n = 0; n < numangle; ang += theta, n++ )
- {
- trigtab->data.fl[n*2] = (float)(cos(ang) * irho);
- trigtab->data.fl[n*2+1] = (float)(sin(ang) * irho);
- }
- ttab = trigtab->data.fl;
- mdata0 = mask->data.ptr;
-
- CV_CALL( cvStartWriteSeq( CV_32SC2, sizeof(CvSeq), sizeof(CvPoint), storage, &writer ));
-
- // stage 1. collect non-zero image points
- for( pt.y = 0, count = 0; pt.y < height; pt.y++ )
- {
- const uchar* data = image->data.ptr + pt.y*image->step;
- uchar* mdata = mdata0 + pt.y*width;
- for( pt.x = 0; pt.x < width; pt.x++ )
- {
- if( data[pt.x] )
- {
- mdata[pt.x] = (uchar)1;
- CV_WRITE_SEQ_ELEM( pt, writer );
- }
- else
- mdata[pt.x] = 0;
- }
- }
-
- seq = cvEndWriteSeq( &writer );
- count = seq->total;
-
- // stage 2. process all the points in random order
- for( ; count > 0; count-- )
- {
- // choose random point out of the remaining ones
- int idx = cvRandInt(&rng) % count;
- int max_val = threshold-1, max_n = 0;
- CvPoint* pt = (CvPoint*)cvGetSeqElem( seq, idx );
- CvPoint line_end[2] = {{0,0}, {0,0}};
- float a, b;
- int* adata = accum->data.i;
- int i, j, k, x0, y0, dx0, dy0, xflag;
- int good_line;
- const int shift = 16;
-
- i = pt->y;
- j = pt->x;
-
- // "remove" it by overriding it with the last element
- *pt = *(CvPoint*)cvGetSeqElem( seq, count-1 );
-
- // check if it has been excluded already (i.e. belongs to some other line)
- if( !mdata0[i*width + j] )
- continue;
-
- // update accumulator, find the most probable line
- for( n = 0; n < numangle; n++, adata += numrho )
- {
- r = cvRound( j * ttab[n*2] + i * ttab[n*2+1] );
- r += (numrho - 1) / 2;
- int val = ++adata[r];
- if( max_val < val )
- {
- max_val = val;
- max_n = n;
- }
- }
-
- // if it is too "weak" candidate, continue with another point
- if( max_val < threshold )
- continue;
-
- // from the current point walk in each direction
- // along the found line and extract the line segment
- a = -ttab[max_n*2+1];
- b = ttab[max_n*2];
- x0 = j;
- y0 = i;
- if( fabs(a) > fabs(b) )
- {
- xflag = 1;
- dx0 = a > 0 ? 1 : -1;
- dy0 = cvRound( b*(1 << shift)/fabs(a) );
- y0 = (y0 << shift) + (1 << (shift-1));
- }
- else
- {
- xflag = 0;
- dy0 = b > 0 ? 1 : -1;
- dx0 = cvRound( a*(1 << shift)/fabs(b) );
- x0 = (x0 << shift) + (1 << (shift-1));
- }
-
- for( k = 0; k < 2; k++ )
- {
- int gap = 0, x = x0, y = y0, dx = dx0, dy = dy0;
-
- if( k > 0 )
- dx = -dx, dy = -dy;
-
- // walk along the line using fixed-point arithmetics,
- // stop at the image border or in case of too big gap
- for( ;; x += dx, y += dy )
- {
- uchar* mdata;
- int i1, j1;
-
- if( xflag )
- {
- j1 = x;
- i1 = y >> shift;
- }
- else
- {
- j1 = x >> shift;
- i1 = y;
- }
-
- if( j1 < 0 || j1 >= width || i1 < 0 || i1 >= height )
- break;
-
- mdata = mdata0 + i1*width + j1;
-
- // for each non-zero point:
- // update line end,
- // clear the mask element
- // reset the gap
- if( *mdata )
- {
- gap = 0;
- line_end[k].y = i1;
- line_end[k].x = j1;
- }
- else if( ++gap > lineGap )
- break;
- }
- }
-
- good_line = abs(line_end[1].x - line_end[0].x) >= lineLength ||
- abs(line_end[1].y - line_end[0].y) >= lineLength;
-
- for( k = 0; k < 2; k++ )
- {
- int x = x0, y = y0, dx = dx0, dy = dy0;
-
- if( k > 0 )
- dx = -dx, dy = -dy;
-
- // walk along the line using fixed-point arithmetics,
- // stop at the image border or in case of too big gap
- for( ;; x += dx, y += dy )
- {
- uchar* mdata;
- int i1, j1;
-
- if( xflag )
- {
- j1 = x;
- i1 = y >> shift;
- }
- else
- {
- j1 = x >> shift;
- i1 = y;
- }
-
- mdata = mdata0 + i1*width + j1;
-
- // for each non-zero point:
- // update line end,
- // clear the mask element
- // reset the gap
- if( *mdata )
- {
- if( good_line )
- {
- adata = accum->data.i;
- for( n = 0; n < numangle; n++, adata += numrho )
- {
- r = cvRound( j1 * ttab[n*2] + i1 * ttab[n*2+1] );
- r += (numrho - 1) / 2;
- adata[r]--;
- }
- }
- *mdata = 0;
- }
-
- if( i1 == line_end[k].y && j1 == line_end[k].x )
- break;
- }
- }
-
- if( good_line )
- {
- CvRect lr = { line_end[0].x, line_end[0].y, line_end[1].x, line_end[1].y };
- cvSeqPush( lines, &lr );
- if( lines->total >= linesMax )
- EXIT;
- }
- }
-
- __END__;
-
- cvReleaseMat( &accum );
- cvReleaseMat( &mask );
- cvReleaseMat( &trigtab );
- cvReleaseMemStorage( &storage );
-}
-
-
-#if defined WIN64 && defined EM64T && _MSC_VER == 1400 && !defined CV_ICC
-#pragma optimize("",on)
-#endif
-
-
-/* Wrapper function for standard hough transform */
-CV_IMPL CvSeq*
-cvHoughLines2( CvArr* src_image, void* lineStorage, int method,
- double rho, double theta, int threshold,
- double param1, double param2 )
-{
- CvSeq* result = 0;
-
- CV_FUNCNAME( "cvHoughLines" );
-
- __BEGIN__;
-
- CvMat stub, *img = (CvMat*)src_image;
- CvMat* mat = 0;
- CvSeq* lines = 0;
- CvSeq lines_header;
- CvSeqBlock lines_block;
- int lineType, elemSize;
- int linesMax = INT_MAX;
- int iparam1, iparam2;
-
- CV_CALL( img = cvGetMat( img, &stub ));
-
- if( !CV_IS_MASK_ARR(img))
- CV_ERROR( CV_StsBadArg, "The source image must be 8-bit, single-channel" );
-
- if( !lineStorage )
- CV_ERROR( CV_StsNullPtr, "NULL destination" );
-
- if( rho <= 0 || theta <= 0 || threshold <= 0 )
- CV_ERROR( CV_StsOutOfRange, "rho, theta and threshold must be positive" );
-
- if( method != CV_HOUGH_PROBABILISTIC )
- {
- lineType = CV_32FC2;
- elemSize = sizeof(float)*2;
- }
- else
- {
- lineType = CV_32SC4;
- elemSize = sizeof(int)*4;
- }
-
- if( CV_IS_STORAGE( lineStorage ))
- {
- CV_CALL( lines = cvCreateSeq( lineType, sizeof(CvSeq), elemSize, (CvMemStorage*)lineStorage ));
- }
- else if( CV_IS_MAT( lineStorage ))
- {
- mat = (CvMat*)lineStorage;
-
- if( !CV_IS_MAT_CONT( mat->type ) || mat->rows != 1 && mat->cols != 1 )
- CV_ERROR( CV_StsBadArg,
- "The destination matrix should be continuous and have a single row or a single column" );
-
- if( CV_MAT_TYPE( mat->type ) != lineType )
- CV_ERROR( CV_StsBadArg,
- "The destination matrix data type is inappropriate, see the manual" );
-
- CV_CALL( lines = cvMakeSeqHeaderForArray( lineType, sizeof(CvSeq), elemSize, mat->data.ptr,
- mat->rows + mat->cols - 1, &lines_header, &lines_block ));
- linesMax = lines->total;
- CV_CALL( cvClearSeq( lines ));
- }
- else
- {
- CV_ERROR( CV_StsBadArg, "Destination is not CvMemStorage* nor CvMat*" );
- }
-
- iparam1 = cvRound(param1);
- iparam2 = cvRound(param2);
-
- switch( method )
- {
- case CV_HOUGH_STANDARD:
- CV_CALL( icvHoughLinesStandard( img, (float)rho,
- (float)theta, threshold, lines, linesMax ));
- break;
- case CV_HOUGH_MULTI_SCALE:
- CV_CALL( icvHoughLinesSDiv( img, (float)rho, (float)theta,
- threshold, iparam1, iparam2, lines, linesMax ));
- break;
- case CV_HOUGH_PROBABILISTIC:
- CV_CALL( icvHoughLinesProbabalistic( img, (float)rho, (float)theta,
- threshold, iparam1, iparam2, lines, linesMax ));
- break;
- default:
- CV_ERROR( CV_StsBadArg, "Unrecognized method id" );
- }
-
- if( mat )
- {
- if( mat->cols > mat->rows )
- mat->cols = lines->total;
- else
- mat->rows = lines->total;
- }
- else
- {
- result = lines;
- }
-
- __END__;
-
- return result;
-}
-
-
-/****************************************************************************************\
-* Circle Detection *
-\****************************************************************************************/
-
-static void
-icvHoughCirclesGradient( CvMat* img, float dp, float min_dist,
- int min_radius, int max_radius,
- int canny_threshold, int acc_threshold,
- CvSeq* circles, int circles_max )
-{
- const int SHIFT = 10, ONE = 1 << SHIFT, R_THRESH = 30;
- CvMat *dx = 0, *dy = 0;
- CvMat *edges = 0;
- CvMat *accum = 0;
- int* sort_buf = 0;
- CvMat* dist_buf = 0;
- CvMemStorage* storage = 0;
-
- CV_FUNCNAME( "icvHoughCirclesGradient" );
-
- __BEGIN__;
-
- int x, y, i, j, center_count, nz_count;
- int rows, cols, arows, acols;
- int astep, *adata;
- float* ddata;
- CvSeq *nz, *centers;
- float idp, dr;
- CvSeqReader reader;
-
- CV_CALL( edges = cvCreateMat( img->rows, img->cols, CV_8UC1 ));
- CV_CALL( cvCanny( img, edges, MAX(canny_threshold/2,1), canny_threshold, 3 ));
-
- CV_CALL( dx = cvCreateMat( img->rows, img->cols, CV_16SC1 ));
- CV_CALL( dy = cvCreateMat( img->rows, img->cols, CV_16SC1 ));
- CV_CALL( cvSobel( img, dx, 1, 0, 3 ));
- CV_CALL( cvSobel( img, dy, 0, 1, 3 ));
-
- if( dp < 1.f )
- dp = 1.f;
- idp = 1.f/dp;
- CV_CALL( accum = cvCreateMat( cvCeil(img->rows*idp)+2, cvCeil(img->cols*idp)+2, CV_32SC1 ));
- CV_CALL( cvZero(accum));
-
- CV_CALL( storage = cvCreateMemStorage() );
- CV_CALL( nz = cvCreateSeq( CV_32SC2, sizeof(CvSeq), sizeof(CvPoint), storage ));
- CV_CALL( centers = cvCreateSeq( CV_32SC1, sizeof(CvSeq), sizeof(int), storage ));
-
- rows = img->rows;
- cols = img->cols;
- arows = accum->rows - 2;
- acols = accum->cols - 2;
- adata = accum->data.i;
- astep = accum->step/sizeof(adata[0]);
-
- for( y = 0; y < rows; y++ )
- {
- const uchar* edges_row = edges->data.ptr + y*edges->step;
- const short* dx_row = (const short*)(dx->data.ptr + y*dx->step);
- const short* dy_row = (const short*)(dy->data.ptr + y*dy->step);
-
- for( x = 0; x < cols; x++ )
- {
- float vx, vy;
- int sx, sy, x0, y0, x1, y1, r, k;
- CvPoint pt;
-
- vx = dx_row[x];
- vy = dy_row[x];
-
- if( !edges_row[x] || vx == 0 && vy == 0 )
- continue;
-
- if( fabs(vx) < fabs(vy) )
- {
- sx = cvRound(vx*ONE/fabs(vy));
- sy = vy < 0 ? -ONE : ONE;
- }
- else
- {
- assert( vx != 0 );
- sy = cvRound(vy*ONE/fabs(vx));
- sx = vx < 0 ? -ONE : ONE;
- }
-
- x0 = cvRound((x*idp)*ONE) + ONE + (ONE/2);
- y0 = cvRound((y*idp)*ONE) + ONE + (ONE/2);
-
- for( k = 0; k < 2; k++ )
- {
- x0 += min_radius * sx;
- y0 += min_radius * sy;
-
- for( x1 = x0, y1 = y0, r = min_radius; r <= max_radius; x1 += sx, y1 += sy, r++ )
- {
- int x2 = x1 >> SHIFT, y2 = y1 >> SHIFT;
- if( (unsigned)x2 >= (unsigned)acols ||
- (unsigned)y2 >= (unsigned)arows )
- break;
- adata[y2*astep + x2]++;
- }
-
- x0 -= min_radius * sx;
- y0 -= min_radius * sy;
- sx = -sx; sy = -sy;
- }
-
- pt.x = x; pt.y = y;
- cvSeqPush( nz, &pt );
- }
- }
-
- nz_count = nz->total;
- if( !nz_count )
- EXIT;
-
- for( y = 1; y < arows - 1; y++ )
- {
- for( x = 1; x < acols - 1; x++ )
- {
- int base = y*(acols+2) + x;
- if( adata[base] > acc_threshold &&
- adata[base] > adata[base-1] && adata[base] > adata[base+1] &&
- adata[base] > adata[base-acols-2] && adata[base] > adata[base+acols+2] )
- cvSeqPush(centers, &base);
- }
- }
-
- center_count = centers->total;
- if( !center_count )
- EXIT;
-
- CV_CALL( sort_buf = (int*)cvAlloc( MAX(center_count,nz_count)*sizeof(sort_buf[0]) ));
- cvCvtSeqToArray( centers, sort_buf );
-
- icvHoughSortDescent32s( sort_buf, center_count, adata );
- cvClearSeq( centers );
- cvSeqPushMulti( centers, sort_buf, center_count );
-
- CV_CALL( dist_buf = cvCreateMat( 1, nz_count, CV_32FC1 ));
- ddata = dist_buf->data.fl;
-
- dr = dp;
- min_dist = MAX( min_dist, dp );
- min_dist *= min_dist;
-
- for( i = 0; i < centers->total; i++ )
- {
- int ofs = *(int*)cvGetSeqElem( centers, i );
- y = ofs/(acols+2) - 1;
- x = ofs - (y+1)*(acols+2) - 1;
- float cx = (float)(x*dp), cy = (float)(y*dp);
- int start_idx = nz_count - 1;
- float start_dist, dist_sum;
- float r_best = 0, c[3];
- int max_count = R_THRESH;
-
- for( j = 0; j < circles->total; j++ )
- {
- float* c = (float*)cvGetSeqElem( circles, j );
- if( (c[0] - cx)*(c[0] - cx) + (c[1] - cy)*(c[1] - cy) < min_dist )
- break;
- }
-
- if( j < circles->total )
- continue;
-
- cvStartReadSeq( nz, &reader );
- for( j = 0; j < nz_count; j++ )
- {
- CvPoint pt;
- float _dx, _dy;
- CV_READ_SEQ_ELEM( pt, reader );
- _dx = cx - pt.x; _dy = cy - pt.y;
- ddata[j] = _dx*_dx + _dy*_dy;
- sort_buf[j] = j;
- }
-
- cvPow( dist_buf, dist_buf, 0.5 );
- icvHoughSortDescent32s( sort_buf, nz_count, (int*)ddata );
-
- dist_sum = start_dist = ddata[sort_buf[nz_count-1]];
- for( j = nz_count - 2; j >= 0; j-- )
- {
- float d = ddata[sort_buf[j]];
-
- if( d > max_radius )
- break;
-
- if( d - start_dist > dr )
- {
- float r_cur = ddata[sort_buf[(j + start_idx)/2]];
- if( (start_idx - j)*r_best >= max_count*r_cur ||
- r_best < FLT_EPSILON && start_idx - j >= max_count )
- {
- r_best = r_cur;
- max_count = start_idx - j;
- }
- start_dist = d;
- start_idx = j;
- dist_sum = 0;
- }
- dist_sum += d;
- }
-
- if( max_count > R_THRESH )
- {
- c[0] = cx;
- c[1] = cy;
- c[2] = (float)r_best;
- cvSeqPush( circles, c );
- if( circles->total > circles_max )
- EXIT;
- }
- }
-
- __END__;
-
- cvReleaseMat( &dist_buf );
- cvFree( &sort_buf );
- cvReleaseMemStorage( &storage );
- cvReleaseMat( &edges );
- cvReleaseMat( &dx );
- cvReleaseMat( &dy );
- cvReleaseMat( &accum );
-}
-
-CV_IMPL CvSeq*
-cvHoughCircles( CvArr* src_image, void* circle_storage,
- int method, double dp, double min_dist,
- double param1, double param2,
- int min_radius, int max_radius )
-{
- CvSeq* result = 0;
-
- CV_FUNCNAME( "cvHoughCircles" );
-
- __BEGIN__;
-
- CvMat stub, *img = (CvMat*)src_image;
- CvMat* mat = 0;
- CvSeq* circles = 0;
- CvSeq circles_header;
- CvSeqBlock circles_block;
- int circles_max = INT_MAX;
- int canny_threshold = cvRound(param1);
- int acc_threshold = cvRound(param2);
-
- CV_CALL( img = cvGetMat( img, &stub ));
-
- if( !CV_IS_MASK_ARR(img))
- CV_ERROR( CV_StsBadArg, "The source image must be 8-bit, single-channel" );
-
- if( !circle_storage )
- CV_ERROR( CV_StsNullPtr, "NULL destination" );
-
- if( dp <= 0 || min_dist <= 0 || canny_threshold <= 0 || acc_threshold <= 0 )
- CV_ERROR( CV_StsOutOfRange, "dp, min_dist, canny_threshold and acc_threshold must be all positive numbers" );
-
- min_radius = MAX( min_radius, 0 );
- if( max_radius <= 0 )
- max_radius = MAX( img->rows, img->cols );
- else if( max_radius <= min_radius )
- max_radius = min_radius + 2;
-
- if( CV_IS_STORAGE( circle_storage ))
- {
- CV_CALL( circles = cvCreateSeq( CV_32FC3, sizeof(CvSeq),
- sizeof(float)*3, (CvMemStorage*)circle_storage ));
- }
- else if( CV_IS_MAT( circle_storage ))
- {
- mat = (CvMat*)circle_storage;
-
- if( !CV_IS_MAT_CONT( mat->type ) || mat->rows != 1 && mat->cols != 1 ||
- CV_MAT_TYPE(mat->type) != CV_32FC3 )
- CV_ERROR( CV_StsBadArg,
- "The destination matrix should be continuous and have a single row or a single column" );
-
- CV_CALL( circles = cvMakeSeqHeaderForArray( CV_32FC3, sizeof(CvSeq), sizeof(float)*3,
- mat->data.ptr, mat->rows + mat->cols - 1, &circles_header, &circles_block ));
- circles_max = circles->total;
- CV_CALL( cvClearSeq( circles ));
- }
- else
- {
- CV_ERROR( CV_StsBadArg, "Destination is not CvMemStorage* nor CvMat*" );
- }
-
- switch( method )
- {
- case CV_HOUGH_GRADIENT:
- CV_CALL( icvHoughCirclesGradient( img, (float)dp, (float)min_dist,
- min_radius, max_radius, canny_threshold,
- acc_threshold, circles, circles_max ));
- break;
- default:
- CV_ERROR( CV_StsBadArg, "Unrecognized method id" );
- }
-
- if( mat )
- {
- if( mat->cols > mat->rows )
- mat->cols = circles->total;
- else
- mat->rows = circles->total;
- }
- else
- result = circles;
-
- __END__;
-
- return result;
-}
-
-/* End of file. */