| /*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" |
| |
| /****************************************************************************************\ |
| * Watershed * |
| \****************************************************************************************/ |
| |
| typedef struct CvWSNode |
| { |
| struct CvWSNode* next; |
| int mask_ofs; |
| int img_ofs; |
| } |
| CvWSNode; |
| |
| typedef struct CvWSQueue |
| { |
| CvWSNode* first; |
| CvWSNode* last; |
| } |
| CvWSQueue; |
| |
| static CvWSNode* |
| icvAllocWSNodes( CvMemStorage* storage ) |
| { |
| CvWSNode* n = 0; |
| |
| CV_FUNCNAME( "icvAllocWSNodes" ); |
| |
| __BEGIN__; |
| |
| int i, count = (storage->block_size - sizeof(CvMemBlock))/sizeof(*n) - 1; |
| |
| CV_CALL( n = (CvWSNode*)cvMemStorageAlloc( storage, count*sizeof(*n) )); |
| for( i = 0; i < count-1; i++ ) |
| n[i].next = n + i + 1; |
| n[count-1].next = 0; |
| |
| __END__; |
| |
| return n; |
| } |
| |
| |
| CV_IMPL void |
| cvWatershed( const CvArr* srcarr, CvArr* dstarr ) |
| { |
| const int IN_QUEUE = -2; |
| const int WSHED = -1; |
| const int NQ = 256; |
| CvMemStorage* storage = 0; |
| |
| CV_FUNCNAME( "cvWatershed" ); |
| |
| __BEGIN__; |
| |
| CvMat sstub, *src; |
| CvMat dstub, *dst; |
| CvSize size; |
| CvWSNode* free_node = 0, *node; |
| CvWSQueue q[NQ]; |
| int active_queue; |
| int i, j; |
| int db, dg, dr; |
| int* mask; |
| uchar* img; |
| int mstep, istep; |
| int subs_tab[513]; |
| |
| // MAX(a,b) = b + MAX(a-b,0) |
| #define ws_max(a,b) ((b) + subs_tab[(a)-(b)+NQ]) |
| // MIN(a,b) = a - MAX(a-b,0) |
| #define ws_min(a,b) ((a) - subs_tab[(a)-(b)+NQ]) |
| |
| #define ws_push(idx,mofs,iofs) \ |
| { \ |
| if( !free_node ) \ |
| CV_CALL( free_node = icvAllocWSNodes( storage ));\ |
| node = free_node; \ |
| free_node = free_node->next;\ |
| node->next = 0; \ |
| node->mask_ofs = mofs; \ |
| node->img_ofs = iofs; \ |
| if( q[idx].last ) \ |
| q[idx].last->next=node; \ |
| else \ |
| q[idx].first = node; \ |
| q[idx].last = node; \ |
| } |
| |
| #define ws_pop(idx,mofs,iofs) \ |
| { \ |
| node = q[idx].first; \ |
| q[idx].first = node->next; \ |
| if( !node->next ) \ |
| q[idx].last = 0; \ |
| node->next = free_node; \ |
| free_node = node; \ |
| mofs = node->mask_ofs; \ |
| iofs = node->img_ofs; \ |
| } |
| |
| #define c_diff(ptr1,ptr2,diff) \ |
| { \ |
| db = abs((ptr1)[0] - (ptr2)[0]);\ |
| dg = abs((ptr1)[1] - (ptr2)[1]);\ |
| dr = abs((ptr1)[2] - (ptr2)[2]);\ |
| diff = ws_max(db,dg); \ |
| diff = ws_max(diff,dr); \ |
| assert( 0 <= diff && diff <= 255 ); \ |
| } |
| |
| CV_CALL( src = cvGetMat( srcarr, &sstub )); |
| CV_CALL( dst = cvGetMat( dstarr, &dstub )); |
| |
| if( CV_MAT_TYPE(src->type) != CV_8UC3 ) |
| CV_ERROR( CV_StsUnsupportedFormat, "Only 8-bit, 3-channel input images are supported" ); |
| |
| if( CV_MAT_TYPE(dst->type) != CV_32SC1 ) |
| CV_ERROR( CV_StsUnsupportedFormat, |
| "Only 32-bit, 1-channel output images are supported" ); |
| |
| if( !CV_ARE_SIZES_EQ( src, dst )) |
| CV_ERROR( CV_StsUnmatchedSizes, "The input and output images must have the same size" ); |
| |
| size = cvGetMatSize(src); |
| |
| CV_CALL( storage = cvCreateMemStorage() ); |
| |
| istep = src->step; |
| img = src->data.ptr; |
| mstep = dst->step / sizeof(mask[0]); |
| mask = dst->data.i; |
| |
| memset( q, 0, NQ*sizeof(q[0]) ); |
| |
| for( i = 0; i < 256; i++ ) |
| subs_tab[i] = 0; |
| for( i = 256; i <= 512; i++ ) |
| subs_tab[i] = i - 256; |
| |
| // draw a pixel-wide border of dummy "watershed" (i.e. boundary) pixels |
| for( j = 0; j < size.width; j++ ) |
| mask[j] = mask[j + mstep*(size.height-1)] = WSHED; |
| |
| // initial phase: put all the neighbor pixels of each marker to the ordered queue - |
| // determine the initial boundaries of the basins |
| for( i = 1; i < size.height-1; i++ ) |
| { |
| img += istep; mask += mstep; |
| mask[0] = mask[size.width-1] = WSHED; |
| |
| for( j = 1; j < size.width-1; j++ ) |
| { |
| int* m = mask + j; |
| if( m[0] < 0 ) m[0] = 0; |
| if( m[0] == 0 && (m[-1] > 0 || m[1] > 0 || m[-mstep] > 0 || m[mstep] > 0) ) |
| { |
| uchar* ptr = img + j*3; |
| int idx = 256, t; |
| if( m[-1] > 0 ) |
| c_diff( ptr, ptr - 3, idx ); |
| if( m[1] > 0 ) |
| { |
| c_diff( ptr, ptr + 3, t ); |
| idx = ws_min( idx, t ); |
| } |
| if( m[-mstep] > 0 ) |
| { |
| c_diff( ptr, ptr - istep, t ); |
| idx = ws_min( idx, t ); |
| } |
| if( m[mstep] > 0 ) |
| { |
| c_diff( ptr, ptr + istep, t ); |
| idx = ws_min( idx, t ); |
| } |
| assert( 0 <= idx && idx <= 255 ); |
| ws_push( idx, i*mstep + j, i*istep + j*3 ); |
| m[0] = IN_QUEUE; |
| } |
| } |
| } |
| |
| // find the first non-empty queue |
| for( i = 0; i < NQ; i++ ) |
| if( q[i].first ) |
| break; |
| |
| // if there is no markers, exit immediately |
| if( i == NQ ) |
| EXIT; |
| |
| active_queue = i; |
| img = src->data.ptr; |
| mask = dst->data.i; |
| |
| // recursively fill the basins |
| for(;;) |
| { |
| int mofs, iofs; |
| int lab = 0, t; |
| int* m; |
| uchar* ptr; |
| |
| if( q[active_queue].first == 0 ) |
| { |
| for( i = active_queue+1; i < NQ; i++ ) |
| if( q[i].first ) |
| break; |
| if( i == NQ ) |
| break; |
| active_queue = i; |
| } |
| |
| ws_pop( active_queue, mofs, iofs ); |
| |
| m = mask + mofs; |
| ptr = img + iofs; |
| t = m[-1]; |
| if( t > 0 ) lab = t; |
| t = m[1]; |
| if( t > 0 ) |
| { |
| if( lab == 0 ) lab = t; |
| else if( t != lab ) lab = WSHED; |
| } |
| t = m[-mstep]; |
| if( t > 0 ) |
| { |
| if( lab == 0 ) lab = t; |
| else if( t != lab ) lab = WSHED; |
| } |
| t = m[mstep]; |
| if( t > 0 ) |
| { |
| if( lab == 0 ) lab = t; |
| else if( t != lab ) lab = WSHED; |
| } |
| assert( lab != 0 ); |
| m[0] = lab; |
| if( lab == WSHED ) |
| continue; |
| |
| if( m[-1] == 0 ) |
| { |
| c_diff( ptr, ptr - 3, t ); |
| ws_push( t, mofs - 1, iofs - 3 ); |
| active_queue = ws_min( active_queue, t ); |
| m[-1] = IN_QUEUE; |
| } |
| if( m[1] == 0 ) |
| { |
| c_diff( ptr, ptr + 3, t ); |
| ws_push( t, mofs + 1, iofs + 3 ); |
| active_queue = ws_min( active_queue, t ); |
| m[1] = IN_QUEUE; |
| } |
| if( m[-mstep] == 0 ) |
| { |
| c_diff( ptr, ptr - istep, t ); |
| ws_push( t, mofs - mstep, iofs - istep ); |
| active_queue = ws_min( active_queue, t ); |
| m[-mstep] = IN_QUEUE; |
| } |
| if( m[mstep] == 0 ) |
| { |
| c_diff( ptr, ptr + 3, t ); |
| ws_push( t, mofs + mstep, iofs + istep ); |
| active_queue = ws_min( active_queue, t ); |
| m[mstep] = IN_QUEUE; |
| } |
| } |
| |
| __END__; |
| |
| cvReleaseMemStorage( &storage ); |
| } |
| |
| |
| /****************************************************************************************\ |
| * Meanshift * |
| \****************************************************************************************/ |
| |
| CV_IMPL void |
| cvPyrMeanShiftFiltering( const CvArr* srcarr, CvArr* dstarr, |
| double sp0, double sr, int max_level, |
| CvTermCriteria termcrit ) |
| { |
| const int cn = 3; |
| const int MAX_LEVELS = 8; |
| CvMat* src_pyramid[MAX_LEVELS+1]; |
| CvMat* dst_pyramid[MAX_LEVELS+1]; |
| CvMat* mask0 = 0; |
| int i, j, level; |
| //uchar* submask = 0; |
| |
| #define cdiff(ofs0) (tab[c0-dptr[ofs0]+255] + \ |
| tab[c1-dptr[(ofs0)+1]+255] + tab[c2-dptr[(ofs0)+2]+255] >= isr22) |
| |
| memset( src_pyramid, 0, sizeof(src_pyramid) ); |
| memset( dst_pyramid, 0, sizeof(dst_pyramid) ); |
| |
| CV_FUNCNAME( "cvPyrMeanShiftFiltering" ); |
| |
| __BEGIN__; |
| |
| double sr2 = sr * sr; |
| int isr2 = cvRound(sr2), isr22 = MAX(isr2,16); |
| int tab[768]; |
| CvMat sstub0, *src0; |
| CvMat dstub0, *dst0; |
| |
| CV_CALL( src0 = cvGetMat( srcarr, &sstub0 )); |
| CV_CALL( dst0 = cvGetMat( dstarr, &dstub0 )); |
| |
| if( CV_MAT_TYPE(src0->type) != CV_8UC3 ) |
| CV_ERROR( CV_StsUnsupportedFormat, "Only 8-bit, 3-channel images are supported" ); |
| |
| if( !CV_ARE_TYPES_EQ( src0, dst0 )) |
| CV_ERROR( CV_StsUnmatchedFormats, "The input and output images must have the same type" ); |
| |
| if( !CV_ARE_SIZES_EQ( src0, dst0 )) |
| CV_ERROR( CV_StsUnmatchedSizes, "The input and output images must have the same size" ); |
| |
| if( (unsigned)max_level > (unsigned)MAX_LEVELS ) |
| CV_ERROR( CV_StsOutOfRange, "The number of pyramid levels is too large or negative" ); |
| |
| if( !(termcrit.type & CV_TERMCRIT_ITER) ) |
| termcrit.max_iter = 5; |
| termcrit.max_iter = MAX(termcrit.max_iter,1); |
| termcrit.max_iter = MIN(termcrit.max_iter,100); |
| if( !(termcrit.type & CV_TERMCRIT_EPS) ) |
| termcrit.epsilon = 1.f; |
| termcrit.epsilon = MAX(termcrit.epsilon, 0.f); |
| |
| for( i = 0; i < 768; i++ ) |
| tab[i] = (i - 255)*(i - 255); |
| |
| // 1. construct pyramid |
| src_pyramid[0] = src0; |
| dst_pyramid[0] = dst0; |
| for( level = 1; level <= max_level; level++ ) |
| { |
| CV_CALL( src_pyramid[level] = cvCreateMat( (src_pyramid[level-1]->rows+1)/2, |
| (src_pyramid[level-1]->cols+1)/2, src_pyramid[level-1]->type )); |
| CV_CALL( dst_pyramid[level] = cvCreateMat( src_pyramid[level]->rows, |
| src_pyramid[level]->cols, src_pyramid[level]->type )); |
| CV_CALL( cvPyrDown( src_pyramid[level-1], src_pyramid[level] )); |
| //CV_CALL( cvResize( src_pyramid[level-1], src_pyramid[level], CV_INTER_AREA )); |
| } |
| |
| CV_CALL( mask0 = cvCreateMat( src0->rows, src0->cols, CV_8UC1 )); |
| //CV_CALL( submask = (uchar*)cvAlloc( (sp+2)*(sp+2) )); |
| |
| // 2. apply meanshift, starting from the pyramid top (i.e. the smallest layer) |
| for( level = max_level; level >= 0; level-- ) |
| { |
| CvMat* src = src_pyramid[level]; |
| CvSize size = cvGetMatSize(src); |
| uchar* sptr = src->data.ptr; |
| int sstep = src->step; |
| uchar* mask = 0; |
| int mstep = 0; |
| uchar* dptr; |
| int dstep; |
| float sp = (float)(sp0 / (1 << level)); |
| sp = MAX( sp, 1 ); |
| |
| if( level < max_level ) |
| { |
| CvSize size1 = cvGetMatSize(dst_pyramid[level+1]); |
| CvMat m = cvMat( size.height, size.width, CV_8UC1, mask0->data.ptr ); |
| dstep = dst_pyramid[level+1]->step; |
| dptr = dst_pyramid[level+1]->data.ptr + dstep + cn; |
| mstep = m.step; |
| mask = m.data.ptr + mstep; |
| //cvResize( dst_pyramid[level+1], dst_pyramid[level], CV_INTER_CUBIC ); |
| cvPyrUp( dst_pyramid[level+1], dst_pyramid[level] ); |
| cvZero( &m ); |
| |
| for( i = 1; i < size1.height-1; i++, dptr += dstep - (size1.width-2)*3, mask += mstep*2 ) |
| { |
| for( j = 1; j < size1.width-1; j++, dptr += cn ) |
| { |
| int c0 = dptr[0], c1 = dptr[1], c2 = dptr[2]; |
| mask[j*2 - 1] = cdiff(-3) || cdiff(3) || cdiff(-dstep-3) || cdiff(-dstep) || |
| cdiff(-dstep+3) || cdiff(dstep-3) || cdiff(dstep) || cdiff(dstep+3); |
| } |
| } |
| |
| cvDilate( &m, &m, 0, 1 ); |
| mask = m.data.ptr; |
| } |
| |
| dptr = dst_pyramid[level]->data.ptr; |
| dstep = dst_pyramid[level]->step; |
| |
| for( i = 0; i < size.height; i++, sptr += sstep - size.width*3, |
| dptr += dstep - size.width*3, |
| mask += mstep ) |
| { |
| for( j = 0; j < size.width; j++, sptr += 3, dptr += 3 ) |
| { |
| int x0 = j, y0 = i, x1, y1, iter; |
| int c0, c1, c2; |
| |
| if( mask && !mask[j] ) |
| continue; |
| |
| c0 = sptr[0], c1 = sptr[1], c2 = sptr[2]; |
| |
| // iterate meanshift procedure |
| for( iter = 0; iter < termcrit.max_iter; iter++ ) |
| { |
| uchar* ptr; |
| int x, y, count = 0; |
| int minx, miny, maxx, maxy; |
| int s0 = 0, s1 = 0, s2 = 0, sx = 0, sy = 0; |
| double icount; |
| int stop_flag; |
| |
| //mean shift: process pixels in window (p-sigmaSp)x(p+sigmaSp) |
| minx = cvRound(x0 - sp); minx = MAX(minx, 0); |
| miny = cvRound(y0 - sp); miny = MAX(miny, 0); |
| maxx = cvRound(x0 + sp); maxx = MIN(maxx, size.width-1); |
| maxy = cvRound(y0 + sp); maxy = MIN(maxy, size.height-1); |
| ptr = sptr + (miny - i)*sstep + (minx - j)*3; |
| |
| for( y = miny; y <= maxy; y++, ptr += sstep - (maxx-minx+1)*3 ) |
| { |
| int row_count = 0; |
| x = minx; |
| for( ; x + 3 <= maxx; x += 4, ptr += 12 ) |
| { |
| int t0 = ptr[0], t1 = ptr[1], t2 = ptr[2]; |
| if( tab[t0-c0+255] + tab[t1-c1+255] + tab[t2-c2+255] <= isr2 ) |
| { |
| s0 += t0; s1 += t1; s2 += t2; |
| sx += x; row_count++; |
| } |
| t0 = ptr[3], t1 = ptr[4], t2 = ptr[5]; |
| if( tab[t0-c0+255] + tab[t1-c1+255] + tab[t2-c2+255] <= isr2 ) |
| { |
| s0 += t0; s1 += t1; s2 += t2; |
| sx += x+1; row_count++; |
| } |
| t0 = ptr[6], t1 = ptr[7], t2 = ptr[8]; |
| if( tab[t0-c0+255] + tab[t1-c1+255] + tab[t2-c2+255] <= isr2 ) |
| { |
| s0 += t0; s1 += t1; s2 += t2; |
| sx += x+2; row_count++; |
| } |
| t0 = ptr[9], t1 = ptr[10], t2 = ptr[11]; |
| if( tab[t0-c0+255] + tab[t1-c1+255] + tab[t2-c2+255] <= isr2 ) |
| { |
| s0 += t0; s1 += t1; s2 += t2; |
| sx += x+3; row_count++; |
| } |
| } |
| |
| for( ; x <= maxx; x++, ptr += 3 ) |
| { |
| int t0 = ptr[0], t1 = ptr[1], t2 = ptr[2]; |
| if( tab[t0-c0+255] + tab[t1-c1+255] + tab[t2-c2+255] <= isr2 ) |
| { |
| s0 += t0; s1 += t1; s2 += t2; |
| sx += x; row_count++; |
| } |
| } |
| count += row_count; |
| sy += y*row_count; |
| } |
| |
| if( count == 0 ) |
| break; |
| |
| icount = 1./count; |
| x1 = cvRound(sx*icount); |
| y1 = cvRound(sy*icount); |
| s0 = cvRound(s0*icount); |
| s1 = cvRound(s1*icount); |
| s2 = cvRound(s2*icount); |
| |
| stop_flag = (x0 == x1 && y0 == y1) || abs(x1-x0) + abs(y1-y0) + |
| tab[s0 - c0 + 255] + tab[s1 - c1 + 255] + |
| tab[s2 - c2 + 255] <= termcrit.epsilon; |
| |
| x0 = x1; y0 = y1; |
| c0 = s0; c1 = s1; c2 = s2; |
| |
| if( stop_flag ) |
| break; |
| } |
| |
| dptr[0] = (uchar)c0; |
| dptr[1] = (uchar)c1; |
| dptr[2] = (uchar)c2; |
| } |
| } |
| } |
| |
| __END__; |
| |
| for( i = 1; i <= MAX_LEVELS; i++ ) |
| { |
| cvReleaseMat( &src_pyramid[i] ); |
| cvReleaseMat( &dst_pyramid[i] ); |
| } |
| cvReleaseMat( &mask0 ); |
| } |
| |