前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >2023-02-25:请用go语言调用ffmpeg,解码mp4文件并保存为YUV420SP格式文件,YUV420P不要转换成YUV420SP。

2023-02-25:请用go语言调用ffmpeg,解码mp4文件并保存为YUV420SP格式文件,YUV420P不要转换成YUV420SP。

原创
作者头像
福大大架构师每日一题
发布2023-02-25 21:31:40
2900
发布2023-02-25 21:31:40
举报

2023-02-25:请用go语言调用ffmpeg,解码mp4文件并保存为YUV420SP格式文件,YUV420P不要转换成YUV420SP。

答案2023-02-25:

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

解码后就是yuv420p,先保存y分量,然后u分量和v分量交替保存,这样就是yuv420sp了。

命令如下:

```bash

go run ./examples/a07.3video_decode_mp42yuv420sp/main.go

```

参考了[07:解码视频并保存为YUV格式文件](https://feater.top/ffmpeg/ffmpeg-decode-video-and-save-as-yuv-file-in-yuv420p-or-yuv420sp),代码用golang编写。代码如下:

```go

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

ptru := uintptr(unsafe.Pointer(yuvFrame.Data[1]))

ptrv := uintptr(unsafe.Pointer(yuvFrame.Data[2]))

for j := int32(0); j < w*h/4; j++ {

bytes = append(bytes, *(*byte)(unsafe.Pointer(ptru)))

ptru++

bytes = append(bytes, *(*byte)(unsafe.Pointer(ptrv)))

ptrv++

}

//写文件

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", "nv12", "-video_size", "640x360", "./out/result.yuv").Output()

if err != nil {

fmt.Println("play err = ", err)

}

}

```

![在这里插入图片描述](https://img-blog.csdnimg.cn/1d5481a62a7845e5a59083a8dd37122d.png)

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
区块链
云链聚未来,协同无边界。腾讯云区块链作为中国领先的区块链服务平台和技术提供商,致力于构建技术、数据、价值、产业互联互通的区块链基础设施,引领区块链底层技术及行业应用创新,助力传统产业转型升级,推动实体经济与数字经济深度融合。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
http://www.vxiaotou.com