java : 调用ImageIO.writer从BufferedImage生成jpeg图像的坑

生成jpeg图像这是个非常非常简单的东西了，网上很多介绍是直接用com.sun.image.codec.jpeg.JPEGImageEncoder来实现，如下：

	/**
	 * 将原图压缩生成jpeg格式的数据
	 * @param source
	 * @return
	 */
	public static byte[] wirteJPEGBytes(BufferedImage source){
		if(null==source)
			throw new NullPointerException();
		ByteArrayOutputStream output = new ByteArrayOutputStream();
		JPEGImageEncoder jencoder = JPEGCodec.createJPEGEncoder(output);
		JPEGEncodeParam param = jencoder.getDefaultJPEGEncodeParam(source);
		param.setQuality(0.75f, true);
		jencoder.setJPEGEncodeParam(param);
		try {
			jencoder.encode(source);
		} catch (ImageFormatException e) {
			throw new RuntimeException(e);
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
		return output.toByteArray();	
	}

JPEGImageEncoder只是sun的jpeg编码实现，并不是标准的java API,只在sun jvm中被支持，但在其他的jvm上，并不会被支持。
而且，虽然上面的代码在java 1.6,1.7上都能正常执行，但在如果使用java 1.8，上面这个代码会报错：

访问限制：由于对必需的库 C:\Program Files\Java\jdk1.8.0_111\jre\lib\rt.jar 具有一定限制，因此无法访问类型 JPEGImageEncoder

所以这个方法是有局限性的。
走捷径是不行的，还是得规规矩矩按java的规范来做，ImageIO类中提供了ImageIO.writer方法可以生成指定的格式的图像，才是正规的实现方式。
但是使用ImageIO.writer方法也是有讲究的。
我原先是这样写的，就是简单的调用ImageIO.writer方法生成jpeg数据：

	/**
	 * 将原图压缩生成jpeg格式的数据
	 * @param source
	 * @return
	 * @see #wirteBytes(BufferedImage, String)
	 */
	public static byte[] wirteJPEGBytes(BufferedImage source){
		return wirteBytes(source,"JPEG");
	}
	/**
	 * 将原图压缩生成jpeg格式的数据
	 * @param source
	 * @return
	 * @see #wirteBytes(BufferedImage, String)
	 */
	public static byte[] wirteJPEGBytes(BufferedImage source){
		return wirteBytes(source,"JPEG");
	}
	/**
	 * 将{@link BufferedImage}生成formatName指定格式的图像数据
	 * @param source
	 * @param formatName 图像格式名，图像格式名错误则抛出异常
	 * @return
	 */
	public static byte[] wirteBytes(BufferedImage source,String formatName){
		Assert.notNull(source, "source");
		Assert.notEmpty(formatName, "formatName");
		ByteArrayOutputStream output = new ByteArrayOutputStream();
		try {			
			if(!ImageIO.write(source, formatName.toLowerCase(), output))
				// 返回false则抛出异常
				throw new IllegalArgumentException(String.format("not found writer for '%s'",formatName));
			
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
		return output.toByteArray();		
	}

处理了几万张图像文件都没问题，遇到一张png图像，ImageIO.write居然返回false,抛出异常了。
究其原因，是ImageIO.wite方法在中调用的私有方法getWriter寻找合适的ImageWriter时不仅与formatName相关，还是输入的原图有关(具体是怎么相关的，因为逻辑关系太复杂没有深究),造成getWriter方法找不到对应的ImageWriter。
参考网上别人的写法改成这样就没问题了：

	/**
	 * 将{@link BufferedImage}生成formatName指定格式的图像数据
	 * @param source
	 * @param formatName 图像格式名，图像格式名错误则抛出异常
	 * @return
	 */
	public static byte[] wirteBytes(BufferedImage source,String formatName){
		Assert.notNull(source, "source");
		Assert.notEmpty(formatName, "formatName");
		ByteArrayOutputStream output = new ByteArrayOutputStream();
		BufferedImage newBufferedImage = new BufferedImage(source.getWidth(),
				source.getHeight(), BufferedImage.TYPE_INT_RGB);
		Graphics2D g = newBufferedImage.createGraphics();
		try {
			g.drawImage(source, 0, 0,null);
			if(!ImageIO.write(newBufferedImage, formatName, output))
				throw new IllegalArgumentException(String.format("not found writer for '%s'",formatName));
		} catch (IOException e) {
			throw new RuntimeException(e);
		}finally{
			g.dispose();
		}
		return output.toByteArray();		
	}

基本的思路就是重创建一个大小相同的BufferedImage，然后用Graphics.drawImage方法将原图写入新的BufferedImage对象，通过这一道转换，抹平了，不同类型图像格式生成的BufferedImage对象之间的区别，再调用 ImageIO.write 对新的ImageIO.write对象进行图像处理就不会有问题了。

改进

在我的项目中图像数据是从互联网上搜索到的，遇到的图像格式绝大部分都是jpeg,但也有少量的png,bmp等格式，对于占绝大多数的jpeg图像来说，我最开始的方法都是有效的，而上面的这个方法多出一道工序就显得有些多余，还浪费资源，所以又改进了上述的方法，基本的原理就是先尝试直接ImageIO.write来生成jpeg，如果失败，就用第二种方式。

	/**
	 * 将{@link BufferedImage}生成formatName指定格式的图像数据
	 * @param source
	 * @param formatName 图像格式名，图像格式名错误则抛出异常
	 * @return
	 */
	public static byte[] wirteBytes(BufferedImage source,String formatName){
		Assert.notNull(source, "source");
		Assert.notEmpty(formatName, "formatName");
		ByteArrayOutputStream output = new ByteArrayOutputStream();
		Graphics2D g = null;
		try {
			for(BufferedImage s=source;!ImageIO.write(s, formatName, output);){
				if(null!=g)
					throw new IllegalArgumentException(String.format("not found writer for '%s'",formatName));
				s = new BufferedImage(source.getWidth(),
						source.getHeight(), BufferedImage.TYPE_INT_RGB);
				g = s.createGraphics();
				g.drawImage(source, 0, 0,null);				
			}				
		} catch (IOException e) {
			throw new RuntimeException(e);
		} finally {
			if (null != g)
				g.dispose();
		}
		return output.toByteArray();		
	}