java:BufferedImage判断图像通道顺序并转RGB/BGR

10km

已于 2024-01-02 11:09:33 修改

阅读量6.9k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： java 文章标签： java ImageIO RGB BGR

于 2017-03-23 16:06:50 首次发布

本文链接：https://blog.youkuaiyun.com/10km/article/details/65444680

java 专栏收录该内容

252 篇文章

订阅专栏

本文介绍如何使用Java ImageIO处理图像并将其转换为BGR格式。针对OpenCV处理需求，提供具体方法实现图像格式从RGB到BGR的转换。

一般来说java ImageIO处理读取图像时，一般是RGB或ARGB格式，但是有的时候，我们需要图像是BGR格式，

比如通过JNI将图像矩阵传递给动态库，动态库里用OpenCV来处理矩阵，而用OpenCV处理图像时默认通道顺序是BGR，这时就需要一个到BGR转换。

翻了好Java API好久，还真没发现有直接将RGB转BGR的方法，于是只好自己写一个，以下是代码片段，用于实现判断BufferedImage图像类型及通道顺序，并将BufferedImage转为RGB或BGR

	/**
	 * @param image
	 * @param bandOffset 用于判断通道顺序
	 * @return
	 */
	private static boolean equalBandOffsetWith3Byte(BufferedImage image,int[] bandOffset){
		if(image.getType()==BufferedImage.TYPE_3BYTE_BGR){
			if(image.getData().getSampleModel() instanceof ComponentSampleModel){
				ComponentSampleModel sampleModel = (ComponentSampleModel)image.getData().getSampleModel();
				if(Arrays.equals(sampleModel.getBandOffsets(), bandOffset)){
					return true;
				}
			}
		}
		return false;		
	}
	/**
	 * 判断图像是否为BGR格式
	 * @return 
	 */
	public static boolean isBGR3Byte(BufferedImage image){
		return equalBandOffsetWith3Byte(image,new int[]{0, 1, 2});
	}
	/**
	 * 判断图像是否为RGB格式
	 * @return 
	 */
	public static boolean isRGB3Byte(BufferedImage image){
		return equalBandOffsetWith3Byte(image,new int[]{2, 1, 0});
	}
	/**
	 * 对图像解码返回RGB格式矩阵数据
	 * @param image
	 * @return 
	 */
	public static byte[] getMatrixRGB(BufferedImage image) {
		if(null==image)
			throw new NullPointerException();
		byte[] matrixRGB;
		if(isRGB3Byte(image)){
			matrixRGB= (byte[]) image.getData().getDataElements(0, 0, image.getWidth(), image.getHeight(), null);
		}else{
			// 转RGB格式
			BufferedImage rgbImage = new BufferedImage(image.getWidth(), image.getHeight(),  
	                BufferedImage.TYPE_3BYTE_BGR);
			new ColorConvertOp(ColorSpace.getInstance(ColorSpace.CS_sRGB), null).filter(image, rgbImage);
			matrixRGB= (byte[]) rgbImage.getData().getDataElements(0, 0, image.getWidth(), image.getHeight(), null);
		} 
		return matrixRGB;
	}
	/**
	 * 对图像解码返回BGR格式矩阵数据
	 * @param image
	 * @return
	 */
	public static byte[] getMatrixBGR(BufferedImage image){
		if(null==image)
			throw new NullPointerException();
		byte[] matrixBGR;
		if(isBGR3Byte(image)){
			matrixBGR= (byte[]) image.getData().getDataElements(0, 0, image.getWidth(), image.getHeight(), null);
		}else{			
			// ARGB格式图像数据
			int intrgb[]=image.getRGB(0, 0, image.getWidth(), image.getHeight(), null, 0, image.getWidth());
			matrixBGR=new byte[image.getWidth() * image.getHeight()*3];
			// ARGB转BGR格式
			for(int i=0,j=0;i<intrgb.length;++i,j+=3){
				matrixBGR[j]=(byte) (intrgb[i]&0xff);
				matrixBGR[j+1]=(byte) ((intrgb[i]>>8)&0xff);
				matrixBGR[j+2]=(byte) ((intrgb[i]>>16)&0xff);
			}
		} 
		return matrixBGR;
	}