opencv连通区域标记程序

//连通区域标C++记程序
//作者：胡柳

//本程序使用了opencv中图像数据结构：IplImage
//算法原理：首先进行第一次扫描，标记连通区域；
//然后对等价标记进行合并，
//最后进行第二次扫描，对等价标记进行重新标记。

//输入：mask为标记模板指针，img1为待标记二值图像指针
//返回值为连通区域数
int label( WORD *mask，IplImage *img1)
{
    BYTE *pb1=(BYTE*)img1->imageData;
    int height=img1->height;
    int width=img1->width;
    int k,j;

    for( k=0; k
        mask[k]=0;

    int cw=100;//初始区域数目
loop:
    bool *col=new bool[cw*cw];
    for(k=0; k
        col[k]=false;

//第一次扫描
    int lab=1;
    for(k=1;k
        for( j=1;j
            if(pb1[k*width+j]==255)
                if((mask[k*width+j-1]+mask[(k-1)*width+j-1]+mask[(k-1)*width+j]+mask[(k-1)*width+j+1])==0)
                {
                    mask[k*width+j]=lab;
                    lab=lab+1;
                    if(lab>cw)
                    {
                        delete col;
                        cw+=200;
                        goto loop;
                    }
                }
                else
                {
                    if(mask[k*width+j-1]!=0)
                        mask[k*width+j]=mask[k*width+j-1];

                    if(mask[(k-1)*width+j-1]!=0)
                        if(mask[k*width+j]==0)
                            mask[k*width+j]=mask[(k-1)*width+j-1];
                        else
                        {
                            col[mask[k*width+j]*cw+mask[(k-1)*width+j-1]]=true;
                            col[mask[(k-1)*width+j-1]*cw+mask[k*width+j]]=true;
                        }

                    if(mask[(k-1)*width+j]!=0)
                        if(mask[k*width+j]==0)
                            mask[k*width+j]=mask[(k-1)*width+j];
                        else
                        {
                            col[mask[k*width+j]*cw+mask[(k-1)*width+j]]=true;
                            col[mask[(k-1)*width+j]*cw+mask[k*width+j]]=true;
                        }

                    if(mask[(k-1)*width+j+1]!=0)
                        if(mask[k*width+j]==0)
                            mask[k*width+j]=mask[(k-1)*width+j+1];
                        else
                        {
                            col[mask[k*width+j]*cw+mask[(k-1)*width+j+1]]=true;
                            col[mask[(k-1)*width+j+1]*cw+mask[k*width+j]]=true;
                        }
                }

    if(lab==1)
        return 0;

//等价关系合并
    bool *col2=new bool[lab*lab];
    for(k=0; k
        col2[k]=false;
    for(k=1;k
        for(j=1;j
            if(col[k*cw+j]==true||k==j)
                col2[k*lab+j]=true;
    delete[] col;

    for(k=1;k
        for(j=1;j
            if(col2[j*lab+k]==true)
                for(int i=1;i
                    col2[j*lab+i]=col2[j*lab+i]||col2[k*lab+i];


    WORD *col3=new WORD[lab];
    WORD *col3_s=new WORD[lab];
    for(k=1;k
    {
        int c1, c2;
        for(j=1;j
            if(col2[k*lab+j]==true)
            {
                c1=j;
                break;
            }
        for(j=1;j
            if(col2[j*lab+k]==true)
            {
                c2=j;
                break;
            }
        col3[k]=min(c1,c2);
        col3_s[k]=col3[k];
    }


    for(k=1;k
    {
        WORD t;
        for(j=1;j
            if(col3_s[j]>col3_s[j+1])
            {
                t=col3_s[j];
                col3_s[j]=col3_s[j+1];
                col3_s[j+1]=t;
            }
    }

    WORD *ind=new WORD[lab];
    ind[col3_s[1]]=1;
    int c=2;
    for(k=1;k
        if(col3_s[k+1]!=col3_s[k])
        {
            ind[col3_s[k+1]]=c;
            c++;
        }

//第2次扫描
    for(k=1;k
        for( j=1;j
            if(mask[k*width+j]!=0)
                mask[k*width+j]=ind[col3[mask[k*width+j]]];

   delete[] col2;
   delete[] col3;
   delete[] col3_s;
   delete[] ind;

    return(c-1);//返回区域数目
 }