音频流采样器类的实现【6】

.h文件

#pragma once

#include <mutex>
#include <iostream>

extern "C"
{

#include "libavcodec/avcodec.h"//解码
#include "libavutil/avutil.h"//工具
#include "libswresample/swresample.h"//重采样
}

#pragma comment(lib,"avutil.lib")//工具库，包含打印错误信息函数
#pragma comment(lib,"avcodec.lib")//解码
#pragma comment(lib,"swresample.lib")//重采样

class XResample
{
public:
	XResample();
	virtual ~XResample();

	//输入和输出参数一致，除了采样格式，输出位S16，qt声卡可以直接播放,释放param
	virtual bool Open(AVCodecParameters* param, bool isNeedClearParam = true);
	
	//
	virtual void Close();

	//返回重采样后的大小,不管成功与否，都需要释放indata的空间
	virtual int Resample(AVFrame* indata, unsigned char* data);

	//输出格式，可以更换
	int outFormat = 1; //AV_SAMPLE_FMT_S16

protected:
	std::mutex mux;

	SwrContext* actx = NULL;//音频格式上下文

};

.cpp文件

#include "XResample.h"

XResample::XResample()
{

}
XResample::~XResample()
{

	this->Close();

}

//*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-//

//输入和输出参数一致，除了采样格式，输出位S16，qt声卡可以直接播放
bool XResample::Open(AVCodecParameters* param, bool isNeedClearParam)
{
	if (param == NULL)
		return false;

	this->Close();

	mux.lock();//上锁

	//actx = swr_alloc();//音频格式上下文,如果actx = NULL，下面的函数会自动分配内存

	//分配内存
	actx = swr_alloc_set_opts(actx,
		av_get_default_channel_layout(param->channels),	//输出双声道,在这里设置输出的通道数，尽量保持一致
		(AVSampleFormat)outFormat,			//AV_SAMPLE_FMT_S16,//输出样本格式
		param->sample_rate,					//输出采样率，1s中的样本数量，保持不变

		av_get_default_channel_layout(param->channels),//输入的音频通道格式
		(AVSampleFormat)param->format,//输入样本格式
		param->sample_rate,					//输入采样率
		0,
		0);

	if (isNeedClearParam)
	{
		avcodec_parameters_free(&param);//如果确定需要清理参数，再清理
	}
	
	int re = swr_init(actx);

	mux.unlock();

	if (re != 0)//0为成功打开文件
	{
		
		char buf[1024] = { 0 };
		av_strerror(re, buf, sizeof(buf) - 1);//打印错误原因
		std::cout << " avcodec_send_packet--" << "--failed! : " << buf << std::endl;
		getchar();
		return -1;
	}

	return true;

}

//*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-//

void XResample::Close()
{
	mux.lock();
	if (actx)
		swr_free(&actx);//释放采样器上下文

	mux.unlock();
}

//*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-//

//返回重采样后的大小,释放frame的空间和采样器上下文的空间
int XResample::Resample(AVFrame* indata, unsigned char* outdata)
{
	if (!indata )//输入不存在
		return 0;

	if (!outdata)//输出不存在
	{
		av_frame_free(&indata);//返回之前释放indatad的空间
		return 0;
	}

	uint8_t* data[2] = { 0 };
	data[0] = outdata;
	int re = swr_convert(actx,
		data,//输出的数据
		indata->nb_samples,//输出的样本数量

		(const uint8_t**)indata->data,//输入的数据
		indata->nb_samples//输入的样本数量
	);


	if (re <= 0)
	{
		av_frame_free(&indata);//释放indatad的空间
		return re;
	}
		

	int outSize =  re * indata->channels * av_get_bytes_per_sample((AVSampleFormat)outFormat);

	av_frame_free(&indata);//释放indatad的空间

	return outSize;
}
//*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-//