297 lines
7.3 KiB
C++
297 lines
7.3 KiB
C++
#include "uicommon.h"
|
|
|
|
#include "modules/Gui.h"
|
|
|
|
#include "df/world.h"
|
|
#include "df/world_raws.h"
|
|
#include "df/building_def.h"
|
|
#include "df/viewscreen_dwarfmodest.h"
|
|
#include "df/building_stockpilest.h"
|
|
#include "modules/Buildings.h"
|
|
#include "modules/Items.h"
|
|
#include "df/ui.h"
|
|
#include "modules/Maps.h"
|
|
#include "modules/World.h"
|
|
#include "df/item_quality.h"
|
|
|
|
using df::global::world;
|
|
using df::global::cursor;
|
|
using df::global::ui;
|
|
using df::building_stockpilest;
|
|
|
|
DFHACK_PLUGIN("automelt");
|
|
#define PLUGIN_VERSION 0.3
|
|
|
|
static const string PERSISTENCE_KEY = "automelt/stockpiles";
|
|
|
|
static void mark_all_in_stockpiles(vector<PersistentStockpileInfo> &stockpiles)
|
|
{
|
|
// Precompute a bitmask with the bad flags
|
|
df::item_flags bad_flags;
|
|
bad_flags.whole = 0;
|
|
|
|
#define F(x) bad_flags.bits.x = true;
|
|
F(dump); F(forbid); F(garbage_collect);
|
|
F(hostile); F(on_fire); F(rotten); F(trader);
|
|
F(in_building); F(construction); F(artifact);
|
|
F(spider_web); F(owned); F(in_job);
|
|
#undef F
|
|
|
|
size_t marked_count = 0;
|
|
auto &melting_items = world->items.other[items_other_id::ANY_MELT_DESIGNATED];
|
|
for (auto it = stockpiles.begin(); it != stockpiles.end(); it++)
|
|
{
|
|
if (!it->isValid())
|
|
continue;
|
|
|
|
Buildings::StockpileIterator stored;
|
|
for (stored.begin(it->getStockpile()); !stored.done(); ++stored)
|
|
{
|
|
df::item *item = *stored;
|
|
if (item->flags.whole & bad_flags.whole)
|
|
continue;
|
|
|
|
if (!can_melt(item))
|
|
continue;
|
|
|
|
if (is_set_to_melt(item))
|
|
continue;
|
|
|
|
++marked_count;
|
|
insert_into_vector(melting_items, &df::item::id, item);
|
|
item->flags.bits.melt = true;
|
|
}
|
|
}
|
|
|
|
if (marked_count)
|
|
Gui::showAnnouncement("Marked " + int_to_string(marked_count) + " items to melt", COLOR_GREEN, false);
|
|
}
|
|
|
|
/*
|
|
* Stockpile Monitoring
|
|
*/
|
|
|
|
class StockpileMonitor
|
|
{
|
|
public:
|
|
bool isMonitored(df::building_stockpilest *sp)
|
|
{
|
|
for (auto it = monitored_stockpiles.begin(); it != monitored_stockpiles.end(); it++)
|
|
{
|
|
if (it->matches(sp))
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void add(df::building_stockpilest *sp)
|
|
{
|
|
auto pile = PersistentStockpileInfo(sp, PERSISTENCE_KEY);
|
|
if (pile.isValid())
|
|
{
|
|
monitored_stockpiles.push_back(PersistentStockpileInfo(pile));
|
|
monitored_stockpiles.back().save();
|
|
}
|
|
}
|
|
|
|
void remove(df::building_stockpilest *sp)
|
|
{
|
|
for (auto it = monitored_stockpiles.begin(); it != monitored_stockpiles.end(); it++)
|
|
{
|
|
if (it->matches(sp))
|
|
{
|
|
it->remove();
|
|
monitored_stockpiles.erase(it);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void doCycle()
|
|
{
|
|
for (auto it = monitored_stockpiles.begin(); it != monitored_stockpiles.end();)
|
|
{
|
|
if (!it->isValid())
|
|
it = monitored_stockpiles.erase(it);
|
|
else
|
|
++it;
|
|
}
|
|
|
|
mark_all_in_stockpiles(monitored_stockpiles);
|
|
}
|
|
|
|
void reset()
|
|
{
|
|
monitored_stockpiles.clear();
|
|
std::vector<PersistentDataItem> items;
|
|
DFHack::World::GetPersistentData(&items, PERSISTENCE_KEY);
|
|
|
|
for (auto i = items.begin(); i != items.end(); i++)
|
|
{
|
|
auto pile = PersistentStockpileInfo(*i, PERSISTENCE_KEY);
|
|
if (pile.load())
|
|
monitored_stockpiles.push_back(PersistentStockpileInfo(pile));
|
|
else
|
|
pile.remove();
|
|
}
|
|
}
|
|
|
|
|
|
private:
|
|
vector<PersistentStockpileInfo> monitored_stockpiles;
|
|
};
|
|
|
|
static StockpileMonitor monitor;
|
|
|
|
#define DELTA_TICKS 610
|
|
|
|
DFhackCExport command_result plugin_onupdate ( color_ostream &out )
|
|
{
|
|
if(!Maps::IsValid())
|
|
return CR_OK;
|
|
|
|
static decltype(world->frame_counter) last_frame_count = 0;
|
|
|
|
if (DFHack::World::ReadPauseState())
|
|
return CR_OK;
|
|
|
|
if (world->frame_counter - last_frame_count < DELTA_TICKS)
|
|
return CR_OK;
|
|
|
|
last_frame_count = world->frame_counter;
|
|
|
|
monitor.doCycle();
|
|
|
|
return CR_OK;
|
|
}
|
|
|
|
|
|
/*
|
|
* Interface
|
|
*/
|
|
|
|
struct melt_hook : public df::viewscreen_dwarfmodest
|
|
{
|
|
typedef df::viewscreen_dwarfmodest interpose_base;
|
|
|
|
bool handleInput(set<df::interface_key> *input)
|
|
{
|
|
building_stockpilest *sp = get_selected_stockpile();
|
|
if (!sp)
|
|
return false;
|
|
|
|
if (input->count(interface_key::CUSTOM_SHIFT_M))
|
|
{
|
|
if (monitor.isMonitored(sp))
|
|
monitor.remove(sp);
|
|
else
|
|
monitor.add(sp);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
DEFINE_VMETHOD_INTERPOSE(void, feed, (set<df::interface_key> *input))
|
|
{
|
|
if (!handleInput(input))
|
|
INTERPOSE_NEXT(feed)(input);
|
|
}
|
|
|
|
DEFINE_VMETHOD_INTERPOSE(void, render, ())
|
|
{
|
|
INTERPOSE_NEXT(render)();
|
|
|
|
building_stockpilest *sp = get_selected_stockpile();
|
|
if (!sp)
|
|
return;
|
|
|
|
auto dims = Gui::getDwarfmodeViewDims();
|
|
int left_margin = dims.menu_x1 + 1;
|
|
int x = left_margin;
|
|
int y = dims.y2 - 6;
|
|
|
|
int links = 0;
|
|
links += sp->links.give_to_pile.size();
|
|
links += sp->links.take_from_pile.size();
|
|
links += sp->links.give_to_workshop.size();
|
|
links += sp->links.take_from_workshop.size();
|
|
bool state = monitor.isMonitored(sp);
|
|
|
|
if (links + 12 >= y) {
|
|
y = dims.y2;
|
|
OutputString(COLOR_WHITE, x, y, "Auto: ");
|
|
OutputString(COLOR_LIGHTRED, x, y, "M");
|
|
OutputString(state? COLOR_LIGHTGREEN: COLOR_GREY, x, y, "elt ");
|
|
} else {
|
|
OutputToggleString(x, y, "Auto melt", "M", state, true, left_margin, COLOR_WHITE, COLOR_LIGHTRED);
|
|
}
|
|
}
|
|
};
|
|
|
|
IMPLEMENT_VMETHOD_INTERPOSE(melt_hook, feed);
|
|
IMPLEMENT_VMETHOD_INTERPOSE(melt_hook, render);
|
|
|
|
|
|
static command_result automelt_cmd(color_ostream &out, vector <string> & parameters)
|
|
{
|
|
if (!parameters.empty())
|
|
{
|
|
if (parameters.size() == 1 && toLower(parameters[0])[0] == 'v')
|
|
{
|
|
out << "Automelt" << endl << "Version: " << PLUGIN_VERSION << endl;
|
|
}
|
|
}
|
|
|
|
return CR_OK;
|
|
}
|
|
|
|
DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_change_event event)
|
|
{
|
|
switch (event)
|
|
{
|
|
case DFHack::SC_MAP_LOADED:
|
|
monitor.reset();
|
|
break;
|
|
case DFHack::SC_MAP_UNLOADED:
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return CR_OK;
|
|
}
|
|
|
|
DFHACK_PLUGIN_IS_ENABLED(is_enabled);
|
|
|
|
DFhackCExport command_result plugin_enable(color_ostream &out, bool enable)
|
|
{
|
|
if (!gps)
|
|
return CR_FAILURE;
|
|
|
|
if (enable != is_enabled)
|
|
{
|
|
if (!INTERPOSE_HOOK(melt_hook, feed).apply(enable) ||
|
|
!INTERPOSE_HOOK(melt_hook, render).apply(enable))
|
|
return CR_FAILURE;
|
|
|
|
is_enabled = enable;
|
|
}
|
|
|
|
return CR_OK;
|
|
}
|
|
|
|
DFhackCExport command_result plugin_init ( color_ostream &out, std::vector <PluginCommand> &commands)
|
|
{
|
|
commands.push_back(
|
|
PluginCommand(
|
|
"automelt", "Automatically flag metal items in marked stockpiles for melting.",
|
|
automelt_cmd, false, ""));
|
|
|
|
return CR_OK;
|
|
}
|
|
|
|
DFhackCExport command_result plugin_shutdown ( color_ostream &out )
|
|
{
|
|
return CR_OK;
|
|
}
|