Java BufferedImage 转 PNG 格式 Base64 字符串

Question

我正在尝试将屏幕截图输出为 base64 编码的字符串，但并没有走得太远。到目前为止，我的代码使用 Base64 库（http://iharder.sourceforge.net/current/java/base64/）：

    Robot robot = new Robot();
    Rectangle r = new Rectangle( Toolkit.getDefaultToolkit().getScreenSize() );
    BufferedImage bi = robot.createScreenCapture(r);
    ByteArrayOutputStream os = new ByteArrayOutputStream();
    OutputStream b64 = new Base64.OutputStream(os);
    ImageIO.write(bi, "png", os);
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    out.writeTo(b64);
    String result = out.toString("UTF-8");

每次我运行这个，“结果”总是一个空字符串，但我不明白为什么。有什么想法吗？

注意：我不想将 png 写入磁盘上的文件。

Answer 1

我关注了xehpuk's 的回答，但在某些浏览器中通过数据 url 渲染时，某些图像的最后几行像素丢失（Chrome 和 Firefox，Safari 似乎可以很好地渲染它们）。我怀疑这是因为浏览器正在尽最大努力解释数据，但最后几个字节的数据丢失了，所以它显示了它可以做什么。

输出流的包装似乎是导致此问题的原因。 Base64.wrap(OutputStream os) 的文档说明：

使用后建议及时关闭返回的输出流，在此期间它会将所有可能的剩余字节刷新到底层输出流中。

因此，根据数据的长度，最后几个字节可能不会从流中刷新，因为没有调用 close()。我对此的解决方案是不用费心包装流，直接对流进行编码：

public static String imgToBase64String(final RenderedImage img, final String formatName)
{
  final ByteArrayOutputStream os = new ByteArrayOutputStream();

  try
  {
    ImageIO.write(img, formatName, os);
    return Base64.getEncoder().encodeToString(os.toByteArray());
  }
  catch (final IOException ioe)
  {
    throw new UncheckedIOException(ioe);
  }
}

这解决了在浏览器中呈现时缺少像素行的问题。

Answer 2

以下语句的工作方向错误：

out.writeTo(b64);

它用out的空字节数组覆盖Base 64数据。

out 的目的到底是什么？我认为你不需要它。

更新：

并且您将图像直接写入os，而不是通过 Base 64 编码器写入。

以下代码应该可以工作：

...
ByteArrayOutputStream os = new ByteArrayOutputStream();
OutputStream b64 = new Base64.OutputStream(os);
ImageIO.write(bi, "png", b64);
String result = os.toString("UTF-8");

Answer 3

使用 Java 8 对图像进行 Base64 编码和解码：

public static String imgToBase64String(final RenderedImage img, final String formatName) {
    final ByteArrayOutputStream os = new ByteArrayOutputStream();
    try (final OutputStream b64os = Base64.getEncoder().wrap(os)) {
        ImageIO.write(img, formatName, b64os);
    } catch (final IOException ioe) {
        throw new UncheckedIOException(ioe);
    }
    return os.toString();
}

public static BufferedImage base64StringToImg(final String base64String) {
    try {
        return ImageIO.read(new ByteArrayInputStream(Base64.getDecoder().decode(base64String)));
    } catch (final IOException ioe) {
        throw new UncheckedIOException(ioe);
    }
}

在您的屏幕截图场景中像这样使用它：

final Robot robot = new Robot();
final Rectangle r = new Rectangle(Toolkit.getDefaultToolkit().getScreenSize());
final BufferedImage bi = robot.createScreenCapture(r);
final String base64String = imgToBase64String(bi, "png");

Answer 4

这对我有用：

将图像编码为 Base64 字符串

public static String encodeToString(BufferedImage image, String type) {
    String imageString = null;
    ByteArrayOutputStream bos = new ByteArrayOutputStream();

    try {
        ImageIO.write(image, type, bos);
        byte[] imageBytes = bos.toByteArray();

        Base64.Encoder encoder = Base64.getEncoder();
        imageString = encoder.encodeToString(imageBytes);

        bos.close();
    } catch (IOException e) {
        e.printStackTrace();
    }
    return imageString;
}

将 Base64 字符串解码为图像

public static BufferedImage decodeToImage(String imageString) {
    BufferedImage image = null;
    byte[] imageByte;
    try {
        Base64.Decoder decoder = Base64.getDecoder();
        imageByte = decoder.decode(imageString);
        ByteArrayInputStream bis = new ByteArrayInputStream(imageByte);
        image = ImageIO.read(bis);
        bis.close();
    } catch (Exception e) {
        e.printStackTrace();
    }
    return image;
}

Answer 5

实际上，两种不同解决方案的组合对我有用。我的用例是从 zip 文件中读取图像。这是对我有用的代码：

BufferedImage bgTileSprite = ImageIO.read(inputStream);
ByteArrayOutputStream os = new ByteArrayOutputStream();
ImageIO.write(bgTileSprite, "png", os);
String result = Base64.getEncoder().encodeToString(os.toByteArray());
LOGGER.info(result);

我通过将结果转换为实际图像来确认结果。效果很好。