From f2dc36be2948f97a630a7e51c8f60b5ca0217c2c Mon Sep 17 00:00:00 2001
From: maximius <none@none>
Date: Fri, 28 Aug 2009 02:42:57 -0700
Subject: *Add new functionality to Player::resetSpells.. not used anywhere
 yet, but it's useful (and it works :P)

--HG--
branch : trunk
---
 src/game/Player.cpp | 38 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

(limited to 'src/game/Player.cpp')

diff --git a/src/game/Player.cpp b/src/game/Player.cpp
index eaf5d9ecb02..41ad8b23982 100644
--- a/src/game/Player.cpp
+++ b/src/game/Player.cpp
@@ -19387,7 +19387,7 @@ void Player::ApplyEquipCooldown( Item * pItem )
     }
 }
 
-void Player::resetSpells()
+void Player::resetSpells(bool myClassOnly)
 {
     // not need after this call
     if(HasAtLoginFlag(AT_LOGIN_RESET_SPELLS))
@@ -19397,8 +19397,44 @@ void Player::resetSpells()
     // and we can't use original map for safe iterative with visit each spell at loop end
     PlayerSpellMap smap = GetSpellMap();
 
+    if(myClassOnly){
+        ChrClassesEntry const* clsEntry = sChrClassesStore.LookupEntry(m_session->GetPlayer()->getClass());
+        if(!clsEntry)
+            return;
+        uint32 family = clsEntry->spellfamily;
+    }
+
     for(PlayerSpellMap::const_iterator iter = smap.begin();iter != smap.end(); ++iter)
+    {
+        if(myClassOnly)
+        {
+            SpellEntry const *spellInfo = sSpellStore.LookupEntry(iter->first);
+            if(!spellInfo)
+                continue;
+
+            // skip server-side/triggered spells
+            if(spellInfo->spellLevel == 0)
+                continue;
+
+            // skip wrong class/race skills
+            if(!m_session->GetPlayer()->IsSpellFitByClassAndRace(spellInfo->Id))
+                continue;
+
+            // skip other spell families
+            if( spellInfo->SpellFamilyName != family)
+                continue;
+
+            // skip spells with first rank learned as talent (and all talents then also)
+            uint32 first_rank = spellmgr.GetFirstSpellInChain(spellInfo->Id);
+            if(GetTalentSpellCost(first_rank) > 0)
+                continue;
+
+            // skip broken spells
+            if(!SpellMgr::IsSpellValid(spellInfo,m_session->GetPlayer(),false))
+                continue;
+        }
         removeSpell(iter->first,false,false);               // only iter->first can be accessed, object by iter->second can be deleted already
+    }
 
     learnDefaultSpells();
     learnQuestRewardedSpells();
-- 
cgit v1.2.3