myvtm/Other/libaudioqueue/libaudioqueue.cpp

// libaudioqueue.cpp : 定义 DLL 应用程序的导出函数。
//

#include "stdafx.h"
#include "libaudioqueue.h"
#include "libsharememory.h"
#include <windows.h>


typedef struct  Qnode
{
	unsigned int videoframeindex;
	unsigned int nextqnodeindex;
}Qnode, *queueptr;

typedef struct linkqueue
{
	unsigned int frontindex;
	unsigned int rearindex;
	unsigned int queuelens;
}linkqueue;

class libaudioqueue_impl
{
private:
		Clibsharememory m_ShareMem;
		LPVOID m_lpMem;
		linkqueue*m_pQueue;
		int m_nQueueAddrLens;
		int m_nQnodeAddrLens;
		int m_nFrameAddrLens;
		int m_nDataAddrlens;
		LPCTSTR szShareMemName;
		unsigned long aQnodeAddr[MAX_AUDIOQUEUE_LENS];
		unsigned long aAudioFrameAddr[MAX_AUDIOQUEUE_LENS];
		unsigned long aAudioDataAddr[MAX_AUDIOQUEUE_LENS];
public:
	libaudioqueue_impl(LPCTSTR audioqueuename,int framesize=MAX_AUDIOQNODE_SIZE)
	{
		m_lpMem = NULL;
		m_pQueue = NULL;
		szShareMemName = NULL;
		m_nQueueAddrLens = 0;
		m_nQnodeAddrLens = 0;
		m_nFrameAddrLens = 0;
		m_nDataAddrlens = 0;
		for(int i=0;i<MAX_AUDIOQUEUE_LENS;i++)
		{
			aQnodeAddr[i] = 0;
			aAudioFrameAddr[i] = 0;
			aAudioDataAddr[i] = 0;
		}
		InitQueue(audioqueuename,framesize);
	}

	~libaudioqueue_impl()
	{
		ClearAudioQueue();
	}
	//初始化队列
	BOOL InitQueue(LPCTSTR szName,int framesize=MAX_AUDIOQNODE_SIZE)
	{
		m_nQueueAddrLens = sizeof(linkqueue);
		m_nQnodeAddrLens = MAX_AUDIOQUEUE_LENS*sizeof(Qnode);
		m_nFrameAddrLens = MAX_AUDIOQUEUE_LENS*sizeof(audio_frame);
		m_nDataAddrlens	= MAX_AUDIOQUEUE_LENS*framesize;
		int nMemTotalNum = m_nQueueAddrLens+m_nQnodeAddrLens+m_nFrameAddrLens+m_nDataAddrlens;
		if (m_ShareMem.Create(szName,nMemTotalNum)&&(m_nDataAddrlens!=0))
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				memset(m_lpMem,0,nMemTotalNum);
				m_pQueue = (linkqueue *)m_lpMem;
				for(int i =0;i<MAX_AUDIOQUEUE_LENS;i++)
				{
					aQnodeAddr[i] = (unsigned long)m_pQueue+m_nQueueAddrLens+sizeof(Qnode)*i;
					aAudioFrameAddr[i] = (unsigned long)m_pQueue+m_nQueueAddrLens+m_nQnodeAddrLens+sizeof(audio_frame)*i;
					aAudioDataAddr[i] = (unsigned long)m_pQueue+m_nQueueAddrLens+m_nQnodeAddrLens+m_nFrameAddrLens+framesize*i;
				}
				m_pQueue->frontindex = m_pQueue->rearindex = 0;
				m_pQueue->queuelens = 0;
				m_ShareMem.Unlock();
			}

		} 
		else if(m_ShareMem.Open(szName))
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				m_pQueue = (linkqueue *)m_lpMem;
				for(int i =0;i<MAX_AUDIOQUEUE_LENS;i++)
				{
					aQnodeAddr[i] = (unsigned long)m_pQueue+m_nQueueAddrLens+sizeof(Qnode)*i;
					aAudioFrameAddr[i] = (unsigned long)m_pQueue+m_nQueueAddrLens+m_nQnodeAddrLens+sizeof(audio_frame)*i;
					aAudioDataAddr[i] = (unsigned long)m_pQueue+m_nQueueAddrLens+m_nQnodeAddrLens+m_nFrameAddrLens+framesize*i;
				}
				m_ShareMem.Unlock();
			}
		}
		return TRUE;
	}
	//返回队列的元素个数，音频队列长度
	int GetAudioLens()
	{
		if(m_ShareMem.IsValid())
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				int num = m_pQueue->queuelens;
				m_ShareMem.Unlock();
				return num;
			}
			else
			{
				return 0;
			}
		}
		else
		{
			return 0;
		}
	}
	//往音频循环队列尾部插节点
	BOOL InsertAudio(audio_frame* Audio)
	{
		if(m_ShareMem.IsValid())
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				unsigned int nRearNextIndex = 0;
				//保存当前对位指针的序列号
				queueptr rearptrfront = (queueptr)aQnodeAddr[m_pQueue->rearindex];
				//如果队列已满
				if(m_pQueue->queuelens == MAX_AUDIOQUEUE_LENS)
				{
					m_pQueue->rearindex = (m_pQueue->rearindex+1)%MAX_AUDIOQUEUE_LENS;
					m_pQueue->frontindex = (m_pQueue->frontindex+1)%MAX_AUDIOQUEUE_LENS;
					m_pQueue->queuelens = MAX_AUDIOQUEUE_LENS;
				}
				else if (m_pQueue->queuelens == 0)
				{
					m_pQueue->rearindex  = 0;
					m_pQueue->frontindex  = 0;
					m_pQueue->queuelens++;
				}
				else
				{
					m_pQueue->rearindex = (m_pQueue->rearindex+1)%MAX_AUDIOQUEUE_LENS;
					m_pQueue->frontindex = m_pQueue->frontindex;
					m_pQueue->queuelens++;
				}
				if (Audio!=NULL)
				{
					queueptr rearqnodetmp			= (queueptr)aQnodeAddr[m_pQueue->rearindex];
					rearqnodetmp->videoframeindex	= m_pQueue->rearindex;
					rearqnodetmp->nextqnodeindex	= 0;
					audio_frame*audiotmp			= (audio_frame*)aAudioFrameAddr[m_pQueue->rearindex];
					audiotmp->data					= (char*)aAudioDataAddr[m_pQueue->rearindex ];
					audiotmp->bitspersample			= Audio->bitspersample;
					audiotmp->format				= Audio->format;
					audiotmp->framesize				= Audio->framesize;
					audiotmp->nchannels				= Audio->nchannels;
					audiotmp->samplespersec			= Audio->samplespersec;
					audiotmp->iseriesnumber			= Audio->iseriesnumber;
					memcpy(audiotmp->data,Audio->data,Audio->framesize);
					rearptrfront->nextqnodeindex = m_pQueue->rearindex;
					//delete Audio->data;
					//delete Audio;
				}
				m_ShareMem.Unlock();
				return TRUE;
			}
			else
			{
				return FALSE;
			}

		}
		else
		{
			return FALSE;
		}
	}
	//读音频队列头部节点
	BOOL GetAudio(audio_frame* Audio)	
	{
		if(m_ShareMem.IsValid())
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				if (m_pQueue->queuelens == 0)
				{
					m_ShareMem.Unlock();
					return FALSE;
				}
				else
				{
					audio_frame*audiotemp = (audio_frame*)aAudioFrameAddr[m_pQueue->frontindex];;
					Audio->format			= audiotemp->format;
					Audio->framesize		= audiotemp->framesize;
					Audio->bitspersample	= audiotemp->bitspersample;
					Audio->nchannels		= audiotemp->nchannels;
					Audio->samplespersec	= audiotemp->samplespersec;
					Audio->iseriesnumber	= audiotemp->iseriesnumber;
					memcpy(Audio->data,(char*)aAudioDataAddr[m_pQueue->frontindex],audiotemp->framesize);
					m_ShareMem.Unlock();
					return TRUE;
				}
			}
			else
			{
				return FALSE;
			}
		}
		else
		{
			return FALSE;
		}		
	}
	//读视频队列头部节点,并删除头部节点
	BOOL GetAudioAndDel(audio_frame* Audio)
	{
		if(m_ShareMem.IsValid())
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				if (m_pQueue->queuelens == 0)
				{
					m_ShareMem.Unlock();
					return FALSE;
				}
				else
				{
					audio_frame*audiotemp = (audio_frame*)aAudioFrameAddr[m_pQueue->frontindex];
					Audio->format			= audiotemp->format;
					Audio->framesize		= audiotemp->framesize;
					Audio->bitspersample	= audiotemp->bitspersample;
					Audio->nchannels		= audiotemp->nchannels;
					Audio->samplespersec	= audiotemp->samplespersec;
					Audio->iseriesnumber    = audiotemp->iseriesnumber;
					memcpy(Audio->data,(char*)aAudioDataAddr[m_pQueue->frontindex],audiotemp->framesize);
					char*data = (char*)aAudioDataAddr[m_pQueue->frontindex];
					memset(data,0,audiotemp->framesize);
					memset(audiotemp,0,sizeof(audio_frame));
					queueptr qnodetmp			= (queueptr)aQnodeAddr[m_pQueue->frontindex];
					m_pQueue->frontindex		= qnodetmp->nextqnodeindex;
					qnodetmp					= (queueptr)aQnodeAddr[m_pQueue->rearindex];
					qnodetmp->nextqnodeindex	= 0;
					m_pQueue->queuelens--;
					m_ShareMem.Unlock();
					return TRUE;
				}
			}
			else
			{
				return FALSE;
			}
		}
		else
		{
			return FALSE;
		}		
	}
	//清除队列
	BOOL ClearAudioQueue()
	{
		if(m_ShareMem.IsValid())
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				if (m_pQueue->queuelens != 0)
				{
					while(m_pQueue->queuelens != 0)
					{
						audio_frame*audiotemp = (audio_frame*)aAudioFrameAddr[m_pQueue->frontindex];
						char*data = (char*)aAudioDataAddr[m_pQueue->frontindex];
						memset(data,0,audiotemp->framesize);
						memset(audiotemp,0,sizeof(audio_frame));
						queueptr qnodetmp			= (queueptr)aQnodeAddr[m_pQueue->frontindex];
						m_pQueue->frontindex		= qnodetmp->nextqnodeindex;
						qnodetmp					= (queueptr)aQnodeAddr[m_pQueue->rearindex];
						qnodetmp->nextqnodeindex	= 0;
						m_pQueue->queuelens--;	
					}					
				}
				m_ShareMem.Unlock();
				return TRUE;
			}
			else
			{
				return FALSE;
			}
		}
		else
		{
			return FALSE;
		}		
	}
	//删除队头的数据
	BOOL DeleteHeadAudio()  
	{
		if(m_ShareMem.IsValid())
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				if (m_pQueue->queuelens != 0)
				{	
					audio_frame*audiotemp = (audio_frame*)aAudioFrameAddr[m_pQueue->frontindex];
					char*data = (char*)aAudioDataAddr[m_pQueue->frontindex];
					memset(data,0,audiotemp->framesize);
					memset(audiotemp,0,sizeof(audio_frame));
					queueptr qnodetmp			= (queueptr)aQnodeAddr[m_pQueue->frontindex];
					m_pQueue->frontindex		= qnodetmp->nextqnodeindex;
					qnodetmp					= (queueptr)aQnodeAddr[m_pQueue->rearindex];
					qnodetmp->nextqnodeindex	= 0;
					m_pQueue->queuelens--;			
				}
				m_ShareMem.Unlock();
				return TRUE;
			}
			else
			{
				return FALSE;
			}
		}
		else
		{
			return FALSE;
		}	
	}
	//获取图像的大小
	int GetFrameSize()
	{
		if(m_ShareMem.IsValid())
		{
			m_lpMem = m_ShareMem.Lock(1000);
			if(m_lpMem != NULL)
			{
				int nFrameSize = 0;
				if (m_pQueue->queuelens != 0)
				{	
					audio_frame*audiotmp	= (audio_frame*)aAudioFrameAddr[m_pQueue->frontindex];
					nFrameSize				= audiotmp->framesize;
				}
				m_ShareMem.Unlock();
				return nFrameSize;
			}
			else
			{
				return 0;
			}
		}
		else
		{
			return 0;
		}
	}
};

// 这是已导出类的构造函数。
// 有关类定义的信息，请参阅 libaudioqueue.h
Clibaudioqueue::Clibaudioqueue(LPCTSTR audioqueuename,int framesize)
{
	m_pImpl = new libaudioqueue_impl(audioqueuename,framesize);
	return;
}

Clibaudioqueue::~Clibaudioqueue()
{
	ClearAudioQueue();
	delete m_pImpl;
	return;
}


BOOL Clibaudioqueue::InsertAudio(audio_frame* Audio)
{
	BOOL bRst = m_pImpl->InsertAudio(Audio);
	return bRst;
}

BOOL Clibaudioqueue::GetAudio(audio_frame* Audio)
{
	BOOL bRst = m_pImpl->GetAudio(Audio);
	return bRst;
}

BOOL Clibaudioqueue::GetAudioAndDel(audio_frame* Audio)
{
	BOOL bRst = m_pImpl->GetAudioAndDel(Audio);
	return bRst;
}

int Clibaudioqueue::GetAudioLens(void)
{
	int i = m_pImpl->GetAudioLens();
	return i; 
}

void Clibaudioqueue::ClearAudioQueue()
{
	m_pImpl->ClearAudioQueue();
	return; 
}

int Clibaudioqueue::GetFrameSize()
{
	int i = m_pImpl->GetFrameSize();
	return i; 
}	

void Clibaudioqueue::DeleteHeadAudio()
{
	m_pImpl->DeleteHeadAudio();
	return;
}