Merge remote-tracking branches 'ab9rf/labormanager-1142', 'ab9rf/labormanager-1162' and 'ab9rf/labormanager-respect-nature' into develop

develop
lethosor 2018-08-04 00:52:59 -04:00
commit d26b1759a2
2 changed files with 80 additions and 26 deletions

@ -516,7 +516,8 @@ public:
if (j->material_category.bits.bone || if (j->material_category.bits.bone ||
j->material_category.bits.horn || j->material_category.bits.horn ||
j->material_category.bits.tooth || j->material_category.bits.tooth ||
j->material_category.bits.shell) j->material_category.bits.shell ||
j->material_category.bits.pearl)
return df::unit_labor::BONE_CARVE; return df::unit_labor::BONE_CARVE;
else else
{ {

@ -70,6 +70,7 @@
#include <df/personality_facet_type.h> #include <df/personality_facet_type.h>
#include <df/cultural_identity.h> #include <df/cultural_identity.h>
#include <df/ethic_type.h> #include <df/ethic_type.h>
#include <df/value_type.h>
#include "labormanager.h" #include "labormanager.h"
#include "joblabormapper.h" #include "joblabormapper.h"
@ -945,6 +946,15 @@ private:
{ {
if (labor >= 0 && labor <= ENUM_LAST_ITEM(unit_labor)) if (labor >= 0 && labor <= ENUM_LAST_ITEM(unit_labor))
{ {
if (!Units::isValidLabor(dwarf->dwarf, labor))
{
debug("WARN(labormanager): Attempted to %s dwarf %s with ineligible labor %s\n",
value ? "set" : "unset",
dwarf->dwarf->name.first_name.c_str(),
ENUM_KEY_STR(unit_labor, labor).c_str());
return;
}
bool old = dwarf->dwarf->status.labors[labor]; bool old = dwarf->dwarf->status.labors[labor];
dwarf->dwarf->status.labors[labor] = value; dwarf->dwarf->status.labors[labor] = value;
if (old != value) if (old != value)
@ -1182,7 +1192,17 @@ private:
{ {
df::unit* cre = *u; df::unit* cre = *u;
if (Units::isCitizen(cre)) // following tests shamelessly stolen from Dwarf Manipulator plugin
bool isAssignable =
(Units::isOwnCiv(cre)) &&
(Units::isOwnGroup(cre)) &&
(Units::isActive(cre)) &&
(!cre->flags2.bits.visitor) &&
(!cre->flags3.bits.ghostly) &&
(ENUM_ATTR(profession, can_assign_labor, cre->profession));
if (isAssignable)
{ {
dwarf_info_t* dwarf = add_dwarf(cre); dwarf_info_t* dwarf = add_dwarf(cre);
@ -1438,7 +1458,7 @@ private:
{ {
if (labor == unit_labor::NONE) if (labor == unit_labor::NONE)
continue; continue;
if (Units::isValidLabor(dwarf->dwarf, labor))
set_labor(dwarf, labor, false); set_labor(dwarf, labor, false);
} }
} }
@ -1518,6 +1538,26 @@ private:
else if (altruism <= 24) else if (altruism <= 24)
score -= 50000; score -= 50000;
} }
// Favor/disfavor BUTCHER (covers slaughtering), HAUL_ANIMALS (covers caging), and CUTWOOD based on NATURE value
if (labor == df::unit_labor::BUTCHER || labor == df::unit_labor::HAUL_ANIMALS || labor == df::unit_labor::CUTWOOD)
{
int nature = 0;
for (auto i = d->dwarf->status.current_soul->personality.values.begin();
i != d->dwarf->status.current_soul->personality.values.end();
i++)
{
if ((*i)->type == df::value_type::NATURE)
nature = (*i)->strength;
}
if (nature <= -11)
score += 5000;
else if (nature >= 26)
score -= 50000;
}
// This should reweight assigning CUTWOOD jobs based on a citizen's ethic toward killing plants // This should reweight assigning CUTWOOD jobs based on a citizen's ethic toward killing plants
if (labor == df::unit_labor::CUTWOOD) if (labor == df::unit_labor::CUTWOOD)
@ -1690,6 +1730,8 @@ public:
{ {
dwarf_info_t* d = (*k); dwarf_info_t* d = (*k);
if (Units::isValidLabor(d->dwarf, df::unit_labor::HAUL_FOOD))
{
int score = score_labor(d, df::unit_labor::HAUL_FOOD); int score = score_labor(d, df::unit_labor::HAUL_FOOD);
if (score > best_score) if (score > best_score)
@ -1698,6 +1740,7 @@ public:
best_score = score; best_score = score;
} }
} }
}
if (best_score > INT_MIN) if (best_score > INT_MIN)
{ {
@ -1708,7 +1751,7 @@ public:
{ {
if (l == df::unit_labor::NONE) if (l == df::unit_labor::NONE)
continue; continue;
if (Units::isValidLabor((*bestdwarf)->dwarf, l))
set_labor(*bestdwarf, l, l == df::unit_labor::HAUL_FOOD); set_labor(*bestdwarf, l, l == df::unit_labor::HAUL_FOOD);
} }
@ -1822,6 +1865,8 @@ public:
for (std::list<dwarf_info_t*>::iterator k = available_dwarfs.begin(); k != available_dwarfs.end(); k++) for (std::list<dwarf_info_t*>::iterator k = available_dwarfs.begin(); k != available_dwarfs.end(); k++)
{ {
dwarf_info_t* d = (*k); dwarf_info_t* d = (*k);
if (Units::isValidLabor(d->dwarf, labor))
{
int score = score_labor(d, labor); int score = score_labor(d, labor);
if (score > best_score) if (score > best_score)
{ {
@ -1831,6 +1876,7 @@ public:
} }
} }
} }
}
if (best_labor == df::unit_labor::NONE) if (best_labor == df::unit_labor::NONE)
break; break;
@ -1845,7 +1891,9 @@ public:
tools_enum t = default_labor_infos[l].tool; tools_enum t = default_labor_infos[l].tool;
if (l == best_labor && (t == TOOL_NONE || tool_in_use[t] < tool_count[t])) if (l == best_labor &&
Units::isValidLabor((*bestdwarf)->dwarf, l) &&
(t == TOOL_NONE || tool_in_use[t] < tool_count[t]))
{ {
set_labor(*bestdwarf, l, true); set_labor(*bestdwarf, l, true);
if (t != TOOL_NONE && !((*bestdwarf)->has_tool[t])) if (t != TOOL_NONE && !((*bestdwarf)->has_tool[t]))
@ -1865,6 +1913,7 @@ public:
} }
else if ((*bestdwarf)->state == IDLE) else if ((*bestdwarf)->state == IDLE)
{ {
if (Units::isValidLabor((*bestdwarf)->dwarf, l))
set_labor(*bestdwarf, l, false); set_labor(*bestdwarf, l, false);
} }
} }
@ -1897,6 +1946,8 @@ public:
continue; continue;
if (labor_needed[l] <= 0) if (labor_needed[l] <= 0)
continue; continue;
if (!Units::isValidLabor((*d)->dwarf, l))
continue;
int score = score_labor(*d, l); int score = score_labor(*d, l);
@ -1934,7 +1985,8 @@ public:
FOR_ENUM_ITEMS(unit_labor, l) FOR_ENUM_ITEMS(unit_labor, l)
{ {
if (l >= df::unit_labor::HAUL_STONE && l <= df::unit_labor::HAUL_ANIMALS && if (l >= df::unit_labor::HAUL_STONE && l <= df::unit_labor::HAUL_ANIMALS &&
canary & (1 << l)) canary & (1 << l) &&
Units::isValidLabor(canary_dwarf->dwarf, l))
set_labor(canary_dwarf, l, true); set_labor(canary_dwarf, l, true);
} }
@ -1967,6 +2019,7 @@ public:
if (l == df::unit_labor::NONE) if (l == df::unit_labor::NONE)
continue; continue;
if (Units::isValidLabor((*d)->dwarf, l))
set_labor(*d, l, set_labor(*d, l,
(l >= df::unit_labor::HAUL_STONE && l <= df::unit_labor::HAUL_ANIMALS) || (l >= df::unit_labor::HAUL_STONE && l <= df::unit_labor::HAUL_ANIMALS) ||
l == df::unit_labor::CLEAN || l == df::unit_labor::CLEAN ||