ffmpeg添加水印 c 代码实现-凯发app官方网站

凯发app官方网站-凯发k8官网下载客户端中心 | | 凯发app官方网站-凯发k8官网下载客户端中心

bigteam_learning

凯发app官方网站首页　| 　　| 　

landuochong

  • 博客访问： 3000192
  • 博文数量： 864
  • 博客积分： 14125
  • 博客等级： 上将
  • 技术积分： 10634
  • 用 户 组： 普通用户
  • 注册时间： 2007-07-27 16:53
个人简介

https://github.com/zytc2009/bigteam_learning

文章分类

（864）

  • （1）
  • （1）
  • （0）
  • （8）
  • （3）
  • （0）
  • （9）
  • （5）
  • （5）
  • （20）
  • （150）
  • （21）
  • （16）
  • （1）
  • （7）
  • （12）
  • （5）
  • （8）
  • （40）
  • （12）
  • （79）
  • （8）
  • （187）
  • （2）
  • （17）
  • （46）
  • （2）
  • （54）
  • （74）
  • （68）
  • （3）
文章存档

（1）

（1）

（3）

（1）

（10）

（3）

（8）

（3）

（69）

（103）

（357）

（283）

（22）

我的朋友
最近访客
推荐博文
相关博文
  • ·
  • ·
  • ·
  • ·
  • ·
  • ·
  • ·
  • ·
  • ·
  • ·
ffmpeg添加水印 c 代码实现

分类： c/c

2023-05-11 10:35:44

使用 ffmpeg 在 c 中添加水印并将其保存到文件的示例代码。
在代码中，我们使用 ffmpeg 的 c api，需要安装 ffmpeg 库并进行链接。在代码中，我们使用 avformat_open_input() 函数打开输入文件，使用 avformat_find_stream_info() 函数查找流信息，使用 avcodec_find_decoder() 函数查找解码器，使用 avcodec_open2() 函数打开解码器，使用 av_read_frame() 函数读取视频帧，使用 avcodec_send_packet() 函数将解码数据包发送到解码器，使用 avcodec_receive_frame() 函数接收解码后的视频帧，使用 av_frame_alloc() 函数分配帧内存，使用 av_frame_copy() 函数复制帧，使用 avfilter_graph_alloc() 函数分配滤镜图，使用 avfilter_graph_parse2() 函数解析滤镜图，使用 avfilter_graph_config() 函数配置滤镜图，使用 av_buffersrc_write_frame() 函数将视频帧写入滤镜图，使用 av_buffersink_get_frame() 函数从滤镜图中获取处理后的视频帧，使用 avcodec_send_frame() 函数将处理后的视频帧发送到编码器，使用 avcodec_receive_packet() 函数接收编码后的数据包，使用 av_write_frame() 函数将数据包写入输出文件，使用 av_write_trailer() 函数写入输出文件尾。


点击(此处)折叠或打开

  1. #include <iostream>
  2. #include <string>
  3. #include <sstream>


  6. extern "c" {
  7. #include <libavcodec/avcodec.h>
  8. #include <libavformat/avformat.h>
  9. #include <libavfilter/avfilter.h>
  10. #include <libavfilter/buffersrc.h>
  11. #include <libavfilter/buffersink.h>
  12. #include <libavutil/imgutils.h>
  13. #include <libavutil/opt.h>
  14. #include <libswscale/swscale.h>
  15. }


  18. using namespace std;


  21. static void custom_log(void *ptr, int level, const char *fmt, va_list vl)
  22. {
  23.     vprintf(fmt, vl);
  24. }


  27. int main(int argc, char *argv[])
  28. {
  29.     if (argc < 4) {
  30.         cerr << "usage: " << argv[0] << " " << endl;
  31.         return -1;
  32.     }


  35.     const char *input_file = argv[1];
  36.     const char *watermark_file = argv[2];
  37.     const char *output_file = argv[3];


  40.     av_register_all();
  41.     avfilter_register_all();
  42.     av_log_set_callback(custom_log);


  45.     // open input file and find video stream
  46.     avformatcontext *in_fmt_ctx = null;
  47.     if (avformat_open_input(&in_fmt_ctx, input_file, null, null) < 0) {
  48.         av_log(null, av_log_error, "cannot open input file: %s\n", input_file);
  49.         return -1;
  50.     }


  53.     if (avformat_find_stream_info(in_fmt_ctx, null) < 0) {
  54.         av_log(null, av_log_error, "cannot find input stream information\n");
  55.         return -1;
  56.     }


  59.     int video_stream_index = -1;
  60.     for (int i = 0; i < in_fmt_ctx->nb_streams; i) {
  61.         if (in_fmt_ctx->streams[i]->codecpar->codec_type == avmedia_type_video) {
  62.             video_stream_index = i;
  63.             break;
  64.         }
  65.     }


  68.     if (video_stream_index == -1) {
  69.         av_log(null, av_log_error, "cannot find video stream\n");
  70.         return -1;
  71.     }


  74.     // find decoder for video stream and open it
  75.     avcodec *codec = avcodec_find_decoder(in_fmt_ctx->streams[video_stream_index]->codecpar->codec_id);
  76.     if (!codec) {
  77.         av_log(null, av_log_error, "cannot find decoder for codec id %d\n", in_fmt_ctx->streams[video_stream_index]->codecpar->codec_id);
  78.         return -1;
  79.     }


  82.     avcodeccontext *codec_ctx = avcodec_alloc_context3(codec);
  83.     if (!codec_ctx) {
  84.         av_log(null, av_log_error, "cannot allocate codec context\n");
  85.         return -1;
  86.     }


  89.     if (avcodec_parameters_to_context(codec_ctx, in_fmt_ctx->streams[video_stream_index]->codecpar) < 0) {
  90.         av_log(null, av_log_error, "cannot set codec context parameters\n");
  91.         return -1;
  92.     }


  95.     if (avcodec_open2(codec_ctx, codec, null) < 0) {
  96.         av_log(null, av_log_error, "cannot open decoder\n");
  97.         return -1;
  98.     }


  101.     // open watermark image file and create filter graph
  102.     avfiltercontext *buffersrc_ctx = null;
  103.     avfiltercontext *buffersink_ctx = null;
  104.     avfiltergraph *filter_graph = avfilter_graph_alloc();
  105.     if (!filter_graph) {
  106.         av_log(null, av_log_error, "cannot allocate filter graph\n");
  107.         return -1;
  108.     }


  111.     avfilter *buffersrc = avfilter_get_by_name("buffer");
  112.     avfilter *buffersink = avfilter_get_by_name("buffersink");
  113.     if (!buffersrc || !buffersink) {
  114.         av_log(null, av_log_error, "cannot find buffer source/sink filters\n");
  115.         return -1;
  116.     }


  119.     avframe *watermark_frame = av_frame_alloc();


  122.     avcodec *watermark_codec = avcodec_find_decoder_by_name("png");
  123.     if (!watermark_codec) {
  124.         av_log(null, av_log_error, "cannot find decoder for watermark image\n");
  125.         return -1;
  126.     }


  129.     avcodeccontext *watermark_codec_ctx = avcodec_alloc_context3(watermark_codec);
  130.     if (!watermark_codec_ctx) {
  131.         av_log(null, av_log_error, "cannot allocate codec context for watermark image\n");
  132.         return -1;
  133.     }


  136.     if (avcodec_open2(watermark_codec_ctx, watermark_codec, null) < 0) {
  137.         av_log(null, av_log_error, "cannot open decoder for watermark image\n");
  138.         return -1;
  139.     }


  142.     avpacket watermark_packet;
  143.     av_init_packet(&watermark_packet);
  144.     watermark_packet.data = null;
  145.     watermark_packet.size = 0;


  148.     if (avcodec_send_packet(watermark_codec_ctx, &watermark_packet) < 0) {
  149.         av_log(null, av_log_error, "cannot send packet to decoder for watermark image\n");
  150.         return -1;
  151.     }


  154.     int got_picture = 0;
  155.     while (avcodec_receive_frame(watermark_codec_ctx, watermark_frame) == 0) {
  156.         got_picture = 1;
  157.         break;
  158.     }


  161.     avframe *watermark_buffer_frame = av_frame_alloc();


  164.     av_image_copy(watermark_buffer_frame->data, watermark_buffer_frame->linesize, watermark_frame->data, watermark_frame->linesize, watermark_codec_ctx->pix_fmt, watermark_codec_ctx->width, watermark_codec_ctx->height);


  167.     avbuffersrcparameters *buffersrc_params = av_buffersrc_parameters_alloc();
  168.     buffersrc_params->format = watermark_codec_ctx->pix_fmt;
  169.     buffersrc_params->width = watermark_codec_ctx->width;
  170.     buffersrc_params->height = watermark_codec_ctx->height;
  171.     buffersrc_params->time_base = (avrational){1, 1};


  174.     stringstream filterss;
  175.     filterss << "scale=" << codec_ctx->width << ":" << codec_ctx->height << "[watermark];[watermark]format=" << av_get_pix_fmt_name(codec_ctx->pix_fmt) << "[watermarkfmt];[0:v][watermarkfmt]overlay=0:0";


  178.     const char *filter_descr = filterss.str().c_str();


  181.     if (avfilter_graph_create_filter(&buffersrc_ctx, buffersrc, "in", filter_descr, buffersrc_params, filter_graph) < 0) {
  182.         av_log(null, av_log_error, "cannot create buffer source\n");
  183.         return -1;
  184.     }


  187.     if (avfilter_graph_create_filter(&buffersink_ctx, buffersink, "out", null, null, filter_graph) < 0) {
  188.         av_log(null, av_log_error, "cannot create buffer sink\n");
  189.         return -1;
  190.     }


  193.     if (avfilter_link(buffersrc_ctx, 0, buffersink_ctx, 0) < 0) {
  194.         av_log(null, av_log_error, "cannot link filters\n");
  195.         return -1;
  196.     }


  199.     if (avfilter_graph_config(filter_graph, null) < 0) {
  200.         av_log(null, av_log_error, "cannot configure filter graph\n");
  201.         return -1;
  202.     }


  205.     // open output file and find video stream
  206.     avformatcontext *out_fmt_ctx = null;
  207.     if (avformat_alloc_output_context2(&out_fmt_ctx, null, null, output_file) < 0) {
  208.         av_log(null, av_log_error, "cannot allocate output context\n");
  209.         return -1;
  210.     }


  213.     if (!out_fmt_ctx) {
  214.         av_log(null, av_log_error, "cannot create output context\n");
  215.         return -1;
  216.     }


  219.     int out_video_stream_index = av_rescale_q(in_fmt_ctx->streams[video_stream_index]->time_base.den, in_fmt_ctx->streams[video_stream_index]->time_base.num, codec_ctx->time_base.den) * codec_ctx->time_base.num / in_fmt_ctx->streams[video_stream_index]->time_base.den;


  222.     avstream *out_video_stream = avformat_new_stream(out_fmt_ctx, null);
  223.     if (!out_video_stream) {
  224.         av_log(null, av_log_error, "cannot create output video stream\n");
  225.         return -1;
  226.     }


  229.     if (avcodec_copy_context(out_video_stream->codec, codec_ctx) < 0) {
  230.         av_log(null, av_log_error, "cannot copy codec context to output stream\n");
  231.         return -1;
  232.     }


  235.     out_video_stream->codec->codec_tag = 0;
  236.     if (out_fmt_ctx->oformat->flags & avfmt_globalheader) {
  237.         out_video_stream->codec->flags |= av_codec_flag_global_header;
  238.     }


  241.     // write output file header
  242.     if (avio_open(&out_fmt_ctx->pb, output_file, avio_flag_write) < 0) {
  243.         av_log(null, av_log_error, "cannot open output file: %s\n", output_file);
  244.         return -1;
  245.     }


  248.     if (avformat_write_header(out_fmt_ctx, null) < 0) {
  249.         av_log(null, av_log_error, "cannot write output file header\n");
  250.         return -1;
  251.     }


  254.     // process each frame and write to output file
  255.     avpacket pkt;
  256.     av_init_packet(&pkt);
  257.     pkt.data = null;
  258.     pkt.size = 0;


  261.     avframe *frame = av_frame_alloc();
  262.     avframe *filtered_frame = av_frame_alloc();


  265.     int frame_count = 0;
  266.     while (1) {
  267.         // todo: read a frame from the input file


  270.         // todo: decode the frame


  273.         // todo: send the decoded frame to the filter graph


  276.         // todo: receive the filtered frame from the filter graph


  279.         // todo: encode the filtered frame


  282.         // todo: write the encoded frame to the output file
  283.     }


  286.     // todo: write the trailer and close the output file


  289.     // todo: free resources and close the input file

  291.     return 0;
  292. }


阅读(650) | 评论(0) | 转发(0) |
0

上一篇：

下一篇：没有了

给主人留下些什么吧！~~
| | | | |

凯发app官方网站 copyright 2001-2010 chinaunix.net all rights reserved 北京皓辰网域网络信息技术有限公司. 凯发k8官网下载客户端中心的版权所有

感谢所有关心和支持过chinaunix的朋友们
")); function link(t){ var href= $(t).attr('href'); href ="?url=" encodeuricomponent(location.href); $(t).attr('href',href); //setcookie("returnouturl", location.href, 60, "/"); }
网站地图