From 2d2196b0d60d4b874af143e370547e72d14ef3ca Mon Sep 17 00:00:00 2001
From: scrawl <scrawl@baseoftrash.de>
Date: Sun, 7 Apr 2013 21:01:02 +0200
Subject: [PATCH] Implemented levelled items

---
 apps/openmw/mwworld/containerstore.cpp | 73 ++++++++++++++++++++++----
 apps/openmw/mwworld/containerstore.hpp |  1 +
 components/esm/loadlevlist.hpp         | 15 +++---
 3 files changed, 73 insertions(+), 16 deletions(-)

diff --git a/apps/openmw/mwworld/containerstore.cpp b/apps/openmw/mwworld/containerstore.cpp
index a377f2bbb..8a5662986 100644
--- a/apps/openmw/mwworld/containerstore.cpp
+++ b/apps/openmw/mwworld/containerstore.cpp
@@ -14,6 +14,8 @@
 #include "../mwbase/world.hpp"
 #include "../mwbase/scriptmanager.hpp"
 
+#include "../mwmechanics/creaturestats.hpp"
+
 #include "manualref.hpp"
 #include "refdata.hpp"
 #include "class.hpp"
@@ -180,21 +182,72 @@ void MWWorld::ContainerStore::fill (const ESM::InventoryList& items, const MWWor
     for (std::vector<ESM::ContItem>::const_iterator iter (items.mList.begin()); iter!=items.mList.end();
         ++iter)
     {
-        ManualRef ref (store, iter->mItem.toString());
-
-        if (ref.getPtr().getTypeName()==typeid (ESM::ItemLevList).name())
-        {
-            /// \todo implement leveled item lists
-            continue;
-        }
-
-        ref.getPtr().getRefData().setCount (std::abs(iter->mCount)); /// \todo implement item restocking (indicated by negative count)
-        addImp (ref.getPtr());
+        std::string id = iter->mItem.toString();
+        addInitialItem(id, iter->mCount);
     }
 
     flagAsModified();
 }
 
+void MWWorld::ContainerStore::addInitialItem (const std::string& id, int count, unsigned char failChance, bool topLevel)
+{
+    count = std::abs(count); /// \todo implement item restocking (indicated by negative count)
+
+    ManualRef ref (MWBase::Environment::get().getWorld()->getStore(), id);
+
+    if (ref.getPtr().getTypeName()==typeid (ESM::ItemLevList).name())
+    {
+        const ESM::ItemLevList* levItem = ref.getPtr().get<ESM::ItemLevList>()->mBase;
+        const std::vector<ESM::LeveledListBase::LevelItem>& items = levItem->mList;
+
+        MWWorld::Ptr player = MWBase::Environment::get().getWorld()->getPlayer().getPlayer();
+        int playerLevel = MWWorld::Class::get(player).getCreatureStats(player).getLevel();
+
+        failChance += levItem->mChanceNone;
+
+        if (topLevel && count > 1 && levItem->mFlags & ESM::ItemLevList::Each)
+        {
+            for (int i=0; i<count; ++i)
+                addInitialItem(id, 1, failChance, false);
+            return;
+        }
+
+        float random = static_cast<float> (std::rand()) / RAND_MAX;
+        if (random >= failChance/100.f)
+        {
+            std::vector<std::string> candidates;
+            int highestLevel = 0;
+            for (std::vector<ESM::LeveledListBase::LevelItem>::const_iterator it = items.begin(); it != items.end(); ++it)
+            {
+                if (it->mLevel > highestLevel)
+                    highestLevel = it->mLevel;
+            }
+
+            std::pair<int, std::string> highest = std::make_pair(-1, "");
+            for (std::vector<ESM::LeveledListBase::LevelItem>::const_iterator it = items.begin(); it != items.end(); ++it)
+            {
+                if (playerLevel >= it->mLevel
+                        && (levItem->mFlags & ESM::ItemLevList::AllLevels || it->mLevel == highestLevel))
+                {
+                    candidates.push_back(it->mId);
+                    if (it->mLevel >= highest.first)
+                        highest = std::make_pair(it->mLevel, it->mId);
+                }
+
+            }
+            if (!candidates.size())
+                return;
+            std::string item = candidates[std::rand()%candidates.size()];
+            addInitialItem(item, count, failChance, false);
+        }
+    }
+    else
+    {
+        ref.getPtr().getRefData().setCount (count);
+        addImp (ref.getPtr());
+    }
+}
+
 void MWWorld::ContainerStore::clear()
 {
     for (ContainerStoreIterator iter (begin()); iter!=end(); ++iter)
diff --git a/apps/openmw/mwworld/containerstore.hpp b/apps/openmw/mwworld/containerstore.hpp
index 9d315d27f..a466c3c2a 100644
--- a/apps/openmw/mwworld/containerstore.hpp
+++ b/apps/openmw/mwworld/containerstore.hpp
@@ -53,6 +53,7 @@ namespace MWWorld
             mutable float mCachedWeight;
             mutable bool mWeightUpToDate;
             ContainerStoreIterator addImp (const Ptr& ptr);
+            void addInitialItem (const std::string& id, int count, unsigned char failChance=0, bool topLevel=true);
 
         public:
 
diff --git a/components/esm/loadlevlist.hpp b/components/esm/loadlevlist.hpp
index b7db5db36..aa9656d72 100644
--- a/components/esm/loadlevlist.hpp
+++ b/components/esm/loadlevlist.hpp
@@ -22,17 +22,20 @@ struct LeveledListBase
 {
     enum Flags
     {
-        AllLevels = 0x01, // Calculate from all levels <= player
-                          // level, not just the closest below
-                          // player.
-        Each = 0x02       // Select a new item each time this
+
+        Each = 0x01,      // Select a new item each time this
                           // list is instantiated, instead of
                           // giving several identical items
-    };                    // (used when a container has more
+                          // (used when a container has more
                           // than one instance of one leveled
                           // list.)
+        AllLevels = 0x02  // Calculate from all levels <= player
+                          // level, not just the closest below
+                          // player.
+    };
+
     int mFlags;
-    unsigned char mChanceNone; // Chance that none are selected (0-255?)
+    unsigned char mChanceNone; // Chance that none are selected (0-100)
     std::string mId;
 
     // Record name used to read references. Must be set before load() is