• FFmpeg 音视频转封装（MP4与FLV互转，流数据转FLV、MP4）

    1.简介

    音视频文件转封装操作就是把一种格式转换为另外一种格式，例如从 flv 转到 MP4，或者把流地址数据转换为MP4。

    本文主要讲解 流地址数据 转为 flv文件。

    2.流程

    2.1 在使用FFmpeg API之前，需要先注册API，然后才能使用API。当然，新版本的库不需要再调用下面的方法。

    av_register_all()

    2.2构建输入AVFormatContext

    声明输入的封装结构体，通过输入文件或者流地址作为封装结构的句柄。韩国电视台的流地址rtmp://mobliestream.c3tv.com:554/live/goodtv.sdp。

    1.     AVFormatContext* inputFmtCtx = nullptr;
    2.     const char* inputUrl = "rtmp://mobliestream.c3tv.com:554/live/goodtv.sdp";
    3.  
    4.     ///打开输入的流，获取数据 begin
    5.     int ret = avformat_open_input(&inputFmtCtx, inputUrl, NULL, NULL);

    2.3 查找音视频流信息，通过下面的接口与AVFormatContext中建立输入文件对应的流信息。

    1.  
    2. 	//查找;
    3. 	if (avformat_find_stream_info(inputFmtCtx, NULL) < 0)
    4. 	{
    5. 		printf("Couldn't find stream information.\n");
    6. 		return -1;
    7. 	}

    2.4构建输出AVFormatContext

    1.     //输出的文件
    2.     AVOutputFormat *ofmt = NULL;
    3.     AVFormatContext *ofmt_ctx = NULL;
    4.     const char* out_filename = "out.flv";
    5.  
    6.     avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, out_filename);
    7.     if (!ofmt_ctx) 
    8.     {
    9.         return -1;
    10.     }

    2.5申请输出的stream信息。

    1.         AVStream* out_stream = NULL;
    2.         //创建一个新的流
    3.         out_stream = avformat_new_stream(ofmt_ctx, NULL); 
    4.         if (!out_stream) 
    5.         {
    6.             return -1;
    7.         }

    2.6信息的复制，输出的stream信息建立完成之后，需要从输入的stream中将信息复制到输出的stream中。

    1.         //输入的流 视频、音频、字幕等
    2.         AVStream* in_stream = ifmt_ctx->streams[i];
    3.         AVCodecParameters* in_codecpar = in_stream->codecpar;
    4.  
    5.         //复制输入的流信息到输出流中
    6.         ret = avcodec_parameters_copy(out_stream->codecpar, in_codecpar);
    7.         if (ret < 0) 
    8.         {
    9.             return -1;
    10.         }

    2.7然后打开文件

    1.         //打开输出文件
    2.         ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE); 
    3.         if (ret < 0) 
    4.         {
    5.             return -1;
    6.         }

    2.8写文件头

    1.     //写入头
    2.     ret = avformat_write_header(ofmt_ctx, NULL);
    3.     if (ret < 0) 
    4.     {
    5.         return -1;
    6.     }

    2.9数据包读取和写入

    输入和输出均已打开，接下来可以从输入格式中读取数据包，然后将数据包写入到输出文件中，当然，随着输入的封装格式与输出的封装格式差别化，时间戳也需要进行计算改变。

    1.     AVPacket pkt;
    2.     while (1) 
    3.     {
    4.         AVStream* in_stream = NULL;
    5.         AVStream* out_stream = NULL;
    6.  
    7.         //从输入流中读取数据到pkt中
    8.         ret = av_read_frame(ifmt_ctx, &pkt);
    9.         if (ret < 0)
    10.             break;
    11.  
    12.         in_stream = ifmt_ctx->streams[pkt.stream_index];
    13.         if (pkt.stream_index >= stream_mapping_size || stream_mapping[pkt.stream_index] < 0) 
    14.         {
    15.             av_packet_unref(&pkt);
    16.             continue;
    17.         }
    18.         pkt.stream_index = stream_mapping[pkt.stream_index];
    19.         out_stream = ofmt_ctx->streams[pkt.stream_index];
    20.  
    21.         /* copy packet */
    22.         pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
    23.         pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
    24.         pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);
    25.         pkt.pos = -1;
    26.  
    27.         ret = av_interleaved_write_frame(ofmt_ctx, &pkt);
    28.         if (ret < 0) 
    29.         {
    30.             fprintf(stderr, "Error muxing packet\n");
    31.             break;
    32.         }
    33.         av_packet_unref(&pkt);
    34.     }

    2.10写文件尾

    1.     //写文件尾
    2.     av_write_trailer(ofmt_ctx);

    2.11收尾，关闭输入输出，释放资源。

    1.     //关闭
    2.     avformat_close_input(&ifmt_ctx);
    3.  
    4.     if (ofmt_ctx && !(ofmt->flags & AVFMT_NOFILE))
    5.         avio_closep(&ofmt_ctx->pb);
    6.  
    7.     avformat_free_context(ofmt_ctx);
    8.     av_freep(&stream_mapping);
    9.     if (ret < 0 && ret != AVERROR_EOF)
    10.     {
    11.         return -1;
    12.     }

    3.源码

    1. #include "pch.h"
    2. #include 
    3.  
    4. extern "C"
    5. {
    6. #include "libavformat/avformat.h"
    7. #include "libavutil/dict.h"
    8. #include "libavutil/opt.h"
    9. #include "libavutil/timestamp.h"
    10. #include "libswscale/swscale.h"
    11. #include "libswresample/swresample.h"
    12. };
    13.  
    14. int main()
    15. {
    16. 	//av_register_all();
    17. 	avformat_network_init();
    18.  
    19.     AVFormatContext* ifmt_ctx = NULL;
    20. 	const char* inputUrl = "rtmp://mobliestream.c3tv.com:554/live/goodtv.sdp";
    21.  
    22. 	///打开输入的流
    23. 	int ret = avformat_open_input(&ifmt_ctx, inputUrl, NULL, NULL);
    24. 	if (ret != 0)
    25. 	{
    26. 		printf("Couldn't open input stream.\n");
    27. 		return -1;
    28. 	}
    29.  
    30. 	//查找流信息
    31. 	if (avformat_find_stream_info(ifmt_ctx, NULL) < 0)
    32. 	{
    33. 		printf("Couldn't find stream information.\n");
    34. 		return -1;
    35. 	}
    36.  
    37.     //输出的文件
    38.     AVOutputFormat *ofmt = NULL;
    39.     AVFormatContext *ofmt_ctx = NULL;
    40.     const char* out_filename = "out.flv";
    41.  
    42.     avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, out_filename);
    43.     if (!ofmt_ctx) 
    44.     {
    45.         return -1;
    46.     }
    47.  
    48.     int stream_mapping_size = ifmt_ctx->nb_streams;
    49.  
    50.     //为数组分配内存
    51.     int* stream_mapping = (int *)av_mallocz_array(stream_mapping_size, sizeof(*stream_mapping));
    52.     if (!stream_mapping) 
    53.     {
    54.         return -1;
    55.     }
    56.  
    57.     int stream_index = 0;
    58.     ofmt = ofmt_ctx->oformat;
    59.     for (int i = 0; i < ifmt_ctx->nb_streams; i++) 
    60.     {
    61.         //输出的流
    62.         AVStream* out_stream = NULL;
    63.  
    64.         //输入的流 视频、音频、字幕等
    65.         AVStream* in_stream = ifmt_ctx->streams[i];
    66.         AVCodecParameters* in_codecpar = in_stream->codecpar;
    67.         if (in_codecpar->codec_type != AVMEDIA_TYPE_AUDIO && in_codecpar->codec_type != AVMEDIA_TYPE_VIDEO && in_codecpar->codec_type != AVMEDIA_TYPE_SUBTITLE) 
    68.         {
    69.             stream_mapping[i] = -1;
    70.             continue;
    71.         }
    72.         stream_mapping[i] = stream_index++;
    73.  
    74.         //创建一个新的流
    75.         out_stream = avformat_new_stream(ofmt_ctx, NULL); 
    76.         if (!out_stream) 
    77.         {
    78.             return -1;
    79.         }
    80.  
    81.         //复制输入的流信息到输出流中
    82.         ret = avcodec_parameters_copy(out_stream->codecpar, in_codecpar);
    83.         if (ret < 0) 
    84.         {
    85.             return -1;
    86.         }
    87.         out_stream->codecpar->codec_tag = 0;
    88.     }
    89.  
    90.     if (!(ofmt->flags & AVFMT_NOFILE)) 
    91.     {
    92.         //打开输出文件
    93.         ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE); 
    94.         if (ret < 0) 
    95.         {
    96.             return -1;
    97.         }
    98.     }
    99.  
    100.     //写入头
    101.     ret = avformat_write_header(ofmt_ctx, NULL);
    102.     if (ret < 0) 
    103.     {
    104.         return -1;
    105.     }
    106.  
    107.     AVPacket pkt;
    108.     while (1) 
    109.     {
    110.         AVStream* in_stream = NULL;
    111.         AVStream* out_stream = NULL;
    112.  
    113.         //从输入流中读取数据到pkt中
    114.         ret = av_read_frame(ifmt_ctx, &pkt);
    115.         if (ret < 0)
    116.             break;
    117.  
    118.         in_stream = ifmt_ctx->streams[pkt.stream_index];
    119.         if (pkt.stream_index >= stream_mapping_size || stream_mapping[pkt.stream_index] < 0) 
    120.         {
    121.             av_packet_unref(&pkt);
    122.             continue;
    123.         }
    124.         pkt.stream_index = stream_mapping[pkt.stream_index];
    125.         out_stream = ofmt_ctx->streams[pkt.stream_index];
    126.  
    127.         /* copy packet */
    128.         pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
    129.         pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
    130.         pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);
    131.         pkt.pos = -1;
    132.  
    133.         ret = av_interleaved_write_frame(ofmt_ctx, &pkt);
    134.         if (ret < 0) 
    135.         {
    136.             fprintf(stderr, "Error muxing packet\n");
    137.             break;
    138.         }
    139.         av_packet_unref(&pkt);
    140.     }
    141.  
    142.     //写文件尾
    143.     av_write_trailer(ofmt_ctx);
    144.  
    145.     //关闭
    146.     avformat_close_input(&ifmt_ctx);
    147.  
    148.     if (ofmt_ctx && !(ofmt->flags & AVFMT_NOFILE))
    149.         avio_closep(&ofmt_ctx->pb);
    150.  
    151.     avformat_free_context(ofmt_ctx);
    152.     av_freep(&stream_mapping);
    153.     if (ret < 0 && ret != AVERROR_EOF)
    154.     {
    155.         return -1;
    156.     }
    157.  
    158.     return 0;
    159. }
    160.

  • 相关阅读:
    Stable Diffusion中的embedding
    Cadence Allegro PCB设计88问解析(十七) 之 Allegro中焊盘的全连接和花焊盘
    Sulfo-CY3 azide水溶性荧光探针 星戈瑞
    【OpenCV】- 多边形将轮廓包围
    postgis 地理化函数
    质量小议15 -- 催
    《向量数据库指南》——Milvus Cloud 版本迭代的依据的是用户的反馈和对市场趋势的判断
    CompletableFuture异步编排（两任务组合——两个任务必须都完成才触发另一个任务 ）
    网络安全笔记-WebShell与文件上传
    【LeetCode】Day132-单词搜索
  • 原文地址：https://blog.csdn.net/wzz953200463/article/details/125793912