compose gpu mode and cpu mode together

xucai.hong · xucai.hong · commit 1af837c1389e · 2021-09-18T09:44:11.000+08:00
diff --git a/ffmpeg/ffmpeg.go b/ffmpeg/ffmpeg.go
@@ -38,7 +38,7 @@ func New(cfg *Config) transcoder.Transcoder {
 }
 
 // Start ...
-func (t *Transcoder) Start(opts transcoder.Options) (<-chan transcoder.Progress, *os.Process, error) {
+func (t *Transcoder) Start(opts transcoder.Options, mode string) (<-chan transcoder.Progress, *os.Process, error) {
 
 	var stderrIn io.ReadCloser
 
@@ -58,7 +58,12 @@ func (t *Transcoder) Start(opts transcoder.Options) (<-chan transcoder.Progress,
 	}
 
 	// Append input file and standard options
-	args := append([]string{"-i", t.input}, opts.GetStrArguments()...)
+	var args []string
+	if mode == "gpu" {
+		args = append([]string{"-vsync", "0", "-hwaccel", "cuvid", "-hwaccel_output_format", "cuda", "-i", t.input}, opts.GetStrArguments()...)
+	} else {
+		args = append([]string{"-i", t.input}, opts.GetStrArguments()...)
+	}
 	outputLength := len(t.output)
 	optionsLength := len(t.options)
 
@@ -94,6 +99,10 @@ func (t *Transcoder) Start(opts transcoder.Options) (<-chan transcoder.Progress,
 		cmd = exec.CommandContext(*t.commandContext, t.config.FfmpegBinPath, args...)
 	}
 
+	if mode != "" {
+		fmt.Printf("ffmpeg command (%s) is: %s\n", mode, cmd.String())
+	}
+
 	// If progresss enabled, get stderr pipe and start progress process
 	if t.config.ProgressEnabled && !t.config.Verbose {
 		stderrIn, err = cmd.StderrPipe()
diff --git a/transcoder.go b/transcoder.go
@@ -8,7 +8,7 @@ import (
 
 // Transcoder ...
 type Transcoder interface {
-	Start(opts Options) (<-chan Progress, *os.Process, error)
+	Start(opts Options, mode string) (<-chan Progress, *os.Process, error)
 	Input(i string) Transcoder
 	InputPipe(w *io.WriteCloser, r *io.ReadCloser) Transcoder
 	Output(o string) Transcoder
