Merge pull request #62 from ghronkrepps/armor-spec

implement armor specialization bonuses
wowsims · Apr 14, 2024 · 819f817 · 819f817
2 parents 429c6cf + 8d5de5b
commit 819f817
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 2 deletions.
diff --git a/sim/core/character.go b/sim/core/character.go
@@ -727,3 +727,24 @@ func FillTalentsProto(data protoreflect.Message, talentsStr string, treeSizes [3
 		offset += treeSizes[treeIdx]
 	}
 }
+
+func (character *Character) EnableArmorSpecialization(primaryStat stats.Stat, armorType proto.ArmorType) bool {
+	hasBonus := true
+
+	if character.Head().ArmorType != armorType ||
+		character.Shoulder().ArmorType != armorType ||
+		character.Chest().ArmorType != armorType ||
+		character.Wrist().ArmorType != armorType ||
+		character.Hands().ArmorType != armorType ||
+		character.Waist().ArmorType != armorType ||
+		character.Legs().ArmorType != armorType ||
+		character.Feet().ArmorType != armorType {
+		hasBonus = false
+	}
+
+	if hasBonus {
+		character.MultiplyStat(primaryStat, 1.05)
+	}
+
+	return hasBonus
+}
diff --git a/sim/core/database.go b/sim/core/database.go
@@ -214,12 +214,36 @@ func (equipment *Equipment) Head() *Item {
 	return &equipment[proto.ItemSlot_ItemSlotHead]
 }
 
+func (equipment *Equipment) Neck() *Item {
+	return &equipment[proto.ItemSlot_ItemSlotNeck]
+}
+
+func (equipment *Equipment) Shoulder() *Item {
+	return &equipment[proto.ItemSlot_ItemSlotShoulder]
+}
+
+func (equipment *Equipment) Chest() *Item {
+	return &equipment[proto.ItemSlot_ItemSlotChest]
+}
+
+func (equipment *Equipment) Wrist() *Item {
+	return &equipment[proto.ItemSlot_ItemSlotWrist]
+}
+
 func (equipment *Equipment) Hands() *Item {
 	return &equipment[proto.ItemSlot_ItemSlotHands]
 }
 
-func (equipment *Equipment) Neck() *Item {
-	return &equipment[proto.ItemSlot_ItemSlotNeck]
+func (equipment *Equipment) Waist() *Item {
+	return &equipment[proto.ItemSlot_ItemSlotWaist]
+}
+
+func (equipment *Equipment) Legs() *Item {
+	return &equipment[proto.ItemSlot_ItemSlotLegs]
+}
+
+func (equipment *Equipment) Feet() *Item {
+	return &equipment[proto.ItemSlot_ItemSlotFeet]
 }
 
 func (equipment *Equipment) Trinket1() *Item {

diff --git a/sim/shaman/shaman.go b/sim/shaman/shaman.go
@@ -5,6 +5,7 @@ import (
 
 	"github.com/wowsims/cata/sim/core"
 	"github.com/wowsims/cata/sim/core/proto"
+	"github.com/wowsims/cata/sim/core/stats"
 )
 
 var TalentTreeSizes = [3]int{19, 19, 20}
@@ -248,6 +249,12 @@ func (shaman *Shaman) Initialize() {
 	// shaman.registerBloodlustCD()
 
 	// shaman.NewTemporaryStatsAura("DC Pre-Pull SP Proc", core.ActionID{SpellID: 60494}, stats.Stats{stats.SpellPower: 765}, time.Second*10)
+
+	if shaman.Spec == proto.Spec_SpecElementalShaman || shaman.Spec == proto.Spec_SpecRestorationShaman {
+		shaman.EnableArmorSpecialization(stats.Intellect, proto.ArmorType_ArmorTypeMail)
+	} else if shaman.Spec == proto.Spec_SpecElementalShaman {
+		shaman.EnableArmorSpecialization(stats.Agility, proto.ArmorType_ArmorTypeMail)
+	}
 }
 
 func (shaman *Shaman) RegisterHealingSpells() {