Merge branch 'master' of https://github.com/angavrilov/dfhack into experimental-dontmerge
Conflicts: dfhack.init-exampledevelop
Warmist
commit
0db873493a
@ -0,0 +1,11 @@
|
||||
local _ENV = mkmodule('plugins.power-meter')
|
||||
|
||||
--[[
|
||||
|
||||
Native functions:
|
||||
|
||||
* makePowerMeter(plate_info,min_power,max_power,invert)
|
||||
|
||||
--]]
|
||||
|
||||
return _ENV
|
||||
@ -0,0 +1,237 @@
|
||||
#include "Core.h"
|
||||
#include <Console.h>
|
||||
#include <Export.h>
|
||||
#include <Error.h>
|
||||
#include <PluginManager.h>
|
||||
#include <modules/Gui.h>
|
||||
#include <modules/Screen.h>
|
||||
#include <modules/Maps.h>
|
||||
#include <modules/World.h>
|
||||
#include <TileTypes.h>
|
||||
#include <vector>
|
||||
#include <cstdio>
|
||||
#include <stack>
|
||||
#include <string>
|
||||
#include <cmath>
|
||||
#include <string.h>
|
||||
|
||||
#include <VTableInterpose.h>
|
||||
#include "df/graphic.h"
|
||||
#include "df/building_trapst.h"
|
||||
#include "df/builtin_mats.h"
|
||||
#include "df/world.h"
|
||||
#include "df/buildings_other_id.h"
|
||||
#include "df/machine.h"
|
||||
#include "df/machine_info.h"
|
||||
#include "df/building_drawbuffer.h"
|
||||
#include "df/ui.h"
|
||||
#include "df/viewscreen_dwarfmodest.h"
|
||||
#include "df/ui_build_selector.h"
|
||||
#include "df/flow_info.h"
|
||||
#include "df/report.h"
|
||||
|
||||
#include "MiscUtils.h"
|
||||
|
||||
using std::vector;
|
||||
using std::string;
|
||||
using std::stack;
|
||||
using namespace DFHack;
|
||||
using namespace df::enums;
|
||||
|
||||
using df::global::gps;
|
||||
using df::global::world;
|
||||
using df::global::ui;
|
||||
using df::global::ui_build_selector;
|
||||
|
||||
DFHACK_PLUGIN("power-meter");
|
||||
|
||||
static const uint32_t METER_BIT = 0x80000000U;
|
||||
|
||||
static void init_plate_info(df::pressure_plate_info &plate_info)
|
||||
{
|
||||
plate_info.water_min = 1;
|
||||
plate_info.water_max = 7;
|
||||
plate_info.flags.whole = METER_BIT;
|
||||
plate_info.flags.bits.water = true;
|
||||
plate_info.flags.bits.resets = true;
|
||||
}
|
||||
|
||||
/*
|
||||
* Hook for the pressure plate itself. Implements core logic.
|
||||
*/
|
||||
|
||||
struct trap_hook : df::building_trapst {
|
||||
typedef df::building_trapst interpose_base;
|
||||
|
||||
// Engine detection
|
||||
|
||||
bool is_power_meter()
|
||||
{
|
||||
return trap_type == trap_type::PressurePlate &&
|
||||
(plate_info.flags.whole & METER_BIT) != 0;
|
||||
}
|
||||
|
||||
inline bool is_fully_built()
|
||||
{
|
||||
return getBuildStage() >= getMaxBuildStage();
|
||||
}
|
||||
|
||||
DEFINE_VMETHOD_INTERPOSE(void, getName, (std::string *buf))
|
||||
{
|
||||
if (is_power_meter())
|
||||
{
|
||||
buf->clear();
|
||||
*buf += "Power Meter";
|
||||
return;
|
||||
}
|
||||
|
||||
INTERPOSE_NEXT(getName)(buf);
|
||||
}
|
||||
|
||||
DEFINE_VMETHOD_INTERPOSE(void, updateAction, ())
|
||||
{
|
||||
if (is_power_meter())
|
||||
{
|
||||
auto pdsgn = Maps::getTileDesignation(centerx,centery,z);
|
||||
|
||||
if (pdsgn)
|
||||
{
|
||||
bool active = false;
|
||||
auto &gears = world->buildings.other[buildings_other_id::GEAR_ASSEMBLY];
|
||||
|
||||
for (size_t i = 0; i < gears.size(); i++)
|
||||
{
|
||||
// Adjacent
|
||||
auto gear = gears[i];
|
||||
int deltaxy = abs(centerx - gear->centerx) + abs(centery - gear->centery);
|
||||
if (gear->z != z || deltaxy != 1)
|
||||
continue;
|
||||
// Linked to machine
|
||||
auto info = gears[i]->getMachineInfo();
|
||||
if (!info || info->machine_id < 0)
|
||||
continue;
|
||||
// an active machine
|
||||
auto machine = df::machine::find(info->machine_id);
|
||||
if (!machine || !machine->flags.bits.active)
|
||||
continue;
|
||||
// with adequate power?
|
||||
int power = machine->cur_power - machine->min_power;
|
||||
if (power < 0 || machine->cur_power <= 0)
|
||||
continue;
|
||||
if (power < plate_info.track_min)
|
||||
continue;
|
||||
if (power > plate_info.track_max && plate_info.track_max >= 0)
|
||||
continue;
|
||||
|
||||
active = true;
|
||||
break;
|
||||
}
|
||||
|
||||
if (plate_info.flags.bits.citizens)
|
||||
active = !active;
|
||||
|
||||
// Temporarily set the tile water amount based on power state
|
||||
auto old_dsgn = *pdsgn;
|
||||
pdsgn->bits.liquid_type = tile_liquid::Water;
|
||||
pdsgn->bits.flow_size = (active ? 7 : 0);
|
||||
|
||||
INTERPOSE_NEXT(updateAction)();
|
||||
|
||||
*pdsgn = old_dsgn;
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
INTERPOSE_NEXT(updateAction)();
|
||||
}
|
||||
|
||||
DEFINE_VMETHOD_INTERPOSE(void, drawBuilding, (df::building_drawbuffer *db, void *unk))
|
||||
{
|
||||
INTERPOSE_NEXT(drawBuilding)(db, unk);
|
||||
|
||||
if (is_power_meter() && is_fully_built())
|
||||
{
|
||||
db->fore[0][0] = 3;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
IMPLEMENT_VMETHOD_INTERPOSE(trap_hook, getName);
|
||||
IMPLEMENT_VMETHOD_INTERPOSE(trap_hook, updateAction);
|
||||
IMPLEMENT_VMETHOD_INTERPOSE(trap_hook, drawBuilding);
|
||||
|
||||
static bool enabled = false;
|
||||
|
||||
static void enable_hooks(bool enable)
|
||||
{
|
||||
enabled = enable;
|
||||
|
||||
INTERPOSE_HOOK(trap_hook, getName).apply(enable);
|
||||
INTERPOSE_HOOK(trap_hook, updateAction).apply(enable);
|
||||
INTERPOSE_HOOK(trap_hook, drawBuilding).apply(enable);
|
||||
}
|
||||
|
||||
static bool makePowerMeter(df::pressure_plate_info *info, int min_power, int max_power, bool invert)
|
||||
{
|
||||
CHECK_NULL_POINTER(info);
|
||||
|
||||
if (!enabled)
|
||||
{
|
||||
auto pworld = Core::getInstance().getWorld();
|
||||
auto entry = pworld->GetPersistentData("power-meter/enabled", NULL);
|
||||
if (!entry.isValid())
|
||||
return false;
|
||||
|
||||
enable_hooks(true);
|
||||
}
|
||||
|
||||
init_plate_info(*info);
|
||||
info->track_min = min_power;
|
||||
info->track_max = max_power;
|
||||
info->flags.bits.citizens = invert;
|
||||
return true;
|
||||
}
|
||||
|
||||
DFHACK_PLUGIN_LUA_FUNCTIONS {
|
||||
DFHACK_LUA_FUNCTION(makePowerMeter),
|
||||
DFHACK_LUA_END
|
||||
};
|
||||
|
||||
DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_change_event event)
|
||||
{
|
||||
switch (event) {
|
||||
case SC_MAP_LOADED:
|
||||
{
|
||||
auto pworld = Core::getInstance().getWorld();
|
||||
bool enable = pworld->GetPersistentData("power-meter/enabled").isValid();
|
||||
|
||||
if (enable)
|
||||
{
|
||||
out.print("Enabling the power meter plugin.\n");
|
||||
enable_hooks(true);
|
||||
}
|
||||
}
|
||||
break;
|
||||
case SC_MAP_UNLOADED:
|
||||
enable_hooks(false);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return CR_OK;
|
||||
}
|
||||
|
||||
DFhackCExport command_result plugin_init ( color_ostream &out, std::vector <PluginCommand> &commands)
|
||||
{
|
||||
if (Core::getInstance().isMapLoaded())
|
||||
plugin_onstatechange(out, SC_MAP_LOADED);
|
||||
|
||||
return CR_OK;
|
||||
}
|
||||
|
||||
DFhackCExport command_result plugin_shutdown ( color_ostream &out )
|
||||
{
|
||||
enable_hooks(false);
|
||||
return CR_OK;
|
||||
}
|
||||
@ -0,0 +1,116 @@
|
||||
-- Interface front-end for power-meter plugin.
|
||||
|
||||
local utils = require 'utils'
|
||||
local gui = require 'gui'
|
||||
local guidm = require 'gui.dwarfmode'
|
||||
local dlg = require 'gui.dialogs'
|
||||
|
||||
local plugin = require('plugins.power-meter')
|
||||
local bselector = df.global.ui_build_selector
|
||||
|
||||
PowerMeter = defclass(PowerMeter, guidm.MenuOverlay)
|
||||
|
||||
PowerMeter.focus_path = 'power-meter'
|
||||
|
||||
function PowerMeter:init()
|
||||
self:init_fields{
|
||||
min_power = 0, max_power = -1, invert = false,
|
||||
}
|
||||
guidm.MenuOverlay.init(self)
|
||||
return self
|
||||
end
|
||||
|
||||
function PowerMeter:onShow()
|
||||
guidm.MenuOverlay.onShow(self)
|
||||
|
||||
-- Send an event to update the errors
|
||||
bselector.plate_info.flags.whole = 0
|
||||
self:sendInputToParent('BUILDING_TRIGGER_ENABLE_WATER')
|
||||
end
|
||||
|
||||
function PowerMeter:onRenderBody(dc)
|
||||
dc:fill(0,0,dc.width-1,13,gui.CLEAR_PEN)
|
||||
dc:seek(1,1):pen(COLOR_WHITE)
|
||||
dc:string("Power Meter"):newline():newline(1)
|
||||
dc:string("Placement"):newline():newline(1)
|
||||
|
||||
dc:string("Excess power range:")
|
||||
|
||||
dc:newline(3):string("as", COLOR_LIGHTGREEN)
|
||||
dc:string(": Min ")
|
||||
if self.min_power <= 0 then
|
||||
dc:string("(any)")
|
||||
else
|
||||
dc:string(''..self.min_power)
|
||||
end
|
||||
|
||||
dc:newline(3):string("zx", COLOR_LIGHTGREEN)
|
||||
dc:string(": Max ")
|
||||
if self.max_power < 0 then
|
||||
dc:string("(any)")
|
||||
else
|
||||
dc:string(''..self.max_power)
|
||||
end
|
||||
dc:newline():newline(1)
|
||||
|
||||
dc:string("i",COLOR_LIGHTGREEN):string(": ")
|
||||
if self.invert then
|
||||
dc:string("Inverted")
|
||||
else
|
||||
dc:string("Not inverted")
|
||||
end
|
||||
end
|
||||
|
||||
function PowerMeter:onInput(keys)
|
||||
if keys.CUSTOM_I then
|
||||
self.invert = not self.invert
|
||||
elseif keys.BUILDING_TRIGGER_MIN_WATER_UP then
|
||||
self.min_power = self.min_power + 10
|
||||
elseif keys.BUILDING_TRIGGER_MIN_WATER_DOWN then
|
||||
self.min_power = math.max(0, self.min_power - 10)
|
||||
elseif keys.BUILDING_TRIGGER_MAX_WATER_UP then
|
||||
if self.max_power < 0 then
|
||||
self.max_power = 0
|
||||
else
|
||||
self.max_power = self.max_power + 10
|
||||
end
|
||||
elseif keys.BUILDING_TRIGGER_MAX_WATER_DOWN then
|
||||
self.max_power = math.max(-1, self.max_power - 10)
|
||||
elseif keys.LEAVESCREEN then
|
||||
self:dismiss()
|
||||
self:sendInputToParent('LEAVESCREEN')
|
||||
elseif keys.SELECT then
|
||||
if #bselector.errors == 0 then
|
||||
if not plugin.makePowerMeter(
|
||||
bselector.plate_info,
|
||||
self.min_power, self.max_power, self.invert
|
||||
)
|
||||
then
|
||||
dlg.showMessage(
|
||||
'Power Meter',
|
||||
'Could not initialize.', COLOR_LIGHTRED
|
||||
)
|
||||
|
||||
self:dismiss()
|
||||
self:sendInputToParent('LEAVESCREEN')
|
||||
return
|
||||
end
|
||||
|
||||
self:sendInputToParent('SELECT')
|
||||
if bselector.stage ~= 1 then
|
||||
self:dismiss()
|
||||
end
|
||||
end
|
||||
elseif self:propagateMoveKeys(keys) then
|
||||
return
|
||||
end
|
||||
end
|
||||
|
||||
if dfhack.gui.getCurFocus() ~= 'dwarfmode/Build/Position/Trap'
|
||||
or bselector.building_subtype ~= df.trap_type.PressurePlate
|
||||
then
|
||||
qerror("This script requires the main dwarfmode view in build pressure plate mode")
|
||||
end
|
||||
|
||||
local list = mkinstance(PowerMeter):init()
|
||||
list:show()
|
||||
Loading…
Reference in New Issue