将2D像素数组转换为BufferedImage

Question

我使用了从this SO接受的答案代码。现在，我想要将其转换回BufferedImage（最好不要使用setRGB（））。我试过这个：

private BufferedImage createImage(int[][] pixelData, BufferedImage outputImage){ 
    final int[] outputImagePixelData = ((DataBufferInt) outputImage.getRaster().getDataBuffer()).getData(); 
    System.arraycopy(pixelData, 0, outputImagePixelData, 0, pixelData.length); 
    return outputImage; 
} 

但它不起作用，因为它需要一维数组作为参数，而不是二维数组。

Answer 1

如果您outputImage已具有良好的类型和格式，那么你可以简单地做使用循环一个2D到1D转换（假设你的阵列编码为[nbRows] [RowLength]）：

private BufferedImage createImage(int[][] pixelData, BufferedImage outputImage) 
    { 
    int[] outputImagePixelData = ((DataBufferInt) outputImage.getRaster().getDataBuffer()).getData() ; 

    final int width = outputImage.getWidth() ; 
    final int height = outputImage.getHeight() ; 

    for (int y=0, pos=0 ; y < height ; y++) 
     for (int x=0 ; x < width ; x++, pos++) 
      outputImagePixelData[pos] = pixelData[y][x] ; 

    return outputImage; 
    } 

但是， INT类型在BufferedImage中没有很好定义。默认情况下，您有TYPE_INT_RGB和TYPE_INT_ARGB，它们将像素编码的R，G，B，A值与单个INT连接起来。如果你想创建类型INT与单通道灰度级的BufferedImage，那么你应该做的：

private BufferedImage createImage(int[][] pixelData) 
    { 
    final int width = pixelData[0].length ; 
    final int height = pixelData.length ; 
    // First I create a BufferedImage with a DataBufferInt, with the appropriate dimensions and number of channels/bands/colors 
    ColorSpace myColorSpace = new FloatCS(ColorSpace.TYPE_GRAY, channel) ; 
    int[] bits = new int[]{32} ; 
    ColorModel myColorModel = new ComponentColorModel(myColorSpace,bits,false,false,ColorModel.OPAQUE,DataBuffer.TYPE_INT) ; 
    BufferedImage outputImage = new BufferedImage(myColorModel, myColorModel.createCompatibleWritableRaster(width, height), false, null) ; 

    int[] outputImagePixelData = ((DataBufferInt) outputImage.getRaster().getDataBuffer()).getData() ; 

    for (int y=0, pos=0 ; y < height ; y++) 
     for (int x=0 ; x < width ; x++, pos++) 
      outputImagePixelData[pos] = pixelData[y][x] ; 

    return outputImage ; 
    } 

随着FloatCS是ColorSpace类。当你想要特定的ColorSpace像Lab，HLS等时，你必须创建你自己的ColorSpace类。

public class FloatCS extends ColorSpace 
{ 

private static final long serialVersionUID = -7713114653902159981L; 

private ColorSpace rgb = ColorSpace.getInstance(ColorSpace.CS_sRGB) ; 

public FloatCS(int type, int channel) 
    { 
    super(type, channel) ; 
    } 


@Override 
public float[] fromCIEXYZ(float[] pixel) 
    { 
    return fromRGB(rgb.fromCIEXYZ(pixel)) ; 
    } 

@Override 
public float[] fromRGB(float[] RGB) 
    { 
    return RGB ; 
    } 

@Override 
public float[] toCIEXYZ(float[] pixel) 
    { 
    return rgb.toCIEXYZ(toRGB(pixel)) ; 
    } 

@Override 
public float[] toRGB(float[] nRGB) 
    { 
    return nRGB ; 
    } 
} 

Answer 2

代替使用System.arraycopy的，可以简单地使用一个双for -loop和迭代x/y，这样的事情：

private BufferedImage createImage(int[][] pixelData, BufferedImage outputImage) { 
    int[] outputImagePixelData = ((DataBufferInt) outputImage.getRaster().getDataBuffer()).getData(); 

    int width = outputImage.getWidth(); 
    int height = outputImage.getHeight(); 

    for (int y = 0; y < height; y++) { 
     for (int x = 0; x < width; x++) { 
      outputImagePixelData[y * width + x] = pixelData[x][y]; 
     } 
    } 

    return outputImage; 
} 

上述代码假定一个数组元素对应于一个像素元件（对于所有BufferedImage.TYPE_INT_*类型都是这种情况）。