Java: BufferedImage到字节数组并返回

Question

我看到很多人都有类似的问题，但是我还没有尝试找到我正在寻找的东西。

那么，我有一个方法读取输入图像并将其转换为字节数组:

    File imgPath = new File(ImageName);
    BufferedImage bufferedImage = ImageIO.read(imgPath);
    WritableRaster raster = bufferedImage .getRaster();
    DataBufferByte data   = (DataBufferByte) raster.getDataBuffer();

我现在要做的是将其转换回BufferedImage(我有一个应用程序需要此功能)。注意"test"是字节数组。

    BufferedImage img = ImageIO.read(new ByteArrayInputStream(test));
    File outputfile = new File("src/image.jpg");
    ImageIO.write(img,"jpg",outputfile);

但是，这会返回以下异常:

    Exception in thread "main" java.lang.IllegalArgumentException: im == null!

这是因为buffereimage为空。我认为这与我从BufferedImage到字节数组的原始转换有关，信息被更改/丢失，因此数据不再被识别为jpg。

谁有什么建议，如何解决这个问题?非常感谢。

Answer 1

建议转换为字节数组

ByteArrayOutputStream baos = new ByteArrayOutputStream();
ImageIO.write(img, "jpg", baos);
byte[] bytes = baos.toByteArray();

Answer 2

请注意，调用close或flush将不会做任何事情，您可以通过查看它们的源代码/doc自己看到这一点:

关闭ByteArrayOutputStream没有效果。

OutputStream的flush方法不执行任何操作。

可以这样写:

ByteArrayOutputStream baos = new ByteArrayOutputStream(THINK_ABOUT_SIZE_HINT);
boolean foundWriter = ImageIO.write(bufferedImage, "jpg", baos);
assert foundWriter; // Not sure about this... with jpg it may work but other formats ?
byte[] bytes = baos.toByteArray();

这里有一些关于尺寸提示的链接:

• Java: Memory efficient ByteArrayOutputStream
• jpg位/像素

当然，一定要阅读你使用的版本的源代码和文档，不要盲目依赖SO答案