avif

internal/builtins/avif.go

@@ -0,0 +1,8 @@
+//go:build avif || core || full
+// +build avif core full
+
+package builtins
+
+import (
+	_ "github.com/coalaura/ffwebp/internal/codec/avif"
+)

internal/codec/avif/avif.go

@@ -0,0 +1,119 @@
+package avif
+
+import (
+	"bytes"
+	"fmt"
+	"image"
+	"io"
+
+	"github.com/gen2brain/avif"
+
+	"github.com/coalaura/ffwebp/internal/codec"
+	"github.com/coalaura/ffwebp/internal/logx"
+	"github.com/coalaura/ffwebp/internal/opts"
+	"github.com/urfave/cli/v3"
+)
+
+var (
+	quality  int
+	qualityA int
+	speed    int
+	chroma   int
+)
+
+func init() {
+	codec.Register(impl{})
+}
+
+type impl struct{}
+
+func (impl) String() string {
+	return "avif"
+}
+
+func (impl) Extensions() []string {
+	return []string{"avif"}
+}
+
+func (impl) Flags(flags []cli.Flag) []cli.Flag {
+	return append(flags,
+		&cli.IntFlag{
+			Name:        "avif.quality-alpha",
+			Usage:       "AVIF: alpha channel quality in range [0-100]",
+			Value:       60,
+			Destination: &qualityA,
+			Validator: func(v int) error {
+				if v < 0 || v > 100 {
+					return fmt.Errorf("invalid avif.quality-alpha: %d", v)
+				}
+				return nil
+			},
+		},
+		&cli.IntFlag{
+			Name:        "avif.speed",
+			Usage:       "AVIF: encoding speed in range [0-10] (0=slowest/best)",
+			Value:       6,
+			Destination: &speed,
+			Validator: func(v int) error {
+				if v < 0 || v > 10 {
+					return fmt.Errorf("invalid avif.speed: %d", v)
+				}
+				return nil
+			},
+		},
+		&cli.IntFlag{
+			Name:        "avif.chroma",
+			Usage:       "AVIF: chroma subsampling (444=best, 422, 420=smallest)",
+			Value:       444,
+			Destination: &chroma,
+			Validator: func(v int) error {
+				if v != 444 && v != 422 && v != 420 {
+					return fmt.Errorf("invalid avif.chroma: %d", v)
+				}
+				return nil
+			},
+		},
+	)
+}
+
+func (impl) Sniff(reader io.ReaderAt) (int, []byte, error) {
+	buf := make([]byte, 12)
+
+	if _, err := reader.ReadAt(buf, 0); err != nil {
+		return 0, nil, err
+	}
+
+	if bytes.Equal(buf[4:12], []byte("ftypavif")) {
+		return 100, buf[:12], nil
+	}
+
+	return 0, nil, nil
+}
+
+func (impl) Decode(reader io.Reader) (image.Image, error) {
+	return avif.Decode(reader)
+}
+
+func (impl) Encode(writer io.Writer, img image.Image, options opts.Common) error {
+	logx.Printf("avif: quality=%d, quality-alpha=%d, speed=%d, chroma=%d\n", options.Quality, qualityA, speed, chroma)
+
+	return avif.Encode(writer, img, avif.Options{
+		Quality:           options.Quality,
+		QualityAlpha:      qualityA,
+		Speed:             speed,
+		ChromaSubsampling: chromaSubsampling(chroma),
+	})
+}
+
+func chromaSubsampling(c int) image.YCbCrSubsampleRatio {
+	switch c {
+	case 444:
+		return image.YCbCrSubsampleRatio444
+	case 422:
+		return image.YCbCrSubsampleRatio422
+	case 420:
+		return image.YCbCrSubsampleRatio420
+	default:
+		return image.YCbCrSubsampleRatio444
+	}
+}

internal/codec/detect.go

@@ -61,7 +61,7 @@ func Sniff(reader io.Reader) (*Sniffed, io.Reader, error) {
 	}
 
 	if guess == nil {
-		return nil, nil, errors.New("unknown format")
+		return nil, nil, errors.New("unknown input format")
 	}
 
 	return &Sniffed{