johanels/photoprism
1
0
Fork 0
mirror of https://github.com/photoprism/photoprism.git synced 2025-12-12 00:34:13 +01:00
Code Issues Packages Projects Releases Wiki Activity

FFmpeg: Refactor extraction of JPEG and PNG images from videos #4604

Browse Source 
Signed-off-by: Michael Mayer <michael@photoprism.app>
This commit is contained in:
Michael Mayer
2025-01-09 10:24:53 +01:00
parent 8b7871fc02
commit e2195d535e
34 changed files with 370 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
internal/ffmpeg/extract_image_cmd_test.go Normal file
View File

@@ -0,0 +1,63 @@
package ffmpeg
import (
"strings"
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/photoprism/photoprism/internal/ffmpeg/encode"
"github.com/photoprism/photoprism/pkg/fs"
)
func TestExtractImageCmd(t *testing.T) {
opt := encode.NewPreviewImageOptions("/usr/bin/ffmpeg", time.Second*9)
srcName := fs.Abs("./testdata/25fps.vp9")
destName := fs.Abs("./testdata/25fps.jpg")
cmd := ExtractImageCmd(srcName, destName, opt)
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 -ss 00:00:03.000 -i SRC -vframes 1 DEST", cmdStr)
RunCommandTest(t, "jpg", srcName, destName, cmd, true)
}
func TestExtractJpegImageCmd(t *testing.T) {
opt := encode.NewPreviewImageOptions("/usr/bin/ffmpeg", time.Second*9)
srcName := fs.Abs("./testdata/25fps.vp9")
destName := fs.Abs("./testdata/25fps.jpeg")
cmd := ExtractJpegImageCmd(srcName, destName, opt)
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 -ss 00:00:03.000 -i SRC -vframes 1 DEST", cmdStr)
RunCommandTest(t, "jpeg", srcName, destName, cmd, true)
}
func TestExtractPngImageCmd(t *testing.T) {
opt := encode.NewPreviewImageOptions("/usr/bin/ffmpeg", time.Second*9)
srcName := fs.Abs("./testdata/25fps.vp9")
destName := fs.Abs("./testdata/25fps.png")
cmd := ExtractPngImageCmd(srcName, destName, opt)
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 -ss 00:00:03.000 -i SRC -vframes 1 DEST", cmdStr)
RunCommandTest(t, "png", srcName, destName, cmd, true)
}