写点什么

2023-02-23:请用 go 语言调用 ffmpeg,解码 mp4 文件并保存为 YUV420P 格式文件。

  • 2023-02-23
    北京
  • 本文字数:2941 字

    阅读完需:约 10 分钟

2023-02-23:请用go语言调用ffmpeg,解码mp4文件并保存为YUV420P格式文件。

2023-02-23:请用 go 语言调用 ffmpeg,解码 mp4 文件并保存为 YUV420P 格式文件。


答案 2023-02-23:


使用 github.com/moonfdd/ffmpeg-go 库。


解码后就是 yuv420p,直接保存 frame 的 yuv 分量即可。


命令如下:


go run ./examples/a07.1video_decode_mp42yuv420p/main.go
复制代码


参考了07:解码视频并保存为YUV格式文件,代码用 golang 编写。代码如下:


package main
import ( "fmt" "os" "os/exec" "unsafe"
"github.com/moonfdd/ffmpeg-go/ffcommon" "github.com/moonfdd/ffmpeg-go/libavcodec" "github.com/moonfdd/ffmpeg-go/libavformat" "github.com/moonfdd/ffmpeg-go/libavutil")
func main() { os.Setenv("Path", os.Getenv("Path")+";./lib") ffcommon.SetAvutilPath("./lib/avutil-56.dll") ffcommon.SetAvcodecPath("./lib/avcodec-58.dll") ffcommon.SetAvdevicePath("./lib/avdevice-56.dll") ffcommon.SetAvfilterPath("./lib/avfilter-56.dll") ffcommon.SetAvformatPath("./lib/avformat-58.dll") ffcommon.SetAvpostprocPath("./lib/postproc-55.dll") ffcommon.SetAvswresamplePath("./lib/swresample-3.dll") ffcommon.SetAvswscalePath("./lib/swscale-5.dll")
genDir := "./out" _, err := os.Stat(genDir) if err != nil { if os.IsNotExist(err) { os.Mkdir(genDir, 0777) // Everyone can read write and execute } }
filePath := "./resources/big_buck_bunny.mp4" //文件地址 videoStreamIndex := -1 //视频流所在流序列中的索引 ret := int32(0) //默认返回值
//需要的变量名并初始化 var fmtCtx *libavformat.AVFormatContext var pkt *libavformat.AVPacket var codecCtx *libavcodec.AVCodecContext var avCodecPara *libavcodec.AVCodecParameters var codec *libavcodec.AVCodec yuvFrame := libavutil.AvFrameAlloc()
for { //=========================== 创建AVFormatContext结构体 ===============================// //分配一个AVFormatContext,FFMPEG所有的操作都要通过这个AVFormatContext来进行 fmtCtx = libavformat.AvformatAllocContext() //==================================== 打开文件 ======================================// ret = libavformat.AvformatOpenInput(&fmtCtx, filePath, nil, nil) if ret != 0 { fmt.Printf("cannot open video file\n") break }
//=================================== 获取视频流信息 ===================================// ret = fmtCtx.AvformatFindStreamInfo(nil) if ret < 0 { fmt.Printf("cannot retrive video info\n") break }
//循环查找视频中包含的流信息,直到找到视频类型的流 //便将其记录下来 保存到videoStreamIndex变量中 for i := uint32(0); i < fmtCtx.NbStreams; i++ { if fmtCtx.GetStream(i).Codecpar.CodecType == libavutil.AVMEDIA_TYPE_VIDEO { videoStreamIndex = int(i) break //找到视频流就退出 } }
//如果videoStream为-1 说明没有找到视频流 if videoStreamIndex == -1 { fmt.Printf("cannot find video stream\n") break }
//打印输入和输出信息:长度 比特率 流格式等 fmtCtx.AvDumpFormat(0, filePath, 0)
//================================= 查找解码器 ===================================// avCodecPara = fmtCtx.GetStream(uint32(videoStreamIndex)).Codecpar codec = libavcodec.AvcodecFindDecoder(avCodecPara.CodecId) if codec == nil { fmt.Printf("cannot find decoder\n") break } //根据解码器参数来创建解码器内容 codecCtx = codec.AvcodecAllocContext3() codecCtx.AvcodecParametersToContext(avCodecPara) if codecCtx == nil { fmt.Printf("Cannot alloc context.") break }
//================================ 打开解码器 ===================================// ret = codecCtx.AvcodecOpen2(codec, nil) if ret < 0 { // 具体采用什么解码器ffmpeg经过封装 我们无须知道 fmt.Printf("cannot open decoder\n") break }
os.Remove("./out/result.yuv") file, err := os.OpenFile("./out/result.yuv", os.O_CREATE|os.O_RDWR, 0777) if err != nil { fmt.Println("open file failed,err:", err) return }
w := codecCtx.Width h := codecCtx.Height
//=========================== 分配AVPacket结构体 ===============================// i := 0 pkt = libavcodec.AvPacketAlloc() //分配一个packet pkt.AvNewPacket(codecCtx.Width * codecCtx.Height) //调整packet的数据
//=========================== 读取视频信息 ===============================// for fmtCtx.AvReadFrame(pkt) >= 0 { //读取的是一帧视频 数据存入一个AVPacket的结构中 if pkt.StreamIndex == uint32(videoStreamIndex) { if codecCtx.AvcodecSendPacket(pkt) == 0 { for codecCtx.AvcodecReceiveFrame(yuvFrame) == 0 { i++ bytes := []byte{} //y ptr := uintptr(unsafe.Pointer(yuvFrame.Data[0])) for j := int32(0); j < w*h; j++ { bytes = append(bytes, *(*byte)(unsafe.Pointer(ptr))) ptr++ } //u ptr = uintptr(unsafe.Pointer(yuvFrame.Data[1])) for j := int32(0); j < w*h/4; j++ { bytes = append(bytes, *(*byte)(unsafe.Pointer(ptr))) ptr++ } //v ptr = uintptr(unsafe.Pointer(yuvFrame.Data[2])) for j := int32(0); j < w*h/4; j++ { bytes = append(bytes, *(*byte)(unsafe.Pointer(ptr))) ptr++ } //写文件 file.Write(bytes) } } } pkt.AvPacketUnref() //重置pkt的内容 } fmt.Printf("There are %d frames int total.\n", i) file.Close() break } // //===========================释放所有指针===============================// libavcodec.AvPacketFree(&pkt) codecCtx.AvcodecClose() libavformat.AvformatCloseInput(&fmtCtx) fmtCtx.AvformatFreeContext() libavutil.AvFrameFree(&yuvFrame)
_, err = exec.Command("./lib/ffplay.exe", "-pixel_format", "yuv420p", "-video_size", "640x360", "./out/result.yuv").Output() if err != nil { fmt.Println("play err = ", err) }}
复制代码



发布于: 刚刚阅读数: 4
用户头像

还未添加个人签名 2021-02-15 加入

还未添加个人简介

评论

发布
暂无评论
2023-02-23:请用go语言调用ffmpeg,解码mp4文件并保存为YUV420P格式文件。_golang_福大大架构师每日一题_InfoQ写作社区