diff --git a/src/Library/CreatureData/CreatureSpecies.cpp b/src/Library/CreatureData/CreatureSpecies.cpp index b5e2dde..16b3fb5 100644 --- a/src/Library/CreatureData/CreatureSpecies.cpp +++ b/src/Library/CreatureData/CreatureSpecies.cpp @@ -7,7 +7,38 @@ CreatureSpecies::CreatureSpecies(uint16_t id, std::string name, const SpeciesVar : _id(id), _genderRate(genderRatio), _growthRate(std::move(growthRate)), _captureRate(captureRate), _variants({{"default", defaultVariant}}), _name(std::move(name)) {} -const SpeciesVariant* CreatureSpecies::GetVariant(const std::string& key) const { return _variants.at(key); } +bool CreatureSpecies::HasVariant(const std::string& name) const { + auto key = name; + std::transform(key.begin(), key.end(), key.end(), ::tolower); + return _variants.find(key) != _variants.end(); +} + +bool CreatureSpecies::TryGetVariant(const std::string& name, const SpeciesVariant*& out) { + auto key = name; + std::transform(key.begin(), key.end(), key.end(), ::tolower); + auto find = _variants.find(key); + if (find != _variants.end()) { + out = find->second; + return true; + } + return false; +} + +const SpeciesVariant* CreatureSpecies::GetVariant(const std::string& name) const { + auto key = name; + std::transform(key.begin(), key.end(), key.end(), ::tolower); + return _variants.at(key); +} + +void CreatureSpecies::SetVariant(const std::string& name, const SpeciesVariant* variant) { + auto key = name; + std::transform(key.begin(), key.end(), key.end(), ::tolower); + auto find = _variants.find(key); + if (find != _variants.end()) { + delete find->second; + } + _variants[key] = variant; +} Gender CreatureSpecies::GetRandomGender(CreatureLib::Core::Random& rand) const { // TODO: Genderless creatures diff --git a/src/Library/CreatureData/CreatureSpecies.hpp b/src/Library/CreatureData/CreatureSpecies.hpp index 9fd4bf2..2c62beb 100644 --- a/src/Library/CreatureData/CreatureSpecies.hpp +++ b/src/Library/CreatureData/CreatureSpecies.hpp @@ -36,17 +36,13 @@ namespace CreatureLib::Library { inline const std::string& GetGrowthRate() const { return _growthRate; } inline uint8_t GetCaptureRate() const { return _captureRate; } + [[nodiscard]] bool HasVariant(const std::string& key) const; + [[nodiscard]] bool TryGetVariant(const std::string& name, const SpeciesVariant*& out); [[nodiscard]] const SpeciesVariant* GetVariant(const std::string& key) const; [[nodiscard]] Gender GetRandomGender(Core::Random& rand) const; [[nodiscard]] const std::string& GetName() const; - void SetVariant(const std::string& name, const SpeciesVariant* variant) { - auto find = _variants.find(name); - if (find != _variants.end()) { - delete find->second; - } - _variants[name] = variant; - } + void SetVariant(const std::string& name, const SpeciesVariant* variant); }; }