java : 调用ImageIO.writer从BufferedImage生成jpeg图像的坑
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();
}腾讯云开发者
