把自定义的demuxer加入ffmpeg源码

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

第一步：

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

mkdemuxer.c：

/*
*实现一个自己的demuxer并加入到demuxer链中去
*作者：缪国凯（MK）
*821486004@qq.com
*2015-6-3
*/

#include "mkdemuxer.h"

typedef struct MKVideoDemuxerContext {
    const AVClass *pclass;     /**< Class for private options. */
    int width, height;        /**< Integers describing video size, set by a private option. */
    char *pixel_format;       /**< Set by a private option. */
    AVRational framerate;     /**< AVRational describing framerate, set by a private option. */
} MKVideoDemuxerContext;

int mkvideo_read_header(AVFormatContext *ctx)
{
    MKVideoDemuxerContext *s = ctx->priv_data;
    enum AVPixelFormat pix_fmt;
    AVStream *st;

    st = avformat_new_stream(ctx, NULL);
    if (!st)
        return AVERROR(ENOMEM);

    st->codec->codec_type = AVMEDIA_TYPE_VIDEO;

    st->codec->codec_id = ctx->iformat->raw_codec_id;

    //这里就简单的直接赋值为420p
    pix_fmt = AV_PIX_FMT_YUV420P;

    st->time_base.num = s->framerate.den;
    st->time_base.den = s->framerate.num;
    st->pts_wrap_bits = 64;


    st->codec->width  = s->width;
    st->codec->height = s->height;
    st->codec->pix_fmt = pix_fmt;

    AVRational tmpRa;
    tmpRa.den = 1;
    tmpRa.num = 8;
    st->codec->bit_rate = av_rescale_q(avpicture_get_size(st->codec->pix_fmt, s->width, s->height),
        tmpRa, st->time_base);

    return 0;
}

int mkvideo_read_packet(AVFormatContext *s, AVPacket *pkt)
{
    int packet_size, ret, width, height;
    AVStream *st = s->streams[0];

    width = st->codec->width;
    height = st->codec->height;

    packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
    if (packet_size < 0)
        return -1;

    ret = av_get_packet(s->pb, pkt, packet_size);
    pkt->pts = pkt->dts = pkt->pos / packet_size;

    pkt->stream_index = 0;
    if (ret < 0)
        return ret;

    return 0;
}

#define OFFSET(x) offsetof(MKVideoDemuxerContext, x)
#define DEC AV_OPT_FLAG_DECODING_PARAM

static const AVOption mk_options[] =
{
    { "video_size", "set frame size", OFFSET(width), AV_OPT_TYPE_IMAGE_SIZE, {.str = NULL}, 0, 0, DEC },
    { "pixel_format", "set pixel format", OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = "yuv420p"}, 0, 0, DEC },
    { "framerate", "set frame rate", OFFSET(framerate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, 0, DEC },
    { NULL },
};

static const AVClass mk_demuxer_class = {
    .class_name = "mk video demuxer",
    .item_name  = av_default_item_name,
    .option     = mk_options,
    .version    = LIBAVUTIL_VERSION_INT,
};

AVInputFormat ff_mk_demuxer = {
    .name           = "mk",
    .long_name      = NULL_IF_CONFIG_SMALL("MK Video Container"),
    .flags          = AVFMT_GENERIC_INDEX,
    .extensions     = "mk",
    .priv_class     = &mk_demuxer_class,
    .raw_codec_id   = AV_CODEC_ID_RAWVIDEO,
    .priv_data_size = sizeof(MKVideoDemuxerContext),

    .read_header    = mkvideo_read_header,
    .read_packet    = mkvideo_read_packet,
};

mkdemuxer.h：

/*
*实现一个自己的demuxer并加入到demuxer链中去
*作者：缪国凯（MK）
*821486004@qq.com
*2015-6-3
*/

#ifndef AVFORMAT_MKDEMUXER_H
#define AVFORMAT_MKDEMUXER_H

#include "libavutil/opt.h"
#include "avformat.h"

int mkvideo_read_header(AVFormatContext *ctx);
int mkvideo_read_packet(AVFormatContext *s, AVPacket *pkt);

#endif //AVFORMAT_MKDEMUXER_H

第二步：

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

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

第三步：

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

OBJS-$(CONFIG_MK_DEMUXER)                  += mkdemuxer.o

第四步：

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

#define CONFIG_MK_DEMUXER 1

第五步：

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

CONFIG_MK_DEMUXER=yes

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

DE mk              mk (MK Video Container)
 E mkvtimestamp_v2 extract pts as timecode v2 format, as defined by mkvtoolnix
DE mlp             raw MLP
D  mlv             Magic Lantern Video (MLV)
D  mm              American Laser Games MM
DE mmf             Yamaha SMAF
 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