2023-02-28：moonfdd/ffmpeg-go是用go语言绑定ffmpeg的库，目前是github上最好用的库。请用go语言将yuv文件编码为h264文件。
Posted 福大大架构师每日一题
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了2023-02-28：moonfdd/ffmpeg-go是用go语言绑定ffmpeg的库，目前是github上最好用的库。请用go语言将yuv文件编码为h264文件。相关的知识，希望对你有一定的参考价值。
2023-02-28：moonfdd/ffmpeg-go是用go语言绑定ffmpeg的库，目前是github上最好用的库。请用go语言将yuv文件编码为h264文件。
答案2023-02-28：
使用 github.com/moonfdd/ffmpeg-go 库。
命令如下：
go run ./examples/a12.video_encode_yuv2h264/main.go
参考了12：yuv编码为h264，代码用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
//是否存在yuv文件
_, err = os.Stat("./out/result.yuv")
if err != nil
if os.IsNotExist(err)
fmt.Println("create yuv file")
exec.Command("./lib/ffmpeg", "-i", "./resources/big_buck_bunny.mp4", "-pix_fmt", "yuv420p", "./out/result.yuv", "-y").CombinedOutput()
ret := int32(0) //默认返回值
//需要的变量名并初始化
var fmtCtx *libavformat.AVFormatContext
var outFmt *libavformat.AVOutputFormat
var vStream *libavformat.AVStream
pkt := libavcodec.AvPacketAlloc()
var codecCtx *libavcodec.AVCodecContext
var codec *libavcodec.AVCodec
var picture_buf uintptr
var picFrame *libavutil.AVFrame
var size ffcommon.FSizeT
//[1]!打开视频文件
in_file, err := os.Open("./out/result.yuv")
if err != nil
fmt.Printf("can not open file!\\n")
return
defer in_file.Close()
for
//[2]!打开输出文件，并填充fmtCtx数据
in_w := int32(640)
in_h := int32(360)
frameCnt := 1440
outFile := "./out/result.h264"
os.Remove(outFile)
if libavformat.AvformatAllocOutputContext2(&fmtCtx, nil, "", outFile) < 0
fmt.Printf("Cannot alloc output file context.\\n")
break
outFmt = fmtCtx.Oformat
//[2]!
//[3]!打开输出文件
if libavformat.AvioOpen(&fmtCtx.Pb, outFile, libavformat.AVIO_FLAG_READ_WRITE) < 0
fmt.Printf("output file open failed.\\n")
break
//[3]!
//[4]!创建h264视频流，并设置参数
vStream = fmtCtx.AvformatNewStream(codec)
if vStream == nil
fmt.Printf("failed create new video stream.\\n")
break
vStream.TimeBase.Den = 25
vStream.TimeBase.Num = 1
//[4]!
//[5]!编码参数相关
codecPara := fmtCtx.GetStream(uint32(vStream.Index)).Codecpar
codecPara.CodecType = libavutil.AVMEDIA_TYPE_VIDEO
codecPara.Width = in_w
codecPara.Height = in_h
//[5]!
//[6]!查找编码器
codec = libavcodec.AvcodecFindEncoder(outFmt.VideoCodec)
if codec == nil
fmt.Printf("Cannot find any endcoder.\\n")
break
//[6]!
//[7]!设置编码器内容
codecCtx = codec.AvcodecAllocContext3()
codecCtx.AvcodecParametersToContext(codecPara)
if codecCtx == nil
fmt.Printf("Cannot alloc context.")
break
codecCtx.CodecId = outFmt.VideoCodec
codecCtx.CodecType = libavutil.AVMEDIA_TYPE_VIDEO
codecCtx.PixFmt = libavutil.AV_PIX_FMT_YUV420P
codecCtx.Width = in_w
codecCtx.Height = in_h
codecCtx.TimeBase.Num = 1
codecCtx.TimeBase.Den = 25
codecCtx.BitRate = 400000
codecCtx.GopSize = 12
if codecCtx.CodecId == libavcodec.AV_CODEC_ID_H264
codecCtx.Qmin = 10
codecCtx.Qmax = 51
codecCtx.Qcompress = 0.6
if codecCtx.CodecId == libavcodec.AV_CODEC_ID_MPEG2VIDEO
codecCtx.MaxBFrames = 2
if codecCtx.CodecId == libavcodec.AV_CODEC_ID_MPEG1VIDEO
codecCtx.MbDecision = 2
//[7]!
//[8]!打开编码器
if codecCtx.AvcodecOpen2(codec, nil) < 0
fmt.Printf("Open encoder failed.\\n")
break
//[8]!
fmtCtx.AvDumpFormat(0, outFile, 1) //输出 输出文件流信息
//初始化帧
picFrame = libavutil.AvFrameAlloc()
picFrame.Width = codecCtx.Width
picFrame.Height = codecCtx.Height
picFrame.Format = codecCtx.PixFmt
size = uint64(libavutil.AvImageGetBufferSize(codecCtx.PixFmt, codecCtx.Width, codecCtx.Height, 1))
picture_buf = libavutil.AvMalloc(size)
libavutil.AvImageFillArrays((*[4]*byte)(unsafe.Pointer(&picFrame.Data)), (*[4]int32)(unsafe.Pointer(&picFrame.Linesize)), (*byte)(unsafe.Pointer(picture_buf)), codecCtx.PixFmt,
codecCtx.Width, codecCtx.Height, 1)
//[9] --写头文件
ret = fmtCtx.AvformatWriteHeader(nil)
//[9]
y_size := codecCtx.Width * codecCtx.Height
pkt.AvNewPacket(int32(size * 3))
buf := make([]byte, size)
//picture_buf = uintptr(unsafe.Pointer(&buf[0]))
//[10] --循环编码每一帧
for i := 0; i < frameCnt; i++
//读入YUV
n, err := in_file.Read(buf)
if err != nil
fmt.Println("read end")
break
if n <= 0
break
for i := 0; i < n; i++
*(*byte)(unsafe.Pointer(picture_buf + uintptr(i))) = buf[i]
picFrame.Data[0] = (*byte)(unsafe.Pointer(picture_buf)) //亮度Y
picFrame.Data[1] = (*byte)(unsafe.Pointer(picture_buf + uintptr(y_size))) // U
picFrame.Data[2] = (*byte)(unsafe.Pointer(picture_buf + uintptr(y_size*5/4))) // V
// AVFrame PTS
picFrame.Pts = int64(i)
//编码
if codecCtx.AvcodecSendFrame(picFrame) >= 0
for codecCtx.AvcodecReceivePacket(pkt) >= 0
fmt.Printf("encoder success!\\n")
// parpare packet for muxing
pkt.StreamIndex = uint32(vStream.Index)
pkt.AvPacketRescaleTs(codecCtx.TimeBase, vStream.TimeBase)
pkt.Pos = -1
ret = fmtCtx.AvInterleavedWriteFrame(pkt)
if ret < 0
fmt.Printf("error is: %s.\\n", libavutil.AvErr2str(ret))
pkt.AvPacketUnref() //刷新缓存
//[10]
//[11] --Flush encoder
ret = flush_encoder(fmtCtx, codecCtx, vStream.Index)
if ret < 0
fmt.Printf("flushing encoder failed!\\n")
break
//[11]
//[12] --写文件尾
fmtCtx.AvWriteTrailer()
//[12]
break
// //===========================释放所有指针===============================//
libavcodec.AvPacketFree(&pkt)
codecCtx.AvcodecClose()
libavutil.AvFree(uintptr(unsafe.Pointer(picFrame)))
libavutil.AvFree(picture_buf)
if fmtCtx != nil
fmtCtx.Pb.AvioClose()
fmtCtx.AvformatFreeContext()
_, err = exec.Command("./lib/ffplay.exe", "./out/result.h264").Output()
if err != nil
fmt.Println("play err = ", err)
//刷新编码器
func flush_encoder(fmtCtx *libavformat.AVFormatContext, codecCtx *libavcodec.AVCodecContext, vStreamIndex int32) int32
var ret int32
enc_pkt := libavcodec.AvPacketAlloc()
enc_pkt.Data = nil
enc_pkt.Size = 0
if codecCtx.Codec.Capabilities&libavcodec.AV_CODEC_CAP_DELAY == 0
return 0
fmt.Printf("Flushing stream #%d encoder\\n", vStreamIndex)
ret = codecCtx.AvcodecSendFrame(nil)
if ret >= 0
for codecCtx.AvcodecReceivePacket(enc_pkt) >= 0
fmt.Printf("success encoder 1 frame.\\n")
// parpare packet for muxing
enc_pkt.StreamIndex = uint32(vStreamIndex)
enc_pkt.AvPacketRescaleTs(codecCtx.TimeBase, fmtCtx.GetStream(uint32(vStreamIndex)).TimeBase)
ret = fmtCtx.AvInterleavedWriteFrame(enc_pkt)
if ret < 0
break
return ret
以上是关于2023-02-28：moonfdd/ffmpeg-go是用go语言绑定ffmpeg的库，目前是github上最好用的库。请用go语言将yuv文件编码为h264文件。的主要内容，如果未能解决你的问题，请参考以下文章
2023-03-12：mp3音频解码为pcm，代码用go语言编写，调用moonfdd/ffmpeg-go库。
2023-03-03：请用go语言调用ffmpeg，摄像头捕获并编码为h264文件，不管音频。
2023-02-21：请用go语言调用ffmpeg，解码mp4文件，输出视频信息和总帧数。