mirror of
https://github.com/mod-playerbots/mod-playerbots
synced 2025-11-29 15:58:20 +08:00
Rewrite stats weight and item score calculation
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -116,9 +116,7 @@ public:
|
||||
void InitSkills();
|
||||
|
||||
static uint32 tradeSkills[];
|
||||
static float CalculateItemScore(uint32 item_id, Player* bot);
|
||||
static float CalculateEnchantScore(uint32 enchant_id, Player* bot);
|
||||
static float CalculateSpellScore(uint32 spell_id, Player* bot, uint32 trigger = ITEM_SPELLTRIGGER_ON_EQUIP);
|
||||
void InitTalentsTree(bool incremental = false, bool use_template = true, bool reset = false);
|
||||
static void InitTalentsBySpecNo(Player* bot, int specNo, bool reset);
|
||||
static void InitTalentsByParsedSpecLink(Player* bot, std::vector<std::vector<uint32>> parsedSpecLink, bool reset);
|
||||
@@ -181,8 +179,6 @@ private:
|
||||
void ApplyEnchantTemplate();
|
||||
void ApplyEnchantTemplate(uint8 spec);
|
||||
std::vector<InventoryType> GetPossibleInventoryTypeListBySlot(EquipmentSlots slot);
|
||||
static bool IsShieldTank(Player* bot);
|
||||
static bool NotSameArmorType(uint32 item_subclass_armor, Player* bot);
|
||||
void IterateItems(IterateItemsVisitor* visitor, IterateItemsMask mask = ITERATE_ITEMS_IN_BAGS);
|
||||
void IterateItemsInBags(IterateItemsVisitor* visitor);
|
||||
void IterateItemsInEquip(IterateItemsVisitor* visitor);
|
||||
407
src/factory/StatsCollector.cpp
Normal file
407
src/factory/StatsCollector.cpp
Normal file
@@ -0,0 +1,407 @@
|
||||
#include "StatsCollector.h"
|
||||
|
||||
#include <cstdint>
|
||||
|
||||
#include "DBCStores.h"
|
||||
#include "ItemTemplate.h"
|
||||
#include "ObjectMgr.h"
|
||||
#include "SpellInfo.h"
|
||||
#include "SpellMgr.h"
|
||||
|
||||
StatsCollector::StatsCollector(CollectorType type) : type_(type) { Reset(); }
|
||||
|
||||
void StatsCollector::Reset()
|
||||
{
|
||||
for (uint32 i = 0; i < STATS_TYPE_MAX; i++)
|
||||
{
|
||||
stats[i] = 0;
|
||||
}
|
||||
}
|
||||
|
||||
void StatsCollector::CollectItemStats(ItemTemplate const* proto)
|
||||
{
|
||||
if (proto->IsRangedWeapon())
|
||||
{
|
||||
uint32 val = (proto->Damage[0].DamageMin + proto->Damage[0].DamageMax) * 1000 / 2 / proto->Delay;
|
||||
stats[STATS_TYPE_RANGED_DPS] += val;
|
||||
}
|
||||
else if (proto->IsWeapon())
|
||||
{
|
||||
uint32 val = (proto->Damage[0].DamageMin + proto->Damage[0].DamageMax) * 1000 / 2 / proto->Delay;
|
||||
stats[STATS_TYPE_MELEE_DPS] += val;
|
||||
}
|
||||
stats[STATS_TYPE_ARMOR] += proto->Armor;
|
||||
stats[STATS_TYPE_BLOCK_VALUE] += proto->Block;
|
||||
for (int i = 0; i < proto->StatsCount; i++)
|
||||
{
|
||||
const _ItemStat& stat = proto->ItemStat[i];
|
||||
const int32& val = stat.ItemStatValue;
|
||||
CollectByItemStatType(stat.ItemStatType, val);
|
||||
}
|
||||
for (uint8 j = 0; j < MAX_ITEM_PROTO_SPELLS; j++)
|
||||
{
|
||||
CollectSpellStats(proto->Spells[j].SpellId, proto->Spells[j].SpellTrigger);
|
||||
}
|
||||
|
||||
if (proto->socketBonus)
|
||||
{
|
||||
if (const SpellItemEnchantmentEntry *enchant = sSpellItemEnchantmentStore.LookupEntry(proto->socketBonus))
|
||||
CollectEnchantStats(enchant);
|
||||
}
|
||||
}
|
||||
|
||||
void StatsCollector::CollectSpellStats(uint32 spellId, uint32 trigger)
|
||||
{
|
||||
SpellInfo const* spellInfo = sSpellMgr->GetSpellInfo(spellId);
|
||||
|
||||
if (!spellInfo)
|
||||
return;
|
||||
|
||||
if (CollectSpecialCaseSpellStats(spellId))
|
||||
return;
|
||||
|
||||
for (int i = 0; i < MAX_SPELL_EFFECTS; i++)
|
||||
{
|
||||
float multiplier = trigger == ITEM_SPELLTRIGGER_ON_EQUIP ? 1.0f : 0.2f;
|
||||
CollectSpellEffectStats(spellInfo->Effects[i], multiplier);
|
||||
}
|
||||
}
|
||||
|
||||
void StatsCollector::CollectSpellEffectStats(const SpellEffectInfo& effectInfo, float multiplier)
|
||||
{
|
||||
if (effectInfo.Effect != SPELL_EFFECT_APPLY_AURA)
|
||||
return;
|
||||
|
||||
int32 val = effectInfo.BasePoints + 1;
|
||||
switch (effectInfo.ApplyAuraName)
|
||||
{
|
||||
case SPELL_AURA_MOD_DAMAGE_DONE:
|
||||
// case SPELL_AURA_MOD_HEALING_DONE is duplicated
|
||||
stats[STATS_TYPE_SPELL_POWER] += val * multiplier;
|
||||
break;
|
||||
case SPELL_AURA_MOD_ATTACK_POWER:
|
||||
stats[STATS_TYPE_ATTACK_POWER] += val * multiplier;
|
||||
break;
|
||||
case SPELL_AURA_MOD_SHIELD_BLOCKVALUE:
|
||||
stats[STATS_TYPE_BLOCK_VALUE] += val * multiplier;
|
||||
break;
|
||||
case SPELL_AURA_MOD_RATING:
|
||||
{
|
||||
for (uint32 rating = CR_WEAPON_SKILL; rating < MAX_COMBAT_RATING; ++rating)
|
||||
{
|
||||
if (effectInfo.MiscValue & (1 << rating))
|
||||
{
|
||||
switch (rating)
|
||||
{
|
||||
case CR_DEFENSE_SKILL:
|
||||
stats[STATS_TYPE_DEFENSE] += val * multiplier;
|
||||
break;
|
||||
case CR_DODGE:
|
||||
stats[STATS_TYPE_DODGE] += val * multiplier;
|
||||
break;
|
||||
case CR_PARRY:
|
||||
stats[STATS_TYPE_PARRY] += val * multiplier;
|
||||
break;
|
||||
case CR_BLOCK:
|
||||
stats[STATS_TYPE_BLOCK_RATING] += val * multiplier;
|
||||
break;
|
||||
case CR_HIT_MELEE:
|
||||
if (type_ == CollectorType::MELEE)
|
||||
stats[STATS_TYPE_HIT] += val * multiplier;
|
||||
break;
|
||||
case CR_HIT_RANGED:
|
||||
if (type_ == CollectorType::RANGED)
|
||||
stats[STATS_TYPE_HIT] += val * multiplier;
|
||||
break;
|
||||
case CR_HIT_SPELL:
|
||||
if (type_ == CollectorType::SPELL)
|
||||
stats[STATS_TYPE_HIT] += val * multiplier;
|
||||
break;
|
||||
case CR_CRIT_MELEE:
|
||||
if (type_ == CollectorType::MELEE)
|
||||
stats[STATS_TYPE_CRIT] += val * multiplier;
|
||||
break;
|
||||
case CR_CRIT_RANGED:
|
||||
if (type_ == CollectorType::RANGED)
|
||||
stats[STATS_TYPE_CRIT] += val * multiplier;
|
||||
break;
|
||||
case CR_CRIT_SPELL:
|
||||
if (type_ == CollectorType::SPELL)
|
||||
stats[STATS_TYPE_CRIT] += val * multiplier;
|
||||
break;
|
||||
case CR_HASTE_MELEE:
|
||||
if (type_ == CollectorType::MELEE)
|
||||
stats[STATS_TYPE_HASTE] += val * multiplier;
|
||||
break;
|
||||
case CR_HASTE_RANGED:
|
||||
if (type_ == CollectorType::RANGED)
|
||||
stats[STATS_TYPE_HASTE] += val * multiplier;
|
||||
break;
|
||||
case CR_HASTE_SPELL:
|
||||
if (type_ == CollectorType::SPELL)
|
||||
stats[STATS_TYPE_HASTE] += val * multiplier;
|
||||
break;
|
||||
case CR_EXPERTISE:
|
||||
stats[STATS_TYPE_EXPERTISE] += val * multiplier;
|
||||
break;
|
||||
case CR_ARMOR_PENETRATION:
|
||||
stats[STATS_TYPE_ARMOR_PENETRATION] += val * multiplier;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
case SPELL_AURA_PROC_TRIGGER_SPELL:
|
||||
{
|
||||
multiplier = 0.2f;
|
||||
if (effectInfo.TriggerSpell)
|
||||
{
|
||||
SpellInfo const* triggerSpellInfo = sSpellMgr->GetSpellInfo(effectInfo.TriggerSpell);
|
||||
if (!triggerSpellInfo)
|
||||
return;
|
||||
for (uint8 k = 0; k < MAX_SPELL_EFFECTS; k++)
|
||||
{
|
||||
CollectSpellEffectStats(triggerSpellInfo->Effects[k], multiplier);
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void StatsCollector::CollectEnchantStats(SpellItemEnchantmentEntry const* enchant)
|
||||
{
|
||||
for (int s = 0; s < MAX_SPELL_ITEM_ENCHANTMENT_EFFECTS; ++s)
|
||||
{
|
||||
uint32 enchant_display_type = enchant->type[s];
|
||||
uint32 enchant_amount = enchant->amount[s];
|
||||
uint32 enchant_spell_id = enchant->spellid[s];
|
||||
|
||||
if (CollectSpecialEnchantSpellStats(enchant_spell_id))
|
||||
continue;
|
||||
|
||||
switch (enchant_display_type)
|
||||
{
|
||||
case ITEM_ENCHANTMENT_TYPE_STAT:
|
||||
{
|
||||
if (!enchant_amount)
|
||||
{
|
||||
break;
|
||||
}
|
||||
CollectByItemStatType(enchant_spell_id, enchant_amount);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// @todo Special case for trinket
|
||||
bool StatsCollector::CollectSpecialCaseSpellStats(uint32 spellId) { return false; }
|
||||
|
||||
bool StatsCollector::CollectSpecialEnchantSpellStats(uint32 enchantSpellId)
|
||||
{
|
||||
switch (enchantSpellId)
|
||||
{
|
||||
case 28093: // mongoose
|
||||
if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
stats[STATS_TYPE_AGILITY] += 40;
|
||||
}
|
||||
return true;
|
||||
case 20007: // crusader
|
||||
if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
stats[STATS_TYPE_STRENGTH] += 30;
|
||||
}
|
||||
return true;
|
||||
case 59620: // Berserk
|
||||
if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
stats[STATS_TYPE_ATTACK_POWER] += 120;
|
||||
}
|
||||
return true;
|
||||
case 64440: // Blade Warding
|
||||
if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
stats[STATS_TYPE_PARRY] += 50;
|
||||
}
|
||||
return true;
|
||||
case 64571:
|
||||
if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
stats[STATS_TYPE_STAMINA] += 50;
|
||||
}
|
||||
return true;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
{
|
||||
int allStatsAmount = 0;
|
||||
switch (enchantSpellId)
|
||||
{
|
||||
case 13624:
|
||||
allStatsAmount = 1;
|
||||
break;
|
||||
case 13625:
|
||||
allStatsAmount = 2;
|
||||
break;
|
||||
case 13824:
|
||||
allStatsAmount = 3;
|
||||
break;
|
||||
case 19988:
|
||||
case 44627:
|
||||
case 56527:
|
||||
allStatsAmount = 4;
|
||||
break;
|
||||
case 27959:
|
||||
case 56529:
|
||||
allStatsAmount = 6;
|
||||
break;
|
||||
case 44624:
|
||||
allStatsAmount = 8;
|
||||
break;
|
||||
case 60694:
|
||||
case 68251:
|
||||
allStatsAmount = 10;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
if (allStatsAmount != 0)
|
||||
{
|
||||
stats[STATS_TYPE_AGILITY] += allStatsAmount;
|
||||
stats[STATS_TYPE_STRENGTH] += allStatsAmount;
|
||||
stats[STATS_TYPE_INTELLECT] += allStatsAmount;
|
||||
stats[STATS_TYPE_SPIRIT] += allStatsAmount;
|
||||
stats[STATS_TYPE_STAMINA] += allStatsAmount;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
void StatsCollector::CollectByItemStatType(uint32 itemStatType, int32 val)
|
||||
{
|
||||
switch (itemStatType)
|
||||
{
|
||||
case ITEM_MOD_MANA:
|
||||
break;
|
||||
case ITEM_MOD_HEALTH:
|
||||
break;
|
||||
case ITEM_MOD_AGILITY:
|
||||
stats[STATS_TYPE_AGILITY] += val;
|
||||
break;
|
||||
case ITEM_MOD_STRENGTH:
|
||||
stats[STATS_TYPE_STRENGTH] += val;
|
||||
break;
|
||||
case ITEM_MOD_INTELLECT:
|
||||
stats[STATS_TYPE_INTELLECT] += val;
|
||||
break;
|
||||
case ITEM_MOD_SPIRIT:
|
||||
stats[STATS_TYPE_SPIRIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_STAMINA:
|
||||
stats[STATS_TYPE_STAMINA] += val;
|
||||
break;
|
||||
case ITEM_MOD_DEFENSE_SKILL_RATING:
|
||||
stats[STATS_TYPE_DEFENSE] += val;
|
||||
break;
|
||||
case ITEM_MOD_DODGE_RATING:
|
||||
stats[STATS_TYPE_DODGE] += val;
|
||||
break;
|
||||
case ITEM_MOD_PARRY_RATING:
|
||||
stats[STATS_TYPE_PARRY] += val;
|
||||
break;
|
||||
case ITEM_MOD_BLOCK_RATING:
|
||||
stats[STATS_TYPE_BLOCK_RATING] += val;
|
||||
break;
|
||||
case ITEM_MOD_HIT_MELEE_RATING:
|
||||
if (type_ == CollectorType::MELEE)
|
||||
stats[STATS_TYPE_HIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_HIT_RANGED_RATING:
|
||||
if (type_ == CollectorType::RANGED)
|
||||
stats[STATS_TYPE_HIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_HIT_SPELL_RATING:
|
||||
if (type_ == CollectorType::SPELL)
|
||||
stats[STATS_TYPE_HIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_CRIT_MELEE_RATING:
|
||||
if (type_ == CollectorType::MELEE)
|
||||
stats[STATS_TYPE_CRIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_CRIT_RANGED_RATING:
|
||||
if (type_ == CollectorType::RANGED)
|
||||
stats[STATS_TYPE_CRIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_CRIT_SPELL_RATING:
|
||||
if (type_ == CollectorType::SPELL)
|
||||
stats[STATS_TYPE_CRIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_HASTE_MELEE_RATING:
|
||||
if (type_ == CollectorType::MELEE)
|
||||
stats[STATS_TYPE_HASTE] += val;
|
||||
break;
|
||||
case ITEM_MOD_HASTE_RANGED_RATING:
|
||||
if (type_ == CollectorType::RANGED)
|
||||
stats[STATS_TYPE_HASTE] += val;
|
||||
break;
|
||||
case ITEM_MOD_HASTE_SPELL_RATING:
|
||||
if (type_ == CollectorType::SPELL)
|
||||
stats[STATS_TYPE_HASTE] += val;
|
||||
break;
|
||||
case ITEM_MOD_HIT_RATING:
|
||||
stats[STATS_TYPE_HIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_CRIT_RATING:
|
||||
stats[STATS_TYPE_CRIT] += val;
|
||||
break;
|
||||
case ITEM_MOD_RESILIENCE_RATING:
|
||||
stats[STATS_TYPE_RESILIENCE] += val;
|
||||
break;
|
||||
case ITEM_MOD_HASTE_RATING:
|
||||
stats[STATS_TYPE_HASTE] += val;
|
||||
break;
|
||||
case ITEM_MOD_EXPERTISE_RATING:
|
||||
stats[STATS_TYPE_EXPERTISE] += val;
|
||||
break;
|
||||
case ITEM_MOD_ATTACK_POWER:
|
||||
stats[STATS_TYPE_ATTACK_POWER] += val;
|
||||
break;
|
||||
case ITEM_MOD_RANGED_ATTACK_POWER:
|
||||
if (type_ == CollectorType::RANGED)
|
||||
stats[STATS_TYPE_ATTACK_POWER] += val;
|
||||
break;
|
||||
case ITEM_MOD_MANA_REGENERATION:
|
||||
stats[STATS_TYPE_MANA_REGENERATION] += val;
|
||||
break;
|
||||
case ITEM_MOD_ARMOR_PENETRATION_RATING:
|
||||
stats[STATS_TYPE_ARMOR_PENETRATION] += val;
|
||||
break;
|
||||
case ITEM_MOD_SPELL_POWER:
|
||||
stats[STATS_TYPE_SPELL_POWER] += val;
|
||||
break;
|
||||
case ITEM_MOD_HEALTH_REGEN:
|
||||
stats[STATS_TYPE_HEALTH_REGENERATION] += val;
|
||||
break;
|
||||
case ITEM_MOD_SPELL_PENETRATION:
|
||||
stats[STATS_TYPE_SPELL_PENETRATION] += val;
|
||||
break;
|
||||
case ITEM_MOD_BLOCK_VALUE:
|
||||
stats[STATS_TYPE_BLOCK_VALUE] += val;
|
||||
break;
|
||||
case ITEM_MOD_SPELL_HEALING_DONE: // deprecated
|
||||
case ITEM_MOD_SPELL_DAMAGE_DONE: // deprecated
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
76
src/factory/StatsCollector.h
Normal file
76
src/factory/StatsCollector.h
Normal file
@@ -0,0 +1,76 @@
|
||||
/*
|
||||
* Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU GPL v2 license, you may redistribute it
|
||||
* and/or modify it under version 2 of the License, or (at your option), any later version.
|
||||
*/
|
||||
|
||||
#ifndef _PLAYERBOT_STATSCOLLECTOR_H
|
||||
#define _PLAYERBOT_STATSCOLLECTOR_H
|
||||
|
||||
#include "ItemTemplate.h"
|
||||
#include "SpellInfo.h"
|
||||
|
||||
enum StatsType : uint8
|
||||
{
|
||||
// Basic stats
|
||||
STATS_TYPE_AGILITY = 0,
|
||||
STATS_TYPE_STRENGTH,
|
||||
STATS_TYPE_INTELLECT,
|
||||
STATS_TYPE_SPIRIT,
|
||||
STATS_TYPE_STAMINA,
|
||||
STATS_TYPE_HIT,
|
||||
STATS_TYPE_CRIT,
|
||||
STATS_TYPE_HASTE,
|
||||
// Stats for tank
|
||||
STATS_TYPE_ARMOR,
|
||||
STATS_TYPE_DEFENSE,
|
||||
STATS_TYPE_DODGE,
|
||||
STATS_TYPE_PARRY,
|
||||
STATS_TYPE_BLOCK_VALUE,
|
||||
STATS_TYPE_BLOCK_RATING,
|
||||
STATS_TYPE_RESILIENCE,
|
||||
STATS_TYPE_HEALTH_REGENERATION,
|
||||
// Stats for spell damage and heal
|
||||
STATS_TYPE_SPELL_POWER,
|
||||
STATS_TYPE_SPELL_PENETRATION,
|
||||
STATS_TYPE_MANA_REGENERATION,
|
||||
// Stats for physical damage and melee
|
||||
STATS_TYPE_ATTACK_POWER,
|
||||
STATS_TYPE_ARMOR_PENETRATION,
|
||||
STATS_TYPE_EXPERTISE,
|
||||
// Stats for weapon dps
|
||||
STATS_TYPE_MELEE_DPS,
|
||||
STATS_TYPE_RANGED_DPS,
|
||||
STATS_TYPE_MAX = 24
|
||||
};
|
||||
|
||||
enum class CollectorType
|
||||
{
|
||||
MELEE,
|
||||
RANGED,
|
||||
SPELL
|
||||
};
|
||||
|
||||
class StatsCollector
|
||||
{
|
||||
public:
|
||||
StatsCollector(CollectorType type);
|
||||
StatsCollector(StatsCollector& stats) = default;
|
||||
void Reset();
|
||||
void CollectItemStats(ItemTemplate const* proto);
|
||||
void CollectSpellStats(uint32 spellId, uint32 trigger = ITEM_SPELLTRIGGER_ON_EQUIP);
|
||||
void CollectSpellEffectStats(const SpellEffectInfo& effectInfo, float multiplier = 1.0f);
|
||||
void CollectEnchantStats(SpellItemEnchantmentEntry const* enchant);
|
||||
|
||||
public:
|
||||
uint32 stats[STATS_TYPE_MAX];
|
||||
|
||||
private:
|
||||
void CollectByItemStatType(uint32 itemStatType, int32 val);
|
||||
bool CollectSpecialCaseSpellStats(uint32 spellId);
|
||||
bool CollectSpecialEnchantSpellStats(uint32 enchantSpellId);
|
||||
|
||||
private:
|
||||
CollectorType type_;
|
||||
};
|
||||
|
||||
#endif
|
||||
467
src/factory/StatsWeightCalculator.cpp
Normal file
467
src/factory/StatsWeightCalculator.cpp
Normal file
@@ -0,0 +1,467 @@
|
||||
/*
|
||||
* Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU GPL v2 license, you may redistribute it
|
||||
* and/or modify it under version 2 of the License, or (at your option), any later version.
|
||||
*/
|
||||
|
||||
#include "StatsWeightCalculator.h"
|
||||
|
||||
#include <memory>
|
||||
|
||||
#include "AiFactory.h"
|
||||
#include "DBCStores.h"
|
||||
#include "ObjectMgr.h"
|
||||
#include "PlayerbotAI.h"
|
||||
#include "SharedDefines.h"
|
||||
#include "StatsCollector.h"
|
||||
#include "Unit.h"
|
||||
|
||||
StatsWeightCalculator::StatsWeightCalculator(Player* player) : player_(player)
|
||||
{
|
||||
if (PlayerbotAI::IsCaster(player))
|
||||
type_ = CollectorType::SPELL;
|
||||
else if (PlayerbotAI::IsMelee(player))
|
||||
type_ = CollectorType::MELEE;
|
||||
else
|
||||
type_ = CollectorType::RANGED;
|
||||
collector_ = std::make_unique<StatsCollector>(type_);
|
||||
|
||||
cls = player->getClass();
|
||||
tab = AiFactory::GetPlayerSpecTab(player);
|
||||
|
||||
enable_overflow_penalty_ = true;
|
||||
enable_item_set_bonus_ = true;
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::Reset()
|
||||
{
|
||||
collector_->Reset();
|
||||
weight_ = 0;
|
||||
for (uint32 i = 0; i < STATS_TYPE_MAX; i++)
|
||||
{
|
||||
stats_weights_[i] = 0;
|
||||
}
|
||||
}
|
||||
|
||||
float StatsWeightCalculator::CalculateItem(uint32 itemId)
|
||||
{
|
||||
ItemTemplate const* proto = &sObjectMgr->GetItemTemplateStore()->at(itemId);
|
||||
|
||||
if (!proto)
|
||||
return 0.0f;
|
||||
|
||||
Reset();
|
||||
|
||||
collector_->CollectItemStats(proto);
|
||||
|
||||
GenerateWeights(player_);
|
||||
for (uint32 i = 0; i < STATS_TYPE_MAX; i++)
|
||||
{
|
||||
weight_ += stats_weights_[i] * collector_->stats[i];
|
||||
}
|
||||
|
||||
CalculateItemTypePenalty(proto);
|
||||
|
||||
if (enable_item_set_bonus_)
|
||||
CalculateItemSetBonus(player_, proto);
|
||||
|
||||
CalculateSocketBonus(player_, proto);
|
||||
|
||||
// Blend with item quality and level
|
||||
weight_ *= (proto->Quality + 1) * proto->ItemLevel;
|
||||
|
||||
return weight_;
|
||||
}
|
||||
|
||||
float StatsWeightCalculator::CalculateEnchant(uint32 enchantId)
|
||||
{
|
||||
SpellItemEnchantmentEntry const* enchant = sSpellItemEnchantmentStore.LookupEntry(enchantId);
|
||||
|
||||
if (!enchant)
|
||||
return 0.0f;
|
||||
|
||||
Reset();
|
||||
|
||||
collector_->CollectEnchantStats(enchant);
|
||||
|
||||
GenerateWeights(player_);
|
||||
for (uint32 i = 0; i < STATS_TYPE_MAX; i++)
|
||||
{
|
||||
weight_ += stats_weights_[i] * collector_->stats[i];
|
||||
}
|
||||
|
||||
return weight_;
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::GenerateWeights(Player* player)
|
||||
{
|
||||
GenerateBasicWeights(player);
|
||||
GenerateAdditionalWeights(player);
|
||||
|
||||
if (enable_overflow_penalty_)
|
||||
ApplyOverflowPenalty(player);
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::GenerateBasicWeights(Player* player)
|
||||
{
|
||||
// Basic weights
|
||||
stats_weights_[STATS_TYPE_STAMINA] += 0.01f;
|
||||
stats_weights_[STATS_TYPE_ARMOR] += 0.001f;
|
||||
|
||||
if (cls == CLASS_HUNTER)
|
||||
{
|
||||
stats_weights_[STATS_TYPE_AGILITY] += 2.5f;
|
||||
stats_weights_[STATS_TYPE_ATTACK_POWER] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_ARMOR_PENETRATION] += 2.5f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_RANGED_DPS] += 5.0f;
|
||||
}
|
||||
else if (cls == CLASS_ROGUE || (cls == CLASS_DRUID && tab == DRUID_TAB_FERAL && !PlayerbotAI::IsTank(player)))
|
||||
{
|
||||
stats_weights_[STATS_TYPE_AGILITY] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_STRENGTH] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_ATTACK_POWER] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_ARMOR_PENETRATION] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] += 2.5f;
|
||||
stats_weights_[STATS_TYPE_MELEE_DPS] += 5.0f;
|
||||
}
|
||||
else if ((cls == CLASS_PALADIN && tab == 2) || // retribution
|
||||
(cls == CLASS_WARRIOR && tab != 2) || // arm / fury
|
||||
(cls == CLASS_DEATH_KNIGHT && tab != 0) // ice / unholy
|
||||
)
|
||||
{
|
||||
stats_weights_[STATS_TYPE_AGILITY] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_STRENGTH] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_ATTACK_POWER] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_ARMOR_PENETRATION] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_MELEE_DPS] += 5.0f;
|
||||
}
|
||||
else if ((cls == CLASS_SHAMAN && tab == SHAMAN_TAB_ENHANCEMENT)) // enhancement
|
||||
{
|
||||
stats_weights_[STATS_TYPE_AGILITY] += 1.6f;
|
||||
stats_weights_[STATS_TYPE_STRENGTH] += 1.1f;
|
||||
stats_weights_[STATS_TYPE_INTELLECT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_ATTACK_POWER] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_SPELL_POWER] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_ARMOR_PENETRATION] += 1.2f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 1.7f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 1.4f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 1.8f;
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_MELEE_DPS] += 5.0f;
|
||||
}
|
||||
else if (cls == CLASS_WARLOCK || cls == CLASS_MAGE ||
|
||||
(cls == CLASS_PRIEST && tab == PRIEST_TAB_SHADOW) || // shadow
|
||||
(cls == CLASS_SHAMAN && tab == SHAMAN_TAB_ELEMENTAL) || // element
|
||||
(cls == CLASS_DRUID && tab == DRUID_TAB_BALANCE)) // balance
|
||||
{
|
||||
stats_weights_[STATS_TYPE_INTELLECT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_SPIRIT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_SPELL_POWER] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_SPELL_PENETRATION] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_RANGED_DPS] += 1.0f;
|
||||
}
|
||||
else if ((cls == CLASS_PALADIN && tab == PALADIN_TAB_HOLY) || // holy
|
||||
(cls == CLASS_PRIEST && tab != PRIEST_TAB_SHADOW) || // discipline / holy
|
||||
(cls == CLASS_SHAMAN && tab == SHAMAN_TAB_RESTORATION) || // heal
|
||||
(cls == CLASS_DRUID && tab == DRUID_TAB_RESTORATION))
|
||||
{
|
||||
stats_weights_[STATS_TYPE_INTELLECT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_SPIRIT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_SPELL_POWER] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_MANA_REGENERATION] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_RANGED_DPS] += 1.0f;
|
||||
}
|
||||
else if ((cls == CLASS_WARRIOR && tab == 2) || (cls == CLASS_PALADIN && tab == 1))
|
||||
{
|
||||
stats_weights_[STATS_TYPE_AGILITY] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_STRENGTH] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_STAMINA] += 3.0f;
|
||||
stats_weights_[STATS_TYPE_ATTACK_POWER] += 0.2f;
|
||||
stats_weights_[STATS_TYPE_DEFENSE] += 2.5f;
|
||||
stats_weights_[STATS_TYPE_PARRY] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_DODGE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_RESILIENCE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_BLOCK_RATING] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_BLOCK_VALUE] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_ARMOR] += 0.3f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 0.2f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] += 3.0f;
|
||||
stats_weights_[STATS_TYPE_MELEE_DPS] += 2.0f;
|
||||
}
|
||||
else if (cls == CLASS_DEATH_KNIGHT && tab == 0)
|
||||
{
|
||||
stats_weights_[STATS_TYPE_AGILITY] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_STRENGTH] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_STAMINA] += 2.5f;
|
||||
stats_weights_[STATS_TYPE_ATTACK_POWER] += 0.2f;
|
||||
stats_weights_[STATS_TYPE_DEFENSE] += 3.5f;
|
||||
stats_weights_[STATS_TYPE_PARRY] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_DODGE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_RESILIENCE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_ARMOR] += 0.3f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] += 3.5f;
|
||||
stats_weights_[STATS_TYPE_MELEE_DPS] += 2.0f;
|
||||
}
|
||||
else
|
||||
{
|
||||
// BEAR DRUID TANK
|
||||
stats_weights_[STATS_TYPE_AGILITY] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_STRENGTH] += 1.0f;
|
||||
stats_weights_[STATS_TYPE_STAMINA] += 1.5f;
|
||||
stats_weights_[STATS_TYPE_ATTACK_POWER] += 0.2f;
|
||||
stats_weights_[STATS_TYPE_DEFENSE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_DODGE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_RESILIENCE] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_ARMOR] += 0.3f;
|
||||
stats_weights_[STATS_TYPE_HIT] += 2.0f;
|
||||
stats_weights_[STATS_TYPE_CRIT] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_HASTE] += 0.5f;
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] += 3.0f;
|
||||
stats_weights_[STATS_TYPE_MELEE_DPS] += 2.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::GenerateAdditionalWeights(Player* player)
|
||||
{
|
||||
uint8 cls = player->getClass();
|
||||
// int tab = AiFactory::GetPlayerSpecTab(player);
|
||||
if (cls == CLASS_HUNTER)
|
||||
{
|
||||
if (player->HasAura(34484))
|
||||
stats_weights_[STATS_TYPE_INTELLECT] += 1.0f;
|
||||
}
|
||||
else if (cls == CLASS_WARRIOR)
|
||||
{
|
||||
if (player->HasAura(61222))
|
||||
stats_weights_[STATS_TYPE_ARMOR] += 0.03f;
|
||||
}
|
||||
else if (cls == CLASS_SHAMAN)
|
||||
{
|
||||
if (player->HasAura(51885))
|
||||
stats_weights_[STATS_TYPE_INTELLECT] += 1.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::CalculateItemSetBonus(Player* player, ItemTemplate const* proto)
|
||||
{
|
||||
uint32 itemSet = proto->ItemSet;
|
||||
if (!itemSet)
|
||||
return;
|
||||
|
||||
float multiplier = 1.0f;
|
||||
size_t i = 0;
|
||||
for (i = 0; i < player->ItemSetEff.size(); i++)
|
||||
{
|
||||
if (player->ItemSetEff[i])
|
||||
{
|
||||
ItemSetEffect* eff = player->ItemSetEff[i];
|
||||
|
||||
uint32 setId = eff->setid;
|
||||
if (itemSet != setId)
|
||||
continue;
|
||||
|
||||
const ItemSetEntry *setEntry = sItemSetStore.LookupEntry(setId);
|
||||
if (!setEntry)
|
||||
continue;
|
||||
|
||||
uint32 itemCount = eff->item_count;
|
||||
uint32 max_items = 0;
|
||||
for (size_t j = 0; j < MAX_ITEM_SET_SPELLS; j++)
|
||||
max_items = std::max(max_items, setEntry->items_to_triggerspell[j]);
|
||||
if (itemCount < max_items)
|
||||
{
|
||||
multiplier += 0.1f * itemCount; // 10% bonus for each item already equipped
|
||||
}
|
||||
else
|
||||
{
|
||||
multiplier = 1.0f; // All item set effect has been triggerred
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (i == player->ItemSetEff.size())
|
||||
multiplier = 1.05f; // this is the first item in the item set
|
||||
|
||||
weight_ *= multiplier;
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::CalculateSocketBonus(Player* player, ItemTemplate const* proto)
|
||||
{
|
||||
uint32 socketNum = 0;
|
||||
for (uint32 enchant_slot = SOCK_ENCHANTMENT_SLOT; enchant_slot < SOCK_ENCHANTMENT_SLOT + MAX_GEM_SOCKETS; ++enchant_slot)
|
||||
{
|
||||
uint8 socketColor = proto->Socket[enchant_slot - SOCK_ENCHANTMENT_SLOT].Color;
|
||||
|
||||
if (!socketColor) // no socket slot
|
||||
continue;
|
||||
|
||||
socketNum++;
|
||||
}
|
||||
|
||||
float multiplier = 1.0f + socketNum * 0.03f; // 3% bonus for socket
|
||||
|
||||
weight_ *= multiplier;
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::CalculateItemTypePenalty(ItemTemplate const* proto)
|
||||
{
|
||||
// penalty for different type armor
|
||||
if (proto->Class == ITEM_CLASS_ARMOR && proto->SubClass >= ITEM_SUBCLASS_ARMOR_CLOTH &&
|
||||
proto->SubClass <= ITEM_SUBCLASS_ARMOR_PLATE && NotBestArmorType(proto->SubClass))
|
||||
{
|
||||
weight_ *= 0.8;
|
||||
}
|
||||
// double hand
|
||||
if (proto->Class == ITEM_CLASS_WEAPON)
|
||||
{
|
||||
bool isDoubleHand = proto->Class == ITEM_CLASS_WEAPON &&
|
||||
!(ITEM_SUBCLASS_MASK_SINGLE_HAND & (1 << proto->SubClass)) &&
|
||||
!(ITEM_SUBCLASS_MASK_WEAPON_RANGED & (1 << proto->SubClass));
|
||||
|
||||
if (isDoubleHand)
|
||||
{
|
||||
weight_ *= 0.5;
|
||||
}
|
||||
// spec without double hand
|
||||
// enhancement, rogue, ice dk, unholy dk, shield tank, fury warrior without titan's grip but with duel wield
|
||||
if (isDoubleHand &&
|
||||
((cls == CLASS_SHAMAN && tab == SHAMAN_TAB_ENHANCEMENT && player_->CanDualWield()) ||
|
||||
(cls == CLASS_ROGUE) || (cls == CLASS_DEATH_KNIGHT && tab != DEATHKNIGT_TAB_BLOOD) ||
|
||||
(cls == CLASS_WARRIOR && tab == WARRIOR_TAB_FURY && !player_->CanTitanGrip() && player_->CanDualWield()) ||
|
||||
(cls == CLASS_WARRIOR && tab == WARRIOR_TAB_PROTECTION) ||
|
||||
(cls == CLASS_PALADIN && tab == PALADIN_TAB_PROTECTION)))
|
||||
{
|
||||
weight_ *= 0.1;
|
||||
}
|
||||
// spec with double hand
|
||||
// fury without duel wield, arms, bear, retribution, blood dk
|
||||
if (isDoubleHand &&
|
||||
((cls == CLASS_WARRIOR && tab == WARRIOR_TAB_FURY && !player_->CanDualWield()) ||
|
||||
(cls == CLASS_WARRIOR && tab == WARRIOR_TAB_ARMS) || (cls == CLASS_DRUID && tab == DRUID_TAB_FERAL) ||
|
||||
(cls == CLASS_PALADIN && tab == PALADIN_TAB_RETRIBUTION) ||
|
||||
(cls == CLASS_DEATH_KNIGHT && tab == DEATHKNIGT_TAB_BLOOD) ||
|
||||
(cls == CLASS_SHAMAN && tab == SHAMAN_TAB_ENHANCEMENT && !player_->CanDualWield())))
|
||||
{
|
||||
weight_ *= 10;
|
||||
}
|
||||
// fury with titan's grip
|
||||
if (isDoubleHand && proto->SubClass != ITEM_SUBCLASS_WEAPON_POLEARM &&
|
||||
(cls == CLASS_WARRIOR && tab == WARRIOR_TAB_FURY && player_->CanTitanGrip()))
|
||||
{
|
||||
weight_ *= 10;
|
||||
}
|
||||
}
|
||||
if (proto->Class == ITEM_CLASS_WEAPON)
|
||||
{
|
||||
if (cls == CLASS_HUNTER && proto->SubClass == ITEM_SUBCLASS_WEAPON_THROWN)
|
||||
{
|
||||
weight_ *= 0.1;
|
||||
}
|
||||
if (cls == CLASS_ROGUE && tab == ROGUE_TAB_ASSASSINATION && proto->SubClass != ITEM_SUBCLASS_WEAPON_DAGGER)
|
||||
{
|
||||
weight_ *= 0.1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool StatsWeightCalculator::NotBestArmorType(uint32 item_subclass_armor)
|
||||
{
|
||||
if (player_->HasSkill(SKILL_PLATE_MAIL))
|
||||
{
|
||||
return item_subclass_armor != ITEM_SUBCLASS_ARMOR_PLATE;
|
||||
}
|
||||
if (player_->HasSkill(SKILL_MAIL))
|
||||
{
|
||||
return item_subclass_armor != ITEM_SUBCLASS_ARMOR_MAIL;
|
||||
}
|
||||
if (player_->HasSkill(SKILL_LEATHER))
|
||||
{
|
||||
return item_subclass_armor != ITEM_SUBCLASS_ARMOR_LEATHER;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
void StatsWeightCalculator::ApplyOverflowPenalty(Player* player)
|
||||
{
|
||||
{
|
||||
float hit_current, hit_overflow;
|
||||
if (type_ == CollectorType::SPELL)
|
||||
{
|
||||
hit_current = player->GetRatingBonusValue(CR_HIT_SPELL);
|
||||
hit_overflow = SPELL_HIT_OVERFLOW;
|
||||
}
|
||||
else if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
hit_current = player->GetRatingBonusValue(CR_HIT_MELEE);
|
||||
hit_overflow = MELEE_HIT_OVERFLOW;
|
||||
}
|
||||
else
|
||||
{
|
||||
hit_current = player->GetRatingBonusValue(CR_HIT_RANGED);
|
||||
hit_overflow = RANGED_HIT_OVERFLOW;
|
||||
}
|
||||
if (hit_current >= hit_overflow)
|
||||
stats_weights_[STATS_TYPE_HIT] = 0.0f;
|
||||
else if (hit_current >= hit_overflow * 0.8)
|
||||
stats_weights_[STATS_TYPE_HIT] /= 1.5;
|
||||
}
|
||||
|
||||
{
|
||||
if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
float expertise_current, expertise_overflow;
|
||||
expertise_current = player->GetRatingBonusValue(CR_EXPERTISE);
|
||||
expertise_overflow = EXPERTISE_OVERFLOW;
|
||||
if (expertise_current >= expertise_overflow)
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] = 0.0f;
|
||||
else if (expertise_current >= expertise_overflow * 0.8)
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] /= 1.5;
|
||||
}
|
||||
}
|
||||
|
||||
{
|
||||
if (type_ == CollectorType::MELEE)
|
||||
{
|
||||
float defense_current, defense_overflow;
|
||||
defense_current = player->GetRatingBonusValue(CR_DEFENSE_SKILL);
|
||||
defense_overflow = EXPERTISE_OVERFLOW;
|
||||
if (defense_current >= defense_overflow)
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] /= 2;
|
||||
else if (defense_current >= defense_overflow * 0.8)
|
||||
stats_weights_[STATS_TYPE_EXPERTISE] /= 1.5;
|
||||
}
|
||||
}
|
||||
|
||||
{
|
||||
if (type_ == CollectorType::MELEE || type_ == CollectorType::RANGED)
|
||||
{
|
||||
float armor_pnetration_current, armor_pnetration_overflow;
|
||||
armor_pnetration_current = player->GetRatingBonusValue(CR_ARMOR_PENETRATION);
|
||||
armor_pnetration_overflow = EXPERTISE_OVERFLOW;
|
||||
if (armor_pnetration_current >= armor_pnetration_overflow)
|
||||
stats_weights_[STATS_TYPE_ARMOR_PENETRATION] = 0.0f;
|
||||
if (armor_pnetration_current >= armor_pnetration_overflow * 0.8)
|
||||
stats_weights_[STATS_TYPE_ARMOR_PENETRATION] /= 1.5;
|
||||
}
|
||||
}
|
||||
}
|
||||
60
src/factory/StatsWeightCalculator.h
Normal file
60
src/factory/StatsWeightCalculator.h
Normal file
@@ -0,0 +1,60 @@
|
||||
/*
|
||||
* Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU GPL v2 license, you may redistribute it
|
||||
* and/or modify it under version 2 of the License, or (at your option), any later version.
|
||||
*/
|
||||
|
||||
#ifndef _PLAYERBOT_GEARSCORECALCULATOR_H
|
||||
#define _PLAYERBOT_GEARSCORECALCULATOR_H
|
||||
|
||||
#include "Player.h"
|
||||
#include "StatsCollector.h"
|
||||
|
||||
#define ITEM_SUBCLASS_MASK_SINGLE_HAND \
|
||||
((1 << ITEM_SUBCLASS_WEAPON_AXE) | (1 << ITEM_SUBCLASS_WEAPON_MACE) | (1 << ITEM_SUBCLASS_WEAPON_SWORD) | \
|
||||
(1 << ITEM_SUBCLASS_WEAPON_DAGGER) | (1 << ITEM_SUBCLASS_WEAPON_FIST))
|
||||
|
||||
enum StatsOverflowThreshold {
|
||||
SPELL_HIT_OVERFLOW = 17,
|
||||
MELEE_HIT_OVERFLOW = 8,
|
||||
RANGED_HIT_OVERFLOW = 8,
|
||||
EXPERTISE_OVERFLOW = 26,
|
||||
DEFENSE_OVERFLOW = 140,
|
||||
ARMOR_PENETRATION_OVERFLOW = 100
|
||||
};
|
||||
|
||||
class StatsWeightCalculator
|
||||
{
|
||||
public:
|
||||
StatsWeightCalculator(Player* player);
|
||||
void Reset();
|
||||
float CalculateItem(uint32 itemId);
|
||||
float CalculateEnchant(uint32 enchantId);
|
||||
|
||||
private:
|
||||
void GenerateWeights(Player* player);
|
||||
void GenerateBasicWeights(Player* player);
|
||||
void GenerateAdditionalWeights(Player* player);
|
||||
|
||||
void CalculateItemSetBonus(Player* player, ItemTemplate const* proto);
|
||||
void CalculateSocketBonus(Player* player, ItemTemplate const* proto);
|
||||
|
||||
void CalculateItemTypePenalty(ItemTemplate const* proto);
|
||||
|
||||
bool NotBestArmorType(uint32 item_subclass_armor);
|
||||
|
||||
void ApplyOverflowPenalty(Player* player);
|
||||
|
||||
private:
|
||||
Player* player_;
|
||||
CollectorType type_;
|
||||
std::unique_ptr<StatsCollector> collector_;
|
||||
uint8 cls;
|
||||
int tab;
|
||||
bool enable_overflow_penalty_;
|
||||
bool enable_item_set_bonus_;
|
||||
|
||||
float weight_;
|
||||
float stats_weights_[STATS_TYPE_MAX];
|
||||
};
|
||||
|
||||
#endif
|
||||
@@ -18,8 +18,6 @@ bool ChangeTalentsAction::Execute(Event event)
|
||||
|
||||
std::ostringstream out;
|
||||
|
||||
TalentSpec botSpec(bot);
|
||||
|
||||
if (!param.empty())
|
||||
{
|
||||
if (param.find("help") != std::string::npos)
|
||||
|
||||
@@ -13,6 +13,7 @@
|
||||
#include "Playerbots.h"
|
||||
#include "RandomItemMgr.h"
|
||||
#include "ServerFacade.h"
|
||||
#include "StatsWeightCalculator.h"
|
||||
|
||||
ItemUsage ItemUsageValue::Calculate()
|
||||
{
|
||||
@@ -190,7 +191,8 @@ ItemUsage ItemUsageValue::QueryItemUsageForEquip(ItemTemplate const* itemProto)
|
||||
|
||||
bool shouldEquip = false;
|
||||
// uint32 statWeight = sRandomItemMgr->GetLiveStatWeight(bot, itemProto->ItemId);
|
||||
float itemScore = PlayerbotFactory::CalculateItemScore(itemProto->ItemId, bot);
|
||||
StatsWeightCalculator calculator(bot);
|
||||
float itemScore = calculator.CalculateItem(itemProto->ItemId);
|
||||
if (itemScore)
|
||||
shouldEquip = true;
|
||||
|
||||
@@ -214,7 +216,7 @@ ItemUsage ItemUsageValue::QueryItemUsageForEquip(ItemTemplate const* itemProto)
|
||||
}
|
||||
|
||||
ItemTemplate const* oldItemProto = oldItem->GetTemplate();
|
||||
float oldScore = PlayerbotFactory::CalculateItemScore(oldItemProto->ItemId, bot);
|
||||
float oldScore = calculator.CalculateItem(oldItemProto->ItemId);
|
||||
if (oldItem)
|
||||
{
|
||||
// uint32 oldStatWeight = sRandomItemMgr->GetLiveStatWeight(bot, oldItemProto->ItemId);
|
||||
|
||||
Reference in New Issue
Block a user