2023-02-13 19:45:26 -07:00
|
|
|
#include "buildingplan.h"
|
2023-02-22 16:08:11 -07:00
|
|
|
#include "buildingtypekey.h"
|
|
|
|
#include "defaultitemfilters.h"
|
|
|
|
#include "plannedbuilding.h"
|
2023-02-13 19:45:26 -07:00
|
|
|
|
2023-01-17 00:49:11 -07:00
|
|
|
#include "Debug.h"
|
2023-01-17 05:05:17 -07:00
|
|
|
#include "LuaTools.h"
|
2023-01-17 00:49:11 -07:00
|
|
|
#include "PluginManager.h"
|
|
|
|
|
|
|
|
#include "modules/World.h"
|
|
|
|
|
2023-03-24 16:26:17 -06:00
|
|
|
#include "df/construction_type.h"
|
2023-01-17 05:05:17 -07:00
|
|
|
#include "df/item.h"
|
|
|
|
#include "df/job_item.h"
|
2023-04-11 04:49:28 -06:00
|
|
|
#include "df/organic_mat_category.h"
|
2023-01-17 00:49:11 -07:00
|
|
|
#include "df/world.h"
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
using std::map;
|
2023-03-02 18:50:12 -07:00
|
|
|
using std::set;
|
2023-01-17 00:49:11 -07:00
|
|
|
using std::string;
|
2023-01-17 05:05:17 -07:00
|
|
|
using std::unordered_map;
|
2023-01-17 00:49:11 -07:00
|
|
|
using std::vector;
|
|
|
|
|
|
|
|
using namespace DFHack;
|
|
|
|
|
|
|
|
DFHACK_PLUGIN("buildingplan");
|
|
|
|
DFHACK_PLUGIN_IS_ENABLED(is_enabled);
|
|
|
|
|
|
|
|
REQUIRE_GLOBAL(world);
|
|
|
|
|
|
|
|
namespace DFHack {
|
|
|
|
DBG_DECLARE(buildingplan, status, DebugCategory::LINFO);
|
|
|
|
DBG_DECLARE(buildingplan, cycle, DebugCategory::LINFO);
|
|
|
|
}
|
|
|
|
|
|
|
|
static const string CONFIG_KEY = string(plugin_name) + "/config";
|
2023-02-22 16:08:11 -07:00
|
|
|
const string FILTER_CONFIG_KEY = string(plugin_name) + "/filter";
|
2023-02-13 19:45:26 -07:00
|
|
|
const string BLD_CONFIG_KEY = string(plugin_name) + "/building";
|
2023-01-17 05:05:17 -07:00
|
|
|
|
2023-02-13 19:45:26 -07:00
|
|
|
int get_config_val(PersistentDataItem &c, int index) {
|
2023-01-17 05:05:17 -07:00
|
|
|
if (!c.isValid())
|
2023-01-17 00:49:11 -07:00
|
|
|
return -1;
|
2023-01-17 05:05:17 -07:00
|
|
|
return c.ival(index);
|
|
|
|
}
|
2023-02-13 19:45:26 -07:00
|
|
|
bool get_config_bool(PersistentDataItem &c, int index) {
|
2023-01-17 05:05:17 -07:00
|
|
|
return get_config_val(c, index) == 1;
|
2023-01-17 00:49:11 -07:00
|
|
|
}
|
2023-02-13 19:45:26 -07:00
|
|
|
void set_config_val(PersistentDataItem &c, int index, int value) {
|
2023-01-17 05:05:17 -07:00
|
|
|
if (c.isValid())
|
|
|
|
c.ival(index) = value;
|
2023-01-17 00:49:11 -07:00
|
|
|
}
|
2023-02-13 19:45:26 -07:00
|
|
|
void set_config_bool(PersistentDataItem &c, int index, bool value) {
|
2023-01-17 05:05:17 -07:00
|
|
|
set_config_val(c, index, value ? 1 : 0);
|
2023-01-17 00:49:11 -07:00
|
|
|
}
|
2023-01-17 05:05:17 -07:00
|
|
|
|
|
|
|
static PersistentDataItem config;
|
2023-02-15 17:54:38 -07:00
|
|
|
// for use in counting available materials for the UI
|
2023-03-02 18:50:12 -07:00
|
|
|
static map<string, std::pair<MaterialInfo, string>> mat_cache;
|
2023-02-22 16:08:11 -07:00
|
|
|
static unordered_map<BuildingTypeKey, vector<const df::job_item *>, BuildingTypeKeyHash> job_item_cache;
|
2023-02-19 01:57:30 -07:00
|
|
|
static unordered_map<BuildingTypeKey, HeatSafety, BuildingTypeKeyHash> cur_heat_safety;
|
2023-02-22 16:08:11 -07:00
|
|
|
static unordered_map<BuildingTypeKey, DefaultItemFilters, BuildingTypeKeyHash> cur_item_filters;
|
2023-01-17 05:05:17 -07:00
|
|
|
// building id -> PlannedBuilding
|
2023-02-15 17:54:38 -07:00
|
|
|
static unordered_map<int32_t, PlannedBuilding> planned_buildings;
|
2023-01-17 05:05:17 -07:00
|
|
|
// vector id -> filter bucket -> queue of (building id, job_item index)
|
2023-02-15 17:54:38 -07:00
|
|
|
static Tasks tasks;
|
2023-01-17 05:05:17 -07:00
|
|
|
|
|
|
|
// note that this just removes the PlannedBuilding. the tasks will get dropped
|
|
|
|
// as we discover them in the tasks queues and they fail to be found in planned_buildings.
|
|
|
|
// this "lazy" task cleaning algorithm works because there is no way to
|
|
|
|
// re-register a building once it has been removed -- if it has been booted out of
|
|
|
|
// planned_buildings, then it has either been built or desroyed. therefore there is
|
|
|
|
// no chance of duplicate tasks getting added to the tasks queues.
|
|
|
|
void PlannedBuilding::remove(color_ostream &out) {
|
|
|
|
DEBUG(status,out).print("removing persistent data for building %d\n", id);
|
2023-02-15 17:54:38 -07:00
|
|
|
World::DeletePersistentData(bld_config);
|
2023-03-27 03:24:56 -06:00
|
|
|
if (planned_buildings.count(id))
|
2023-01-17 05:05:17 -07:00
|
|
|
planned_buildings.erase(id);
|
2023-01-17 00:49:11 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
static const int32_t CYCLE_TICKS = 600; // twice per game day
|
|
|
|
static int32_t cycle_timestamp = 0; // world->frame_counter at last cycle
|
|
|
|
|
2023-02-21 19:05:15 -07:00
|
|
|
static bool call_buildingplan_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 buildingplan 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.buildingplan", fn_name,
|
|
|
|
nargs, nres,
|
|
|
|
std::forward<Lua::LuaLambda&&>(args_lambda),
|
|
|
|
std::forward<Lua::LuaLambda&&>(res_lambda));
|
|
|
|
}
|
|
|
|
|
|
|
|
static int get_num_filters(color_ostream &out, BuildingTypeKey key) {
|
|
|
|
int num_filters = 0;
|
|
|
|
if (!call_buildingplan_lua(&out, "get_num_filters", 3, 1,
|
|
|
|
[&](lua_State *L) {
|
|
|
|
Lua::Push(L, std::get<0>(key));
|
|
|
|
Lua::Push(L, std::get<1>(key));
|
|
|
|
Lua::Push(L, std::get<2>(key));
|
|
|
|
},
|
|
|
|
[&](lua_State *L) {
|
|
|
|
num_filters = lua_tonumber(L, -1);
|
|
|
|
})) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return num_filters;
|
|
|
|
}
|
|
|
|
|
2023-02-22 16:08:11 -07:00
|
|
|
static const vector<const df::job_item *> & get_job_items(color_ostream &out, BuildingTypeKey key) {
|
2023-02-21 19:05:15 -07:00
|
|
|
if (job_item_cache.count(key))
|
|
|
|
return job_item_cache[key];
|
|
|
|
const int num_filters = get_num_filters(out, key);
|
|
|
|
auto &jitems = job_item_cache[key];
|
|
|
|
for (int index = 0; index < num_filters; ++index) {
|
|
|
|
bool failed = false;
|
|
|
|
if (!call_buildingplan_lua(&out, "get_job_item", 4, 1,
|
|
|
|
[&](lua_State *L) {
|
|
|
|
Lua::Push(L, std::get<0>(key));
|
|
|
|
Lua::Push(L, std::get<1>(key));
|
|
|
|
Lua::Push(L, std::get<2>(key));
|
|
|
|
Lua::Push(L, index+1);
|
|
|
|
},
|
|
|
|
[&](lua_State *L) {
|
|
|
|
df::job_item *jitem = Lua::GetDFObject<df::job_item>(L, -1);
|
2023-03-23 11:48:33 -06:00
|
|
|
DEBUG(status,out).print("retrieving job_item for (%d, %d, %d) index=%d: 0x%p\n",
|
2023-02-21 19:05:15 -07:00
|
|
|
std::get<0>(key), std::get<1>(key), std::get<2>(key), index, jitem);
|
|
|
|
if (!jitem)
|
|
|
|
failed = true;
|
|
|
|
else
|
|
|
|
jitems.emplace_back(jitem);
|
|
|
|
}) || failed) {
|
|
|
|
jitems.clear();
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return jitems;
|
|
|
|
}
|
|
|
|
|
2023-03-05 19:17:10 -07:00
|
|
|
static const df::dfhack_material_category stone_cat(df::dfhack_material_category::mask_stone);
|
|
|
|
static const df::dfhack_material_category wood_cat(df::dfhack_material_category::mask_wood);
|
|
|
|
static const df::dfhack_material_category metal_cat(df::dfhack_material_category::mask_metal);
|
|
|
|
static const df::dfhack_material_category glass_cat(df::dfhack_material_category::mask_glass);
|
2023-04-17 02:05:25 -06:00
|
|
|
static const df::dfhack_material_category gem_cat(df::dfhack_material_category::mask_gem);
|
2023-03-12 16:13:57 -06:00
|
|
|
static const df::dfhack_material_category clay_cat(df::dfhack_material_category::mask_clay);
|
2023-04-11 04:49:28 -06:00
|
|
|
static const df::dfhack_material_category cloth_cat(df::dfhack_material_category::mask_cloth);
|
|
|
|
static const df::dfhack_material_category silk_cat(df::dfhack_material_category::mask_silk);
|
2023-04-21 12:54:38 -06:00
|
|
|
static const df::dfhack_material_category yarn_cat(df::dfhack_material_category::mask_yarn);
|
2023-03-02 05:38:40 -07:00
|
|
|
|
2023-03-05 19:17:10 -07:00
|
|
|
static void cache_matched(int16_t type, int32_t index) {
|
2023-03-02 05:38:40 -07:00
|
|
|
MaterialInfo mi;
|
|
|
|
mi.decode(type, index);
|
2023-03-02 18:50:12 -07:00
|
|
|
if (mi.matches(stone_cat)) {
|
2023-03-12 16:13:57 -06:00
|
|
|
DEBUG(status).print("cached stone material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
2023-03-02 18:50:12 -07:00
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "stone"));
|
|
|
|
} else if (mi.matches(wood_cat)) {
|
2023-03-12 16:13:57 -06:00
|
|
|
DEBUG(status).print("cached wood material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
2023-03-02 18:50:12 -07:00
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "wood"));
|
|
|
|
} else if (mi.matches(metal_cat)) {
|
2023-03-12 16:13:57 -06:00
|
|
|
DEBUG(status).print("cached metal material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
2023-03-02 18:50:12 -07:00
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "metal"));
|
2023-03-05 19:17:10 -07:00
|
|
|
} else if (mi.matches(glass_cat)) {
|
2023-03-12 16:13:57 -06:00
|
|
|
DEBUG(status).print("cached glass material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
2023-03-05 19:17:10 -07:00
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "glass"));
|
2023-04-17 02:05:25 -06:00
|
|
|
} else if (mi.matches(gem_cat)) {
|
|
|
|
DEBUG(status).print("cached gem material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "gem"));
|
2023-03-12 16:13:57 -06:00
|
|
|
} else if (mi.matches(clay_cat)) {
|
|
|
|
DEBUG(status).print("cached clay material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "clay"));
|
2023-04-11 04:49:28 -06:00
|
|
|
} else if (mi.matches(cloth_cat)) {
|
|
|
|
DEBUG(status).print("cached cloth material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "cloth"));
|
|
|
|
} else if (mi.matches(silk_cat)) {
|
|
|
|
DEBUG(status).print("cached silk material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "silk"));
|
2023-04-21 12:54:38 -06:00
|
|
|
} else if (mi.matches(yarn_cat)) {
|
|
|
|
DEBUG(status).print("cached yarn material: %s (%d, %d)\n", mi.toString().c_str(), type, index);
|
|
|
|
mat_cache.emplace(mi.toString(), std::make_pair(mi, "yarn"));
|
2023-03-02 05:38:40 -07:00
|
|
|
}
|
|
|
|
else
|
|
|
|
TRACE(status).print("not matched: %s\n", mi.toString().c_str());
|
|
|
|
}
|
|
|
|
|
2023-04-11 04:49:28 -06:00
|
|
|
static void load_organic_material_cache(df::organic_mat_category cat) {
|
|
|
|
auto& mat_tab = world->raws.mat_table;
|
|
|
|
auto& cat_vec = mat_tab.organic_types[cat];
|
|
|
|
auto& cat_indices = mat_tab.organic_indexes[cat];
|
|
|
|
for (size_t i = 0; i < cat_vec.size(); i++) {
|
|
|
|
cache_matched(cat_vec[i], cat_indices[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-03-02 05:38:40 -07:00
|
|
|
static void load_material_cache() {
|
|
|
|
df::world_raws &raws = world->raws;
|
|
|
|
for (int i = 1; i < DFHack::MaterialInfo::NUM_BUILTIN; ++i)
|
|
|
|
if (raws.mat_table.builtin[i])
|
|
|
|
cache_matched(i, -1);
|
|
|
|
|
|
|
|
for (size_t i = 0; i < raws.inorganics.size(); i++)
|
|
|
|
cache_matched(0, i);
|
|
|
|
|
2023-04-11 04:49:28 -06:00
|
|
|
load_organic_material_cache(df::organic_mat_category::Wood);
|
|
|
|
load_organic_material_cache(df::organic_mat_category::PlantFiber);
|
|
|
|
load_organic_material_cache(df::organic_mat_category::Silk);
|
2023-04-21 12:54:38 -06:00
|
|
|
load_organic_material_cache(df::organic_mat_category::Yarn);
|
2023-03-02 05:38:40 -07:00
|
|
|
}
|
|
|
|
|
2023-02-21 19:05:15 -07:00
|
|
|
static HeatSafety get_heat_safety_filter(const BuildingTypeKey &key) {
|
2023-09-03 12:49:13 -06:00
|
|
|
if (cur_heat_safety.count(key))
|
|
|
|
return cur_heat_safety.at(key);
|
2023-02-21 19:05:15 -07:00
|
|
|
return HEAT_SAFETY_ANY;
|
|
|
|
}
|
|
|
|
|
2023-02-22 16:08:11 -07:00
|
|
|
static DefaultItemFilters & get_item_filters(color_ostream &out, const BuildingTypeKey &key) {
|
2023-02-21 19:05:15 -07:00
|
|
|
if (cur_item_filters.count(key))
|
2023-02-22 16:08:11 -07:00
|
|
|
return cur_item_filters.at(key);
|
|
|
|
cur_item_filters.emplace(key, DefaultItemFilters(out, key, get_job_items(out, key)));
|
|
|
|
return cur_item_filters.at(key);
|
2023-02-21 19:05:15 -07:00
|
|
|
}
|
|
|
|
|
2023-01-17 00:49:11 -07:00
|
|
|
static command_result do_command(color_ostream &out, vector<string> ¶meters);
|
2023-02-13 19:45:26 -07:00
|
|
|
void buildingplan_cycle(color_ostream &out, Tasks &tasks,
|
2023-03-27 03:24:56 -06:00
|
|
|
unordered_map<int32_t, PlannedBuilding> &planned_buildings, bool unsuspend_on_finalize);
|
2023-02-13 19:45:26 -07:00
|
|
|
|
2023-03-27 03:24:56 -06:00
|
|
|
static bool registerPlannedBuilding(color_ostream &out, PlannedBuilding & pb, bool unsuspend_on_finalize);
|
2023-01-17 00:49:11 -07:00
|
|
|
|
|
|
|
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,
|
|
|
|
"Plan building placement before you have materials.",
|
|
|
|
do_command));
|
|
|
|
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_enable(color_ostream &out, bool enable) {
|
|
|
|
if (enable != is_enabled) {
|
|
|
|
is_enabled = enable;
|
|
|
|
DEBUG(status,out).print("%s from the API; persisting\n",
|
|
|
|
is_enabled ? "enabled" : "disabled");
|
|
|
|
} 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;
|
|
|
|
}
|
|
|
|
|
2023-02-15 17:54:38 -07:00
|
|
|
static void validate_config(color_ostream &out, bool verbose = false) {
|
|
|
|
if (get_config_bool(config, CONFIG_BLOCKS)
|
|
|
|
|| get_config_bool(config, CONFIG_BOULDERS)
|
|
|
|
|| get_config_bool(config, CONFIG_LOGS)
|
|
|
|
|| get_config_bool(config, CONFIG_BARS))
|
|
|
|
return;
|
|
|
|
|
|
|
|
if (verbose)
|
|
|
|
out.printerr("all contruction materials disabled; resetting config\n");
|
|
|
|
|
|
|
|
set_config_bool(config, CONFIG_BLOCKS, true);
|
|
|
|
set_config_bool(config, CONFIG_BOULDERS, true);
|
|
|
|
set_config_bool(config, CONFIG_LOGS, true);
|
|
|
|
set_config_bool(config, CONFIG_BARS, false);
|
|
|
|
}
|
|
|
|
|
2023-03-15 15:02:59 -06:00
|
|
|
static void reset_filters(color_ostream &out) {
|
|
|
|
cur_heat_safety.clear();
|
|
|
|
cur_item_filters.clear();
|
2023-02-17 00:02:34 -07:00
|
|
|
call_buildingplan_lua(&out, "signal_reset");
|
2023-03-15 15:02:59 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
static void clear_state(color_ostream &out) {
|
2023-02-17 00:02:34 -07:00
|
|
|
planned_buildings.clear();
|
|
|
|
tasks.clear();
|
2023-02-21 19:05:15 -07:00
|
|
|
for (auto &entry : job_item_cache ) {
|
2023-02-15 17:54:38 -07:00
|
|
|
for (auto &jitem : entry.second) {
|
|
|
|
delete jitem;
|
|
|
|
}
|
|
|
|
}
|
2023-02-21 19:05:15 -07:00
|
|
|
job_item_cache.clear();
|
2023-03-02 05:38:40 -07:00
|
|
|
mat_cache.clear();
|
2023-03-15 15:02:59 -06:00
|
|
|
reset_filters(out);
|
|
|
|
call_buildingplan_lua(&out, "reload_pens");
|
2023-02-15 17:54:38 -07:00
|
|
|
}
|
|
|
|
|
2023-03-24 16:26:17 -06:00
|
|
|
static int16_t get_subtype(df::building *bld) {
|
|
|
|
if (!bld)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
int16_t subtype = bld->getSubtype();
|
|
|
|
if (bld->getType() == df::building_type::Construction &&
|
|
|
|
subtype >= df::construction_type::UpStair &&
|
|
|
|
subtype <= df::construction_type::UpDownStair)
|
|
|
|
subtype = df::construction_type::UpDownStair;
|
|
|
|
return subtype;
|
|
|
|
}
|
|
|
|
|
2023-03-27 03:24:56 -06:00
|
|
|
static bool is_suspendmanager_enabled(color_ostream &out) {
|
|
|
|
bool suspendmanager_enabled = false;
|
|
|
|
call_buildingplan_lua(&out, "is_suspendmanager_enabled", 0, 1,
|
|
|
|
Lua::DEFAULT_LUA_LAMBDA,
|
|
|
|
[&](lua_State *L){
|
|
|
|
suspendmanager_enabled = lua_toboolean(L, -1);
|
|
|
|
});
|
|
|
|
return suspendmanager_enabled;
|
|
|
|
}
|
|
|
|
|
2023-01-17 00:49:11 -07:00
|
|
|
DFhackCExport command_result plugin_load_data (color_ostream &out) {
|
2023-02-03 00:44:33 -07:00
|
|
|
cycle_timestamp = 0;
|
2023-01-17 00:49:11 -07:00
|
|
|
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);
|
|
|
|
}
|
2023-02-15 17:54:38 -07:00
|
|
|
validate_config(out);
|
2023-01-17 00:49:11 -07:00
|
|
|
|
2023-01-17 22:55:27 -07:00
|
|
|
DEBUG(status,out).print("loading persisted state\n");
|
2023-02-17 00:02:34 -07:00
|
|
|
clear_state(out);
|
2023-02-22 16:08:11 -07:00
|
|
|
|
2023-03-02 05:38:40 -07:00
|
|
|
load_material_cache();
|
|
|
|
|
2023-02-22 16:08:11 -07:00
|
|
|
vector<PersistentDataItem> filter_configs;
|
|
|
|
World::GetPersistentData(&filter_configs, FILTER_CONFIG_KEY);
|
|
|
|
for (auto &cfg : filter_configs) {
|
|
|
|
BuildingTypeKey key = DefaultItemFilters::getKey(cfg);
|
|
|
|
cur_item_filters.emplace(key, DefaultItemFilters(out, cfg, get_job_items(out, key)));
|
|
|
|
}
|
|
|
|
|
2023-01-17 22:55:27 -07:00
|
|
|
vector<PersistentDataItem> building_configs;
|
|
|
|
World::GetPersistentData(&building_configs, BLD_CONFIG_KEY);
|
2023-01-17 05:05:17 -07:00
|
|
|
const size_t num_building_configs = building_configs.size();
|
2023-03-27 03:24:56 -06:00
|
|
|
bool unsuspend_on_finalize = !is_suspendmanager_enabled(out);
|
2023-01-18 01:08:21 -07:00
|
|
|
for (size_t idx = 0; idx < num_building_configs; ++idx) {
|
2023-02-15 20:10:42 -07:00
|
|
|
PlannedBuilding pb(out, building_configs[idx]);
|
2023-02-21 19:05:15 -07:00
|
|
|
df::building *bld = df::building::find(pb.id);
|
|
|
|
if (!bld) {
|
2023-03-23 11:48:33 -06:00
|
|
|
DEBUG(status,out).print("building %d no longer exists; skipping\n", pb.id);
|
2023-02-22 16:08:11 -07:00
|
|
|
pb.remove(out);
|
|
|
|
continue;
|
2023-02-21 19:05:15 -07:00
|
|
|
}
|
2023-03-24 16:26:17 -06:00
|
|
|
BuildingTypeKey key(bld->getType(), get_subtype(bld), bld->getCustomType());
|
2023-02-22 16:08:11 -07:00
|
|
|
if (pb.item_filters.size() != get_item_filters(out, key).getItemFilters().size()) {
|
2023-02-21 19:05:15 -07:00
|
|
|
WARN(status).print("loaded state for building %d doesn't match world\n", pb.id);
|
2023-02-22 16:08:11 -07:00
|
|
|
pb.remove(out);
|
|
|
|
continue;
|
|
|
|
}
|
2023-03-27 03:24:56 -06:00
|
|
|
registerPlannedBuilding(out, pb, unsuspend_on_finalize);
|
2023-01-18 01:08:21 -07:00
|
|
|
}
|
2023-01-17 05:05:17 -07:00
|
|
|
|
2023-01-17 00:49:11 -07:00
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
2023-02-15 17:54:38 -07:00
|
|
|
static bool cycle_requested = false;
|
|
|
|
|
|
|
|
static void do_cycle(color_ostream &out) {
|
|
|
|
// mark that we have recently run
|
|
|
|
cycle_timestamp = world->frame_counter;
|
|
|
|
cycle_requested = false;
|
|
|
|
|
2023-03-27 03:24:56 -06:00
|
|
|
bool unsuspend_on_finalize = !is_suspendmanager_enabled(out);
|
|
|
|
buildingplan_cycle(out, tasks, planned_buildings, unsuspend_on_finalize);
|
2023-02-16 22:17:55 -07:00
|
|
|
call_buildingplan_lua(&out, "signal_reset");
|
2023-02-15 17:54:38 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_onupdate(color_ostream &out) {
|
|
|
|
if (!Core::getInstance().isWorldLoaded())
|
|
|
|
return CR_OK;
|
|
|
|
|
|
|
|
if (is_enabled &&
|
|
|
|
(cycle_requested || world->frame_counter - cycle_timestamp >= CYCLE_TICKS))
|
|
|
|
do_cycle(out);
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
2023-01-17 00:49:11 -07:00
|
|
|
static command_result do_command(color_ostream &out, vector<string> ¶meters) {
|
|
|
|
CoreSuspender suspend;
|
|
|
|
|
|
|
|
if (!Core::getInstance().isWorldLoaded()) {
|
2023-01-17 22:55:27 -07:00
|
|
|
out.printerr("Cannot configure %s without a loaded world.\n", plugin_name);
|
2023-01-17 00:49:11 -07:00
|
|
|
return CR_FAILURE;
|
|
|
|
}
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
bool show_help = false;
|
|
|
|
if (!call_buildingplan_lua(&out, "parse_commandline", parameters.size(), 1,
|
|
|
|
[&](lua_State *L) {
|
|
|
|
for (const string ¶m : parameters)
|
|
|
|
Lua::Push(L, param);
|
|
|
|
},
|
|
|
|
[&](lua_State *L) {
|
|
|
|
show_help = !lua_toboolean(L, -1);
|
|
|
|
})) {
|
|
|
|
return CR_FAILURE;
|
|
|
|
}
|
2023-01-17 00:49:11 -07:00
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
return show_help ? CR_WRONG_USAGE : CR_OK;
|
2023-01-17 00:49:11 -07:00
|
|
|
}
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
/////////////////////////////////////////////////////
|
|
|
|
// Lua API
|
|
|
|
// core will already be suspended when coming in through here
|
|
|
|
//
|
2023-01-17 00:49:11 -07:00
|
|
|
|
2023-02-27 00:38:00 -07:00
|
|
|
static string getBucket(const df::job_item & ji, const PlannedBuilding & pb, int idx) {
|
2023-03-13 14:35:12 -06:00
|
|
|
if (idx < 0 || (size_t)idx >= pb.item_filters.size())
|
2023-02-27 00:38:00 -07:00
|
|
|
return "INVALID";
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
std::ostringstream ser;
|
2023-01-17 00:49:11 -07:00
|
|
|
|
2023-02-27 00:38:00 -07:00
|
|
|
// put elements in front that significantly affect the difficulty of matching
|
|
|
|
// the filter. ensure the lexicographically "less" value is the pickier value.
|
|
|
|
const ItemFilter & item_filter = pb.item_filters[idx];
|
|
|
|
|
|
|
|
if (item_filter.getDecoratedOnly())
|
|
|
|
ser << "Da";
|
|
|
|
else
|
|
|
|
ser << "Db";
|
|
|
|
|
|
|
|
if (ji.flags2.bits.magma_safe || pb.heat_safety == HEAT_SAFETY_MAGMA)
|
|
|
|
ser << "Ha";
|
|
|
|
else if (ji.flags2.bits.fire_safe || pb.heat_safety == HEAT_SAFETY_FIRE)
|
|
|
|
ser << "Hb";
|
|
|
|
else
|
|
|
|
ser << "Hc";
|
|
|
|
|
|
|
|
size_t num_materials = item_filter.getMaterials().size();
|
2023-03-15 01:29:27 -06:00
|
|
|
if (num_materials == 0 || num_materials >= 9 || !item_filter.getMaterialMask().whole)
|
2023-02-27 00:38:00 -07:00
|
|
|
ser << "M9";
|
|
|
|
else
|
|
|
|
ser << "M" << num_materials;
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
// pull out and serialize only known relevant fields. if we miss a few, then
|
|
|
|
// the filter bucket will be slighly less specific than it could be, but
|
|
|
|
// that's probably ok. we'll just end up bucketing slightly different items
|
|
|
|
// together. this is only a problem if the different filter at the front of
|
|
|
|
// the queue doesn't match any available items and blocks filters behind it
|
|
|
|
// that could be matched.
|
|
|
|
ser << ji.item_type << ':' << ji.item_subtype << ':' << ji.mat_type << ':'
|
|
|
|
<< ji.mat_index << ':' << ji.flags1.whole << ':' << ji.flags2.whole
|
|
|
|
<< ':' << ji.flags3.whole << ':' << ji.flags4 << ':' << ji.flags5 << ':'
|
|
|
|
<< ji.metal_ore << ':' << ji.has_tool_use;
|
|
|
|
|
2023-02-27 00:38:00 -07:00
|
|
|
ser << ':' << item_filter.serialize();
|
|
|
|
|
2023-03-15 01:29:27 -06:00
|
|
|
for (auto &special : pb.specials)
|
|
|
|
ser << ':' << special;
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
return ser.str();
|
2023-01-17 00:49:11 -07:00
|
|
|
}
|
2023-01-17 05:05:17 -07:00
|
|
|
|
|
|
|
// get a list of item vectors that we should search for matches
|
2023-03-24 15:36:06 -06:00
|
|
|
vector<df::job_item_vector_id> getVectorIds(color_ostream &out, const df::job_item *job_item, bool ignore_filters) {
|
2023-01-17 05:05:17 -07:00
|
|
|
std::vector<df::job_item_vector_id> ret;
|
|
|
|
|
|
|
|
// if the filter already has the vector_id set to something specific, use it
|
|
|
|
if (job_item->vector_id > df::job_item_vector_id::IN_PLAY)
|
|
|
|
{
|
|
|
|
DEBUG(status,out).print("using vector_id from job_item: %s\n",
|
|
|
|
ENUM_KEY_STR(job_item_vector_id, job_item->vector_id).c_str());
|
|
|
|
ret.push_back(job_item->vector_id);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2023-03-13 14:35:12 -06:00
|
|
|
// if the filter is for building material, refer to our global settings for
|
2023-01-17 05:05:17 -07:00
|
|
|
// which vectors to search
|
|
|
|
if (job_item->flags2.bits.building_material)
|
|
|
|
{
|
2023-03-24 15:36:06 -06:00
|
|
|
if (ignore_filters || get_config_bool(config, CONFIG_BLOCKS))
|
2023-01-17 05:05:17 -07:00
|
|
|
ret.push_back(df::job_item_vector_id::BLOCKS);
|
2023-03-24 15:36:06 -06:00
|
|
|
if (ignore_filters || get_config_bool(config, CONFIG_BOULDERS))
|
2023-01-17 05:05:17 -07:00
|
|
|
ret.push_back(df::job_item_vector_id::BOULDER);
|
2023-03-24 15:36:06 -06:00
|
|
|
if (ignore_filters || get_config_bool(config, CONFIG_LOGS))
|
2023-01-17 05:05:17 -07:00
|
|
|
ret.push_back(df::job_item_vector_id::WOOD);
|
2023-03-24 15:36:06 -06:00
|
|
|
if (ignore_filters || get_config_bool(config, CONFIG_BARS))
|
2023-01-17 05:05:17 -07:00
|
|
|
ret.push_back(df::job_item_vector_id::BAR);
|
|
|
|
}
|
|
|
|
|
|
|
|
// fall back to IN_PLAY if no other vector was appropriate
|
|
|
|
if (ret.empty())
|
|
|
|
ret.push_back(df::job_item_vector_id::IN_PLAY);
|
|
|
|
return ret;
|
|
|
|
}
|
2023-02-15 20:10:42 -07:00
|
|
|
|
2023-03-27 03:24:56 -06:00
|
|
|
static bool registerPlannedBuilding(color_ostream &out, PlannedBuilding & pb, bool unsuspend_on_finalize) {
|
2023-01-17 05:05:17 -07:00
|
|
|
df::building * bld = pb.getBuildingIfValidOrRemoveIfNot(out);
|
|
|
|
if (!bld)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (bld->jobs.size() != 1) {
|
|
|
|
DEBUG(status,out).print("unexpected number of jobs: want 1, got %zu\n", bld->jobs.size());
|
|
|
|
return false;
|
|
|
|
}
|
2023-02-17 20:16:45 -07:00
|
|
|
|
2023-03-27 03:24:56 -06:00
|
|
|
// suspend jobs
|
|
|
|
for (auto job : bld->jobs)
|
|
|
|
job->flags.bits.suspend = true;
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
auto job_items = bld->jobs[0]->job_items;
|
2023-02-17 20:16:45 -07:00
|
|
|
if (isJobReady(out, job_items)) {
|
|
|
|
// all items are already attached
|
2023-03-27 03:24:56 -06:00
|
|
|
finalizeBuilding(out, bld, unsuspend_on_finalize);
|
|
|
|
pb.remove(out);
|
2023-02-17 20:16:45 -07:00
|
|
|
return true;
|
2023-01-17 05:05:17 -07:00
|
|
|
}
|
2023-02-17 20:16:45 -07:00
|
|
|
|
2023-03-13 14:35:12 -06:00
|
|
|
int num_job_items = (int)job_items.size();
|
2023-01-17 05:05:17 -07:00
|
|
|
int32_t id = bld->id;
|
|
|
|
for (int job_item_idx = 0; job_item_idx < num_job_items; ++job_item_idx) {
|
2023-03-13 14:35:12 -06:00
|
|
|
int rev_jitem_index = num_job_items - (job_item_idx+1);
|
|
|
|
auto job_item = job_items[rev_jitem_index];
|
2023-02-27 00:38:00 -07:00
|
|
|
auto bucket = getBucket(*job_item, pb, job_item_idx);
|
2023-01-17 05:05:17 -07:00
|
|
|
|
|
|
|
// if there are multiple vector_ids, schedule duplicate tasks. after
|
|
|
|
// the correct number of items are matched, the extras will get popped
|
|
|
|
// as invalid
|
2023-02-16 18:25:15 -07:00
|
|
|
for (auto vector_id : pb.vector_ids[job_item_idx]) {
|
2023-01-17 05:05:17 -07:00
|
|
|
for (int item_num = 0; item_num < job_item->quantity; ++item_num) {
|
2023-03-13 14:35:12 -06:00
|
|
|
tasks[vector_id][bucket].emplace_back(id, rev_jitem_index);
|
2023-01-17 05:05:17 -07:00
|
|
|
DEBUG(status,out).print("added task: %s/%s/%d,%d; "
|
2023-03-16 20:55:12 -06:00
|
|
|
"%zu vector(s), %zu filter bucket(s), %zu task(s) in bucket\n",
|
2023-01-17 05:05:17 -07:00
|
|
|
ENUM_KEY_STR(job_item_vector_id, vector_id).c_str(),
|
2023-03-13 14:35:12 -06:00
|
|
|
bucket.c_str(), id, rev_jitem_index, tasks.size(),
|
2023-01-17 05:05:17 -07:00
|
|
|
tasks[vector_id].size(), tasks[vector_id][bucket].size());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// add the planned buildings to our register
|
|
|
|
planned_buildings.emplace(bld->id, pb);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-02-16 04:08:55 -07:00
|
|
|
static string get_desc_string(color_ostream &out, df::job_item *jitem,
|
|
|
|
const vector<df::job_item_vector_id> &vec_ids) {
|
|
|
|
vector<string> descs;
|
|
|
|
for (auto &vec_id : vec_ids) {
|
|
|
|
df::job_item jitem_copy = *jitem;
|
|
|
|
jitem_copy.vector_id = vec_id;
|
|
|
|
call_buildingplan_lua(&out, "get_desc", 1, 1,
|
|
|
|
[&](lua_State *L) { Lua::Push(L, &jitem_copy); },
|
|
|
|
[&](lua_State *L) {
|
|
|
|
descs.emplace_back(lua_tostring(L, -1)); });
|
|
|
|
}
|
|
|
|
return join_strings(" or ", descs);
|
|
|
|
}
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
static void printStatus(color_ostream &out) {
|
|
|
|
DEBUG(status,out).print("entering buildingplan_printStatus\n");
|
|
|
|
out.print("buildingplan is %s\n\n", is_enabled ? "enabled" : "disabled");
|
|
|
|
out.print("Current settings:\n");
|
|
|
|
out.print(" use blocks: %s\n", get_config_bool(config, CONFIG_BLOCKS) ? "yes" : "no");
|
|
|
|
out.print(" use boulders: %s\n", get_config_bool(config, CONFIG_BOULDERS) ? "yes" : "no");
|
|
|
|
out.print(" use logs: %s\n", get_config_bool(config, CONFIG_LOGS) ? "yes" : "no");
|
|
|
|
out.print(" use bars: %s\n", get_config_bool(config, CONFIG_BARS) ? "yes" : "no");
|
|
|
|
out.print("\n");
|
2023-02-08 19:47:10 -07:00
|
|
|
|
2023-03-10 02:04:23 -07:00
|
|
|
size_t bld_count = 0;
|
2023-02-08 19:47:10 -07:00
|
|
|
map<string, int32_t> counts;
|
|
|
|
int32_t total = 0;
|
2023-02-16 04:08:55 -07:00
|
|
|
for (auto &entry : planned_buildings) {
|
|
|
|
auto &pb = entry.second;
|
2023-03-10 02:04:23 -07:00
|
|
|
// don't actually remove bad buildings from the list while we're
|
|
|
|
// actively iterating through that list
|
|
|
|
auto bld = pb.getBuildingIfValidOrRemoveIfNot(out, true);
|
2023-02-16 04:08:55 -07:00
|
|
|
if (!bld || bld->jobs.size() != 1)
|
|
|
|
continue;
|
|
|
|
auto &job_items = bld->jobs[0]->job_items;
|
2023-03-13 14:35:12 -06:00
|
|
|
const size_t num_job_items = job_items.size();
|
|
|
|
if (num_job_items != pb.vector_ids.size())
|
2023-02-16 04:08:55 -07:00
|
|
|
continue;
|
2023-03-10 02:04:23 -07:00
|
|
|
++bld_count;
|
2023-02-16 04:08:55 -07:00
|
|
|
int job_item_idx = 0;
|
|
|
|
for (auto &vec_ids : pb.vector_ids) {
|
2023-03-13 14:35:12 -06:00
|
|
|
auto &jitem = job_items[num_job_items - (job_item_idx+1)];
|
2023-02-16 04:08:55 -07:00
|
|
|
int32_t quantity = jitem->quantity;
|
|
|
|
if (quantity) {
|
|
|
|
counts[get_desc_string(out, jitem, vec_ids)] += quantity;
|
|
|
|
total += quantity;
|
2023-02-08 19:47:10 -07:00
|
|
|
}
|
2023-03-13 14:35:12 -06:00
|
|
|
++job_item_idx;
|
2023-02-08 19:47:10 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-03-10 02:04:23 -07:00
|
|
|
if (bld_count) {
|
2023-02-16 18:25:15 -07:00
|
|
|
out.print("Waiting for %d item(s) to be produced for %zd building(s):\n",
|
2023-03-10 02:04:23 -07:00
|
|
|
total, bld_count);
|
2023-02-16 18:25:15 -07:00
|
|
|
for (auto &count : counts)
|
2023-02-18 02:25:24 -07:00
|
|
|
out.print(" %3d %s%s\n", count.second, count.first.c_str(), count.second == 1 ? "" : "s");
|
2023-02-16 18:25:15 -07:00
|
|
|
} else {
|
|
|
|
out.print("Currently no planned buildings\n");
|
|
|
|
}
|
2023-02-08 19:47:10 -07:00
|
|
|
out.print("\n");
|
2023-01-17 05:05:17 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
static bool setSetting(color_ostream &out, string name, bool value) {
|
2023-01-18 01:08:21 -07:00
|
|
|
DEBUG(status,out).print("entering setSetting (%s -> %s)\n", name.c_str(), value ? "true" : "false");
|
2023-01-17 05:05:17 -07:00
|
|
|
if (name == "blocks")
|
|
|
|
set_config_bool(config, CONFIG_BLOCKS, value);
|
|
|
|
else if (name == "boulders")
|
|
|
|
set_config_bool(config, CONFIG_BOULDERS, value);
|
|
|
|
else if (name == "logs")
|
|
|
|
set_config_bool(config, CONFIG_LOGS, value);
|
|
|
|
else if (name == "bars")
|
|
|
|
set_config_bool(config, CONFIG_BARS, value);
|
|
|
|
else {
|
|
|
|
out.printerr("unrecognized setting: '%s'\n", name.c_str());
|
|
|
|
return false;
|
|
|
|
}
|
2023-02-15 17:54:38 -07:00
|
|
|
|
|
|
|
validate_config(out, true);
|
2023-02-16 22:17:55 -07:00
|
|
|
call_buildingplan_lua(&out, "signal_reset");
|
2023-01-17 05:05:17 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-03-15 15:02:59 -06:00
|
|
|
static void resetFilters(color_ostream &out) {
|
|
|
|
DEBUG(status,out).print("entering resetFilters\n");
|
|
|
|
reset_filters(out);
|
|
|
|
}
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
static bool isPlannableBuilding(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom) {
|
|
|
|
DEBUG(status,out).print("entering isPlannableBuilding\n");
|
2023-02-21 19:05:15 -07:00
|
|
|
return get_num_filters(out, BuildingTypeKey(type, subtype, custom)) >= 1;
|
2023-01-17 05:05:17 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
static bool isPlannedBuilding(color_ostream &out, df::building *bld) {
|
|
|
|
TRACE(status,out).print("entering isPlannedBuilding\n");
|
2023-02-21 19:05:15 -07:00
|
|
|
return bld && planned_buildings.count(bld->id);
|
2023-02-19 01:57:30 -07:00
|
|
|
}
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
static bool addPlannedBuilding(color_ostream &out, df::building *bld) {
|
|
|
|
DEBUG(status,out).print("entering addPlannedBuilding\n");
|
2023-03-24 16:26:17 -06:00
|
|
|
if (!bld || planned_buildings.count(bld->id))
|
2023-01-17 05:05:17 -07:00
|
|
|
return false;
|
2023-03-24 16:26:17 -06:00
|
|
|
|
|
|
|
int16_t subtype = get_subtype(bld);
|
|
|
|
|
|
|
|
if (!isPlannableBuilding(out, bld->getType(), subtype, bld->getCustomType()))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
BuildingTypeKey key(bld->getType(), subtype, bld->getCustomType());
|
2023-03-15 01:29:27 -06:00
|
|
|
PlannedBuilding pb(out, bld, get_heat_safety_filter(key), get_item_filters(out, key));
|
2023-03-27 03:24:56 -06:00
|
|
|
|
|
|
|
bool unsuspend_on_finalize = !is_suspendmanager_enabled(out);
|
|
|
|
return registerPlannedBuilding(out, pb, unsuspend_on_finalize);
|
2023-01-17 05:05:17 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
static void doCycle(color_ostream &out) {
|
|
|
|
DEBUG(status,out).print("entering doCycle\n");
|
|
|
|
do_cycle(out);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void scheduleCycle(color_ostream &out) {
|
|
|
|
DEBUG(status,out).print("entering scheduleCycle\n");
|
|
|
|
cycle_requested = true;
|
|
|
|
}
|
|
|
|
|
2023-02-18 02:09:54 -07:00
|
|
|
static int scanAvailableItems(color_ostream &out, df::building_type type, int16_t subtype,
|
2023-11-03 17:33:08 -06:00
|
|
|
int32_t custom, int index, bool ignore_filters, bool ignore_quality, HeatSafety *heat_override = NULL,
|
|
|
|
vector<int> *item_ids = NULL, map<MaterialInfo, int32_t> *counts = NULL)
|
|
|
|
{
|
2023-02-15 17:54:38 -07:00
|
|
|
DEBUG(status,out).print(
|
2023-10-16 00:20:31 -06:00
|
|
|
"entering scanAvailableItems building_type=%d subtype=%d custom=%d index=%d\n",
|
2023-02-15 17:54:38 -07:00
|
|
|
type, subtype, custom, index);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
2023-11-03 17:33:08 -06:00
|
|
|
HeatSafety heat = heat_override ? *heat_override : get_heat_safety_filter(key);
|
2023-02-21 19:05:15 -07:00
|
|
|
auto &job_items = get_job_items(out, key);
|
2023-02-22 09:34:55 -07:00
|
|
|
if (index < 0 || job_items.size() <= (size_t)index)
|
2023-02-21 19:05:15 -07:00
|
|
|
return 0;
|
2023-03-15 01:29:27 -06:00
|
|
|
auto &item_filters = get_item_filters(out, key);
|
|
|
|
auto &filters = item_filters.getItemFilters();
|
|
|
|
auto &specials = item_filters.getSpecials();
|
2023-02-15 17:54:38 -07:00
|
|
|
|
|
|
|
auto &jitem = job_items[index];
|
2023-03-24 15:36:06 -06:00
|
|
|
auto vector_ids = getVectorIds(out, jitem, ignore_filters);
|
2023-02-15 17:54:38 -07:00
|
|
|
|
|
|
|
int count = 0;
|
|
|
|
for (auto vector_id : vector_ids) {
|
|
|
|
auto other_id = ENUM_ATTR(job_item_vector_id, other, vector_id);
|
|
|
|
for (auto &item : df::global::world->items.other[other_id]) {
|
2023-03-15 01:29:27 -06:00
|
|
|
ItemFilter filter = filters[index];
|
2023-03-31 05:45:59 -06:00
|
|
|
set<string> special = specials;
|
|
|
|
if (ignore_filters || counts) {
|
2023-03-02 18:50:12 -07:00
|
|
|
// don't filter by material; we want counts for all materials
|
|
|
|
filter.setMaterialMask(0);
|
|
|
|
filter.setMaterials(set<MaterialInfo>());
|
2023-03-31 05:45:59 -06:00
|
|
|
special.clear();
|
2023-03-02 18:50:12 -07:00
|
|
|
}
|
2023-11-03 17:33:08 -06:00
|
|
|
if (ignore_quality) {
|
|
|
|
filter.setMinQuality(df::item_quality::Ordinary);
|
|
|
|
filter.setMaxQuality(df::item_quality::Artifact);
|
|
|
|
}
|
2023-04-26 22:01:21 -06:00
|
|
|
if (itemPassesScreen(out, item) && matchesFilters(item, jitem, heat, filter, special)) {
|
2023-02-18 02:09:54 -07:00
|
|
|
if (item_ids)
|
|
|
|
item_ids->emplace_back(item->id);
|
2023-03-02 18:50:12 -07:00
|
|
|
if (counts) {
|
|
|
|
MaterialInfo mi;
|
|
|
|
mi.decode(item);
|
|
|
|
(*counts)[mi]++;
|
|
|
|
}
|
2023-02-15 17:54:38 -07:00
|
|
|
++count;
|
2023-02-18 02:09:54 -07:00
|
|
|
}
|
2023-02-15 17:54:38 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
DEBUG(status,out).print("found matches %d\n", count);
|
|
|
|
return count;
|
2023-02-13 17:24:10 -07:00
|
|
|
}
|
|
|
|
|
2023-02-18 02:09:54 -07:00
|
|
|
static int getAvailableItems(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
int index = luaL_checkint(L, 4);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getAvailableItems building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
|
|
|
vector<int> item_ids;
|
2023-11-03 17:33:08 -06:00
|
|
|
scanAvailableItems(*out, type, subtype, custom, index, true, false, NULL, &item_ids);
|
|
|
|
Lua::PushVector(L, item_ids);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int getAvailableItemsByHeat(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
int index = luaL_checkint(L, 4);
|
|
|
|
HeatSafety heat = (HeatSafety)luaL_checkint(L, 5);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getAvailableItemsByHeat building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
|
|
|
vector<int> item_ids;
|
|
|
|
scanAvailableItems(*out, type, subtype, custom, index, true, true, &heat, &item_ids);
|
2023-02-18 02:09:54 -07:00
|
|
|
Lua::PushVector(L, item_ids);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2023-02-27 11:16:58 -07:00
|
|
|
static int getGlobalSettings(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
DEBUG(status,*out).print("entering getGlobalSettings\n");
|
|
|
|
map<string, bool> settings;
|
|
|
|
settings.emplace("blocks", get_config_bool(config, CONFIG_BLOCKS));
|
|
|
|
settings.emplace("logs", get_config_bool(config, CONFIG_LOGS));
|
|
|
|
settings.emplace("boulders", get_config_bool(config, CONFIG_BOULDERS));
|
|
|
|
settings.emplace("bars", get_config_bool(config, CONFIG_BARS));
|
|
|
|
Lua::Push(L, settings);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2023-02-18 02:09:54 -07:00
|
|
|
static int countAvailableItems(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom, int index) {
|
|
|
|
DEBUG(status,out).print(
|
|
|
|
"entering countAvailableItems building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
2023-11-03 17:33:08 -06:00
|
|
|
int count = scanAvailableItems(out, type, subtype, custom, index, false, false);
|
2023-10-16 00:20:31 -06:00
|
|
|
if (count)
|
|
|
|
return count;
|
|
|
|
|
|
|
|
// nothing in stock; return how many are waiting in line as a negative
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &job_items = get_job_items(out, key);
|
|
|
|
if (index < 0 || job_items.size() <= (size_t)index)
|
|
|
|
return 0;
|
|
|
|
auto &jitem = job_items[index];
|
|
|
|
|
|
|
|
for (auto &entry : planned_buildings) {
|
|
|
|
auto &pb = entry.second;
|
|
|
|
// don't actually remove bad buildings from the list while we're
|
|
|
|
// actively iterating through that list
|
|
|
|
auto bld = pb.getBuildingIfValidOrRemoveIfNot(out, true);
|
|
|
|
if (!bld || bld->jobs.size() != 1)
|
|
|
|
continue;
|
|
|
|
for (auto pb_jitem : bld->jobs[0]->job_items) {
|
|
|
|
if (pb_jitem->item_type == jitem->item_type && pb_jitem->item_subtype == jitem->item_subtype)
|
|
|
|
count -= pb_jitem->quantity;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return count;
|
2023-02-18 02:09:54 -07:00
|
|
|
}
|
|
|
|
|
2023-02-21 14:04:53 -07:00
|
|
|
static bool hasFilter(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom, int index) {
|
2023-02-22 16:08:11 -07:00
|
|
|
TRACE(status,out).print("entering hasFilter\n");
|
2023-04-07 03:07:45 -06:00
|
|
|
if (!Core::getInstance().isWorldLoaded())
|
|
|
|
return false;
|
2023-02-22 16:08:11 -07:00
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(out, key);
|
2023-03-13 14:35:12 -06:00
|
|
|
if (index < 0 || filters.getItemFilters().size() <= (size_t)index)
|
|
|
|
return false;
|
|
|
|
return !filters.getItemFilters()[index].isEmpty();
|
2023-02-17 20:16:45 -07:00
|
|
|
}
|
|
|
|
|
2023-02-27 13:27:21 -07:00
|
|
|
static void clearFilter(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom, int index) {
|
|
|
|
TRACE(status,out).print("entering clearFilter\n");
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(out, key);
|
2023-03-02 05:38:40 -07:00
|
|
|
if (index < 0 || filters.getItemFilters().size() <= (size_t)index)
|
2023-02-27 13:27:21 -07:00
|
|
|
return;
|
2023-03-06 00:04:03 -07:00
|
|
|
ItemFilter filter = filters.getItemFilters()[index];
|
|
|
|
filter.clear();
|
|
|
|
filters.setItemFilter(out, filter, index);
|
2023-02-27 13:27:21 -07:00
|
|
|
call_buildingplan_lua(&out, "signal_reset");
|
|
|
|
}
|
|
|
|
|
2023-03-02 18:50:12 -07:00
|
|
|
static int setMaterialMaskFilter(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
int index = luaL_checkint(L, 4);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering setMaterialMaskFilter building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(*out, key).getItemFilters();
|
|
|
|
if (index < 0 || filters.size() <= (size_t)index)
|
|
|
|
return 0;
|
|
|
|
uint32_t mask = 0;
|
|
|
|
vector<string> cats;
|
2023-03-02 20:32:25 -07:00
|
|
|
Lua::GetVector(L, cats, 5);
|
2023-03-02 18:50:12 -07:00
|
|
|
for (auto &cat : cats) {
|
|
|
|
if (cat == "stone")
|
2023-03-05 19:17:10 -07:00
|
|
|
mask |= stone_cat.whole;
|
2023-03-02 18:50:12 -07:00
|
|
|
else if (cat == "wood")
|
2023-03-05 19:17:10 -07:00
|
|
|
mask |= wood_cat.whole;
|
2023-03-02 18:50:12 -07:00
|
|
|
else if (cat == "metal")
|
2023-03-05 19:17:10 -07:00
|
|
|
mask |= metal_cat.whole;
|
|
|
|
else if (cat == "glass")
|
|
|
|
mask |= glass_cat.whole;
|
2023-04-17 02:05:25 -06:00
|
|
|
else if (cat == "gem")
|
|
|
|
mask |= gem_cat.whole;
|
2023-03-12 16:13:57 -06:00
|
|
|
else if (cat == "clay")
|
|
|
|
mask |= clay_cat.whole;
|
2023-04-11 04:49:28 -06:00
|
|
|
else if (cat == "cloth")
|
|
|
|
mask |= cloth_cat.whole;
|
|
|
|
else if (cat == "silk")
|
|
|
|
mask |= silk_cat.whole;
|
2023-04-21 12:54:38 -06:00
|
|
|
else if (cat == "yarn")
|
|
|
|
mask |= yarn_cat.whole;
|
2023-03-02 18:50:12 -07:00
|
|
|
}
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"setting material mask filter for building_type=%d subtype=%d custom=%d index=%d to %x\n",
|
|
|
|
type, subtype, custom, index, mask);
|
|
|
|
ItemFilter filter = filters[index];
|
|
|
|
filter.setMaterialMask(mask);
|
2023-03-05 20:15:04 -07:00
|
|
|
set<MaterialInfo> new_mats;
|
2023-03-05 19:17:10 -07:00
|
|
|
if (mask) {
|
|
|
|
// remove materials from the list that don't match the mask
|
|
|
|
const auto &mats = filter.getMaterials();
|
|
|
|
const df::dfhack_material_category mat_mask(mask);
|
|
|
|
for (auto & mat : mats) {
|
|
|
|
if (mat.matches(mat_mask))
|
|
|
|
new_mats.emplace(mat);
|
|
|
|
}
|
|
|
|
}
|
2023-03-05 20:15:04 -07:00
|
|
|
filter.setMaterials(new_mats);
|
2023-03-02 18:50:12 -07:00
|
|
|
get_item_filters(*out, key).setItemFilter(*out, filter, index);
|
|
|
|
call_buildingplan_lua(out, "signal_reset");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int getMaterialMaskFilter(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
int index = luaL_checkint(L, 4);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getMaterialFilter building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(*out, key);
|
|
|
|
if (index < 0 || filters.getItemFilters().size() <= (size_t)index)
|
|
|
|
return 0;
|
2023-03-05 19:17:10 -07:00
|
|
|
map<string, bool> ret;
|
2023-03-02 18:50:12 -07:00
|
|
|
uint32_t bits = filters.getItemFilters()[index].getMaterialMask().whole;
|
2023-03-05 19:17:10 -07:00
|
|
|
ret.emplace("unset", !bits);
|
|
|
|
ret.emplace("stone", !bits || bits & stone_cat.whole);
|
|
|
|
ret.emplace("wood", !bits || bits & wood_cat.whole);
|
|
|
|
ret.emplace("metal", !bits || bits & metal_cat.whole);
|
|
|
|
ret.emplace("glass", !bits || bits & glass_cat.whole);
|
2023-04-17 02:05:25 -06:00
|
|
|
ret.emplace("gem", !bits || bits & gem_cat.whole);
|
2023-03-12 16:13:57 -06:00
|
|
|
ret.emplace("clay", !bits || bits & clay_cat.whole);
|
2023-04-11 04:49:28 -06:00
|
|
|
ret.emplace("cloth", !bits || bits & cloth_cat.whole);
|
|
|
|
ret.emplace("silk", !bits || bits & silk_cat.whole);
|
2023-04-21 12:54:38 -06:00
|
|
|
ret.emplace("yarn", !bits || bits & yarn_cat.whole);
|
2023-03-05 19:17:10 -07:00
|
|
|
Lua::Push(L, ret);
|
2023-03-02 18:50:12 -07:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int setMaterialFilter(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
int index = luaL_checkint(L, 4);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering setMaterialFilter building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(*out, key).getItemFilters();
|
|
|
|
if (index < 0 || filters.size() <= (size_t)index)
|
|
|
|
return 0;
|
|
|
|
set<MaterialInfo> mats;
|
|
|
|
vector<string> matstrs;
|
2023-03-05 19:17:10 -07:00
|
|
|
Lua::GetVector(L, matstrs, 5);
|
2023-03-02 18:50:12 -07:00
|
|
|
for (auto &mat : matstrs) {
|
|
|
|
if (mat_cache.count(mat))
|
|
|
|
mats.emplace(mat_cache.at(mat).first);
|
|
|
|
}
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"setting material filter for building_type=%d subtype=%d custom=%d index=%d to %zd materials\n",
|
|
|
|
type, subtype, custom, index, mats.size());
|
|
|
|
ItemFilter filter = filters[index];
|
|
|
|
filter.setMaterials(mats);
|
2023-03-05 19:17:10 -07:00
|
|
|
// ensure relevant masks are explicitly enabled
|
|
|
|
df::dfhack_material_category mask = filter.getMaterialMask();
|
2023-03-05 20:15:04 -07:00
|
|
|
if (!mats.size())
|
|
|
|
mask.whole = 0; // if all materials are disabled, reset the mask
|
2023-03-05 19:17:10 -07:00
|
|
|
for (auto & mat : mats) {
|
|
|
|
if (mat.matches(stone_cat))
|
|
|
|
mask.whole |= stone_cat.whole;
|
|
|
|
else if (mat.matches(wood_cat))
|
|
|
|
mask.whole |= wood_cat.whole;
|
|
|
|
else if (mat.matches(metal_cat))
|
|
|
|
mask.whole |= metal_cat.whole;
|
|
|
|
else if (mat.matches(glass_cat))
|
|
|
|
mask.whole |= glass_cat.whole;
|
2023-04-17 02:05:25 -06:00
|
|
|
else if (mat.matches(gem_cat))
|
|
|
|
mask.whole |= gem_cat.whole;
|
2023-03-12 16:13:57 -06:00
|
|
|
else if (mat.matches(clay_cat))
|
|
|
|
mask.whole |= clay_cat.whole;
|
2023-04-11 04:49:28 -06:00
|
|
|
else if (mat.matches(cloth_cat))
|
|
|
|
mask.whole |= cloth_cat.whole;
|
|
|
|
else if (mat.matches(silk_cat))
|
|
|
|
mask.whole |= silk_cat.whole;
|
2023-04-21 12:54:38 -06:00
|
|
|
else if (mat.matches(yarn_cat))
|
|
|
|
mask.whole |= yarn_cat.whole;
|
2023-03-05 19:17:10 -07:00
|
|
|
}
|
|
|
|
filter.setMaterialMask(mask.whole);
|
2023-03-02 18:50:12 -07:00
|
|
|
get_item_filters(*out, key).setItemFilter(*out, filter, index);
|
|
|
|
call_buildingplan_lua(out, "signal_reset");
|
|
|
|
return 0;
|
2023-02-19 01:57:30 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
static int getMaterialFilter(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
int index = luaL_checkint(L, 4);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getMaterialFilter building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
2023-03-02 18:50:12 -07:00
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(*out, key).getItemFilters();
|
|
|
|
if (index < 0 || filters.size() <= (size_t)index)
|
|
|
|
return 0;
|
|
|
|
const auto &mat_filter = filters[index].getMaterials();
|
|
|
|
map<MaterialInfo, int32_t> counts;
|
2023-11-03 17:33:08 -06:00
|
|
|
scanAvailableItems(*out, type, subtype, custom, index, false, false, NULL, NULL, &counts);
|
2023-03-10 03:34:05 -07:00
|
|
|
HeatSafety heat = get_heat_safety_filter(key);
|
2023-09-03 12:49:13 -06:00
|
|
|
const df::job_item *jitem = get_job_items(*out, key)[index];
|
2023-03-10 03:34:05 -07:00
|
|
|
// name -> {count=int, enabled=bool, category=string, heat=string}
|
2023-03-02 18:50:12 -07:00
|
|
|
map<string, map<string, string>> ret;
|
|
|
|
for (auto & entry : mat_cache) {
|
|
|
|
auto &mat = entry.second.first;
|
2023-09-03 12:49:13 -06:00
|
|
|
if (!mat.matches(jitem))
|
|
|
|
continue;
|
|
|
|
if (!matchesHeatSafety(mat.type, mat.index, heat))
|
2023-03-10 03:34:05 -07:00
|
|
|
continue;
|
|
|
|
string heat_safety = "";
|
2023-09-03 12:49:13 -06:00
|
|
|
if (matchesHeatSafety(mat.type, mat.index, HEAT_SAFETY_MAGMA))
|
2023-03-10 03:34:05 -07:00
|
|
|
heat_safety = "magma-safe";
|
2023-09-03 12:49:13 -06:00
|
|
|
else if (matchesHeatSafety(mat.type, mat.index, HEAT_SAFETY_FIRE))
|
2023-03-10 03:34:05 -07:00
|
|
|
heat_safety = "fire-safe";
|
|
|
|
auto &name = entry.first;
|
2023-03-02 18:50:12 -07:00
|
|
|
auto &cat = entry.second.second;
|
|
|
|
map<string, string> props;
|
|
|
|
string count = "0";
|
|
|
|
if (counts.count(mat))
|
|
|
|
count = int_to_string(counts.at(mat));
|
|
|
|
props.emplace("count", count);
|
|
|
|
props.emplace("enabled", (!mat_filter.size() || mat_filter.count(mat)) ? "true" : "false");
|
|
|
|
props.emplace("category", cat);
|
|
|
|
ret.emplace(name, props);
|
|
|
|
}
|
|
|
|
Lua::Push(L, ret);
|
2023-02-19 01:57:30 -07:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2023-03-29 00:51:52 -06:00
|
|
|
static void setChooseItems(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom, int choose) {
|
|
|
|
DEBUG(status,out).print(
|
|
|
|
"entering setChooseItems building_type=%d subtype=%d custom=%d choose=%d\n",
|
|
|
|
type, subtype, custom, choose);
|
2023-03-14 22:55:43 -06:00
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
2023-03-14 23:26:56 -06:00
|
|
|
auto &filters = get_item_filters(out, key);
|
|
|
|
filters.setChooseItems(choose);
|
2023-03-14 22:55:43 -06:00
|
|
|
// no need to reset signal; no change to the state of any other UI element
|
|
|
|
}
|
|
|
|
|
|
|
|
static int getChooseItems(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getChooseItems building_type=%d subtype=%d custom=%d\n",
|
|
|
|
type, subtype, custom);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
2023-03-14 23:26:56 -06:00
|
|
|
Lua::Push(L, get_item_filters(*out, key).getChooseItems());
|
2023-03-14 22:55:43 -06:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2023-02-19 01:57:30 -07:00
|
|
|
static void setHeatSafetyFilter(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom, int heat) {
|
|
|
|
DEBUG(status,out).print("entering setHeatSafetyFilter\n");
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
if (heat == HEAT_SAFETY_FIRE || heat == HEAT_SAFETY_MAGMA)
|
|
|
|
cur_heat_safety[key] = (HeatSafety)heat;
|
|
|
|
else
|
|
|
|
cur_heat_safety.erase(key);
|
|
|
|
call_buildingplan_lua(&out, "signal_reset");
|
2023-02-17 20:16:45 -07:00
|
|
|
}
|
|
|
|
|
2023-02-19 01:57:30 -07:00
|
|
|
static int getHeatSafetyFilter(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getHeatSafetyFilter building_type=%d subtype=%d custom=%d\n",
|
|
|
|
type, subtype, custom);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
HeatSafety heat = get_heat_safety_filter(key);
|
|
|
|
Lua::Push(L, heat);
|
|
|
|
return 1;
|
2023-02-17 20:16:45 -07:00
|
|
|
}
|
|
|
|
|
2023-03-15 01:29:27 -06:00
|
|
|
static void setSpecial(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom, string special, bool val) {
|
|
|
|
DEBUG(status,out).print("entering setSpecial\n");
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(out, key);
|
|
|
|
filters.setSpecial(special, val);
|
|
|
|
call_buildingplan_lua(&out, "signal_reset");
|
|
|
|
}
|
|
|
|
|
|
|
|
static int getSpecials(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getSpecials building_type=%d subtype=%d custom=%d\n",
|
|
|
|
type, subtype, custom);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
Lua::Push(L, get_item_filters(*out, key).getSpecials());
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2023-02-27 13:27:21 -07:00
|
|
|
static void setQualityFilter(color_ostream &out, df::building_type type, int16_t subtype, int32_t custom, int index,
|
|
|
|
int decorated, int min_quality, int max_quality) {
|
|
|
|
DEBUG(status,out).print("entering setQualityFilter\n");
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(out, key).getItemFilters();
|
2023-03-02 05:38:40 -07:00
|
|
|
if (index < 0 || filters.size() <= (size_t)index)
|
2023-02-27 13:27:21 -07:00
|
|
|
return;
|
|
|
|
ItemFilter filter = filters[index];
|
|
|
|
filter.setDecoratedOnly(decorated != 0);
|
|
|
|
filter.setMinQuality(min_quality);
|
|
|
|
filter.setMaxQuality(max_quality);
|
|
|
|
get_item_filters(out, key).setItemFilter(out, filter, index);
|
|
|
|
call_buildingplan_lua(&out, "signal_reset");
|
|
|
|
}
|
|
|
|
|
|
|
|
static int getQualityFilter(lua_State *L) {
|
|
|
|
color_ostream *out = Lua::GetOutput(L);
|
|
|
|
if (!out)
|
|
|
|
out = &Core::getInstance().getConsole();
|
|
|
|
df::building_type type = (df::building_type)luaL_checkint(L, 1);
|
|
|
|
int16_t subtype = luaL_checkint(L, 2);
|
|
|
|
int32_t custom = luaL_checkint(L, 3);
|
|
|
|
int index = luaL_checkint(L, 4);
|
|
|
|
DEBUG(status,*out).print(
|
|
|
|
"entering getQualityFilter building_type=%d subtype=%d custom=%d index=%d\n",
|
|
|
|
type, subtype, custom, index);
|
|
|
|
BuildingTypeKey key(type, subtype, custom);
|
|
|
|
auto &filters = get_item_filters(*out, key).getItemFilters();
|
2023-03-02 06:29:33 -07:00
|
|
|
if (index < 0 || filters.size() <= (size_t)index)
|
2023-02-27 13:27:21 -07:00
|
|
|
return 0;
|
|
|
|
auto &filter = filters[index];
|
|
|
|
map<string, int> ret;
|
|
|
|
ret.emplace("decorated", filter.getDecoratedOnly());
|
|
|
|
ret.emplace("min_quality", filter.getMinQuality());
|
|
|
|
ret.emplace("max_quality", filter.getMaxQuality());
|
|
|
|
Lua::Push(L, ret);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2023-02-16 04:08:55 -07:00
|
|
|
static bool validate_pb(color_ostream &out, df::building *bld, int index) {
|
2023-02-15 20:10:42 -07:00
|
|
|
if (!isPlannedBuilding(out, bld) || bld->jobs.size() != 1)
|
2023-02-16 04:08:55 -07:00
|
|
|
return false;
|
2023-02-15 20:10:42 -07:00
|
|
|
|
|
|
|
auto &job_items = bld->jobs[0]->job_items;
|
2023-02-16 22:42:58 -07:00
|
|
|
if ((int)job_items.size() <= index)
|
2023-02-16 04:08:55 -07:00
|
|
|
return false;
|
2023-02-15 20:10:42 -07:00
|
|
|
|
|
|
|
PlannedBuilding &pb = planned_buildings.at(bld->id);
|
2023-02-16 22:42:58 -07:00
|
|
|
if ((int)pb.vector_ids.size() <= index)
|
2023-02-16 04:08:55 -07:00
|
|
|
return false;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
static string getDescString(color_ostream &out, df::building *bld, int index) {
|
|
|
|
DEBUG(status,out).print("entering getDescString\n");
|
|
|
|
if (!validate_pb(out, bld, index))
|
2023-03-13 14:35:12 -06:00
|
|
|
return "INVALID";
|
2023-02-15 20:10:42 -07:00
|
|
|
|
2023-02-16 04:08:55 -07:00
|
|
|
PlannedBuilding &pb = planned_buildings.at(bld->id);
|
2023-03-13 14:35:12 -06:00
|
|
|
auto & jitems = bld->jobs[0]->job_items;
|
|
|
|
const size_t num_job_items = jitems.size();
|
|
|
|
int rev_index = num_job_items - (index + 1);
|
|
|
|
auto &jitem = jitems[rev_index];
|
2023-03-10 02:04:23 -07:00
|
|
|
return int_to_string(jitem->quantity) + " " + get_desc_string(out, jitem, pb.vector_ids[index]);
|
2023-02-16 04:08:55 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
static int getQueuePosition(color_ostream &out, df::building *bld, int index) {
|
2023-03-11 16:36:47 -07:00
|
|
|
TRACE(status,out).print("entering getQueuePosition\n");
|
2023-02-16 04:08:55 -07:00
|
|
|
if (!validate_pb(out, bld, index))
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
PlannedBuilding &pb = planned_buildings.at(bld->id);
|
2023-03-13 14:35:12 -06:00
|
|
|
auto & jitems = bld->jobs[0]->job_items;
|
|
|
|
const size_t num_job_items = jitems.size();
|
|
|
|
int rev_index = num_job_items - (index + 1);
|
|
|
|
auto &job_item = jitems[rev_index];
|
2023-02-16 04:08:55 -07:00
|
|
|
|
|
|
|
if (job_item->quantity <= 0)
|
|
|
|
return 0;
|
2023-02-15 20:10:42 -07:00
|
|
|
|
|
|
|
int min_pos = -1;
|
|
|
|
for (auto &vec_id : pb.vector_ids[index]) {
|
|
|
|
if (!tasks.count(vec_id))
|
|
|
|
continue;
|
|
|
|
auto &buckets = tasks.at(vec_id);
|
2023-02-27 00:38:00 -07:00
|
|
|
string bucket_id = getBucket(*job_item, pb, index);
|
2023-02-15 20:10:42 -07:00
|
|
|
if (!buckets.count(bucket_id))
|
|
|
|
continue;
|
|
|
|
int bucket_pos = -1;
|
|
|
|
for (auto &task : buckets.at(bucket_id)) {
|
|
|
|
++bucket_pos;
|
2023-03-13 14:35:12 -06:00
|
|
|
if (bld->id == task.first && rev_index == task.second)
|
2023-02-15 20:10:42 -07:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (bucket_pos++ >= 0)
|
|
|
|
min_pos = min_pos < 0 ? bucket_pos : std::min(min_pos, bucket_pos);
|
|
|
|
}
|
|
|
|
|
|
|
|
return min_pos < 0 ? 0 : min_pos;
|
|
|
|
}
|
|
|
|
|
2023-02-16 19:02:15 -07:00
|
|
|
static void makeTopPriority(color_ostream &out, df::building *bld) {
|
|
|
|
DEBUG(status,out).print("entering makeTopPriority\n");
|
|
|
|
if (!validate_pb(out, bld, 0))
|
|
|
|
return;
|
|
|
|
|
|
|
|
PlannedBuilding &pb = planned_buildings.at(bld->id);
|
|
|
|
auto &job_items = bld->jobs[0]->job_items;
|
2023-03-13 14:35:12 -06:00
|
|
|
const int num_job_items = (int)job_items.size();
|
2023-02-16 19:02:15 -07:00
|
|
|
|
2023-03-13 14:35:12 -06:00
|
|
|
for (int index = 0; index < num_job_items; ++index) {
|
|
|
|
int rev_index = num_job_items - (index + 1);
|
2023-02-16 19:02:15 -07:00
|
|
|
for (auto &vec_id : pb.vector_ids[index]) {
|
|
|
|
if (!tasks.count(vec_id))
|
|
|
|
continue;
|
|
|
|
auto &buckets = tasks.at(vec_id);
|
2023-03-13 14:35:12 -06:00
|
|
|
string bucket_id = getBucket(*job_items[rev_index], pb, index);
|
2023-02-16 19:02:15 -07:00
|
|
|
if (!buckets.count(bucket_id))
|
|
|
|
continue;
|
|
|
|
auto &bucket = buckets.at(bucket_id);
|
|
|
|
for (auto taskit = bucket.begin(); taskit != bucket.end(); ++taskit) {
|
2023-03-13 14:35:12 -06:00
|
|
|
if (bld->id == taskit->first && rev_index == taskit->second) {
|
2023-02-16 19:02:15 -07:00
|
|
|
auto task_bld_id = taskit->first;
|
|
|
|
auto task_job_item_idx = taskit->second;
|
|
|
|
bucket.erase(taskit);
|
|
|
|
bucket.emplace_front(task_bld_id, task_job_item_idx);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-01-17 05:05:17 -07:00
|
|
|
DFHACK_PLUGIN_LUA_FUNCTIONS {
|
|
|
|
DFHACK_LUA_FUNCTION(printStatus),
|
|
|
|
DFHACK_LUA_FUNCTION(setSetting),
|
2023-03-15 15:02:59 -06:00
|
|
|
DFHACK_LUA_FUNCTION(resetFilters),
|
2023-01-17 05:05:17 -07:00
|
|
|
DFHACK_LUA_FUNCTION(isPlannableBuilding),
|
|
|
|
DFHACK_LUA_FUNCTION(isPlannedBuilding),
|
|
|
|
DFHACK_LUA_FUNCTION(addPlannedBuilding),
|
|
|
|
DFHACK_LUA_FUNCTION(doCycle),
|
2023-03-02 05:38:40 -07:00
|
|
|
DFHACK_LUA_FUNCTION(scheduleCycle),
|
|
|
|
DFHACK_LUA_FUNCTION(countAvailableItems),
|
2023-02-21 14:04:53 -07:00
|
|
|
DFHACK_LUA_FUNCTION(hasFilter),
|
2023-02-27 13:27:21 -07:00
|
|
|
DFHACK_LUA_FUNCTION(clearFilter),
|
2023-03-14 22:55:43 -06:00
|
|
|
DFHACK_LUA_FUNCTION(setChooseItems),
|
2023-02-19 01:57:30 -07:00
|
|
|
DFHACK_LUA_FUNCTION(setHeatSafetyFilter),
|
2023-03-15 01:29:27 -06:00
|
|
|
DFHACK_LUA_FUNCTION(setSpecial),
|
2023-02-27 13:27:21 -07:00
|
|
|
DFHACK_LUA_FUNCTION(setQualityFilter),
|
2023-02-16 04:08:55 -07:00
|
|
|
DFHACK_LUA_FUNCTION(getDescString),
|
2023-02-15 20:10:42 -07:00
|
|
|
DFHACK_LUA_FUNCTION(getQueuePosition),
|
2023-02-16 19:02:15 -07:00
|
|
|
DFHACK_LUA_FUNCTION(makeTopPriority),
|
2023-01-17 05:05:17 -07:00
|
|
|
DFHACK_LUA_END
|
|
|
|
};
|
2023-02-18 02:09:54 -07:00
|
|
|
|
|
|
|
DFHACK_PLUGIN_LUA_COMMANDS {
|
2023-02-27 11:16:58 -07:00
|
|
|
DFHACK_LUA_COMMAND(getGlobalSettings),
|
2023-02-18 02:09:54 -07:00
|
|
|
DFHACK_LUA_COMMAND(getAvailableItems),
|
2023-11-03 17:33:08 -06:00
|
|
|
DFHACK_LUA_COMMAND(getAvailableItemsByHeat),
|
2023-03-02 18:50:12 -07:00
|
|
|
DFHACK_LUA_COMMAND(setMaterialMaskFilter),
|
|
|
|
DFHACK_LUA_COMMAND(getMaterialMaskFilter),
|
|
|
|
DFHACK_LUA_COMMAND(setMaterialFilter),
|
2023-02-19 01:57:30 -07:00
|
|
|
DFHACK_LUA_COMMAND(getMaterialFilter),
|
2023-03-14 22:55:43 -06:00
|
|
|
DFHACK_LUA_COMMAND(getChooseItems),
|
2023-02-19 01:57:30 -07:00
|
|
|
DFHACK_LUA_COMMAND(getHeatSafetyFilter),
|
2023-03-15 01:29:27 -06:00
|
|
|
DFHACK_LUA_COMMAND(getSpecials),
|
2023-02-27 13:27:21 -07:00
|
|
|
DFHACK_LUA_COMMAND(getQualityFilter),
|
2023-02-18 02:09:54 -07:00
|
|
|
DFHACK_LUA_END
|
|
|
|
};
|