Python&C++相互混合调用编程全面实战-32完成python调用扩展库pyffmpeg的decode解码函数

最新推荐文章于 2024-08-03 18:36:56 发布

虚坏叔叔

最新推荐文章于 2024-08-03 18:36:56 发布

阅读量370

点赞数

分类专栏： Python&C++混合编程文章标签： c++ python 解码 ffmpeg 帧数

本文链接：https://blog.csdn.net/biggbang/article/details/125574308

版权

Python&C++混合编程专栏收录该内容

31 篇文章 30 订阅

订阅专栏

作者：虚坏叔叔
博客：https://xuhss.com

早餐店不会开到晚上，想吃的人早就来了！😄

完成python调用扩展库pyffmpeg的decode解码函数

在这里插入图片描述

一、添加解码函数

XFFmpeg.h添加decode函数和成员变量

	bool Decode();
	AVFrame *frame = 0;

完整代码如下：

#pragma once
struct AVFormatContext;
struct AVPacket;
struct AVCodecContext;
struct AVFrame;
class XFFmpeg
{
public:
	// 打开视频
	bool Open(const char *url);
	// 读取一阵视频 在内部存储
	bool Read();

	bool Decode();

	void Close();
	XFFmpeg();
	~XFFmpeg();
	int totalms = 0;
protected:
	// 解封装上下文
	AVFormatContext *ic = 0;
	// 读取视频帧
	AVPacket *pkt = 0;
	// 解码器上下文
	AVCodecContext *vcodec = 0;
	// 解码后的yuv帧
	AVFrame *frame = 0;
};

XFFmpeg.cpp添加代码：

包括avframe申请以及帧数据的清理

完整代码如下：

#include "XFFmpeg.h"
#include <stdio.h>
extern "C" {
#include "libavformat\avformat.h"
}

void XFFmpeg::Close()
{
	printf("XFFmpeg::Close\n");
	if (ic)
	{
		avformat_close_input(&ic);

	}
	if (vcodec)
	{
		avcodec_free_context(&vcodec);
	}
	if (pkt)
	{
		av_packet_free(&pkt);
	}
	if (frame)
	{
		av_frame_free(&frame);
	}
}

bool XFFmpeg::Open(const char *url)
{
	printf("XFFmpeg::open %s\n", url);
	// 打开视频 解封装
	int re =avformat_open_input(&ic, url, 0, 0);
	if (re != 0)
	{
		char buf[1024] = { 0 };
		av_strerror(re, buf, 1023);
		printf("avformat open fail:%s\n", buf);
		return false;
	}

	// 获取流
	avformat_find_stream_info(ic, 0);

	// 获取视频总时长
	this->totalms = ic->duration / (AV_TIME_BASE / 1000);
	printf("Total Ms =%d\n", totalms);

	// 打开解码器 0视频 1音频
	AVCodecParameters *para = ic->streams[0]->codecpar;

	// 找到解码器
	AVCodec * codec = avcodec_find_decoder(para->codec_id);

	if (!codec)
	{
		printf("avcodec_find_decoder %d failed!\n", para->codec_id);
	}

	// 分配解码器上下文
	vcodec = avcodec_alloc_context3(codec);

	// 配置解码器上下文
	avcodec_parameters_to_context(vcodec, para);
	// 多线程解码
	vcodec->thread_count = 8;

	// 打开上下文
	re = avcodec_open2(vcodec, 0, 0);
	if (re != 0)
	{
		avcodec_free_context(&vcodec);
		char buf[1024];
		av_strerror(re, buf, sizeof(buf) - 1);
		printf("avcodec_open2 failec:%s!\n", buf);
	}
	printf("avcodec_open2 successed \n");

	return true;
}

bool XFFmpeg::Read()
{
	if (!ic)
		return false;
	// 视频帧的存储空间
	if (!pkt)
	{
		// 分配对象空间
		pkt = av_packet_alloc();
	}
	else
	{
		// 引用计数 -1 清理视频帧
		av_packet_unref(pkt);
	}
	int re = 0;
	bool isFindVideo = false;
	// 音频数据丢掉
	for (int i = 0; i < 20; i++)
	{

		// 读取一帧数据
		re = av_read_frame(ic, pkt);

		// 读取失败或者读取到文件结尾
		if (re != 0)
		{
			return false;
		}

		// 是否是视频帧
		if (pkt->stream_index == 0)
		{
			isFindVideo = true;
			break;
		}

		// 音频帧 清理packet
		av_packet_unref(pkt);
	}

	return isFindVideo;

}


bool XFFmpeg::Decode() {
	if (!vcodec || !pkt) return false;

	// 发送到解码线程
	int re = avcodec_send_packet(vcodec, pkt);
	if (re != 0) return false;
	if (!frame)
		frame = av_frame_alloc();

	// 取到解码后的数据 yuv 前几帧会失败 失败不代表解码失败 只表示解码缓冲没有数据
	re = avcodec_receive_frame(vcodec, frame);
	if (re != 0) return false;

	return true;
}

XFFmpeg::XFFmpeg()
{
	printf("Create XFFmpeg\n");
}


XFFmpeg::~XFFmpeg()
{
	printf("Delete XFFmpeg\n");
}

二、开放解码器调用接口

pyFFmpeg.h 开放接口

	static PyObject *Decode(PyFFmpeg*self, PyObject*args);

pyFFmpeg. cpp添加接口的调用和注册：

PyObject *PyFFmpeg::Decode(PyFFmpeg*self, PyObject*args)
{

	if (!self->ff)
		Py_RETURN_FALSE;
	if (self->ff->Decode())
		Py_RETURN_TRUE;
	Py_RETURN_FALSE;
}

// 模块入口 模块名称 pyffmpeg
PyMODINIT_FUNC PyInit_pyffmpeg(void)
{
	PyObject *m = NULL;
	static PyModuleDef ffmod = {
		PyModuleDef_HEAD_INIT,
		"pyffmpeg",
		"", -1, 0
	};
	m = PyModule_Create(&ffmod);

	// 添加PyFFmpeg_python类
	static PyTypeObject type;
	memset(&type, 0, sizeof(PyFFmpeg));
	type.ob_base = { PyObject_HEAD_INIT(NULL) 0 };
	type.tp_name = "";
	type.tp_basicsize = sizeof(PyFFmpeg);
	type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE;
	type.tp_new = PyFFmpeg::Create;
	type.tp_init = (initproc)PyFFmpeg::Init;
	type.tp_dealloc = (destructor)PyFFmpeg::Close;

	static PyMethodDef ffmeth[] = {
		{ "open", (PyCFunction)PyFFmpeg::Open, METH_VARARGS, "" },
		{ "read", (PyCFunction)PyFFmpeg::Read, METH_NOARGS, "" },
		{ "decode", (PyCFunction)PyFFmpeg::Decode, METH_NOARGS, "" },
		{ NULL }
	};
	type.tp_methods = ffmeth;

	static PyGetSetDef sets[] = {
		{ "totalms", (getter)PyFFmpeg::GetTotalms, 0, 0,0 },
		{ NULL }
	};
	type.tp_getset = sets;

	// 初始化类型
	if (PyType_Ready(&type) < 0) {
		return NULL;
	}

	PyModule_AddObject(m, "PyFFmpeg", (PyObject*)&type);

	printf("Pyinit_pyffmpeg\n");
	return m;
}

三、python调用解码函数

pyqt.py调用解码函数

#主函数 在子线程中调用，线程是c++创建
def main():
    print("Python main")
    global ff;
    while isRunning:
        #print(ff.read())
        re = ff.read()
        if re:
            print(".", end='', flush = True) #flush输出缓冲
            print(ff.decode(), end='', flush = True)
            time.sleep(0.02)
        else:
            time.sleep(1)

运行：

可以看到一开始的几帧解码失败，这是正常的，后面就解码成功了。

在这里插入图片描述

四、总结

本文完成python调用扩展库pyffmpeg的decode解码函数。
如果觉得文章对你有用处，记得 点赞 收藏 转发 一波哦，博主也支持为铁粉丝制作专属动态壁纸哦~

💬 往期优质文章分享

🚀 优质教程分享 🚀

🎄如果感觉文章看完了不过瘾，可以来我的其他专栏看一下哦~
🎄比如以下几个专栏：Python实战微信订餐小程序、Python量化交易实战、C++ QT实战类项目和算法学习专栏
🎄可以学习更多的关于C++/Python的相关内容哦！直接点击下面颜色字体就可以跳转啦！

学习路线指引（点击解锁）	知识定位	人群定位
🧡 Python实战微信订餐小程序 🧡	进阶级	本课程是python flask+微信小程序的完美结合，从项目搭建到腾讯云部署上线，打造一个全栈订餐系统。
💛Python量化交易实战 💛	入门级	手把手带你打造一个易扩展、更安全、效率更高的量化交易系统
❤️ C++ QT结合FFmpeg实战开发视频播放器❤️	难度偏高	分享学习QT成品的视频播放器源码，需要有扎实的C++知识！
💚 游戏爱好者九万人社区💚	互助/吹水	九万人游戏爱好者社区，聊天互助，白嫖奖品
💙 Python零基础到入门 💙	Python初学者	针对没有经过系统学习的小伙伴，核心目的就是让我们能够快速学习Python的知识以达到入门

🚀 资料白嫖，温馨提示 🚀

关注下面卡片即刻获取更多编程知识，包括各种语言学习资料，上千套PPT模板和各种游戏源码素材等等资料。更多内容可自行查看哦！

请添加图片描述

虚坏叔叔

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
打赏
0
评论
Python&C++相互混合调用编程全面实战-32完成python调用扩展库pyffmpeg的decode解码函数

作者：虚坏叔叔博客：https://xuhss.com添加函数和成员变量完整代码如下：添加代码：包括avframe申请以及帧数据的清理完整代码如下：二、开放解码器调用接口开放接口添加接口的调用和注册：三、python调用解码函数调用解码函数运行：可以看到一开始的几帧解码失败，这是正常的，后面就解码成功了。关注下面卡片即刻获取更多编程知识，包括各种语言学习资料，上千套PPT模板和各种游戏源码素材等等资料。更多内容可自行查看哦！...
复制链接

扫一扫