|
|
@ -7,6 +7,7 @@
|
|
|
|
#include "df/world_raws.h"
|
|
|
|
#include "df/world_raws.h"
|
|
|
|
#include "df/creature_raw.h"
|
|
|
|
#include "df/creature_raw.h"
|
|
|
|
#include "df/caste_raw.h"
|
|
|
|
#include "df/caste_raw.h"
|
|
|
|
|
|
|
|
#include "modules/World.h"
|
|
|
|
|
|
|
|
|
|
|
|
//#include "df/world.h"
|
|
|
|
//#include "df/world.h"
|
|
|
|
|
|
|
|
|
|
|
@ -15,26 +16,34 @@ using namespace DFHack;
|
|
|
|
DFHACK_PLUGIN("generated-creature-renamer");
|
|
|
|
DFHACK_PLUGIN("generated-creature-renamer");
|
|
|
|
REQUIRE_GLOBAL(world);
|
|
|
|
REQUIRE_GLOBAL(world);
|
|
|
|
|
|
|
|
|
|
|
|
command_result rename_creatures (color_ostream &out, std::vector <std::string> & parameters);
|
|
|
|
command_result rename_creatures(color_ostream &out, std::vector <std::string> & parameters);
|
|
|
|
|
|
|
|
command_result list_creatures(color_ostream &out, std::vector <std::string> & parameters);
|
|
|
|
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_init (color_ostream &out, std::vector <PluginCommand> &commands)
|
|
|
|
DFhackCExport command_result plugin_init(color_ostream &out, std::vector <PluginCommand> &commands)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
commands.push_back(PluginCommand(
|
|
|
|
commands.push_back(PluginCommand(
|
|
|
|
"rename_creatures",
|
|
|
|
"rename-generated",
|
|
|
|
"Renames generated creature tags to something friendlier to modders",
|
|
|
|
"Renames generated creature tags to something friendlier to modders",
|
|
|
|
rename_creatures,
|
|
|
|
rename_creatures,
|
|
|
|
false, //allow non-interactive use
|
|
|
|
false, //allow non-interactive use
|
|
|
|
"longHelpString"
|
|
|
|
"Renames generated creature tags to something friendlier to modders"
|
|
|
|
|
|
|
|
));
|
|
|
|
|
|
|
|
commands.push_back(PluginCommand(
|
|
|
|
|
|
|
|
"list-generated",
|
|
|
|
|
|
|
|
"Prints a list of generated creature tokens. Use \"list-generated detailed\" to show descriptions.",
|
|
|
|
|
|
|
|
list_creatures,
|
|
|
|
|
|
|
|
false, //allow non-interactive use
|
|
|
|
|
|
|
|
"Prints a list of generated creature tokens. Use \"list-generated detailed\" to show descriptions."
|
|
|
|
));
|
|
|
|
));
|
|
|
|
return CR_OK;
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_shutdown (color_ostream &out)
|
|
|
|
DFhackCExport command_result plugin_shutdown(color_ostream &out)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
return CR_OK;
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
std::string descriptors[] = { "blob", "quadruped", "humanoid", "silverfish", "mayfly", "dragonfly",
|
|
|
|
std::string descriptors[221] = { "blob", "quadruped", "humanoid", "silverfish", "mayfly", "dragonfly",
|
|
|
|
"damselfly", "stonefly", "earwig", "grasshopper", "cricket", "stick insect", "cockroach", "termite",
|
|
|
|
"damselfly", "stonefly", "earwig", "grasshopper", "cricket", "stick insect", "cockroach", "termite",
|
|
|
|
"mantis", "louse", "thrips", "aphid", "cicada", "assassin bug", "wasp", "hornet", "tiger beetle",
|
|
|
|
"mantis", "louse", "thrips", "aphid", "cicada", "assassin bug", "wasp", "hornet", "tiger beetle",
|
|
|
|
"ladybug", "weevil", "darkling beetle", "click beetle", "firefly", "scarab beetle", "stag beetle",
|
|
|
|
"ladybug", "weevil", "darkling beetle", "click beetle", "firefly", "scarab beetle", "stag beetle",
|
|
|
@ -61,21 +70,83 @@ std::string descriptors[] = { "blob", "quadruped", "humanoid", "silverfish", "ma
|
|
|
|
"hippopotamus", "camel", "llama", "giraffe", "deer", "moose", "antelope", "sheep", "goat",
|
|
|
|
"hippopotamus", "camel", "llama", "giraffe", "deer", "moose", "antelope", "sheep", "goat",
|
|
|
|
"bison", "buffalo", "bull" };
|
|
|
|
"bison", "buffalo", "bull" };
|
|
|
|
|
|
|
|
|
|
|
|
command_result rename_creatures (color_ostream &out, std::vector <std::string> & parameters)
|
|
|
|
command_result rename_creatures(color_ostream &out, std::vector <std::string> & parameters)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
if (!parameters.empty())
|
|
|
|
if (!parameters.empty())
|
|
|
|
return CR_WRONG_USAGE;
|
|
|
|
return CR_WRONG_USAGE;
|
|
|
|
CoreSuspender suspend;
|
|
|
|
CoreSuspender suspend;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int descriptorCount[221] = { 0 };
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (World::GetPersistentData("AlreadyRenamedCreatures").isValid())
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
return CR_OK;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
for (int i = 0; i < world->raws.creatures.all.size(); i++)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
auto creatureRaw = world->raws.creatures.all[i];
|
|
|
|
|
|
|
|
if (!creatureRaw->flags.is_set(df::enums::creature_raw_flags::GENERATED))
|
|
|
|
|
|
|
|
continue;
|
|
|
|
|
|
|
|
size_t minPos = std::string::npos;
|
|
|
|
|
|
|
|
size_t foundIndex = std::string::npos;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
for (size_t j = 0; j < 221; j++)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
size_t pos = creatureRaw->caste[0]->description.find(descriptors[j]);
|
|
|
|
|
|
|
|
if (pos < minPos)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
minPos = pos;
|
|
|
|
|
|
|
|
foundIndex = j;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (foundIndex < std::string::npos)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
size_t digitPos = creatureRaw->creature_id.find_first_of("0123456789");
|
|
|
|
|
|
|
|
if (digitPos > creatureRaw->creature_id.length())
|
|
|
|
|
|
|
|
digitPos = creatureRaw->creature_id.length();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
creatureRaw->creature_id.replace(digitPos, std::string::npos, descriptors[foundIndex]);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (descriptorCount[foundIndex] > 0)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
creatureRaw->creature_id.append(std::to_string(descriptorCount[foundIndex]));
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
descriptorCount[foundIndex]++;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
World::AddPersistentData("AlreadyRenamedCreatures");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return CR_OK;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
command_result list_creatures(color_ostream &out, std::vector <std::string> & parameters)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
bool detailed = false;
|
|
|
|
|
|
|
|
if (!parameters.empty())
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
if (parameters.size() > 1)
|
|
|
|
|
|
|
|
return CR_WRONG_USAGE;
|
|
|
|
|
|
|
|
if(parameters[0].compare("detailed") != 0)
|
|
|
|
|
|
|
|
return CR_WRONG_USAGE;
|
|
|
|
|
|
|
|
detailed = true;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
CoreSuspender suspend;
|
|
|
|
for (int i = 0; i < world->raws.creatures.all.size(); i++)
|
|
|
|
for (int i = 0; i < world->raws.creatures.all.size(); i++)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
auto creatureRaw = world->raws.creatures.all[i];
|
|
|
|
auto creatureRaw = world->raws.creatures.all[i];
|
|
|
|
if (!creatureRaw->flags.is_set(df::enums::creature_raw_flags::GENERATED))
|
|
|
|
if (!creatureRaw->flags.is_set(df::enums::creature_raw_flags::GENERATED))
|
|
|
|
continue;
|
|
|
|
continue;
|
|
|
|
out.print(creatureRaw->creature_id.c_str());
|
|
|
|
out.print(creatureRaw->creature_id.c_str());
|
|
|
|
out.print("\n");
|
|
|
|
if (detailed)
|
|
|
|
out.print(creatureRaw->caste[0]->description.c_str());
|
|
|
|
{
|
|
|
|
|
|
|
|
out.print("\t");
|
|
|
|
|
|
|
|
out.print(creatureRaw->caste[0]->description.c_str());
|
|
|
|
|
|
|
|
}
|
|
|
|
out.print("\n");
|
|
|
|
out.print("\n");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return CR_OK;
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|