Read EXIF data and apply image rotation (#3413)

This commit is contained in:
Laurenz 2024-02-14 15:12:28 +01:00 committed by GitHub
parent fcf64d0ee0
commit b89348b92a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 60 additions and 2 deletions

16
Cargo.lock generated
View File

@ -1175,6 +1175,15 @@ dependencies = [
"wasm-bindgen",
]
[[package]]
name = "kamadak-exif"
version = "0.5.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ef4fc70d0ab7e5b6bafa30216a6b48705ea964cdfc29c050f2412295eba58077"
dependencies = [
"mutate_once",
]
[[package]]
name = "kqueue"
version = "1.0.8"
@ -1375,6 +1384,12 @@ dependencies = [
"windows-sys 0.48.0",
]
[[package]]
name = "mutate_once"
version = "0.1.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "16cf681a23b4d0a43fc35024c176437f9dcd818db34e0f42ab456a0ee5ad497b"
[[package]]
name = "native-tls"
version = "0.2.11"
@ -2556,6 +2571,7 @@ dependencies = [
"icu_segmenter",
"image",
"indexmap 2.1.0",
"kamadak-exif",
"kurbo",
"lipsum",
"log",

View File

@ -58,6 +58,7 @@ if_chain = "1"
image = { version = "0.24", default-features = false, features = ["png", "jpeg", "gif"] }
include_dir = "0.7"
indexmap = { version = "2", features = ["serde"] }
kamadak-exif = "0.5"
kurbo = "0.9" # in sync with usvg
libfuzzer-sys = "0.4"
lipsum = "0.9"

BIN
assets/files/f2t.jpg Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 441 B

View File

@ -36,6 +36,7 @@ icu_provider_blob = { workspace = true }
icu_segmenter = { workspace = true }
image = { workspace = true }
indexmap = { workspace = true }
kamadak-exif = { workspace = true }
kurbo = { workspace = true }
lipsum = { workspace = true }
log = { workspace = true }

View File

@ -7,7 +7,7 @@ use image::codecs::gif::GifDecoder;
use image::codecs::jpeg::JpegDecoder;
use image::codecs::png::PngDecoder;
use image::io::Limits;
use image::{guess_format, ImageDecoder, ImageResult};
use image::{guess_format, DynamicImage, ImageDecoder, ImageResult};
use crate::diag::{bail, StrResult};
use crate::foundations::{Bytes, Cast};
@ -39,13 +39,17 @@ impl RasterImage {
}
let cursor = io::Cursor::new(&data);
let (dynamic, icc) = match format {
let (mut dynamic, icc) = match format {
RasterFormat::Jpg => decode_with(JpegDecoder::new(cursor)),
RasterFormat::Png => decode_with(PngDecoder::new(cursor)),
RasterFormat::Gif => decode_with(GifDecoder::new(cursor)),
}
.map_err(format_image_error)?;
if let Some(rotation) = exif_rotation(&data) {
apply_rotation(&mut dynamic, rotation);
}
Ok(Self(Arc::new(Repr { data, format, dynamic, icc })))
}
@ -129,6 +133,36 @@ impl TryFrom<image::ImageFormat> for RasterFormat {
}
}
/// Get rotation from EXIF metadata.
fn exif_rotation(data: &[u8]) -> Option<u32> {
let reader = exif::Reader::new();
let mut cursor = std::io::Cursor::new(data);
let exif = reader.read_from_container(&mut cursor).ok()?;
let orient = exif.get_field(exif::Tag::Orientation, exif::In::PRIMARY)?;
orient.value.get_uint(0)
}
/// Apply an EXIF rotation to a dynamic image.
fn apply_rotation(image: &mut DynamicImage, rotation: u32) {
use image::imageops as ops;
match rotation {
2 => ops::flip_horizontal_in_place(image),
3 => ops::rotate180_in_place(image),
4 => ops::flip_vertical_in_place(image),
5 => {
ops::flip_horizontal_in_place(image);
*image = image.rotate270();
}
6 => *image = image.rotate90(),
7 => {
ops::flip_horizontal_in_place(image);
*image = image.rotate90();
}
8 => *image = image.rotate270(),
_ => {}
}
}
/// Format the user-facing raster graphic decoding error message.
fn format_image_error(error: image::ImageError) -> EcoString {
match error {

Binary file not shown.

After

Width:  |  Height:  |  Size: 296 B

View File

@ -0,0 +1,6 @@
// Ensure that EXIF rotation is applied.
// https://github.com/image-rs/image/issues/1045
---
// Files is from https://magnushoff.com/articles/jpeg-orientation/
#image("/files/f2t.jpg", width: 10pt)