114 lines
5.7 KiB
C++
114 lines
5.7 KiB
C++
#include "BuildSpecies.hpp"
|
|
#include <fstream>
|
|
#include <iostream>
|
|
|
|
#define GET(o, objectKey, key) \
|
|
auto _##objectKey = o[#objectKey]; \
|
|
if (_##objectKey.is_null()) { \
|
|
std::cout << "Failed to retrieve key '" << #objectKey << "' for object with key '" << key << "' in file '" \
|
|
<< path << "'\n"; \
|
|
return nullptr; \
|
|
}
|
|
|
|
PkmnLib::Library::SpeciesLibrary* BuildSpecies::BuildLibrary(const std::string& path,
|
|
const CreatureLib::Library::TypeLibrary* types) {
|
|
std::ifstream fileStream(path.c_str());
|
|
if (fileStream.fail()) {
|
|
std::cout << "Failed to load Pokemon data file at '" << path << "'\n";
|
|
return nullptr;
|
|
}
|
|
auto lib = new PkmnLib::Library::SpeciesLibrary();
|
|
json j;
|
|
fileStream >> j;
|
|
|
|
for (json::iterator it = j.begin(); it != j.end(); ++it) {
|
|
auto val = it.value();
|
|
GET(val, id, it.key());
|
|
GET(val, species, it.key());
|
|
GET(val, genderRatio, it.key());
|
|
GET(val, growthRate, it.key());
|
|
GET(val, baseHappiness, it.key());
|
|
GET(val, catchRate, it.key());
|
|
GET(val, color, it.key());
|
|
GET(val, genderDifference, it.key());
|
|
GET(val, eggGroups, it.key());
|
|
GET(val, eggCycles, it.key());
|
|
GET(val, tags, it.key());
|
|
GET(val, formes, it.key());
|
|
|
|
PkmnLib::Library::PokemonSpecies* species = nullptr;
|
|
|
|
auto defaultForme = _formes["default"];
|
|
if (!defaultForme.is_null()) {
|
|
auto forme = BuildForme("default", defaultForme, it.key(), path, types);
|
|
species = new PkmnLib::Library::PokemonSpecies(
|
|
_id.get<uint16_t>(), Arbutils::CaseInsensitiveConstString(_species.get<std::string>()), forme,
|
|
_genderRatio.get<int8_t>() / static_cast<float>(100),
|
|
Arbutils::CaseInsensitiveConstString(_growthRate.get<std::string>()), _catchRate.get<uint8_t>(),
|
|
_baseHappiness.get<uint8_t>());
|
|
}
|
|
|
|
for (json::iterator formeIt = _formes.begin(); formeIt != _formes.end(); ++formeIt) {
|
|
if (formeIt.key() == "default") {
|
|
continue;
|
|
}
|
|
auto forme = BuildForme(formeIt.key(), formeIt.value(), it.key(), path, types);
|
|
if (forme == nullptr)
|
|
return nullptr;
|
|
if (species == nullptr) {
|
|
species = new PkmnLib::Library::PokemonSpecies(
|
|
_id.get<uint16_t>(), Arbutils::CaseInsensitiveConstString(_species.get<std::string>()), forme,
|
|
static_cast<float>(_genderRatio.get<int8_t>()) / static_cast<float>(100),
|
|
Arbutils::CaseInsensitiveConstString(_growthRate.get<std::string>()), _catchRate.get<uint8_t>(),
|
|
_baseHappiness.get<uint8_t>());
|
|
} else {
|
|
if (species->HasForme(Arbutils::CaseInsensitiveConstString(formeIt.key()))) {
|
|
std::cout << "Species '" << it.key() << "' has duplicate forme '" << formeIt.key()
|
|
<< "'. Skipping.\n";
|
|
delete forme;
|
|
continue;
|
|
}
|
|
species->SetVariant(Arbutils::CaseInsensitiveConstString(formeIt.key()), forme);
|
|
}
|
|
}
|
|
if (species == nullptr) {
|
|
std::cout << "Pokemon with key '" << it.key() << "' does not have any formes.\n";
|
|
return nullptr;
|
|
}
|
|
lib->Insert(Arbutils::CaseInsensitiveConstString(it.key()), species);
|
|
}
|
|
return lib;
|
|
}
|
|
|
|
static CreatureLib::Core::StatisticSet<uint16_t> ParseStatistics(json& json) {
|
|
return CreatureLib::Core::StatisticSet<uint16_t>(
|
|
json["hp"].get<uint16_t>(), json["attack"].get<uint16_t>(), json["defense"].get<uint16_t>(),
|
|
json["specialAttack"].get<uint16_t>(), json["specialDefense"].get<uint16_t>(), json["speed"].get<uint16_t>());
|
|
}
|
|
|
|
const PkmnLib::Library::PokemonForme* BuildSpecies::BuildForme(const std::string& name, json& forme,
|
|
const std::string& baseKeyName, const std::string& path,
|
|
const CreatureLib::Library::TypeLibrary* typeLibrary) {
|
|
GET(forme, abilities, baseKeyName << " -> " << name);
|
|
GET(forme, hiddenAbilities, baseKeyName << " -> " << name);
|
|
GET(forme, baseStats, baseKeyName << " -> " << name);
|
|
GET(forme, evReward, baseKeyName << " -> " << name);
|
|
GET(forme, types, baseKeyName << " -> " << name);
|
|
GET(forme, height, baseKeyName << " -> " << name);
|
|
GET(forme, weight, baseKeyName << " -> " << name);
|
|
GET(forme, baseExp, baseKeyName << " -> " << name);
|
|
GET(forme, moves, baseKeyName << " -> " << name);
|
|
|
|
auto typeStrings = _types.get<std::vector<std::string>>();
|
|
auto types = std::vector<uint8_t>(typeStrings.size());
|
|
for (auto i = 0; i < typeStrings.size(); i++) {
|
|
types[i] = typeLibrary->GetTypeId(typeStrings[i]);
|
|
}
|
|
auto stats = ParseStatistics(_baseStats);
|
|
|
|
return new PkmnLib::Library::PokemonForme(
|
|
name, _height.get<float>(), _weight.get<float>(), _baseExp.get<uint32_t>(), types, stats,
|
|
_abilities.get<std::vector<std::string>>(), _hiddenAbilities.get<std::vector<std::string>>(), nullptr);
|
|
}
|
|
|
|
#undef GET |