使用ffmepg实现多路视频流合并

2024-08-29 05:58

本文主要是介绍使用ffmepg实现多路视频流合并,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

做视频会议系统的时候,有时需要实现多路视频画面合并后推流功能,要直接底层实现这样的功能还是不太容易的,如果借助ffmpeg就方便多了,使用ffmpeg的滤镜功能就能实现多路合并的效果。

首先说明需要用到的ffmpeg对象,以及一些必要的字段。

ffmpeg版本:

version 4.3

所用到的头文件:

#include <libavutil/avassert.h>
#include <libavutil/opt.h>
#include <libavfilter/avfilter.h>

需要的数据结构如下:

每一条输入流需要如下的字段

typedef struct Stream{int x;int y;int width;int height;int format;//参考AVPixelFormatAVFilterContext* buffersrc_ctx;AVFilter* buffersrc;AVFilterInOut* output;AVFrame* inputFrame;
} Stream;

输出流需要如下字段

typedef struct Merge {AVFilterGraph* filter_graph;AVFilterContext* buffersink_ctx;AVFilter* buffersink;AVFilterInOut* input;const char* filters_descr;AVFrame* outputFrame;unsigned char* outputBuffer;int outputWidth;int outputHeight;int outputFormat;//参考AVPixelFormatStream *streams[128];int streamCount;
} Merge;

主要流程如下:

1、构造输出流及输入流

构造输出流,输出流需要设置分辨率以及输出的像素格式

Merge* Merge_Create(int outputWidth, int outputHeight, int outputFormat) {Merge* merge = malloc(sizeof(Merge));memset(merge, 0, sizeof(Merge));merge->outputWidth = outputWidth;merge->outputHeight = outputHeight;merge->outputFormat = outputFormat;return merge;
}

添加输入流,输入流需要设置在输出流中的位置和大小,以及输入流像素格式 

Stream* Merge_AddStream(Merge* merge, int x, int y, int width, int height, int format) {Stream* stream = malloc(sizeof(Stream));memset(merge, 0, sizeof(Stream));stream->x = x;stream->y = y;stream->width = width;stream->height = height;stream->format = format;merge->streams[merge->streamCount++] = stream;return stream;
}

2、初始化滤镜

主要用到的滤镜是filters_descr = "[in0]pad=1280:640:0:0:black[x0];[x0][in1]overlay=640:0[x1];[x1][in2]overlay=600:0[x2];[x2]null[out]";

//初始化Merge
int Merge_Init(Merge* merge) {char args[512];char name[8];char* filters_descr = NULL;int ret;//avfilter_register_all();//旧版可能用到此行merge->buffersink = avfilter_get_by_name("buffersink");av_assert0(merge->buffersink);merge->input = avfilter_inout_alloc();if (merge->input == NULL){printf("alloc inout  fail\n");goto fail;}merge->filter_graph = avfilter_graph_alloc();if (merge->input == NULL){printf("alloc graph  fail\n");goto fail;}ret = avfilter_graph_create_filter(&merge->buffersink_ctx, merge->buffersink, "out", NULL, NULL, merge->filter_graph);if (ret < 0){printf("graph create  fail\n");goto fail;}merge->input->name = av_strdup("out");merge->input->filter_ctx = merge->buffersink_ctx;merge->input->pad_idx = 0;merge->input->next = NULL;merge->outputFrame = av_frame_alloc();if (merge->outputFrame == NULL){printf("alloc frame  fail\n");goto fail;}merge->outputBuffer = (unsigned char*)av_malloc(av_image_get_buffer_size(merge->outputFormat, merge->outputWidth, merge->outputHeight, 1));if (merge->outputBuffer == NULL){printf("alloc buffer  fail\n");goto fail;}enum AVPixelFormat pix_fmts[2] = { 0, AV_PIX_FMT_NONE };pix_fmts[0] = merge->outputFormat;ret = av_opt_set_int_list(merge->buffersink_ctx, "pix_fmts", pix_fmts, AV_PIX_FMT_NONE, AV_OPT_SEARCH_CHILDREN);if (ret < 0) {printf("set opt fail\n");goto fail;}Stream** streams = merge->streams;for (int i = 0; i < merge->streamCount; i++){streams[i]->buffersrc = avfilter_get_by_name("buffer");av_assert0(streams[i]->buffersrc);snprintf(args, sizeof(args), "video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d", streams[i]->width, streams[i]->height, streams[i]->format, 1, 90000, 1, 1);snprintf(name, sizeof(name), "in%d", i);ret = avfilter_graph_create_filter(&streams[i]->buffersrc_ctx, streams[i]->buffersrc, name, args, NULL, merge->filter_graph);if (ret < 0){printf("stream graph create fail\n");goto fail;}streams[i]->output = avfilter_inout_alloc();streams[i]->output->name = av_strdup(name);streams[i]->output->filter_ctx = streams[i]->buffersrc_ctx;streams[i]->output->pad_idx = 0;streams[i]->output->next = NULL;streams[i]->inputFrame = av_frame_alloc();if (streams[i]->inputFrame == NULL){printf("alloc frame  fail\n");goto fail;}streams[i]->inputFrame->format = streams[i]->format;streams[i]->inputFrame->width = streams[i]->width;streams[i]->inputFrame->height = streams[i]->height;if (i > 0){streams[i - 1]->output->next = streams[i]->output;}}/*filters_descr = "[in0]pad=1280:640:0:0:black[x0];[x0][in1]overlay=640:0[x1];[x1][in2]overlay=600:0[x2];[x2]null[out]";*/filters_descr = malloc(sizeof(char) * merge->streamCount * 128);if (filters_descr == NULL){printf("alloc string  fail\n");goto fail;}char sigle_descr[128];snprintf(sigle_descr, sizeof(sigle_descr), "[in0]pad=%d:%d:%d:%d:black[x0];", merge->outputWidth, merge->outputHeight, streams[0]->x, streams[0]->y);strcpy(filters_descr, sigle_descr);int i = 1;for (; i < merge->streamCount; i++){snprintf(sigle_descr, sizeof(sigle_descr), "[x%d][in%d]overlay=%d:%d[x%d];", i - 1, i, streams[i]->x, streams[i]->y, i);strcat(filters_descr, sigle_descr);}snprintf(sigle_descr, sizeof(sigle_descr), "[x%d]null[out]", i - 1);strcat(filters_descr, sigle_descr);ret = avfilter_graph_parse_ptr(merge->filter_graph, filters_descr, &merge->input, &streams[0]->output, NULL);if (ret < 0){printf("graph parse fail\n");goto fail;}// 过滤配置初始化ret = avfilter_graph_config(merge->filter_graph, NULL);if (ret < 0){printf("graph config fail\n");goto fail;}if (filters_descr != NULL)free(filters_descr);return 0;
fail:if (filters_descr != NULL)free(filters_descr);Merge_Deinit(merge); //执行反初始化return -1;
}

3、写输入流

void Merge_WriteStream(Merge* merge, Stream* stream, const unsigned char* buffer, int timestamp) {av_image_fill_arrays(stream->inputFrame->data, stream->inputFrame->linesize, buffer, stream->format, stream->width, stream->height, 1);stream->inputFrame->pts = timestamp;if (av_buffersrc_write_frame(stream->buffersrc_ctx, stream->inputFrame) < 0) {printf("Error while add frame.\n");}
}

4、合并流

调用下列方法,即可得到合并后的一帧的数据。可以按照一定帧率调用3、4方法。

const unsigned char* Merge_Merge(Merge* merge) {int	ret = av_buffersink_get_frame(merge->buffersink_ctx, merge->outputFrame);if (ret < 0) {printf("Error while get frame.\n");return NULL;}av_image_copy_to_buffer(merge->outputBuffer,av_image_get_buffer_size(merge->outputFormat, merge->outputWidth, merge->outputHeight, 1),merge->outputFrame->data, merge->outputFrame->linesize, merge->outputFrame->format, merge->outputFrame->width, merge->outputFrame->height, 1);av_frame_unref(merge->outputFrame);return  merge->outputBuffer;
}

5、结束,反初始化,销毁对象

static void Merge_Deinit(Merge* merge)
{if (merge->input != NULL)avfilter_inout_free(&merge->input);if (merge->filter_graph != NULL)avfilter_graph_free(&merge->filter_graph);if (merge->outputFrame != NULL)av_frame_free(&merge->outputFrame);if (merge->outputBuffer != NULL)av_free(merge->outputBuffer);for (int i = 0; i < merge->streamCount; i++){Stream* stream = merge->streams[i]; if (stream->inputFrame != NULL)av_frame_free(&stream->inputFrame);free(stream);}merge->streamCount = 0;
}
void Merge_Destroy(Merge* merge) {Merge_Deinit(merge);free(merge);
}

调用流程示例:

int main() {int flag = 1;Merge* merge = Merge_Create(1920, 1080, 0);Stream* stream1 = Merge_AddStream(merge, 0, 0, 960, 540, 0);Stream* stream2 = Merge_AddStream(merge, 960, 540, 960, 540, 0);Stream* stream3 = Merge_AddStream(merge, 0, 540, 960, 540, 0);if (Merge_Init(merge) != 0){Merge_Destroy(merge);return -1;}while (flag){clock_t time = clock();unsigned char* buffer1;unsigned char* buffer2;unsigned char* buffer3;//获取每路流的数据...//获取每路流的数据-endMerge_WriteStream(merge, stream1, buffer1, time);Merge_WriteStream(merge, stream2, buffer2, time);Merge_WriteStream(merge, stream3, buffer3, time);unsigned char* mergedBuffer = Merge_Merge(merge);//显示或编码推流...//显示或编码推流-end}Merge_Destroy(merge);
}

需要注意的是上述方法最好在单线程中使用,多线程使用可能需要另外做修改,或者参考:

https://download.csdn.net/download/u013113678/32899063

效果如下

这篇关于使用ffmepg实现多路视频流合并的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1117038

相关文章

Spring Boot整合Redis注解实现增删改查功能(Redis注解使用)

《SpringBoot整合Redis注解实现增删改查功能(Redis注解使用)》文章介绍了如何使用SpringBoot整合Redis注解实现增删改查功能,包括配置、实体类、Repository、Se... 目录配置Redis连接定义实体类创建Repository接口增删改查操作示例插入数据查询数据删除数据更

Java Lettuce 客户端入门到生产的实现步骤

《JavaLettuce客户端入门到生产的实现步骤》本文主要介绍了JavaLettuce客户端入门到生产的实现步骤,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要... 目录1 安装依赖MavenGradle2 最小化连接示例3 核心特性速览4 生产环境配置建议5 常见问题

使用python生成固定格式序号的方法详解

《使用python生成固定格式序号的方法详解》这篇文章主要为大家详细介绍了如何使用python生成固定格式序号,文中的示例代码讲解详细,具有一定的借鉴价值,有需要的小伙伴可以参考一下... 目录生成结果验证完整生成代码扩展说明1. 保存到文本文件2. 转换为jsON格式3. 处理特殊序号格式(如带圈数字)4

Java使用Swing生成一个最大公约数计算器

《Java使用Swing生成一个最大公约数计算器》这篇文章主要为大家详细介绍了Java使用Swing生成一个最大公约数计算器的相关知识,文中的示例代码讲解详细,感兴趣的小伙伴可以了解一下... 目录第一步:利用欧几里得算法计算最大公约数欧几里得算法的证明情形 1:b=0情形 2:b>0完成相关代码第二步:加

linux ssh如何实现增加访问端口

《linuxssh如何实现增加访问端口》Linux中SSH默认使用22端口,为了增强安全性或满足特定需求,可以通过修改SSH配置来增加或更改SSH访问端口,具体步骤包括修改SSH配置文件、增加或修改... 目录1. 修改 SSH 配置文件2. 增加或修改端口3. 保存并退出编辑器4. 更新防火墙规则使用uf

Java 的ArrayList集合底层实现与最佳实践

《Java的ArrayList集合底层实现与最佳实践》本文主要介绍了Java的ArrayList集合类的核心概念、底层实现、关键成员变量、初始化机制、容量演变、扩容机制、性能分析、核心方法源码解析、... 目录1. 核心概念与底层实现1.1 ArrayList 的本质1.1.1 底层数据结构JDK 1.7

Java中流式并行操作parallelStream的原理和使用方法

《Java中流式并行操作parallelStream的原理和使用方法》本文详细介绍了Java中的并行流(parallelStream)的原理、正确使用方法以及在实际业务中的应用案例,并指出在使用并行流... 目录Java中流式并行操作parallelStream0. 问题的产生1. 什么是parallelS

C++中unordered_set哈希集合的实现

《C++中unordered_set哈希集合的实现》std::unordered_set是C++标准库中的无序关联容器,基于哈希表实现,具有元素唯一性和无序性特点,本文就来详细的介绍一下unorder... 目录一、概述二、头文件与命名空间三、常用方法与示例1. 构造与析构2. 迭代器与遍历3. 容量相关4

Linux join命令的使用及说明

《Linuxjoin命令的使用及说明》`join`命令用于在Linux中按字段将两个文件进行连接,类似于SQL的JOIN,它需要两个文件按用于匹配的字段排序,并且第一个文件的换行符必须是LF,`jo... 目录一. 基本语法二. 数据准备三. 指定文件的连接key四.-a输出指定文件的所有行五.-o指定输出

Linux jq命令的使用解读

《Linuxjq命令的使用解读》jq是一个强大的命令行工具,用于处理JSON数据,它可以用来查看、过滤、修改、格式化JSON数据,通过使用各种选项和过滤器,可以实现复杂的JSON处理任务... 目录一. 简介二. 选项2.1.2.2-c2.3-r2.4-R三. 字段提取3.1 普通字段3.2 数组字段四.