ffmpeg如何添加水印c语言

FFmpeg添加水印的C语言实现方法包括：使用FFmpeg库、理解视频处理流程、编写添加水印的具体代码。FFmpeg是一个强大的多媒体处理库，可以通过C语言调用其API来完成各种媒体处理任务，包括在视频中添加水印。以下是详细描述其中一个步骤的方法。

一、FFmpeg库介绍

FFmpeg是一个开源的跨平台多媒体处理工具，可以处理音频、视频以及其他多媒体文件和流。它提供了一套丰富的API，供开发者使用C语言进行多媒体处理。通过FFmpeg的API，开发者可以实现视频解码、编码、转码、剪辑、滤镜等功能。

二、准备工作

在开始编写代码之前，确保你已经安装了FFmpeg库，并且能够在你的开发环境中正确链接和使用它。你可以通过以下命令来安装FFmpeg库：

sudo apt-get install ffmpeg libavcodec-dev libavformat-dev libavfilter-dev libswscale-dev libavutil-dev

三、初始化FFmpeg库

在使用FFmpeg库之前，首先需要初始化库。在C语言中，可以使用以下代码进行初始化：

#include <libavformat/avformat.h>
int main(int argc, char *argv[]) {
    av_register_all();
    avfilter_register_all();
    // Your code here
    return 0;
}

av_register_all函数用于注册所有的编解码器、封装器和协议，avfilter_register_all函数用于注册所有的滤镜。

四、打开输入视频文件

要处理视频文件，首先需要打开输入视频文件。可以使用以下代码来实现：

AVFormatContext *pFormatCtx = NULL;
if (avformat_open_input(&pFormatCtx, "input.mp4", NULL, NULL) != 0) {
    fprintf(stderr, "Could not open input file.n");
    return -1;
}
if (avformat_find_stream_info(pFormatCtx, NULL) < 0) {
    fprintf(stderr, "Could not find stream information.n");
    return -1;
}

avformat_open_input函数用于打开输入视频文件，avformat_find_stream_info函数用于读取视频流信息。

五、查找视频流

在打开输入视频文件之后，需要找到视频流。可以使用以下代码来实现：

int videoStream = -1;
for (int i = 0; i < pFormatCtx->nb_streams; i++) {
    if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
        videoStream = i;
        break;
    }
}
if (videoStream == -1) {
    fprintf(stderr, "Could not find video stream.n");
    return -1;
}

该代码遍历所有的流，找到第一个视频流。

六、获取解码器和解码器上下文

找到视频流之后，需要获取解码器和解码器上下文。可以使用以下代码来实现：

AVCodecParameters *pCodecPar = pFormatCtx->streams[videoStream]->codecpar;
AVCodec *pCodec = avcodec_find_decoder(pCodecPar->codec_id);
if (pCodec == NULL) {
    fprintf(stderr, "Unsupported codec.n");
    return -1;
}
AVCodecContext *pCodecCtx = avcodec_alloc_context3(pCodec);
if (avcodec_parameters_to_context(pCodecCtx, pCodecPar) < 0) {
    fprintf(stderr, "Could not copy codec parameters to context.n");
    return -1;
}
if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0) {
    fprintf(stderr, "Could not open codec.n");
    return -1;
}

七、读取帧并添加水印

在获取解码器和解码器上下文之后，可以读取帧并添加水印。可以使用以下代码来实现：

AVFrame *pFrame = av_frame_alloc();
AVFrame *pFrameRGB = av_frame_alloc();
if (pFrame == NULL || pFrameRGB == NULL) {
    fprintf(stderr, "Could not allocate frame.n");
    return -1;
}
int numBytes = av_image_get_buffer_size(AV_PIX_FMT_RGB24, pCodecCtx->width, pCodecCtx->height, 1);
uint8_t *buffer = (uint8_t *)av_malloc(numBytes * sizeof(uint8_t));
av_image_fill_arrays(pFrameRGB->data, pFrameRGB->linesize, buffer, AV_PIX_FMT_RGB24, pCodecCtx->width, pCodecCtx->height, 1);
struct SwsContext *sws_ctx = sws_getContext(
    pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt,
    pCodecCtx->width, pCodecCtx->height, AV_PIX_FMT_RGB24,
    SWS_BILINEAR, NULL, NULL, NULL
);
AVPacket packet;
while (av_read_frame(pFormatCtx, &packet) >= 0) {
    if (packet.stream_index == videoStream) {
        if (avcodec_send_packet(pCodecCtx, &packet) == 0) {
            while (avcodec_receive_frame(pCodecCtx, pFrame) == 0) {
                sws_scale(
                    sws_ctx,
                    (uint8_t const * const *)pFrame->data,
                    pFrame->linesize,
                    0,
                    pCodecCtx->height,
                    pFrameRGB->data,
                    pFrameRGB->linesize
                );
                // Add watermark here
                add_watermark(pFrameRGB);
                // Save frame to output file
                save_frame(pFrameRGB, pCodecCtx->width, pCodecCtx->height);
            }
        }
    }
    av_packet_unref(&packet);
}

在该代码中，sws_getContext函数用于创建图像转换上下文，sws_scale函数用于将帧转换为RGB格式。你可以在add_watermark函数中添加水印，并在save_frame函数中保存处理后的帧。

八、实现添加水印的函数

以下是一个简单的添加水印的函数实现：

void add_watermark(AVFrame *pFrame) {
    int x, y;
    for (y = 0; y < 100; y++) {
        for (x = 0; x < 100; x++) {
            pFrame->data[0][y * pFrame->linesize[0] + x * 3 + 0] = 255;  // Red
            pFrame->data[0][y * pFrame->linesize[0] + x * 3 + 1] = 0;    // Green
            pFrame->data[0][y * pFrame->linesize[0] + x * 3 + 2] = 0;    // Blue
        }
    }
}

该函数在帧的左上角添加了一个100×100的红色矩形作为水印。

九、实现保存帧的函数

以下是一个简单的保存帧的函数实现：

void save_frame(AVFrame *pFrame, int width, int height) {
    FILE *pFile = fopen("output.ppm", "wb");
    if (pFile == NULL) {
        return;
    }
    fprintf(pFile, "P6n%d %dn255n", width, height);
    for (int y = 0; y < height; y++) {
        fwrite(pFrame->data[0] + y * pFrame->linesize[0], 1, width * 3, pFile);
    }
    fclose(pFile);
}

该函数将帧保存为PPM格式的图像文件。

十、清理资源

在程序结束时，需要清理资源。可以使用以下代码来实现：

av_free(buffer);
av_frame_free(&pFrame);
av_frame_free(&pFrameRGB);
avcodec_close(pCodecCtx);
avformat_close_input(&pFormatCtx);

总结

通过以上步骤，你可以使用FFmpeg库在C语言中实现添加水印的功能。从初始化库、打开输入文件、查找视频流、获取解码器、读取帧、添加水印到保存帧，整个过程涉及多个步骤和函数调用。FFmpeg提供了丰富的API，供开发者灵活使用。通过深入理解和灵活运用这些API，你可以实现各种复杂的多媒体处理任务。

在实际应用中，你可以根据具体需求调整和优化代码。例如，你可以将水印的位置、大小和内容参数化，以便更灵活地控制水印的添加。同时，你还可以将处理后的帧编码并保存为视频文件，而不仅仅是保存为图像文件。通过深入学习和探索FFmpeg库，你可以实现更加复杂和多样化的多媒体处理功能。