myquery.cpp

该程序实现FIRE足球机器人竞赛中的3：3比赛源码

// MyQuery.cpp: implementation of the CMyQuery class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "MyQuery.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CMyQueue::CMyQueue()
{

}

CMyQueue::~CMyQueue()
{
	// 清空队列
	ClearQueue();
	if (front != NULL) {
		delete front;
		front = NULL;
	}
	if (rear != NULL) {
		delete rear;
		rear = NULL;
	}
}
/*************************************************************
// 构造队列
// iLength：队列的长度
// DataSize：数据内存区域的大小
// nBitCount：颜色的位数
*************************************************************/
CMyQueue::CMyQueue(int iLength, CSize DataSize, int nBitCount)//50,640x480,8
{
	if (iLength <= 0) {
		m_QueryLength = 0;
		m_nDataSum = 0;
		rear = front = NULL;
		return;
	}
	m_QueryLength = iLength;
	m_nDataSum = 0;
	m_DataSize.cx = DataSize.cx;
	m_DataSize.cy = DataSize.cy;
	m_nBitCount = nBitCount;

	front = new( ImageData );
	rear = new( ImageData );
	
	ImageData* s;
	ImageData* r = front;

	int nSize = m_DataSize.cx*m_DataSize.cy*(m_nBitCount>>3);

	for(int i=0; i<iLength; i++)
	{
		s = new(ImageData);
		s->pData = new byte[nSize];
		memset(s->pData, 0, nSize);//内存数据初始化为0
		s->DataSize.cx = m_DataSize.cx;
		s->DataSize.cy = m_DataSize.cy;
		s->nBitCount = m_nBitCount;		
		r->next = s;
		r = s;
	}
	//将末尾节点的NEXT指针指向头指针指向的节点
	r->next = front->next;
	//初始状态尾指针和头指针指向同一个节点
	rear->next = front->next;
}

//清空队列
void CMyQueue::ClearQueue()
{
	ImageData* p = NULL;
	for(int i=0; i<m_QueryLength; i++)
	{
		// 将p指向头节点指向节点的下一个节点，然后删除头节点指向的节点
		p = front->next;
		front->next = p->next;

		// 删除图像内存
		delete p->pData;
		p->pData = NULL;

		//删除节点内存
		delete p;
		p = NULL;
	}
}

//取头指针指向的节点
ImageData* CMyQueue::GetFront()
{
	// 如果图像数据节点为0
	if(m_nDataSum == 0)
	{
		return NULL;
	}
	else
	{
		return front->next;
	}
}

//新增一个节点，插入队尾，队列的总长度+1
void CMyQueue::AddImageDataNode()
{
	m_QueryLength++;
	ImageData* s = new(ImageData);
	s->pData = new BYTE[m_DataSize.cx*m_DataSize.cy*(m_nBitCount>>3)];
	s->DataSize.cx = m_DataSize.cx;
	s->DataSize.cy = m_DataSize.cy;
	s->nBitCount = m_nBitCount;
	
	s->next = rear->next->next;
	rear->next->next = s;
	rear->next = s;
}

//进队列
bool CMyQueue::InQueue(PBYTE pData)
{
	// 如果尾指针指向的节点与头指针指向的节点相同，说明循环链队列已满
	// 这时不用再添加节点
	if (rear->next == front->next && m_nDataSum > 0) { 
	//或者++m_nDataSum >= m_QueryLength
	//if(++m_nDataSum >= m_QueryLength){
		return ERROR;
	}

	memcpy(rear->next->pData, pData, m_DataSize.cx*m_DataSize.cy*(m_nBitCount>>3));
	rear->next->DataSize.cx = m_DataSize.cx;
	rear->next->DataSize.cy = m_DataSize.cy;
	rear->next->nBitCount = m_nBitCount;
	
	rear->next = rear->next->next;
	m_nDataSum++;
	
	return OK;
}

//出队列，返回图像的数据内存
PBYTE CMyQueue::OutQueue()
{
	//表示队列里无存储数据
	if(m_nDataSum == 0)
	{
		return NULL;
	}

	m_nDataSum--;

	//memset(front->next->pData, 0, m_DataSize.cx*m_DataSize.cy*(m_nBitCount>>3));

	ImageData* s = front->next;	
	front->next = s->next;

	return s->pData;
}