diff --git a/examples/go.mod b/examples/go.mod new file mode 100644 index 0000000..7ce3ade --- /dev/null +++ b/examples/go.mod @@ -0,0 +1,7 @@ +module github.com/u2takey/ffmpeg-go/examples + +go 1.14 + +replace github.com/u2takey/ffmpeg-go => ../../ffmpeg-go + +require github.com/u2takey/ffmpeg-go v0.0.0-00010101000000-000000000000 diff --git a/examples/go.sum b/examples/go.sum new file mode 100644 index 0000000..fdd0fa1 --- /dev/null +++ b/examples/go.sum @@ -0,0 +1,44 @@ +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/docker/distribution v2.7.1+incompatible/go.mod h1:J2gT2udsDAN96Uj4KfcMRqY0/ypR+oyYUYmja8H+y+w= +github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= +github.com/go-logr/logr v0.1.0/go.mod h1:ixOQHD9gLJUVQQ2ZOR7zLEifBX6tGkNJF4QyIY7sIas= +github.com/golang/groupcache v0.0.0-20190129154638-5b532d6fd5ef/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc= +github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8= +github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/opencontainers/go-digest v1.0.0-rc1/go.mod h1:cMLVZDEM3+U2I4VmLI6N8jQYUd2OVphdqWwCJHrFt2s= +github.com/opencontainers/selinux v1.5.2/go.mod h1:yTcKuYAh6R95iDpefGLQaPaRwJFwyzAJufJyiTt7s0g= +github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/spf13/afero v1.2.2/go.mod h1:9ZxEEn6pIJ8Rxe320qSDBk6AsU0r9pR7Q4OcevTdifk= +github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk= +github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4= +github.com/tidwall/gjson v1.6.3 h1:aHoiiem0dr7GHkW001T1SMTJ7X5PvyekH5WX0whWGnI= +github.com/tidwall/gjson v1.6.3/go.mod h1:BaHyNc5bjzYkPqgLq7mdVzeiRtULKULXLgZFKsxEHI0= +github.com/tidwall/match v1.0.1 h1:PnKP62LPNxHKTwvHHZZzdOAOCtsJTjo6dZLCwpKm5xc= +github.com/tidwall/match v1.0.1/go.mod h1:LujAq0jyVjBy028G1WhWfIzbpQfMO8bBZ6Tyb0+pL9E= +github.com/tidwall/pretty v1.0.2 h1:Z7S3cePv9Jwm1KwS0513MRaoUe3S01WPbLNV40pwWZU= +github.com/tidwall/pretty v1.0.2/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk= +github.com/u2takey/go-utils v0.0.0-20200713025200-4704d09fc2c7 h1:PT7mE8HJE1mwaSazrOdSeByJ1FoV33/fHUZrBB+zwVU= +github.com/u2takey/go-utils v0.0.0-20200713025200-4704d09fc2c7/go.mod h1:ATqKFpgjUIlhGRs8j59gXmu8Cmpo1QQEHV6vwu1hs28= +golang.org/x/sys v0.0.0-20191115151921-52ab43148777/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= +golang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10= +gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +sigs.k8s.io/yaml v1.2.0/go.mod h1:yfXDCHCao9+ENCvLSE62v9VSji2MKu5jeNfTrofGhJc= diff --git a/examples/in1.mp4 b/examples/in1.mp4 new file mode 100644 index 0000000..2c7d59e Binary files /dev/null and b/examples/in1.mp4 differ diff --git a/examples/stream.go b/examples/stream.go new file mode 100644 index 0000000..755ae7e --- /dev/null +++ b/examples/stream.go @@ -0,0 +1,155 @@ +package main + +import ( + "bytes" + "encoding/json" + "flag" + "fmt" + "io" + "log" + "sync" + + ffmpeg "github.com/u2takey/ffmpeg-go" +) + +var ( + InputFile = flag.String("in_filename", "./in1.mp4", "Input filename") + OutputFile = flag.String("out_filename", "./out.mp4", "Input filename") + Dream = flag.Bool("dream", false, "Use DeepDream frame processing (requires tensorflow)") +) + +// Buffer is a goroutine safe bytes.Buffer +type Buffer struct { + buffer bytes.Buffer + mutex sync.Mutex +} + +func (s *Buffer) Write(p []byte) (n int, err error) { + s.mutex.Lock() + defer s.mutex.Unlock() + return s.buffer.Write(p) +} + +func (s *Buffer) Read(p []byte) (n int, err error) { + s.mutex.Lock() + defer s.mutex.Unlock() + return s.buffer.Read(p) +} + +func getVideoSize(fileName string) (int, int) { + log.Println("Getting video size for", fileName) + data, err := ffmpeg.Probe(fileName) + if err != nil { + panic(err) + } + log.Println("got video info", data) + type VideoInfo struct { + Streams []struct { + CodecType string `json:"codec_type"` + Width int + Height int + } `json:"streams"` + } + vInfo := &VideoInfo{} + err = json.Unmarshal([]byte(data), vInfo) + if err != nil { + panic(err) + } + for _, s := range vInfo.Streams { + if s.CodecType == "video" { + return s.Width, s.Height + } + } + return 0, 0 +} + +func startFFmpegProcess1(infileName string, writer io.WriteCloser) <-chan error { + log.Println("Starting ffmpeg process1") + done := make(chan error) + go func() { + err := ffmpeg.Input(infileName). + Output("pipe:", + ffmpeg.KwArgs{ + "format": "rawvideo", "pix_fmt": "rgb24", + }). + WithOutput(writer). + Run() + log.Println("ffmpeg process1 done") + _ = writer.Close() + done <- err + close(done) + }() + return done +} + +func startFFmpegProcess2(outfileName string, buf io.Reader, width, height int) <-chan error { + log.Println("Starting ffmpeg process2") + done := make(chan error) + go func() { + err := ffmpeg.Input("pipe:", + ffmpeg.KwArgs{"format": "rawvideo", + "pix_fmt": "rgb24", "s": fmt.Sprintf("%dx%d", width, height), + }). + Output(outfileName, ffmpeg.KwArgs{"pix_fmt": "yuv420p"}). + OverWriteOutput(). + WithInput(buf). + Run() + log.Println("ffmpeg process2 done") + done <- err + close(done) + }() + return done +} + +func process(reader io.ReadCloser, writer io.WriteCloser, w, h int) { + go func() { + frameSize := w * h * 3 + buf := make([]byte, frameSize, frameSize) + for { + n, err := io.ReadFull(reader, buf) + if n == 0 || err == io.EOF { + _ = writer.Close() + return + } else if n != frameSize || err != nil { + panic(fmt.Sprintf("read error: %d, %s", n, err)) + } + for i := range buf { + buf[i] = buf[i] / 3 + } + n, err = writer.Write(buf) + if n != frameSize || err != nil { + panic(fmt.Sprintf("write error: %d, %s", n, err)) + } + } + }() + return +} + +func run(inFile, outFile string) { + w, h := getVideoSize(inFile) + log.Println(w, h) + + pr1, pw1 := io.Pipe() + pr2, pw2 := io.Pipe() + done1 := startFFmpegProcess1(inFile, pw1) + process(pr1, pw2, w, h) + done2 := startFFmpegProcess2(outFile, pr2, w, h) + err := <-done1 + if err != nil { + panic(err) + } + err = <-done2 + if err != nil { + panic(err) + } + log.Println("Done") +} + +func main() { + flag.Parse() + if *Dream == true { + fmt.Println("tensorflow mode not implemented, todo") + return + } + run(*InputFile, *OutputFile) +} diff --git a/probe.go b/probe.go index 578bfa0..855c882 100644 --- a/probe.go +++ b/probe.go @@ -15,8 +15,8 @@ import ( // The stderr output can be retrieved by accessing the // ``stderr`` property of the exception. -func Probe(fileName string, kwargs KwArgs) (string, error) { - return ProbeWithTimeout(fileName, 0, kwargs) +func Probe(fileName string, kwargs ...KwArgs) (string, error) { + return ProbeWithTimeout(fileName, 0, MergeKwArgs(kwargs)) } func ProbeWithTimeout(fileName string, timeOut time.Duration, kwargs KwArgs) (string, error) {