MetzNet
/
dfhack
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added command to show status of preservetombs (is enabled or disabled)

develop
Najeeb Al-Shabibi 2023-09-29 13:38:52 +01:00
parent e713e63ea3
commit b0a15b2e8a
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
plugins/preserve-tombs.cpp
Unescape Escape View File

@ -68,12 +68,27 @@ static void set_config_bool(PersistentDataItem &c, int index, bool value) {
static bool assign_to_tomb(int32_t unit_id, int32_t building_id);
static void update_tomb_assignments(color_ostream& out);
void onUnitDeath(color_ostream& out, void* ptr);
static command_result do_command(color_ostream& out, std::vector<std::string>& params);
DFhackCExport command_result plugin_init(color_ostream &out, std::vector <PluginCommand> &commands) {
tomb_assignments.clear();
commands.push_back(PluginCommand(
plugin_name,
"Preserves tomb assignments to units when they die.",
do_command));
return CR_OK;
}
static command_result do_command(color_ostream& out, std::vector<std::string>& params) {
if (params.size() != 1 || params[0] != "status") {
out.print("%s wrong usage", plugin_name);
return CR_WRONG_USAGE;
}
else {
out.print("%s is currently %s", plugin_name, is_enabled ? "enabled" : "disabled");
return CR_OK;
}
}
// event listener
EventManager::EventHandler assign_tomb_handler(onUnitDeath, 0);
@ -154,7 +169,7 @@ DFhackCExport command_result plugin_onupdate(color_ostream &out) {
void onUnitDeath(color_ostream& out, void* ptr) {
// input is void* that contains the unit id
int32_t unit_id = reinterpret_cast<std::intptr_t>(ptr);
// check if unit was assigned a tomb in life
auto it = tomb_assignments.find(unit_id);
if (it == tomb_assignments.end()) return;
@ -199,10 +214,10 @@ static void update_tomb_assignments(color_ostream &out) {
}
// now check our civzones for unassignment / deleted zone /
// now check our civzones for unassignment / deleted zone /
for (auto it = tomb_assignments.begin(); it != tomb_assignments.end(); ++it){
auto &[unit_id, building_id] = *it;
const size_t tomb_idx = binsearch_index(world->buildings.other.ZONE_TOMB, building_id);
if (tomb_idx == -1) {
out.print("%s tomb missing: %d - removing\n", plugin_name, building_id);
@ -229,13 +244,13 @@ static void update_tomb_assignments(color_ostream &out) {
//
//
static bool assign_to_tomb(int32_t unit_id, int32_t building_id) {
size_t unit_idx = Units::findIndexById(unit_id);
if (unit_idx == -1) return false;
df::unit* unit = world->units.all[unit_idx];
if (!Units::isDead(unit)) return false;
size_t tomb_idx = binsearch_index(world->buildings.other.ZONE_TOMB, building_id);
if (tomb_idx == -1) return false;