ffmpeg用代码实现自己的muxer

曾经终败给现在 2022-07-17 15:26 205阅读 0赞

1、概述

本代码实现了一个muxer并加入到ffmpeg的AVOutputFormat 链表中去，使代码能直接调用。实现的意义一是了解ffmpeg代码结构，二是可以自己整自己的视频格式，让别人播放不了。

2、代码

简单看下代码：

**\[cpp\]**  [view plain][]   [copy][view plain]

1.  /\* 
2.  \*本程序主要实现一个自己的muxer并加入到muxer链中去，使其可用,只有视频 
3.  \*作者：缪国凯（MK） 
4.  \*821486004@qq.com 
5.  \*2015-6-2 
6.  \*/  
7.    
8.  \#include "stdafx.h"  
9.    
10. \#ifdef \_\_cplusplus  
11. **extern** "C"  
12. \{  
13. \#endif  
14. \#include <libavformat/avformat.h>  
15. \#include <libavcodec/avcodec.h>  
16. \#  
17. \#ifdef \_\_cplusplus  
18. \};  
19. \#endif  
20.   
21. \#pragma comment(lib, "avcodec.lib")  
22. \#pragma comment(lib, "avformat.lib")  
23. \#pragma comment(lib, "avutil.lib")  
24. //\#pragma comment(lib, "avdevice.lib")  
25. //\#pragma comment(lib, "avfilter.lib")  
26. //\#pragma comment(lib, "postproc.lib")  
27. //\#pragma comment(lib, "swresample.lib")  
28. //\#pragma comment(lib, "swscale.lib")  
29.   
30. **int** mk\_write\_header(**struct** AVFormatContext \*fmt)  
31. \{  
32.     //这些地方就可以加入自己的格式定义  
33.     **return** 0;  
34. \}  
35.   
36. **int** mk\_write\_packet(**struct** AVFormatContext \*fmt, AVPacket \*pkt)  
37. \{  
38.     avio\_write(fmt->pb, pkt->data, pkt->size);//简单的用file协议写文件  
39.     **return** 0;  
40. \}  
41.   
42. **int** mk\_write\_trailer(**struct** AVFormatContext \*fmt)  
43. \{  
44.     //这些地方就可以加入自己的格式定义  
45.     **return** 0;  
46. \}  
47.   
48. AVOutputFormat ff\_mk\_muxer =   
49. \{  
50.     /\*.name         = \*/"mk",  
51.     /\*.long\_name        = \*/"mk (MK Video Container)",  
52.     /\*.mime\_type        = \*/"mkvideo/x-msvideo",  
53.     /\*.extensions       = \*/"mk",  
54.   
55.     /\*.audio\_codec      = \*/AV\_CODEC\_ID\_NONE,  
56.     /\*.video\_codec      = \*/AV\_CODEC\_ID\_RAWVIDEO,//这里先用ffmpeg自带的yuv编码器，以后改成自己的  
57.     /\*.subtitle\_codec   = \*/AV\_CODEC\_ID\_NONE,  
58.   
59.     /\*.flags        = \*/AVFMT\_NOTIMESTAMPS,  
60.     /\*.codec\_tag        = \*/NULL,  
61.     /\*.priv\_class       = \*/NULL,  
62.     /\*.next             = \*/NULL,  
63.     /\*.priv\_data\_size   = \*/0,    
64.   
65.     /\*.write\_header     = \*/mk\_write\_header,  
66.     /\*.write\_packet     = \*/mk\_write\_packet,  
67.     /\*.write\_trailer    = \*/mk\_write\_trailer,  
68. \};  
69.   
70. **void** help()  
71. \{  
72.     printf("\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\\n");  
73.     printf("Usage:\\n");  
74.     printf("    MyMuxer \[inputfile\] \[outputfile.mk\]\\n");  
75.     printf("\\n");  
76.     printf("Examples: \\n");  
77.     printf("    MyMuxer a.avi a.mk\\n");  
78.     printf("\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\\n");    
79. \}  
80.   
81. **int** \_tmain(**int** argc, \_TCHAR\* argv\[\])  
82. \{  
83.     **if**(argc < 3 || (!strcmp(argv\[1\],"--help")))  
84.     \{  
85.         help();  
86.         **return** 0;  
87.     \}  
88.   
89.     AVFormatContext \*in\_fxt = NULL, \*out\_fxt = NULL;  
90.     AVStream \*out\_stream = NULL;  
91.     **int** video\_index = -1;  
92.   
93.     av\_register\_all();  
94.     av\_register\_output\_format(&ff\_mk\_muxer);//把自己的muxer加入链表  
95.   
96.     **if** (avformat\_open\_input(&in\_fxt, argv\[1\], NULL, NULL) < 0)  
97.     \{  
98.         printf("can not open the input file context!\\n");  
99.         **goto** end;  
100.     \}  
101.     **if** (avformat\_find\_stream\_info(in\_fxt, NULL) < 0)  
102.     \{  
103.         printf("can not find the stream info!\\n");  
104.         **goto** end;  
105.     \}  
106.   
107.     **if**(avformat\_alloc\_output\_context2(&out\_fxt, NULL, NULL, argv\[2\]) < 0)  
108.     \{  
109.         printf("can not alloc output context!\\n");  
110.         **goto** end;  
111.     \}  
112.   
113.     **for** (**int** i = 0; i < in\_fxt->nb\_streams; i++)  
114.     \{  
115.         **if** (in\_fxt->streams\[i\]->codec->codec\_type == AVMEDIA\_TYPE\_VIDEO)  
116.         \{  
117.             //open decoder  
118.             **if**(0 > avcodec\_open2(in\_fxt->streams\[i\]->codec, avcodec\_find\_decoder(in\_fxt->streams\[i\]->codec->codec\_id), NULL))  
119.             \{  
120.                 printf("can not find or open decoder!\\n");  
121.                 **goto** end;  
122.             \}  
123.             video\_index = i;  
124.             //new stream  
125.             out\_stream = avformat\_new\_stream(out\_fxt, NULL);  
126.             **if** (!out\_stream)  
127.             \{  
128.                 printf("can not new stream for output!\\n");  
129.                 **goto** end;  
130.             \}  
131.             //set codec context param  
132.             out\_stream->codec->codec = avcodec\_find\_encoder(out\_fxt->oformat->video\_codec);  
133.             out\_stream->codec->height = in\_fxt->streams\[i\]->codec->height;  
134.             out\_stream->codec->width = in\_fxt->streams\[i\]->codec->width;  
135.   
136.             out\_stream->codec->time\_base.num = in\_fxt->streams\[i\]->avg\_frame\_rate.den;  
137.             out\_stream->codec->time\_base.den = in\_fxt->streams\[i\]->avg\_frame\_rate.num;  
138.   
139.             out\_stream->codec->sample\_aspect\_ratio = in\_fxt->streams\[i\]->codec->sample\_aspect\_ratio;           
140.             out\_stream->codec->pix\_fmt = in\_fxt->streams\[i\]->codec->pix\_fmt;  
141.             **if** (!out\_stream->codec->codec)  
142.             \{  
143.                 printf("can not find the encoder!\\n");  
144.                 **goto** end;  
145.             \}  
146.             **if** ((avcodec\_open2(out\_stream->codec, out\_stream->codec->codec, NULL)) < 0)  
147.             \{  
148.                 printf("can not open the encoder\\n");  
149.                 **goto** end;  
150.             \}  
151.             **if** (out\_fxt->oformat->flags & AVFMT\_GLOBALHEADER)  
152.                 out\_stream->codec->flags |= CODEC\_FLAG\_GLOBAL\_HEADER;  
153.             **break**;  
154.         \}  
155.     \}  
156.   
157.     **if** (-1 == video\_index)  
158.     \{  
159.         printf("found no video stream in input file!\\n");  
160.         **goto** end;  
161.     \}  
162.   
163.     **if** (!(out\_fxt->oformat->flags & AVFMT\_NOFILE))  
164.     \{  
165.         **if**(avio\_open(&out\_fxt->pb, argv\[2\], AVIO\_FLAG\_WRITE) < 0)  
166.         \{  
167.             printf("can not open output file handle!\\n");  
168.             **goto** end;  
169.         \}  
170.     \}  
171.   
172.     **if**(avformat\_write\_header(out\_fxt, NULL) < 0)  
173.     \{  
174.         printf("can not write the header of the output file!\\n");  
175.         **goto** end;  
176.     \}  
177.   
178.     AVPacket pkt\_in, pkt\_out;  
179.     AVFrame \*frame;  
180.     frame = av\_frame\_alloc();  
181.     av\_init\_packet(&pkt\_in);  
182.     av\_init\_packet(&pkt\_out);  
183.     **int** got\_frame, got\_picture;  
184.     **int** i = 0, frame\_index = 0;  
185.     **while**(1)  
186.     \{  
187.         got\_frame = -1;  
188.         got\_picture = -1;  
189.         **if** (av\_read\_frame(in\_fxt, &pkt\_in) < 0)  
190.         \{  
191.             **break**;  
192.         \}  
193.         **if** (avcodec\_decode\_video2(in\_fxt->streams\[video\_index\]->codec, frame, &got\_frame, &pkt\_in) < 0)  
194.         \{  
195.             printf("can not decoder a frame");  
196.             **break**;  
197.         \}  
198.         av\_free\_packet(&pkt\_in);  
199.   
200.         **if** (got\_frame)  
201.         \{  
202.             frame->pts = i++;  
203.             pkt\_out.data = NULL;//主要这里必须自己初始化，或者必须置为null，不然ff\_alloc\_packet2函数会报错  
204.             pkt\_out.size = 0;  
205.             **if** (avcodec\_encode\_video2(out\_stream->codec, &pkt\_out, frame, &got\_picture) < 0)  
206.             \{  
207.                 printf("can not encode a frame!\\n");  
208.                 **break**;  
209.             \}  
210.   
211.             **if** (got\_picture)  
212.             \{  
213.                 printf("Succeed to encode frame: %5d\\tsize:%5d\\n",frame\_index,pkt\_out.size);  
214.                 pkt\_out.stream\_index = out\_stream->index;  
215.                 frame\_index++;  
216.                 av\_write\_frame(out\_fxt, &pkt\_out);  
217.                 av\_free\_packet(&pkt\_out);  
218.             \}  
219.         \}  
220.     \}  
221.     av\_frame\_free(&frame);  
222.   
223.     av\_write\_trailer(out\_fxt);  
224.   
225.     //clean  
226.     avcodec\_close(out\_stream->codec);  
227.     avcodec\_close(out\_fxt->streams\[video\_index\]->codec);  
228. end:  
229.     avformat\_close\_input(&in\_fxt);  
230.   
231.     **if** (out\_fxt && !(out\_fxt->oformat->flags & AVFMT\_NOFILE))  
232.     \{  
233.         avio\_close(out\_fxt->pb);  
234.     \}  
235.     avformat\_free\_context(out\_fxt);  
236.   
237.     **return** 0;  
238. \}

3、解释

简单说下基本原理。

首先自己实现一个AVOutputFormat，然后把这个muxer加入到链表中去，其他的根本不用自己管了，ffmpeg这套框架做的的确很漂亮，会一层层的掉用到你的muxer来，函数avformat\_alloc\_output\_context2会根据输入的格式或后缀来在AVOutputFormat链中查找最适合的muxer，并把这个muxer的指针用oformat保存，下面看看格式匹配的代码：

**\[cpp\]**  [view plain][]   [copy][view plain]

1.  AVOutputFormat \*av\_guess\_format(**const** **char** \*short\_name, **const** **char** \*filename,  
2.                                  **const** **char** \*mime\_type)  
3.  \{  
4.      AVOutputFormat \*fmt = NULL, \*fmt\_found;  
5.      **int** score\_max, score;  
6.    
7.      /\* specific test for image sequences \*/  
8.  \#if CONFIG\_IMAGE2\_MUXER  
9.      **if** (!short\_name && filename &&  
10.         av\_filename\_number\_test(filename) &&  
11.         ff\_guess\_image2\_codec(filename) != AV\_CODEC\_ID\_NONE) \{  
12.         **return** av\_guess\_format("image2", NULL, NULL);  
13.     \}  
14. \#endif  
15.     /\* Find the proper file type. \*/  
16.     fmt\_found = NULL;  
17.     score\_max = 0;  
18.     **while** ((fmt = av\_oformat\_next(fmt))) \{  
19.         score = 0;  
20.         **if** (fmt->name && short\_name && av\_match\_name(short\_name, fmt->name))  
21.             score += 100;  
22.         **if** (fmt->mime\_type && mime\_type && !strcmp(fmt->mime\_type, mime\_type))  
23.             score += 10;  
24.         **if** (filename && fmt->extensions &&  
25.             av\_match\_ext(filename, fmt->extensions)) \{  
26.             score += 5;  
27.         \}  
28.         **if** (score > score\_max) \{  
29.             score\_max = score;  
30.             fmt\_found = fmt;  
31.         \}  
32.     \}  
33.     **return** fmt\_found;  
34. \}

遍历整个链表，得分最高的就是最匹配的格式，name和传入的shotname（就是传入的格式）匹配得分100，mime\_type匹配得分10，extensions和传入的文件名后缀匹配得分5，是不是很有意思。

再来看看avformat\_write\_header的代码：

**\[cpp\]**  [view plain][]   [copy][view plain]

1.  **int** avformat\_write\_header(AVFormatContext \*s, AVDictionary \*\*options)  
2.  \{  
3.      **int** ret = 0;  
4.    
5.      **if** (ret = init\_muxer(s, options))  
6.          **return** ret;  
7.    
8.      **if** (s->oformat->write\_header) \{  
9.          ret = s->oformat->write\_header(s);  
10.         **if** (ret >= 0 && s->pb && s->pb->error < 0)  
11.             ret = s->pb->error;  
12.         **if** (ret < 0)  
13.             **return** ret;  
14.         **if** (s->flush\_packets && s->pb && s->pb->error >= 0 && s->flags & AVFMT\_FLAG\_FLUSH\_PACKETS)  
15.             avio\_flush(s->pb);  
16.     \}  
17.   
18.     **if** ((ret = init\_pts(s)) < 0)  
19.         **return** ret;  
20.   
21.     **if** (s->avoid\_negative\_ts < 0) \{  
22.         av\_assert2(s->avoid\_negative\_ts == AVFMT\_AVOID\_NEG\_TS\_AUTO);  
23.         **if** (s->oformat->flags & (AVFMT\_TS\_NEGATIVE | AVFMT\_NOTIMESTAMPS)) \{  
24.             s->avoid\_negative\_ts = 0;  
25.         \} **else**  
26.             s->avoid\_negative\_ts = AVFMT\_AVOID\_NEG\_TS\_MAKE\_NON\_NEGATIVE;  
27.     \}  
28.   
29.     **return** 0;  
30. \}

会调用oformat的 write\_header。  
  
 再看 av\_write\_frame 的代码：

**\[cpp\]**  [view plain][]   [copy][view plain]

1.  **int** av\_write\_frame(AVFormatContext \*s, AVPacket \*pkt)  
2.  \{  
3.      **int** ret;  
4.    
5.      ret = check\_packet(s, pkt);  
6.      **if** (ret < 0)  
7.          **return** ret;  
8.    
9.      **if** (!pkt) \{  
10.         **if** (s->oformat->flags & AVFMT\_ALLOW\_FLUSH) \{  
11.             ret = s->oformat->write\_packet(s, NULL);  
12.             **if** (s->flush\_packets && s->pb && s->pb->error >= 0 && s->flags & AVFMT\_FLAG\_FLUSH\_PACKETS)  
13.                 avio\_flush(s->pb);  
14.             **if** (ret >= 0 && s->pb && s->pb->error < 0)  
15.                 ret = s->pb->error;  
16.             **return** ret;  
17.         \}  
18.         **return** 1;  
19.     \}  
20.   
21.     ret = compute\_pkt\_fields2(s, s->streams\[pkt->stream\_index\], pkt);  
22.   
23.     **if** (ret < 0 && !(s->oformat->flags & AVFMT\_NOTIMESTAMPS))  
24.         **return** ret;  
25.   
26.     ret = write\_packet(s, pkt);  
27.     **if** (ret >= 0 && s->pb && s->pb->error < 0)  
28.         ret = s->pb->error;  
29.   
30.     **if** (ret >= 0)  
31.         s->streams\[pkt->stream\_index\]->nb\_frames++;  
32.     **return** ret;  
33. \}

会调用oformat的 write\_packet。

最后看看av\_write\_trailer的代码：

**\[cpp\]**  [view plain][]   [copy][view plain]

1.  **int** av\_write\_trailer(AVFormatContext \*s)  
2.  \{  
3.      **int** ret, i;  
4.    
5.      **for** (;; ) \{  
6.          AVPacket pkt;  
7.          ret = interleave\_packet(s, &pkt, NULL, 1);  
8.          **if** (ret < 0)  
9.              **goto** fail;  
10.         **if** (!ret)  
11.             **break**;  
12.   
13.         ret = write\_packet(s, &pkt);  
14.         **if** (ret >= 0)  
15.             s->streams\[pkt.stream\_index\]->nb\_frames++;  
16.   
17.         av\_free\_packet(&pkt);  
18.   
19.         **if** (ret < 0)  
20.             **goto** fail;  
21.         **if**(s->pb && s->pb->error)  
22.             **goto** fail;  
23.     \}  
24.   
25. fail:  
26.     **if** (s->oformat->write\_trailer)  
27.         **if** (ret >= 0) \{  
28.         ret = s->oformat->write\_trailer(s);  
29.         \} **else** \{  
30.             s->oformat->write\_trailer(s);  
31.         \}  
32.   
33.     **if** (s->pb)  
34.        avio\_flush(s->pb);  
35.     **if** (ret == 0)  
36.        ret = s->pb ? s->pb->error : 0;  
37.     **for** (i = 0; i < s->nb\_streams; i++) \{  
38.         av\_freep(&s->streams\[i\]->priv\_data);  
39.         av\_freep(&s->streams\[i\]->index\_entries);  
40.     \}  
41.     **if** (s->oformat->priv\_class)  
42.         av\_opt\_free(s->priv\_data);  
43.     av\_freep(&s->priv\_data);  
44.     **return** ret;  
45. \}

![save_snippets.png][]

会调用oformat的 write\_trailer。

所以说，只要你按照ffmpeg的格式去定义muxer并加入链表，其他的都不用管了。

最后验证一下你的结果，运行程序随便转一个结果出来，把后缀.mk改为.yuv，用yuv播放器放一下，ok了，大功告成。下一篇介绍如何把这个muxer加入ffmpeg源码，使ffmpeg命令行可以用这个muxer。

链接：http://blog.csdn.net/dancing\_night/article/details/46328837

[view plain]: http://blog.csdn.net/dancing_night/article/details/46328837#
[save_snippets.png]: /images/20220717/c9da87e564804094b4976b6610b7e143.png