音视频

程序步骤

1. 处理参数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
char* src;  //输入数据
char* dst;  //输出数据

int *stream_map = NULL;

AVFormatContext *pFmtCtx = NULL;
AVFormatContext *oFmtCtx = NULL;

AVPacket pkt;

av_log_set_level(AV_LOG_DEBUG);
if(argc < 3){ //argv[0] 就是这个extra_audio名字
    av_log(NULL, AV_LOG_INFO, "arguments must be more than 3!");
    exit(-1);
}

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

2. 配置输入上下文，打开多媒体文件

通过 avformat_open_input 打开输入的多媒体文件，并初始化输入上下文：

1
2
3
4
if((ret = avformat_open_input(&pFmtCtx, src, NULL, NULL)) < 0){
    av_log(NULL, AV_LOG_ERROR, "%s\n", av_err2str(ret));
    exit(-1);
}

3. 配置输出上下文

通过 avformat_alloc_output_context2 为目标文件创建输出上下文：

1
2
3
4
5
avformat_alloc_output_context2(&oFmtCtx, NULL, NULL, dst);
if(!oFmtCtx){
    av_log(NULL ,AV_LOG_ERROR, "NO MEMORY");
    goto __ERROR;
}

4. 打开输出文件

通过 avio_open2 打开目标文件，用于写入处理后的多媒体数据：

1
2
3
4
5
ret = avio_open2(&oFmtCtx->pb, dst, AVIO_FLAG_WRITE, NULL, NULL);
if(ret < 0){
    av_log(oFmtCtx, AV_LOG_ERROR, "%s", av_err2str(ret));
    goto __ERROR;
}

5. 在输出上下文创建流

遍历输入文件的所有流，根据需要创建对应的输出流，同时初始化 stream_map：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
stream_map = av_calloc(pFmtCtx->nb_streams, sizeof(int));
if(!stream_map){
    av_log(NULL ,AV_LOG_ERROR, "NO MEMORY");
    goto __ERROR;
}

for(int i = 0; i < pFmtCtx->nb_streams; i++){
    AVStream *outStream = NULL;
    AVStream *inStream = pFmtCtx->streams[i];
    AVCodecParameters *inCodecPar = inStream->codecpar;
    if(inCodecPar->codec_type != AVMEDIA_TYPE_AUDIO && inCodecPar->codec_type != AVMEDIA_TYPE_VIDEO && inCodecPar->codec_type != AVMEDIA_TYPE_SUBTITLE){
        stream_map[i] = -1;
        continue;
    }
    stream_map[i] = stream_idx++;
    outStream = avformat_new_stream(oFmtCtx, NULL);
    if(!outStream){
        av_log(NULL ,AV_LOG_ERROR, "NO MEMORY");
        goto __ERROR;
    }

    //设置输出视频参数
    avcodec_parameters_copy(outStream->codecpar, inStream->codecpar);
    outStream->codecpar->codec_tag = 0; //自动设置编解码器，对于输出流在这里是自动设置编码器
}

6. 写多媒体文件头到目的文件

1
2
3
4
5
ret = avformat_write_header(oFmtCtx, NULL);
if(ret < 0){
    av_log(oFmtCtx, AV_LOG_ERROR, "%s", av_err2str(ret));
    goto __ERROR;
}

7. 从源多媒体文件中读取数据到目的文件中

通过 av_read_frame 读取输入文件的帧数据，并通过 av_interleaved_write_frame 将处理后的数据写入目标文件：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
while(av_read_frame(pFmtCtx, &pkt) >= 0){
    AVStream *inStream, *outStream;
    if(stream_map[pkt.stream_index] < 0){
        av_packet_unref(&pkt);
        continue;
    }
    inStream = pFmtCtx->streams[pkt.stream_index];
    pkt.stream_index = stream_map[pkt.stream_index];
    outStream = oFmtCtx->streams[pkt.stream_index];
    av_packet_rescale_ts(&pkt, inStream->time_base, outStream->time_base);
    //查看ffmpeg源码可以知道在写入包之前会有一个guess_pkt_duration(s, st, pkt),这个函数会自动设置duration
    //pkt.pos = -1; //用于索引标识，在特殊的场景下可以使用，比如合并多个文件。
    av_interleaved_write_frame(oFmtCtx, &pkt);
    av_packet_unref(&pkt);
}

8. 写多媒体文件尾到文件中

1
av_write_trailer(oFmtCtx);

9. 将申请的资源释放掉

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
__ERROR:
if(pFmtCtx){
    avformat_close_input(&pFmtCtx);
    pFmtCtx = NULL;
}
if(oFmtCtx->pb){
    avio_close(oFmtCtx->pb);
}
if(oFmtCtx){
    avformat_free_context(oFmtCtx);
    oFmtCtx = NULL;
}
if(stream_map){
    av_free(stream_map);
}

说些什么吧！

valine