ffmpeg代码实现自定义encoder

1、概述

本文主要讲述如何用ffmpeg代码实现自己的encoder。

2、代码

1. /*
2. *本程序主要实现一个自己的encoder并加入到encoder链中去,供api调用
3. *作者：缪国凯（MK）
4. *821486004@qq.com
5. *2015-6-4
6. */
7. #include "stdafx.h"
8. #ifdef __cplusplus
9. extern "C"
10. {
11. #endif
12. #include <libavformat/avformat.h>
13. #include <libavcodec/avcodec.h>
14. #include <libavutil/pixdesc.h>
15. #
16. #ifdef __cplusplus
17. };
18. #endif
19. #pragma comment(lib, "avcodec.lib")
20. #pragma comment(lib, "avformat.lib")
21. #pragma comment(lib, "avutil.lib")
22. //#pragma comment(lib, "avdevice.lib")
23. //#pragma comment(lib, "avfilter.lib")
24. //#pragma comment(lib, "postproc.lib")
25. //#pragma comment(lib, "swresample.lib")
26. //#pragma comment(lib, "swscale.lib")
27. static av_cold int mk_encode_init(AVCodecContext *avctx)
28. {
29. const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(avctx->pix_fmt);
30. avctx->coded_frame = av_frame_alloc();
31. avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
32. avctx->bits_per_coded_sample = av_get_bits_per_pixel(desc);
33. if(!avctx->codec_tag)
34. avctx->codec_tag = avcodec_pix_fmt_to_codec_tag(avctx->pix_fmt);
35. return 0;
36. }
37. static int mk_encode(AVCodecContext *avctx, AVPacket *pkt,
38. const AVFrame *frame, int *got_packet)
39. {
40. int ret = avpicture_get_size(avctx->pix_fmt, avctx->width, avctx->height);
41. if (ret < 0)
42. return ret;
43. if (pkt->data == NULL && pkt->size == 0)
44. {
45. av_new_packet(pkt,ret);
46. pkt->size = ret;
47. }
48. //  if ((ret = ff_alloc_packet2(avctx, pkt, ret)) < 0)
49. //      return ret;
50. if ((ret = avpicture_layout((const AVPicture *)frame, avctx->pix_fmt, avctx->width,
51. avctx->height, pkt->data, pkt->size)) < 0)
52. return ret;
53. //  if(avctx->codec_tag == AV_RL32("yuv2") && ret > 0 &&
54. //      avctx->pix_fmt   == AV_PIX_FMT_YUYV422)
55. //  {
56. //          int x;
57. //          for(x = 1; x < avctx->height*avctx->width*2; x += 2)
58. //              pkt->data[x] ^= 0x80;
59. //  }
60. pkt->flags |= AV_PKT_FLAG_KEY;
61. *got_packet = 1;
62. return 0;
63. }
64. static av_cold int mk_close(AVCodecContext *avctx)
65. {
66. av_frame_free(&avctx->coded_frame);
67. return 0;
68. }
69. AVCodec ff_mkvideo_encoder = {
70. /*.name           = */"mkvideo",
71. /*.long_name      = */"mk video",
72. /*.type           = */AVMEDIA_TYPE_VIDEO,
73. /*.id             = */AV_CODEC_ID_MKVIDEO,
74. /*.capabilities = */0,
75. /*.supported_framerates = */NULL,
76. /*.pix_fmts     = */NULL,
77. /*.supported_samplerates = */NULL,
78. /*.sample_fmts  = */NULL,
79. /*.channel_layouts = */NULL,
80. /*.max_lowres       = */0,
81. /*.priv_class       = */NULL,
82. /*.profiles     = */NULL,
83. /*.priv_data_size   = */0,
84. /*.next         = */NULL,
85. /*.init_thread_copy = */NULL,
86. /*.update_thread_context = */NULL,
87. /*.defaults     = */NULL,
88. /*.init_static_data = */NULL,
89. /*.init           = */mk_encode_init,
90. /*.encode_sub       = */NULL,
91. /*.encode2        = */mk_encode,
92. /*.decode           = */NULL,
93. /*.close          = */mk_close,
94. };
95. void help()
96. {
97. printf("**********************************************\n");
98. printf("Usage:\n");
99. printf("    MyMuxer [inputfile] [outputfile] \n");
100. printf("\n");
101. printf("Examples: \n");
102. printf("    MyMuxer a.avi a.yuv \n");
103. printf("**********************************************\n");
104. }
105. int _tmain(int argc, _TCHAR* argv[])
106. {
107. if(argc < 3 || (!strcmp(argv[1],"--help")))
108. {
109. help();
110. return 0;
111. }
112. av_register_all();
113. avcodec_register(&ff_mkvideo_encoder);
114. AVFormatContext *in_fxt = NULL, *out_fxt = NULL;
115. AVStream *out_stream = NULL;
116. int video_index = -1;
117. if (avformat_open_input(&in_fxt, argv[1], NULL, NULL) < 0)
118. {
119. printf("can not open the input file context!\n");
120. goto end;
121. }
122. if (avformat_find_stream_info(in_fxt, NULL) < 0)
123. {
124. printf("can not find the stream info!\n");
125. goto end;
126. }
127. if(avformat_alloc_output_context2(&out_fxt, NULL, NULL, argv[2]) < 0)
128. {
129. printf("can not alloc output context!\n");
130. goto end;
131. }
132. for (int i = 0; i < in_fxt->nb_streams; i++)
133. {
134. if (in_fxt->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)
135. {
136. //open decoder
137. if(0 > avcodec_open2(in_fxt->streams[i]->codec, avcodec_find_decoder(in_fxt->streams[i]->codec->codec_id), NULL))
138. {
139. printf("can not find or open decoder!\n");
140. goto end;
141. }
142. video_index = i;
143. //new stream
144. out_stream = avformat_new_stream(out_fxt, NULL);
145. if (!out_stream)
146. {
147. printf("can not new stream for output!\n");
148. goto end;
149. }
150. //set codec context param
151. out_stream->codec->codec = avcodec_find_encoder(/*out_fxt->oformat->video_codec*/AV_CODEC_ID_MKVIDEO);
152. out_stream->codec->height = in_fxt->streams[i]->codec->height;
153. out_stream->codec->width = in_fxt->streams[i]->codec->width;
154. out_stream->codec->time_base = in_fxt->streams[i]->time_base;
155. //out_stream->codec->time_base.den = 25;
156. out_stream->codec->sample_aspect_ratio = in_fxt->streams[i]->codec->sample_aspect_ratio;
157. out_stream->codec->pix_fmt = in_fxt->streams[i]->codec->pix_fmt;
158. out_stream->avg_frame_rate.den = out_stream->codec->time_base.num;
159. out_stream->avg_frame_rate.num = out_stream->codec->time_base.den;
160. if (!out_stream->codec->codec)
161. {
162. printf("can not find the encoder!\n");
163. goto end;
164. }
165. if ((avcodec_open2(out_stream->codec, out_stream->codec->codec, NULL)) < 0)
166. {
167. printf("can not open the encoder\n");
168. goto end;
169. }
170. if (out_fxt->oformat->flags & AVFMT_GLOBALHEADER)
171. out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
172. break;
173. }
174. }
175. if (-1 == video_index)
176. {
177. printf("found no video stream in input file!\n");
178. goto end;
179. }
180. if (!(out_fxt->oformat->flags & AVFMT_NOFILE))
181. {
182. if(avio_open(&out_fxt->pb, argv[2], AVIO_FLAG_WRITE) < 0)
183. {
184. printf("can not open output file handle!\n");
185. goto end;
186. }
187. }
188. if(avformat_write_header(out_fxt, NULL) < 0)
189. {
190. printf("can not write the header of the output file!\n");
191. goto end;
192. }
193. AVPacket pkt_in, pkt_out;
194. AVFrame *frame;
195. frame = av_frame_alloc();
196. av_init_packet(&pkt_in);
197. av_init_packet(&pkt_out);
198. int got_frame, got_picture;
199. int i = 0, frame_index = 0;
200. while(1)
201. {
202. got_frame = -1;
203. got_picture = -1;
204. if (av_read_frame(in_fxt, &pkt_in) < 0)
205. {
206. break;
207. }
208. if (avcodec_decode_video2(in_fxt->streams[video_index]->codec, frame, &got_frame, &pkt_in) < 0)
209. {
210. printf("can not decoder a frame");
211. break;
212. }
213. av_free_packet(&pkt_in);
214. if (got_frame)
215. {
216. frame->pts = i++;
217. pkt_out.data = NULL;//主要这里必须自己初始化，或者必须置为null，不然ff_alloc_packet2函数会报错
218. pkt_out.size = 0;
219. if (avcodec_encode_video2(out_stream->codec, &pkt_out, frame, &got_picture) < 0)
220. {
221. printf("can not encode a frame!\n");
222. break;
223. }
224. if (got_picture)
225. {
226. printf("Succeed to encode frame: %5d\tsize:%5d\n",frame_index,pkt_out.size);
227. pkt_out.stream_index = out_stream->index;
228. frame_index++;
229. av_write_frame(out_fxt, &pkt_out);
230. av_free_packet(&pkt_out);
231. }
232. }
233. }
234. av_frame_free(&frame);
235. av_write_trailer(out_fxt);
236. //clean
237. avcodec_close(out_stream->codec);
238. avcodec_close(out_fxt->streams[video_index]->codec);
239. end:
240. avformat_close_input(&in_fxt);
241. if (out_fxt && !(out_fxt->oformat->flags & AVFMT_NOFILE))
242. {
243. avio_close(out_fxt->pb);
244. }
245. avformat_free_context(out_fxt);
246. return 0;
247. }

3、解释

原理和前面的自定义的muxer、demuxer一样，在这里就不多说了。

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