FFmpeg進階教程之給視頻添加文字水印
更新時間：2022年11月03日 09:52:49 作者：碼農(nóng)飛飛
FFmpeg是一套可以用來記錄、轉換數(shù)字音頻、視頻，并能將其轉化為流的開源計算機程序,下面這篇文章主要給大家介紹了關于FFmpeg進階教程之給視頻添加文字水印的相關資料,需要的朋友可以參考下
前言
和圖片水印一樣，很多時候為了聲明視頻的原創(chuàng)性，我們會給視頻添加文字水印進行版權保護。添加文字水印和添加圖片水印的流程相似，但又略有不同，這里介紹一下如何通過FFmpeg給視頻添加文字水印。添加文字水印的流程圖如下圖所示:
文字水印配置項
在講文字水印之前先介紹一下文字水印支持的那些配置，方便大家的使用。
| 項目 | 介紹 |
|---|---|
| 使用格式 | drawtext=fontfile=font_f:text=text1…(通過冒號分割配置項,通過=給配置項賦值) |
| fontfile | 用于繪制文本的字體文件的正確路徑，強制參數(shù) |
| text | 要繪制的文本字符串，必須是UTF-8編碼的字符序列 |
| x,y | 繪制的位置的起始坐標值 |
| fontcolor | 字體顏色名稱或0xRRGGBB[AA]格式的顏色，默認為黑色 |
| fontsize | 要繪制的文本字體大小，默認值為16 |
| tabsize | 用于呈現(xiàn)選項卡的空間大小，默認值為4 |
| line_h,lh | 每個文本行的高度 |
| main_h,h,H | 輸入的高度 |
| main_w,w,W | 輸入的寬度 |
常用的配置項主要有這些，如果需要其他的配置可以參考官方文檔介紹。
文字水印關鍵點
中文的支持
和QT一樣，F(xiàn)Fmpeg繪制文字水印也存在中文亂碼的問題。在windows下解決中文亂碼主要需要以下幾點:
1.將源碼文件修改為utf-8編碼
2.將編譯編碼類型修改為utf-8編碼對應的配置如下:
#pragma execution_character_set("utf-8")同時我們還應該確保使用的字體支持中文。
字體路徑問題
指定字體文件路徑是強制參數(shù)，可以使用絕對路徑和相對路徑
//使用工程相對路徑下的字體 fontfile=.//simsun.ttc //使用D盤絕對路徑下的字體,要對斜杠進行轉義 fontfile=D\\\\:simun.ttc
定義濾鏡實現(xiàn)
文字水印對應的繪制流程圖如下圖所示:
文字水印濾鏡的實現(xiàn)如下:
int InitFilter(AVCodecContext * codecContext)
{
char args[512];
int ret = 0;
//緩存輸入和緩存輸出
const AVFilter *buffersrc = avfilter_get_by_name("buffer");
const AVFilter *buffersink = avfilter_get_by_name("buffersink");
//創(chuàng)建輸入輸出參數(shù)
AVFilterInOut *outputs = avfilter_inout_alloc();
AVFilterInOut *inputs = avfilter_inout_alloc();
//濾鏡的描述
//使用simhei字體,繪制的字體大小為100,文本內容為"鬼滅之刃",繪制位置為(100,100)
//繪制的字體顏色為白色
string filters_descr = "drawtext=fontfile=.//simsun.ttc:fontsize=100:text=鬼滅之刃:x=100:y=100:fontcolor=0xFFFFFF";
enum AVPixelFormat pix_fmts[] = { AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV420P };
//創(chuàng)建濾鏡容器
filter_graph = avfilter_graph_alloc();
if (!outputs || !inputs || !filter_graph)
{
ret = AVERROR(ENOMEM);
goto end;
}
//初始化數(shù)據(jù)幀的格式
sprintf_s(args, sizeof(args),
"video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d",
codecContext->width, codecContext->height, codecContext->pix_fmt,
codecContext->time_base.num, codecContext->time_base.den,
codecContext->sample_aspect_ratio.num, codecContext->sample_aspect_ratio.den);
//輸入數(shù)據(jù)緩存
ret = avfilter_graph_create_filter(&buffersrc_ctx, buffersrc, "in",
args, NULL, filter_graph);
if (ret < 0) {
goto end;
}
//輸出數(shù)據(jù)緩存
ret = avfilter_graph_create_filter(&buffersink_ctx, buffersink, "out",
NULL, NULL, filter_graph);
if (ret < 0)
{
av_log(NULL, AV_LOG_ERROR, "Cannot create buffer sink\n");
goto end;
}
//設置元素樣式
ret = av_opt_set_int_list(buffersink_ctx, "pix_fmts", pix_fmts,
AV_PIX_FMT_YUV420P, AV_OPT_SEARCH_CHILDREN);
if (ret < 0)
{
av_log(NULL, AV_LOG_ERROR, "Cannot set output pixel format\n");
goto end;
}
//設置濾鏡的端點
outputs->name = av_strdup("in");
outputs->filter_ctx = buffersrc_ctx;
outputs->pad_idx = 0;
outputs->next = NULL;
inputs->name = av_strdup("out");
inputs->filter_ctx = buffersink_ctx;
inputs->pad_idx = 0;
inputs->next = NULL;
//初始化濾鏡
if ((ret = avfilter_graph_parse_ptr(filter_graph, filters_descr.c_str(),
&inputs, &outputs, NULL)) < 0)
goto end;
//濾鏡生效
if ((ret = avfilter_graph_config(filter_graph, NULL)) < 0)
goto end;
end:
//釋放對應的輸入輸出
avfilter_inout_free(&inputs);
avfilter_inout_free(&outputs);
return ret;
}
項目工程源碼
給視頻文件添加文字水印的工程源碼如下，歡迎參考，如有問題歡迎反饋。
#pragma execution_character_set("utf-8")
#include <string>
#include <iostream>
#include <thread>
#include <memory>
#include <iostream>
#include <fstream>
extern "C"
{
#include <libavcodec/avcodec.h>
#include <libavdevice/avdevice.h>
#include <libavfilter/avfilter.h>
#include <libavformat/avformat.h>
#include <libavformat/avio.h>
#include <libavutil/avutil.h>
#include <libswresample/swresample.h>
#include <libswscale/swscale.h>
#include <libavutil/frame.h>
#include <libavutil/imgutils.h>
#include <libavformat/avformat.h>
#include <libavutil/time.h>
#include <libavfilter/avfilter.h>
#include <libavfilter/buffersink.h>
#include <libavfilter/buffersrc.h>
#include <libswscale/swscale.h>
#include <libswresample/swresample.h>
#include <libavdevice/avdevice.h>
}
using namespace std;
//輸入媒體文件的上下文
AVFormatContext * input_format_ctx = nullptr;
//輸出媒體文件的上下文
AVFormatContext* output_format_ctx;
//輸出視頻編碼器
AVCodecContext* ouput_video_encode_ctx = NULL;
//音視頻解碼器
AVCodecContext *video_decode_ctx = NULL;
AVCodecContext *audio_decode_ctx = NULL;
//視頻索引和音頻索引
int video_stream_index = -1;
int audio_stream_index = -1;
//輸出編碼器
static AVCodec * output_video_codec;
//濾鏡容器和緩存
AVFilterGraph * filter_graph = nullptr;
AVFilterContext *buffersink_ctx = nullptr;;
AVFilterContext *buffersrc_ctx = nullptr;;
AVPacket packet;
//起始時間
static int64_t startTime;
int OpenOutput(char *fileName)
{
//創(chuàng)建輸出流,輸出flv格式視頻
int ret = 0;
ret = avformat_alloc_output_context2(&output_format_ctx, NULL, "flv", fileName);
if (ret < 0)
{
return -1;
}
//打開輸出流
ret = avio_open(&output_format_ctx->pb, fileName, AVIO_FLAG_READ_WRITE);
if (ret < 0)
{
return -2;
}
//創(chuàng)建輸出流
for (int index = 0; index < input_format_ctx->nb_streams; index++)
{
if (index == video_stream_index)
{
AVStream * stream = avformat_new_stream(output_format_ctx, output_video_codec);
avcodec_parameters_from_context(stream->codecpar, ouput_video_encode_ctx);
stream->codecpar->codec_tag = 0;
}
else if (index == audio_stream_index)
{
AVStream * stream = avformat_new_stream(output_format_ctx, NULL);
stream->codecpar = input_format_ctx->streams[audio_stream_index]->codecpar;
stream->codecpar->codec_tag = 0;
}
}
//寫文件頭
ret = avformat_write_header(output_format_ctx, nullptr);
if (ret < 0)
{
return -3;
}
if (ret >= 0)
cout << "open output stream successfully" << endl;
return ret;
}
//初始化輸出視頻的編碼器
int InitEncoderCodec(int iWidth, int iHeight)
{
output_video_codec = avcodec_find_encoder(AV_CODEC_ID_H264);
if (NULL == output_video_codec)
{
return -1;
}
//指定編碼器的參數(shù)
ouput_video_encode_ctx = avcodec_alloc_context3(output_video_codec);
ouput_video_encode_ctx->time_base = input_format_ctx->streams[video_stream_index]->time_base;
ouput_video_encode_ctx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
ouput_video_encode_ctx->sample_fmt = AV_SAMPLE_FMT_S16;
ouput_video_encode_ctx->width = iWidth;
ouput_video_encode_ctx->height = iHeight;
ouput_video_encode_ctx->bit_rate = input_format_ctx->streams[video_stream_index]->codecpar->bit_rate;
ouput_video_encode_ctx->pix_fmt = (AVPixelFormat)*output_video_codec->pix_fmts;
ouput_video_encode_ctx->profile = FF_PROFILE_H264_MAIN;
ouput_video_encode_ctx->level = 41;
ouput_video_encode_ctx->thread_count = 8;
return 0;
}
int InitFilter(AVCodecContext * codecContext)
{
char args[512];
int ret = 0;
//緩存輸入和緩存輸出
const AVFilter *buffersrc = avfilter_get_by_name("buffer");
const AVFilter *buffersink = avfilter_get_by_name("buffersink");
//創(chuàng)建輸入輸出參數(shù)
AVFilterInOut *outputs = avfilter_inout_alloc();
AVFilterInOut *inputs = avfilter_inout_alloc();
//濾鏡的描述
//使用simhei字體,繪制的字體大小為100,文本內容為"鬼滅之刃",繪制位置為(100,100)
//繪制的字體顏色為白色
string filters_descr = "drawtext=fontfile=.//simsun.ttc:fontsize=100:text=鬼滅之刃:x=100:y=100:fontcolor=0xFFFFFF";
enum AVPixelFormat pix_fmts[] = { AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV420P };
//創(chuàng)建濾鏡容器
filter_graph = avfilter_graph_alloc();
if (!outputs || !inputs || !filter_graph)
{
ret = AVERROR(ENOMEM);
goto end;
}
//初始化數(shù)據(jù)幀的格式
sprintf_s(args, sizeof(args),
"video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d",
codecContext->width, codecContext->height, codecContext->pix_fmt,
codecContext->time_base.num, codecContext->time_base.den,
codecContext->sample_aspect_ratio.num, codecContext->sample_aspect_ratio.den);
//輸入數(shù)據(jù)緩存
ret = avfilter_graph_create_filter(&buffersrc_ctx, buffersrc, "in",
args, NULL, filter_graph);
if (ret < 0) {
goto end;
}
//輸出數(shù)據(jù)緩存
ret = avfilter_graph_create_filter(&buffersink_ctx, buffersink, "out",
NULL, NULL, filter_graph);
if (ret < 0)
{
av_log(NULL, AV_LOG_ERROR, "Cannot create buffer sink\n");
goto end;
}
//設置元素樣式
ret = av_opt_set_int_list(buffersink_ctx, "pix_fmts", pix_fmts,
AV_PIX_FMT_YUV420P, AV_OPT_SEARCH_CHILDREN);
if (ret < 0)
{
av_log(NULL, AV_LOG_ERROR, "Cannot set output pixel format\n");
goto end;
}
//設置濾鏡的端點
outputs->name = av_strdup("in");
outputs->filter_ctx = buffersrc_ctx;
outputs->pad_idx = 0;
outputs->next = NULL;
inputs->name = av_strdup("out");
inputs->filter_ctx = buffersink_ctx;
inputs->pad_idx = 0;
inputs->next = NULL;
//初始化濾鏡
if ((ret = avfilter_graph_parse_ptr(filter_graph, filters_descr.c_str(),
&inputs, &outputs, NULL)) < 0)
goto end;
//濾鏡生效
if ((ret = avfilter_graph_config(filter_graph, NULL)) < 0)
goto end;
end:
//釋放對應的輸入輸出
avfilter_inout_free(&inputs);
avfilter_inout_free(&outputs);
return ret;
}
//將加水印之后的圖像幀輸出到文件中
static int output_frame(AVFrame *frame, AVRational time_base)
{
int code;
AVPacket packet = { 0 };
av_init_packet(&packet);
int ret = avcodec_send_frame(ouput_video_encode_ctx, frame);
if (ret < 0)
{
printf("Error sending a frame for encoding\n");
return -1;
}
while (ret >= 0)
{
ret = avcodec_receive_packet(ouput_video_encode_ctx, &packet);
if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
{
return (ret == AVERROR(EAGAIN)) ? 0 : 1;
}
else if (ret < 0) {
printf("Error during encoding\n");
exit(1);
}
AVRational avTimeBaseQ = { 1, AV_TIME_BASE };
int64_t ptsTime = av_rescale_q(frame->pts, input_format_ctx->streams[video_stream_index]->time_base, avTimeBaseQ);
int64_t nowTime = av_gettime() - startTime;
if ((ptsTime > nowTime))
{
int64_t sleepTime = ptsTime - nowTime;
av_usleep((sleepTime));
}
else
{
printf("not sleeping\n");
}
packet.pts = av_rescale_q_rnd(packet.pts, time_base, output_format_ctx->streams[video_stream_index]->time_base, (AVRounding)(AV_ROUND_INF | AV_ROUND_PASS_MINMAX));
packet.dts = av_rescale_q_rnd(packet.dts, time_base, output_format_ctx->streams[video_stream_index]->time_base, (AVRounding)(AV_ROUND_INF | AV_ROUND_PASS_MINMAX));
packet.stream_index = video_stream_index;
code = av_interleaved_write_frame(output_format_ctx, &packet);
av_packet_unref(&packet);
if (code < 0)
{
av_log(NULL, AV_LOG_ERROR, "[ERROR] Writing Live Stream Interleaved Frame");
}
if (ret < 0) {
exit(1);
}
av_packet_unref(&packet);
}
}
int main(int argc, char* argv[])
{
if (argc != 3)
{
printf("usage:%1 input filepath %2 outputfilepath");
return -1;
}
//輸入文件地址、輸出文件地址
string fileInput = std::string(argv[1]);
string fileOutput = std::string(argv[2]);
//初始化各種配置
avformat_network_init();
av_log_set_level(AV_LOG_ERROR);
//打開輸入文件
int ret = avformat_open_input(&input_format_ctx, fileInput.c_str(), NULL, NULL);
if (ret < 0)
{
return ret;
}
ret = avformat_find_stream_info(input_format_ctx, NULL);
if (ret < 0)
{
return ret;
}
//查找音視頻流的索引
for (int index = 0; index < input_format_ctx->nb_streams; ++index)
{
if (index == AVMEDIA_TYPE_AUDIO)
{
audio_stream_index = index;
}
else if (index == AVMEDIA_TYPE_VIDEO)
{
video_stream_index = index;
}
}
//打開視頻解碼器
const AVCodec* codec = avcodec_find_decoder(input_format_ctx->streams[video_stream_index]->codecpar->codec_id);
if (!codec)
{
return -1;
}
video_decode_ctx = avcodec_alloc_context3(codec);
if (!video_decode_ctx)
{
fprintf(stderr, "Could not allocate video codec context\n");
return -2;
}
avcodec_parameters_to_context(video_decode_ctx, input_format_ctx->streams[video_stream_index]->codecpar);
if (codec->capabilities & AV_CODEC_CAP_TRUNCATED)
video_decode_ctx->flags |= AV_CODEC_FLAG_TRUNCATED;
ret = avcodec_open2(video_decode_ctx, codec, NULL);
if (ret < 0)
{
av_free(video_decode_ctx);
return -3;
}
//初始化視頻編碼器
ret = InitEncoderCodec(video_decode_ctx->width, video_decode_ctx->height);
if (ret < 0)
{
return 0;
}
//初始化濾鏡
ret = InitFilter(ouput_video_encode_ctx);
//打開編碼器
ret = avcodec_open2(ouput_video_encode_ctx, output_video_codec, NULL);
if (ret < 0)
{
return ret;
}
//初始化輸出
if (OpenOutput((char *)fileOutput.c_str()) < 0)
{
cout << "Open file Output failed!" << endl;
this_thread::sleep_for(chrono::seconds(10));
return 0;
}
AVFrame* pSrcFrame = av_frame_alloc();
AVFrame* filterFrame = av_frame_alloc();
av_init_packet(&packet);
startTime = av_gettime();
while (true)
{
int ret = av_read_frame(input_format_ctx, &packet);
if (ret < 0)
{
break;
}
//視頻幀通過濾鏡處理之后編碼輸出
if (packet.stream_index == video_stream_index)
{
int ret = avcodec_send_packet(video_decode_ctx, &packet);
if (ret < 0)
{
break;
}
while (ret >= 0)
{
ret = avcodec_receive_frame(video_decode_ctx, pSrcFrame);
if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
{
break;
}
else if (ret < 0)
{
goto End;
}
pSrcFrame->pts = pSrcFrame->best_effort_timestamp;
//添加到濾鏡中
if (av_buffersrc_add_frame_flags(buffersrc_ctx, pSrcFrame, AV_BUFFERSRC_FLAG_KEEP_REF) < 0)
{
break;
}
while (1)
{
//獲取濾鏡輸出
int ret = av_buffersink_get_frame(buffersink_ctx, filterFrame);
if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
{
break;
}
else if (ret < 0)
{
goto End;
}
//編碼之后輸出
output_frame(filterFrame, buffersink_ctx->inputs[0]->time_base);
av_frame_unref(filterFrame);
}
av_frame_unref(pSrcFrame);
}
}
else if (packet.stream_index == audio_stream_index)
{
packet.pts = av_rescale_q_rnd(packet.pts, input_format_ctx->streams[audio_stream_index]->time_base, output_format_ctx->streams[audio_stream_index]->time_base, (AVRounding)(AV_ROUND_INF | AV_ROUND_PASS_MINMAX));
packet.dts = av_rescale_q_rnd(packet.dts, input_format_ctx->streams[audio_stream_index]->time_base, output_format_ctx->streams[audio_stream_index]->time_base, (AVRounding)(AV_ROUND_INF | AV_ROUND_PASS_MINMAX));
packet.stream_index = audio_stream_index;
av_interleaved_write_frame(output_format_ctx, &packet);
}
av_packet_unref(&packet);
}
av_write_trailer(output_format_ctx);
End:
//結束的時候清理資源
avfilter_graph_free(&filter_graph);
if (input_format_ctx != NULL)
{
avformat_close_input(&input_format_ctx);
}
avcodec_free_context(&video_decode_ctx);
avcodec_free_context(&ouput_video_encode_ctx);
return 0;
}
使用效果
沒有添加水印之前的視頻截圖如下:
添加水印之后的效果圖如下圖所示:
總結
到此這篇關于FFmpeg進階教程之給視頻添加文字水印的文章就介紹到這了,更多相關FFmpeg視頻加文字水印內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
詳解如何在code block創(chuàng)建一個C語言的項目
這篇文章主要介紹了詳解如何在code block創(chuàng)建一個C語言的項目，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-12-12
這篇文章主要為大家詳細介紹了C語言實現(xiàn)簡單酒店管理系統(tǒng)，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-03-03
C++計算任意權值的單源最短路徑（Bellman-Ford）
這篇文章主要為大家詳細介紹了C++計算任意權值的單源最短路徑，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2020-04-04
C++的繼承機制相對其他語言是比較復雜的一種，不同于java只支持單繼承，C++不僅支持單繼承，也支持多繼承。本文將詳細講解C++中單繼承與多繼承的使用，需要的可以參考一下2022-04-04
這篇文章主要介紹了C語言中二級指針的實例詳解的相關資料,希望通過本文能幫助到大家，讓大家掌握理解二級指針的知識，需要的朋友可以參考下2017-10-10
本篇文章是對uCOS中全局變量的使用進行了詳細的分析介紹，需要的朋友參考下2013-05-05
這篇文章主要介紹了基于C++?OpenCV實現(xiàn)的車道檢測，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-01-01
C++實現(xiàn)LeetCode(162.求數(shù)組的局部峰值)
這篇文章主要介紹了C++實現(xiàn)LeetCode(162.求數(shù)組的局部峰值),本篇文章通過簡要的案例,講解了該項技術的了解與使用,以下就是詳細內容,需要的朋友可以參考下2021-07-07
這篇文章主要分享了C++友元函數(shù)講解，C++提供了一種形式的訪問權限，叫做友元，友元有三種，分別是友元函數(shù)、友元類和友元成員函數(shù)，下面將詳細介紹該內容，需要的小伙伴可以參考一下2022-01-01
C++?正式分類方法是直接按語法分類，分為：隱式轉換和顯示轉換。這篇文章主要介紹了C++?轉換的非正式分類,需要的朋友可以參考下2022-01-01