我是Go的新手!
我正在做一个简单的测试,即从ffmpeg读取输出并写入文件。
我知道我可以用不同的方式来做,只需转换,但这是一个项目的开始,我想稍后操作读取的字节,更改它们然后将它们发送到输出。并且输入将是UDP并且输出也将是UDP,也就是说,我将得到ffmpeg输出我将按照我希望的方式处理字节,然后我将这些字节作为输入投入另一个ffmpeg进程,其输出为UDP as好。
通过这个简单的测试,文件的结果不能在VLC中运行,我相信我正确地在输出文件中写入字节,但输出文件总是比输入文件少1MB。
我希望得到一些帮助来阐明我正在做的这个测试的最佳方式是什么,基于我可以离开这个地方。我不知道这是不是错了,但我的印象是它。
输入文件是4K,h264的视频,我相信输出应该是相同的,因为在这个简单的测试中,我只是简单地阅读文件中cmd写入的内容。
按照分析代码:
package main
import (
"os/exec"
"os"
)
func verificaErro(e error) {
if e != nil {
panic(e)
}
}
func main() {
dir, _ := os.Getwd()
cmdName := "ffmpeg"
args := []string{
"-hide_banner",
"-re",
"-i",
dir + "\\teste-4k.mp4",
"-preset",
"superfast",
"-c:v",
"h264",
"-crf",
"0",
"-c",
"copy",
"-f", "rawvideo", "-",
}
cmd := exec.Command(cmdName, args...)
stdout, err := cmd.StdoutPipe()
verificaErro(err)
err2 := cmd.Start()
verificaErro(err2)
fileOutput := dir + "/out.raw"
var _, err3 = os.Stat(fileOutput)
if os.IsNotExist(err3) {
var file, err = os.Create(fileOutput)
verificaErro(err)
defer file.Close()
}
f, err4 := os.OpenFile(dir+"/out.raw", os.O_RDWR|os.O_APPEND, 0666)
verificaErro(err4)
bytes := make([]byte, 1024)
for {
_, err5 := stdout.Read(bytes)
if err5 != nil {
continue
}
if len(bytes) > 0 {
_, err6 := f.Write(bytes)
verificaErro(err6)
} else {
break
}
}
f.Close()
}
答案 0 :(得分:2)
您必须检查stdout.Read
的返回值。请注意,读取的字节数(nr)可能小于缓冲区大小,因此您需要重新切片缓冲区以获取有效内容。修改阅读循环如下:
chunk := make([]byte, 40*1024)
for {
nr, err5 := stdout.Read(chunk)
fmt.Printf("Read %d bytes\n", nr)
//do something with the data
//e.g. write to file
if nr > 0 {
validData := chunk[:nr]
nw, err6 := f.Write(validData)
fmt.Printf("Write %d bytes\n", nw)
verificaErro(err6)
}
if err5 != nil {
//Reach end of file (stream), exit from loop
if err5 == io.EOF {
break
}
fmt.Printf("Error = %v\n", err5)
continue
}
}
if err := cmd.Wait(); err != nil {
fmt.Printf("Wait command error: %v\n", err)
}
另一种解决方案是利用io.Copy
将整个输出复制到golang缓冲区。代码段如下所示:
var buf bytes.Buffer
n, err := io.Copy(&buf, stdout)
verificaErro(err)
fmt.Printf("Copied %d bytes\n", n)
err = cmd.Wait()
fmt.Printf("Wait error %v\n", err)
//do something with the data
data := buf.Bytes()
f, err4 := os.OpenFile(dir+"/out.raw", os.O_RDWR|os.O_APPEND, 0666)
verificaErro(err4)
defer f.Close()
nw, err := f.Write(data)
f.Sync()
fmt.Printf("Write size %d bytes\n", nw)