astra/Vidio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

buildSelectExpression func moved from video to utils

Browse source
This commit is contained in:
Krzysztofz01 2023-08-29 10:44:41 +02:00
parent 4fe8cad509
commit a1d63d0f9f
2 changed files with 41 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

41
utils.go
View file

@ -214,3 +214,44 @@ func getDevicesWindows() ([]string, error) {
devices := parseDevices(builder.String()) devices := parseDevices(builder.String())
return devices, nil return devices, nil
} }
// Error representing a strings.Builder failure in the buildSelectExpression func.
var errExpressionBuilder = fmt.Errorf("vidio: failed to write tokens to the frame select expresion")
// Helper function used to generate a "-vf select" expression that specifies which video frames should be exported.
func buildSelectExpression(n ...int) (string, error) {
sb := strings.Builder{}
if _, err := sb.WriteString("select='"); err != nil {
return "", errExpressionBuilder
}
for index, frame := range n {
if index != 0 {
if _, err := sb.WriteRune('+'); err != nil {
return "", errExpressionBuilder
}
}
if _, err := sb.WriteString("eq(n\\,"); err != nil {
return "", errExpressionBuilder
}
if _, err := sb.WriteString(strconv.Itoa(frame)); err != nil {
return "", errExpressionBuilder
}
if _, err := sb.WriteRune(')'); err != nil {
return "", errExpressionBuilder
}
}
if _, err := sb.WriteRune('\''); err != nil {
return "", errExpressionBuilder
}
return sb.String(), nil
}

42
video.go
View file

@ -6,7 +6,6 @@ import (
"os" "os"
"os/exec" "os/exec"
"os/signal" "os/signal"
"strconv"
"strings" "strings"
"syscall" "syscall"
) )
@ -405,44 +404,3 @@ func (video *Video) cleanup() {
os.Exit(1) os.Exit(1)
}() }()
} }
// Error representing a strings.Builder failure in the buildSelectExpression func.
var errExpressionBuilder = fmt.Errorf("vidio: failed to write tokens to the frame select expresion")
// Helper function used to generate a "-vf select" expression that specifies which video frames should be exported.
func buildSelectExpression(n ...int) (string, error) {
sb := strings.Builder{}
if _, err := sb.WriteString("select='"); err != nil {
return "", errExpressionBuilder
}
for index, frame := range n {
if index != 0 {
if _, err := sb.WriteRune('+'); err != nil {
return "", errExpressionBuilder
}
}
if _, err := sb.WriteString("eq(n\\,"); err != nil {
return "", errExpressionBuilder
}
if _, err := sb.WriteString(strconv.Itoa(frame)); err != nil {
return "", errExpressionBuilder
}
if _, err := sb.WriteRune(')'); err != nil {
return "", errExpressionBuilder
}
}
if _, err := sb.WriteRune('\''); err != nil {
return "", errExpressionBuilder
}
return sb.String(), nil
}