2023-03-30:用Go语言改写FFmpeg示例decode_audio.c,实现高效音频解码。

2023-03-30:用Go语言改写FFmpeg示例decode_audio.c,实现高效音频解码。

答案2023-03-30:

这个程序的主要功能是将 MP2 音频文件解码为 PCM 格式,并输出到指定的输出文件中。下面是该程序的详细步骤:

1.导入所需的包
通过import语句导入了一些第三方库和FFmpeg相关的包。

2.定义变量
定义了一些必要的变量和常量,如输入和输出文件名、音频编解码器、编解码器上下文、音频解析器上下文、解析缓冲区、音频数据帧、采样格式等。

3.解析命令行参数
读取命令行传入的输入文件名和输出文件名。

4.初始化解析器和编码器
通过 AVCodecFindDecoder() 函数查找 MPEG 音频解码器并得到其指针,如果为空则表示未找到对应的解码器。接着调用 AVParserInit() 函数初始化一个解析器,用于从输入文件中解析出音频数据帧。同时也需要分配一个编解码器上下文(AVCodecContext)对象,并调用 AVCodecOpen2() 函数打开编解码器。

5.打开输入文件和输出文件
使用 os.Open() 函数打开输入文件,如果失败则退出程序。使用 os.Create() 函数创建输出文件,如果失败则需要释放相关资源并退出程序。

6.逐帧解码
循环读取输入文件,每次读取 AUDIO_INBUF_SIZE 大小的数据,然后使用 AVParserParse2() 函数将数据解析成音频数据帧 AVPacket,并调用解码函数 decode() 进行解码,将解码后的 PCM 数据输出到输出文件中。读取结束时需要调用 AVCodecSendPacket() 函数和 AVCodecReceiveFrame() 函数进行“flush”,以确保所有剩余的音频数据帧都被解码。

7.输出 PCM 文件信息
在程序结束前,输出 PCM 文件的格式信息(包括采样率、声道数、采样格式等),以供用户使用 ffplay 命令播放该文件。

8.释放资源
关闭输入文件和输出文件,释放编解码器上下文、解析器上下文、解析缓冲区、音频数据帧以及 AVPacket 等资源。

总体来说,这个程序通过FFmpeg库提供的API从输入文件中逐帧解码音频数据,并将解码后的PCM数据输出到指定的输出文件中。此外,它还提供了一些基本的错误处理和输出格式信息的功能。

执行命令:

./lib/ffmpeg -i ./resources/big_buck_bunny.mp4 -c:a mp2 ./out/big_buck_bunny.mp2

go run ./examples/internalexamples/decode_audio/main.go ./out/big_buck_bunny.mp2 ./out/big_buck_bunny.pcm

./lib/ffplay -f s16le -ac 2 -ar 22050 ./out/big_buck_bunny.pcm

golang代码如下:

package main

import (
    "fmt"
    "os"
    "unsafe"

    "github.com/moonfdd/ffmpeg-go/ffcommon"
    "github.com/moonfdd/ffmpeg-go/libavcodec"
    "github.com/moonfdd/ffmpeg-go/libavutil"
)

func main0() (ret ffcommon.FInt) {
    // ./lib/ffmpeg -i ./resources/big_buck_bunny.mp4 -c:a mp2 ./out/big_buck_bunny.mp2
    // go run ./examples/internalexamples/decode_audio/main.go ./out/big_buck_bunny.mp2 ./out/big_buck_bunny.pcm
    // ./lib/ffplay -f s16le -ac 2 -ar 22050 ./out/big_buck_bunny.pcm

    var outfilename, filename string
    var codec *libavcodec.AVCodec
    var c *libavcodec.AVCodecContext
    var parser *libavcodec.AVCodecParserContext
    var len0 ffcommon.FInt
    var f, outfile *os.File
    var inbuf [AUDIO_INBUF_SIZE + libavcodec.AV_INPUT_BUFFER_PADDING_SIZE]ffcommon.FUint8T
    var data *ffcommon.FUint8T
    var data_size ffcommon.FSizeT
    var pkt *libavcodec.AVPacket
    var decoded_frame *libavutil.AVFrame
    var sfmt libavutil.AVSampleFormat
    var n_channels ffcommon.FInt = 0
    var fmt0 string

    if len(os.Args) <= 2 {
        fmt.Printf("Usage: %s <input file> <output file>\n", os.Args[0])
        os.Exit(0)
    }
    filename = os.Args[1]
    outfilename = os.Args[2]

    pkt = libavcodec.AvPacketAlloc()

    /* find the MPEG audio decoder */
    codec = libavcodec.AvcodecFindDecoder(libavcodec.AV_CODEC_ID_MP2)
    if codec == nil {
        fmt.Printf("Codec not found\n")
        os.Exit(1)
    }

    parser = libavcodec.AvParserInit(int32(codec.Id))
    if parser == nil {
        fmt.Printf("Parser not found\n")
        os.Exit(1)
    }

    c = codec.AvcodecAllocContext3()
    if c == nil {
        fmt.Printf("Could not allocate audio codec context\n")
        os.Exit(1)
    }

    /* open it */
    if c.AvcodecOpen2(codec, nil) < 0 {
        fmt.Printf("Could not open codec\n")
        os.Exit(1)
    }

    var err error
    f, err = os.Open(filename)
    if err != nil {
        fmt.Printf("Could not open %s\n", filename)
        os.Exit(1)
    }

    outfile, err = os.Create(outfilename)
    if err != nil {
        libavutil.AvFree(uintptr(unsafe.Pointer(c)))
        os.Exit(1)
    }

    /* decode until eof */
    data = (*byte)(unsafe.Pointer(&inbuf))
    var n int
    n, _ = f.Read(inbuf[0:AUDIO_INBUF_SIZE])
    data_size = uint64(n)

    for data_size > 0 {
        if decoded_frame == nil {
            decoded_frame = libavutil.AvFrameAlloc()
            if decoded_frame == nil {
                fmt.Printf("Could not allocate audio frame\n")
                os.Exit(1)
            }
        }

        ret = parser.AvParserParse2(c, &pkt.Data, (*int32)(unsafe.Pointer(&pkt.Size)),
            data, int32(data_size),
            libavutil.AV_NOPTS_VALUE, libavutil.AV_NOPTS_VALUE, 0)
        if ret < 0 {
            fmt.Printf("Error while parsing\n")
            os.Exit(1)
        }
        data = (*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(data)) + uintptr(ret)))
        data_size -= uint64(ret)

        if pkt.Size != 0 {
            decode(c, pkt, decoded_frame, outfile)
        }

        if data_size < AUDIO_REFILL_THRESH {
            for i := uint64(0); i < data_size; i++ {
                inbuf[i] = *(*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(data)) + uintptr(i)))
            }
            data = (*byte)(unsafe.Pointer(&inbuf))
            n, _ = f.Read(inbuf[data_size:AUDIO_INBUF_SIZE])
            len0 = int32(n)
            if len0 > 0 {
                data_size += uint64(len0)
            }
        }
    }

    /* flush the decoder */
    pkt.Data = nil
    pkt.Size = 0
    decode(c, pkt, decoded_frame, outfile)

    /* print output pcm infomations, because there have no metadata of pcm */
    sfmt = c.SampleFmt

    if libavutil.AvSampleFmtIsPlanar(sfmt) != 0 {
        packed := libavutil.AvGetSampleFmtName(sfmt)
        pa := ""
        if packed == "" {
            pa = "?"
        } else {
            pa = packed
        }
        fmt.Printf("Warning: the sample format the decoder produced is planar (%s). This example will output the first channel only.\n", pa)
        sfmt = libavutil.AvGetPackedSampleFmt(sfmt)
    }

    n_channels = c.Channels
    for {
        ret = get_format_from_sample_fmt(&fmt0, sfmt)
        if ret < 0 {
            break
        }

        fmt.Printf("Play the output audio file with the command:\nffplay -f %s -ac %d -ar %d %s\n",
            fmt0, n_channels, c.SampleRate,
            outfilename)
        break
    }
    // end:
    outfile.Close()
    f.Close()

    libavcodec.AvcodecFreeContext(&c)
    parser.AvParserClose()
    libavutil.AvFrameFree(&decoded_frame)
    libavcodec.AvPacketFree(&pkt)

    return 0
}

const AUDIO_INBUF_SIZE = 20480
const AUDIO_REFILL_THRESH = 4096

func get_format_from_sample_fmt(fmt0 *string, sample_fmt libavutil.AVSampleFormat) (ret ffcommon.FInt) {
    switch sample_fmt {
    case libavutil.AV_SAMPLE_FMT_U8:
        *fmt0 = "u8"
    case libavutil.AV_SAMPLE_FMT_S16:
        *fmt0 = "s16le"
    case libavutil.AV_SAMPLE_FMT_S32:
        *fmt0 = "s32le"
    case libavutil.AV_SAMPLE_FMT_FLT:
        *fmt0 = "f32le"
    case libavutil.AV_SAMPLE_FMT_DBL:
        *fmt0 = "f64le"
    default:
        fmt.Printf("sample format %s is not supported as output format\n",
            libavutil.AvGetSampleFmtName(sample_fmt))
        ret = -1
    }
    return
}

func decode(dec_ctx *libavcodec.AVCodecContext, pkt *libavcodec.AVPacket, frame *libavutil.AVFrame, outfile *os.File) {
    var i, ch ffcommon.FInt
    var ret, data_size ffcommon.FInt

    /* send the packet with the compressed data to the decoder */
    ret = dec_ctx.AvcodecSendPacket(pkt)
    if ret < 0 {
        fmt.Printf("Error submitting the packet to the decoder\n")
        os.Exit(1)
    }

    /* read all the output frames (in general there may be any number of them */
    for ret >= 0 {
        ret = dec_ctx.AvcodecReceiveFrame(frame)
        if ret == -libavutil.EAGAIN || ret == libavutil.AVERROR_EOF {
            return
        } else if ret < 0 {
            fmt.Printf("Error during decoding\n")
            os.Exit(1)
        }
        data_size = libavutil.AvGetBytesPerSample(dec_ctx.SampleFmt)
        if data_size < 0 {
            /* This should not occur, checking just for paranoia */
            fmt.Printf("Failed to calculate data size\n")
            os.Exit(1)
        }
        bytes := []byte{}
        for i = 0; i < frame.NbSamples; i++ {
            for ch = 0; ch < dec_ctx.Channels; ch++ {
                ptr := uintptr(unsafe.Pointer(frame.Data[ch])) + uintptr(data_size*i)
                for k := int32(0); k < data_size; k++ {
                    bytes = append(bytes, *(*byte)(unsafe.Pointer(ptr)))
                    ptr++
                }
            }
        }
        outfile.Write(bytes)
    }
}

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-58.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
        }
    }

    main0()
}

1680181769268.jpg
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 230,578评论 6 544
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 99,701评论 3 429
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 178,691评论 0 383
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 63,974评论 1 318
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 72,694评论 6 413
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 56,026评论 1 329
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 44,015评论 3 450
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 43,193评论 0 290
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 49,719评论 1 336
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 41,442评论 3 360
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 43,668评论 1 374
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 39,151评论 5 365
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 44,846评论 3 351
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 35,255评论 0 28
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 36,592评论 1 295
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 52,394评论 3 400
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 48,635评论 2 380

推荐阅读更多精彩内容