From f43a10b7b99c6520873a8b36eefe4a8209281aee Mon Sep 17 00:00:00 2001
From: scrawl <scrawl@baseoftrash.de>
Date: Tue, 15 Jul 2014 21:10:30 +0200
Subject: [PATCH] Adjust disease contraction according to Hrnchamd's research

---
 apps/openmw/mwmechanics/disease.hpp | 58 ++++++++++++++++++++---------
 1 file changed, 41 insertions(+), 17 deletions(-)

diff --git a/apps/openmw/mwmechanics/disease.hpp b/apps/openmw/mwmechanics/disease.hpp
index 5f73e8acd..05ce1c7ae 100644
--- a/apps/openmw/mwmechanics/disease.hpp
+++ b/apps/openmw/mwmechanics/disease.hpp
@@ -14,9 +14,11 @@ namespace MWMechanics
 {
 
     /// Call when \a actor has got in contact with \a carrier (e.g. hit by him, or loots him)
+    /// @param actor The actor that will potentially catch diseases. Currently only the player can catch diseases.
+    /// @param carrier The disease carrier.
     inline void diseaseContact (MWWorld::Ptr actor, MWWorld::Ptr carrier)
     {
-        if (!carrier.getClass().isActor())
+        if (!carrier.getClass().isActor() || actor != MWBase::Environment::get().getWorld()->getPlayerPtr())
             return;
 
         float fDiseaseXferChance =
@@ -27,25 +29,47 @@ namespace MWMechanics
         for (Spells::TIterator it = spells.begin(); it != spells.end(); ++it)
         {
             const ESM::Spell* spell = MWBase::Environment::get().getWorld()->getStore().get<ESM::Spell>().find(it->first);
-            if (spell->mData.mType == ESM::Spell::ST_Disease
-                    || spell->mData.mType == ESM::Spell::ST_Blight)
-            {
-                float roll = std::rand()/ (static_cast<double> (RAND_MAX) + 1) * 100; // [0, 99]
-                if (roll < fDiseaseXferChance)
-                {
-                    // Contracted disease!
-                    actor.getClass().getCreatureStats(actor).getSpells().add(it->first);
 
-                    if (actor.getRefData().getHandle() == "player")
-                    {
-                        std::string msg = "sMagicContractDisease";
-                        msg = MWBase::Environment::get().getWorld()->getStore().get<ESM::GameSetting>().find(msg)->getString();
-                        if (msg.find("%s") != std::string::npos)
-                            msg.replace(msg.find("%s"), 2, spell->mName);
-                        MWBase::Environment::get().getWindowManager()->messageBox(msg);
-                    }
+            if (actor.getClass().getCreatureStats(actor).getSpells().hasSpell(spell->mId))
+                continue;
+
+            bool hasCorprusEffect = false;
+            for (std::vector<ESM::ENAMstruct>::const_iterator effectIt = spell->mEffects.mList.begin(); effectIt != spell->mEffects.mList.end(); ++effectIt)
+            {
+                if (effectIt->mEffectID == ESM::MagicEffect::Corprus)
+                {
+                    hasCorprusEffect = true;
+                    break;
                 }
             }
+
+            float resist = 0.f;
+            if (hasCorprusEffect)
+                resist = 1.f - 0.01 * (actor.getClass().getCreatureStats(actor).getMagicEffects().get(ESM::MagicEffect::ResistCorprusDisease).mMagnitude
+                                        - actor.getClass().getCreatureStats(actor).getMagicEffects().get(ESM::MagicEffect::WeaknessToCorprusDisease).mMagnitude);
+            else if (spell->mData.mType == ESM::Spell::ST_Disease)
+                resist = 1.f - 0.01 * (actor.getClass().getCreatureStats(actor).getMagicEffects().get(ESM::MagicEffect::ResistCommonDisease).mMagnitude
+                                        - actor.getClass().getCreatureStats(actor).getMagicEffects().get(ESM::MagicEffect::WeaknessToCommonDisease).mMagnitude);
+            else if (spell->mData.mType == ESM::Spell::ST_Blight)
+                resist = 1.f - 0.01 * (actor.getClass().getCreatureStats(actor).getMagicEffects().get(ESM::MagicEffect::ResistBlightDisease).mMagnitude
+                                        - actor.getClass().getCreatureStats(actor).getMagicEffects().get(ESM::MagicEffect::WeaknessToBlightDisease).mMagnitude);
+            else
+                continue;
+
+            int x = fDiseaseXferChance * 100 * resist;
+            float roll = std::rand()/ (static_cast<double> (RAND_MAX) + 1) * 10000; // [0, 9999]
+
+            if (roll < x)
+            {
+                // Contracted disease!
+                actor.getClass().getCreatureStats(actor).getSpells().add(it->first);
+
+                std::string msg = "sMagicContractDisease";
+                msg = MWBase::Environment::get().getWorld()->getStore().get<ESM::GameSetting>().find(msg)->getString();
+                if (msg.find("%s") != std::string::npos)
+                    msg.replace(msg.find("%s"), 2, spell->mName);
+                MWBase::Environment::get().getWindowManager()->messageBox(msg);
+            }
         }
     }
 }