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

落日映苍穹つ 2022-07-11 05:40 230阅读 0赞

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

  1. /**
  2. * 将原图压缩生成jpeg格式的数据
  3. * @param source
  4. * @return
  5. */
  6. public static byte[] wirteJPEGBytes(BufferedImage source){
  7. if(null==source)
  8. throw new NullPointerException();
  9. ByteArrayOutputStream output = new ByteArrayOutputStream();
  10. JPEGImageEncoder jencoder = JPEGCodec.createJPEGEncoder(output);
  11. JPEGEncodeParam param = jencoder.getDefaultJPEGEncodeParam(source);
  12. param.setQuality(0.75f, true);
  13. jencoder.setJPEGEncodeParam(param);
  14. try {
  15. jencoder.encode(source);
  16. } catch (ImageFormatException e) {
  17. throw new RuntimeException(e);
  18. } catch (IOException e) {
  19. throw new RuntimeException(e);
  20. }
  21. return output.toByteArray();
  22. }

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数据:

  1. /**
  2. * 将原图压缩生成jpeg格式的数据
  3. * @param source
  4. * @return
  5. * @see #wirteBytes(BufferedImage, String)
  6. */
  7. public static byte[] wirteJPEGBytes(BufferedImage source){
  8. return wirteBytes(source,"JPEG");
  9. }
  10. /**
  11. * 将原图压缩生成jpeg格式的数据
  12. * @param source
  13. * @return
  14. * @see #wirteBytes(BufferedImage, String)
  15. */
  16. public static byte[] wirteJPEGBytes(BufferedImage source){
  17. return wirteBytes(source,"JPEG");
  18. }
  19. /**
  20. * 将{@link BufferedImage}生成formatName指定格式的图像数据
  21. * @param source
  22. * @param formatName 图像格式名,图像格式名错误则抛出异常
  23. * @return
  24. */
  25. public static byte[] wirteBytes(BufferedImage source,String formatName){
  26. Assert.notNull(source, "source");
  27. Assert.notEmpty(formatName, "formatName");
  28. ByteArrayOutputStream output = new ByteArrayOutputStream();
  29. try {
  30. if(!ImageIO.write(source, formatName.toLowerCase(), output))
  31. // 返回false则抛出异常
  32. throw new IllegalArgumentException(String.format("not found writer for '%s'",formatName));
  33. } catch (IOException e) {
  34. throw new RuntimeException(e);
  35. }
  36. return output.toByteArray();
  37. }

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

  1. /**
  2. * 将{@link BufferedImage}生成formatName指定格式的图像数据
  3. * @param source
  4. * @param formatName 图像格式名,图像格式名错误则抛出异常
  5. * @return
  6. */
  7. public static byte[] wirteBytes(BufferedImage source,String formatName){
  8. Assert.notNull(source, "source");
  9. Assert.notEmpty(formatName, "formatName");
  10. ByteArrayOutputStream output = new ByteArrayOutputStream();
  11. BufferedImage newBufferedImage = new BufferedImage(source.getWidth(),
  12. source.getHeight(), BufferedImage.TYPE_INT_RGB);
  13. Graphics2D g = newBufferedImage.createGraphics();
  14. try {
  15. g.drawImage(source, 0, 0,null);
  16. if(!ImageIO.write(newBufferedImage, formatName, output))
  17. throw new IllegalArgumentException(String.format("not found writer for '%s'",formatName));
  18. } catch (IOException e) {
  19. throw new RuntimeException(e);
  20. }finally{
  21. g.dispose();
  22. }
  23. return output.toByteArray();
  24. }

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

改进

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

  1. /**
  2. * 将{@link BufferedImage}生成formatName指定格式的图像数据
  3. * @param source
  4. * @param formatName 图像格式名,图像格式名错误则抛出异常
  5. * @return
  6. */
  7. public static byte[] wirteBytes(BufferedImage source,String formatName){
  8. Assert.notNull(source, "source");
  9. Assert.notEmpty(formatName, "formatName");
  10. ByteArrayOutputStream output = new ByteArrayOutputStream();
  11. Graphics2D g = null;
  12. try {
  13. for(BufferedImage s=source;!ImageIO.write(s, formatName, output);){
  14. if(null!=g)
  15. throw new IllegalArgumentException(String.format("not found writer for '%s'",formatName));
  16. s = new BufferedImage(source.getWidth(),
  17. source.getHeight(), BufferedImage.TYPE_INT_RGB);
  18. g = s.createGraphics();
  19. g.drawImage(source, 0, 0,null);
  20. }
  21. } catch (IOException e) {
  22. throw new RuntimeException(e);
  23. } finally {
  24. if (null != g)
  25. g.dispose();
  26. }
  27. return output.toByteArray();
  28. }

发表评论

表情:
评论列表 (有 0 条评论,230人围观)

还没有评论,来说两句吧...

相关阅读