From 8cd2c43afe043a85451d9e3e5aba71b8b7c6e0e9 Mon Sep 17 00:00:00 2001
From: Yohann Ferreira <yohann_dot_ferreira_at_orange_dot_efer>
Date: Fri, 2 Dec 2011 23:51:07 +0100
Subject: Fixed the chr_inv_count function to handle equipment.

the function can now count in the inventory
and/or the player's equipment.

I also fixed the script function and added a use case
in the example map.

+ Fixes after Ablu's review.
+ 2nd fix after Ablu's review: Fix the inventory remove behaviour.

Resolves: Mana-Mantis #288

Reviewed-by: Ablu
---
 src/game-server/inventory.cpp | 42 +++++++++++++++++++++++++++++++++---------
 src/game-server/inventory.h   |  5 ++++-
 src/scripting/lua.cpp         | 17 +++++++++++------
 3 files changed, 48 insertions(+), 16 deletions(-)

(limited to 'src')

diff --git a/src/game-server/inventory.cpp b/src/game-server/inventory.cpp
index 4d447d3a..1b5e67e8 100644
--- a/src/game-server/inventory.cpp
+++ b/src/game-server/inventory.cpp
@@ -217,14 +217,34 @@ unsigned int Inventory::insert(unsigned int itemId, unsigned int amount)
     return amount;
 }
 
-unsigned int Inventory::count(unsigned int itemId) const
+unsigned int Inventory::count(unsigned int itemId,
+                              bool inInventory, bool inEquipment) const
 {
     unsigned int nb = 0;
-    for (InventoryData::iterator it = mPoss->inventory.begin(),
-                                 it_end = mPoss->inventory.end();
-         it != it_end; ++it)
-        if (it->second.itemId == itemId)
-            nb += it->second.amount;
+    if (inInventory)
+    {
+        for (InventoryData::iterator it = mPoss->inventory.begin(),
+            it_end = mPoss->inventory.end(); it != it_end; ++it)
+            if (it->second.itemId == itemId)
+                nb += it->second.amount;
+    }
+
+    if (inEquipment)
+    {
+        std::set<unsigned> itemInstances;
+        for (EquipData::iterator it = mPoss->equipSlots.begin(),
+            it_end = mPoss->equipSlots.end(); it != it_end; ++it)
+        {
+            if (it->second.itemId != itemId || !it->second.itemInstance)
+                continue;
+
+            // If the insertion was successful, then it was the first time,
+            // and can be counted.
+            if ((itemInstances.insert(it->second.itemInstance)).second)
+                ++nb;
+        }
+    }
+
     return nb;
 }
 
@@ -248,9 +268,10 @@ unsigned int Inventory::remove(unsigned int itemId, unsigned int amount)
 
     MessageOut invMsg(GPMSG_INVENTORY);
     bool triggerLeaveInventory = true;
-    for (InventoryData::iterator it = mPoss->inventory.begin(),
-             it_end = mPoss->inventory.end(); it != it_end; ++it)
+    for (InventoryData::iterator it = mPoss->inventory.begin();
+         it != mPoss->inventory.end();)
     {
+        LOG_DEBUG("Remove: Treating slot id: " << it->first);
         if (it->second.itemId == itemId)
         {
             if (amount)
@@ -273,8 +294,10 @@ unsigned int Inventory::remove(unsigned int itemId, unsigned int amount)
                 else
                 {
                     invMsg.writeInt16(0);
-                    mPoss->inventory.erase(it);
+                    // Ensure the slot is set empty.
                     LOG_DEBUG("Slot id: " << it->first << " is now empty.");
+                    mPoss->inventory.erase(it++);
+                    continue;
                 }
             }
             else
@@ -284,6 +307,7 @@ unsigned int Inventory::remove(unsigned int itemId, unsigned int amount)
                 triggerLeaveInventory = false;
             }
         }
+        ++it;
     }
 
     if (triggerLeaveInventory)
diff --git a/src/game-server/inventory.h b/src/game-server/inventory.h
index 812c142d..4593add1 100644
--- a/src/game-server/inventory.h
+++ b/src/game-server/inventory.h
@@ -112,8 +112,11 @@ class Inventory
 
         /**
          * Counts number of items with given Id.
+         * @param inInventory Search in player's inventory.
+         * @param inEquipment Search in player's equipment.
          */
-        unsigned int count(unsigned int itemId) const;
+        unsigned int count(unsigned int itemId, bool inInventory = true,
+                           bool inEquipment = true) const;
 
         /**
          * Gets the ID of the items in a given slot.
diff --git a/src/scripting/lua.cpp b/src/scripting/lua.cpp
index f681133f..14658736 100644
--- a/src/scripting/lua.cpp
+++ b/src/scripting/lua.cpp
@@ -435,23 +435,28 @@ static int chr_inv_change(lua_State *s)
 }
 
 /**
- * mana.chr_inv_count(Character*, int item_id...): int count...
+ * mana.chr_inv_count(Character*, bool in inventory, bool in equipment,
+ *                    int item_id...): int count...
  * Callback for counting items in inventory.
+ * The function can search in inventory and/or in the equipment.
  */
 static int chr_inv_count(lua_State *s)
 {
     Character *q = getCharacter(s, 1);
-    if (!q)
+    if (!q || !lua_isboolean(s, 2) || !lua_isboolean(s, 3))
     {
         raiseScriptError(s, "chr_inv_count called with incorrect parameters.");
         return 0;
     }
-    int nb_items = lua_gettop(s) - 1;
-    lua_checkstack(s, nb_items);
+
+    bool inInventory = lua_toboolean(s, 2);
+    bool inEquipment = lua_toboolean(s, 3);
+
+    int nb_items = lua_gettop(s) - 3;
     Inventory inv(q);
 
     int id, nb = 0;
-    for (int i = 2; i <= nb_items + 1; ++i)
+    for (int i = 4; i < nb_items + 4; ++i)
     {
         ItemClass *it;
         if (lua_isnumber(s, i))
@@ -474,7 +479,7 @@ static int chr_inv_count(lua_State *s)
         }
         else
         {
-            nb = inv.count(id);
+            nb = inv.count(id, inInventory, inEquipment);
             lua_pushinteger(s, nb);
         }
     }
-- 
cgit v1.2.3-70-g09d2