Implementation for SWS scale

README.md

@@ -27,6 +27,7 @@ Examples are located in the [examples](examples) directory and mirror as much as
 |Filtering|[see](examples/filtering/main.go)|[see](https://github.com/FFmpeg/FFmpeg/blob/n5.1.2/doc/examples/filtering_video.c)
 |Hardware Decoding|[see](examples/hardware_decoding/main.go)|[see](https://github.com/FFmpeg/FFmpeg/blob/n5.1.2/doc/examples/hw_decode.c)
 |Remuxing|[see](examples/remuxing/main.go)|[see](https://github.com/FFmpeg/FFmpeg/blob/n5.1.2/doc/examples/remuxing.c)
+|Scaling|[see](examples/scaling/main.go)|[see](https://github.com/FFmpeg/FFmpeg/blob/n5.1.2/doc/examples/scaling_video.c)
 |Transcoding|[see](examples/transcoding/main.go)|[see](https://github.com/FFmpeg/FFmpeg/blob/n5.1.2/doc/examples/transcoding.c)
 
 *Tip: you can use the video sample located in the `testdata` directory for your tests*

examples/scaling/main.go

@@ -0,0 +1,98 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"image/png"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/asticode/go-astiav"
+)
+
+var (
+	output    = flag.String("o", "", "the png output path")
+	dstWidth  = flag.Int("w", 50, "destination width")
+	dstHeight = flag.Int("h", 50, "destination height")
+)
+
+func main() {
+	// Handle ffmpeg logs
+	astiav.SetLogLevel(astiav.LogLevelDebug)
+	astiav.SetLogCallback(func(l astiav.LogLevel, fmt, msg, parent string) {
+		log.Printf("ffmpeg log: %s (level: %d)\n", strings.TrimSpace(msg), l)
+	})
+
+	// Parse flags
+	flag.Parse()
+
+	// Usage
+	if *output == "" || *dstWidth <= 0 || *dstHeight <= 0 {
+		log.Println("Usage: <binary path> -o <output path> -w <output width> -h <output height>")
+		return
+	}
+
+	// Create destination file
+	dstFile, err := os.Create(*output)
+	if err != nil {
+		log.Fatal(fmt.Errorf("main: creating %s failed: %w", *output, err))
+	}
+	defer dstFile.Close()
+
+	// Create source frame
+	srcFrame := astiav.AllocFrame()
+	defer srcFrame.Free()
+	srcFrame.SetWidth(320)
+	srcFrame.SetHeight(240)
+	srcFrame.SetPixelFormat(astiav.PixelFormatYuv420P)
+	if err = srcFrame.AllocBuffer(1); err != nil {
+		log.Fatal(fmt.Errorf("main: allocating source frame buffer failed: %w", err))
+	}
+	if err = srcFrame.ImageFillBlack(); err != nil {
+		log.Fatal(fmt.Errorf("main: filling source frame with black image failed: %w", err))
+	}
+
+	// Create destination frame
+	dstFrame := astiav.AllocFrame()
+	defer dstFrame.Free()
+
+	// Create software scale context
+	swsCtx, err := astiav.CreateSoftwareScaleContext(
+		srcFrame.Width(),
+		srcFrame.Height(),
+		srcFrame.PixelFormat(),
+		*dstWidth,
+		*dstHeight,
+		astiav.PixelFormatRgba,
+		astiav.NewSoftwareScaleContextFlags(astiav.SoftwareScaleContextFlagBilinear),
+	)
+	if err != nil {
+		log.Fatal(fmt.Errorf("main: creating software scale context failed: %w", err))
+	}
+	defer swsCtx.Free()
+
+	// Scale frame
+	if err := swsCtx.ScaleFrame(srcFrame, dstFrame); err != nil {
+		log.Fatal(fmt.Errorf("main: scaling frame failed: %w", err))
+	}
+
+	// Guess destination image format
+	img, err := dstFrame.Data().GuessImageFormat()
+	if err != nil {
+		log.Fatal(fmt.Errorf("main: guessing destination image format failed: %w", err))
+	}
+
+	// Copy frame data to destination image
+	if err = dstFrame.Data().ToImage(img); err != nil {
+		log.Fatal(fmt.Errorf("main: copying frame data to destination image failed: %w", err))
+	}
+
+	// Encode to png
+	if err = png.Encode(dstFile, img); err != nil {
+		log.Fatal(fmt.Errorf("main: encoding to png failed: %w", err))
+	}
+
+	// Success
+	log.Println("success")
+}

internal/cmd/flags/main.go

@@ -21,13 +21,14 @@ var list = []listItem{
 	{Name: "CodecHardwareConfigMethod"},
 	{Name: "Dictionary"},
 	{Name: "FilterCommand"},
-	{Name: "FormatContextCtx"},
 	{Name: "FormatContext"},
+	{Name: "FormatContextCtx"},
 	{Name: "FormatEvent"},
 	{Name: "IOContext"},
 	{Name: "IOFormat"},
 	{Name: "Packet"},
 	{Name: "Seek"},
+	{Name: "SoftwareScaleContext"},
 	{Name: "StreamEvent"},
 }