-
-
Save steinarb/0ab59ad7c17b7ac15b757dd67368dad9 to your computer and use it in GitHub Desktop.
Set exif metadata when downloading a JPEG
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File downloadImageUrlToTempFile(AlbumEntry albumEntry, Path tempDir) { | |
var imageUrl = albumEntry.getImageUrl(); | |
if (imageUrl == null || imageUrl.isEmpty()) { | |
throw new OldAlbumException(String.format("Unable to download album entry matching id=%d, imageUrl is missing", albumEntry.getId())); | |
} | |
var fileName = findFileNamePartOfUrl(imageUrl); | |
var tempfile = tempDir.resolve(fileName).toFile(); | |
IIOImage image = null; | |
ImageWriter writer = null; | |
try { | |
HttpURLConnection connection = getConnectionFactory().connect(imageUrl); | |
connection.setRequestMethod("GET"); | |
try(var inputStream = ImageIO.createImageInputStream(connection.getInputStream())) { | |
var readers = ImageIO.getImageReaders(inputStream); | |
if (readers.hasNext()) { | |
var reader = readers.next(); | |
writer = ImageIO.getImageWriter(reader); | |
reader.setInput(inputStream); | |
image = reader.readAll(0, null); | |
} | |
} | |
} catch (IOException e) { | |
throw new OldAlbumException(String.format("Unable to download album entry matching id=%d from url=\"%s\"", albumEntry.getId(), albumEntry.getImageUrl()), e); | |
} | |
var metadata = image.getMetadata(); | |
var metadataAsTree = (IIOMetadataNode) metadata.getAsTree("javax_imageio_1.0"); | |
findJfifCommentNode(metadataAsTree) | |
.ifPresent(node -> node.setAttribute("value", albumEntry.getDescription())); | |
Collection<Entry> entries = new ArrayList<>(); | |
var exifDateTimeFormat = new SimpleDateFormat("yyyy:MM:dd hh:mm:ss"); | |
exifDateTimeFormat.setTimeZone(TimeZone.getTimeZone("Europe/Oslo")); | |
var datetime = exifDateTimeFormat.format(albumEntry.getLastModified()); | |
entries.add(new TIFFEntry(TIFF.TAG_DATE_TIME, datetime)); | |
var markerSequence = (IIOMetadataNode) metadataAsTree.getElementsByTagName("markerSequence").item(0); | |
if (markerSequence == null) { | |
markerSequence = new IIOMetadataNode("markerSequence"); | |
metadataAsTree.appendChild(markerSequence); | |
} | |
ByteArrayOutputStream bytes = new ByteArrayOutputStream(); | |
try { | |
bytes.write("Exif".getBytes(StandardCharsets.US_ASCII)); | |
bytes.write(new byte[2]); | |
new TIFFWriter().write(entries, new MemoryCacheImageOutputStream(bytes)); | |
IIOMetadataNode exif = new IIOMetadataNode("unknown"); | |
exif.setAttribute("MarkerTag", String.valueOf(0xE1)); // APP1 or "225" | |
exif.setUserObject(bytes.toByteArray()); | |
markerSequence.appendChild(exif); | |
} catch (IOException e) { | |
throw new OldAlbumException(String.format("Failed to create EXIF structure when saving local copy of album entry matching id=%d from url=\"%s\"", albumEntry.getId(), albumEntry.getImageUrl()), e); | |
} | |
try (var outputStream = ImageIO.createImageOutputStream(new FileOutputStream(tempfile))){ | |
writer.setOutput(outputStream); | |
var param = writer.getDefaultWriteParam(); | |
var modifiedMetadata = writer.getDefaultImageMetadata(ImageTypeSpecifiers.createFromRenderedImage(image.getRenderedImage()), param); | |
modifiedMetadata.setFromTree("javax_imageio_1.0", metadataAsTree); | |
image.setMetadata(modifiedMetadata); | |
writer.write(image); | |
Files.setLastModifiedTime(tempfile.toPath(), FileTime.from(albumEntry.getLastModified().toInstant())); | |
return tempfile; | |
} catch (IOException e) { | |
throw new OldAlbumException(String.format("Unable to save local copy of album entry matching id=%d from url=\"%s\"", albumEntry.getId(), albumEntry.getImageUrl()), e); | |
} | |
} | |
Optional<IIOMetadataNode> findJfifCommentNode(Node metadataAsTree) { | |
return StreamSupport.stream(iterable(metadataAsTree.getChildNodes()).spliterator(), false) | |
.filter(n -> "Text".equals(n.getNodeName())) | |
.findFirst() | |
.flatMap(n -> StreamSupport.stream(iterable(n.getChildNodes()).spliterator(), false).findFirst()); | |
} | |
public static Iterable<IIOMetadataNode> iterable(final NodeList nodeList) { | |
return () -> new Iterator<IIOMetadataNode>() { | |
private int index = 0; | |
@Override | |
public boolean hasNext() { | |
return index < nodeList.getLength(); | |
} | |
@Override | |
public IIOMetadataNode next() { | |
if (!hasNext()) | |
throw new NoSuchElementException(); | |
return (IIOMetadataNode) nodeList.item(index++); | |
} | |
}; | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment