From 93be0f975f10bc84a8c520ce831b8eec3200076e Mon Sep 17 00:00:00 2001
From: Ahmed Samy <f.fallen45@gmail.com>
Date: Tue, 14 Jan 2014 01:45:57 +0200
Subject: [PATCH] getCreatures / getSpawns

---
 src/client/creatures.cpp    | 17 ++++++++++++++++-
 src/client/creatures.h      |  2 ++
 src/client/declarations.h   |  2 ++
 src/client/luafunctions.cpp |  2 ++
 4 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/src/client/creatures.cpp b/src/client/creatures.cpp
index d6498775..8325cac0 100644
--- a/src/client/creatures.cpp
+++ b/src/client/creatures.cpp
@@ -144,6 +144,14 @@ void Spawn::removeCreature(const Position& pos)
     }
 }
 
+std::vector<CreatureTypePtr> Spawn::getCreatures()
+{
+    std::vector<CreatureTypePtr> creatures;
+    for (auto p : m_creatures)
+        creatures.push_back(p.second);
+    return creatures;
+}
+
 CreaturePtr CreatureType::cast()
 {
     CreaturePtr ret(new Creature);
@@ -399,5 +407,12 @@ SpawnPtr CreatureManager::addSpawn(const Position& centerPos, int radius)
     return ret;
 }
 
-/* vim: set ts=4 sw=4 et: */
+std::vector<SpawnPtr> CreatureManager::getSpawns()
+{
+    std::vector<SpawnPtr> spawns;
+    for (auto p : m_spawns)
+        spawns.push_back(p.second);
+    return spawns;
+}
 
+/* vim: set ts=4 sw=4 et: */
diff --git a/src/client/creatures.h b/src/client/creatures.h
index 4e2fb531..5ddc5bf6 100644
--- a/src/client/creatures.h
+++ b/src/client/creatures.h
@@ -61,6 +61,7 @@ public:
     void setCenterPos(const Position& pos) { m_attribs.set(SpawnAttrCenter, pos); }
     Position getCenterPos() { return m_attribs.get<Position>(SpawnAttrCenter); }
 
+    std::vector<CreatureTypePtr> getCreatures();
     void addCreature(const Position& placePos, const CreatureTypePtr& cType);
     void removeCreature(const Position& pos);
     void clear() { m_creatures.clear(); }
@@ -120,6 +121,7 @@ public:
     const CreatureTypePtr& getCreatureByName(std::string name);
     const CreatureTypePtr& getCreatureByLook(int look);
 
+    std::vector<SpawnPtr> getSpawns();
     SpawnPtr getSpawn(const Position& centerPos);
     SpawnPtr getSpawnForPlacePos(const Position& pos);
     SpawnPtr addSpawn(const Position& centerPos, int radius);
diff --git a/src/client/declarations.h b/src/client/declarations.h
index c3dc86c7..17c4fe17 100644
--- a/src/client/declarations.h
+++ b/src/client/declarations.h
@@ -82,6 +82,8 @@ typedef std::list<TownPtr> TownList;
 typedef std::list<ItemPtr> ItemList;
 typedef std::vector<ItemPtr> ItemVector;
 typedef std::unordered_map<Position, TilePtr, PositionHasher> TileMap;
+typedef std::unordered_map<Position, CreatureTypePtr, PositionHasher> CreatureMap;
+typedef std::unordered_map<Position, SpawnPtr, PositionHasher> SpawnMap;
 
 // net
 class ProtocolLogin;
diff --git a/src/client/luafunctions.cpp b/src/client/luafunctions.cpp
index 2f86a9fd..965079e1 100644
--- a/src/client/luafunctions.cpp
+++ b/src/client/luafunctions.cpp
@@ -172,6 +172,7 @@ void Client::registerLuaFunctions()
     g_lua.bindSingletonFunction("g_creatures", "isSpawnLoaded", &CreatureManager::isSpawnLoaded, &g_creatures);
     g_lua.bindSingletonFunction("g_creatures", "clear", &CreatureManager::clear, &g_creatures);
     g_lua.bindSingletonFunction("g_creatures", "clearSpawns", &CreatureManager::clearSpawns, &g_creatures);
+    g_lua.bindSingletonFunction("g_creatures", "getSpawns", &CreatureManager::getSpawns, &g_creatures);
 
     g_lua.registerSingletonClass("g_game");
     g_lua.bindSingletonFunction("g_game", "loginWorld", &Game::loginWorld, &g_game);
@@ -398,6 +399,7 @@ void Client::registerLuaFunctions()
     g_lua.bindClassMemberFunction<Spawn>("getCenterPos", &Spawn::getCenterPos);
     g_lua.bindClassMemberFunction<Spawn>("addCreature", &Spawn::addCreature);
     g_lua.bindClassMemberFunction<Spawn>("removeCreature", &Spawn::removeCreature);
+    g_lua.bindClassMemberFunction<Spawn>("getCreatures", &Spawn::getCreatures);
 
     g_lua.registerClass<Town>();
     g_lua.bindClassStaticFunction<Town>("create", []{ return TownPtr(new Town); });