java:BufferedImage判断图像通道顺序并转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;
	}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

10km

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值