Java 使用 BufferedImage 从 URL 获取图像扩展/类型
我熟悉处理图像。我从URL中检索/读取图像,其中URL没有文件扩展名。然后,我希望将图像写入/保存到本地存储,但我必须指定图像文件扩展名(即.JPG,PNG等),我无法通过BufferedImage检索其扩展名。
能否指出如何做到这一点?任何其他方法都可以。
我熟悉处理图像。我从URL中检索/读取图像,其中URL没有文件扩展名。然后,我希望将图像写入/保存到本地存储,但我必须指定图像文件扩展名(即.JPG,PNG等),我无法通过BufferedImage检索其扩展名。
能否指出如何做到这一点?任何其他方法都可以。
Use ImageReader.getFormatName()
您可以使用 ImageIO.getImageReaders(Object input) 获取文件的图像读取器。
我自己还没有测试过,但你可以试试这个:
ImageInputStream iis = ImageIO.createImageInputStream(file);
Iterator<ImageReader> imageReaders = ImageIO.getImageReaders(iis);
while (imageReaders.hasNext()) {
ImageReader reader = (ImageReader) imageReaders.next();
System.out.printf("formatName: %s%n", reader.getFormatName());
}
如果对象是 URL,则使用 ImageIO.createImageInputStream(obj) 的建议将不起作用。
一种替代方法是使用URLConnection.guessContentTypeFromStream(InputStream stream)方法。此方法通过检查流的前 12 个字节来猜测内容类型。
使用此方法的一个复杂之处在于,它要求标记支持的给定流参数,并且不支持 java url.openStream() 返回的流。
此外,如果要确定内容类型并将图像下载到 BufferedImage,则最好是解决方案仅下载内容一次(而不是进行两次传递,一次用于确定内容类型,第二次用于下载图像)。
一种解决方案是使用 PushbackInputStream。PushbackInputStream 可用于下载第一个初始字节以确定内容类型。然后,可以将字节推回流上,以便 ImageIO.read(流) 可以完整地读取流。
可能的解决方案:
// URLConnection.guessContentTypeFromStream only needs the first 12 bytes, but
// just to be safe from future java api enhancements, we'll use a larger number
int pushbackLimit = 100;
InputStream urlStream = url.openStream();
PushbackInputStream pushUrlStream = new PushbackInputStream(urlStream, pushbackLimit);
byte [] firstBytes = new byte[pushbackLimit];
// download the first initial bytes into a byte array, which we will later pass to
// URLConnection.guessContentTypeFromStream
pushUrlStream.read(firstBytes);
// push the bytes back onto the PushbackInputStream so that the stream can be read
// by ImageIO reader in its entirety
pushUrlStream.unread(firstBytes);
String imageType = null;
// Pass the initial bytes to URLConnection.guessContentTypeFromStream in the form of a
// ByteArrayInputStream, which is mark supported.
ByteArrayInputStream bais = new ByteArrayInputStream(firstBytes);
String mimeType = URLConnection.guessContentTypeFromStream(bais);
if (mimeType.startsWith("image/"))
imageType = mimeType.substring("image/".length());
// else handle failure here
// read in image
BufferedImage inputImage = ImageIO.read(pushUrlStream);