PkmnLib/src/Battling/Pokemon/Pokemon.hpp

#ifndef PKMNLIB_POKEMON_HPP
#define PKMNLIB_POKEMON_HPP

#include <CreatureLib/Battling/Models/Creature.hpp>
#include <utility>
#include "../../Library/Statistic.hpp"
#include "../Library/BattleLibrary.hpp"
#include "LearnedMove.hpp"

namespace PkmnLib::Battling {
    class Pokemon : public CreatureLib::Battling::Creature {
    private:
        CreatureLib::Library::StatisticSet<uint8_t> _individualValues;
        CreatureLib::Library::StatisticSet<uint8_t> _effortValues;

        ArbUt::BorrowedPtr<const PkmnLib::Library::Nature> _nature;
        std::unique_ptr<CreatureLib::Battling::Script> _statusScript = nullptr;

    public:
        Pokemon(ArbUt::BorrowedPtr<const BattleLibrary> library,
                const ArbUt::BorrowedPtr<const Library::PokemonSpecies>& species,
                const ArbUt::BorrowedPtr<const Library::PokemonForme>& forme, uint8_t level, uint32_t experience,
                uint32_t uid, CreatureLib::Library::Gender gender, uint8_t coloring,
                ArbUt::BorrowedPtr<const Library::Item> heldItem, const std::string& nickname,
                const CreatureLib::Library::TalentIndex& talent,
                const std::vector<CreatureLib::Battling::LearnedAttack*>& moves,
                CreatureLib::Library::StatisticSet<uint8_t> individualValues,
                CreatureLib::Library::StatisticSet<uint8_t> effortValues,
                ArbUt::BorrowedPtr<const PkmnLib::Library::Nature> nature, bool allowedExperienceGain = true)
            : CreatureLib::Battling::Creature(library.ForceAs<const CreatureLib::Battling::BattleLibrary>(),
                                              species.ForceAs<const CreatureLib::Library::CreatureSpecies>(),
                                              forme.As<const CreatureLib::Library::SpeciesVariant>(), level, experience,
                                              uid, gender, coloring, heldItem.As<const CreatureLib::Library::Item>(),
                                              nickname, talent, moves, allowedExperienceGain),
              _individualValues(individualValues), _effortValues(effortValues), _nature(nature) {}

        const ArbUt::BorrowedPtr<const Library::PokemonForme> GetForme() const {
            return _variant.As<const Library::PokemonForme>();
        }

        inline bool IsShiny() const noexcept { return _coloring == 1; }

        const ArbUt::List<LearnedMove*>& GetMoves() const {
            return reinterpret_cast<const ArbUt::List<LearnedMove*>&>(_attacks);
        }

        inline const ArbUt::BorrowedPtr<const PkmnLib::Library::Nature>& GetNature() const noexcept { return _nature; }
        inline uint8_t GetIndividualValue(CreatureLib::Library::Statistic stat) const {
            return _individualValues.GetStat(stat);
        }
        inline uint8_t GetEffortValue(CreatureLib::Library::Statistic stat) const {
            return _effortValues.GetStat(stat);
        }

        inline ArbUt::BorrowedPtr<const PkmnLib::Library::PokemonSpecies> GetPokemonSpecies() const noexcept {
            return _species.As<const PkmnLib::Library::PokemonSpecies>();
        }

        void Evolve(ArbUt::BorrowedPtr<const Library::PokemonSpecies> mon,
                    ArbUt::BorrowedPtr<const Library::PokemonForme> forme);

        void SetStatus(const ArbUt::StringView& name);
        void ClearStatus();
        const ArbUt::StringView& GetStatusName() noexcept {
            if (_statusScript == nullptr)
                return ArbUt::StringView::Empty();
            return _statusScript->GetName();
        }
    };
}

#endif // PKMNLIB_POKEMON_HPP