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

来源:互联网 发布:linux启动nodejs项目 编辑:程序博客网 时间:2024/05/29 05:53

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)方法实现代码也差不多,你可以自己选择。最终的结果都是一样的。

3 0