洪水填充C ++

时间:2011-11-30 00:02:46

标签: c++ cimg flooding

执行洪水填充时遇到问题。
任务是要求用户点击图像的白色部分(表示种子点),他想填充黑色。
操作应该在二进制图像上完成。
我正在使用CImg库。
不能使用递归算法。我想出了一些东西,但它没有正常工作(间隙只在种子点变黑)。我根本不熟悉队列,所以问题可能在于它们的实现。

void floodfill(int x, int y, int c, int b, CImg <unsigned char>image)
{
    //c-black
    //b-white
    CImg<unsigned char> kopia(image.width(),image.height());

    for (int p=1; p<image.height()-1; p++)
    {
        for (int q=1; q<image.width()-1; q++)
        {
            kopia(p,q)=255; //setting kopia2 all white
        }
    }

    queue <pair<int,int> > a;
    int p;
    if(image(x, y) == c)
    {
        cout<<"Already black"<<endl;
        return;
    }
    else
    {
        a.push(make_pair(x, y));
        while(!a.empty())
        {
            a.pop();
            p=image(x+1, y);
            if((p == b) && (x < image.width()))
            {
                a.push(make_pair(x+1, y));
                kopia(x+1, y)=c;
                image(x+1, y)=c;
            }
            p = image(x-1, y);
            if((p == c) && (x > 0))
            {
                a.push(make_pair(x-1, y));
                kopia(x-1, y)=c;
                image(x-1, y)=c;
            }
            p=image(x, y+1);
            if((p == b) && (y < image.height()))
            {
                a.push(make_pair(x, y+1));
                kopia(x, y+1)=c;
                image(x, y+1)=c;
            }
            p=image(x, y-1);
            if((p == b) && (y > 0))
            {
                a.push(make_pair(x, y-1));
                kopia(x, y-1)=c;
                image(x, y-1)=c;
            }
        }
        saving(kopia);
    }
}

void hole (CImg <unsigned char>image)
{
    CImgDisplay image_disp(image,"Click a point");

    int c_x=0; //coordinates
    int c_y=0;

    while (!image_disp.is_closed())
    {
        image_disp.wait();
        if (image_disp.button())
        {
            c_x=image_disp.mouse_x();  //reads coordinates indicated by user
            c_y=image_disp.mouse_y();
        }
    }

    floodfill(c_x, c_y,0,255,image);
}

3 个答案:

答案 0 :(得分:1)

1)

    while(!a.empty())
    {
        x = a.front().first; //fixed as per ChristianRau's code
        y = a.front().second; //fixed as per ChristianRau's code
        a.pop();

你只是将当前的x,y坐标从堆栈中弹出而不看它们是什么。

2)

        p = image(x-1, y);
        if((p == c) && (x > 0))

你的意思是检查它是否是白色的,就像你对其他方向一样吗?

3)来电者传递黑白,如果图像的一部分是蓝色,会发生什么?更好的是传递填充颜色(黑色),无论你有白色,用非黑色替换它。

答案 1 :(得分:1)

您是否意识到自己始终使用相同的xy并且a.pop()没有返回任何内容? std::queue::pop仅弹出队列的前端,但不返回它。您必须事先使用std::queue::front进行查询。所以只需添加

x = a.front().first;
y = a.front().second;

在while循环内a.pop()之前。

顺便说一下,在推送初始对之前,您可能还想在else块的开头将image(x, y)(可能是kopia(x, y))设置为c,尽管可能也可以通过邻居的迭代来设定。

答案 2 :(得分:1)

此外,CImg中还有一个内置函数可以执行您想要的操作:CImg :: draw_fill()。