dfhack/plugins/autochop.cpp

879 lines
31 KiB
C++

// automatically chop trees
#include "Debug.h"
#include "LuaTools.h"
#include "PluginManager.h"
#include "TileTypes.h"
#include "modules/Burrows.h"
#include "modules/Designations.h"
#include "modules/Maps.h"
#include "modules/Persistence.h"
#include "modules/Units.h"
#include "modules/World.h"
#include "df/burrow.h"
#include "df/item.h"
#include "df/map_block.h"
#include "df/plant.h"
#include "df/plant_tree_info.h"
#include "df/plant_tree_tile.h"
#include "df/plotinfost.h"
#include "df/world.h"
#include <map>
#include <unordered_map>
using std::map;
using std::multimap;
using std::pair;
using std::string;
using std::unordered_map;
using std::vector;
using namespace DFHack;
using namespace df::enums;
DFHACK_PLUGIN("autochop");
DFHACK_PLUGIN_IS_ENABLED(is_enabled);
REQUIRE_GLOBAL(world);
REQUIRE_GLOBAL(plotinfo);
namespace DFHack {
// for configuration-related logging
DBG_DECLARE(autochop, status, DebugCategory::LINFO);
// for logging during the periodic scan
DBG_DECLARE(autochop, cycle, DebugCategory::LINFO);
}
static const string CONFIG_KEY = string(plugin_name) + "/config";
static const string BURROW_CONFIG_KEY_PREFIX = string(plugin_name) + "/burrow/";
static PersistentDataItem config;
static vector<PersistentDataItem> watched_burrows;
static unordered_map<int, size_t> watched_burrows_indices;
enum ConfigValues {
CONFIG_IS_ENABLED = 0,
CONFIG_MAX_LOGS = 1,
CONFIG_MIN_LOGS = 2,
CONFIG_WAITING_FOR_MIN = 3,
};
enum BurrowConfigValues {
BURROW_CONFIG_ID = 0,
BURROW_CONFIG_CHOP = 1,
BURROW_CONFIG_CLEARCUT = 2,
BURROW_CONFIG_PROTECT_BREWABLE = 3,
BURROW_CONFIG_PROTECT_EDIBLE = 4,
BURROW_CONFIG_PROTECT_COOKABLE = 5,
};
static int get_config_val(PersistentDataItem &c, int index) {
if (!c.isValid())
return -1;
return c.ival(index);
}
static bool get_config_bool(PersistentDataItem &c, int index) {
return get_config_val(c, index) == 1;
}
static void set_config_val(PersistentDataItem &c, int index, int value) {
if (c.isValid())
c.ival(index) = value;
}
static void set_config_bool(PersistentDataItem &c, int index, bool value) {
set_config_val(c, index, value ? 1 : 0);
}
static PersistentDataItem & ensure_burrow_config(color_ostream &out, int id) {
if (watched_burrows_indices.count(id))
return watched_burrows[watched_burrows_indices[id]];
string keyname = BURROW_CONFIG_KEY_PREFIX + int_to_string(id);
DEBUG(status,out).print("creating new persistent key for burrow %d\n", id);
watched_burrows.emplace_back(World::GetPersistentData(keyname, NULL));
size_t idx = watched_burrows.size()-1;
watched_burrows_indices.emplace(id, idx);
return watched_burrows[idx];
}
static void remove_burrow_config(color_ostream &out, int id) {
if (!watched_burrows_indices.count(id))
return;
DEBUG(status,out).print("removing persistent key for burrow %d\n", id);
size_t idx = watched_burrows_indices[id];
World::DeletePersistentData(watched_burrows[idx]);
watched_burrows.erase(watched_burrows.begin()+idx);
watched_burrows_indices.erase(id);
}
static void validate_burrow_configs(color_ostream &out) {
for (int32_t idx = watched_burrows.size()-1; idx >=0; --idx) {
int id = get_config_val(watched_burrows[idx], BURROW_CONFIG_ID);
if (!df::burrow::find(id)) {
remove_burrow_config(out, id);
}
}
}
static const int32_t CYCLE_TICKS = 1200;
static int32_t cycle_timestamp = 0; // world->frame_counter at last cycle
static command_result do_command(color_ostream &out, vector<string> &parameters);
static int32_t do_cycle(color_ostream &out, bool force_designate = false);
DFhackCExport command_result plugin_init(color_ostream &out, std::vector <PluginCommand> &commands) {
DEBUG(status,out).print("initializing %s\n", plugin_name);
// provide a configuration interface for the plugin
commands.push_back(PluginCommand(
plugin_name,
"Auto-harvest trees when low on stockpiled logs.",
do_command));
return CR_OK;
}
DFhackCExport command_result plugin_enable(color_ostream &out, bool enable) {
if (!Core::getInstance().isWorldLoaded()) {
out.printerr("Cannot enable %s without a loaded world.\n", plugin_name);
return CR_FAILURE;
}
if (enable != is_enabled) {
is_enabled = enable;
DEBUG(status,out).print("%s from the API; persisting\n",
is_enabled ? "enabled" : "disabled");
set_config_bool(config, CONFIG_IS_ENABLED, is_enabled);
} else {
DEBUG(status,out).print("%s from the API, but already %s; no action\n",
is_enabled ? "enabled" : "disabled",
is_enabled ? "enabled" : "disabled");
}
return CR_OK;
}
DFhackCExport command_result plugin_shutdown (color_ostream &out) {
DEBUG(status,out).print("shutting down %s\n", plugin_name);
return CR_OK;
}
DFhackCExport command_result plugin_load_data (color_ostream &out) {
config = World::GetPersistentData(CONFIG_KEY);
if (!config.isValid()) {
DEBUG(status,out).print("no config found in this save; initializing\n");
config = World::AddPersistentData(CONFIG_KEY);
set_config_bool(config, CONFIG_IS_ENABLED, is_enabled);
set_config_val(config, CONFIG_MAX_LOGS, 200);
set_config_val(config, CONFIG_MIN_LOGS, 160);
set_config_bool(config, CONFIG_WAITING_FOR_MIN, false);
}
// we have to copy our enabled flag into the global plugin variable, but
// all the other state we can directly read/modify from the persistent
// data structure.
is_enabled = get_config_bool(config, CONFIG_IS_ENABLED);
DEBUG(status,out).print("loading persisted enabled state: %s\n",
is_enabled ? "true" : "false");
World::GetPersistentData(&watched_burrows, BURROW_CONFIG_KEY_PREFIX, true);
watched_burrows_indices.clear();
const size_t num_watched_burrows = watched_burrows.size();
for (size_t idx = 0; idx < num_watched_burrows; ++idx) {
auto &c = watched_burrows[idx];
watched_burrows_indices.emplace(get_config_val(c, BURROW_CONFIG_ID), idx);
}
validate_burrow_configs(out);
return CR_OK;
}
DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_change_event event) {
if (event == DFHack::SC_WORLD_UNLOADED) {
if (is_enabled) {
DEBUG(status,out).print("world unloaded; disabling %s\n",
plugin_name);
is_enabled = false;
}
}
return CR_OK;
}
DFhackCExport command_result plugin_onupdate(color_ostream &out) {
if (is_enabled && world->frame_counter - cycle_timestamp >= CYCLE_TICKS) {
int32_t designated = do_cycle(out);
if (0 < designated)
out.print("autochop: designated %d tree(s) for chopping\n", designated);
}
return CR_OK;
}
static bool call_autochop_lua(color_ostream *out, const char *fn_name,
int nargs = 0, int nres = 0,
Lua::LuaLambda && args_lambda = Lua::DEFAULT_LUA_LAMBDA,
Lua::LuaLambda && res_lambda = Lua::DEFAULT_LUA_LAMBDA) {
DEBUG(status).print("calling autochop lua function: '%s'\n", fn_name);
CoreSuspender guard;
auto L = Lua::Core::State;
Lua::StackUnwinder top(L);
if (!out)
out = &Core::getInstance().getConsole();
return Lua::CallLuaModuleFunction(*out, L, "plugins.autochop", fn_name,
nargs, nres,
std::forward<Lua::LuaLambda&&>(args_lambda),
std::forward<Lua::LuaLambda&&>(res_lambda));
}
static command_result do_command(color_ostream &out, vector<string> &parameters) {
CoreSuspender suspend;
if (!Core::getInstance().isWorldLoaded()) {
out.printerr("Cannot run %s without a loaded world.\n", plugin_name);
return CR_FAILURE;
}
bool show_help = false;
if (!call_autochop_lua(&out, "parse_commandline", parameters.size(), 1,
[&](lua_State *L) {
for (const string &param : parameters)
Lua::Push(L, param);
},
[&](lua_State *L) {
show_help = !lua_toboolean(L, -1);
})) {
return CR_FAILURE;
}
return show_help ? CR_WRONG_USAGE : CR_OK;
}
/////////////////////////////////////////////////////
// cycle logic
//
static bool is_accessible_item(const df::coord &pos, const vector<df::unit *> &citizens) {
for (auto &unit : citizens) {
if (Maps::canWalkBetween(unit->pos, pos))
return true;
}
return false;
}
// at least one member of the fort can reach a position adjacent to the given pos
static bool is_accessible_tree(const df::coord &pos, const vector<df::unit *> &citizens) {
for (auto &unit : citizens) {
if (Maps::canWalkBetween(unit->pos, df::coord(pos.x-1, pos.y-1, pos.z))
|| Maps::canWalkBetween(unit->pos, df::coord(pos.x, pos.y-1, pos.z))
|| Maps::canWalkBetween(unit->pos, df::coord(pos.x+1, pos.y-1, pos.z))
|| Maps::canWalkBetween(unit->pos, df::coord(pos.x-1, pos.y, pos.z))
|| Maps::canWalkBetween(unit->pos, df::coord(pos.x+1, pos.y, pos.z))
|| Maps::canWalkBetween(unit->pos, df::coord(pos.x-1, pos.y+1, pos.z))
|| Maps::canWalkBetween(unit->pos, df::coord(pos.x, pos.y+1, pos.z))
|| Maps::canWalkBetween(unit->pos, df::coord(pos.x+1, pos.y+1, pos.z)))
return true;
}
return false;
}
static bool is_valid_tree(const df::plant *plant) {
// Skip all non-trees immediately.
if (plant->flags.bits.is_shrub)
return false;
// Skip plants with invalid tile.
df::map_block *block = Maps::getTileBlock(plant->pos);
if (!block)
return false;
int x = plant->pos.x % 16;
int y = plant->pos.y % 16;
// Skip all unrevealed plants.
if (block->designation[x][y].bits.hidden)
return false;
if (tileMaterial(block->tiletype[x][y]) != tiletype_material::TREE)
return false;
return true;
}
static bool is_protected(const df::plant * plant, PersistentDataItem &c) {
const df::plant_raw *plant_raw = df::plant_raw::find(plant->material);
bool protect_brewable = get_config_bool(c, BURROW_CONFIG_PROTECT_BREWABLE);
bool protect_edible = get_config_bool(c, BURROW_CONFIG_PROTECT_EDIBLE);
bool protect_cookable = get_config_bool(c, BURROW_CONFIG_PROTECT_COOKABLE);
if (protect_brewable && plant_raw->material_defs.type[plant_material_def::drink] != -1)
return true;
if (protect_edible || protect_cookable) {
for (df::material * mat : plant_raw->material) {
if (protect_edible && mat->flags.is_set(material_flags::EDIBLE_RAW))
return true;
if (protect_cookable && mat->flags.is_set(material_flags::EDIBLE_COOKED))
return true;
}
}
return false;
}
static int32_t estimate_logs(const df::plant *plant) {
//adapted from code by aljohnston112 @ github
df::plant_tree_tile** tiles = plant->tree_info->body;
df::plant_tree_tile* tilesRow;
int trunks = 0;
for (int i = 0; i < plant->tree_info->body_height; i++) {
tilesRow = tiles[i];
for (int j = 0; j < plant->tree_info->dim_y*plant->tree_info->dim_x; j++) {
trunks += tilesRow[j].bits.trunk;
}
}
return trunks;
}
static void bucket_tree(df::plant *plant, bool designate_clearcut, bool *designated, bool *can_chop,
map<int32_t, int32_t> *tree_counts, map<int32_t, int32_t> *designated_tree_counts,
map<int, PersistentDataItem *> &clearcut_burrows,
map<int, PersistentDataItem *> &chop_burrows) {
for (auto &burrow : plotinfo->burrows.list) {
if (!Burrows::isAssignedTile(burrow, plant->pos))
continue;
int id = burrow->id;
if (tree_counts)
++(*tree_counts)[id];
if (*designated) {
if (designated_tree_counts)
++(*designated_tree_counts)[id];
} else if (clearcut_burrows.count(id) && !is_protected(plant, *clearcut_burrows[id])) {
if (designate_clearcut && Designations::markPlant(plant)) {
*designated = true;
if (designated_tree_counts)
++(*designated_tree_counts)[id];
}
} else if (chop_burrows.count(id) && !is_protected(plant, *chop_burrows[id])) {
*can_chop = true;
}
}
if (!*designated && chop_burrows.empty())
*can_chop = true;
}
static void bucket_watched_burrows(color_ostream & out,
map<int, PersistentDataItem *> &clearcut_burrows,
map<int, PersistentDataItem *> &chop_burrows) {
for (auto &c : watched_burrows) {
int id = get_config_val(c, BURROW_CONFIG_ID);
if (get_config_bool(c, BURROW_CONFIG_CLEARCUT))
clearcut_burrows.emplace(id, &c);
else if (get_config_bool(c, BURROW_CONFIG_CHOP))
chop_burrows.emplace(id, &c);
}
}
typedef multimap<int, df::plant *, std::greater<int>> TreesBySize;
// returns the number of trees that were newly marked
static int32_t scan_trees(color_ostream & out, int32_t *expected_yield,
TreesBySize *designatable_trees_by_size, bool designate_clearcut,
const vector<df::unit *> &citizens, int32_t *accessible_trees = NULL,
int32_t *inaccessible_trees = NULL, int32_t *designated_trees = NULL,
int32_t *accessible_yield = NULL,
map<int32_t, int32_t> *tree_counts = NULL,
map<int32_t, int32_t> *designated_tree_counts = NULL) {
int32_t newly_marked = 0;
if (accessible_trees)
*accessible_trees = 0;
if (inaccessible_trees)
*inaccessible_trees = 0;
if (designated_trees)
*designated_trees = 0;
if (expected_yield)
*expected_yield = 0;
if (accessible_yield)
*accessible_yield = 0;
if (tree_counts)
tree_counts->clear();
if (designated_tree_counts)
designated_tree_counts->clear();
map<int, PersistentDataItem *> clearcut_burrows, chop_burrows;
bucket_watched_burrows(out, clearcut_burrows, chop_burrows);
for (auto plant : world->plants.all) {
if (!is_valid_tree(plant))
continue;
bool accessible = is_accessible_tree(plant->pos, citizens);
int32_t yield = estimate_logs(plant);
if (accessible) {
if (accessible_trees)
++*accessible_trees;
if (accessible_yield)
*accessible_yield += yield;
} else {
if (inaccessible_trees)
++*inaccessible_trees;
}
bool can_chop = false;
bool designated = Designations::isPlantMarked(plant);
bool was_designated = designated;
bucket_tree(plant, designate_clearcut, &designated, &can_chop, tree_counts,
designated_tree_counts, clearcut_burrows, chop_burrows);
if (designated) {
if (!was_designated)
++newly_marked;
if (designated_trees)
++*designated_trees;
if (expected_yield)
*expected_yield += yield;
} else if (can_chop && accessible) {
if (designatable_trees_by_size)
designatable_trees_by_size->emplace(yield, plant);
}
}
return newly_marked;
}
// TODO: does this actually catch anything above the bad_flag check?
static bool is_valid_item(df::item *item) {
for (size_t i = 0; i < item->general_refs.size(); i++) {
df::general_ref *ref = item->general_refs[i];
switch (ref->getType()) {
case general_ref_type::CONTAINED_IN_ITEM:
return false;
case general_ref_type::UNIT_HOLDER:
return false;
case general_ref_type::BUILDING_HOLDER:
return false;
default:
break;
}
}
for (size_t i = 0; i < item->specific_refs.size(); i++) {
df::specific_ref *ref = item->specific_refs[i];
if (ref->type == specific_ref_type::JOB) {
// Ignore any items assigned to a job
return false;
}
}
return true;
}
struct BadFlags
{
uint32_t whole;
BadFlags()
{
df::item_flags flags;
#define F(x) flags.bits.x = true;
F(dump); F(forbid); F(garbage_collect);
F(hostile); F(on_fire); F(rotten); F(trader);
F(in_building); F(construction); F(artifact);
F(in_job); F(owned); F(in_chest); F(removed);
F(encased); F(spider_web);
#undef F
whole = flags.whole;
}
};
static void scan_logs(int32_t *usable_logs, const vector<df::unit *> &citizens, int32_t *inaccessible_logs = NULL) {
static const BadFlags bad_flags;
if (usable_logs)
*usable_logs = 0;
if (inaccessible_logs)
*inaccessible_logs = 0;
for (auto &item : world->items.other[items_other_id::IN_PLAY]) {
if (item->flags.whole & bad_flags.whole)
continue;
if (item->getType() != item_type::WOOD)
continue;
if (!is_valid_item(item))
continue;
if (!is_accessible_item(item->pos, citizens)) {
if (inaccessible_logs)
++*inaccessible_logs;
} else if (usable_logs) {
++*usable_logs;
}
}
}
static int32_t do_cycle(color_ostream &out, bool force_designate) {
DEBUG(cycle,out).print("running %s cycle\n", plugin_name);
// mark that we have recently run
cycle_timestamp = world->frame_counter;
validate_burrow_configs(out);
// scan trees and clearcut marked burrows
int32_t expected_yield;
TreesBySize designatable_trees_by_size;
vector<df::unit *> citizens;
Units::getCitizens(citizens);
int32_t newly_marked = scan_trees(out, &expected_yield,
&designatable_trees_by_size, true, citizens);
// check how many logs we have already
int32_t usable_logs;
scan_logs(&usable_logs, citizens);
if (get_config_bool(config, CONFIG_WAITING_FOR_MIN)
&& usable_logs <= get_config_val(config, CONFIG_MIN_LOGS)) {
DEBUG(cycle,out).print("minimum threshold crossed\n");
set_config_bool(config, CONFIG_WAITING_FOR_MIN, false);
}
else if (!get_config_bool(config, CONFIG_WAITING_FOR_MIN)
&& usable_logs > get_config_val(config, CONFIG_MAX_LOGS)) {
DEBUG(cycle,out).print("maximum threshold crossed\n");
set_config_bool(config, CONFIG_WAITING_FOR_MIN, true);
}
// if we already have designated enough, we're done
int32_t limit = force_designate || !get_config_bool(config, CONFIG_WAITING_FOR_MIN) ?
get_config_val(config, CONFIG_MAX_LOGS) :
get_config_val(config, CONFIG_MIN_LOGS);
if (usable_logs + expected_yield > limit) {
return newly_marked;
}
// designate until the expected yield gets us to our target or we run out
// of accessible trees
int32_t needed = get_config_val(config, CONFIG_MAX_LOGS) -
(usable_logs + expected_yield);
DEBUG(cycle,out).print("needed logs for this cycle: %d\n", needed);
for (auto & entry : designatable_trees_by_size) {
if (!Designations::markPlant(entry.second))
continue;
++newly_marked;
needed -= entry.first;
if (needed <= 0) {
return newly_marked;
}
}
out.print("autochop: insufficient accessible trees to reach log target! Still need %d logs!\n",
needed);
return newly_marked;
}
/////////////////////////////////////////////////////
// Lua API
// core will already be suspended when coming in through here
//
static const char * get_protect_str(bool protect_brewable, bool protect_edible, bool protect_cookable) {
if (!protect_brewable && !protect_edible && !protect_cookable)
return " ";
if (!protect_brewable && !protect_edible && protect_cookable)
return " z";
if (!protect_brewable && protect_edible && !protect_cookable)
return " e ";
if (!protect_brewable && protect_edible && protect_cookable)
return " ez";
if (protect_brewable && !protect_edible && !protect_cookable)
return "b ";
if (protect_brewable && !protect_edible && protect_cookable)
return "b z";
if (protect_brewable && protect_edible && !protect_cookable)
return "be ";
if (protect_brewable && protect_edible && protect_cookable)
return "bez";
return "";
}
static void autochop_printStatus(color_ostream &out) {
DEBUG(status,out).print("entering autochop_printStatus\n");
validate_burrow_configs(out);
out.print("autochop is %s\n\n", is_enabled ? "enabled" : "disabled");
out.print(" keeping log counts between %d and %d\n",
get_config_val(config, CONFIG_MIN_LOGS), get_config_val(config, CONFIG_MAX_LOGS));
if (get_config_bool(config, CONFIG_WAITING_FOR_MIN))
out.print(" currently waiting for min threshold to be crossed before designating more trees\n");
else
out.print(" currently designating trees until max threshold is crossed\n");
out.print("\n");
int32_t usable_logs, inaccessible_logs;
int32_t accessible_trees, inaccessible_trees;
int32_t designated_trees, expected_yield, accessible_yield;
map<int32_t, int32_t> tree_counts, designated_tree_counts;
vector<df::unit *> citizens;
Units::getCitizens(citizens);
scan_logs(&usable_logs, citizens, &inaccessible_logs);
scan_trees(out, &expected_yield, NULL, false, citizens, &accessible_trees, &inaccessible_trees,
&designated_trees, &accessible_yield, &tree_counts, &designated_tree_counts);
out.print("summary:\n");
out.print(" accessible logs (usable stock): %d\n", usable_logs);
out.print(" inaccessible logs: %d\n", inaccessible_logs);
out.print(" total visible logs: %d\n", usable_logs + inaccessible_logs);
out.print("\n");
out.print(" accessible trees: %d\n", accessible_trees);
out.print(" inaccessible trees: %d\n", inaccessible_trees);
out.print(" total visible trees: %d\n", accessible_trees + inaccessible_trees);
out.print("\n");
out.print(" designated trees: %d\n", designated_trees);
out.print(" expected logs from designated trees: %d\n", expected_yield);
out.print(" expected logs from all accessible trees: %d\n", accessible_yield);
out.print("\n");
out.print(" total trees harvested: %d\n", plotinfo->trees_removed);
out.print("\n");
if (!plotinfo->burrows.list.size()) {
out.print("no burrows defined\n");
return;
}
out.print("\n");
int name_width = 11;
for (auto &burrow : plotinfo->burrows.list) {
name_width = std::max(name_width, (int)burrow->name.size());
}
name_width = -name_width; // left justify
const char *fmt = "%*s %4s %4s %8s %5s %6s %7s\n";
out.print(fmt, name_width, "burrow name", " id ", "chop", "clearcut", "trees", "marked", "protect");
out.print(fmt, name_width, "-----------", "----", "----", "--------", "-----", "------", "-------");
for (auto &burrow : plotinfo->burrows.list) {
bool chop = false;
bool clearcut = false;
bool protect_brewable = false;
bool protect_edible = false;
bool protect_cookable = false;
if (watched_burrows_indices.count(burrow->id)) {
auto &c = watched_burrows[watched_burrows_indices[burrow->id]];
chop = get_config_bool(c, BURROW_CONFIG_CHOP);
clearcut = get_config_bool(c, BURROW_CONFIG_CLEARCUT);
protect_brewable = get_config_bool(c, BURROW_CONFIG_PROTECT_BREWABLE);
protect_edible = get_config_bool(c, BURROW_CONFIG_PROTECT_EDIBLE);
protect_cookable = get_config_bool(c, BURROW_CONFIG_PROTECT_COOKABLE);
}
out.print(fmt, name_width, burrow->name.c_str(), int_to_string(burrow->id).c_str(),
chop ? "[x]" : "[ ]", clearcut ? "[x]" : "[ ]",
int_to_string(tree_counts[burrow->id]).c_str(),
int_to_string(designated_tree_counts[burrow->id]).c_str(),
get_protect_str(protect_brewable, protect_edible, protect_cookable));
}
}
static void autochop_designate(color_ostream &out) {
DEBUG(status,out).print("entering autochop_designate\n");
out.print("designated %d tree(s) for chopping\n", do_cycle(out, true));
}
static void autochop_undesignate(color_ostream &out) {
DEBUG(status,out).print("entering autochop_undesignate\n");
int32_t count = 0;
for (auto plant : world->plants.all) {
if (is_valid_tree(plant) && Designations::unmarkPlant(plant))
++count;
}
out.print("undesignated %d tree(s)\n", count);
}
static void autochop_setTargets(color_ostream &out, int32_t max_logs, int32_t min_logs) {
DEBUG(status,out).print("entering autochop_setTargets\n");
if (max_logs < min_logs || min_logs < 0) {
out.printerr("max and min must be at least 0 and max must be greater than min\n");
return;
}
set_config_val(config, CONFIG_MAX_LOGS, max_logs);
set_config_val(config, CONFIG_MIN_LOGS, min_logs);
// check limits and designate up to the new maximum
autochop_designate(out);
}
static int autochop_getTargets(lua_State *L) {
color_ostream *out = Lua::GetOutput(L);
if (!out)
out = &Core::getInstance().getConsole();
DEBUG(status,*out).print("entering autochop_getTargets\n");
Lua::Push(L, get_config_val(config, CONFIG_MAX_LOGS));
Lua::Push(L, get_config_val(config, CONFIG_MIN_LOGS));
return 2;
}
static int autochop_getLogCounts(lua_State *L) {
color_ostream *out = Lua::GetOutput(L);
if (!out)
out = &Core::getInstance().getConsole();
DEBUG(status,*out).print("entering autochop_getNumLogs\n");
int32_t usable_logs, inaccessible_logs;
vector<df::unit *> citizens;
Units::getCitizens(citizens);
scan_logs(&usable_logs, citizens, &inaccessible_logs);
Lua::Push(L, usable_logs);
Lua::Push(L, inaccessible_logs);
return 2;
}
static void push_burrow_config(lua_State *L, int id, bool chop = false,
bool clearcut = false, bool protect_brewable = false,
bool protect_edible = false, bool protect_cookable = false) {
map<string, int32_t> burrow_config;
burrow_config.emplace("id", id);
burrow_config.emplace("chop", chop);
burrow_config.emplace("clearcut", clearcut);
burrow_config.emplace("protect_brewable", protect_brewable);
burrow_config.emplace("protect_edible", protect_edible);
burrow_config.emplace("protect_cookable", protect_cookable);
Lua::Push(L, burrow_config);
}
static void push_burrow_config(lua_State *L, PersistentDataItem &c) {
push_burrow_config(L, get_config_val(c, BURROW_CONFIG_ID),
get_config_bool(c, BURROW_CONFIG_CHOP),
get_config_bool(c, BURROW_CONFIG_CLEARCUT),
get_config_bool(c, BURROW_CONFIG_PROTECT_BREWABLE),
get_config_bool(c, BURROW_CONFIG_PROTECT_EDIBLE),
get_config_bool(c, BURROW_CONFIG_PROTECT_COOKABLE));
}
static int autochop_getTreeCountsAndBurrowConfigs(lua_State *L) {
color_ostream *out = Lua::GetOutput(L);
if (!out)
out = &Core::getInstance().getConsole();
DEBUG(status,*out).print("entering autochop_getTreeCountsAndBurrowConfigs\n");
validate_burrow_configs(*out);
int32_t accessible_trees, inaccessible_trees;
int32_t designated_trees, expected_yield, accessible_yield;
map<int32_t, int32_t> tree_counts, designated_tree_counts;
vector<df::unit *> citizens;
Units::getCitizens(citizens);
scan_trees(*out, &expected_yield, NULL, false, citizens, &accessible_trees, &inaccessible_trees,
&designated_trees, &accessible_yield, &tree_counts, &designated_tree_counts);
map<string, int32_t> summary;
summary.emplace("accessible_trees", accessible_trees);
summary.emplace("inaccessible_trees", inaccessible_trees);
summary.emplace("designated_trees", designated_trees);
summary.emplace("expected_yield", expected_yield);
summary.emplace("accessible_yield", accessible_yield);
Lua::Push(L, summary);
Lua::Push(L, tree_counts);
Lua::Push(L, designated_tree_counts);
for (auto &burrow : plotinfo->burrows.list) {
int id = burrow->id;
if (watched_burrows_indices.count(id)) {
push_burrow_config(L, watched_burrows[watched_burrows_indices[id]]);
} else {
push_burrow_config(L, id);
}
}
return 3 + plotinfo->burrows.list.size();
}
static int autochop_getBurrowConfig(lua_State *L) {
color_ostream *out = Lua::GetOutput(L);
if (!out)
out = &Core::getInstance().getConsole();
DEBUG(status,*out).print("entering autochop_getBurrowConfig\n");
validate_burrow_configs(*out);
// param can be a name or an id
int id;
if (lua_isnumber(L, -1)) {
id = lua_tointeger(L, -1);
if (!df::burrow::find(id))
return 0;
} else {
const char * name = lua_tostring(L, -1);
if (!name)
return 0;
string nameStr = name;
bool found = false;
for (auto &burrow : plotinfo->burrows.list) {
if (nameStr == burrow->name) {
id = burrow->id;
found = true;
break;
}
}
if (!found)
return 0;
}
if (watched_burrows_indices.count(id)) {
push_burrow_config(L, watched_burrows[watched_burrows_indices[id]]);
} else {
push_burrow_config(L, id);
}
return 1;
}
static void autochop_setBurrowConfig(color_ostream &out, int id, bool chop,
bool clearcut, bool protect_brewable, bool protect_edible,
bool protect_cookable) {
DEBUG(status,out).print("entering autochop_setBurrowConfig\n");
validate_burrow_configs(out);
bool isInvalidBurrow = !df::burrow::find(id);
bool hasNoData = !chop && !clearcut && !protect_brewable && !protect_edible
&& !protect_cookable;
if (isInvalidBurrow || hasNoData) {
remove_burrow_config(out, id);
return;
}
PersistentDataItem &c = ensure_burrow_config(out, id);
set_config_val(c, BURROW_CONFIG_ID, id);
set_config_bool(c, BURROW_CONFIG_CHOP, chop);
set_config_bool(c, BURROW_CONFIG_CLEARCUT, clearcut);
set_config_bool(c, BURROW_CONFIG_PROTECT_BREWABLE, protect_brewable);
set_config_bool(c, BURROW_CONFIG_PROTECT_EDIBLE, protect_edible);
set_config_bool(c, BURROW_CONFIG_PROTECT_COOKABLE, protect_cookable);
}
DFHACK_PLUGIN_LUA_FUNCTIONS {
DFHACK_LUA_FUNCTION(autochop_printStatus),
DFHACK_LUA_FUNCTION(autochop_designate),
DFHACK_LUA_FUNCTION(autochop_undesignate),
DFHACK_LUA_FUNCTION(autochop_setTargets),
DFHACK_LUA_FUNCTION(autochop_setBurrowConfig),
DFHACK_LUA_END
};
DFHACK_PLUGIN_LUA_COMMANDS {
DFHACK_LUA_COMMAND(autochop_getTargets),
DFHACK_LUA_COMMAND(autochop_getLogCounts),
DFHACK_LUA_COMMAND(autochop_getBurrowConfig),
DFHACK_LUA_COMMAND(autochop_getTreeCountsAndBurrowConfigs),
DFHACK_LUA_END
};