Update .gitignore to include log files and database journal files. Modify go.mod to update dependencies for go-sqlite3 and cloud.google.com/go/compute/metadata. Enhance Makefile to include logging options for manager and runner commands. Introduce new job token handling in auth package and implement database migration scripts. Refactor manager and runner components to improve job processing and metadata extraction. Add support for video preview in frontend components and enhance WebSocket management for channel subscriptions.
This commit is contained in:
980
internal/runner/encoding/encoders_test.go
Normal file
980
internal/runner/encoding/encoders_test.go
Normal file
@@ -0,0 +1,980 @@
|
||||
package encoding
|
||||
|
||||
import (
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestSoftwareEncoder_BuildCommand_H264_EXR(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: true,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
if cmd == nil {
|
||||
t.Fatal("BuildCommand returned nil")
|
||||
}
|
||||
|
||||
if !strings.Contains(cmd.Path, "ffmpeg") {
|
||||
t.Errorf("Expected command path to contain 'ffmpeg', got '%s'", cmd.Path)
|
||||
}
|
||||
|
||||
if cmd.Dir != "/tmp" {
|
||||
t.Errorf("Expected work dir '/tmp', got '%s'", cmd.Dir)
|
||||
}
|
||||
|
||||
args := cmd.Args[1:] // Skip "ffmpeg"
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Check required arguments
|
||||
checks := []struct {
|
||||
name string
|
||||
expected string
|
||||
}{
|
||||
{"-y flag", "-y"},
|
||||
{"image2 format", "-f image2"},
|
||||
{"start number", "-start_number 1"},
|
||||
{"framerate", "-framerate 24.00"},
|
||||
{"input pattern", "-i frame_%04d.exr"},
|
||||
{"codec", "-c:v libx264"},
|
||||
{"pixel format", "-pix_fmt yuv420p"}, // EXR now treated as SDR (like PNG)
|
||||
{"frame rate", "-r 24.00"},
|
||||
{"color primaries", "-color_primaries bt709"}, // EXR now uses bt709 (SDR)
|
||||
{"color trc", "-color_trc bt709"}, // EXR now uses bt709 (SDR)
|
||||
{"colorspace", "-colorspace bt709"},
|
||||
{"color range", "-color_range tv"},
|
||||
{"video filter", "-vf"},
|
||||
{"preset", "-preset veryslow"},
|
||||
{"crf", "-crf 15"},
|
||||
{"profile", "-profile:v high"}, // EXR now uses high profile (SDR)
|
||||
{"pass 2", "-pass 2"},
|
||||
{"output path", "output.mp4"},
|
||||
}
|
||||
|
||||
for _, check := range checks {
|
||||
if !strings.Contains(argsStr, check.expected) {
|
||||
t.Errorf("Missing expected argument: %s", check.expected)
|
||||
}
|
||||
}
|
||||
|
||||
// Verify filter is present for EXR (linear RGB to sRGB conversion, like Krita does)
|
||||
if !strings.Contains(argsStr, "format=gbrpf32le") {
|
||||
t.Error("Expected format conversion filter for EXR source, but not found")
|
||||
}
|
||||
if !strings.Contains(argsStr, "zscale=transferin=8:transfer=13") {
|
||||
t.Error("Expected linear to sRGB conversion for EXR source, but not found")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_BuildCommand_H264_PNG(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.png",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: true,
|
||||
SourceFormat: "png",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// PNG should NOT have video filter
|
||||
if strings.Contains(argsStr, "-vf") {
|
||||
t.Error("PNG source should not have video filter, but -vf was found")
|
||||
}
|
||||
|
||||
// Should still have all other required args
|
||||
if !strings.Contains(argsStr, "-c:v libx264") {
|
||||
t.Error("Missing codec argument")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_BuildCommand_AV1_WithAlpha(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libaom-av1"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 100,
|
||||
FrameRate: 30.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: true,
|
||||
TwoPass: true,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Check alpha-specific settings
|
||||
if !strings.Contains(argsStr, "-pix_fmt yuva420p") {
|
||||
t.Error("Expected yuva420p pixel format for alpha, but not found")
|
||||
}
|
||||
|
||||
// Check AV1-specific arguments
|
||||
av1Checks := []string{
|
||||
"-c:v libaom-av1",
|
||||
"-crf 30",
|
||||
"-b:v 0",
|
||||
"-tiles 2x2",
|
||||
"-g 240",
|
||||
}
|
||||
|
||||
for _, check := range av1Checks {
|
||||
if !strings.Contains(argsStr, check) {
|
||||
t.Errorf("Missing AV1 argument: %s", check)
|
||||
}
|
||||
}
|
||||
|
||||
// Check tonemap filter includes alpha format
|
||||
if !strings.Contains(argsStr, "format=yuva420p") {
|
||||
t.Error("Expected tonemap filter to output yuva420p for alpha, but not found")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_BuildCommand_VP9(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libvpx-vp9"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.webm",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: true,
|
||||
TwoPass: true,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Check VP9-specific arguments
|
||||
vp9Checks := []string{
|
||||
"-c:v libvpx-vp9",
|
||||
"-crf 30",
|
||||
"-b:v 0",
|
||||
"-row-mt 1",
|
||||
"-g 240",
|
||||
}
|
||||
|
||||
for _, check := range vp9Checks {
|
||||
if !strings.Contains(argsStr, check) {
|
||||
t.Errorf("Missing VP9 argument: %s", check)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_BuildPass1Command(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: true,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildPass1Command(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Pass 1 should have -pass 1 and output to null
|
||||
if !strings.Contains(argsStr, "-pass 1") {
|
||||
t.Error("Pass 1 command should include '-pass 1'")
|
||||
}
|
||||
|
||||
if !strings.Contains(argsStr, "-f null") {
|
||||
t.Error("Pass 1 command should include '-f null'")
|
||||
}
|
||||
|
||||
if !strings.Contains(argsStr, "/dev/null") {
|
||||
t.Error("Pass 1 command should output to /dev/null")
|
||||
}
|
||||
|
||||
// Should NOT have output path
|
||||
if strings.Contains(argsStr, "output.mp4") {
|
||||
t.Error("Pass 1 command should not include output path")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_BuildPass1Command_AV1(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libaom-av1"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: true,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildPass1Command(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Pass 1 should have -pass 1 and output to null
|
||||
if !strings.Contains(argsStr, "-pass 1") {
|
||||
t.Error("Pass 1 command should include '-pass 1'")
|
||||
}
|
||||
|
||||
if !strings.Contains(argsStr, "-f null") {
|
||||
t.Error("Pass 1 command should include '-f null'")
|
||||
}
|
||||
|
||||
if !strings.Contains(argsStr, "/dev/null") {
|
||||
t.Error("Pass 1 command should output to /dev/null")
|
||||
}
|
||||
|
||||
// Check AV1-specific arguments in pass 1
|
||||
av1Checks := []string{
|
||||
"-c:v libaom-av1",
|
||||
"-crf 30",
|
||||
"-b:v 0",
|
||||
"-tiles 2x2",
|
||||
"-g 240",
|
||||
}
|
||||
|
||||
for _, check := range av1Checks {
|
||||
if !strings.Contains(argsStr, check) {
|
||||
t.Errorf("Missing AV1 argument in pass 1: %s", check)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_BuildPass1Command_VP9(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libvpx-vp9"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.webm",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: true,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildPass1Command(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Pass 1 should have -pass 1 and output to null
|
||||
if !strings.Contains(argsStr, "-pass 1") {
|
||||
t.Error("Pass 1 command should include '-pass 1'")
|
||||
}
|
||||
|
||||
if !strings.Contains(argsStr, "-f null") {
|
||||
t.Error("Pass 1 command should include '-f null'")
|
||||
}
|
||||
|
||||
if !strings.Contains(argsStr, "/dev/null") {
|
||||
t.Error("Pass 1 command should output to /dev/null")
|
||||
}
|
||||
|
||||
// Check VP9-specific arguments in pass 1
|
||||
vp9Checks := []string{
|
||||
"-c:v libvpx-vp9",
|
||||
"-crf 30",
|
||||
"-b:v 0",
|
||||
"-row-mt 1",
|
||||
"-g 240",
|
||||
}
|
||||
|
||||
for _, check := range vp9Checks {
|
||||
if !strings.Contains(argsStr, check) {
|
||||
t.Errorf("Missing VP9 argument in pass 1: %s", check)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_BuildCommand_NoTwoPass(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: false,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Should NOT have -pass flag when TwoPass is false
|
||||
if strings.Contains(argsStr, "-pass") {
|
||||
t.Error("Command should not include -pass flag when TwoPass is false")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSelector_SelectH264(t *testing.T) {
|
||||
selector := NewSelector()
|
||||
encoder := selector.SelectH264()
|
||||
|
||||
if encoder == nil {
|
||||
t.Fatal("SelectH264 returned nil")
|
||||
}
|
||||
|
||||
if encoder.Codec() != "libx264" {
|
||||
t.Errorf("Expected codec 'libx264', got '%s'", encoder.Codec())
|
||||
}
|
||||
|
||||
if encoder.Name() != "software" {
|
||||
t.Errorf("Expected name 'software', got '%s'", encoder.Name())
|
||||
}
|
||||
}
|
||||
|
||||
func TestSelector_SelectAV1(t *testing.T) {
|
||||
selector := NewSelector()
|
||||
encoder := selector.SelectAV1()
|
||||
|
||||
if encoder == nil {
|
||||
t.Fatal("SelectAV1 returned nil")
|
||||
}
|
||||
|
||||
if encoder.Codec() != "libaom-av1" {
|
||||
t.Errorf("Expected codec 'libaom-av1', got '%s'", encoder.Codec())
|
||||
}
|
||||
}
|
||||
|
||||
func TestSelector_SelectVP9(t *testing.T) {
|
||||
selector := NewSelector()
|
||||
encoder := selector.SelectVP9()
|
||||
|
||||
if encoder == nil {
|
||||
t.Fatal("SelectVP9 returned nil")
|
||||
}
|
||||
|
||||
if encoder.Codec() != "libvpx-vp9" {
|
||||
t.Errorf("Expected codec 'libvpx-vp9', got '%s'", encoder.Codec())
|
||||
}
|
||||
}
|
||||
|
||||
func TestTonemapFilter_WithAlpha(t *testing.T) {
|
||||
filter := tonemapFilter(true)
|
||||
|
||||
// Filter should convert from gbrpf32le to yuva420p10le with proper colorspace conversion
|
||||
if !strings.Contains(filter, "yuva420p10le") {
|
||||
t.Error("Tonemap filter with alpha should output yuva420p10le format for HDR")
|
||||
}
|
||||
|
||||
if !strings.Contains(filter, "gbrpf32le") {
|
||||
t.Error("Tonemap filter should start with gbrpf32le format")
|
||||
}
|
||||
|
||||
// Should use zscale for colorspace conversion from linear RGB to bt2020 YUV
|
||||
if !strings.Contains(filter, "zscale") {
|
||||
t.Error("Tonemap filter should use zscale for colorspace conversion")
|
||||
}
|
||||
|
||||
// Check for HLG transfer function (numeric value 18 or string arib-std-b67)
|
||||
if !strings.Contains(filter, "transfer=18") && !strings.Contains(filter, "transfer=arib-std-b67") {
|
||||
t.Error("Tonemap filter should use HLG transfer function (18 or arib-std-b67)")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTonemapFilter_WithoutAlpha(t *testing.T) {
|
||||
filter := tonemapFilter(false)
|
||||
|
||||
// Filter should convert from gbrpf32le to yuv420p10le with proper colorspace conversion
|
||||
if !strings.Contains(filter, "yuv420p10le") {
|
||||
t.Error("Tonemap filter without alpha should output yuv420p10le format for HDR")
|
||||
}
|
||||
|
||||
if strings.Contains(filter, "yuva420p") {
|
||||
t.Error("Tonemap filter without alpha should not output yuva420p format")
|
||||
}
|
||||
|
||||
if !strings.Contains(filter, "gbrpf32le") {
|
||||
t.Error("Tonemap filter should start with gbrpf32le format")
|
||||
}
|
||||
|
||||
// Should use zscale for colorspace conversion from linear RGB to bt2020 YUV
|
||||
if !strings.Contains(filter, "zscale") {
|
||||
t.Error("Tonemap filter should use zscale for colorspace conversion")
|
||||
}
|
||||
|
||||
// Check for HLG transfer function (numeric value 18 or string arib-std-b67)
|
||||
if !strings.Contains(filter, "transfer=18") && !strings.Contains(filter, "transfer=arib-std-b67") {
|
||||
t.Error("Tonemap filter should use HLG transfer function (18 or arib-std-b67)")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSoftwareEncoder_Available(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
if !encoder.Available() {
|
||||
t.Error("Software encoder should always be available")
|
||||
}
|
||||
}
|
||||
|
||||
func TestEncodeConfig_DefaultSourceFormat(t *testing.T) {
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: false,
|
||||
// SourceFormat not set, should default to empty string (treated as exr)
|
||||
}
|
||||
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := strings.Join(cmd.Args[1:], " ")
|
||||
|
||||
// Should still have tonemap filter when SourceFormat is empty (defaults to exr behavior)
|
||||
if !strings.Contains(args, "-vf") {
|
||||
t.Error("Empty SourceFormat should default to EXR behavior with tonemap filter")
|
||||
}
|
||||
}
|
||||
|
||||
func TestCommandOrder(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: true,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := cmd.Args[1:]
|
||||
|
||||
// Verify argument order: input should come before codec
|
||||
inputIdx := -1
|
||||
codecIdx := -1
|
||||
vfIdx := -1
|
||||
|
||||
for i, arg := range args {
|
||||
if arg == "-i" && i+1 < len(args) && args[i+1] == "frame_%04d.exr" {
|
||||
inputIdx = i
|
||||
}
|
||||
if arg == "-c:v" && i+1 < len(args) && args[i+1] == "libx264" {
|
||||
codecIdx = i
|
||||
}
|
||||
if arg == "-vf" {
|
||||
vfIdx = i
|
||||
}
|
||||
}
|
||||
|
||||
if inputIdx == -1 {
|
||||
t.Fatal("Input pattern not found in command")
|
||||
}
|
||||
if codecIdx == -1 {
|
||||
t.Fatal("Codec not found in command")
|
||||
}
|
||||
if vfIdx == -1 {
|
||||
t.Fatal("Video filter not found in command")
|
||||
}
|
||||
|
||||
// Input should come before codec
|
||||
if inputIdx >= codecIdx {
|
||||
t.Error("Input pattern should come before codec in command")
|
||||
}
|
||||
|
||||
// Video filter should come after input (order: input -> codec -> colorspace -> filter -> codec args)
|
||||
// In practice, the filter comes after codec and colorspace metadata but before codec-specific args
|
||||
if vfIdx <= inputIdx {
|
||||
t.Error("Video filter should come after input")
|
||||
}
|
||||
}
|
||||
|
||||
func TestCommand_ColorspaceMetadata(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: false,
|
||||
SourceFormat: "exr",
|
||||
PreserveHDR: false, // SDR encoding
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Verify all SDR colorspace metadata is present for EXR (SDR encoding)
|
||||
colorspaceArgs := []string{
|
||||
"-color_primaries bt709", // EXR uses bt709 (SDR)
|
||||
"-color_trc bt709", // EXR uses bt709 (SDR)
|
||||
"-colorspace bt709",
|
||||
"-color_range tv",
|
||||
}
|
||||
|
||||
for _, arg := range colorspaceArgs {
|
||||
if !strings.Contains(argsStr, arg) {
|
||||
t.Errorf("Missing colorspace metadata: %s", arg)
|
||||
}
|
||||
}
|
||||
|
||||
// Verify SDR pixel format
|
||||
if !strings.Contains(argsStr, "-pix_fmt yuv420p") {
|
||||
t.Error("SDR encoding should use yuv420p pixel format")
|
||||
}
|
||||
|
||||
// Verify H.264 high profile (not high10)
|
||||
if !strings.Contains(argsStr, "-profile:v high") {
|
||||
t.Error("SDR encoding should use high profile")
|
||||
}
|
||||
if strings.Contains(argsStr, "-profile:v high10") {
|
||||
t.Error("SDR encoding should not use high10 profile")
|
||||
}
|
||||
}
|
||||
|
||||
func TestCommand_HDR_ColorspaceMetadata(t *testing.T) {
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: "frame_%04d.exr",
|
||||
OutputPath: "output.mp4",
|
||||
StartFrame: 1,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: "/tmp",
|
||||
UseAlpha: false,
|
||||
TwoPass: false,
|
||||
SourceFormat: "exr",
|
||||
PreserveHDR: true, // HDR encoding
|
||||
}
|
||||
|
||||
cmd := encoder.BuildCommand(config)
|
||||
args := cmd.Args[1:]
|
||||
argsStr := strings.Join(args, " ")
|
||||
|
||||
// Verify all HDR colorspace metadata is present for EXR (HDR encoding)
|
||||
colorspaceArgs := []string{
|
||||
"-color_primaries bt709", // bt709 primaries to match PNG color appearance
|
||||
"-color_trc arib-std-b67", // HLG transfer function for HDR/SDR compatibility
|
||||
"-colorspace bt709", // bt709 colorspace to match PNG
|
||||
"-color_range tv",
|
||||
}
|
||||
|
||||
for _, arg := range colorspaceArgs {
|
||||
if !strings.Contains(argsStr, arg) {
|
||||
t.Errorf("Missing HDR colorspace metadata: %s", arg)
|
||||
}
|
||||
}
|
||||
|
||||
// Verify HDR pixel format (10-bit)
|
||||
if !strings.Contains(argsStr, "-pix_fmt yuv420p10le") {
|
||||
t.Error("HDR encoding should use yuv420p10le pixel format")
|
||||
}
|
||||
|
||||
// Verify H.264 high10 profile (for 10-bit)
|
||||
if !strings.Contains(argsStr, "-profile:v high10") {
|
||||
t.Error("HDR encoding should use high10 profile")
|
||||
}
|
||||
|
||||
// Verify HDR filter chain (linear -> sRGB -> HLG)
|
||||
if !strings.Contains(argsStr, "-vf") {
|
||||
t.Fatal("HDR encoding should have video filter")
|
||||
}
|
||||
vfIdx := -1
|
||||
for i, arg := range args {
|
||||
if arg == "-vf" && i+1 < len(args) {
|
||||
vfIdx = i + 1
|
||||
break
|
||||
}
|
||||
}
|
||||
if vfIdx == -1 {
|
||||
t.Fatal("Video filter not found")
|
||||
}
|
||||
filter := args[vfIdx]
|
||||
if !strings.Contains(filter, "transfer=18") {
|
||||
t.Error("HDR filter should convert to HLG (transfer=18)")
|
||||
}
|
||||
if !strings.Contains(filter, "yuv420p10le") {
|
||||
t.Error("HDR filter should output yuv420p10le format")
|
||||
}
|
||||
}
|
||||
|
||||
// Integration tests using example files
|
||||
func TestIntegration_Encode_EXR_H264(t *testing.T) {
|
||||
if testing.Short() {
|
||||
t.Skip("Skipping integration test in short mode")
|
||||
}
|
||||
|
||||
// Check if example file exists
|
||||
exampleDir := filepath.Join("..", "..", "..", "examples")
|
||||
exrFile := filepath.Join(exampleDir, "frame_0800.exr")
|
||||
if _, err := os.Stat(exrFile); os.IsNotExist(err) {
|
||||
t.Skipf("Example file not found: %s", exrFile)
|
||||
}
|
||||
|
||||
// Get absolute paths
|
||||
workspaceRoot, err := filepath.Abs(filepath.Join("..", "..", ".."))
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get workspace root: %v", err)
|
||||
}
|
||||
exampleDirAbs, err := filepath.Abs(exampleDir)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get example directory: %v", err)
|
||||
}
|
||||
tmpDir := filepath.Join(workspaceRoot, "tmp")
|
||||
if err := os.MkdirAll(tmpDir, 0755); err != nil {
|
||||
t.Fatalf("Failed to create tmp directory: %v", err)
|
||||
}
|
||||
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: filepath.Join(exampleDirAbs, "frame_%04d.exr"),
|
||||
OutputPath: filepath.Join(tmpDir, "test_exr_h264.mp4"),
|
||||
StartFrame: 800,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: tmpDir,
|
||||
UseAlpha: false,
|
||||
TwoPass: false, // Use single pass for faster testing
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
// Build and run command
|
||||
cmd := encoder.BuildCommand(config)
|
||||
if cmd == nil {
|
||||
t.Fatal("BuildCommand returned nil")
|
||||
}
|
||||
|
||||
// Capture stderr to see what went wrong
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Errorf("FFmpeg command failed: %v\nCommand output: %s", err, string(output))
|
||||
return
|
||||
}
|
||||
|
||||
// Verify output file was created
|
||||
if _, err := os.Stat(config.OutputPath); os.IsNotExist(err) {
|
||||
t.Errorf("Output file was not created: %s\nCommand output: %s", config.OutputPath, string(output))
|
||||
} else {
|
||||
t.Logf("Successfully created output file: %s", config.OutputPath)
|
||||
// Verify file has content
|
||||
info, _ := os.Stat(config.OutputPath)
|
||||
if info.Size() == 0 {
|
||||
t.Errorf("Output file was created but is empty\nCommand output: %s", string(output))
|
||||
} else {
|
||||
t.Logf("Output file size: %d bytes", info.Size())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestIntegration_Encode_PNG_H264(t *testing.T) {
|
||||
if testing.Short() {
|
||||
t.Skip("Skipping integration test in short mode")
|
||||
}
|
||||
|
||||
// Check if example file exists
|
||||
exampleDir := filepath.Join("..", "..", "..", "examples")
|
||||
pngFile := filepath.Join(exampleDir, "frame_0800.png")
|
||||
if _, err := os.Stat(pngFile); os.IsNotExist(err) {
|
||||
t.Skipf("Example file not found: %s", pngFile)
|
||||
}
|
||||
|
||||
// Get absolute paths
|
||||
workspaceRoot, err := filepath.Abs(filepath.Join("..", "..", ".."))
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get workspace root: %v", err)
|
||||
}
|
||||
exampleDirAbs, err := filepath.Abs(exampleDir)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get example directory: %v", err)
|
||||
}
|
||||
tmpDir := filepath.Join(workspaceRoot, "tmp")
|
||||
if err := os.MkdirAll(tmpDir, 0755); err != nil {
|
||||
t.Fatalf("Failed to create tmp directory: %v", err)
|
||||
}
|
||||
|
||||
encoder := &SoftwareEncoder{codec: "libx264"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: filepath.Join(exampleDirAbs, "frame_%04d.png"),
|
||||
OutputPath: filepath.Join(tmpDir, "test_png_h264.mp4"),
|
||||
StartFrame: 800,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: tmpDir,
|
||||
UseAlpha: false,
|
||||
TwoPass: false, // Use single pass for faster testing
|
||||
SourceFormat: "png",
|
||||
}
|
||||
|
||||
// Build and run command
|
||||
cmd := encoder.BuildCommand(config)
|
||||
if cmd == nil {
|
||||
t.Fatal("BuildCommand returned nil")
|
||||
}
|
||||
|
||||
// Verify no video filter is used for PNG
|
||||
argsStr := strings.Join(cmd.Args, " ")
|
||||
if strings.Contains(argsStr, "-vf") {
|
||||
t.Error("PNG encoding should not use video filter, but -vf was found in command")
|
||||
}
|
||||
|
||||
// Run the command
|
||||
cmdOutput, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Errorf("FFmpeg command failed: %v\nCommand output: %s", err, string(cmdOutput))
|
||||
return
|
||||
}
|
||||
|
||||
// Verify output file was created
|
||||
if _, err := os.Stat(config.OutputPath); os.IsNotExist(err) {
|
||||
t.Errorf("Output file was not created: %s\nCommand output: %s", config.OutputPath, string(cmdOutput))
|
||||
} else {
|
||||
t.Logf("Successfully created output file: %s", config.OutputPath)
|
||||
info, _ := os.Stat(config.OutputPath)
|
||||
if info.Size() == 0 {
|
||||
t.Error("Output file was created but is empty")
|
||||
} else {
|
||||
t.Logf("Output file size: %d bytes", info.Size())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestIntegration_Encode_EXR_VP9(t *testing.T) {
|
||||
if testing.Short() {
|
||||
t.Skip("Skipping integration test in short mode")
|
||||
}
|
||||
|
||||
// Check if example file exists
|
||||
exampleDir := filepath.Join("..", "..", "..", "examples")
|
||||
exrFile := filepath.Join(exampleDir, "frame_0800.exr")
|
||||
if _, err := os.Stat(exrFile); os.IsNotExist(err) {
|
||||
t.Skipf("Example file not found: %s", exrFile)
|
||||
}
|
||||
|
||||
// Check if VP9 encoder is available
|
||||
checkCmd := exec.Command("ffmpeg", "-hide_banner", "-encoders")
|
||||
checkOutput, err := checkCmd.CombinedOutput()
|
||||
if err != nil || !strings.Contains(string(checkOutput), "libvpx-vp9") {
|
||||
t.Skip("VP9 encoder (libvpx-vp9) not available in ffmpeg")
|
||||
}
|
||||
|
||||
// Get absolute paths
|
||||
workspaceRoot, err := filepath.Abs(filepath.Join("..", "..", ".."))
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get workspace root: %v", err)
|
||||
}
|
||||
exampleDirAbs, err := filepath.Abs(exampleDir)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get example directory: %v", err)
|
||||
}
|
||||
tmpDir := filepath.Join(workspaceRoot, "tmp")
|
||||
if err := os.MkdirAll(tmpDir, 0755); err != nil {
|
||||
t.Fatalf("Failed to create tmp directory: %v", err)
|
||||
}
|
||||
|
||||
encoder := &SoftwareEncoder{codec: "libvpx-vp9"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: filepath.Join(exampleDirAbs, "frame_%04d.exr"),
|
||||
OutputPath: filepath.Join(tmpDir, "test_exr_vp9.webm"),
|
||||
StartFrame: 800,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: tmpDir,
|
||||
UseAlpha: false,
|
||||
TwoPass: false, // Use single pass for faster testing
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
// Build and run command
|
||||
cmd := encoder.BuildCommand(config)
|
||||
if cmd == nil {
|
||||
t.Fatal("BuildCommand returned nil")
|
||||
}
|
||||
|
||||
// Capture stderr to see what went wrong
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Errorf("FFmpeg command failed: %v\nCommand output: %s", err, string(output))
|
||||
return
|
||||
}
|
||||
|
||||
// Verify output file was created
|
||||
if _, err := os.Stat(config.OutputPath); os.IsNotExist(err) {
|
||||
t.Errorf("Output file was not created: %s\nCommand output: %s", config.OutputPath, string(output))
|
||||
} else {
|
||||
t.Logf("Successfully created output file: %s", config.OutputPath)
|
||||
// Verify file has content
|
||||
info, _ := os.Stat(config.OutputPath)
|
||||
if info.Size() == 0 {
|
||||
t.Errorf("Output file was created but is empty\nCommand output: %s", string(output))
|
||||
} else {
|
||||
t.Logf("Output file size: %d bytes", info.Size())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestIntegration_Encode_EXR_AV1(t *testing.T) {
|
||||
if testing.Short() {
|
||||
t.Skip("Skipping integration test in short mode")
|
||||
}
|
||||
|
||||
// Check if example file exists
|
||||
exampleDir := filepath.Join("..", "..", "..", "examples")
|
||||
exrFile := filepath.Join(exampleDir, "frame_0800.exr")
|
||||
if _, err := os.Stat(exrFile); os.IsNotExist(err) {
|
||||
t.Skipf("Example file not found: %s", exrFile)
|
||||
}
|
||||
|
||||
// Check if AV1 encoder is available
|
||||
checkCmd := exec.Command("ffmpeg", "-hide_banner", "-encoders")
|
||||
output, err := checkCmd.CombinedOutput()
|
||||
if err != nil || !strings.Contains(string(output), "libaom-av1") {
|
||||
t.Skip("AV1 encoder (libaom-av1) not available in ffmpeg")
|
||||
}
|
||||
|
||||
// Get absolute paths
|
||||
workspaceRoot, err := filepath.Abs(filepath.Join("..", "..", ".."))
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get workspace root: %v", err)
|
||||
}
|
||||
exampleDirAbs, err := filepath.Abs(exampleDir)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get example directory: %v", err)
|
||||
}
|
||||
tmpDir := filepath.Join(workspaceRoot, "tmp")
|
||||
if err := os.MkdirAll(tmpDir, 0755); err != nil {
|
||||
t.Fatalf("Failed to create tmp directory: %v", err)
|
||||
}
|
||||
|
||||
encoder := &SoftwareEncoder{codec: "libaom-av1"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: filepath.Join(exampleDirAbs, "frame_%04d.exr"),
|
||||
OutputPath: filepath.Join(tmpDir, "test_exr_av1.mp4"),
|
||||
StartFrame: 800,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: tmpDir,
|
||||
UseAlpha: false,
|
||||
TwoPass: false,
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
// Build and run command
|
||||
cmd := encoder.BuildCommand(config)
|
||||
cmdOutput, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Errorf("FFmpeg command failed: %v\nCommand output: %s", err, string(cmdOutput))
|
||||
return
|
||||
}
|
||||
|
||||
// Verify output file was created
|
||||
if _, err := os.Stat(config.OutputPath); os.IsNotExist(err) {
|
||||
t.Errorf("Output file was not created: %s\nCommand output: %s", config.OutputPath, string(cmdOutput))
|
||||
} else {
|
||||
t.Logf("Successfully created AV1 output file: %s", config.OutputPath)
|
||||
info, _ := os.Stat(config.OutputPath)
|
||||
if info.Size() == 0 {
|
||||
t.Errorf("Output file was created but is empty\nCommand output: %s", string(cmdOutput))
|
||||
} else {
|
||||
t.Logf("Output file size: %d bytes", info.Size())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestIntegration_Encode_EXR_VP9_WithAlpha(t *testing.T) {
|
||||
if testing.Short() {
|
||||
t.Skip("Skipping integration test in short mode")
|
||||
}
|
||||
|
||||
// Check if example file exists
|
||||
exampleDir := filepath.Join("..", "..", "..", "examples")
|
||||
exrFile := filepath.Join(exampleDir, "frame_0800.exr")
|
||||
if _, err := os.Stat(exrFile); os.IsNotExist(err) {
|
||||
t.Skipf("Example file not found: %s", exrFile)
|
||||
}
|
||||
|
||||
// Check if VP9 encoder is available
|
||||
checkCmd := exec.Command("ffmpeg", "-hide_banner", "-encoders")
|
||||
output, err := checkCmd.CombinedOutput()
|
||||
if err != nil || !strings.Contains(string(output), "libvpx-vp9") {
|
||||
t.Skip("VP9 encoder (libvpx-vp9) not available in ffmpeg")
|
||||
}
|
||||
|
||||
// Get absolute paths
|
||||
workspaceRoot, err := filepath.Abs(filepath.Join("..", "..", ".."))
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get workspace root: %v", err)
|
||||
}
|
||||
exampleDirAbs, err := filepath.Abs(exampleDir)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get example directory: %v", err)
|
||||
}
|
||||
tmpDir := filepath.Join(workspaceRoot, "tmp")
|
||||
if err := os.MkdirAll(tmpDir, 0755); err != nil {
|
||||
t.Fatalf("Failed to create tmp directory: %v", err)
|
||||
}
|
||||
|
||||
encoder := &SoftwareEncoder{codec: "libvpx-vp9"}
|
||||
config := &EncodeConfig{
|
||||
InputPattern: filepath.Join(exampleDirAbs, "frame_%04d.exr"),
|
||||
OutputPath: filepath.Join(tmpDir, "test_exr_vp9_alpha.webm"),
|
||||
StartFrame: 800,
|
||||
FrameRate: 24.0,
|
||||
WorkDir: tmpDir,
|
||||
UseAlpha: true, // Test with alpha
|
||||
TwoPass: false, // Use single pass for faster testing
|
||||
SourceFormat: "exr",
|
||||
}
|
||||
|
||||
// Build and run command
|
||||
cmd := encoder.BuildCommand(config)
|
||||
if cmd == nil {
|
||||
t.Fatal("BuildCommand returned nil")
|
||||
}
|
||||
|
||||
// Capture stderr to see what went wrong
|
||||
cmdOutput, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Errorf("FFmpeg command failed: %v\nCommand output: %s", err, string(cmdOutput))
|
||||
return
|
||||
}
|
||||
|
||||
// Verify output file was created
|
||||
if _, err := os.Stat(config.OutputPath); os.IsNotExist(err) {
|
||||
t.Errorf("Output file was not created: %s\nCommand output: %s", config.OutputPath, string(cmdOutput))
|
||||
} else {
|
||||
t.Logf("Successfully created VP9 output file with alpha: %s", config.OutputPath)
|
||||
info, _ := os.Stat(config.OutputPath)
|
||||
if info.Size() == 0 {
|
||||
t.Errorf("Output file was created but is empty\nCommand output: %s", string(cmdOutput))
|
||||
} else {
|
||||
t.Logf("Output file size: %d bytes", info.Size())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Helper function to copy files
|
||||
func copyFile(src, dst string) error {
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return os.WriteFile(dst, data, 0644)
|
||||
}
|
||||
Reference in New Issue
Block a user