ffmpeg 代码实现视频添加水印功能.pdf

《ffmpeg 代码实现视频添加水印功能.pdf》由会员分享，可在线阅读，更多相关《ffmpeg 代码实现视频添加水印功能.pdf（15页珍藏版）》请在淘文阁 - 分享文档赚钱的网站上搜索。

1、ffmpeg 代码实现视频添加水印功能备忘:用 ffmpeg 给视频添加水印的命令为:ffmpeg-i in.mp4-ixxx.png-filter_complex overlay=5:5out.mp4 代码实现如下,主要是用了参考了 ffmpeg.c 里面的 filter_complex 的代码:cpp view plaincopy#include<unistd.h>#include<stdio.h>#include<assert.h>#include<libavcodec/avcodec.h>#include<libavformat/a

2、vformat.h>#include<libavfilter/avfiltergraph.h>#include<libavfilter/avcodec.h>#include<libavfilter/buffersink.h>#include<libavfilter/buffersrc.h>#include<libswscale/swscale.h>const char*filter_descr=overlay=5:5;#define ENABLE_YUV_FILE 1AVFormatContext*input_fmt_ctx;A

3、VCodecContext*input_dec_ctx;AVFormatContext*overlay_fmt_ctx;AVCodecContext*overlay_dec_ctx;intinput_video_stream_idx,overlay_video_stream_idx;AVFilterGraph*filter_graph;AVFilterInOut*inputs;AVFilterInOut*outputs;AVFilterContext*buffersrc_ctx;AVFilterContext*bufferoverlay_ctx;AVFilterContext*buffersi

4、nk_ctx;int ret;int got_frame;intvideo_eof_reached=0;int overlay_eof_reached=0;int active_stream_index=-1;FILE*fp_yuv;voidyuv420p_save(AVFrame*pFrame);intvideo_transcode_step(AVFrame*mVideoFrame);intoverlay_transcode_step(AVFrame*mOverlayFrame);intvideo_output_eof_packet(const char*tag,AVStream*ist,A

5、VFilterContext*ifilter);static intopen_input_file(const char*filename)int ret;AVCodec*dec;if(ret=avformat_open_input(&input_fmt_ctx,filename,NULL,NULL)<0)av_log(NULL,AV_LOG_ERROR,Cannot open input filen);return ret;if(ret=avformat_find_stream_info(input_fmt_ctx,NULL)<0)av_log(NULL,AV_LOG_E

6、RROR,Cannot findstream informationn);return ret;/*select the video stream*/ret=av_find_best_stream(input_fmt_ctx,AVMEDIA_TYPE_VIDEO,-1,-1,&dec,0);if(ret<0)av_log(NULL,AV_LOG_ERROR,Cannot find a video stream in the input filen);return ret;input_video_stream_idx=ret;input_dec_ctx=input_fmt_ctx-

7、>streamsinput_video_stream_idx->codec;/*init the video decoder*/if(ret=avcodec_open2(input_dec_ctx,dec,NULL)<0)av_log(NULL,AV_LOG_ERROR,Cannotopen video decodern);return ret;return 0;static int open_overlay_file(const char*filename)int ret;AVCodec*dec;if(ret=avformat_open_input(&overlay

8、_fmt_ctx,filename,NULL,NULL)<0)av_log(NULL,AV_LOG_ERROR,Cannot open input filen);return ret;if(ret=avformat_find_stream_info(overlay_fmt_ctx,NULL)<0)av_log(NULL,AV_LOG_ERROR,Cannot findstream informationn);return ret;/*select the video stream*/ret=av_find_best_stream(overlay_fmt_ctx,AVMEDIA_TY

9、PE_VIDEO,-1,-1,&dec,0);if(ret<0)av_log(NULL,AV_LOG_ERROR,Cannot find a video stream in the input filen);return ret;overlay_video_stream_idx=ret;overlay_dec_ctx=overlay_fmt_ctx->streamsoverlay_video_stream_idx->codec;/*init the video decoder*/if(ret=avcodec_open2(overlay_dec_ctx,dec,NULL

10、)<0)av_log(NULL,AV_LOG_ERROR,Cannotopen video decodern);return ret;printf(overlay format=%sn,overlay_fmt_ctx->iformat->name);return0;static intvideo_config_input_filter(AVFilterInOut*inputs,AVFilterContext*input_filter_ctx)char args512;memset(args,0,sizeof(args);AVFilterContext*first_filter

11、=inputs->filter_ctx;int pad_idx=inputs->pad_idx;AVFilter*filter=avfilter_get_by_name(buffer);/AVRationaltime_base=input_dec_ctx->time_base;AVStream*video_st=input_fmt_ctx->streamsinput_video_stream_idx;AVRational time_base=video_st->time_base;snprintf(args,sizeof(args),video_size=%dx%

12、d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d:sws_param=flags=%d:frame_rate=%d/%d,input_dec_ctx->width,input_dec_ctx->height,input_dec_ctx->pix_fmt,input_dec_ctx->time_base.num,input_dec_ctx->time_base.den,input_dec_ctx->sample_aspect_ratio.num,input_dec_ctx->sample_aspect_rat

13、io.den,SWS_BILINEAR+(video_st->codec->flags&CODEC_FLAG_BITEXACT)?SWS_BITEXACT:0),video_st->r_frame_rate.num,video_st->r_frame_rate.den);printf(inputargs=%sn,args);ret=avfilter_graph_create_filter(input_filter_ctx,filter,src_in,args,NULL,filter_graph);if(ret<0)printf(video config i

14、nput filter fail.n);return-1;ret=avfilter_link(*input_filter_ctx,0,first_filter,pad_idx);assert(ret>=0);printf(video_config_input_filter avfilter_link ret=%dn,ret);return ret;static intvideo_config_overlay_filter(AVFilterInOut*inputs,AVFilterContext*overlay_filter_ctx)charargs512;memset(args,0,si

15、zeof(args);AVFilterContext*first_filter=inputs->filter_ctx;int pad_idx=inputs->pad_idx;AVFilter*filter=avfilter_get_by_name(buffer);/AVRationaltime_base=overlay_dec_ctx->time_base;AVStream*overlay_st=overlay_fmt_ctx->streamsoverlay_video_stream_idx;AVRational time_base=overlay_st->tim

16、e_base;snprintf(args,sizeof(args),video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d:sws_param=flags=%d:frame_rate=%d/%d,overlay_dec_ctx->width,overlay_dec_ctx->height,overlay_dec_ctx->pix_fmt,time_base.num,time_base.den,overlay_dec_ctx->sample_aspect_ratio.num,overlay_dec_ct

17、x->sample_aspect_ratio.den,SWS_BILINEAR+(overlay_st->codec->flags&CODEC_FLAG_BITEXACT)?SWS_BITEXACT:0),overlay_st->r_frame_rate.num,overlay_st->r_frame_rate.den);printf(overlay args=%sn,args);ret=avfilter_graph_create_filter(overlay_filter_ctx,filter,overlay_in,args,NULL,filter_gr

18、aph);if(ret<0)printf(video config overlay filter fail.n);return-1;ret=avfilter_link(*overlay_filter_ctx,0,first_filter,pad_idx);assert(ret>=0);printf(video_config_overlay_filter ret=%dn,ret);avfilter_inout_free(&inputs);return ret;static int video_config_output_filter(AVFilterInOut*outputs

19、,AVFilterContext*out_filter_ctx)char args512;AVFilterContext*last_filter=outputs->filter_ctx;int pad_idx=outputs->pad_idx;AVFilter*buffersink=avfilter_get_by_name(ffbuffersink);int ret=avfilter_graph_create_filter(out_filter_ctx,buffersink,video_out,NULL,NULL,filter_graph);assert(ret>=0);if

20、(ret<0)returnret;ret=avfilter_link(last_filter,pad_idx,*out_filter_ctx,0);assert(ret>=0);if(ret<0)return ret;avfilter_inout_free(&outputs);return 0;static int init_input_filters()filter_graph->scale_sws_opts=av_strdup(flags=0 x4);av_opt_set(filter_graph,aresample_swr_opts,0);ret=avfi

21、lter_graph_parse2(filter_graph,filter_descr,&inputs,&outputs);assert(inputs&&inputs->next&&!inputs->next->next);ret=video_config_input_filter(inputs,&buffersrc_ctx);ret=video_config_overlay_filter(inputs->next,&bufferoverlay_ctx);return ret;staticint init_

22、output_filters()returnvideo_config_output_filter(outputs,&buffersink_ctx);int reap_filters()AVFilterBufferRef*picref;while(1)ret=av_buffersink_get_buffer_ref(buffersink_ctx,&picref,AV_BUFFERSINK_FLAG_NO_REQUEST);if(ret=AVERROR(EAGAIN)|ret=AVERROR_EOF)/printf(reap_filters fail ret=%dn,ret);re

23、turn 0;/no frame filtered.printf(samplesref-n);AVFrame*filtered_frame=avcodec_alloc_frame();avcodec_get_frame_defaults(filtered_frame);avfilter_copy_buf_props(filtered_frame,picref);yuv420p_save(filtered_frame);avfilter_unref_bufferp(&picref);inttranscode_from_filter(AVFilterContext*ifilters,int

24、*eof_reached_arr,int*active_stream_indext)int ret=0;ret=avfilter_graph_request_oldest(filter_graph);if(ret>=0)return ret;if(ret=AVERROR_EOF)return ret;if(ret!=AVERROR(EAGAIN)return ret;int nb_requests_max=0;int i;for(i=0;i<2;i+)int eof_reached=eof_reached_arri;if(eof_reached)continue;AVFilterC

25、ontext*ifilter=ifiltersi;intnb_requests=av_buffersrc_get_nb_failed_requests(ifilter);if(nb_requests>nb_requests_max)nb_requests_max=nb_requests;*active_stream_indext=i;return ret;int main()avcodec_register_all();av_register_all();avfilter_register_all();avformat_network_init();char*video_file=out

26、FileSrc.mp4;char*overlay_video_file=my_logo.png;/light1.mp4#ifENABLE_YUV_FILEconst char*yuvFile=outWater.yuv;fp_yuv=fopen(yuvFile,wb);#endifopen_input_file(video_file);open_overlay_file(overlay_video_file);filter_graph=avfilter_graph_alloc();if(!filter_graph)printf(filter graph alloc fail.n);return-

27、1;init_input_filters();init_output_filters();if(ret=avfilter_graph_config(filter_graph,NULL)<0)return ret;AVFrame*mVideoFrame=avcodec_alloc_frame();AVFrame*mOverlayFrame=avcodec_alloc_frame();while(1)if(video_eof_reached&&overlay_eof_reached)printf(stream EOF.n);break;AVFilterContext*ifil

28、ters=buffersrc_ctx,bufferoverlay_ctx;inteof_reacheds=video_eof_reached,overlay_eof_reached;ret=transcode_from_filter(ifilters,eof_reacheds,&active_stream_index);if(ret>=0)ret=reap_filters();assert(ret>=0);continue;if(ret=AVERROR_EOF)ret=reap_filters();assert(ret>=0);continue;if(ret=AVER

29、ROR(EAGAIN)&&active_stream_index<0)continue;assert(active_stream_index>=0);printf(active_stream_index=%dn,active_stream_index);if(active_stream_index=0)video_transcode_step(mVideoFrame);continue;overlay_transcode_step(mOverlayFrame);if(input_dec_ctx)avcodec_close(input_dec_ctx);avforma

30、t_close_input(&input_fmt_ctx);if(overlay_dec_ctx)avcodec_close(overlay_dec_ctx);avformat_close_input(&overlay_fmt_ctx);printf(my_filtering_video3 end-n);return 0;int video_transcode_step(AVFrame*mVideoFrame)int ret=0;AVPacket pkt;ret=av_read_frame(input_fmt_ctx,&pkt);if(ret=AVERROR(EAGAI

31、N)return 0;if(ret<0)video_eof_reached=1;assert(ret=AVERROR_EOF);ret=video_output_eof_packet(video_eof,input_fmt_ctx->streamsinput_video_stream_idx,buffersrc_ctx);assert(ret>=0);return ret;if(pkt.stream_index!=input_video_stream_idx)/av_free(&pkt);return ret;ret=avcodec_decode_video2(inp

32、ut_dec_ctx,mVideoFrame,&got_frame,&pkt);if(ret<0)printf(Error decoding input videon);if(got_frame)int64_t best_effort_timestamp=av_frame_get_best_effort_timestamp(mVideoFrame);mVideoFrame->pts=best_effort_timestamp;if(av_buffersrc_add_frame(buffersrc_ctx,mVideoFrame,AV_BUFFERSRC_FLAG_P

33、USH)<0)av_log(NULL,AV_LOG_ERROR,Errorwhile feeding the video filtergraphn);return-1;reap_filters();return 0;int overlay_transcode_step(AVFrame*mOverlayFrame)int ret=0;AVPacket pkt;ret=av_read_frame(overlay_fmt_ctx,&pkt);if(ret=AVERROR(EAGAIN)return 0;if(ret<0)overlay_eof_reached=1;ret=vide

34、o_output_eof_packet(overlay_eof,overlay_fmt_ctx->streamsinput_video_stream_idx,bufferoverlay_ctx);assert(ret>=0);return ret;if(pkt.stream_index!=overlay_video_stream_idx)av_free_packet(&pkt);returnret;ret=avcodec_decode_video2(overlay_dec_ctx,mOverlayFrame,&got_frame,&pkt);if(ret&l

35、t;0)printf(Error decoding overlayvideon);if(got_frame)int64_tbest_effort_timestamp=av_frame_get_best_effort_timestamp(mOverlayFrame);mOverlayFrame->pts=best_effort_timestamp;if(av_buffersrc_add_frame(bufferoverlay_ctx,mOverlayFrame,AV_BUFFERSRC_FLAG_PUSH)<0)av_log(NULL,AV_LOG_ERROR,Errorwhile

36、feeding the audio filtergraphn);return-1;return 0;/*output EOF packet to filter to flush*/intvideo_output_eof_packet(const char*tag,AVStream*ist,AVFilterContext*ifilter)int ret=0;/alloc frame if NULLAVFrame*decoded_frame=avcodec_alloc_frame();AVPacket pkt;av_init_packet(&pkt);pkt.data=NULL;pkt.s

37、ize=0;int got_frame=0;ret=avcodec_decode_video2(ist->codec,decoded_frame,&got_frame,&pkt);/EOF,assert gotnothing and ret is 0./TODO:here we still got frame,different to ffmpeg.assert(ret>=0);/flush filterav_buffersrc_add_ref(ifilter,NULL,0);printf(%s filter->eof packet.n,tag);return

38、 ret;/*save yuv420p frame YUV*/void yuv420p_save(AVFrame*pFrame)int i=0;int width=pFrame->width,height=pFrame->height;int height_half=height/2,width_half=width/2;int y_wrap=pFrame->linesize0;int u_wrap=pFrame->linesize1;int v_wrap=pFrame->linesize2;unsigned char*y_buf=pFrame->data0

39、;unsigned char*u_buf=pFrame->data1;unsigned char*v_buf=pFrame->data2;/save yfor(i=0;i<height;i+)fwrite(y_buf+i*y_wrap,1,width,fp_yuv);/save ufor(i=0;i<height_half;i+)fwrite(u_buf+i*u_wrap,1,width_half,fp_yuv);/save vfor(i=0;i<height_half;i+)fwrite(v_buf+i*v_wrap,1,width_half,fp_yuv);fflush(fp_yuv);PS:用到 avfilter 的，如果有两个输入，不管音频还是视频，代码基本都是这样(如 视频的 overlay音频的 amix等等)cpp view plaincopy