123

CryEntitySystem/EntityIt.h

@@ -0,0 +1,63 @@
+// EntityIt.h: interface for the CEntityIt class.
+//
+//////////////////////////////////////////////////////////////////////
+
+#if !defined(AFX_ENTITYIT_H__95AE38A4_7F15_4069_A97A_2F3A1F06F670__INCLUDED_)
+#define AFX_ENTITYIT_H__95AE38A4_7F15_4069_A97A_2F3A1F06F670__INCLUDED_
+
+#if _MSC_VER > 1000
+#pragma once
+#endif // _MSC_VER > 1000
+
+class CEntity;
+struct IEntity;
+
+#include "EntitySystem.h"
+
+class CEntityItMap : public IEntityIt
+{
+public:
+	CEntityItMap(EntityMap *pMap)
+	{
+		m_nRefCount=0;
+		m_pEntityMap = pMap;
+		MoveFirst();
+	};
+
+	bool IsEnd() { return (m_itEntityMap == m_pEntityMap->end()); };
+	IEntity * Next() { return IsEnd() ? NULL : (IEntity *) (* m_itEntityMap++).second; };
+	void MoveFirst() { m_itEntityMap = m_pEntityMap->begin(); };
+	void AddRef(){m_nRefCount++;}
+	void Release() { --m_nRefCount; if(m_nRefCount<=0){delete this;} };
+	
+protected:
+	int m_nRefCount;
+	EntityMap *m_pEntityMap;
+	EntityMapItor m_itEntityMap;
+
+};
+
+class CEntityItVec : public IEntityIt
+{
+public:
+	CEntityItVec(EntityVector *pVec)
+	{
+		m_nRefCount=0;
+		m_pEntityVec = pVec;
+		MoveFirst();
+	};
+
+	bool IsEnd() { return (m_itEntityVec == m_pEntityVec->end()); };
+	IEntity * Next() { return IsEnd() ? NULL : (*m_itEntityVec++); };
+	void MoveFirst() { m_itEntityVec = m_pEntityVec->begin(); };
+	void AddRef(){m_nRefCount++;}
+	void Release() { --m_nRefCount; if(m_nRefCount<=0){delete this;} };
+
+protected:
+	int m_nRefCount;
+	EntityVector *m_pEntityVec;
+	EntityVectorItor m_itEntityVec;
+
+};
+
+#endif // !defined(AFX_ENTITYIT_H__95AE38A4_7F15_4069_A97A_2F3A1F06F670__INCLUDED_)