Updated code to support new MapTrainerTeamMember item representation

1 files changed, 5 insertions, 4 deletions

diff --git a/sigscript/MapTrainerTeamMemberWrapper.cpp b/sigscript/MapTrainerTeamMemberWrapper.cpp
index 6309c1d7..d7b97cc4 100644
--- a/sigscript/MapTrainerTeamMemberWrapper.cpp
+++ b/sigscript/MapTrainerTeamMemberWrapper.cpp
@@ -56,12 +56,13 @@ QList<Sigscript::AbilityWrapper*> Sigscript::MapTrainerTeamMemberWrapper::abilit
     return abilities;
 }
 
-QList<Sigscript::ItemWrapper*> Sigscript::MapTrainerTeamMemberWrapper::items()
+QMap<Sigscript::ItemWrapper*, int> Sigscript::MapTrainerTeamMemberWrapper::items()
 {
-    QList<int> itemIds = m_teamMember->item();
-    QList<ItemWrapper*> items;
+    const QMap<int, int>& itemMap = m_teamMember->item();
+    QList<int> itemIds = itemMap.keys();
+    QMap<ItemWrapper*, int> items;
     foreach (int id, itemIds)
-        items.append(sigmod()->item(id));
+        items[sigmod()->item(id)] = itemMap[id];
     return items;
 }