把自定义的demuxer加入ffmpeg源码

Posted 2020-10-02 DoubleLi

.简介：把上一篇文章中的demuxer加入ffmpeg源码中去，使可以用命令行方式调用自定义的demuxer

第一步：

在libavformat目录下新建mkdemuxer.c和mkdemuxer.h，代码如下：

mkdemuxer.c：

 

[cpp] view plain copy

 

1. /*  
2. *实现一个自己的demuxer并加入到demuxer链中去 
3. *作者：缪国凯（MK）  
4. *[email protected]  
5. *2015-6-3  
6. */   
7.   
8. #include "mkdemuxer.h"  
9.   
10. typedef struct MKVideoDemuxerContext {  
11.     const AVClass *pclass;     /**< Class for private options. */  
12.     int width, height;        /**< Integers describing video size, set by a private option. */  
13.     char *pixel_format;       /**< Set by a private option. */  
14.     AVRational framerate;     /**< AVRational describing framerate, set by a private option. */  
15. } MKVideoDemuxerContext;  
16.   
17. int mkvideo_read_header(AVFormatContext *ctx)  
18. {  
19.     MKVideoDemuxerContext *s = ctx->priv_data;  
20.     enum AVPixelFormat pix_fmt;  
21.     AVStream *st;  
22.   
23.     st = avformat_new_stream(ctx, NULL);  
24.     if (!st)  
25.         return AVERROR(ENOMEM);  
26.   
27.     st->codec->codec_type = AVMEDIA_TYPE_VIDEO;  
28.   
29.     st->codec->codec_id = ctx->iformat->raw_codec_id;  
30.   
31.     //这里就简单的直接赋值为420p  
32.     pix_fmt = AV_PIX_FMT_YUV420P;  
33.   
34.     st->time_base.num = s->framerate.den;  
35.     st->time_base.den = s->framerate.num;  
36.     st->pts_wrap_bits = 64;  
37.       
38.   
39.     st->codec->width  = s->width;  
40.     st->codec->height = s->height;  
41.     st->codec->pix_fmt = pix_fmt;  
42.   
43.     AVRational tmpRa;  
44.     tmpRa.den = 1;  
45.     tmpRa.num = 8;  
46.     st->codec->bit_rate = av_rescale_q(avpicture_get_size(st->codec->pix_fmt, s->width, s->height),  
47.         tmpRa, st->time_base);  
48.   
49.     return 0;  
50. }  
51.   
52. int mkvideo_read_packet(AVFormatContext *s, AVPacket *pkt)  
53. {  
54.     int packet_size, ret, width, height;  
55.     AVStream *st = s->streams[0];  
56.   
57.     width = st->codec->width;  
58.     height = st->codec->height;  
59.   
60.     packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);  
61.     if (packet_size < 0)  
62.         return -1;  
63.   
64.     ret = av_get_packet(s->pb, pkt, packet_size);  
65.     pkt->pts = pkt->dts = pkt->pos / packet_size;  
66.   
67.     pkt->stream_index = 0;  
68.     if (ret < 0)  
69.         return ret;  
70.       
71.     return 0;  
72. }  
73.   
74. #define OFFSET(x) offsetof(MKVideoDemuxerContext, x)  
75. #define DEC AV_OPT_FLAG_DECODING_PARAM  
76.   
77. static const AVOption mk_options[] =   
78. {  
79.     { "video_size", "set frame size", OFFSET(width), AV_OPT_TYPE_IMAGE_SIZE, {.str = NULL}, 0, 0, DEC },  
80.     { "pixel_format", "set pixel format", OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = "yuv420p"}, 0, 0, DEC },  
81.     { "framerate", "set frame rate", OFFSET(framerate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, 0, DEC },  
82.     { NULL },  
83. };  
84.   
85. static const AVClass mk_demuxer_class = {  
86.     .class_name = "mk video demuxer",  
87.     .item_name  = av_default_item_name,  
88.     .option     = mk_options,  
89.     .version    = LIBAVUTIL_VERSION_INT,  
90. };  
91.   
92. AVInputFormat ff_mk_demuxer = {  
93.     .name           = "mk",  
94.     .long_name      = NULL_IF_CONFIG_SMALL("MK Video Container"),  
95.     .flags          = AVFMT_GENERIC_INDEX,  
96.     .extensions     = "mk",   
97.     .priv_class     = &mk_demuxer_class,      
98.     .raw_codec_id   = AV_CODEC_ID_RAWVIDEO,  
99.     .priv_data_size = sizeof(MKVideoDemuxerContext),  
100.       
101.     .read_header    = mkvideo_read_header,  
102.     .read_packet    = mkvideo_read_packet,  
103. };  

 

 

 

mkdemuxer.h：

 

[cpp] view plain copy

 

1. /*  
2. *实现一个自己的demuxer并加入到demuxer链中去 
3. *作者：缪国凯（MK）  
4. *[email protected]  
5. *2015-6-3  
6. */   
7.   
8. #ifndef AVFORMAT_MKDEMUXER_H  
9. #define AVFORMAT_MKDEMUXER_H  
10.   
11. #include "libavutil/opt.h"  
12. #include "avformat.h"  
13.   
14. int mkvideo_read_header(AVFormatContext *ctx);  
15. int mkvideo_read_packet(AVFormatContext *s, AVPacket *pkt);  
16.   
17. #endif //AVFORMAT_MKDEMUXER_H  

 

 

 

 

第二步：

在libavformat下的allformats.c的av_register_all里加入：

 

[cpp] view plain copy

 

1. REGISTER_DEMUXER   (MK,              mk);//add ByMK for mk demuxer    

 

第三步：

在libavformat目录下的makefile文件的# muxers/demuxers后加入：

 

[cpp] view plain copy

 

1. OBJS-$(CONFIG_MK_DEMUXER)                  += mkdemuxer.o   

 

第四步：

在ffmpeg根目录的config.h里加入：

 

[cpp] view plain copy

 

1. #define CONFIG_MK_DEMUXER 1    

 

 

 

第五步：

如果不想重新configure，则在ffmpeg根目录的config.mak文件中加入：

[cpp] view plain copy

 

1. CONFIG_MK_DEMUXER=yes  

 

OK，按照正常的编译去编译ffmpeg，然后运行ffmpeg.exe输入ffmpeg -formats 看到有mk这个muxer了，如下：

 

[cpp] view plain copy

 

1. DE mk              mk (MK Video Container)  
2.  E mkvtimestamp_v2 extract pts as timecode v2 format, as defined by mkvtoolnix  
3. DE mlp             raw MLP  
4. D  mlv             Magic Lantern Video (MLV)  
5. D  mm              American Laser Games MM  
6. DE mmf             Yamaha SMAF  
7.  E mov             QuickTime / MOV  

 

 

再试一下命令：ffmpeg -s 1280x720 -i test.mk -y test.avi。

注意：当.mk格式作为输入的时候，必须加输入参数-s 分辨率，就和yuv作为输入一样（因为从本质上讲就是yuv...）。

 

成功！OK，大功告成。

 

原理在上一篇的博文中已经讲到了，下一步，实现自己的encoder。

from:http://blog.csdn.net/dancing_night/article/details/46348515