feat(cli): added avif codec

src/cli/codecs/avif.rs

@@ -0,0 +1,27 @@
+use clap::{arg, value_parser, Command};
+use indoc::indoc;
+
+pub fn avif() -> Command {
+    Command::new("avif")
+        .about("Encode images into AVIF format")
+        .args([
+            arg!(-q --quality <NUM> "Quality which the image will be encoded with")
+                .value_parser(value_parser!(u8).range(1..=100))
+                .default_value("50"),
+            arg!(--alpha_quality <NUM> "Separate alpha quality which the image will be encoded with")
+                .value_parser(value_parser!(u8).range(1..=100)),
+            arg!(--speed <NUM> "Compression speed (effort)")
+                .long_help(indoc! {"Compression speed (effort)
+
+                1 = very very slow, but max compression
+                10 = quick, but larger file sizes and lower quality."})
+                .value_parser(value_parser!(u8).range(1..=10))
+                .default_value("6"),
+            arg!(--colorspace <COLOR> "Set color space of AVIF being written")
+                .value_parser(["ycbcr", "rgb"])
+                .default_value("ycbcr"),
+            arg!(--alpha_mode <MODE> "Configure handling of color channels in transparent images")
+                .value_parser(["UnassociatedDirty", "UnassociatedClean", "Premultiplied"])
+                .default_value("UnassociatedClean")
+        ])
+}

src/cli/codecs/mod.rs

@@ -1,10 +1,11 @@
 use clap::Command;
 
 use self::{
-    farbfeld::farbfeld, jpeg::jpeg, jpeg_xl::jpeg_xl, mozjpeg::mozjpeg, oxipng::oxipng, png::png,
-    ppm::ppm, qoi::qoi,
+    avif::avif, farbfeld::farbfeld, jpeg::jpeg, jpeg_xl::jpeg_xl, mozjpeg::mozjpeg, oxipng::oxipng,
+    png::png, ppm::ppm, qoi::qoi,
 };
 
+mod avif;
 mod farbfeld;
 mod jpeg;
 mod jpeg_xl;
@@ -17,6 +18,7 @@ mod qoi;
 impl Codecs for Command {
     fn codecs(self) -> Self {
         self.subcommands([
+            avif(),
             farbfeld(),
             jpeg(),
             jpeg_xl(),

src/cli/pipeline.rs

@@ -3,6 +3,7 @@ use std::{collections::BTreeMap, fs::read, path::Path};
 use clap::ArgMatches;
 use mozjpeg::qtable;
 use rimage::codecs::{
+    avif::{AvifEncoder, AvifOptions},
     mozjpeg::{MozJpegEncoder, MozJpegOptions},
     oxipng::{OxiPngEncoder, OxiPngOptions},
 };
@@ -226,6 +227,30 @@ pub fn encoder(matches: &ArgMatches) -> Result<(Box<dyn EncoderTrait>, &'static
 
                 Ok((Box::new(OxiPngEncoder::new_with_options(options)), "png"))
             }
+            "avif" => {
+                let options = AvifOptions {
+                    quality: *matches.get_one::<u8>("quality").unwrap() as f32,
+                    alpha_quality: matches.get_one::<u8>("alpha_quality").map(|q| *q as f32),
+                    speed: *matches.get_one::<u8>("speed").unwrap(),
+                    color_space: match matches.get_one::<String>("colorspace").unwrap().as_str() {
+                        "ycbcr" => ravif::ColorSpace::YCbCr,
+                        "rgb" => ravif::ColorSpace::RGB,
+                        _ => unreachable!(),
+                    },
+                    alpha_color_mode: match matches
+                        .get_one::<String>("alpha_mode")
+                        .unwrap()
+                        .as_str()
+                    {
+                        "UnassociatedDirty" => ravif::AlphaColorMode::UnassociatedDirty,
+                        "UnassociatedClean" => ravif::AlphaColorMode::UnassociatedClean,
+                        "Premultiplied" => ravif::AlphaColorMode::Premultiplied,
+                        _ => unreachable!(),
+                    },
+                };
+
+                Ok((Box::new(AvifEncoder::new_with_options(options)), "avif"))
+            }
             "png" => Ok((Box::new(PngEncoder::new()), "png")),
             "ppm" => Ok((Box::new(PPMEncoder::new()), "ppm")),
             "qoi" => Ok((Box::new(QoiEncoder::new()), "qoi")),