boost shared_ptr 及C++内存管理的一些总结

先从实际问题开始讨论, 有个视频采集线程,采集到一帧视频数据, 这帧数据要给存储模块(线程)、RTSP模块,RTMP模块使用。 像这样的情况,内存的申请是在采集线程, 但释放就不能是在采集线程了,所以要实现,谁最后使用,谁释放。如图所示:

boost shared_ptr 及C++内存管理的一些总结_第1张图片

 

需求明确,但是代码怎么实现呢? 答案是用到智能指针实现!

智能指针又是怎么实现呢? 智能指针使用引用技术实现, 当指针传递时,引用加1,当指针使用结束,引用减1,如果引用数等于0,则析构对象,释放内存 。

 在上面的例子中,采集模块new出来对象,对象的引用计数为1,对象的指针传给存储模块时,引用计数加1;传给RTSP模块时,引用计数加1;传给RTMP模块时,引用计数加1, 如果几个模块都没有使用结束,此时对象的引用计数为4。采集模块使用结束,引用减1;存储模块使用结束,引用减1;RTSP模块使用结束,引用减1;RRTMP模块使用结束,引用减1, 此时引用计数为0, 析构对象,释放内存。

引用计数的智能指针的具体实现请看:

引用计数的智能指针的实现与思考

 

智能指针在boost里有现成的, 也就是shared_ptr。 怎么使用boost:: shared_ptr来实现我们想要的功能呢? 直接看代码可能更容易理解:

 AVData.h

#ifndef AV_DATA__H
#define AV_DATA__H

#include 
#include 
#include 
#include 
#include 
#include 

typedef enum _AVDataType
{
	AVDataType_Audio = 1,
	AVDataType_Video,
	AVDataType_Both
}AVDataType;

class AVData
{
public:
	AVData();
	AVData(int i);
	~AVData();

	void TestFunc();
	int	 SetAVData(AVDataType type, const char* buffer, const int size);

	char* GetAVData();
	int	  GetAVDataSize();

private:
	char* m_pBuffer;
	int	  m_iBufferSize;

	int	  m_iCount;
};

typedef boost::shared_ptr AVDataPtr;

#endif // AV_DATA__H
AVDataPool.h

#ifndef AV_DATA_POOL__H
#define AV_DATA_POOL__H

#include "AVData.h"
#include 

typedef std::list AVDataList;

class AVDataPool
{
public:
	AVDataPool();
	AVDataPool(int max);
	~AVDataPool();

	void PutAVDataPool(AVDataPtr ptr);
	void GetAVDataPool(AVDataPtr &ptr);

	void Flush();

private:
	AVDataList m_List;
	boost::mutex m_Mutex;
	int	  m_iMaxCount;
};

#endif // AV_DATA_POOL__H
main
#include 
#include 
#include 
#include 
#include 
#include 

#include 
#include 
#include 
#include 

#include "AVData.h"
#include "AVDataPool.h"

void my_handler(int sig);
void ThreadStoreVideo(void *p);
void ThreadRtsp(void *p);
void ThreadRtmp(void *p);


int global_value_exit = 0;

class ThreadParams
{
public:
	ThreadParams()
		:m_pAVDataPool(NULL)
	{

	}

	~ThreadParams()
	{
		if (m_pAVDataPool)
		{
			delete m_pAVDataPool;
			m_pAVDataPool = NULL;
		}
	}

	void InitThreadParams(int count)
	{
		m_pAVDataPool = new AVDataPool(count);
	}

public:
	AVDataPool		*m_pAVDataPool;
};

int main(int argc, char* argv[])
{
	printf("shared_ptr test begin\n");

	struct sigaction sigIntHandler; // 接收Ctrl + C 结束
	sigIntHandler.sa_handler = my_handler;
	sigemptyset(&sigIntHandler.sa_mask);
	sigIntHandler.sa_flags = 0;
	sigaction(SIGINT, &sigIntHandler, NULL);

	ThreadParams *store_params = new ThreadParams();
	store_params->InitThreadParams(3);

	ThreadParams *rtsp_params = new ThreadParams();
	rtsp_params->InitThreadParams(4);

	ThreadParams *rtmp_params = new ThreadParams();
	rtmp_params->InitThreadParams(5);

	boost::thread thread_store(ThreadStoreVideo, store_params);
	boost::thread thread_rtsp(ThreadRtsp, rtsp_params);
	boost::thread thread_rtmp(ThreadRtmp, rtmp_params);

	int i = 0;
	char *buffer = (char*)malloc(4096);
	int size;
	char data = 'a';
	while(!global_value_exit)
	{
		if (data > 'z')
		{
			data = 'a';
		}

		AVDataPtr pAVData(new AVData(i++));
		size = 1024 + rand()%1024; // 生成随机数 1024~2048
		memset(buffer, data, size);

		pAVData->SetAVData(AVDataType_Both, buffer, size);

		data++;

		store_params->m_pAVDataPool->PutAVDataPool(pAVData);
		rtsp_params->m_pAVDataPool->PutAVDataPool(pAVData);
		rtmp_params->m_pAVDataPool->PutAVDataPool(pAVData);

		usleep(200000);
	}

	thread_store.join();
	thread_rtsp.join();
	thread_rtmp.join();

	if (buffer)
	{
		free(buffer);
		buffer = NULL;
	}
	return 0;
}

void ThreadStoreVideo(void *p)
{
	ThreadParams *params = (ThreadParams*)p;

	while(!global_value_exit)
	{
		AVDataPtr ptr;
		params->m_pAVDataPool->GetAVDataPool(ptr);

		if (ptr)
		{
			//ptr->TestFunc();
			char *buffer = ptr->GetAVData();
			printf("%s : %c\n", __func__, *buffer);
		}

		usleep(15000);
	}
}

void ThreadRtsp(void *p)
{
	ThreadParams *params = (ThreadParams*)p;

	while(!global_value_exit)
	{
		AVDataPtr ptr;
		params->m_pAVDataPool->GetAVDataPool(ptr);

		if (ptr)
		{
			//ptr->TestFunc();
			char *buffer = ptr->GetAVData();
			printf("%s : %c\n", __func__, *buffer);
		}

		usleep(15000);
	}
}

void ThreadRtmp(void *p)
{
	ThreadParams *params = (ThreadParams*)p;

	while(!global_value_exit)
	{
		AVDataPtr ptr;
		params->m_pAVDataPool->GetAVDataPool(ptr);

		if (ptr)
		{
			//ptr->TestFunc();
			char *buffer = ptr->GetAVData();
			printf("%s : %c\n", __func__, *buffer);
		}

		usleep(15000);
	}
}

void my_handler(int sig)
{
	if (sig == SIGINT)
	{
		printf("Caught signal %d\n", sig);
		global_value_exit = 1;
	}
}

 

以上就是C++内存管理的一种方式, 当然C++内存管理方式多种多样。 还可以在一开始的时候,申请一大块内存,  然后再通过修改 uint8_t *rptr, *wptr来改变读写内存的位置, 来实现数据的写入和读出。如果代码可以参考ffmpeg  libavutil/fifo.c,实现原理如下图所示:

 boost shared_ptr 及C++内存管理的一些总结_第2张图片

哪种实现性能更优呢?  第二种方法看似更好一些, 一次申请了一大块内存, 不用跟系统频繁申请释放内存, 但是别忘了第二种方法要加锁。没做过性能测试, 不知道哪种方法更好些,  目前开发的系统使用的是第一种方法。 欢迎做过比较测试的朋友告诉我哈

 

 

 

你可能感兴趣的:(C/C++笔记,流媒体)