java:从sRGB字节流(byte[])创建BufferedImage

有时候我们需要从字节流byte[]创建一个BufferedImage对象,比如将jni层返回的图像数据转为BufferedImage对象并显示出来。BufferedImage类中的BufferedImage(java.awt.image.ColorModel, WritableRaster, boolean, java.util.Hashtable)构造函数可以满足这个要求。
不过你看到这个构造函数，所要求的参数完全不是byte[]，所以需要做一些对象创建的工作才能达到我们的目的。
以RGB格式的图像矩阵数据为例，首先要构造 sRGB标准的ColorModel对象，然后再从存储图像矩阵的字节数组(byte[])构造WritableRaster。做完这两步就可以调用BufferedImage(java.awt.image.ColorModel, WritableRaster, boolean, java.util.Hashtable)构造生成 BufferedImage对象了。
下面是完整的代码：

package test;

import java.awt.Transparency;
import java.awt.color.ColorSpace;
import java.awt.image.BufferedImage;
import java.awt.image.ComponentColorModel;
import java.awt.image.DataBuffer;
import java.awt.image.DataBufferByte;
import java.awt.image.Raster;
import java.awt.image.WritableRaster;

public class RGBtoBufferedImage {
	/**
	 * 根据指定的参数创建一个RGB格式的BufferedImage
	 * @param matrixRGB RGB格式的图像矩阵
	 * @param width 图像宽度
	 * @param height 图像高度
	 * @return
	 * @see DataBufferByte#DataBufferByte(byte[], int)
	 * @see ColorSpace#getInstance(int)
	 * @see ComponentColorModel#ComponentColorModel(ColorSpace, boolean, boolean, int, int)
	 * @see Raster#createInterleavedRaster(DataBuffer, int, int, int, int, int[], java.awt.Point)
	 * @see BufferedImage#BufferedImage(java.awt.image.ColorModel, WritableRaster, boolean, java.util.Hashtable)
	 */
	public static BufferedImage createRGBImage(byte[] matrixRGB,int width,int height){
		// 检测参数合法性
		if(null==matrixRGB||matrixRGB.length!=width*height*3)
			throw new IllegalArgumentException("invalid image description");
		// 将byte[]转为DataBufferByte用于后续创建BufferedImage对象
	    DataBufferByte dataBuffer = new DataBufferByte(matrixRGB, matrixRGB.length);
	    // sRGB色彩空间对象
        ColorSpace cs = ColorSpace.getInstance(ColorSpace.CS_sRGB);
        int[] nBits = {8, 8, 8};
        int[] bOffs = {0, 1, 2};
        ComponentColorModel colorModel = new ComponentColorModel(cs, nBits, false, false,
                                             Transparency.OPAQUE,
                                             DataBuffer.TYPE_BYTE);		   
        WritableRaster raster = Raster.createInterleavedRaster(dataBuffer, width, height, width*3, 3, bOffs, null);
        BufferedImage newImg = new BufferedImage(colorModel,raster,false,null);
       /* try {
        	//写入文件测试查看结果
			ImageIO.write(newImg, "bmp", new File(System.getProperty("user.dir"),"test.bmp"));
		} catch (IOException e) {
			e.printStackTrace();
		}*/
	    return  newImg;	    
	}
}

如果你的图像数据是Gray或ARGB格式的，如何构造一个BufferedImage对象呢？
其实也差不多，
可以参照BufferedImage中构造函数BufferedImage(int width, int height, int imageType)的源码，耐心研究一下就明白了。