dfhack/plugins/showmood.cpp

284 lines
10 KiB
C++

// Show details of currently active strange mood, if any
#include "Core.h"
#include "Console.h"
#include "Export.h"
#include "PluginManager.h"
#include "modules/Materials.h"
#include "modules/Translation.h"
#include "modules/Items.h"
#include "DataDefs.h"
#include "df/world.h"
#include "df/job.h"
#include "df/job_item.h"
#include "df/general_ref.h"
#include "df/builtin_mats.h"
#include "df/inorganic_raw.h"
#include "df/matter_state.h"
#include "df/unit.h"
#include "df/building.h"
using std::string;
using std::vector;
using namespace DFHack;
using namespace df::enums;
using df::global::world;
command_result df_showmood (color_ostream &out, vector <string> & parameters)
{
if (!parameters.empty())
return CR_WRONG_USAGE;
if (!Translation::IsValid())
{
out.printerr("Translation data unavailable!\n");
return CR_FAILURE;
}
CoreSuspender suspend;
bool found = false;
for (df::job_list_link *cur = world->job_list.next; cur != NULL; cur = cur->next)
{
df::job *job = cur->item;
if ((job->job_type < job_type::StrangeMoodCrafter) || (job->job_type > job_type::StrangeMoodMechanics))
continue;
found = true;
df::unit *unit = NULL;
df::building *building = NULL;
for (size_t i = 0; i < job->references.size(); i++)
{
df::general_ref *ref = job->references[i];
if (ref->getType() == general_ref_type::UNIT_WORKER)
unit = ref->getUnit();
if (ref->getType() == general_ref_type::BUILDING_HOLDER)
building = ref->getBuilding();
}
if (!unit)
{
out.printerr("Found strange mood not attached to any dwarf!\n");
continue;
}
if (unit->mood == mood_type::None)
{
out.printerr("Dwarf with strange mood does not have a mood type!\n");
continue;
}
out.print("%s is currently ", Translation::TranslateName(&unit->name, false).c_str());
switch (unit->mood)
{
case mood_type::Macabre:
out.print("in a macabre mood");
if (job->job_type != job_type::StrangeMoodBrooding)
out.print(" (but isn't actually in a macabre mood?)");
break;
case mood_type::Fell:
out.print("in a fell mood");
if (job->job_type != job_type::StrangeMoodFell)
out.print(" (but isn't actually in a fell mood?)");
break;
case mood_type::Fey:
case mood_type::Secretive:
case mood_type::Possessed:
switch (unit->mood)
{
case mood_type::Fey:
out.print("in a fey mood");
break;
case mood_type::Secretive:
out.print("in a secretive mood");
break;
case mood_type::Possessed:
out.print("possessed");
break;
}
out.print(" with intent to ");
switch (job->job_type)
{
case job_type::StrangeMoodCrafter:
out.print("become a Craftsdwarf (or Engraver)");
break;
case job_type::StrangeMoodJeweller:
out.print("become a Jeweler");
break;
case job_type::StrangeMoodForge:
out.print("become a Metalworker");
break;
case job_type::StrangeMoodMagmaForge:
out.print("become a Metalworker using a Magma Forge");
break;
case job_type::StrangeMoodCarpenter:
out.print("become a Carpenter");
break;
case job_type::StrangeMoodMason:
out.print("become a Mason (or Miner)");
break;
case job_type::StrangeMoodBowyer:
out.print("become a Bowyer");
break;
case job_type::StrangeMoodTanner:
out.print("become a Leatherworker (or Tanner)");
break;
case job_type::StrangeMoodWeaver:
out.print("become a Clothier (or Weaver)");
break;
case job_type::StrangeMoodGlassmaker:
out.print("become a Glassmaker");
break;
case job_type::StrangeMoodMechanics:
out.print("become a Mechanic");
break;
case job_type::StrangeMoodBrooding:
out.print("enter a macabre mood?");
break;
case job_type::StrangeMoodFell:
out.print("enter a fell mood?");
break;
default:
out.print("do something else...");
break;
}
break;
default:
out.print("insane?");
break;
}
if (building)
{
string name;
building->getName(&name);
out.print(" and has claimed a %s\n", name.c_str());
}
else
out.print(" and has not yet claimed a workshop\n");
for (size_t i = 0; i < job->job_items.size(); i++)
{
df::job_item *item = job->job_items[i];
out.print("Item %i: ", i + 1);
MaterialInfo matinfo(item->mat_type, item->mat_index);
string mat_name = matinfo.toString();
switch (item->item_type)
{
case item_type::BOULDER:
out.print("%s boulder", mat_name.c_str());
break;
case item_type::BLOCKS:
out.print("%s blocks", mat_name.c_str());
break;
case item_type::WOOD:
out.print("%s logs", mat_name.c_str());
break;
case item_type::BAR:
if (matinfo.isInorganicWildcard())
mat_name = "metal";
if (matinfo.inorganic && matinfo.inorganic->flags.is_set(inorganic_flags::WAFERS))
out.print("%s wafers", mat_name.c_str());
else
out.print("%s bars", mat_name.c_str());
break;
case item_type::SMALLGEM:
out.print("%s cut gems", mat_name.c_str());
break;
case item_type::ROUGH:
if (matinfo.isAnyInorganic())
{
if (matinfo.isInorganicWildcard())
mat_name = "any";
out.print("%s rough gems", mat_name.c_str());
}
else
out.print("raw %s", mat_name.c_str());
break;
case item_type::SKIN_TANNED:
out.print("%s leather", mat_name.c_str());
break;
case item_type::CLOTH:
if (matinfo.isNone())
{
if (item->flags2.bits.plant)
mat_name = "any plant fiber";
else if (item->flags2.bits.silk)
mat_name = "any silk";
else if (item->flags2.bits.yarn)
mat_name = "any yarn";
}
out.print("%s cloth", mat_name.c_str());
break;
case item_type::REMAINS:
out.print("%s remains", mat_name.c_str());
break;
case item_type::CORPSE:
out.print("%s %scorpse", mat_name.c_str(), (item->flags1.bits.murdered ? "murdered " : ""));
break;
case item_type::NONE:
if (item->flags2.bits.body_part)
{
if (item->flags2.bits.bone)
out.print("%s bones", mat_name.c_str());
else if (item->flags2.bits.shell)
out.print("%s shells", mat_name.c_str());
else if (item->flags2.bits.horn)
out.print("%s horns", mat_name.c_str());
else if (item->flags2.bits.pearl)
out.print("%s pearls", mat_name.c_str());
else if (item->flags2.bits.ivory_tooth)
out.print("%s ivory/teeth", mat_name.c_str());
else
out.print("%s unknown body parts (%s:%s:%s)",
mat_name.c_str(),
bitfield_to_string(item->flags1).c_str(),
bitfield_to_string(item->flags2).c_str(),
bitfield_to_string(item->flags3).c_str());
}
else
out.print("indeterminate %s item (%s:%s:%s)",
mat_name.c_str(),
bitfield_to_string(item->flags1).c_str(),
bitfield_to_string(item->flags2).c_str(),
bitfield_to_string(item->flags3).c_str());
break;
default:
{
ItemTypeInfo itinfo(item->item_type, item->item_subtype);
out.print("item %s material %s flags (%s:%s:%s)",
itinfo.toString().c_str(), mat_name.c_str(),
bitfield_to_string(item->flags1).c_str(),
bitfield_to_string(item->flags2).c_str(),
bitfield_to_string(item->flags3).c_str());
break;
}
}
out.print(", quantity %i\n", item->quantity);
}
}
if (!found)
out.print("No strange moods currently active.\n");
return CR_OK;
}
DFHACK_PLUGIN("showmood");
DFhackCExport command_result plugin_init (color_ostream &out, std::vector<PluginCommand> &commands)
{
commands.clear();
commands.push_back(PluginCommand("showmood", "Shows items needed for current strange mood.", df_showmood));
return CR_OK;
}
DFhackCExport command_result plugin_shutdown ( color_ostream &out )
{
return CR_OK;
}