111 lines
4.6 KiB
C++
111 lines
4.6 KiB
C++
#ifndef POKEMONSCRIPTTESTER_BATTLEBUILDER_HPP
|
|
#define POKEMONSCRIPTTESTER_BATTLEBUILDER_HPP
|
|
|
|
#include <CreatureLib/Battling/TurnChoices/AttackTurnChoice.hpp>
|
|
#include <CreatureLib/Battling/TurnChoices/PassTurnChoice.hpp>
|
|
#include <PkmnLib/Battling/Battle/Battle.hpp>
|
|
#include <PkmnLib/Battling/Pokemon/CreatePokemon.hpp>
|
|
#include <angelscript.h>
|
|
#include "../TestEnvironment.hpp"
|
|
|
|
class BattleBuilder {
|
|
static PkmnLib::Battling::Battle* CreateSimpleBattle(u32 seed, const ArbUt::StringView& species1,
|
|
const ArbUt::StringView& species2, u8 level) {
|
|
auto* ctx = asGetActiveContext();
|
|
TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData());
|
|
auto lib = Globals::Library.GetValue();
|
|
|
|
auto mon1 = PkmnLib::Battling::CreatePokemon(lib, species1, level)
|
|
.WithEffortValues(0, 0, 0, 0, 0, 0)
|
|
.WithIndividualValues(31, 31, 31, 31, 31, 31)
|
|
.WithNature("hardy"_cnc)
|
|
.WithGender(CreatureLib::Library::Gender::Male)
|
|
.IsAllowedExperienceGain(false)
|
|
.Build();
|
|
auto p1 = new CreatureLib::Battling::CreatureParty(1);
|
|
p1->SwapInto(0, mon1);
|
|
|
|
auto mon2 = PkmnLib::Battling::CreatePokemon(lib, species2, level)
|
|
.WithEffortValues(0, 0, 0, 0, 0, 0)
|
|
.WithIndividualValues(31, 31, 31, 31, 31, 31)
|
|
.WithNature("hardy"_cnc)
|
|
.WithGender(CreatureLib::Library::Gender::Male)
|
|
.IsAllowedExperienceGain(false)
|
|
.Build();
|
|
auto p2 = new CreatureLib::Battling::CreatureParty(1);
|
|
p1->SwapInto(0, mon2);
|
|
|
|
auto battle = new PkmnLib::Battling::Battle(
|
|
lib,
|
|
{new CreatureLib::Battling::BattleParty(p1, {CreatureLib::Battling::CreatureIndex(0, 0)}),
|
|
new CreatureLib::Battling::BattleParty(p2, {CreatureLib::Battling::CreatureIndex(1, 0)})},
|
|
true, // can flee
|
|
2, // 2 sides
|
|
1, // 1 mon per side
|
|
seed // with seed
|
|
);
|
|
battle->SwitchCreature(0, 0, mon1);
|
|
battle->SwitchCreature(1, 0, mon2);
|
|
|
|
env->AddGarbage(battle);
|
|
env->AddGarbage(p1);
|
|
env->AddGarbage(p2);
|
|
|
|
return battle;
|
|
}
|
|
|
|
static bool UseMove(PkmnLib::Battling::Pokemon* user, const ArbUt::StringView& moveName, u8 sideTarget, u8 target) {
|
|
auto battle = user->GetBattle();
|
|
if (!battle.HasValue()) {
|
|
return false;
|
|
}
|
|
|
|
auto move = Globals::Library.GetValue()->GetMoveLibrary()->TryGet(moveName);
|
|
if (!move.has_value()) {
|
|
return false;
|
|
}
|
|
auto* ctx = asGetActiveContext();
|
|
TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData());
|
|
|
|
auto learnedMove =
|
|
new PkmnLib::Battling::LearnedMove(move.value(), CreatureLib::Battling::AttackLearnMethod::Unknown);
|
|
env->AddGarbage(learnedMove);
|
|
auto choice = new CreatureLib::Battling::AttackTurnChoice(
|
|
user, learnedMove, CreatureLib::Battling::CreatureIndex(sideTarget, target));
|
|
auto b = battle.GetValue()->TrySetChoice(choice);
|
|
if (!b) {
|
|
delete choice;
|
|
}
|
|
return b;
|
|
}
|
|
|
|
static bool PassTurn(PkmnLib::Battling::Pokemon* user) {
|
|
auto battle = user->GetBattle();
|
|
if (!battle.HasValue()) {
|
|
return false;
|
|
}
|
|
|
|
auto choice = new CreatureLib::Battling::PassTurnChoice(user);
|
|
auto b = battle.GetValue()->TrySetChoice(choice);
|
|
if (!b) {
|
|
delete choice;
|
|
}
|
|
return b;
|
|
}
|
|
|
|
public:
|
|
static void Register(AngelScriptResolver* scriptResolver) {
|
|
auto engine = scriptResolver->GetBuilder().GetEngine();
|
|
Ensure(engine->RegisterGlobalFunction("Battle@ CreateSimpleBattle(uint seed, const constString&in species1, "
|
|
"const constString&in species2, uint8 level)",
|
|
asFUNCTION(CreateSimpleBattle), asCALL_CDECL) >= 0);
|
|
Ensure(engine->RegisterObjectMethod("Pokemon",
|
|
"bool UseMove(const constString&in move, uint8 side, uint8 index)",
|
|
asFUNCTION(UseMove), asCALL_CDECL_OBJFIRST) >= 0);
|
|
Ensure(engine->RegisterObjectMethod("Pokemon", "bool PassTurn()", asFUNCTION(PassTurn),
|
|
asCALL_CDECL_OBJFIRST) >= 0);
|
|
}
|
|
};
|
|
|
|
#endif // POKEMONSCRIPTTESTER_BATTLEBUILDER_HPP
|