Initial commit

options.go

@@ -0,0 +1,149 @@
+package main
+
+import (
+	"image"
+	"image/gif"
+	"image/jpeg"
+
+	"github.com/gen2brain/avif"
+	"github.com/gen2brain/jpegxl"
+	"github.com/gen2brain/webp"
+	"golang.org/x/image/tiff"
+)
+
+func GetWebPOptions() webp.Options {
+	return webp.Options{
+		Lossless: arguments.GetBool("l", "lossless", false),
+		Quality:  int(arguments.GetUint64("q", "quality", 100, 0, 100)),
+		Method:   int(arguments.GetUint64("m", "method", 4, 0, 6)),
+		Exact:    arguments.GetBool("x", "exact", false),
+	}
+}
+
+func LogWebPOptions(options webp.Options) {
+	info("Using output options:")
+	info(" - lossless: %v", options.Lossless)
+	info(" - quality:  %v", options.Quality)
+	info(" - method:   %v", options.Method)
+	info(" - exact:    %v", options.Exact)
+}
+
+func GetJpegOptions() *jpeg.Options {
+	return &jpeg.Options{
+		Quality: int(arguments.GetUint64("q", "quality", 100, 0, 100)),
+	}
+}
+
+func LogJpegOptions(options *jpeg.Options) {
+	info("Using output options:")
+	info(" - quality: %v", options.Quality)
+}
+
+func GetGifOptions() *gif.Options {
+	return &gif.Options{
+		NumColors: int(arguments.GetUint64("c", "colors", 256, 0, 256)),
+	}
+}
+
+func LogGifOptions(options *gif.Options) {
+	info("Using output options:")
+	info(" - colors: %v", options.NumColors)
+}
+
+func GetTiffOptions() *tiff.Options {
+	return &tiff.Options{
+		Compression: GetTiffCompressionType(),
+	}
+}
+
+func LogTiffOptions(options *tiff.Options) {
+	info("Using output options:")
+	info(" - compression: %s", TiffCompressionTypeToString(options.Compression))
+}
+
+func GetAvifOptions() avif.Options {
+	return avif.Options{
+		Quality:           int(arguments.GetUint64("q", "quality", 100, 0, 100)),
+		QualityAlpha:      int(arguments.GetUint64("qa", "quality-alpha", 100, 0, 100)),
+		Speed:             int(arguments.GetUint64("s", "speed", 6, 0, 10)),
+		ChromaSubsampling: GetAvifYCbCrSubsampleRatio(),
+	}
+}
+
+func LogAvifOptions(options avif.Options) {
+	info("Using output options:")
+	info(" - quality: %v", options.Quality)
+	info(" - quality-alpha: %v", options.QualityAlpha)
+	info(" - speed: %v", options.Speed)
+	info(" - chroma subsampling: %s", options.ChromaSubsampling.String())
+}
+
+func GetJxlOptions() jpegxl.Options {
+	return jpegxl.Options{
+		Quality: int(arguments.GetUint64("q", "quality", 100, 0, 100)),
+		Effort:  int(arguments.GetUint64("e", "effort", 7, 0, 10)),
+	}
+}
+
+func LogJxlOptions(options jpegxl.Options) {
+	info("Using output options:")
+	info(" - quality: %v", options.Quality)
+	info(" - effort: %v", options.Effort)
+}
+
+func GetTiffCompressionType() tiff.CompressionType {
+	compression := arguments.GetUint64("z", "compression", 1, 0, 4)
+
+	switch compression {
+	case 0:
+		return tiff.Uncompressed
+	case 1:
+		return tiff.Deflate
+	case 2:
+		return tiff.LZW
+	case 3:
+		return tiff.CCITTGroup3
+	case 4:
+		return tiff.CCITTGroup4
+	}
+
+	return tiff.Deflate
+}
+
+func TiffCompressionTypeToString(compression tiff.CompressionType) string {
+	switch compression {
+	case tiff.Uncompressed:
+		return "uncompressed"
+	case tiff.Deflate:
+		return "deflate"
+	case tiff.LZW:
+		return "lzw"
+	case tiff.CCITTGroup3:
+		return "ccittgroup3"
+	case tiff.CCITTGroup4:
+		return "ccittgroup4"
+	default:
+		return "unknown"
+	}
+}
+
+func GetAvifYCbCrSubsampleRatio() image.YCbCrSubsampleRatio {
+	sampleRatio := arguments.GetUint64("r", "sample-ratio", 0, 0, 5)
+
+	switch sampleRatio {
+	case 0:
+		return image.YCbCrSubsampleRatio444
+	case 1:
+		return image.YCbCrSubsampleRatio422
+	case 2:
+		return image.YCbCrSubsampleRatio420
+	case 3:
+		return image.YCbCrSubsampleRatio440
+	case 4:
+		return image.YCbCrSubsampleRatio411
+	case 5:
+		return image.YCbCrSubsampleRatio410
+	}
+
+	return image.YCbCrSubsampleRatio444
+}