Implements GetRandomAttack method on LearnableAttacks.
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2020-04-19 17:42:32 +02:00
parent c6ce5fe8a7
commit ad9cbf0670
4 changed files with 13 additions and 44 deletions

View File

@@ -4,6 +4,7 @@
#include <Arbutils/Assert.hpp>
#include <Arbutils/Collections/Dictionary.hpp>
#include <Arbutils/Collections/List.hpp>
#include <Arbutils/Random.hpp>
#include <unordered_map>
#include "../Attacks/AttackData.hpp"
@@ -12,19 +13,29 @@ using namespace Arbutils::Collections;
namespace CreatureLib::Library {
class LearnableAttacks {
Dictionary<uint8_t, List<const AttackData*>> _learnedByLevel;
std::unordered_set<const AttackData*> _distinctAttacks;
public:
LearnableAttacks(size_t levelAttackCapacity)
explicit LearnableAttacks(size_t levelAttackCapacity)
: _learnedByLevel(Dictionary<uint8_t, List<const AttackData*>>(levelAttackCapacity)) {
for (auto kv : _learnedByLevel) {
for (auto attack : kv.second)
for (auto attack : kv.second) {
AssertNotNull(attack)
_distinctAttacks.insert(attack);
}
}
}
virtual ~LearnableAttacks() = default;
void AddLevelMove(uint8_t level, const AttackData* attack);
const List<const AttackData*>& GetAttacksForLevel(uint8_t level) const;
virtual const AttackData* GetRandomAttack(Arbutils::Random rand = Arbutils::Random()) {
auto val = rand.Get(_distinctAttacks.size());
return *std::next(_distinctAttacks.begin(), val);
}
};
}