From 1dbfad04fd9f010e355b8658e7defb97e756493a Mon Sep 17 00:00:00 2001 From: "Dylan M. Taylor" Date: Fri, 13 Nov 2026 00:08:05 -0500 Subject: [PATCH] Support extension-based raster encoding --- crates/raster-codecs/src/common.rs | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/crates/raster-codecs/src/common.rs b/crates/raster-codecs/src/common.rs index b653db687..da3c1c39d 100644 --- a/crates/raster-codecs/src/common.rs +++ b/crates/raster-codecs/src/common.rs @@ -73,6 +73,16 @@ pub fn encode_jpeg(pixels: &RgbaImage, quality: u8) -> Result, CodecErro Ok(out.into_inner()) } +/// Encode pixels according to a destination filename's extension (`png`, `jpg`, or `jpeg`). +pub fn encode_for_path(path: impl AsRef, pixels: &RgbaImage) -> Result, CodecError> { + let extension = path.as_ref().extension().and_then(std::ffi::OsStr::to_str).unwrap_or_default(); + match extension.to_ascii_lowercase().as_str() { + "png" => encode_png(pixels), + "jpg" | "jpeg" => encode_jpeg(pixels, 90), + other => Err(CodecError::unsupported("image", format!("cannot encode files with the .{other} extension"))), + } +} + fn is_heif(bytes: &[u8]) -> bool { bytes.get(4..8) == Some(b"ftyp") && matches!(bytes.get(8..12), Some(b"heic" | b"heix" | b"hevc" | b"mif1" | b"msf1" | b"avif")) }