完成扩展库pyffmpeg的尺寸和像素格式转换上下文初始化

在这里插入图片描述

一、定义上下文并支持解码函数宽高修改

XFFmepg.h添加成员变量尺寸格式转换上下文

	// 尺寸和像素格式转换上下文 yuv->rgb
	SwsContext *sws = 0;

对应的解码函数支持修改宽高

	bool Decode(int outwidth, int outheight);

完整代码如下：

#pragma once
struct AVFormatContext;
struct AVPacket;
struct AVCodecContext;
struct AVFrame;
struct SwsContext;
class XFFmpeg
{
public:
	// 打开视频
	bool Open(const char *url);
	// 读取一阵视频 在内部存储
	bool Read();
	bool Decode(int outwidth, int outheight);
	void Close();
	XFFmpeg();
	~XFFmpeg();
	int totalms = 0;
protected:
	// 解封装上下文
	AVFormatContext *ic = 0;
	// 读取视频帧
	AVPacket *pkt = 0;
	// 解码器上下文
	AVCodecContext *vcodec = 0;
	// 解码后的yuv帧
	AVFrame *frame = 0;
	// 尺寸和像素格式转换上下文 yuv->rgb
	SwsContext *sws = 0;
};

XFFmepg.cpp添加头文件包含。因为要引入修改视频宽高的api

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

库文件添加：

avformat.lib
avutil.lib
avcodec.lib
swscale.lib

在这里插入图片描述

对应的解码函数的实现完整代码如下：

bool XFFmpeg::Decode(int outwidth, int outheight) {
	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;

	// 像素格式和尺寸转换上下文
	// 将yuv转为rgb 并调整尺寸
	sws = sws_getCachedContext(sws,
		vcodec->width, vcodec->height, vcodec->pix_fmt, //原宽高/像素格式（YUV420p）
		outwidth, outheight, AV_PIX_FMT_BGRA, //目标宽高/像素格式（RGB）
		SWS_BICUBIC, // 转换的算法
		0, 0, 0
	);
	if (!sws)
	{
		// 出错一般都是参数格式有问题
		printf("sws_getCachedContext failed\n");
	}

	return true;
}

二、修改导出接口，支持python调用

由于添加了2个参数，PyFFmpeg.cpp修改导出的注册

PyObject *PyFFmpeg::Decode(PyFFmpeg*self, PyObject*args)
{
	if (!self->ff)
		Py_RETURN_FALSE;
	int w = 0;
	int h = 0;
	if (!PyArg_ParseTuple(args, "ii", &w, &h))
	{
		Py_RETURN_FALSE;
	}
	if (self->ff->Decode(w, h))
		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_VARARGS, "" },
		{ 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中添加宽高的调用：

isRunning = True
#输出的宽高，根据窗口尺寸
winWidth = 1280
winHeight = 720

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

运行：

在这里插入图片描述

只能保证没有出错。现在还看不到效果。

四、总结

本文完成扩展库pyffmpeg的尺寸和像素格式转换上下文初始化。
如果觉得文章对你有用处，记得 点赞 收藏 转发 一波哦，博主也支持为铁粉丝制作专属动态壁纸哦~

💬 往期优质文章分享

🚀 优质教程分享 🚀

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

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