把自定义的demuxer加入ffmpeg源码

.简介：把上一篇文章中的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. #include "mkdemuxer.h"
8. typedef struct MKVideoDemuxerContext {
9. const AVClass *pclass;     /**< Class for private options. */
10. int width, height;        /**< Integers describing video size, set by a private option. */
11. char *pixel_format;       /**< Set by a private option. */
12. AVRational framerate;     /**< AVRational describing framerate, set by a private option. */
13. } MKVideoDemuxerContext;
14. int mkvideo_read_header(AVFormatContext *ctx)
15. {
16. MKVideoDemuxerContext *s = ctx->priv_data;
17. enum AVPixelFormat pix_fmt;
18. AVStream *st;
19. st = avformat_new_stream(ctx, NULL);
20. if (!st)
21. return AVERROR(ENOMEM);
22. st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
23. st->codec->codec_id = ctx->iformat->raw_codec_id;
24. //这里就简单的直接赋值为420p
25. pix_fmt = AV_PIX_FMT_YUV420P;
26. st->time_base.num = s->framerate.den;
27. st->time_base.den = s->framerate.num;
28. st->pts_wrap_bits = 64;
29. st->codec->width  = s->width;
30. st->codec->height = s->height;
31. st->codec->pix_fmt = pix_fmt;
32. AVRational tmpRa;
33. tmpRa.den = 1;
34. tmpRa.num = 8;
35. st->codec->bit_rate = av_rescale_q(avpicture_get_size(st->codec->pix_fmt, s->width, s->height),
36. tmpRa, st->time_base);
37. return 0;
38. }
39. int mkvideo_read_packet(AVFormatContext *s, AVPacket *pkt)
40. {
41. int packet_size, ret, width, height;
42. AVStream *st = s->streams[0];
43. width = st->codec->width;
44. height = st->codec->height;
45. packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
46. if (packet_size < 0)
47. return -1;
48. ret = av_get_packet(s->pb, pkt, packet_size);
49. pkt->pts = pkt->dts = pkt->pos / packet_size;
50. pkt->stream_index = 0;
51. if (ret < 0)
52. return ret;
53. return 0;
54. }
55. #define OFFSET(x) offsetof(MKVideoDemuxerContext, x)
56. #define DEC AV_OPT_FLAG_DECODING_PARAM
57. static const AVOption mk_options[] =
58. {
59. { "video_size", "set frame size", OFFSET(width), AV_OPT_TYPE_IMAGE_SIZE, {.str = NULL}, 0, 0, DEC },
60. { "pixel_format", "set pixel format", OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = "yuv420p"}, 0, 0, DEC },
61. { "framerate", "set frame rate", OFFSET(framerate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, 0, DEC },
62. { NULL },
63. };
64. static const AVClass mk_demuxer_class = {
65. .class_name = "mk video demuxer",
66. .item_name  = av_default_item_name,
67. .option     = mk_options,
68. .version    = LIBAVUTIL_VERSION_INT,
69. };
70. AVInputFormat ff_mk_demuxer = {
71. .name           = "mk",
72. .long_name      = NULL_IF_CONFIG_SMALL("MK Video Container"),
73. .flags          = AVFMT_GENERIC_INDEX,
74. .extensions     = "mk",
75. .priv_class     = &mk_demuxer_class,
76. .raw_codec_id   = AV_CODEC_ID_RAWVIDEO,
77. .priv_data_size = sizeof(MKVideoDemuxerContext),
78. .read_header    = mkvideo_read_header,
79. .read_packet    = mkvideo_read_packet,
80. };

mkdemuxer.h：

[cpp] view plain copy

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