diff --git a/src/Library/CreatureData/LearnableAttacks.cpp b/src/Library/CreatureData/LearnableAttacks.cpp index a5587f7..7234941 100644 --- a/src/Library/CreatureData/LearnableAttacks.cpp +++ b/src/Library/CreatureData/LearnableAttacks.cpp @@ -2,7 +2,7 @@ using namespace CreatureLib::Library; -void LearnableAttacks::AddLevelMove(uint8_t level, const AttackData* attack) { +void LearnableAttacks::AddLevelAttack(uint8_t level, const AttackData* attack) { List levelData; if (_learnedByLevel.TryGet(level, levelData)) { @@ -11,6 +11,7 @@ void LearnableAttacks::AddLevelMove(uint8_t level, const AttackData* attack) { levelData = {attack}; _learnedByLevel.Insert(level, levelData); } + _distinctAttacks.insert(attack); } const List& LearnableAttacks::GetAttacksForLevel(uint8_t level) const { diff --git a/src/Library/CreatureData/LearnableAttacks.hpp b/src/Library/CreatureData/LearnableAttacks.hpp index 60b0315..e55ff8e 100644 --- a/src/Library/CreatureData/LearnableAttacks.hpp +++ b/src/Library/CreatureData/LearnableAttacks.hpp @@ -12,6 +12,7 @@ using namespace Arbutils::Collections; namespace CreatureLib::Library { class LearnableAttacks { + protected: Dictionary> _learnedByLevel; std::unordered_set _distinctAttacks; @@ -28,7 +29,7 @@ namespace CreatureLib::Library { virtual ~LearnableAttacks() = default; - void AddLevelMove(uint8_t level, const AttackData* attack); + void AddLevelAttack(uint8_t level, const AttackData* attack); const List& GetAttacksForLevel(uint8_t level) const;