FFmpeg学习记录（二）—— ffmpeg多媒体文件处理

news2024/10/7 2:23:53

1.日志系统

常用的日志级别：

AV_LOG_ERROR
AV_LOG_WARNING
AV_LOG_INFO
AV_LOG_DEBUG

#include <stdio.h>
#include <libavutil/log.h>

int main(int argc, char *argv[])
{
    av_log_set_level(AV_LOG_DEBUG);

    av_log(NULL, AV_LOG_DEBUG, "hello world!\n");

    return 0;
}

2.文件的删除与重命名

文件删除与重命名的API：

avpriv_io_delete()
avpriv_io_move()

	int ret;
    // 重命名
    ret = avpriv_io_move("111.txt", "222.txt");
    if(ret < 0) {
        av_log(NULL, AV_LOG_ERROR, "Failed to rename\n");
        return -1;
    }
    av_log(NULL, AV_LOG_INFO, "Success to renmae\n");


    // 删除
    ret = avpriv_io_delete("./test.txt");
    if(ret < 0) {
        av_log(NULL, AV_LOG_ERROR, "Failed to delete\n");
        return -1;
    }
    av_log(NULL, AV_LOG_INFO, "Success to delete\n");

3.目录操作

操作目录重要函数:

avio_open_dir()
avio_read_dir()
avio_close_dir()

操作目录重要结构体：

AVIODirContext
AVIODirEntry

    avdevice_register_all();

    int ret;
    AVIODirContext *ctx = NULL;
    AVIODirEntry *entry = NULL;
    av_log_set_level(AV_LOG_INFO);

    ret = avio_open_dir(&ctx, "./", NULL);
    if(ret < 0)
    {
        av_log(NULL, AV_LOG_ERROR, "Can't open dir: %s\n", av_err2str(ret));
        return -1;
    }

    while(1) {
        avio_read_dir(ctx, &entry);
        if(ret < 0)
        {
            av_log(NULL, AV_LOG_ERROR, "Can't read dir: %s\n", av_err2str(ret));
            goto __fail;
        }

        if(!entry) {
            break;
        }

        av_log(NULL, AV_LOG_INFO,"%-9s %12"PRId64"  %s\n", entry->size, entry->name);

        avio_free_directory_entry(&entry);
    }


__fail:
    avio_close_dir(&ctx);

4.抽取音频数据

1.处理一些参数
2.打开多媒体文件
3.从多媒体文件中找到音频流
4.打开目的文件的上下文
5.为目的文件，创建一个新的音频流
6.设置输出音频参数
7.写多媒体文件头到目的文件
8…从源多媒体文件中读音频数据到目的文件中
9.写多媒体文件尾到文件中
10.将申请的资源释放掉

步骤比较多，一点一点来实现：

 // 1.处理一些参数
    if(argc < 3) {
        av_log(NULL, AV_LOG_INFO, "argument must be 3");
        exit(-1);
    }

    src = argv[1];
    dst = argv[2];

    // 2.打开多媒体文件
    ret = avformat_open_input(&pFmtCtx, src, NULL, NULL);
    if(ret < 0) {
        av_log(NULL, AV_LOG_ERROR, "%s\n", av_err2str(ret));
        exit(-1);
    }

// 3.从多媒体文件中找到音频流
    idx = av_find_best_stream(pFmtCtx, AVMEDIA_TYPE_AUDIO, -1, -1, NULL, 0);
    if(idx < 0) {
        av_log(pFmtCtx, AV_LOG_ERROR, "Does not include audio\n");
        goto _ERROR;
    }

    // 4.打开目的文件的上下文
    oFmtCtx = avformat_alloc_context();
    if(!oFmtCtx) {
        av_log(NULL, AV_LOG_ERROR, "No Memory\n");
        goto _ERROR;
    }
    outFmt = av_guess_format(NULL, dst, NULL);
    oFmtCtx->oformat = outFmt;

    // 5.为目的文件，创建一个新的音频流
    outStream = av_format_new_stream(oFmtCtx, NULL);

    // 6.设置输出音频参数
    inStream = pFmtCtx->stream[idx];
    avcodec_parameters_copy(outStream->codecpar, inStream->codecpar);
    outStream->codecpar->codec_tag = 0;

    // 7.写多媒体文件头到目的文件
    ret = avformat_write_header(oFmtCtx, NULL);
    if(ret < 0) {
        av_log(NULL, AV_LOG_ERROR, "%s\n", av_err2str(ret));
        goto _ERROR;
    }

    // 8.聪源多媒体文件中读到音频数据到目的文件中
    while(av_read_frame(pFmtCtx, &pkt) >= 0) {
        if(pkt.stream_index == idx) {
            pkt.pts = av_rescale_q_rnd(pkt.pts, inStream->time_base, outStream->time_base, AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX);
            pkt.dts = pkt.dts;
            pkt.duration = av_rescale_q(pkt.duration, inStream->time_base, outStream->time_base);
            pkt.stream_index = 0;
            pkt.pos = -1;
            av_interleaved_write_frame(oFmtCtx, &pkt);
            av_packet_unref(&pkt);
        }
    }
    // 9.写多媒体文件尾到文件中
    av_write_trailer(oFmtCtx);

    // 10.将申请的资源释放掉
_ERROR:
    if(pFmtCtx) {
        avformat_close_input(&pFmtCtx);
        pFmtCtx = NULL;
    }

    if(oFmtCtx->pb) {
        avio_close(oFmtCtx->pb);
    }

    if(oFmtCtx) {
        avformat_free_context(oFmtCtx);
        oFmtCtx = NULL;
    }

6.抽取视频数据

与处理音频流程相同，简单修改即可

 // 3.从多媒体文件中找到视频流
    idx = av_find_best_stream(pFmtCtx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
    if(idx < 0) {
        av_log(pFmtCtx, AV_LOG_ERROR, "Does not include VIDEO\n");
        goto _ERROR;
    }

... ...
  // 8.从源多媒体文件中读到视频数据到目的文件中
  while(av_read_frame(pFmtCtx, &pkt) >= 0) {
        if(pkt.stream_index == idx) {
            pkt.pts = av_rescale_q_rnd(pkt.pts, inStream->time_base, outStream->time_base, AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX);
            pkt.dts = av_rescale_q_rnd(pkt.dts, inStream->time_base, outStream->time_base, AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX);
            // pkt.dts = pkt.dts;
            pkt.duration = av_rescale_q(pkt.duration, inStream->time_base, outStream->time_base);
            pkt.stream_index = 0;
            pkt.pos = -1;
            av_interleaved_write_frame(oFmtCtx, &pkt);
            av_packet_unref(&pkt);
        }
    }