From a5806492243120e402c1ef9f6ae7a755410fcddf Mon Sep 17 00:00:00 2001
From: Deukhoofd <deukhoofd@gmail.com>
Date: Sun, 19 Apr 2020 19:11:48 +0200
Subject: [PATCH] Ensure AddLevelAttack adds to distinctMoves

---
 src/Library/CreatureData/LearnableAttacks.cpp | 3 ++-
 src/Library/CreatureData/LearnableAttacks.hpp | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

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<const AttackData*> 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<const AttackData*>& 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<uint8_t, List<const AttackData*>> _learnedByLevel;
         std::unordered_set<const AttackData*> _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<const AttackData*>& GetAttacksForLevel(uint8_t level) const;