163 lines
6.4 KiB
C++
163 lines
6.4 KiB
C++
#include "Console.h"
|
|
#include "Core.h"
|
|
#include "DataDefs.h"
|
|
#include "Export.h"
|
|
#include "PluginManager.h"
|
|
#include "df/world.h"
|
|
#include "df/world_raws.h"
|
|
#include "df/creature_raw.h"
|
|
#include "df/caste_raw.h"
|
|
#include "modules/World.h"
|
|
|
|
//#include "df/world.h"
|
|
|
|
using namespace DFHack;
|
|
|
|
DFHACK_PLUGIN("generated-creature-renamer");
|
|
REQUIRE_GLOBAL(world);
|
|
|
|
#define RENAMER_VERSION 1
|
|
|
|
command_result list_creatures(color_ostream &out, std::vector <std::string> & parameters);
|
|
|
|
DFhackCExport command_result plugin_init(color_ostream &out, std::vector <PluginCommand> &commands)
|
|
{
|
|
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;
|
|
}
|
|
|
|
DFhackCExport command_result plugin_shutdown(color_ostream &out)
|
|
{
|
|
return CR_OK;
|
|
}
|
|
|
|
DFHACK_PLUGIN_IS_ENABLED(is_enabled);
|
|
|
|
DFhackCExport command_result plugin_enable(color_ostream &out, bool enable)
|
|
{
|
|
is_enabled = enable;
|
|
return CR_OK;
|
|
}
|
|
|
|
std::vector<std::string> descriptors = { "blob", "quadruped", "humanoid", "silverfish", "mayfly", "dragonfly",
|
|
"damselfly", "stonefly", "earwig", "grasshopper", "cricket", "stick insect", "cockroach", "termite",
|
|
"mantis", "louse", "thrips", "aphid", "cicada", "assassin bug", "wasp", "hornet", "tiger beetle",
|
|
"ladybug", "weevil", "darkling beetle", "click beetle", "firefly", "scarab beetle", "stag beetle",
|
|
"dung beetle", "rhinoceros beetle", "rove beetle", "snakefly", "lacewing", "antlion larva",
|
|
"mosquito", "flea", "scorpionfly", "caddisfly", "butterfly", "moth", "caterpillar", "maggot",
|
|
"spider", "tarantula", "scorpion", "tick", "mite", "shrimp", "lobster", "crab", "nematode",
|
|
"snail", "slug", "earthworm", "leech", "bristleworm", "ribbon worm", "flat worm", "toad", "frog",
|
|
"salamander", "newt", "alligator", "crocodile", "lizard", "chameleon", "iguana", "gecko", "skink",
|
|
"gila monster", "monitor", "serpent", "viper", "rattlesnake", "cobra", "python", "anaconda", "turtle",
|
|
"tortoise", "pterosaur", "dimetrodon", "sauropod", "theropod", "iguanodont", "hadrosaurid", "stegosaurid",
|
|
"ceratopsid", "ankylosaurid", "duck", "goose", "swan", "turkey", "grouse", "chicken", "quail", "pheasant",
|
|
"gull", "loon", "grebe", "albatross", "petrel", "penguin", "pelican", "stork", "vulture", "flamingo",
|
|
"falcon", "kestrel", "condor", "osprey", "buzzard", "eagle", "harrier", "kite", "crane", "dove",
|
|
"pigeon", "parrot", "cockatoo", "cuckoo", "nightjar", "swift", "hummingbird", "kingfisher",
|
|
"hornbill", "quetzal", "toucan", "woodpecker", "lyrebird", "thornbill", "honeyeater", "oriole",
|
|
"fantail", "shrike", "crow", "raven", "magpie", "kinglet", "lark", "swallow", "martin", "bushtit",
|
|
"warbler", "thrush", "oxpecker", "starling", "mockingbird", "wren", "nuthatch", "sparrow", "tanager",
|
|
"cardinal", "bunting", "finch", "titmouse", "chickadee", "waxwing", "flycatcher", "opossum", "koala",
|
|
"wombat", "kangaroo", "sloth", "anteater", "armadillo", "squirrel", "marmot", "beaver", "gopher",
|
|
"mouse", "porcupine", "chinchilla", "cavy", "capybara", "rabbit", "hare", "lemur", "loris", "monkey",
|
|
"hedgehog", "shrew", "mole", "fruit bat", "wolf", "coyote", "jackal", "raccoon", "coati",
|
|
"weasel", "otter", "badger", "skunk", "bear", "panda", "panther", "mongoose", "hyena", "civet",
|
|
"walrus", "pangolin", "elephant", "mammoth", "horse", "zebra", "tapir", "rhinoceros", "warthog",
|
|
"hippopotamus", "camel", "llama", "giraffe", "deer", "moose", "antelope", "sheep", "goat",
|
|
"bison", "buffalo", "bull", "ape", "ant", "bat", "owl", "pig", "bee"};
|
|
|
|
|
|
DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_change_event event)
|
|
{
|
|
if (!is_enabled)
|
|
return CR_OK;
|
|
switch (event)
|
|
{
|
|
case DFHack::SC_WORLD_LOADED:
|
|
CoreSuspender suspend;
|
|
|
|
std::vector<int> descriptorCount = std::vector<int>(descriptors.size());
|
|
|
|
auto version = World::GetPersistentData("AlreadyRenamedCreatures");
|
|
if (version.isValid() && version.ival(1) >= RENAMER_VERSION)
|
|
{
|
|
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 < descriptors.size(); 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], 1, std::string::npos);
|
|
|
|
if (descriptorCount[foundIndex] > 0)
|
|
{
|
|
creatureRaw->creature_id.append(std::to_string(descriptorCount[foundIndex]));
|
|
}
|
|
|
|
descriptorCount[foundIndex]++;
|
|
}
|
|
}
|
|
version = World::AddPersistentData("AlreadyRenamedCreatures");
|
|
version.ival(1) = RENAMER_VERSION;
|
|
break;
|
|
}
|
|
|
|
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++)
|
|
{
|
|
auto creatureRaw = world->raws.creatures.all[i];
|
|
if (!creatureRaw->flags.is_set(df::enums::creature_raw_flags::GENERATED))
|
|
continue;
|
|
out.print(creatureRaw->creature_id.c_str());
|
|
if (detailed)
|
|
{
|
|
out.print("\t");
|
|
out.print(creatureRaw->caste[0]->description.c_str());
|
|
}
|
|
out.print("\n");
|
|
}
|
|
}
|
|
|