新增AVPacket Read();方法


MyFFmpeg.h文件:

#pragma once
extern "C"{
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>
}
#include <string>
#include <QMutex>
class MyFFmpeg
{
public:
	/*设置成为单件模式*/
	static MyFFmpeg *Get()
	{
		static MyFFmpeg ff;
		return &ff;
	}

	/**
	 *打开指定路径的视频文件,如果已有打开的视频文件则先关闭
	 */
	bool Open(const char *path);

	//关闭之前打开的视频文件
	void Close();

	//读取视频帧
	AVPacket Read();

	/*获取相关错误信息*/
	std::string GetError();

	/*类析构函数*/
	virtual ~MyFFmpeg();

	/*视频文件总的毫秒数*/
	int totalMs = 0;
protected:
	/*相关错误信息*/
	char errorbuf[1024];

	//应对多线程访问时的同步锁
	QMutex mutex;

	AVFormatContext *ac = NULL;

	/*设置成为单件模式,所以要把构造函数设置为私有*/
	MyFFmpeg();
};


MyFFmpeg.cpp文件:

#include "MyFFmpeg.h"
#pragma comment(lib,"avformat.lib")
#pragma comment(lib,"avutil.lib")
#pragma comment(lib,"avcodec.lib")
#pragma comment(lib,"swscale.lib")
bool MyFFmpeg::Open(const char *path){
	Close();
	mutex.lock();
	int re = avformat_open_input(&ac, path, 0, 0);
	if (re != 0){//打开文件失败
		mutex.unlock();
		av_strerror(re, errorbuf, sizeof(errorbuf));
		return false;
	}
	//得到视频总时长的毫秒数
	totalMs = ((ac->duration / AV_TIME_BASE)*1000);
	mutex.unlock();
	return true;
}
void MyFFmpeg::Close(){
	mutex.lock();
	if (ac) avformat_close_input(&ac);
	mutex.unlock();
}
std::string MyFFmpeg::GetError(){
	mutex.lock();
	std::string re = this->errorbuf;
	mutex.unlock();
	return re;
}

AVPacket MyFFmpeg::Read(){
	AVPacket pkt;
	memset(&pkt, 0, sizeof(AVPacket));
	mutex.lock();
	if (!ac){
		mutex.unlock();
		return pkt;
	}
	int err = av_read_frame(ac, &pkt);
	if (err != 0){//读取失败
		av_strerror(err, errorbuf, sizeof(errorbuf));
	}
	mutex.unlock();
	return pkt;
}

MyFFmpeg::MyFFmpeg()
{
	errorbuf[0] = '\0';
	av_register_all();
}


MyFFmpeg::~MyFFmpeg()
{
}

main.cpp调用:

#include "myplayer.h"
#include <QtWidgets/QApplication>
#include "MyFFmpeg.h"

int main(int argc, char *argv[])
{
	char path[1024] = "test.mp4";
	if (MyFFmpeg::Get()->Open(path)){
		printf("文件[%s]打开成功",path);
	}
	else
	{
		printf("\n文件[%s]打开失败;错误信息:%s", path,MyFFmpeg::Get()->GetError().c_str());
		getchar();
		return -1;
	}
	for (;;){
		AVPacket pkt = MyFFmpeg::Get()->Read();
		if (pkt.size == 0){
			break;
		}
		printf("pts = %lld\n",pkt.pts);
		//释放空间
		av_packet_unref(&pkt);
	}

	QApplication a(argc, argv);
	MyPlayer w;
	w.show();
	return a.exec();
}


运行结果:








更多推荐