Allow specifying constraints on item amounts, and match jobs to them.

develop
Alexander Gavrilov 2012-01-09 19:20:28 +04:00
parent 3e147fe902
commit ba338be350
1 changed files with 297 additions and 22 deletions

@ -5,6 +5,7 @@
#include <MiscUtils.h> #include <MiscUtils.h>
#include <modules/Materials.h> #include <modules/Materials.h>
#include <modules/Items.h>
#include <modules/Gui.h> #include <modules/Gui.h>
#include <modules/Job.h> #include <modules/Job.h>
#include <modules/World.h> #include <modules/World.h>
@ -60,8 +61,14 @@ DFhackCExport command_result plugin_init (Core *c, std::vector <PluginCommand> &
" workflow enable\n" " workflow enable\n"
" workflow disable\n" " workflow disable\n"
" Enable or disable the plugin.\n" " Enable or disable the plugin.\n"
" workflow list-jobs\n" " workflow jobs\n"
" List workflow-controlled jobs (if in a workshop, filtered by it).\n" " List workflow-controlled jobs (if in a workshop, filtered by it).\n"
" workflow list\n"
" List active constraints, and their job counts.\n"
" workflow limit <constraint-spec> <cnt-limit> [cnt-gap]\n"
" Set a constraint.\n"
" workflow unlimit <constraint-spec>\n"
" Delete a constraint.\n"
"Function:\n" "Function:\n"
" When the plugin is enabled, it protects all repeat jobs from removal.\n" " When the plugin is enabled, it protects all repeat jobs from removal.\n"
" If they do disappear due to any cause, they are immediately re-added\n" " If they do disappear due to any cause, they are immediately re-added\n"
@ -101,6 +108,8 @@ DFhackCExport command_result plugin_onstatechange(Core* c, state_change_event ev
/*******************************/ /*******************************/
struct ItemConstraint;
struct ProtectedJob { struct ProtectedJob {
int id; int id;
int building_id; int building_id;
@ -110,12 +119,17 @@ struct ProtectedJob {
df::building *holder; df::building *holder;
df::job *job_copy; df::job *job_copy;
df::job *actual_job;
std::vector<ItemConstraint*> constraints;
ProtectedJob(df::job *job) : id(job->id), live(true) ProtectedJob(df::job *job) : id(job->id), live(true)
{ {
check_idx = 0; check_idx = 0;
holder = getJobHolder(job); holder = getJobHolder(job);
building_id = holder ? holder->id : -1; building_id = holder ? holder->id : -1;
job_copy = cloneJobStruct(job); job_copy = cloneJobStruct(job);
actual_job = job;
} }
~ProtectedJob() ~ProtectedJob()
@ -125,6 +139,7 @@ struct ProtectedJob {
void update(df::job *job) void update(df::job *job)
{ {
actual_job = job;
if (*job == *job_copy) if (*job == *job_copy)
return; return;
@ -133,6 +148,29 @@ struct ProtectedJob {
} }
}; };
struct ItemConstraint {
PersistentDataItem config;
ItemTypeInfo item;
MaterialInfo material;
df::job_material_category mat_mask;
int weight;
std::vector<ProtectedJob*> jobs;
std::map<std::pair<int,int>, bool> material_cache;
int goalCount() { return config.ival(0); }
void setGoalCount(int v) { config.ival(0) = v; }
int goalGap() {
int gcnt = std::max(1, goalCount()/2);
return std::min(gcnt, config.ival(1) <= 0 ? 5 : config.ival(1));
}
void setGoalGap(int v) { config.ival(1) = v; }
};
/*******************************/ /*******************************/
static bool enabled = false; static bool enabled = false;
@ -146,6 +184,7 @@ typedef std::map<int, ProtectedJob*> TKnownJobs;
static TKnownJobs known_jobs; static TKnownJobs known_jobs;
static std::vector<ProtectedJob*> pending_recover; static std::vector<ProtectedJob*> pending_recover;
static std::vector<ItemConstraint*> constraints;
/*******************************/ /*******************************/
@ -189,9 +228,14 @@ static void cleanup_state(Core *c)
config = PersistentDataItem(); config = PersistentDataItem();
stop_protect(c); stop_protect(c);
for (unsigned i = 0; i < constraints.size(); i++)
delete constraints[i];
constraints.clear();
} }
static bool check_lost_jobs(Core *c); static bool check_lost_jobs(Core *c);
static ItemConstraint *get_constraint(Core *c, const std::string &str, PersistentDataItem *cfg = NULL);
static void start_protect(Core *c) static void start_protect(Core *c)
{ {
@ -208,6 +252,18 @@ static void init_state(Core *c)
enabled = config.isValid() && config.ival(0) != -1 && enabled = config.isValid() && config.ival(0) != -1 &&
(config.ival(0) & CF_ENABLED); (config.ival(0) & CF_ENABLED);
// Parse constraints
std::vector<PersistentDataItem> items;
c->getWorld()->GetPersistentData(&items, "workflow/constraints");
for (int i = items.size()-1; i >= 0; i--) {
if (get_constraint(c, items[i].val(), &items[i]))
continue;
c->con.printerr("Lost constraint %s\n", items[i].val().c_str());
c->getWorld()->DeletePersistentData(items[i]);
}
if (!enabled) if (!enabled)
return; return;
@ -287,6 +343,7 @@ static bool recover_job(Core *c, ProtectedJob *pj)
pj->holder->jobs.push_back(recovered); pj->holder->jobs.push_back(recovered);
// Done // Done
pj->actual_job = recovered;
pj->live = true; pj->live = true;
return true; return true;
} }
@ -326,6 +383,7 @@ static bool check_lost_jobs(Core *c)
continue; continue;
it->second->live = false; it->second->live = false;
it->second->actual_job = NULL;
pending_recover.push_back(it->second); pending_recover.push_back(it->second);
any_lost = true; any_lost = true;
} }
@ -358,33 +416,207 @@ DFhackCExport command_result plugin_onupdate(Core* c)
return CR_OK; return CR_OK;
static unsigned cnt = 0; static unsigned cnt = 0;
static unsigned last_rlen = 0;
cnt++; cnt++;
bool force_recover = false; if ((cnt % 5) == 0) {
if ((cnt % 5) == 0) check_lost_jobs(c);
if (pending_recover.size() != last_rlen || (cnt % 50) == 0)
{
recover_jobs(c);
last_rlen = pending_recover.size();
if ((cnt % 500) == 0)
update_job_data(c);
}
}
return CR_OK;
}
/*******************************/
static ItemConstraint *get_constraint(Core *c, const std::string &str, PersistentDataItem *cfg)
{
std::vector<std::string> tokens;
split_string(&tokens, str, "/");
if (tokens.size() > 3)
return NULL;
int weight = 0;
ItemTypeInfo item;
if (!item.find(tokens[0]) || !item.isValid()) {
c->con.printerr("Cannot find item type: %s\n", tokens[0].c_str());
return NULL;
}
if (item.subtype >= 0)
weight += 10000;
MaterialInfo material;
std::string matstr = vector_get(tokens,1);
if (!matstr.empty() && (!material.find(matstr) || !material.isValid())) {
c->con.printerr("Cannot find material: %s\n", matstr.c_str());
return NULL;
}
if (material.type >= 0)
weight += (material.index >= 0 ? 5000 : 1000);
df::job_material_category mat_mask;
std::string maskstr = vector_get(tokens,2);
if (!maskstr.empty() && !parseJobMaterialCategory(&mat_mask, maskstr)) {
c->con.printerr("Cannot decode material mask: %s\n", maskstr.c_str());
return NULL;
}
if (mat_mask.whole && material.isValid() && !material.matches(mat_mask)) {
c->con.printerr("Material %s doesn't match mask %s\n", matstr.c_str(), maskstr.c_str());
return NULL;
}
if (mat_mask.whole != 0)
weight += 100;
for (unsigned i = 0; i < constraints.size(); i++)
{ {
force_recover = check_lost_jobs(c); ItemConstraint *ct = constraints[i];
if (ct->item == item && ct->material == material &&
ct->mat_mask.whole == mat_mask.whole)
return ct;
} }
if (force_recover || (cnt % 50) == 0) ItemConstraint *nct = new ItemConstraint;
nct->item = item;
nct->material = material;
nct->mat_mask = mat_mask;
nct->weight = weight;
if (cfg)
nct->config = *cfg;
else
{ {
recover_jobs(c); nct->config = c->getWorld()->AddPersistentData("workflow/constraints");
nct->config.val() = str;
} }
if ((cnt % 500) == 0) constraints.push_back(nct);
update_job_data(c); return nct;
}
return CR_OK; static void delete_constraint(Core *c, ItemConstraint *cv)
{
int idx = linear_index(constraints, cv);
if (idx >= 0)
vector_erase_at(constraints, idx);
c->getWorld()->DeletePersistentData(cv->config);
delete cv;
}
static void print_constraint(Core *c, ItemConstraint *cv, bool no_job = false, std::string prefix = "")
{
c->con << prefix << "Constraint " << cv->config.val() << ": "
<< cv->goalCount() << " (gap " << cv->goalGap() << ")" << endl;
if (no_job) return;
if (cv->jobs.empty())
c->con.printerr(" (no jobs)\n");
for (int i = 0; i < cv->jobs.size(); i++)
{
ProtectedJob *pj = cv->jobs[i];
df::job *job = pj->actual_job;
c->con << prefix << " job " << job->id << ": "
<< ENUM_KEY_STR(job_type, job->job_type);
if (job->flags.bits.suspend)
c->con << " (suspended)";
c->con << endl;
}
}
static void print_job(Core *c, ProtectedJob *pj)
{
if (!pj)
return;
printJobDetails(c, pj->job_copy);
for (int i = 0; i < pj->constraints.size(); i++)
print_constraint(c, pj->constraints[i], true, " ");
}
static void map_job_constraints(Core *c)
{
for (unsigned i = 0; i < constraints.size(); i++)
constraints[i]->jobs.clear();
for (TKnownJobs::const_iterator it = known_jobs.begin(); it != known_jobs.end(); ++it)
{
it->second->constraints.clear();
if (!it->second->live)
continue;
df::job *job = it->second->job_copy;
df::item_type itype = ENUM_ATTR(job_type, item, job->job_type);
if (itype == item_type::NONE)
continue;
int16_t isubtype = job->item_subtype;
int16_t mat_type = job->mat_type;
int32_t mat_index = job->mat_index;
if (itype == item_type::FOOD)
mat_type = -1;
if (mat_type == -1 && job->job_items.size() == 1) {
mat_type = job->job_items[0]->mat_type;
mat_index = job->job_items[0]->mat_index;
}
MaterialInfo mat(mat_type, mat_index);
for (unsigned i = 0; i < constraints.size(); i++)
{
ItemConstraint *ct = constraints[i];
if (ct->item.type != itype ||
(ct->item.subtype != -1 && ct->item.subtype != isubtype))
continue;
if (ct->material.isValid() && ct->material != mat)
continue;
if (ct->mat_mask.whole)
{
if (mat.isValid() && !mat.matches(ct->mat_mask))
continue;
else if (!(job->material_category.whole & ct->mat_mask.whole))
continue;
}
ct->jobs.push_back(it->second);
it->second->constraints.push_back(ct);
}
}
} }
static command_result workflow_cmd(Core *c, vector <string> & parameters) static command_result workflow_cmd(Core *c, vector <string> & parameters)
{ {
CoreSuspender suspend(c); CoreSuspender suspend(c);
update_job_data(c); if (enabled) {
check_lost_jobs(c);
if (parameters.empty()) recover_jobs(c);
return CR_WRONG_USAGE; update_job_data(c);
map_job_constraints(c);
}
df::building *workshop = NULL; df::building *workshop = NULL;
df::job *job = NULL; df::job *job = NULL;
@ -396,10 +628,7 @@ static command_result workflow_cmd(Core *c, vector <string> & parameters)
job = getSelectedWorkshopJob(c, true); job = getSelectedWorkshopJob(c, true);
} }
std::map<int, df::job*> jobs; std::string cmd = parameters.empty() ? "list" : parameters[0];
enumLiveJobs(jobs);
std::string cmd = parameters[0];
if (cmd == "enable") if (cmd == "enable")
{ {
@ -429,19 +658,18 @@ static command_result workflow_cmd(Core *c, vector <string> & parameters)
if (!enabled) if (!enabled)
c->con << "Note: the plugin is not enabled." << endl; c->con << "Note: the plugin is not enabled." << endl;
if (cmd == "list-jobs") if (cmd == "jobs")
{ {
if (workshop) if (workshop)
{ {
for (unsigned i = 0; i < workshop->jobs.size(); i++) for (unsigned i = 0; i < workshop->jobs.size(); i++)
if (get_known(workshop->jobs[i]->id)) print_job(c, get_known(workshop->jobs[i]->id));
printJobDetails(c, workshop->jobs[i]);
} }
else else
{ {
for (TKnownJobs::iterator it = known_jobs.begin(); it != known_jobs.end(); ++it) for (TKnownJobs::iterator it = known_jobs.begin(); it != known_jobs.end(); ++it)
if (df::job *job = jobs[it->first]) if (it->second->live)
printJobDetails(c, job); print_job(c, it->second);
} }
bool pending = false; bool pending = false;
@ -462,6 +690,53 @@ static command_result workflow_cmd(Core *c, vector <string> & parameters)
return CR_OK; return CR_OK;
} }
else if (cmd == "list")
{
for (int i = 0; i < constraints.size(); i++)
print_constraint(c, constraints[i]);
return CR_OK;
}
else if (cmd == "limit")
{
if (parameters.size() < 3)
return CR_WRONG_USAGE;
int limit = atoi(parameters[2].c_str());
if (limit <= 0) {
c->con.printerr("Invalid limit value.\n");
return CR_FAILURE;
}
ItemConstraint *icv = get_constraint(c, parameters[1]);
if (!icv)
return CR_FAILURE;
icv->setGoalCount(limit);
if (parameters.size() >= 4)
icv->setGoalGap(atoi(parameters[3].c_str()));
map_job_constraints(c);
print_constraint(c, icv);
return CR_OK;
}
else if (cmd == "unlimit")
{
if (parameters.size() != 2)
return CR_WRONG_USAGE;
for (int i = 0; i < constraints.size(); i++)
{
if (constraints[i]->config.val() != parameters[1])
continue;
delete_constraint(c, constraints[i]);
return CR_OK;
}
c->con.printerr("Constraint not found: %s\n", parameters[1].c_str());
return CR_FAILURE;
}
else else
return CR_WRONG_USAGE; return CR_WRONG_USAGE;
} }