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

Large change to how autolabor assigns labors. Military dwarves and 

children/nobles will never be assigned labors. Labor assignments are
more understandable.
develop
RossM 2012-04-02 20:55:02 -07:00
parent b21d813c73
commit ecb1ca0972
1 changed files with 159 additions and 118 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

273
plugins/autolabor.cpp
Unescape Escape View File

@ -31,6 +31,8 @@ using df::global::world;
static int enable_autolabor = 0; static int enable_autolabor = 0;
static bool print_debug = 0;
// Here go all the command declarations... // Here go all the command declarations...
// mostly to allow having the mandatory stuff on top of the file and commands on the bottom // mostly to allow having the mandatory stuff on top of the file and commands on the bottom
@ -41,10 +43,12 @@ command_result autolabor (color_ostream &out, std::vector <std::string> & parame
DFHACK_PLUGIN("autolabor"); DFHACK_PLUGIN("autolabor");
enum labor_mode { enum labor_mode {
FIXED, DISABLE,
AUTOMATIC,
EVERYONE,
HAULERS, HAULERS,
EVERYONE,
MINIMUM,
EXACTLY,
MAXIMUM
}; };
enum dwarf_state { enum dwarf_state {
@ -64,6 +68,14 @@ enum dwarf_state {
OTHER OTHER
}; };
static const char *state_names[] = {
"IDLE",
"BUSY",
"MILITARY",
"CHILD",
"OTHER",
};
static const dwarf_state dwarf_states[] = { static const dwarf_state dwarf_states[] = {
BUSY /* CarveFortification */, BUSY /* CarveFortification */,
BUSY /* DetailWall */, BUSY /* DetailWall */,
@ -303,7 +315,7 @@ struct labor_info
static struct labor_info* labor_infos; static struct labor_info* labor_infos;
static const struct labor_info default_labor_infos[] = { static const struct labor_info default_labor_infos[] = {
/* MINE */ {AUTOMATIC, true, 2}, /* MINE */ {MINIMUM, true, 2},
/* HAUL_STONE */ {HAULERS, false, 1}, /* HAUL_STONE */ {HAULERS, false, 1},
/* HAUL_WOOD */ {HAULERS, false, 1}, /* HAUL_WOOD */ {HAULERS, false, 1},
/* HAUL_BODY */ {HAULERS, false, 1}, /* HAUL_BODY */ {HAULERS, false, 1},
@ -313,69 +325,69 @@ static const struct labor_info default_labor_infos[] = {
/* HAUL_FURNITURE */ {HAULERS, false, 1}, /* HAUL_FURNITURE */ {HAULERS, false, 1},
/* HAUL_ANIMAL */ {HAULERS, false, 1}, /* HAUL_ANIMAL */ {HAULERS, false, 1},
/* CLEAN */ {HAULERS, false, 1}, /* CLEAN */ {HAULERS, false, 1},
/* CUTWOOD */ {AUTOMATIC, true, 1}, /* CUTWOOD */ {MINIMUM, true, 1},
/* CARPENTER */ {AUTOMATIC, false, 1}, /* CARPENTER */ {MINIMUM, false, 1},
/* DETAIL */ {AUTOMATIC, false, 1}, /* DETAIL */ {MINIMUM, false, 1},
/* MASON */ {AUTOMATIC, false, 1}, /* MASON */ {MINIMUM, false, 1},
/* ARCHITECT */ {AUTOMATIC, false, 1}, /* ARCHITECT */ {MINIMUM, false, 1},
/* ANIMALTRAIN */ {AUTOMATIC, false, 1}, /* ANIMALTRAIN */ {MINIMUM, false, 1},
/* ANIMALCARE */ {AUTOMATIC, false, 1}, /* ANIMALCARE */ {MINIMUM, false, 1},
/* DIAGNOSE */ {AUTOMATIC, false, 1}, /* DIAGNOSE */ {MINIMUM, false, 1},
/* SURGERY */ {AUTOMATIC, false, 1}, /* SURGERY */ {MINIMUM, false, 1},
/* BONE_SETTING */ {AUTOMATIC, false, 1}, /* BONE_SETTING */ {MINIMUM, false, 1},
/* SUTURING */ {AUTOMATIC, false, 1}, /* SUTURING */ {MINIMUM, false, 1},
/* DRESSING_WOUNDS */ {AUTOMATIC, false, 1}, /* DRESSING_WOUNDS */ {MINIMUM, false, 1},
/* FEED_WATER_CIVILIANS */ {EVERYONE, false, 1}, /* FEED_WATER_CIVILIANS */ {EVERYONE, false, 1},
/* RECOVER_WOUNDED */ {HAULERS, false, 1}, /* RECOVER_WOUNDED */ {HAULERS, false, 1},
/* BUTCHER */ {AUTOMATIC, false, 1}, /* BUTCHER */ {MINIMUM, false, 1},
/* TRAPPER */ {AUTOMATIC, false, 1}, /* TRAPPER */ {MINIMUM, false, 1},
/* DISSECT_VERMIN */ {AUTOMATIC, false, 1}, /* DISSECT_VERMIN */ {MINIMUM, false, 1},
/* LEATHER */ {AUTOMATIC, false, 1}, /* LEATHER */ {MINIMUM, false, 1},
/* TANNER */ {AUTOMATIC, false, 1}, /* TANNER */ {MINIMUM, false, 1},
/* BREWER */ {AUTOMATIC, false, 1}, /* BREWER */ {MINIMUM, false, 1},
/* ALCHEMIST */ {AUTOMATIC, false, 1}, /* ALCHEMIST */ {MINIMUM, false, 1},
/* SOAP_MAKER */ {AUTOMATIC, false, 1}, /* SOAP_MAKER */ {MINIMUM, false, 1},
/* WEAVER */ {AUTOMATIC, false, 1}, /* WEAVER */ {MINIMUM, false, 1},
/* CLOTHESMAKER */ {AUTOMATIC, false, 1}, /* CLOTHESMAKER */ {MINIMUM, false, 1},
/* MILLER */ {AUTOMATIC, false, 1}, /* MILLER */ {MINIMUM, false, 1},
/* PROCESS_PLANT */ {AUTOMATIC, false, 1}, /* PROCESS_PLANT */ {MINIMUM, false, 1},
/* MAKE_CHEESE */ {AUTOMATIC, false, 1}, /* MAKE_CHEESE */ {MINIMUM, false, 1},
/* MILK */ {AUTOMATIC, false, 1}, /* MILK */ {MINIMUM, false, 1},
/* COOK */ {AUTOMATIC, false, 1}, /* COOK */ {MINIMUM, false, 1},
/* PLANT */ {AUTOMATIC, false, 1}, /* PLANT */ {MINIMUM, false, 1},
/* HERBALIST */ {AUTOMATIC, false, 1}, /* HERBALIST */ {MINIMUM, false, 1},
/* FISH */ {FIXED, false, 1}, /* FISH */ {EXACTLY, false, 1},
/* CLEAN_FISH */ {AUTOMATIC, false, 1}, /* CLEAN_FISH */ {MINIMUM, false, 1},
/* DISSECT_FISH */ {AUTOMATIC, false, 1}, /* DISSECT_FISH */ {MINIMUM, false, 1},
/* HUNT */ {FIXED, true, 1}, /* HUNT */ {EXACTLY, true, 1},
/* SMELT */ {AUTOMATIC, false, 1}, /* SMELT */ {MINIMUM, false, 1},
/* FORGE_WEAPON */ {AUTOMATIC, false, 1}, /* FORGE_WEAPON */ {MINIMUM, false, 1},
/* FORGE_ARMOR */ {AUTOMATIC, false, 1}, /* FORGE_ARMOR */ {MINIMUM, false, 1},
/* FORGE_FURNITURE */ {AUTOMATIC, false, 1}, /* FORGE_FURNITURE */ {MINIMUM, false, 1},
/* METAL_CRAFT */ {AUTOMATIC, false, 1}, /* METAL_CRAFT */ {MINIMUM, false, 1},
/* CUT_GEM */ {AUTOMATIC, false, 1}, /* CUT_GEM */ {MINIMUM, false, 1},
/* ENCRUST_GEM */ {AUTOMATIC, false, 1}, /* ENCRUST_GEM */ {MINIMUM, false, 1},
/* WOOD_CRAFT */ {AUTOMATIC, false, 1}, /* WOOD_CRAFT */ {MINIMUM, false, 1},
/* STONE_CRAFT */ {AUTOMATIC, false, 1}, /* STONE_CRAFT */ {MINIMUM, false, 1},
/* BONE_CARVE */ {AUTOMATIC, false, 1}, /* BONE_CARVE */ {MINIMUM, false, 1},
/* GLASSMAKER */ {AUTOMATIC, false, 1}, /* GLASSMAKER */ {MINIMUM, false, 1},
/* EXTRACT_STRAND */ {AUTOMATIC, false, 1}, /* EXTRACT_STRAND */ {MINIMUM, false, 1},
/* SIEGECRAFT */ {AUTOMATIC, false, 1}, /* SIEGECRAFT */ {MINIMUM, false, 1},
/* SIEGEOPERATE */ {AUTOMATIC, false, 1}, /* SIEGEOPERATE */ {MINIMUM, false, 1},
/* BOWYER */ {AUTOMATIC, false, 1}, /* BOWYER */ {MINIMUM, false, 1},
/* MECHANIC */ {AUTOMATIC, false, 1}, /* MECHANIC */ {MINIMUM, false, 1},
/* POTASH_MAKING */ {AUTOMATIC, false, 1}, /* POTASH_MAKING */ {MINIMUM, false, 1},
/* LYE_MAKING */ {AUTOMATIC, false, 1}, /* LYE_MAKING */ {MINIMUM, false, 1},
/* DYER */ {AUTOMATIC, false, 1}, /* DYER */ {MINIMUM, false, 1},
/* BURN_WOOD */ {AUTOMATIC, false, 1}, /* BURN_WOOD */ {MINIMUM, false, 1},
/* OPERATE_PUMP */ {AUTOMATIC, false, 1}, /* OPERATE_PUMP */ {MINIMUM, false, 1},
/* SHEARER */ {AUTOMATIC, false, 1}, /* SHEARER */ {MINIMUM, false, 1},
/* SPINNER */ {AUTOMATIC, false, 1}, /* SPINNER */ {MINIMUM, false, 1},
/* POTTERY */ {AUTOMATIC, false, 1}, /* POTTERY */ {MINIMUM, false, 1},
/* GLAZING */ {AUTOMATIC, false, 1}, /* GLAZING */ {MINIMUM, false, 1},
/* PRESSING */ {AUTOMATIC, false, 1}, /* PRESSING */ {MINIMUM, false, 1},
/* BEEKEEPING */ {AUTOMATIC, false, 1}, /* BEEKEEPING */ {MINIMUM, false, 1},
/* WAX_WORKING */ {AUTOMATIC, false, 1}, /* WAX_WORKING */ {MINIMUM, false, 1},
}; };
static const df::job_skill noble_skills[] = { static const df::job_skill noble_skills[] = {
@ -651,6 +663,9 @@ DFhackCExport command_result plugin_onupdate ( color_ostream &out )
} }
state_count[dwarf_info[dwarf].state]++; state_count[dwarf_info[dwarf].state]++;
if (print_debug)
out.print("Dwarf %i \"%s\": penalty %i, state %s\n", dwarf, dwarfs[dwarf]->name.first_name.c_str(), dwarf_info[dwarf].mastery_penalty, state_names[dwarf_info[dwarf].state]);
} }
// Generate labor -> skill mapping // Generate labor -> skill mapping
@ -690,6 +705,26 @@ DFhackCExport command_result plugin_onupdate ( color_ostream &out )
laborinfo_sorter lasorter; laborinfo_sorter lasorter;
std::sort(labors.begin(), labors.end(), lasorter); std::sort(labors.begin(), labors.end(), lasorter);
// Handle DISABLED skills (just bookkeeping)
for (auto lp = labors.begin(); lp != labors.end(); ++lp)
{
auto labor = *lp;
if (labor_infos[labor].mode != DISABLE)
continue;
for (int dwarf = 0; dwarf < n_dwarfs; dwarf++)
{
if (dwarfs[dwarf]->status.labors[labor])
{
if (labor_infos[labor].is_exclusive)
dwarf_info[dwarf].has_exclusive_labor = true;
dwarf_info[dwarf].assigned_jobs++;
}
}
}
// Handle all skills except those marked HAULERS // Handle all skills except those marked HAULERS
for (auto lp = labors.begin(); lp != labors.end(); ++lp) for (auto lp = labors.begin(); lp != labors.end(); ++lp)
@ -705,28 +740,53 @@ DFhackCExport command_result plugin_onupdate ( color_ostream &out )
if (labor_infos[labor].mode == HAULERS) if (labor_infos[labor].mode == HAULERS)
continue; continue;
if (labor_infos[labor].mode == DISABLE)
continue;
int best_dwarf = 0; int best_dwarf = 0;
int best_value = -10000; int best_value = -10000;
std::vector<int> values(n_dwarfs); std::vector<int> values(n_dwarfs);
std::vector<int> candidates; std::vector<int> candidates;
std::vector<int> backup_candidates;
std::map<int, int> dwarf_skill; std::map<int, int> dwarf_skill;
auto mode = labor_infos[labor].mode; auto mode = labor_infos[labor].mode;
if (AUTOMATIC == mode && state_count[IDLE] == 0)
mode = FIXED; int min_dwarfs = 0;
int max_dwarfs = 0;
switch (mode)
{
case EVERYONE:
min_dwarfs = max_dwarfs = n_dwarfs;
break;
case MINIMUM:
min_dwarfs = labor_infos[labor].minimum_dwarfs;
max_dwarfs = n_dwarfs;
break;
case EXACTLY:
min_dwarfs = max_dwarfs = labor_infos[labor].minimum_dwarfs;
break;
case MAXIMUM:
min_dwarfs = 0;
max_dwarfs = labor_infos[labor].minimum_dwarfs;
break;
}
for (int dwarf = 0; dwarf < n_dwarfs; dwarf++) for (int dwarf = 0; dwarf < n_dwarfs; dwarf++)
{ {
if (dwarf_info[dwarf].state != IDLE && dwarf_info[dwarf].state != BUSY && mode != EVERYONE) if (dwarf_info[dwarf].state == CHILD)
continue;
if (dwarf_info[dwarf].state == MILITARY)
continue; continue;
if (labor_infos[labor].is_exclusive && dwarf_info[dwarf].has_exclusive_labor) if (labor_infos[labor].is_exclusive && dwarf_info[dwarf].has_exclusive_labor)
continue; continue;
int value = dwarf_info[dwarf].mastery_penalty - dwarf_info[dwarf].assigned_jobs * 50; int value = dwarf_info[dwarf].mastery_penalty; // - dwarf_info[dwarf].assigned_jobs * 50;
if (skill != df::enums::job_skill::NONE) if (skill != df::enums::job_skill::NONE)
{ {
@ -766,71 +826,33 @@ DFhackCExport command_result plugin_onupdate ( color_ostream &out )
values[dwarf] = value; values[dwarf] = value;
if (mode == AUTOMATIC && dwarf_info[dwarf].state != IDLE)
backup_candidates.push_back(dwarf);
else
candidates.push_back(dwarf); candidates.push_back(dwarf);
} }
if (candidates.size() == 0)
{
candidates = backup_candidates;
mode = FIXED;
}
if (labor_infos[labor].mode != EVERYONE) if (labor_infos[labor].mode != EVERYONE)
{ {
values_sorter ivs(values); values_sorter ivs(values);
std::sort(candidates.begin(), candidates.end(), ivs); std::sort(candidates.begin(), candidates.end(), ivs);
} }
for (int dwarf = 0; dwarf < n_dwarfs; dwarf++) int active_dwarfs = 0;
{
bool allow_labor = false;
if (dwarf_info[dwarf].state == BUSY &&
mode == AUTOMATIC &&
(labor_infos[labor].is_exclusive || dwarf_skill[dwarf] > 0))
{
allow_labor = true;
}
if (dwarf_info[dwarf].state == OTHER &&
mode == AUTOMATIC &&
dwarf_skill[dwarf] > 0 &&
!dwarf_info[dwarf].is_best_noble)
{
allow_labor = true;
}
if (dwarfs[dwarf]->status.labors[labor] && for (int dwarf = 0; dwarf < n_dwarfs; dwarf++)
allow_labor &&
!(labor_infos[labor].is_exclusive && dwarf_info[dwarf].has_exclusive_labor))
{ {
if (labor_infos[labor].is_exclusive) if (dwarf_info[dwarf].state == CHILD)
dwarf_info[dwarf].has_exclusive_labor = true; continue;
dwarf_info[dwarf].assigned_jobs++;
}
else
{
dwarfs[dwarf]->status.labors[labor] = false; dwarfs[dwarf]->status.labors[labor] = false;
} }
}
int minimum_dwarfs = labor_infos[labor].minimum_dwarfs;
if (labor_infos[labor].mode == EVERYONE)
minimum_dwarfs = n_dwarfs;
// Special - don't assign hunt without a butchers, or fish without a fishery // Special - don't assign hunt without a butchers, or fish without a fishery
if (df::enums::unit_labor::HUNT == labor && !has_butchers) if (df::enums::unit_labor::HUNT == labor && !has_butchers)
minimum_dwarfs = 0; min_dwarfs = max_dwarfs = 0;
if (df::enums::unit_labor::FISH == labor && !has_fishery) if (df::enums::unit_labor::FISH == labor && !has_fishery)
minimum_dwarfs = 0; min_dwarfs = max_dwarfs = 0;
for (int i = 0; i < candidates.size() && i < minimum_dwarfs; i++) for (int i = 0; i < candidates.size() && active_dwarfs < max_dwarfs; i++)
{ {
int dwarf = candidates[i]; int dwarf = candidates[i];
@ -848,6 +870,15 @@ DFhackCExport command_result plugin_onupdate ( color_ostream &out )
// all the exclusive labors require equipment so this should force the dorf to reequip if needed // all the exclusive labors require equipment so this should force the dorf to reequip if needed
dwarfs[dwarf]->military.pickup_flags.bits.update = 1; dwarfs[dwarf]->military.pickup_flags.bits.update = 1;
} }
if (print_debug)
out.print("Dwarf %i \"%s\" assigned %s: value %i\n", dwarf, dwarfs[dwarf]->name.first_name.c_str(), ENUM_KEY_STR(unit_labor, labor).c_str(), values[dwarf]);
if (dwarf_info[dwarf].state == IDLE || dwarf_info[dwarf].state == BUSY)
active_dwarfs++;
if (active_dwarfs >= min_dwarfs && (dwarf_info[dwarf].state == IDLE || state_count[IDLE] < 2))
break;
} }
} }
@ -896,6 +927,9 @@ DFhackCExport command_result plugin_onupdate ( color_ostream &out )
assert(dwarf < n_dwarfs); assert(dwarf < n_dwarfs);
dwarfs[dwarf]->status.labors[labor] = true; dwarfs[dwarf]->status.labors[labor] = true;
dwarf_info[dwarf].assigned_jobs++; dwarf_info[dwarf].assigned_jobs++;
if (print_debug)
out.print("Dwarf %i \"%s\" assigned %s: hauler\n", dwarf, dwarfs[dwarf]->name.first_name.c_str(), ENUM_KEY_STR(unit_labor, labor).c_str());
} }
for (int i = num_haulers; i < hauler_ids.size(); i++) for (int i = num_haulers; i < hauler_ids.size(); i++)
@ -910,6 +944,9 @@ DFhackCExport command_result plugin_onupdate ( color_ostream &out )
dwarfs[dwarf]->status.labors[labor] = false; dwarfs[dwarf]->status.labors[labor] = false;
} }
} }
print_debug = 0;
return CR_OK; return CR_OK;
} }
@ -935,7 +972,11 @@ command_result autolabor (color_ostream &out, std::vector <std::string> & parame
out.print("Syntax: autolabor miners <n>, where n is 0 or more.\n" out.print("Syntax: autolabor miners <n>, where n is 0 or more.\n"
"Current miner count: %d\n", labor_infos[0].minimum_dwarfs); "Current miner count: %d\n", labor_infos[0].minimum_dwarfs);
} }
} else }
else if (parameters.size() == 1 && parameters[0] == "debug") {
print_debug = 1;
}
else
{ {
out.print("Automatically assigns labors to dwarves.\n" out.print("Automatically assigns labors to dwarves.\n"
"Activate with 'autolabor 1', deactivate with 'autolabor 0'.\n" "Activate with 'autolabor 1', deactivate with 'autolabor 0'.\n"