把自定义的decoder加入ffmpeg源码

第一步：

在libavcodec目录下新建mkdecoder.c，并加入一下代码：

[cpp] view plain copy

1. /*
2. *实现一个自己的decoder,编码工作其实就是把pkt的数据拷贝到frame
3. *作者：缪国凯（MK）
4. *[email protected]
5. *2015-6-5
6. */
7. #include "avcodec.h"
8. static av_cold int mk_init_decoder(AVCodecContext *avctx)
9. {
10. printf("init mk decoder\n");
11. return 0;
12. }
13. static int mk_decode(AVCodecContext *avctx, void *data, int *got_frame,
14. AVPacket *avpkt)
15. {
16. AVFrame   *frame   = (AVFrame*)data;
17. AVPicture *picture = (AVPicture*)data;
18. const uint8_t *buf             = avpkt->data;
19. int buf_size                   = avpkt->size;
20. int size = avpicture_get_size(avctx->pix_fmt, avctx->width,
21. avctx->height);
22. frame->pict_type        = AV_PICTURE_TYPE_I;
23. frame->key_frame        = 1;
24. frame->buf[0] = av_buffer_alloc(size);
25. memcpy(frame->buf[0]->data, buf, buf_size);
26. int res = 0;
27. if ((res = avpicture_fill(picture, frame->buf[0]->data, avctx->pix_fmt,
28. avctx->width, avctx->height)) < 0)
29. {
30. av_buffer_unref(&frame->buf[0]);
31. return res;
32. }
33. *got_frame = 1;
34. return 0;
35. }
36. static av_cold int mk_close_decoder(AVCodecContext *avctx)
37. {
38. printf("init mk decoder\n");
39. return 0;
40. }
41. AVCodec ff_mkvideo_decoder = {
42. .name           = "mkvideo",
43. .long_name      = "mk video",
44. .type           = AVMEDIA_TYPE_VIDEO,
45. .id             = AV_CODEC_ID_MKVIDEO,
46. .capabilities   = CODEC_CAP_PARAM_CHANGE,
47. .init           = mk_init_decoder,
48. .decode         = mk_decode,
49. .close          = mk_close_decoder,
50. };

第二步：

在avcodec.h里的 enum AVCodecID 最后加入：

[cpp] view plain copy

1. AV_CODEC_ID_MKVIDEO,

第三步：

在allcodec.c的void avcodec_register_all(void)函数中加入

[cpp] view plain copy

1. REGISTER_DECODER(MKVIDEO,          mkvideo);

第四步：

在libavcodec的makefile里加入：

[cpp] view plain copy

1. OBJS-$(CONFIG_MKVIDEO_DECODER)              += mkdecoder.o

第五步：

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

[cpp] view plain copy

1. #define CONFIG_MKVIDEO_DECODER 1

第六步：

在codec_desc.c的static const AVCodecDescriptor codec_descriptors[] 中加入：

[cpp] view plain copy

1. {
2. .id        = AV_CODEC_ID_MKVIDEO,
3. .type      = AVMEDIA_TYPE_VIDEO,
4. .name      = "mkvideo",
5. .long_name = NULL_IF_CONFIG_SMALL("mk video"),
6. .props     = AV_CODEC_PROP_INTRA_ONLY | AV_CODEC_PROP_LOSSLESS,
7. },

第七步：

把以前实现的mkdemuxer的codeid改为

[cpp] view plain copy

1. AV_CODEC_ID_MKVIDEO

第八步：

重新编译ffmpeg。

测试：

运行命令：ffmpeg -decoders：在显示中找到自己的decoder：

[cpp] view plain copy

1. VF...D mimic                Mimic
2. V....D mjpeg                MJPEG (Motion JPEG)
3. V....D mjpegb               Apple MJPEG-B
4. V..... mkvideo              mk video

运行命令：

[cpp] view plain copy

1. ffmpeg -s 1280x720 -i test.mk -y test.avi

生成test.avi，并在转码过程中打出在代码中添加的标记：

[cpp] view plain copy

1. init mk decoder

播放test.avi，能正确播放，ok，成功！