java:构建ImageInputStream利用ImageReader对内存字节流进行图像解码

版权声明:本文为博主原创文章,转载请注明源地址。 https://blog.csdn.net/10km/article/details/52119508

java提供了一个非常方便的图像工具类javax.imageio.ImageIO,用它的javax.imageio.ImageIO.read方法可以很方便的将一个图像文件进行解码。 javax.imageio.ImageIO.read方法有多个重载方法,支持File,InputStream,URL等参数,但这些方法有可能会在解码过程中使用文件系统做cache,具体原因这里不展开讲了,好长,你要研究java源码。 有了磁盘IO势必会影响解码效率,这在性能敏感的应用环境是不能容忍的, 如果要实现完全基于内存的图像解码,就不能简单使用javax.imageio.ImageIO.read方法。需要利用javax.imageio.stream.MemoryCacheImageInputStream来实现内存cache。来实现完全的内存解码,以下是完整的代码,

package test;

import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Iterator;

import javax.imageio.ImageIO;
import javax.imageio.ImageReader;
import javax.imageio.stream.ImageInputStream;
import javax.imageio.stream.MemoryCacheImageInputStream;

import org.junit.Test;

public class TestReadMemoryImage {
    @Test
    public final void testreadMemoryImage() throws IllegalArgumentException, IOException {
        // 将图像文件加读取到内存成为字节数组
        byte[] imgBytes = readBytes(TestReadMemoryImage.class.getResourceAsStream("/images/he049.jpg"));
        BufferedImage bufImg = readMemoryImage(imgBytes);
        System.out.printf("decode success,width=%d,heigh=%d\n", bufImg.getWidth(),bufImg.getHeight());
    }
    @Test
    public final void testreadMemoryImage1() throws IllegalArgumentException, IOException {
        // 将图像文件加读取到内存成为字节数组
        byte[] imgBytes = readBytes(TestReadMemoryImage.class.getResourceAsStream("/images/he049.jpg"));
        BufferedImage bufImg = readMemoryImage1(imgBytes);
        System.out.printf("decode success,width=%d,heigh=%d\n", bufImg.getWidth(),bufImg.getHeight());
    }
    /**
     * 从内存字节数组中读取图像
     * 
     * @param imgBytes
     *            未解码的图像数据
     * @return 返回 {@link BufferedImage}
     * @throws IOException
     *             当读写错误或不识别的格式时抛出
     */
    public static final BufferedImage readMemoryImage(byte[] imgBytes) throws IOException {
        if (null == imgBytes || 0 == imgBytes.length)
            throw new NullPointerException("the argument 'imgBytes' must not be null or empty");
        // 将字节数组转为InputStream,再转为MemoryCacheImageInputStream
        ImageInputStream imageInputstream = new MemoryCacheImageInputStream(new ByteArrayInputStream(imgBytes));
        // 获取所有能识别数据流格式的ImageReader对象
        Iterator<ImageReader> it = ImageIO.getImageReaders(imageInputstream);
        // 迭代器遍历尝试用ImageReader对象进行解码
        while (it.hasNext()) {
            ImageReader imageReader = it.next();
            // 设置解码器的输入流
            imageReader.setInput(imageInputstream, true, true);
            // 图像文件格式后缀
            String suffix = imageReader.getFormatName().trim().toLowerCase();
            // 图像宽度
            int width = imageReader.getWidth(0);
            // 图像高度
            int height = imageReader.getHeight(0);
            System.out.printf("format %s,%dx%d\n", suffix, width, height);
            try {
                // 解码成功返回BufferedImage对象
                // 0即为对第0张图像解码(gif格式会有多张图像),前面获取宽度高度的方法中的参数0也是同样的意思
                 return imageReader.read(0, imageReader.getDefaultReadParam());
            } catch (Exception e) {
                imageReader.dispose();
                // 如果解码失败尝试用下一个ImageReader解码
            }
        }
        imageInputstream.close();
        // 没有能识别此数据的图像ImageReader对象,抛出异常
        throw new IOException("unsupported image format");
    }
    public static final BufferedImage readMemoryImage1(byte[] imgBytes) throws IOException {
        if (null == imgBytes || 0 == imgBytes.length)
            throw new NullPointerException("the argument 'imgBytes' must not be null or empty");
        // 将字节数组转为InputStream,再转为MemoryCacheImageInputStream
        ImageInputStream imageInputstream = new MemoryCacheImageInputStream(new ByteArrayInputStream(imgBytes));
        // 直接调用ImageIO.read方法解码
        BufferedImage bufImg = ImageIO.read(imageInputstream);
        if(null==bufImg)
            // 没有能识别此数据的图像ImageReader对象,抛出异常
            throw new IOException("unsupported image format");
        return bufImg;
    }
    /**
     * 从{@link InputStream}读取字节数组<br>
     * 结束时会关闭{@link InputStream}<br>
     * {@code in}为{@code null}时抛出{@link NullPointerException}
     * 
     * @param in
     * @return 字节数组
     * @throws IOException
     */
    public static final byte[] readBytes(InputStream in) throws IOException {
        if (null == in)
            throw new NullPointerException("the argument 'in' must not be null");
        try {
            int buffSize = Math.max(in.available(), 1024 * 8);
            byte[] temp = new byte[buffSize];
            ByteArrayOutputStream out = new ByteArrayOutputStream(buffSize);
            int size = 0;
            while ((size = in.read(temp)) != -1) {
                out.write(temp, 0, size);
            }
            return out.toByteArray();
        } finally {
            in.close();
        }
    }
}

代码中提供了两个方法来实现内存解码(readMemoryImagereadMemoryImage1),共同点就是都要基于byte数组构建一个MemoryCacheImageInputStream对象作为 ImageInputStream。 不同点就是readMemoryImage1方法直接使用javax.imageio.ImageIO.read(ImageInputStream stream)方法来解码。 而readMemoryImage则是寻找合适的ImageReader来实现解码, 其实逻辑上与javax.imageio.ImageIO.read(ImageInputStream stream)方法实现代码也差不多,你可以自己选择。最终的结果都是一样的。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏程序猿DD

Spring Boot 2.x基础教程:快速入门

在您第1次接触和学习Spring框架的时候,是否因为其繁杂的配置而退却了?在你第n次使用Spring框架的时候,是否觉得一堆反复黏贴的配置有一些厌烦?那么您就不...

21220
来自专栏代码男人

让JNI告诉你 你的应用为什么被卸载

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/huangliniqng/article/detail...

11340
来自专栏云计算与大数据

Configuring Spring Boot's Server, GZip compression, HTTP/2

Spring Boot is powerful yet flexible. It tries to auto-configure most of the stu...

34650
来自专栏Java学习录

Redis中的事务与Lua脚本

Redis本身提供了multi关键字用来开启事务,exec用来关闭事务。Redis这两个关键字之间的操作是原子性的。

27450
来自专栏移动开发之家

Flutter完整开发实战详解(十一、全面深入理解Stream)

Stream 在 Flutter 是属于非常关键的概念,在 Flutter 中,状态管理除了 InheritedWidget 之外,无论 rxdart,Bloc...

26040
来自专栏那些年我们学过的前端

Rollup

30920
来自专栏京程一灯

在同一基准下对前端框架进行比较[每日前端夜话0x58]

翻译:疯狂的技术宅 原文:https://medium.freecodecamp.org/a-realworld-comparison-of-front-end...

12220
来自专栏成猿之路

使用lombok编写优雅的Bean对象

使用java编写代码,十之八九都是在写java类,从而构建java对象。lombok之前也说了不少,但使用了这么多年,感觉还是有很多技巧可以使用的。

11130
来自专栏信安之路

burp 日志插件从原理到实践

Logger++ 是 nccgroup 开源的一个 burp 扩展,主要功能是记录经过 Burp Suite 的所有 HTTP 请求 和 HTTP 响应。

30620
来自专栏业余草

Java实现图片上传到服务器,并把上传的图片读取出来

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/xmt1139057136/article/detai...

86930

扫码关注云+社区

领取腾讯云代金券

年度创作总结 领取年终奖励