首页 > 代码库 > ffmpeg代码实现自定义encoder

ffmpeg代码实现自定义encoder

2024-11-22 14:29:02   202人阅读

1、概述

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

2、代码

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

3、解释

原理和前面的自定义的muxer、demuxer一样，在这里就不多说了。
from:http://blog.csdn.net/dancing_night/article/details/46360851

ffmpeg代码实现自定义encoder


联系
我们