Java BufferedImage 地图颜色

Question

我正在尝试编写一个程序，该程序将 bufferedImage 作为输入并将所有接近黑色的颜色（R

public static BufferedImage BlackAndWhite(BufferedImage image) {

    ColorModel model = new BlackWhiteColorModel(DataBuffer.TYPE_INT);
    WritableRaster raster = image.getRaster();

    BufferedImage newImage = new BufferedImage(model, raster, false, null);

    return newImage;
}

BlackWhiteColorModel 被定义为

public class BlackWhiteColorModel extends ColorModel {

public BlackWhiteColorModel(int bits) {
    super(bits);
}

@Override
public int getRed(int pixel) {
    int[] rgb = getRgb(pixel);

    if (rgb[0] < 32 && rgb[1] < 32 && rgb[2] < 32) {
        return 0;
    } else {
        return 255;
    }
}

@Override
public int getGreen(int pixel) {
    int[] rgb = getRgb(pixel);

    if (rgb[0] < 32 && rgb[1] < 32 && rgb[2] < 32) {
        return 0;
    } else {
        return 255;
    }
}

@Override
public int getBlue(int pixel) {
    int[] rgb = getRgb(pixel);

    if (rgb[0] < 32 && rgb[1] < 32 && rgb[2] < 32) {
        return 0;
    } else {
        return 255;
    }
}

@Override
public int getAlpha(int pixel) {
    return pixel;
}

private int[] getRgb(int pixel) {
    int r = (pixel) & 0xFF;
    int g = (pixel >> 8) & 0xFF;
    int b = (pixel >> 16) & 0xFF;
    int a = (pixel >> 24) & 0xFF;

    return new int[]{r, g, b, a};
}



}

但是，我最终得到了 isCompatibleRasterException。谁能给我一些建议？

Answer 1

(1) 你没有实现合约的方法之一：isCompatibleRaster。鉴于您的问题陈述，这是一个过于乐观的实现，运行良好：

@Override
public boolean isCompatibleRaster(Raster raster) {
    return true;
}

(2) getRed、getGreen 和 getBlue 具有相同的代码。这是一个完整的程序，它将公共代码分解为一个名为getColor 的新方法。该程序运行无误：

import java.awt.*;
import java.awt.image.*;

public class BlackWhiteColorModel extends ColorModel {
    public static void main(String[] args) {
        BufferedImage bufferedImage = new BufferedImage(200,200,BufferedImage.TYPE_INT_RGB);
         Graphics g = bufferedImage.getGraphics();
         g.drawString("Hello, World", 20,20);
        blackAndWhite(bufferedImage);
    }

    public static BufferedImage blackAndWhite(BufferedImage image) {
        ColorModel model = new BlackWhiteColorModel(DataBuffer.TYPE_INT);
        WritableRaster raster = image.getRaster();

        BufferedImage newImage = new BufferedImage(model, raster, false, null);

        return newImage;
    }

    public BlackWhiteColorModel(int bits) {
        super(bits);
    }

    private int getColor(int pixel) {
        int[] rgb = getRgb(pixel);

        if (rgb[0] < 32 && rgb[1] < 32 && rgb[2] < 32) {
            return 0;
        } else {
            return 255;
        }
    }

    @Override
    public int getRed(int pixel) {
        return getColor(pixel);
    }

    @Override
    public int getGreen(int pixel) {
        return getColor(pixel);
    }

    @Override
    public int getBlue(int pixel) {
        return getColor(pixel);
    }

    @Override
    public int getAlpha(int pixel) {
        return pixel;
    }

    @Override
    public boolean isCompatibleRaster(Raster raster) {
        return true;
    }

    private int[] getRgb(int pixel) {
        int r = (pixel) & 0xFF;
        int g = (pixel >> 8) & 0xFF;
        int b = (pixel >> 16) & 0xFF;
        int a = (pixel >> 24) & 0xFF;

        return new int[]{r, g, b, a};
    }
}

Answer 2

你想要的是一个阈值操作，例如：

我猜你不想因为性能问题而对像素进行迭代。如果这是真的，我强烈建议您像这样使用OpenCV：

BufferedImage image = null; //use your image here
//Convert image to OpenCv Mat
byte[] pixels = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
Mat mat = new Mat(image.getHeight(), image.getWidth(), CvType.CV_8UC3);
mat.put(0, 0, pixels);

//do something with the Mat e.g:
Imgproc.threshold(...); 

//Convert back
mat.get(0, 0, pixels);

如果您进行更多图像处理，性能至少会快 10 倍