#include "CreatureSpecies.hpp"

using namespace CreatureLib::Library;

CreatureSpecies::CreatureSpecies(uint16_t id, std::string name, const SpeciesVariant* defaultVariant, float genderRatio,
                                 std::string growthRate, uint8_t captureRate)
    : _id(id), _genderRate(genderRatio), _growthRate(std::move(growthRate)), _captureRate(captureRate),
      _variants({{"default", defaultVariant}}), _name(std::move(name)) {}

bool CreatureSpecies::HasVariant(const std::string& name) const {
    auto key = name;
    std::transform(key.begin(), key.end(), key.begin(), ::tolower);
    return _variants.find(key) != _variants.end();
}

bool CreatureSpecies::TryGetVariant(const std::string& name, const SpeciesVariant*& out) const {
    auto key = name;
    std::transform(key.begin(), key.end(), key.begin(), ::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.begin(), ::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.begin(), ::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
    auto val = rand.GetDouble();
    if (val >= this->_genderRate)
        return Gender ::Female;
    return Gender ::Male;
}

const std::string& CreatureSpecies::GetName() const { return _name; }