mirror of
https://github.com/coalaura/ffwebp.git
synced 2025-09-08 05:49:54 +00:00
123 lines
2.5 KiB
Go
123 lines
2.5 KiB
Go
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 (
|
|
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) CanEncode() bool {
|
|
return true
|
|
}
|
|
|
|
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
|
|
}
|
|
}
|