67 lines
4.0 KiB
C++
67 lines
4.0 KiB
C++
|
#ifndef POKEMONSCRIPTTESTER_POKEMONBUILDER_HPP
|
||
|
#define POKEMONSCRIPTTESTER_POKEMONBUILDER_HPP
|
||
|
#include <PkmnLib/Battling/Pokemon/CreatePokemon.hpp>
|
||
|
#include <angelscript.h>
|
||
|
|
||
|
class PokemonBuilderRegistration {
|
||
|
public:
|
||
|
static PkmnLib::Battling::CreatePokemon* CreatePokemonBuilder(const ArbUt::StringView& species, uint8_t level) {
|
||
|
auto* ctx = asGetActiveContext();
|
||
|
TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData(684));
|
||
|
auto lib = Globals::Library.GetValue();
|
||
|
|
||
|
auto builder = new PkmnLib::Battling::CreatePokemon(lib, species, level);
|
||
|
env->AddGarbage(builder);
|
||
|
return builder;
|
||
|
}
|
||
|
|
||
|
static PkmnLib::Battling::CreatePokemon& WithGenderWrapper(PkmnLib::Battling::CreatePokemon* builder, i32 gender) {
|
||
|
return builder->WithGender((CreatureLib::Library::Gender)gender);
|
||
|
}
|
||
|
|
||
|
static PkmnLib::Battling::Pokemon* BuildWrapper(PkmnLib::Battling::CreatePokemon* builder, u32 seed) {
|
||
|
auto rand = ArbUt::Random(seed);
|
||
|
auto p = builder->Build(rand);
|
||
|
auto* ctx = asGetActiveContext();
|
||
|
TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData(684));
|
||
|
env->AddGarbage(p);
|
||
|
return p;
|
||
|
}
|
||
|
|
||
|
static void Register(AngelScriptResolver* resolver) {
|
||
|
auto engine = resolver->GetEngine();
|
||
|
Ensure(engine->RegisterObjectType("PokemonBuilder", 0, asOBJ_REF | asOBJ_NOCOUNT) >= 0);
|
||
|
Ensure(engine->RegisterGlobalFunction(
|
||
|
"PokemonBuilder@ CreatePokemonBuilder(const constString&in species, uint8 level)",
|
||
|
asFUNCTION(CreatePokemonBuilder), asCALL_CDECL) >= 0);
|
||
|
|
||
|
Ensure(engine->RegisterObjectMethod("PokemonBuilder", "PokemonBuilder@ WithForme(const constString&in forme)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, WithForme), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod("PokemonBuilder", "PokemonBuilder@ WithGender(Gender gender)",
|
||
|
asFUNCTION(WithGenderWrapper), asCALL_CDECL_OBJFIRST));
|
||
|
Ensure(engine->RegisterObjectMethod("PokemonBuilder", "PokemonBuilder@ IsShiny(bool value)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, IsShiny), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod("PokemonBuilder", "PokemonBuilder@ WithHeldItem(const constString&in item)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, WithHeldItem), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod(
|
||
|
"PokemonBuilder", "PokemonBuilder@ LearnMove(const constString&in move, MoveLearnMethod method)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, LearnMove), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod(
|
||
|
"PokemonBuilder",
|
||
|
"PokemonBuilder@ WithIndividualValues(uint8 hp,uint8 att,uint8 def,uint8 spa,uint8 spd,uint8 speed)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, WithIndividualValues), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod(
|
||
|
"PokemonBuilder",
|
||
|
"PokemonBuilder@ WithEffortValues(uint8 hp,uint8 att,uint8 def,uint8 spa,uint8 spd,uint8 speed)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, WithEffortValues), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod("PokemonBuilder", "PokemonBuilder@ WithNature(const constString&in nature)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, WithNature), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod("PokemonBuilder", "PokemonBuilder@ WithNickname(const string&in name)",
|
||
|
asMETHOD(PkmnLib::Battling::CreatePokemon, WithNickname), asCALL_THISCALL));
|
||
|
Ensure(engine->RegisterObjectMethod("PokemonBuilder", "Pokemon@ Build(uint seed = 0)", asFUNCTION(BuildWrapper),
|
||
|
asCALL_CDECL_OBJFIRST));
|
||
|
}
|
||
|
};
|
||
|
|
||
|
#endif // POKEMONSCRIPTTESTER_POKEMONBUILDER_HPP
|