使用Flood Fill算法创建数组

时间:2013-06-20 19:43:16

标签: java recursion flood-fill

我正在使用填充填充算法对图像进行排序。如果它遇到相同的颜色,我希望它将该像素复制到一个名为filled的相同大小的数组中。然后将填充的数组转换回图像并保存为jpg。但是,当我打开jpg时,它看起来完全是黑色的。

public static void findFace(int[][] image) throws IOException {
    int height = image.length;
    int width = image[0].length;

    Color centerStart = new Color(image[width / 2][height / 2]);
    int[][] filled = new int[width][height];

    floodFill(width / 2, height / 2, centerStart, image, filled);

    //construct the filled array as image. Show if the face was found.
    BufferedImage bufferImage2 = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);

    for (int y = 0; y < height; y++) {
        for (int x = 0; x < width; x++) {
            int Pixel = filled[x][y] << 16 | filled[x][y] << 8 | filled[x][y];
            bufferImage2.setRGB(x, y, Pixel);
        }
    }

    //save filled array as image file
    File outputfile = new File("/home/lily/Pictures/APicaDay/saved.jpg");
    ImageIO.write(bufferImage2, "jpg", outputfile);
}

public static int[][] floodFill(int x, int y, Color targetColor, int[][] image, int[][] filled) {
    if (image[x][y] != targetColor.getRGB()) {
        return filled;
    }

    filled[x][y] = image[x][y];

    floodFill(x - 1, y, targetColor, image, filled);
    floodFill(x + 1, y, targetColor, image, filled);
    floodFill(x, y - 1, targetColor, image, filled);
    floodFill(x, y + 1, targetColor, image, filled);

    return filled;
}

红利问题:我希望洪水填充也能接受相似但不完全相同的颜色,因为我正在处理照片。

1 个答案:

答案 0 :(得分:1)

您发布的floodFill功能缺少两个重要元素:

  1. 如果包含与第一个像素相同颜色的区域一直延伸到图像的边界,则该函数将尝试在无效索引处访问image。您可以通过首先检查您正在检查的像素的x和y坐标来修复此问题,并在它们超出范围时立即返回。
  2. 如果有多个相同颜色的相邻像素,该函数将无限递归,因为初始调用将在第二个像素上调用floodFill,然后调用floodFill在第一个像素上,依此类推。您需要一种方法来确保只在特定像素上调用floodFill一次。
  3. 由于您没有观察到这两种症状中的任何一种,并且您没有观察到生成的图像中的任何内容,我想初始像素的颜色检查不正确。将整数传递给Color构造函数时,您确定它使用该整数的RBG解释吗?