PhysicalDevice/Machine/MachineryECOM/Trunk/DIOS.Dev.Machinery.Motion.Driver/DIOS.Dev.Machinery.Client/deliver/include/MsgVector.h

#pragma once
#include "DiosLock.h"

#include <vector>
#include <algorithm>


using namespace std;

template<typename T> class MsgVector
{
	DiosLock m_Lock;
	HANDLE m_NotifyInVec;
	vector<T> m_DataVector;

public:


	MsgVector(void)
	{
		m_NotifyInVec = CreateEvent(0, 1, 0, 0);
	};


	~MsgVector(void)
	{
		Clear();

		CloseHandle(m_NotifyInVec);
	};
	
	//suppose it locked already...
	T &operator [](size_t idx)
	{
		return m_DataVector[idx];
	};
	
	//suppose it locked already...
	T &operator [](int idx)
	{
		return m_DataVector[idx];
	};

	void Lock()
	{
		m_Lock.Thread_Lock();
	};
	
	void UnLock()
	{
		m_Lock.Thread_UnLock();
	};
	
	size_t Size()
	{
		Lock();

		size_t size = m_DataVector.size();

		UnLock();

		return size;
	};

	bool Clear()
	{
		ResetEvent(m_NotifyInVec);
		Lock();

		m_DataVector.clear();

		UnLock();

		return true;
	};

	bool PopBack(T &rMsg)
	{
		bool ret = false;
		Lock();

		//DebugPrint("before deQue size:%d\n",m_DataVector.size());
		if (m_DataVector.size() > 0)
		{
			rMsg = m_DataVector.pop_back();

			ret = true;
		}
		//DebugPrint("after deQue size:%d\n",m_DataVector.size());
		if (m_DataVector.size() == 0)
		{
			ResetEvent(m_NotifyInVec);
		}

		UnLock();

		return ret;
	};

	bool FindAndClear(T &rMsg)
	{
		bool ret = false;
		Lock();
		vector<T>::iterator result1 = find(m_DataVector.begin(), m_DataVector.end(), rMsg);
		if (result1 != m_DataVector.end())
		{
			m_DataVector.erase(result1);
			ret = true;
		}
		if (m_DataVector.size() == 0)
		{
			ResetEvent(m_NotifyInVec);
		}
		UnLock();

		return ret;

	};

	bool PushBack(T &rMsg)
	{
		Lock();
		m_DataVector.push_back(rMsg);
		UnLock();

		SetEvent(m_NotifyInVec);

		return true;
	};

	DWORD WaitForInVec(DWORD timeout)
	{
		DWORD ret = WAIT_TIMEOUT;
		Lock();

		if (m_DataVector.size() > 0)
		{
			ret = WAIT_OBJECT_0;
		}

		UnLock();

		if (ret == WAIT_TIMEOUT)
		{
			return WaitForSingleObject(m_NotifyInVec, timeout);
		}

		return ret;

	};

};