mirror of
https://github.com/photoprism/photoprism.git
synced 2025-09-27 05:08:13 +08:00
198 lines
8.4 KiB
Go
198 lines
8.4 KiB
Go
package ffmpeg
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/photoprism/photoprism/internal/ffmpeg/encode"
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
)
|
|
|
|
func TestTranscodeCmd(t *testing.T) {
|
|
ffmpegBin := "/usr/bin/ffmpeg"
|
|
|
|
t.Run("NoSource", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions("", encode.IntelAvc, 1500, "50M", "", "")
|
|
_, _, err := TranscodeCmd("", "", opt)
|
|
|
|
assert.Equal(t, "empty source filename", err.Error())
|
|
})
|
|
t.Run("NoDestination", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions("", encode.IntelAvc, 1500, "50M", "", "")
|
|
_, _, err := TranscodeCmd("VID123.mov", "", opt)
|
|
|
|
assert.Equal(t, "empty destination filename", err.Error())
|
|
})
|
|
t.Run("Animation", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions("", encode.IntelAvc, 1500, "50M", "", "")
|
|
r, _, err := TranscodeCmd("VID123.gif", "VID123.gif.avc", opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Contains(t, r.String(), "bin/ffmpeg -y -strict -2 -i VID123.gif -pix_fmt yuv420p -vf scale='trunc(iw/2)*2:trunc(ih/2)*2' -f mp4 -movflags +faststart VID123.gif.avc")
|
|
})
|
|
t.Run("VP9toAVC", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions(ffmpegBin, encode.SoftwareAvc, 1500, "50M", "", "")
|
|
|
|
srcName := fs.Abs("./testdata/25fps.vp9")
|
|
destName := fs.Abs("./testdata/25fps.avc")
|
|
|
|
cmd, _, err := TranscodeCmd(srcName, destName, opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmdStr := cmd.String()
|
|
cmdStr = strings.Replace(cmdStr, srcName, "SRC", 1)
|
|
cmdStr = strings.Replace(cmdStr, destName, "DEST", 1)
|
|
|
|
assert.Equal(t, "/usr/bin/ffmpeg -y -strict -2 -i SRC -c:v libx264 -map 0:v:0 -map 0:a:0? -c:a aac -vf scale='if(gte(iw,ih), min(1500, iw), -2):if(gte(iw,ih), -2, min(1500, ih))',format=yuv420p -max_muxing_queue_size 1024 -crf 23 -r 30 -b:v 50M -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart DEST", cmdStr)
|
|
|
|
// Run generated command to test software transcoding.
|
|
RunCommandTest(t, opt.Encoder, srcName, destName, cmd, true)
|
|
})
|
|
t.Run("Vaapi", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions(ffmpegBin, encode.VaapiAvc, 1500, "50M", "", "")
|
|
|
|
srcName := fs.Abs("./testdata/25fps.vp9")
|
|
destName := fs.Abs("./testdata/25fps.vaapi.avc")
|
|
|
|
cmd, _, err := TranscodeCmd(srcName, destName, opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmdStr := cmd.String()
|
|
cmdStr = strings.Replace(cmdStr, srcName, "SRC", 1)
|
|
cmdStr = strings.Replace(cmdStr, destName, "DEST", 1)
|
|
|
|
assert.Equal(t, "/usr/bin/ffmpeg -y -strict -2 -hwaccel vaapi -i SRC -c:a aac -vf scale='if(gte(iw,ih), min(1500, iw), -2):if(gte(iw,ih), -2, min(1500, ih))',format=nv12,hwupload -c:v h264_vaapi -map 0:v:0 -map 0:a:0? -r 30 -b:v 50M -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart DEST", cmdStr)
|
|
|
|
// This transcoding test requires a supported hardware device that is properly configured:
|
|
if os.Getenv("PHOTOPRISM_FFMPEG_ENCODER") == "vaapi" {
|
|
RunCommandTest(t, encode.VaapiAvc, srcName, destName, cmd, true)
|
|
}
|
|
})
|
|
t.Run("IntelHvc", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions(ffmpegBin, encode.IntelAvc, 1500, "50M", "", "")
|
|
|
|
// QuickTime MOV container with HVC1 (HEVC) codec.
|
|
srcName := fs.Abs("./testdata/30fps.mov")
|
|
destName := fs.Abs("./testdata/30fps.intel.avc")
|
|
|
|
cmd, _, err := TranscodeCmd(srcName, destName, opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmdStr := cmd.String()
|
|
cmdStr = strings.Replace(cmdStr, srcName, "SRC", 1)
|
|
cmdStr = strings.Replace(cmdStr, destName, "DEST", 1)
|
|
|
|
assert.Equal(t, "/usr/bin/ffmpeg -y -strict -2 -hwaccel qsv -hwaccel_output_format qsv -i SRC -c:a aac -vf scale_qsv=w='if(gte(iw,ih), min(1500, iw), -1)':h='if(gte(iw,ih), -1, min(1500, ih))':format=nv12 -c:v h264_qsv -map 0:v:0 -map 0:a:0? -r 30 -b:v 50M -bitrate 50M -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart DEST", cmdStr)
|
|
|
|
// This transcoding test requires a supported hardware device that is properly configured:
|
|
if os.Getenv("PHOTOPRISM_FFMPEG_ENCODER") == "intel" {
|
|
RunCommandTest(t, encode.IntelAvc, srcName, destName, cmd, true)
|
|
}
|
|
})
|
|
t.Run("IntelVp9", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions(ffmpegBin, encode.IntelAvc, 1500, "50M", "", "")
|
|
|
|
srcName := fs.Abs("./testdata/25fps.vp9")
|
|
destName := fs.Abs("./testdata/25fps.intel.avc")
|
|
|
|
cmd, _, err := TranscodeCmd(srcName, destName, opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmdStr := cmd.String()
|
|
cmdStr = strings.Replace(cmdStr, srcName, "SRC", 1)
|
|
cmdStr = strings.Replace(cmdStr, destName, "DEST", 1)
|
|
|
|
assert.Equal(t, "/usr/bin/ffmpeg -y -strict -2 -hwaccel qsv -hwaccel_output_format qsv -i SRC -c:a aac -vf scale_qsv=w='if(gte(iw,ih), min(1500, iw), -1)':h='if(gte(iw,ih), -1, min(1500, ih))':format=nv12 -c:v h264_qsv -map 0:v:0 -map 0:a:0? -r 30 -b:v 50M -bitrate 50M -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart DEST", cmdStr)
|
|
|
|
// This transcoding test requires a supported hardware device that is properly configured:
|
|
if os.Getenv("PHOTOPRISM_FFMPEG_ENCODER") == "intel" {
|
|
RunCommandTest(t, encode.IntelAvc, srcName, destName, cmd, true)
|
|
}
|
|
})
|
|
t.Run("NvidiaHvc", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions(ffmpegBin, encode.NvidiaAvc, 1500, "50M", "", "")
|
|
|
|
// QuickTime MOV container with HVC1 (HEVC) codec.
|
|
srcName := fs.Abs("./testdata/30fps.mov")
|
|
destName := fs.Abs("./testdata/30fps.nvidia.avc")
|
|
|
|
cmd, _, err := TranscodeCmd(srcName, destName, opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmdStr := cmd.String()
|
|
cmdStr = strings.Replace(cmdStr, srcName, "SRC", 1)
|
|
cmdStr = strings.Replace(cmdStr, destName, "DEST", 1)
|
|
|
|
assert.Equal(t, "/usr/bin/ffmpeg -y -strict -2 -hwaccel auto -i SRC -pix_fmt yuv420p -c:v h264_nvenc -map 0:v:0 -map 0:a:0? -c:a aac -preset 15 -pixel_format yuv420p -gpu any -vf scale='if(gte(iw,ih), min(1500, iw), -2):if(gte(iw,ih), -2, min(1500, ih))',format=yuv420p -rc:v constqp -cq 0 -tune 2 -r 30 -b:v 50M -profile:v 1 -level:v auto -coder:v 1 -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart DEST", cmdStr)
|
|
|
|
// This transcoding test requires a supported hardware device that is properly configured:
|
|
if os.Getenv("PHOTOPRISM_FFMPEG_ENCODER") == "nvidia" {
|
|
RunCommandTest(t, encode.NvidiaAvc, srcName, destName, cmd, true)
|
|
}
|
|
})
|
|
t.Run("NvidiaVp9", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions(ffmpegBin, encode.NvidiaAvc, 1500, "50M", "", "")
|
|
|
|
srcName := fs.Abs("./testdata/25fps.vp9")
|
|
destName := fs.Abs("./testdata/25fps.nvidia.avc")
|
|
|
|
cmd, _, err := TranscodeCmd(srcName, destName, opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmdStr := cmd.String()
|
|
cmdStr = strings.Replace(cmdStr, srcName, "SRC", 1)
|
|
cmdStr = strings.Replace(cmdStr, destName, "DEST", 1)
|
|
|
|
assert.Equal(t, "/usr/bin/ffmpeg -y -strict -2 -hwaccel auto -i SRC -pix_fmt yuv420p -c:v h264_nvenc -map 0:v:0 -map 0:a:0? -c:a aac -preset 15 -pixel_format yuv420p -gpu any -vf scale='if(gte(iw,ih), min(1500, iw), -2):if(gte(iw,ih), -2, min(1500, ih))',format=yuv420p -rc:v constqp -cq 0 -tune 2 -r 30 -b:v 50M -profile:v 1 -level:v auto -coder:v 1 -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart DEST", cmdStr)
|
|
|
|
// This transcoding test requires a supported hardware device that is properly configured:
|
|
if os.Getenv("PHOTOPRISM_FFMPEG_ENCODER") == "nvidia" {
|
|
RunCommandTest(t, encode.NvidiaAvc, srcName, destName, cmd, true)
|
|
}
|
|
})
|
|
t.Run("Apple", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions("", encode.AppleAvc, 1500, "50M", "", "")
|
|
r, _, err := TranscodeCmd("VID123.mov", "VID123.mov.avc", opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Contains(t, r.String(), "bin/ffmpeg -y -strict -2 -i VID123.mov -c:v h264_videotoolbox -map 0:v:0 -map 0:a:0? -c:a aac -vf scale='if(gte(iw,ih), min(1500, iw), -2):if(gte(iw,ih), -2, min(1500, ih))',format=yuv420p -profile high -level 51 -r 30 -b:v 50M -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart VID123.mov.avc")
|
|
})
|
|
t.Run("Video4Linux", func(t *testing.T) {
|
|
opt := encode.NewVideoOptions("", encode.V4LAvc, 1500, "50M", "", "")
|
|
r, _, err := TranscodeCmd("VID123.mov", "VID123.mov.avc", opt)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Contains(t, r.String(), "bin/ffmpeg -y -strict -2 -i VID123.mov -c:v h264_v4l2m2m -map 0:v:0 -map 0:a:0? -c:a aac -vf scale='if(gte(iw,ih), min(1500, iw), -2):if(gte(iw,ih), -2, min(1500, ih))',format=yuv420p -num_output_buffers 72 -num_capture_buffers 64 -max_muxing_queue_size 1024 -crf 23 -r 30 -b:v 50M -f mp4 -movflags frag_keyframe+empty_moov+default_base_moof+faststart VID123.mov.avc")
|
|
})
|
|
}
|