C++音视频03:媒体信息打印、提取音视频、转封装、剪辑 ...

版本：FFMpeg6.0
媒体信息打印

1. #include <libavformat/avformat.h>
2. #include <libavutil/log.h>
4. int main (int argc, char *argv[]) {
5.     AVFormatContext *fmt_cxt = NULL;
6.     int ret;
7.     av_log_set_level(AV_LOG_INFO);
9.     ret = avformat_open_input(&fmt_cxt, "/root/resource/1.mp4", NULL, NULL);
10.     if (ret < 0) {
11.         av_log(NULL, AV_LOG_ERROR, "Can't open file: %s", av_err2str(ret));
12.         return -1;
13.     }
14.         // 打印媒体信息
15.     av_dump_format(fmt_cxt, 0, "/root/resource/1.mp4", 0);
17.     avformat_close_input(&fmt_cxt);
18.    
19.     return 0;
20. }

复制代码

提取音视频

音频和视频的区别在于av_find_best_stream时将AVMEDIA_TYPE_VIDEO改为AVMEDIA_TYPE_AUDIO

1. #include <libavformat/avformat.h>
2. #include <libavutil/log.h>
3. #include <libavutil/avutil.h>
5. int main (int argc, char *argv[]) {
6.     // 1. 处理参数
7.     char *src;
8.     char *dst;
9.     int ret;
10.     int idx = -1;
11.     AVFormatContext *pFmtCtx = NULL;
12.     AVFormatContext *oFmtCtx = NULL;
14.     const AVOutputFormat *outFmt = NULL;
15.     AVStream *outStream = NULL;
16.     AVStream *inStream = NULL;
17.     AVPacket pkt;
19.     av_log_set_level(AV_LOG_DEBUG);
20.     if (argc < 3) {
21.         av_log(NULL, AV_LOG_ERROR, "argument must be more than 3\n");
22.         exit(-1);
23.     }
25.     src = argv[1];
26.     dst = argv[2];
28.     // 2. 打开多媒体文件
29.     ret = avformat_open_input(&pFmtCtx, src, NULL, NULL);
30.     if (ret < 0) {
31.         av_log(NULL, AV_LOG_ERROR, "%s\n", av_err2str(ret));
32.         exit(-1);
33.     }
34.     // 3. 从多媒体文件中找到音频流
35.     idx = av_find_best_stream(pFmtCtx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
36.     if (idx < 0) {
37.         av_log(pFmtCtx, AV_LOG_ERROR, "Does not include audio stream\n");
38.         goto _ERROR;
39.     }
40.     // 4. 打开目的文件的上下文
41.     oFmtCtx = avformat_alloc_context();
42.     if (!oFmtCtx) {
43.         av_log(pFmtCtx, AV_LOG_ERROR, "No memory\n");
44.         goto _ERROR;
45.     }
46.     outFmt = av_guess_format(NULL, dst, NULL);
47.     oFmtCtx->oformat = outFmt;
49.     // 5. 为目的文件创建新的音频流
50.     outStream = avformat_new_stream(oFmtCtx, NULL);
51.     // 6. 设置输出视频参数
52.     inStream = pFmtCtx->streams[idx];
53.     avcodec_parameters_copy(outStream->codecpar, inStream->codecpar);
54.     // 设置为0，会自动去找对应的编码器
55.     outStream->codecpar->codec_tag = 0;
57.     ret = avio_open2(&oFmtCtx->pb, dst, AVIO_FLAG_WRITE, NULL, NULL);
58.     if (ret < 0) {
59.         av_log(oFmtCtx, AV_LOG_ERROR, "%s\n", av_err2str(ret));
60.         goto _ERROR;
61.     }
62.    
63.     // 7. 写多媒体文件头到目的文件
64.     ret = avformat_write_header(oFmtCtx, NULL);
65.     if (ret < 0) {
66.         av_log(oFmtCtx, AV_LOG_ERROR, "%s\n", av_err2str(ret));
67.         goto _ERROR;
68.     }
69.     // 8. 从源多媒体文件中读音频数据到目的文件中
70.     while (av_read_frame(pFmtCtx, &pkt) >= 0) {
71.         if (pkt.stream_index == idx) {
72.             // 修改时间戳
73.             pkt.pts = av_rescale_q_rnd(pkt.pts, inStream->time_base, outStream->time_base, (AV_ROUND_INF | AV_ROUND_PASS_MINMAX));
74.             pkt.dts = av_rescale_q_rnd(pkt.dts, inStream->time_base, outStream->time_base, (AV_ROUND_INF | AV_ROUND_PASS_MINMAX));
75.             pkt.duration = av_rescale_q(pkt.duration, inStream->time_base, outStream->time_base);
76.             pkt.stream_index = 0;
77.             // 设置成-1，自己会去计算
78.             pkt.pos = -1;
79.             av_interleaved_write_frame(oFmtCtx, &pkt);
80.             av_packet_unref(&pkt);
81.         }
82.     }
83.     // 9. 写多媒体文件尾到目的文件
84.     av_write_trailer(oFmtCtx);
85.     // 10. 将申请的资源释放掉
86. _ERROR:
87.     if (pFmtCtx) {
88.         avformat_close_input(&pFmtCtx);
89.         pFmtCtx = NULL;
90.     }   
91.     if (oFmtCtx->pb) {
92.         avio_close(oFmtCtx->pb);
93.     }
94.     if (oFmtCtx) {
95.         avformat_free_context(oFmtCtx);
96.         oFmtCtx = NULL;
97.     }
98.     return 0;
99. }

复制代码

转封装

1. #include <libavformat/avformat.h>
2. #include <libavutil/log.h>
3. #include <libavutil/avutil.h>
5. int main (int argc, char *argv[]) {
6.     // 1. 处理参数
7.     char *src;
8.     char *dst;
10.     int *stream_map = NULL;
11.     int stream_idx = 0;
13.     int ret;
14.     int idx = -1;
15.     int i = 0;
16.     AVFormatContext *pFmtCtx = NULL;
17.     AVFormatContext *oFmtCtx = NULL;
19.     const AVOutputFormat *outFmt = NULL;
20.    
21.     AVPacket pkt;
23.     av_log_set_level(AV_LOG_DEBUG);
24.     if (argc < 3) {
25.         av_log(NULL, AV_LOG_ERROR, "argument must be more than 3\n");
26.         exit(-1);
27.     }
29.     src = argv[1];
30.     dst = argv[2];
32.     // 2. 打开多媒体文件
33.     ret = avformat_open_input(&pFmtCtx, src, NULL, NULL);
34.     if (ret < 0) {
35.         av_log(NULL, AV_LOG_ERROR, "%s\n", av_err2str(ret));
36.         exit(-1);
37.     }
38.    
39.     // 3. 打开目的文件的上下文
40.     avformat_alloc_output_context2(&oFmtCtx, NULL, NULL, dst);
41.     if (!oFmtCtx) {
42.         av_log(NULL, AV_LOG_ERROR, "NO MEMORY\n");
43.         goto _ERROR;
44.     }
46.     stream_map = av_calloc(pFmtCtx->nb_streams, sizeof(int));
47.     if (!stream_map) {
48.         av_log(NULL, AV_LOG_ERROR, "NO MEMORY\n");
49.         goto _ERROR;
50.     }
52.     for (i = 0; i < pFmtCtx->nb_streams; ++i) {
53.         AVStream *outStream = NULL;
54.         AVStream *inStream = pFmtCtx->streams[i];
55.         AVCodecParameters *inCodecPar = inStream->codecpar;
56.         if (inCodecPar->codec_type != AVMEDIA_TYPE_AUDIO &&
57.             inCodecPar->codec_type != AVMEDIA_TYPE_VIDEO &&
58.             inCodecPar->codec_type != AVMEDIA_TYPE_SUBTITLE) {
59.             stream_map[i] = -1;
60.             continue;
61.         }
62.         stream_map[i] = stream_idx++;
64.         // 4. 为目的文件创建新的音频流
65.         outStream = avformat_new_stream(oFmtCtx, NULL);
66.         if (!outStream) {
67.             av_log(oFmtCtx, AV_LOG_ERROR, "NO MEMORY\n");
68.             goto _ERROR;
69.         }
71.         avcodec_parameters_copy(outStream->codecpar, inStream->codecpar);
72.         // 设置为0，会自动去找对应的编码器
73.         outStream->codecpar->codec_tag = 0;
75.     }
76.    
77.     ret = avio_open2(&oFmtCtx->pb, dst, AVIO_FLAG_WRITE, NULL, NULL);
78.     if (ret < 0) {
79.         av_log(oFmtCtx, AV_LOG_ERROR, "%s\n", av_err2str(ret));
80.         goto _ERROR;
81.     }
82.    
83.     // 5. 写多媒体文件头到目的文件
84.     ret = avformat_write_header(oFmtCtx, NULL);
85.     if (ret < 0) {
86.         av_log(oFmtCtx, AV_LOG_ERROR, "%s\n", av_err2str(ret));
87.         goto _ERROR;
88.     }
89.     // 6. 从源多媒体文件中读音频数据到目的文件中
90.     while (av_read_frame(pFmtCtx, &pkt) >= 0) {
91.         AVStream *outStream = NULL;
92.         AVStream *inStream = NULL;
93.         
94.         if (stream_map[pkt.stream_index] < 0) {
95.             av_packet_unref(&pkt);
96.             continue;
97.         }
98.         inStream = pFmtCtx->streams[pkt.stream_index];
100.         pkt.stream_index = stream_map[pkt.stream_index];
102.         outStream = oFmtCtx->streams[pkt.stream_index];
103.         av_packet_rescale_ts(&pkt, inStream->time_base, outStream->time_base);
104.         // 设置成-1，自己会去计算
105.         pkt.pos = -1;
106.         av_interleaved_write_frame(oFmtCtx, &pkt);
107.         av_packet_unref(&pkt);
108.         
109.     }
110.     // 7. 写多媒体文件尾到目的文件
111.     av_write_trailer(oFmtCtx);
112.     // 8. 将申请的资源释放掉
113. _ERROR:
114.     if (pFmtCtx) {
115.         avformat_close_input(&pFmtCtx);
116.         pFmtCtx = NULL;
117.     }   
118.     if (oFmtCtx->pb) {
119.         avio_close(oFmtCtx->pb);
120.     }
121.     if (oFmtCtx) {
122.         avformat_free_context(oFmtCtx);
123.         oFmtCtx = NULL;
124.     }
125.     if (stream_map) {
126.         av_free(stream_map);
127.     }
128.     return 0;
129. }

复制代码

剪辑

1. #include <libavformat/avformat.h>
2. #include <libavutil/log.h>
3. #include <libavutil/avutil.h>
4. #include <stdlib.h>
6. int main (int argc, char *argv[]) {
7.     // 1. 处理参数
8.     char *src;
9.     char *dst;
11.     int *stream_map = NULL;
12.     int stream_idx = 0;
14.     int ret;
15.     int idx = -1;
16.     int i = 0;
17.     double startTime = 0;
18.     double endTime = 0;
19.     int64_t *dts_start_time = NULL;
20.     int64_t *pts_start_time = NULL;
22.     AVFormatContext *pFmtCtx = NULL;
23.     AVFormatContext *oFmtCtx = NULL;
25.     const AVOutputFormat *outFmt = NULL;
26.    
27.     AVPacket pkt;
29.     av_log_set_level(AV_LOG_DEBUG);
30.     // cut src dst start end
31.     if (argc < 5) {
32.         av_log(NULL, AV_LOG_ERROR, "argument must be more than 5\n");
33.         exit(-1);
34.     }
36.     src = argv[1];
37.     dst = argv[2];
38.     startTime = atof(argv[3]);
39.     endTime = atof(argv[4]);
41.     av_log(NULL, AV_LOG_INFO, "hello\n");
43.     // 2. 打开多媒体文件
44.     ret = avformat_open_input(&pFmtCtx, src, NULL, NULL);
45.     if (ret < 0) {
46.         av_log(NULL, AV_LOG_ERROR, "%s\n", av_err2str(ret));
47.         exit(-1);
48.     }
49.    
50.     // 4. 打开目的文件的上下文
51.     avformat_alloc_output_context2(&oFmtCtx, NULL, NULL, dst);
52.     if (!oFmtCtx) {
53.         av_log(NULL, AV_LOG_ERROR, "NO MEMORY\n");
54.         goto _ERROR;
55.     }
57.     stream_map = av_calloc(pFmtCtx->nb_streams, sizeof(int));
58.     if (!stream_map) {
59.         av_log(NULL, AV_LOG_ERROR, "NO MEMORY\n");
60.         goto _ERROR;
61.     }
63.     for (i = 0; i < pFmtCtx->nb_streams; ++i) {
64.         AVStream *outStream = NULL;
65.         AVStream *inStream = pFmtCtx->streams[i];
66.         AVCodecParameters *inCodecPar = inStream->codecpar;
67.         if (inCodecPar->codec_type != AVMEDIA_TYPE_AUDIO &&
68.             inCodecPar->codec_type != AVMEDIA_TYPE_VIDEO &&
69.             inCodecPar->codec_type != AVMEDIA_TYPE_SUBTITLE) {
70.             stream_map[i] = -1;
71.             continue;
72.         }
73.         stream_map[i] = stream_idx++;
75.         // 5. 为目的文件创建新的音频流
76.         outStream = avformat_new_stream(oFmtCtx, NULL);
77.         if (!outStream) {
78.             av_log(oFmtCtx, AV_LOG_ERROR, "NO MEMORY\n");
79.             goto _ERROR;
80.         }
82.         avcodec_parameters_copy(outStream->codecpar, inStream->codecpar);
83.         // 设置为0，会自动去找对应的编码器
84.         outStream->codecpar->codec_tag = 0;
86.     }
87.    
88.     ret = avio_open2(&oFmtCtx->pb, dst, AVIO_FLAG_WRITE, NULL, NULL);
89.     if (ret < 0) {
90.         av_log(oFmtCtx, AV_LOG_ERROR, "%s\n", av_err2str(ret));
91.         goto _ERROR;
92.     }
93.    
94.     // 7. 写多媒体文件头到目的文件
95.     ret = avformat_write_header(oFmtCtx, NULL);
96.     if (ret < 0) {
97.         av_log(oFmtCtx, AV_LOG_ERROR, "%s\n", av_err2str(ret));
98.         goto _ERROR;
99.     }
101.     // seek
102.     ret = av_seek_frame(pFmtCtx, -1, startTime * AV_TIME_BASE, AVSEEK_FLAG_BACKWARD);
103.     if (ret < 0) {
104.         av_log(oFmtCtx, AV_LOG_ERROR, "%s\n", av_err2str(ret));
105.         goto _ERROR;
106.     }
108.     dts_start_time = av_calloc(pFmtCtx->nb_streams, sizeof(int64_t));
109.     pts_start_time = av_calloc(pFmtCtx->nb_streams, sizeof(int64_t));
110.     for (int t = 0; t < pFmtCtx->nb_streams; ++t) {
111.         dts_start_time[t] = -1;
112.         pts_start_time[t] = -1;
113.     }
115.     // 8. 从源多媒体文件中读音频数据到目的文件中
116.     while (av_read_frame(pFmtCtx, &pkt) >= 0) {
117.         AVStream *outStream = NULL;
118.         AVStream *inStream = NULL;
119.         
120.         if (dts_start_time[pkt.stream_index] == -1 && pkt.dts > 0) {
121.             dts_start_time[pkt.stream_index] = pkt.dts;
122.         }
124.         if (pts_start_time[pkt.stream_index] == -1 && pkt.pts > 0) {
125.             pts_start_time[pkt.stream_index] = pkt.pts;
126.         }
128.         if (stream_map[pkt.stream_index] < 0) {
129.             av_packet_unref(&pkt);
130.             continue;
131.         }
133.         pkt.pts = pkt.pts - pts_start_time[pkt.stream_index];
134.         pkt.dts = pkt.dts - dts_start_time[pkt.stream_index];
136.         if (pkt.dts > pkt.pts) {
137.             pkt.pts = pkt.dts;
138.         }
140.         inStream = pFmtCtx->streams[pkt.stream_index];
141.         if (av_q2d(inStream->time_base) * pkt.pts > endTime) {
142.             av_log(oFmtCtx, AV_LOG_INFO, "success\n");
143.             break;
144.         }
145.         pkt.stream_index = stream_map[pkt.stream_index];
147.         outStream = oFmtCtx->streams[pkt.stream_index];
148.         av_packet_rescale_ts(&pkt, inStream->time_base, outStream->time_base);
150.         // 设置成-1，自己会去计算
151.         pkt.pos = -1;
152.         av_interleaved_write_frame(oFmtCtx, &pkt);
153.         av_packet_unref(&pkt);
154.         
155.     }
156.     // 9. 写多媒体文件尾到目的文件
157.     av_write_trailer(oFmtCtx);
158.     // 10. 将申请的资源释放掉
159. _ERROR:
160.     if (pFmtCtx) {
161.         avformat_close_input(&pFmtCtx);
162.         pFmtCtx = NULL;
163.     }   
164.     if (oFmtCtx->pb) {
165.         avio_close(oFmtCtx->pb);
166.     }
167.     if (oFmtCtx) {
168.         avformat_free_context(oFmtCtx);
169.         oFmtCtx = NULL;
170.     }
171.     if (stream_map) {
172.         av_free(stream_map);
173.     }
175.     if (pts_start_time) {
176.         av_free(pts_start_time);
177.     }
179.     if (dts_start_time) {
180.         av_free(dts_start_time);
181.     }
183.     return 0;
184. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。