labormanager: Refactor source, separating the job labor mapper into a separate source file.
Kelly Kinkade
parent
9ba459f21e
commit
15ae72edec
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,4 @@
|
||||
#pragma once
|
||||
|
||||
void debug(const char* fmt, ...);
|
||||
void debug_pause();
|
||||
@ -0,0 +1,919 @@
|
||||
/*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 "labormanager_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,21 @@
|
||||
#pragma once
|
||||
|
||||
|
||||
class jlfunc;
|
||||
|
||||
class JobLaborMapper {
|
||||
|
||||
private:
|
||||
std::map<df::job_type, jlfunc*> job_to_labor_table;
|
||||
std::map<df::unit_labor, jlfunc*> jlf_cache;
|
||||
|
||||
jlfunc* JobLaborMapper::jlf_const(df::unit_labor l);
|
||||
|
||||
public:
|
||||
~JobLaborMapper();
|
||||
JobLaborMapper();
|
||||
|
||||
df::unit_labor find_job_labor(df::job* j);
|
||||
|
||||
|
||||
};
|
||||
Loading…
Reference in New Issue