文章目录
显示
作者:虚坏叔叔
博客: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解码函数。
- 如果觉得文章对你有用处,记得
点赞
收藏
转发
一波哦,博主也支持为铁粉丝制作专属动态壁纸哦~
? 往期优质文章分享
- C++ QT结合FFmpeg实战开发视频播放器-01环境的安装和项目部署
- 解决QT问题:运行qmake:Project ERROR: Cannot run compiler ‘cl‘. Output:
- 解决安装QT后MSVC2015 64bit配置无编译器和调试器问题
- Qt中的套件提示no complier set in kit和no debugger,出现黄色感叹号问题解决(MSVC2017)
- Python+selenium 自动化 - 实现自动导入、上传外部文件(不弹出windows窗口)
? 优质教程分享 ?
- ?如果感觉文章看完了不过瘾,可以来我的其他 专栏 看一下哦~
- ?比如以下几个专栏:Python实战微信订餐小程序、Python量化交易实战、C++ QT实战类项目 和 算法学习专栏
- ?可以学习更多的关于C++/Python的相关内容哦!直接点击下面颜色字体就可以跳转啦!
学习路线指引(点击解锁) | 知识定位 | 人群定位 |
---|---|---|
? Python实战微信订餐小程序 ? | 进阶级 | 本课程是python flask+微信小程序的完美结合,从项目搭建到腾讯云部署上线,打造一个全栈订餐系统。 |
?Python量化交易实战 ? | 入门级 | 手把手带你打造一个易扩展、更安全、效率更高的量化交易系统 |
❤️ C++ QT结合FFmpeg实战开发视频播放器❤️ | 难度偏高 | 分享学习QT成品的视频播放器源码,需要有扎实的C++知识! |
? 游戏爱好者九万人社区? | 互助/吹水 | 九万人游戏爱好者社区,聊天互助,白嫖奖品 |
? Python零基础到入门 ? | Python初学者 | 针对没有经过系统学习的小伙伴,核心目的就是让我们能够快速学习Python的知识以达到入门 |
? 资料白嫖,温馨提示 ?
关注下面卡片即刻获取更多编程知识,包括各种语言学习资料,上千套PPT模板和各种游戏源码素材等等资料。更多内容可自行查看哦!
转载请注明:xuhss » Python&C++相互混合调用编程全面实战-32完成python调用扩展库pyffmpeg的decode解码函数