Merge remote-tracking branch 'ab9rf/labormanager-1103' into develop
lethosor
commit
eb7e9dcf1f
@ -0,0 +1,34 @@
|
|||||||
|
PROJECT (labormanager)
|
||||||
|
# A list of source files
|
||||||
|
SET(PROJECT_SRCS
|
||||||
|
labormanager.cpp
|
||||||
|
joblabormapper.cpp
|
||||||
|
)
|
||||||
|
# A list of headers
|
||||||
|
SET(PROJECT_HDRS
|
||||||
|
labormanager.h
|
||||||
|
joblabormapper.h
|
||||||
|
)
|
||||||
|
SET_SOURCE_FILES_PROPERTIES( ${PROJECT_HDRS} PROPERTIES HEADER_FILE_ONLY TRUE)
|
||||||
|
|
||||||
|
# mash them together (headers are marked as headers and nothing will try to compile them)
|
||||||
|
LIST(APPEND PROJECT_SRCS ${PROJECT_HDRS})
|
||||||
|
|
||||||
|
#linux
|
||||||
|
IF(UNIX)
|
||||||
|
add_definitions(-DLINUX_BUILD)
|
||||||
|
SET(PROJECT_LIBS
|
||||||
|
# add any extra linux libs here
|
||||||
|
${PROJECT_LIBS}
|
||||||
|
)
|
||||||
|
# windows
|
||||||
|
ELSE(UNIX)
|
||||||
|
SET(PROJECT_LIBS
|
||||||
|
# add any extra linux libs here
|
||||||
|
${PROJECT_LIBS}
|
||||||
|
$(NOINHERIT)
|
||||||
|
)
|
||||||
|
ENDIF(UNIX)
|
||||||
|
# this makes sure all the stuff is put in proper places and linked to dfhack
|
||||||
|
|
||||||
|
DFHACK_PLUGIN(labormanager ${PROJECT_SRCS} LINK_LIBRARIES ${PROJECT_LIBS})
|
||||||
@ -0,0 +1,920 @@
|
|||||||
|
/*a
|
||||||
|
* This file contains the logic to attempt to intuit the labor required for
|
||||||
|
* a given job. This is way more complicated than it should be, but I have
|
||||||
|
* not figured out how to make it simpler.
|
||||||
|
*
|
||||||
|
* Usage:
|
||||||
|
* Instantiate an instance of the JobLaborMapper class
|
||||||
|
* Call the find_job_labor method of that class instance,
|
||||||
|
* passing the job as the only argument, to determine the labor for that job
|
||||||
|
* When done, destroy the instance
|
||||||
|
*
|
||||||
|
* The class should allow you to create multiple instances, although there is
|
||||||
|
* little benefit to doing so. jlfuncs are not reused across instances.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
#include "DataDefs.h"
|
||||||
|
#include "MiscUtils.h"
|
||||||
|
#include "modules/Materials.h"
|
||||||
|
|
||||||
|
#include <df/building.h>
|
||||||
|
#include <df/building_actual.h>
|
||||||
|
#include <df/building_def.h>
|
||||||
|
#include <df/building_design.h>
|
||||||
|
#include <df/building_furnacest.h>
|
||||||
|
#include <df/building_type.h>
|
||||||
|
#include <df/building_workshopst.h>
|
||||||
|
|
||||||
|
#include <df/furnace_type.h>
|
||||||
|
|
||||||
|
#include <df/general_ref.h>
|
||||||
|
#include <df/general_ref_building_holderst.h>
|
||||||
|
#include <df/general_ref_contains_itemst.h>
|
||||||
|
|
||||||
|
#include <df/item.h>
|
||||||
|
#include <df/item_type.h>
|
||||||
|
|
||||||
|
#include <df/job.h>
|
||||||
|
#include <df/job_item.h>
|
||||||
|
#include <df/job_item_ref.h>
|
||||||
|
|
||||||
|
#include <df/material_flags.h>
|
||||||
|
|
||||||
|
#include <df/reaction.h>
|
||||||
|
|
||||||
|
#include <df/unit_labor.h>
|
||||||
|
|
||||||
|
#include <df/world.h>
|
||||||
|
|
||||||
|
#include <vector>
|
||||||
|
#include <set>
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
using std::string;
|
||||||
|
using std::endl;
|
||||||
|
using namespace DFHack;
|
||||||
|
using namespace df::enums;
|
||||||
|
using df::global::ui;
|
||||||
|
using df::global::world;
|
||||||
|
|
||||||
|
#include "labormanager.h"
|
||||||
|
#include "joblabormapper.h"
|
||||||
|
|
||||||
|
static df::unit_labor hauling_labor_map[] =
|
||||||
|
{
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BAR */
|
||||||
|
df::unit_labor::HAUL_STONE, /* SMALLGEM */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BLOCKS */
|
||||||
|
df::unit_labor::HAUL_STONE, /* ROUGH */
|
||||||
|
df::unit_labor::HAUL_STONE, /* BOULDER */
|
||||||
|
df::unit_labor::HAUL_WOOD, /* WOOD */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* DOOR */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* FLOODGATE */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* BED */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* CHAIR */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* CHAIN */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* FLASK */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* GOBLET */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* INSTRUMENT */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* TOY */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* WINDOW */
|
||||||
|
df::unit_labor::HAUL_ANIMALS, /* CAGE */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BARREL */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BUCKET */
|
||||||
|
df::unit_labor::HAUL_ANIMALS, /* ANIMALTRAP */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* TABLE */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* COFFIN */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* STATUE */
|
||||||
|
df::unit_labor::HAUL_REFUSE, /* CORPSE */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* WEAPON */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* ARMOR */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* SHOES */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* SHIELD */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* HELM */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* GLOVES */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* BOX */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BIN */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* ARMORSTAND */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* WEAPONRACK */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* CABINET */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* FIGURINE */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* AMULET */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* SCEPTER */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* AMMO */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* CROWN */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* RING */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* EARRING */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BRACELET */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* GEM */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* ANVIL */
|
||||||
|
df::unit_labor::HAUL_REFUSE, /* CORPSEPIECE */
|
||||||
|
df::unit_labor::HAUL_REFUSE, /* REMAINS */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* MEAT */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* FISH */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* FISH_RAW */
|
||||||
|
df::unit_labor::HAUL_REFUSE, /* VERMIN */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* PET */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* SEEDS */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* PLANT */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* SKIN_TANNED */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* LEAVES */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* THREAD */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* CLOTH */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* TOTEM */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* PANTS */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BACKPACK */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* QUIVER */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* CATAPULTPARTS */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* BALLISTAPARTS */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* SIEGEAMMO */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* BALLISTAARROWHEAD */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* TRAPPARTS */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* TRAPCOMP */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* DRINK */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* POWDER_MISC */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* CHEESE */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* FOOD */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* LIQUID_MISC */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* COIN */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* GLOB */
|
||||||
|
df::unit_labor::HAUL_STONE, /* ROCK */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* PIPE_SECTION */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* HATCH_COVER */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* GRATE */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* QUERN */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* MILLSTONE */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* SPLINT */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* CRUTCH */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* TRACTION_BENCH */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* ORTHOPEDIC_CAST */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* TOOL */
|
||||||
|
df::unit_labor::HAUL_FURNITURE, /* SLAB */
|
||||||
|
df::unit_labor::HAUL_FOOD, /* EGG */
|
||||||
|
df::unit_labor::HAUL_ITEM, /* BOOK */
|
||||||
|
};
|
||||||
|
|
||||||
|
static df::unit_labor workshop_build_labor[] =
|
||||||
|
{
|
||||||
|
/* Carpenters */ df::unit_labor::CARPENTER,
|
||||||
|
/* Farmers */ df::unit_labor::PROCESS_PLANT,
|
||||||
|
/* Masons */ df::unit_labor::MASON,
|
||||||
|
/* Craftsdwarfs */ df::unit_labor::STONE_CRAFT,
|
||||||
|
/* Jewelers */ df::unit_labor::CUT_GEM,
|
||||||
|
/* MetalsmithsForge */ df::unit_labor::METAL_CRAFT,
|
||||||
|
/* MagmaForge */ df::unit_labor::METAL_CRAFT,
|
||||||
|
/* Bowyers */ df::unit_labor::BOWYER,
|
||||||
|
/* Mechanics */ df::unit_labor::MECHANIC,
|
||||||
|
/* Siege */ df::unit_labor::SIEGECRAFT,
|
||||||
|
/* Butchers */ df::unit_labor::BUTCHER,
|
||||||
|
/* Leatherworks */ df::unit_labor::LEATHER,
|
||||||
|
/* Tanners */ df::unit_labor::TANNER,
|
||||||
|
/* Clothiers */ df::unit_labor::CLOTHESMAKER,
|
||||||
|
/* Fishery */ df::unit_labor::CLEAN_FISH,
|
||||||
|
/* Still */ df::unit_labor::BREWER,
|
||||||
|
/* Loom */ df::unit_labor::WEAVER,
|
||||||
|
/* Quern */ df::unit_labor::MILLER,
|
||||||
|
/* Kennels */ df::unit_labor::ANIMALTRAIN,
|
||||||
|
/* Kitchen */ df::unit_labor::COOK,
|
||||||
|
/* Ashery */ df::unit_labor::LYE_MAKING,
|
||||||
|
/* Dyers */ df::unit_labor::DYER,
|
||||||
|
/* Millstone */ df::unit_labor::MILLER,
|
||||||
|
/* Custom */ df::unit_labor::NONE,
|
||||||
|
/* Tool */ df::unit_labor::NONE
|
||||||
|
};
|
||||||
|
|
||||||
|
static df::building* get_building_from_job(df::job* j)
|
||||||
|
{
|
||||||
|
for (auto r = j->general_refs.begin(); r != j->general_refs.end(); r++)
|
||||||
|
{
|
||||||
|
if ((*r)->getType() == df::general_ref_type::BUILDING_HOLDER)
|
||||||
|
{
|
||||||
|
int32_t id = ((df::general_ref_building_holderst*)(*r))->building_id;
|
||||||
|
df::building* bld = binsearch_in_vector(world->buildings.all, id);
|
||||||
|
return bld;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static df::unit_labor construction_build_labor(df::building_actual* b)
|
||||||
|
{
|
||||||
|
if (b->getType() == df::building_type::RoadPaved)
|
||||||
|
return df::unit_labor::BUILD_ROAD;
|
||||||
|
// Find last item in building with use mode appropriate to the building's constructions state
|
||||||
|
// For screw pumps contained_items[0] = pipe, 1 corkscrew, 2 block
|
||||||
|
// For wells 0 mechanism, 1 rope, 2 bucket, 3 block
|
||||||
|
// Trade depots and bridges use the last one too
|
||||||
|
// Must check use mode b/c buildings may have items in them that are not part of the building
|
||||||
|
|
||||||
|
df::item* i = 0;
|
||||||
|
for (auto p = b->contained_items.begin(); p != b->contained_items.end(); p++)
|
||||||
|
if (b->construction_stage > 0 && (*p)->use_mode == 2 ||
|
||||||
|
b->construction_stage == 0 && (*p)->use_mode == 0)
|
||||||
|
i = (*p)->item;
|
||||||
|
|
||||||
|
MaterialInfo matinfo;
|
||||||
|
if (i && matinfo.decode(i))
|
||||||
|
{
|
||||||
|
if (matinfo.material->flags.is_set(df::material_flags::IS_METAL))
|
||||||
|
return df::unit_labor::METAL_CRAFT;
|
||||||
|
if (matinfo.material->flags.is_set(df::material_flags::WOOD))
|
||||||
|
return df::unit_labor::CARPENTER;
|
||||||
|
}
|
||||||
|
return df::unit_labor::MASON;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
class jlfunc
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
virtual df::unit_labor get_labor(df::job* j) = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
class jlfunc_const : public jlfunc
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
df::unit_labor labor;
|
||||||
|
public:
|
||||||
|
df::unit_labor get_labor(df::job* j)
|
||||||
|
{
|
||||||
|
return labor;
|
||||||
|
}
|
||||||
|
jlfunc_const(df::unit_labor l) : labor(l) {};
|
||||||
|
};
|
||||||
|
|
||||||
|
class jlfunc_hauling : public jlfunc
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
df::unit_labor get_labor(df::job* j)
|
||||||
|
{
|
||||||
|
df::item* item = 0;
|
||||||
|
if (j->job_type == df::job_type::StoreItemInStockpile && j->item_subtype != -1)
|
||||||
|
return (df::unit_labor) j->item_subtype;
|
||||||
|
|
||||||
|
for (auto i = j->items.begin(); i != j->items.end(); i++)
|
||||||
|
{
|
||||||
|
if ((*i)->role == 7)
|
||||||
|
{
|
||||||
|
item = (*i)->item;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (item && item->flags.bits.container)
|
||||||
|
{
|
||||||
|
for (auto a = item->general_refs.begin(); a != item->general_refs.end(); a++)
|
||||||
|
{
|
||||||
|
if ((*a)->getType() == df::general_ref_type::CONTAINS_ITEM)
|
||||||
|
{
|
||||||
|
int item_id = ((df::general_ref_contains_itemst *) (*a))->item_id;
|
||||||
|
item = binsearch_in_vector(world->items.all, item_id);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
df::unit_labor l = item ? hauling_labor_map[item->getType()] : df::unit_labor::HAUL_ITEM;
|
||||||
|
if (item && l == df::unit_labor::HAUL_REFUSE && item->flags.bits.dead_dwarf)
|
||||||
|
l = df::unit_labor::HAUL_BODY;
|
||||||
|
return l;
|
||||||
|
}
|
||||||
|
jlfunc_hauling() {};
|
||||||
|
};
|
||||||
|
|
||||||
|
class jlfunc_construct_bld : public jlfunc
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
df::unit_labor get_labor(df::job* j)
|
||||||
|
{
|
||||||
|
if (j->flags.bits.item_lost)
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
|
||||||
|
df::building* bld = get_building_from_job(j);
|
||||||
|
switch (bld->getType())
|
||||||
|
{
|
||||||
|
case df::building_type::Hive:
|
||||||
|
return df::unit_labor::BEEKEEPING;
|
||||||
|
case df::building_type::Workshop:
|
||||||
|
{
|
||||||
|
df::building_workshopst* ws = (df::building_workshopst*) bld;
|
||||||
|
if (ws->design && !ws->design->flags.bits.designed)
|
||||||
|
return df::unit_labor::ARCHITECT;
|
||||||
|
if (ws->type == df::workshop_type::Custom)
|
||||||
|
{
|
||||||
|
df::building_def* def = df::building_def::find(ws->custom_type);
|
||||||
|
return def->build_labors[0];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return workshop_build_labor[ws->type];
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case df::building_type::Construction:
|
||||||
|
return df::unit_labor::BUILD_CONSTRUCTION;
|
||||||
|
case df::building_type::Furnace:
|
||||||
|
case df::building_type::TradeDepot:
|
||||||
|
case df::building_type::Bridge:
|
||||||
|
case df::building_type::ArcheryTarget:
|
||||||
|
case df::building_type::WaterWheel:
|
||||||
|
case df::building_type::RoadPaved:
|
||||||
|
case df::building_type::Well:
|
||||||
|
case df::building_type::ScrewPump:
|
||||||
|
case df::building_type::Wagon:
|
||||||
|
case df::building_type::Shop:
|
||||||
|
case df::building_type::Support:
|
||||||
|
case df::building_type::Windmill:
|
||||||
|
{
|
||||||
|
df::building_actual* b = (df::building_actual*) bld;
|
||||||
|
if (b->design && !b->design->flags.bits.designed)
|
||||||
|
return df::unit_labor::ARCHITECT;
|
||||||
|
return construction_build_labor(b);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case df::building_type::FarmPlot:
|
||||||
|
return df::unit_labor::PLANT;
|
||||||
|
case df::building_type::Chair:
|
||||||
|
case df::building_type::Bed:
|
||||||
|
case df::building_type::Table:
|
||||||
|
case df::building_type::Coffin:
|
||||||
|
case df::building_type::Door:
|
||||||
|
case df::building_type::Floodgate:
|
||||||
|
case df::building_type::Box:
|
||||||
|
case df::building_type::Weaponrack:
|
||||||
|
case df::building_type::Armorstand:
|
||||||
|
case df::building_type::Cabinet:
|
||||||
|
case df::building_type::Statue:
|
||||||
|
case df::building_type::WindowGlass:
|
||||||
|
case df::building_type::WindowGem:
|
||||||
|
case df::building_type::Cage:
|
||||||
|
case df::building_type::NestBox:
|
||||||
|
case df::building_type::TractionBench:
|
||||||
|
case df::building_type::Slab:
|
||||||
|
case df::building_type::Chain:
|
||||||
|
case df::building_type::GrateFloor:
|
||||||
|
case df::building_type::Hatch:
|
||||||
|
case df::building_type::BarsFloor:
|
||||||
|
case df::building_type::BarsVertical:
|
||||||
|
case df::building_type::GrateWall:
|
||||||
|
case df::building_type::Bookcase:
|
||||||
|
case df::building_type::Instrument:
|
||||||
|
return df::unit_labor::HAUL_FURNITURE;
|
||||||
|
case df::building_type::Trap:
|
||||||
|
case df::building_type::GearAssembly:
|
||||||
|
case df::building_type::AxleHorizontal:
|
||||||
|
case df::building_type::AxleVertical:
|
||||||
|
case df::building_type::Rollers:
|
||||||
|
return df::unit_labor::MECHANIC;
|
||||||
|
case df::building_type::AnimalTrap:
|
||||||
|
return df::unit_labor::TRAPPER;
|
||||||
|
case df::building_type::Civzone:
|
||||||
|
case df::building_type::Nest:
|
||||||
|
case df::building_type::Stockpile:
|
||||||
|
case df::building_type::Weapon:
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
case df::building_type::SiegeEngine:
|
||||||
|
return df::unit_labor::SIEGECRAFT;
|
||||||
|
case df::building_type::RoadDirt:
|
||||||
|
return df::unit_labor::BUILD_ROAD;
|
||||||
|
}
|
||||||
|
|
||||||
|
debug("LABORMANAGER: Cannot deduce labor for construct building job of type %s\n",
|
||||||
|
ENUM_KEY_STR(building_type, bld->getType()).c_str());
|
||||||
|
debug_pause();
|
||||||
|
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
jlfunc_construct_bld() {}
|
||||||
|
};
|
||||||
|
|
||||||
|
class jlfunc_destroy_bld : public jlfunc
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
df::unit_labor get_labor(df::job* j)
|
||||||
|
{
|
||||||
|
df::building* bld = get_building_from_job(j);
|
||||||
|
df::building_type type = bld->getType();
|
||||||
|
|
||||||
|
switch (bld->getType())
|
||||||
|
{
|
||||||
|
case df::building_type::Hive:
|
||||||
|
return df::unit_labor::BEEKEEPING;
|
||||||
|
case df::building_type::Workshop:
|
||||||
|
{
|
||||||
|
df::building_workshopst* ws = (df::building_workshopst*) bld;
|
||||||
|
if (ws->type == df::workshop_type::Custom)
|
||||||
|
{
|
||||||
|
df::building_def* def = df::building_def::find(ws->custom_type);
|
||||||
|
return def->build_labors[0];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return workshop_build_labor[ws->type];
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case df::building_type::Construction:
|
||||||
|
return df::unit_labor::REMOVE_CONSTRUCTION;
|
||||||
|
case df::building_type::Furnace:
|
||||||
|
case df::building_type::TradeDepot:
|
||||||
|
case df::building_type::Wagon:
|
||||||
|
case df::building_type::Bridge:
|
||||||
|
case df::building_type::ScrewPump:
|
||||||
|
case df::building_type::ArcheryTarget:
|
||||||
|
case df::building_type::RoadPaved:
|
||||||
|
case df::building_type::Shop:
|
||||||
|
case df::building_type::Support:
|
||||||
|
case df::building_type::WaterWheel:
|
||||||
|
case df::building_type::Well:
|
||||||
|
case df::building_type::Windmill:
|
||||||
|
{
|
||||||
|
auto b = (df::building_actual*) bld;
|
||||||
|
return construction_build_labor(b);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case df::building_type::FarmPlot:
|
||||||
|
return df::unit_labor::PLANT;
|
||||||
|
case df::building_type::Trap:
|
||||||
|
case df::building_type::AxleHorizontal:
|
||||||
|
case df::building_type::AxleVertical:
|
||||||
|
case df::building_type::GearAssembly:
|
||||||
|
case df::building_type::Rollers:
|
||||||
|
return df::unit_labor::MECHANIC;
|
||||||
|
case df::building_type::Chair:
|
||||||
|
case df::building_type::Bed:
|
||||||
|
case df::building_type::Table:
|
||||||
|
case df::building_type::Coffin:
|
||||||
|
case df::building_type::Door:
|
||||||
|
case df::building_type::Floodgate:
|
||||||
|
case df::building_type::Box:
|
||||||
|
case df::building_type::Weaponrack:
|
||||||
|
case df::building_type::Armorstand:
|
||||||
|
case df::building_type::Cabinet:
|
||||||
|
case df::building_type::Statue:
|
||||||
|
case df::building_type::WindowGlass:
|
||||||
|
case df::building_type::WindowGem:
|
||||||
|
case df::building_type::Cage:
|
||||||
|
case df::building_type::NestBox:
|
||||||
|
case df::building_type::TractionBench:
|
||||||
|
case df::building_type::Slab:
|
||||||
|
case df::building_type::Chain:
|
||||||
|
case df::building_type::Hatch:
|
||||||
|
case df::building_type::BarsFloor:
|
||||||
|
case df::building_type::BarsVertical:
|
||||||
|
case df::building_type::GrateFloor:
|
||||||
|
case df::building_type::GrateWall:
|
||||||
|
case df::building_type::Bookcase:
|
||||||
|
case df::building_type::Instrument:
|
||||||
|
return df::unit_labor::HAUL_FURNITURE;
|
||||||
|
case df::building_type::AnimalTrap:
|
||||||
|
return df::unit_labor::TRAPPER;
|
||||||
|
case df::building_type::Civzone:
|
||||||
|
case df::building_type::Nest:
|
||||||
|
case df::building_type::RoadDirt:
|
||||||
|
case df::building_type::Stockpile:
|
||||||
|
case df::building_type::Weapon:
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
case df::building_type::SiegeEngine:
|
||||||
|
return df::unit_labor::SIEGECRAFT;
|
||||||
|
}
|
||||||
|
|
||||||
|
debug("LABORMANAGER: Cannot deduce labor for destroy building job of type %s\n",
|
||||||
|
ENUM_KEY_STR(building_type, bld->getType()).c_str());
|
||||||
|
debug_pause();
|
||||||
|
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
jlfunc_destroy_bld() {}
|
||||||
|
};
|
||||||
|
|
||||||
|
class jlfunc_make : public jlfunc
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
df::unit_labor metaltype;
|
||||||
|
public:
|
||||||
|
df::unit_labor get_labor(df::job* j)
|
||||||
|
{
|
||||||
|
df::building* bld = get_building_from_job(j);
|
||||||
|
if (bld->getType() == df::building_type::Workshop)
|
||||||
|
{
|
||||||
|
df::workshop_type type = ((df::building_workshopst*)(bld))->type;
|
||||||
|
switch (type)
|
||||||
|
{
|
||||||
|
case df::workshop_type::Craftsdwarfs:
|
||||||
|
{
|
||||||
|
df::item_type jobitem = j->job_items[0]->item_type;
|
||||||
|
switch (jobitem)
|
||||||
|
{
|
||||||
|
case df::item_type::BOULDER:
|
||||||
|
return df::unit_labor::STONE_CRAFT;
|
||||||
|
case df::item_type::NONE:
|
||||||
|
if (j->material_category.bits.bone ||
|
||||||
|
j->material_category.bits.horn ||
|
||||||
|
j->material_category.bits.tooth ||
|
||||||
|
j->material_category.bits.shell)
|
||||||
|
return df::unit_labor::BONE_CARVE;
|
||||||
|
else
|
||||||
|
{
|
||||||
|
debug("LABORMANAGER: Cannot deduce labor for make crafts job (not bone)\n");
|
||||||
|
debug_pause();
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
case df::item_type::WOOD:
|
||||||
|
return df::unit_labor::WOOD_CRAFT;
|
||||||
|
case df::item_type::CLOTH:
|
||||||
|
return df::unit_labor::CLOTHESMAKER;
|
||||||
|
case df::item_type::SKIN_TANNED:
|
||||||
|
return df::unit_labor::LEATHER;
|
||||||
|
default:
|
||||||
|
debug("LABORMANAGER: Cannot deduce labor for make crafts job, item type %s\n",
|
||||||
|
ENUM_KEY_STR(item_type, jobitem).c_str());
|
||||||
|
debug_pause();
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case df::workshop_type::Masons:
|
||||||
|
return df::unit_labor::MASON;
|
||||||
|
case df::workshop_type::Carpenters:
|
||||||
|
return df::unit_labor::CARPENTER;
|
||||||
|
case df::workshop_type::Leatherworks:
|
||||||
|
return df::unit_labor::LEATHER;
|
||||||
|
case df::workshop_type::Clothiers:
|
||||||
|
return df::unit_labor::CLOTHESMAKER;
|
||||||
|
case df::workshop_type::Bowyers:
|
||||||
|
return df::unit_labor::BOWYER;
|
||||||
|
case df::workshop_type::MagmaForge:
|
||||||
|
case df::workshop_type::MetalsmithsForge:
|
||||||
|
return metaltype;
|
||||||
|
default:
|
||||||
|
debug("LABORMANAGER: Cannot deduce labor for make job, workshop type %s\n",
|
||||||
|
ENUM_KEY_STR(workshop_type, type).c_str());
|
||||||
|
debug_pause();
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (bld->getType() == df::building_type::Furnace)
|
||||||
|
{
|
||||||
|
df::furnace_type type = ((df::building_furnacest*)(bld))->type;
|
||||||
|
switch (type)
|
||||||
|
{
|
||||||
|
case df::furnace_type::MagmaGlassFurnace:
|
||||||
|
case df::furnace_type::GlassFurnace:
|
||||||
|
return df::unit_labor::GLASSMAKER;
|
||||||
|
default:
|
||||||
|
debug("LABORMANAGER: Cannot deduce labor for make job, furnace type %s\n",
|
||||||
|
ENUM_KEY_STR(furnace_type, type).c_str());
|
||||||
|
debug_pause();
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
debug("LABORMANAGER: Cannot deduce labor for make job, building type %s\n",
|
||||||
|
ENUM_KEY_STR(building_type, bld->getType()).c_str());
|
||||||
|
debug_pause();
|
||||||
|
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
|
||||||
|
jlfunc_make(df::unit_labor mt) : metaltype(mt) {}
|
||||||
|
};
|
||||||
|
|
||||||
|
class jlfunc_custom : public jlfunc
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
df::unit_labor get_labor(df::job* j)
|
||||||
|
{
|
||||||
|
for (auto r = world->raws.reactions.begin(); r != world->raws.reactions.end(); r++)
|
||||||
|
{
|
||||||
|
if ((*r)->code == j->reaction_name)
|
||||||
|
{
|
||||||
|
df::job_skill skill = (*r)->skill;
|
||||||
|
df::unit_labor labor = ENUM_ATTR(job_skill, labor, skill);
|
||||||
|
return labor;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
jlfunc_custom() {}
|
||||||
|
};
|
||||||
|
|
||||||
|
jlfunc* JobLaborMapper::jlf_const(df::unit_labor l) {
|
||||||
|
jlfunc* jlf;
|
||||||
|
if (jlf_cache.count(l) == 0)
|
||||||
|
{
|
||||||
|
jlf = new jlfunc_const(l);
|
||||||
|
jlf_cache[l] = jlf;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
jlf = jlf_cache[l];
|
||||||
|
|
||||||
|
return jlf;
|
||||||
|
}
|
||||||
|
|
||||||
|
JobLaborMapper::~JobLaborMapper()
|
||||||
|
{
|
||||||
|
std::set<jlfunc*> log;
|
||||||
|
|
||||||
|
for (auto i = jlf_cache.begin(); i != jlf_cache.end(); i++)
|
||||||
|
{
|
||||||
|
if (!log.count(i->second))
|
||||||
|
{
|
||||||
|
log.insert(i->second);
|
||||||
|
delete i->second;
|
||||||
|
}
|
||||||
|
i->second = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
FOR_ENUM_ITEMS(job_type, j)
|
||||||
|
{
|
||||||
|
if (j < 0)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
jlfunc* p = job_to_labor_table[j];
|
||||||
|
if (!log.count(p))
|
||||||
|
{
|
||||||
|
log.insert(p);
|
||||||
|
delete p;
|
||||||
|
}
|
||||||
|
job_to_labor_table[j] = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
JobLaborMapper::JobLaborMapper()
|
||||||
|
{
|
||||||
|
jlfunc* jlf_hauling = new jlfunc_hauling();
|
||||||
|
jlfunc* jlf_make_furniture = new jlfunc_make(df::unit_labor::FORGE_FURNITURE);
|
||||||
|
jlfunc* jlf_make_object = new jlfunc_make(df::unit_labor::METAL_CRAFT);
|
||||||
|
jlfunc* jlf_make_armor = new jlfunc_make(df::unit_labor::FORGE_ARMOR);
|
||||||
|
jlfunc* jlf_make_weapon = new jlfunc_make(df::unit_labor::FORGE_WEAPON);
|
||||||
|
|
||||||
|
jlfunc* jlf_no_labor = jlf_const(df::unit_labor::NONE);
|
||||||
|
|
||||||
|
job_to_labor_table[df::job_type::CarveFortification] = jlf_const(df::unit_labor::DETAIL);
|
||||||
|
job_to_labor_table[df::job_type::DetailWall] = jlf_const(df::unit_labor::DETAIL);
|
||||||
|
job_to_labor_table[df::job_type::DetailFloor] = jlf_const(df::unit_labor::DETAIL);
|
||||||
|
job_to_labor_table[df::job_type::Dig] = jlf_const(df::unit_labor::MINE);
|
||||||
|
job_to_labor_table[df::job_type::CarveUpwardStaircase] = jlf_const(df::unit_labor::MINE);
|
||||||
|
job_to_labor_table[df::job_type::CarveDownwardStaircase] = jlf_const(df::unit_labor::MINE);
|
||||||
|
job_to_labor_table[df::job_type::CarveUpDownStaircase] = jlf_const(df::unit_labor::MINE);
|
||||||
|
job_to_labor_table[df::job_type::CarveRamp] = jlf_const(df::unit_labor::MINE);
|
||||||
|
job_to_labor_table[df::job_type::DigChannel] = jlf_const(df::unit_labor::MINE);
|
||||||
|
job_to_labor_table[df::job_type::FellTree] = jlf_const(df::unit_labor::CUTWOOD);
|
||||||
|
job_to_labor_table[df::job_type::GatherPlants] = jlf_const(df::unit_labor::HERBALIST);
|
||||||
|
job_to_labor_table[df::job_type::RemoveConstruction] = jlf_const(df::unit_labor::REMOVE_CONSTRUCTION);
|
||||||
|
job_to_labor_table[df::job_type::CollectWebs] = jlf_const(df::unit_labor::WEAVER);
|
||||||
|
job_to_labor_table[df::job_type::BringItemToDepot] = jlf_const(df::unit_labor::HAUL_TRADE);
|
||||||
|
job_to_labor_table[df::job_type::BringItemToShop] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::Eat] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::GetProvisions] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::Drink] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::Drink2] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::FillWaterskin] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::FillWaterskin2] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::Sleep] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::CollectSand] = jlf_const(df::unit_labor::HAUL_ITEM);
|
||||||
|
job_to_labor_table[df::job_type::Fish] = jlf_const(df::unit_labor::FISH);
|
||||||
|
job_to_labor_table[df::job_type::Hunt] = jlf_const(df::unit_labor::HUNT);
|
||||||
|
job_to_labor_table[df::job_type::HuntVermin] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::Kidnap] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::BeatCriminal] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StartingFistFight] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::CollectTaxes] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::GuardTaxCollector] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::CatchLiveLandAnimal] = jlf_const(df::unit_labor::HUNT);
|
||||||
|
job_to_labor_table[df::job_type::CatchLiveFish] = jlf_const(df::unit_labor::FISH);
|
||||||
|
job_to_labor_table[df::job_type::ReturnKill] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::CheckChest] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StoreOwnedItem] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::PlaceItemInTomb] = jlf_const(df::unit_labor::HAUL_BODY);
|
||||||
|
job_to_labor_table[df::job_type::StoreItemInStockpile] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::StoreItemInBag] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::StoreItemInHospital] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::StoreWeapon] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::StoreArmor] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::StoreItemInBarrel] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::StoreItemInBin] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::SeekArtifact] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::SeekInfant] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::AttendParty] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::GoShopping] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::GoShopping2] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::Clean] = jlf_const(df::unit_labor::CLEAN);
|
||||||
|
job_to_labor_table[df::job_type::Rest] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::PickupEquipment] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::DumpItem] = jlf_const(df::unit_labor::HAUL_REFUSE);
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodCrafter] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodJeweller] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodForge] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodMagmaForge] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodBrooding] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodFell] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodCarpenter] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodMason] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodBowyer] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodTanner] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodWeaver] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodGlassmaker] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::StrangeMoodMechanics] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::ConstructBuilding] = new jlfunc_construct_bld();
|
||||||
|
job_to_labor_table[df::job_type::ConstructDoor] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructFloodgate] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructBed] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructThrone] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructCoffin] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructTable] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructChest] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructBin] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructArmorStand] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructWeaponRack] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructCabinet] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructStatue] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructBlocks] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::MakeRawGlass] = jlf_const(df::unit_labor::GLASSMAKER);
|
||||||
|
job_to_labor_table[df::job_type::MakeCrafts] = jlf_make_object; job_to_labor_table[df::job_type::MintCoins] = jlf_const(df::unit_labor::METAL_CRAFT);
|
||||||
|
job_to_labor_table[df::job_type::CutGems] = jlf_const(df::unit_labor::CUT_GEM);
|
||||||
|
job_to_labor_table[df::job_type::CutGlass] = jlf_const(df::unit_labor::CUT_GEM);
|
||||||
|
job_to_labor_table[df::job_type::EncrustWithGems] = jlf_const(df::unit_labor::ENCRUST_GEM);
|
||||||
|
job_to_labor_table[df::job_type::EncrustWithGlass] = jlf_const(df::unit_labor::ENCRUST_GEM);
|
||||||
|
job_to_labor_table[df::job_type::DestroyBuilding] = new jlfunc_destroy_bld();
|
||||||
|
job_to_labor_table[df::job_type::SmeltOre] = jlf_const(df::unit_labor::SMELT);
|
||||||
|
job_to_labor_table[df::job_type::MeltMetalObject] = jlf_const(df::unit_labor::SMELT);
|
||||||
|
job_to_labor_table[df::job_type::ExtractMetalStrands] = jlf_const(df::unit_labor::EXTRACT_STRAND);
|
||||||
|
job_to_labor_table[df::job_type::PlantSeeds] = jlf_const(df::unit_labor::PLANT);
|
||||||
|
job_to_labor_table[df::job_type::HarvestPlants] = jlf_const(df::unit_labor::PLANT);
|
||||||
|
job_to_labor_table[df::job_type::TrainHuntingAnimal] = jlf_const(df::unit_labor::ANIMALTRAIN);
|
||||||
|
job_to_labor_table[df::job_type::TrainWarAnimal] = jlf_const(df::unit_labor::ANIMALTRAIN);
|
||||||
|
job_to_labor_table[df::job_type::MakeWeapon] = jlf_make_weapon;
|
||||||
|
job_to_labor_table[df::job_type::ForgeAnvil] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructCatapultParts] = jlf_const(df::unit_labor::SIEGECRAFT);
|
||||||
|
job_to_labor_table[df::job_type::ConstructBallistaParts] = jlf_const(df::unit_labor::SIEGECRAFT);
|
||||||
|
job_to_labor_table[df::job_type::MakeArmor] = jlf_make_armor;
|
||||||
|
job_to_labor_table[df::job_type::MakeHelm] = jlf_make_armor;
|
||||||
|
job_to_labor_table[df::job_type::MakePants] = jlf_make_armor;
|
||||||
|
job_to_labor_table[df::job_type::StudWith] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::ButcherAnimal] = jlf_const(df::unit_labor::BUTCHER);
|
||||||
|
job_to_labor_table[df::job_type::PrepareRawFish] = jlf_const(df::unit_labor::CLEAN_FISH);
|
||||||
|
job_to_labor_table[df::job_type::MillPlants] = jlf_const(df::unit_labor::MILLER);
|
||||||
|
job_to_labor_table[df::job_type::BaitTrap] = jlf_const(df::unit_labor::TRAPPER);
|
||||||
|
job_to_labor_table[df::job_type::MilkCreature] = jlf_const(df::unit_labor::MILK);
|
||||||
|
job_to_labor_table[df::job_type::MakeCheese] = jlf_const(df::unit_labor::MAKE_CHEESE);
|
||||||
|
job_to_labor_table[df::job_type::ProcessPlants] = jlf_const(df::unit_labor::PROCESS_PLANT);
|
||||||
|
job_to_labor_table[df::job_type::ProcessPlantsVial] = jlf_const(df::unit_labor::PROCESS_PLANT);
|
||||||
|
job_to_labor_table[df::job_type::ProcessPlantsBarrel] = jlf_const(df::unit_labor::PROCESS_PLANT);
|
||||||
|
job_to_labor_table[df::job_type::PrepareMeal] = jlf_const(df::unit_labor::COOK);
|
||||||
|
job_to_labor_table[df::job_type::WeaveCloth] = jlf_const(df::unit_labor::WEAVER);
|
||||||
|
job_to_labor_table[df::job_type::MakeGloves] = jlf_make_armor;
|
||||||
|
job_to_labor_table[df::job_type::MakeShoes] = jlf_make_armor;
|
||||||
|
job_to_labor_table[df::job_type::MakeShield] = jlf_make_armor;
|
||||||
|
job_to_labor_table[df::job_type::MakeCage] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::MakeChain] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeFlask] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeGoblet] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeToy] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeAnimalTrap] = jlf_const(df::unit_labor::TRAPPER);
|
||||||
|
job_to_labor_table[df::job_type::MakeBarrel] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::MakeBucket] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::MakeWindow] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::MakeTotem] = jlf_const(df::unit_labor::BONE_CARVE);
|
||||||
|
job_to_labor_table[df::job_type::MakeAmmo] = jlf_make_weapon;
|
||||||
|
job_to_labor_table[df::job_type::DecorateWith] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeBackpack] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeQuiver] = jlf_make_armor;
|
||||||
|
job_to_labor_table[df::job_type::MakeBallistaArrowHead] = jlf_make_weapon;
|
||||||
|
job_to_labor_table[df::job_type::AssembleSiegeAmmo] = jlf_const(df::unit_labor::SIEGECRAFT);
|
||||||
|
job_to_labor_table[df::job_type::LoadCatapult] = jlf_const(df::unit_labor::SIEGEOPERATE);
|
||||||
|
job_to_labor_table[df::job_type::LoadBallista] = jlf_const(df::unit_labor::SIEGEOPERATE);
|
||||||
|
job_to_labor_table[df::job_type::FireCatapult] = jlf_const(df::unit_labor::SIEGEOPERATE);
|
||||||
|
job_to_labor_table[df::job_type::FireBallista] = jlf_const(df::unit_labor::SIEGEOPERATE);
|
||||||
|
job_to_labor_table[df::job_type::ConstructMechanisms] = jlf_const(df::unit_labor::MECHANIC);
|
||||||
|
job_to_labor_table[df::job_type::MakeTrapComponent] = jlf_make_weapon;
|
||||||
|
job_to_labor_table[df::job_type::LoadCageTrap] = jlf_const(df::unit_labor::MECHANIC);
|
||||||
|
job_to_labor_table[df::job_type::LoadStoneTrap] = jlf_const(df::unit_labor::MECHANIC);
|
||||||
|
job_to_labor_table[df::job_type::LoadWeaponTrap] = jlf_const(df::unit_labor::MECHANIC);
|
||||||
|
job_to_labor_table[df::job_type::CleanTrap] = jlf_const(df::unit_labor::MECHANIC);
|
||||||
|
job_to_labor_table[df::job_type::CastSpell] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::LinkBuildingToTrigger] = jlf_const(df::unit_labor::MECHANIC);
|
||||||
|
job_to_labor_table[df::job_type::PullLever] = jlf_const(df::unit_labor::PULL_LEVER);
|
||||||
|
job_to_labor_table[df::job_type::ExtractFromPlants] = jlf_const(df::unit_labor::HERBALIST);
|
||||||
|
job_to_labor_table[df::job_type::ExtractFromRawFish] = jlf_const(df::unit_labor::DISSECT_FISH);
|
||||||
|
job_to_labor_table[df::job_type::ExtractFromLandAnimal] = jlf_const(df::unit_labor::DISSECT_VERMIN);
|
||||||
|
job_to_labor_table[df::job_type::TameVermin] = jlf_const(df::unit_labor::ANIMALTRAIN);
|
||||||
|
job_to_labor_table[df::job_type::TameAnimal] = jlf_const(df::unit_labor::ANIMALTRAIN);
|
||||||
|
job_to_labor_table[df::job_type::ChainAnimal] = jlf_const(df::unit_labor::HAUL_ANIMALS);
|
||||||
|
job_to_labor_table[df::job_type::UnchainAnimal] = jlf_const(df::unit_labor::HAUL_ANIMALS);
|
||||||
|
job_to_labor_table[df::job_type::UnchainPet] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::ReleaseLargeCreature] = jlf_const(df::unit_labor::HAUL_ANIMALS);
|
||||||
|
job_to_labor_table[df::job_type::ReleasePet] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::ReleaseSmallCreature] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::HandleSmallCreature] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::HandleLargeCreature] = jlf_const(df::unit_labor::HAUL_ANIMALS);
|
||||||
|
job_to_labor_table[df::job_type::CageLargeCreature] = jlf_const(df::unit_labor::HAUL_ANIMALS);
|
||||||
|
job_to_labor_table[df::job_type::CageSmallCreature] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::RecoverWounded] = jlf_const(df::unit_labor::RECOVER_WOUNDED);
|
||||||
|
job_to_labor_table[df::job_type::DiagnosePatient] = jlf_const(df::unit_labor::DIAGNOSE);
|
||||||
|
job_to_labor_table[df::job_type::ImmobilizeBreak] = jlf_const(df::unit_labor::BONE_SETTING);
|
||||||
|
job_to_labor_table[df::job_type::DressWound] = jlf_const(df::unit_labor::DRESSING_WOUNDS);
|
||||||
|
job_to_labor_table[df::job_type::CleanPatient] = jlf_const(df::unit_labor::DRESSING_WOUNDS);
|
||||||
|
job_to_labor_table[df::job_type::Surgery] = jlf_const(df::unit_labor::SURGERY);
|
||||||
|
job_to_labor_table[df::job_type::Suture] = jlf_const(df::unit_labor::SUTURING);
|
||||||
|
job_to_labor_table[df::job_type::SetBone] = jlf_const(df::unit_labor::BONE_SETTING);
|
||||||
|
job_to_labor_table[df::job_type::PlaceInTraction] = jlf_const(df::unit_labor::BONE_SETTING);
|
||||||
|
job_to_labor_table[df::job_type::DrainAquarium] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::FillAquarium] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::FillPond] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::GiveWater] = jlf_const(df::unit_labor::FEED_WATER_CIVILIANS);
|
||||||
|
job_to_labor_table[df::job_type::GiveFood] = jlf_const(df::unit_labor::FEED_WATER_CIVILIANS);
|
||||||
|
job_to_labor_table[df::job_type::GiveWater2] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::GiveFood2] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::RecoverPet] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::PitLargeAnimal] = jlf_const(df::unit_labor::HAUL_ANIMALS);
|
||||||
|
job_to_labor_table[df::job_type::PitSmallAnimal] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::SlaughterAnimal] = jlf_const(df::unit_labor::BUTCHER);
|
||||||
|
job_to_labor_table[df::job_type::MakeCharcoal] = jlf_const(df::unit_labor::BURN_WOOD);
|
||||||
|
job_to_labor_table[df::job_type::MakeAsh] = jlf_const(df::unit_labor::BURN_WOOD);
|
||||||
|
job_to_labor_table[df::job_type::MakeLye] = jlf_const(df::unit_labor::LYE_MAKING);
|
||||||
|
job_to_labor_table[df::job_type::MakePotashFromLye] = jlf_const(df::unit_labor::POTASH_MAKING);
|
||||||
|
job_to_labor_table[df::job_type::FertilizeField] = jlf_const(df::unit_labor::PLANT);
|
||||||
|
job_to_labor_table[df::job_type::MakePotashFromAsh] = jlf_const(df::unit_labor::POTASH_MAKING);
|
||||||
|
job_to_labor_table[df::job_type::DyeThread] = jlf_const(df::unit_labor::DYER);
|
||||||
|
job_to_labor_table[df::job_type::DyeCloth] = jlf_const(df::unit_labor::DYER);
|
||||||
|
job_to_labor_table[df::job_type::SewImage] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakePipeSection] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::OperatePump] = jlf_const(df::unit_labor::OPERATE_PUMP);
|
||||||
|
job_to_labor_table[df::job_type::ManageWorkOrders] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::UpdateStockpileRecords] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::TradeAtDepot] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::ConstructHatchCover] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructGrate] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::RemoveStairs] = jlf_const(df::unit_labor::MINE);
|
||||||
|
job_to_labor_table[df::job_type::ConstructQuern] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructMillstone] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructSplint] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructCrutch] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::ConstructTractionBench] = jlf_const(df::unit_labor::MECHANIC);
|
||||||
|
job_to_labor_table[df::job_type::CleanSelf] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::BringCrutch] = jlf_const(df::unit_labor::BONE_SETTING);
|
||||||
|
job_to_labor_table[df::job_type::ApplyCast] = jlf_const(df::unit_labor::BONE_SETTING);
|
||||||
|
job_to_labor_table[df::job_type::CustomReaction] = new jlfunc_custom();
|
||||||
|
job_to_labor_table[df::job_type::ConstructSlab] = jlf_make_furniture;
|
||||||
|
job_to_labor_table[df::job_type::EngraveSlab] = jlf_const(df::unit_labor::DETAIL);
|
||||||
|
job_to_labor_table[df::job_type::ShearCreature] = jlf_const(df::unit_labor::SHEARER);
|
||||||
|
job_to_labor_table[df::job_type::SpinThread] = jlf_const(df::unit_labor::SPINNER);
|
||||||
|
job_to_labor_table[df::job_type::PenLargeAnimal] = jlf_const(df::unit_labor::HAUL_ANIMALS);
|
||||||
|
job_to_labor_table[df::job_type::PenSmallAnimal] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::MakeTool] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::CollectClay] = jlf_const(df::unit_labor::POTTERY);
|
||||||
|
job_to_labor_table[df::job_type::InstallColonyInHive] = jlf_const(df::unit_labor::BEEKEEPING);
|
||||||
|
job_to_labor_table[df::job_type::CollectHiveProducts] = jlf_const(df::unit_labor::BEEKEEPING);
|
||||||
|
job_to_labor_table[df::job_type::CauseTrouble] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::DrinkBlood] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::ReportCrime] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::ExecuteCriminal] = jlf_no_labor;
|
||||||
|
job_to_labor_table[df::job_type::TrainAnimal] = jlf_const(df::unit_labor::ANIMALTRAIN);
|
||||||
|
job_to_labor_table[df::job_type::CarveTrack] = jlf_const(df::unit_labor::DETAIL);
|
||||||
|
job_to_labor_table[df::job_type::PushTrackVehicle] = jlf_const(df::unit_labor::HANDLE_VEHICLES);
|
||||||
|
job_to_labor_table[df::job_type::PlaceTrackVehicle] = jlf_const(df::unit_labor::HANDLE_VEHICLES);
|
||||||
|
job_to_labor_table[df::job_type::StoreItemInVehicle] = jlf_hauling;
|
||||||
|
job_to_labor_table[df::job_type::GeldAnimal] = jlf_const(df::unit_labor::GELD);
|
||||||
|
job_to_labor_table[df::job_type::MakeFigurine] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeAmulet] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeScepter] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeCrown] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeRing] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeEarring] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeBracelet] = jlf_make_object;
|
||||||
|
job_to_labor_table[df::job_type::MakeGem] = jlf_make_object;
|
||||||
|
|
||||||
|
job_to_labor_table[df::job_type::StoreItemInLocation] = jlf_no_labor; // StoreItemInLocation
|
||||||
|
};
|
||||||
|
|
||||||
|
df::unit_labor JobLaborMapper::find_job_labor(df::job* j)
|
||||||
|
{
|
||||||
|
if (j->job_type == df::job_type::CustomReaction)
|
||||||
|
{
|
||||||
|
for (auto r = world->raws.reactions.begin(); r != world->raws.reactions.end(); r++)
|
||||||
|
{
|
||||||
|
if ((*r)->code == j->reaction_name)
|
||||||
|
{
|
||||||
|
df::job_skill skill = (*r)->skill;
|
||||||
|
return ENUM_ATTR(job_skill, labor, skill);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
df::unit_labor labor;
|
||||||
|
if (job_to_labor_table.count(j->job_type) == 0)
|
||||||
|
{
|
||||||
|
debug("LABORMANAGER: job has no job to labor table entry: %s (%d)\n", ENUM_KEY_STR(job_type, j->job_type).c_str(), j->job_type);
|
||||||
|
debug_pause();
|
||||||
|
labor = df::unit_labor::NONE;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
|
||||||
|
labor = job_to_labor_table[j->job_type]->get_labor(j);
|
||||||
|
}
|
||||||
|
|
||||||
|
return labor;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* End of labor deducer */
|
||||||
@ -0,0 +1,32 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#include <Core.h>
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
using namespace df::enums;
|
||||||
|
|
||||||
|
#include "df/job.h"
|
||||||
|
#include "df/job_type.h"
|
||||||
|
#include "df/unit_labor.h"
|
||||||
|
|
||||||
|
|
||||||
|
class jlfunc;
|
||||||
|
|
||||||
|
class JobLaborMapper {
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::map<df::job_type, jlfunc*> job_to_labor_table;
|
||||||
|
std::map<df::unit_labor, jlfunc*> jlf_cache;
|
||||||
|
|
||||||
|
jlfunc* jlf_const(df::unit_labor l);
|
||||||
|
|
||||||
|
public:
|
||||||
|
~JobLaborMapper();
|
||||||
|
JobLaborMapper();
|
||||||
|
|
||||||
|
df::unit_labor find_job_labor(df::job* j);
|
||||||
|
|
||||||
|
|
||||||
|
};
|
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,4 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
void debug(const char* fmt, ...);
|
||||||
|
void debug_pause();
|
||||||
Loading…
Reference in New Issue