Add a script that implements a linked mechanism browser.
Alexander Gavrilov
parent
38a07a4ca5
commit
601a3a7927
@ -0,0 +1,119 @@
|
||||
-- Support for messing with the dwarfmode screen
|
||||
|
||||
local _ENV = mkmodule('gui.dwarfmode')
|
||||
|
||||
local gui = require('gui')
|
||||
local dscreen = dfhack.screen
|
||||
|
||||
AREA_MAP_WIDTH = 23
|
||||
MENU_WIDTH = 30
|
||||
|
||||
function getPanelLayout()
|
||||
local sw, sh = dscreen.getWindowSize()
|
||||
local view_height = sh-2
|
||||
local view_rb = sw-1
|
||||
local area_x2 = sw-AREA_MAP_WIDTH-2
|
||||
local menu_x2 = sw-MENU_WIDTH-2
|
||||
local menu_x1 = area_x2-MENU_WIDTH-1
|
||||
local area_pos = df.global.ui_area_map_width
|
||||
local menu_pos = df.global.ui_menu_width
|
||||
local rv = {}
|
||||
|
||||
if area_pos < 3 then
|
||||
rv.area_map = gui.mkdims_xy(area_x2+1,1,view_rb-1,view_height)
|
||||
view_rb = area_x2
|
||||
end
|
||||
if menu_pos < area_pos or df.global.ui.main.mode ~= 0 then
|
||||
if menu_pos >= area_pos then
|
||||
rv.menu_forced = true
|
||||
menu_pos = area_pos-1
|
||||
end
|
||||
local menu_x = menu_x2
|
||||
if menu_pos < 2 then menu_x = menu_x1 end
|
||||
rv.menu = gui.mkdims_xy(menu_x+1,1,view_rb-1,view_height)
|
||||
view_rb = menu_x
|
||||
end
|
||||
rv.area_pos = area_pos
|
||||
rv.menu_pos = menu_pos
|
||||
rv.map = gui.mkdims_xy(1,1,view_rb-1,view_height)
|
||||
return rv
|
||||
end
|
||||
|
||||
function getViewportPos()
|
||||
return {
|
||||
x = df.global.window_x,
|
||||
y = df.global.window_y,
|
||||
z = df.global.window_z
|
||||
}
|
||||
end
|
||||
|
||||
function getCursorPos()
|
||||
return copyall(df.global.cursor)
|
||||
end
|
||||
|
||||
function setCursorPos(cursor)
|
||||
df.global.cursor = cursor
|
||||
end
|
||||
|
||||
function setViewportPos(layout,view)
|
||||
local map = df.global.world.map
|
||||
df.global.window_x = math.max(0, math.min(view.x, map.x_count-layout.map.width))
|
||||
df.global.window_y = math.max(0, math.min(view.y, map.y_count-layout.map.height))
|
||||
df.global.window_z = math.max(0, math.min(view.z, map.z_count-1))
|
||||
return getViewportPos()
|
||||
end
|
||||
|
||||
function centerViewportOn(layout,target)
|
||||
local mapsz = layout.map
|
||||
local view = xyz2pos(
|
||||
target.x-math.floor(layout.map.width/2),
|
||||
target.y-math.floor(layout.map.height/2),
|
||||
target.z
|
||||
)
|
||||
return setViewportPos(layout, view)
|
||||
end
|
||||
|
||||
function isInViewport(layout,view,target,gap)
|
||||
gap = gap or 0
|
||||
return target.x-gap >= view.x and target.x+gap < view.x+layout.map.width
|
||||
and target.y-gap >= view.y and target.y+gap < view.y+layout.map.height
|
||||
and target.z == view.z
|
||||
end
|
||||
|
||||
DwarfOverlay = defclass(DwarfOverlay, gui.Screen)
|
||||
|
||||
function DwarfOverlay:updateLayout()
|
||||
self.df_layout = getPanelLayout()
|
||||
self.df_viewport = getViewportPos()
|
||||
self.df_cursor = getCursorPos()
|
||||
end
|
||||
|
||||
local move_keys = {
|
||||
'CURSOR_UP', 'CURSOR_DOWN', 'CURSOR_LEFT', 'CURSOR_RIGHT',
|
||||
'CURSOR_UPLEFT', 'CURSOR_UPRIGHT', 'CURSOR_DOWNLEFT', 'CURSOR_DOWNRIGHT',
|
||||
'CURSOR_UP_FAST', 'CURSOR_DOWN_FAST', 'CURSOR_LEFT_FAST', 'CURSOR_RIGHT_FAST',
|
||||
'CURSOR_UPLEFT_FAST', 'CURSOR_UPRIGHT_FAST', 'CURSOR_DOWNLEFT_FAST', 'CURSOR_DOWNRIGHT_FAST',
|
||||
'CURSOR_UP_Z', 'CURSOR_DOWN_Z', 'CURSOR_UP_Z_AUX', 'CURSOR_DOWN_Z_AUX'
|
||||
}
|
||||
|
||||
function DwarfOverlay:propagateMoveKeys(keys)
|
||||
for _,v in ipairs(move_keys) do
|
||||
if keys[v] then
|
||||
self:inputToParent(v)
|
||||
return
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function DwarfOverlay:onIdle()
|
||||
dscreen.invalidate()
|
||||
end
|
||||
|
||||
function DwarfOverlay:clearMenu()
|
||||
local menu = self.df_layout.menu
|
||||
if not menu then return nil end
|
||||
dscreen.fillRect(gui.CLEAR_PEN,menu.x1,menu.y1,menu.x2,menu.y2)
|
||||
return menu.x1,menu.y1,menu.width,menu.height
|
||||
end
|
||||
|
||||
return _ENV
|
||||
@ -0,0 +1,143 @@
|
||||
-- Shows mechanisms linked to the current building.
|
||||
|
||||
local gui = require 'gui'
|
||||
local guidm = require 'gui.dwarfmode'
|
||||
|
||||
function getBuildingName(building)
|
||||
return dfhack.with_temp_object(
|
||||
df.new "string",
|
||||
function(str)
|
||||
building:getName(str)
|
||||
return str.value
|
||||
end
|
||||
)
|
||||
end
|
||||
|
||||
function listMechanismLinks(building)
|
||||
local lst = {}
|
||||
local function push(item, mode)
|
||||
if item then
|
||||
lst[#lst+1] = { obj = item, name = getBuildingName(item), mode = mode }
|
||||
end
|
||||
end
|
||||
|
||||
push(building, 'self')
|
||||
|
||||
if not df.building_actual:is_instance(building) then
|
||||
return lst
|
||||
end
|
||||
|
||||
local item, tref, tgt
|
||||
for _,v in ipairs(building.contained_items) do
|
||||
item = v.item
|
||||
if df.item_trappartsst:is_instance(item) then
|
||||
tref = dfhack.items.getGeneralRef(item, df.general_ref_type.BUILDING_TRIGGER)
|
||||
if tref then
|
||||
push(tref:getBuilding(), 'trigger')
|
||||
end
|
||||
tref = dfhack.items.getGeneralRef(item, df.general_ref_type.BUILDING_TRIGGERTARGET)
|
||||
if tref then
|
||||
push(tref:getBuilding(), 'target')
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return lst
|
||||
end
|
||||
|
||||
MechanismList = defclass(MechanismList, guidm.DwarfOverlay)
|
||||
|
||||
MechanismList.focus_path = 'mechanisms'
|
||||
|
||||
function MechanismList.new(building)
|
||||
local self = {
|
||||
old_cursor = guidm.getCursorPos(),
|
||||
links = listMechanismLinks(building),
|
||||
selected = 1
|
||||
}
|
||||
self.links[1].viewport = guidm.getViewportPos()
|
||||
self.links[1].cursor = guidm.getCursorPos()
|
||||
return mkinstance(MechanismList, self)
|
||||
end
|
||||
|
||||
local colors = {
|
||||
self = COLOR_CYAN, trigger = COLOR_MAGENTA, target = COLOR_GREEN
|
||||
}
|
||||
local icons = {
|
||||
self = 128, trigger = 17, target = 16
|
||||
}
|
||||
|
||||
function MechanismList:onRender()
|
||||
self:renderParent()
|
||||
self:updateLayout()
|
||||
local x,y,w,h = self:clearMenu()
|
||||
|
||||
self.paintString({fg=COLOR_WHITE},x+1,y+1,"Mechanism Links")
|
||||
|
||||
for i,v in ipairs(self.links) do
|
||||
local pen = { fg=colors[v.mode], bold = (i == self.selected) }
|
||||
self.paintTile(pen, x+2, y+2+i, icons[v.mode])
|
||||
self.paintString(pen, x+4, y+2+i, v.name)
|
||||
end
|
||||
|
||||
local nlinks = #self.links
|
||||
local line = y+4+nlinks
|
||||
|
||||
if nlinks <= 1 then
|
||||
self.paintString({fg=COLOR_LIGHTRED},x+1,line,"This building has no links")
|
||||
line = line+2
|
||||
end
|
||||
|
||||
self.paintString({fg=COLOR_LIGHTGREEN},x+1,line,"Esc")
|
||||
self.paintString({fg=COLOR_WHITE},x+4,line,": Back,")
|
||||
self.paintString({fg=COLOR_LIGHTGREEN},x+12,line,"Enter")
|
||||
self.paintString({fg=COLOR_WHITE},x+17,line,": Switch")
|
||||
end
|
||||
|
||||
function MechanismList:zoomToLink(link)
|
||||
df.global.world.selected_building = link.obj
|
||||
|
||||
local cursor = link.cursor
|
||||
if not cursor then
|
||||
cursor = xyz2pos(link.obj.centerx, link.obj.centery, link.obj.z)
|
||||
end
|
||||
guidm.setCursorPos(cursor)
|
||||
|
||||
if not guidm.isInViewport(self.df_layout, self.df_viewport, cursor, 5) then
|
||||
local vp = link.viewport
|
||||
if vp then
|
||||
guidm.setViewportPos(self.df_layout,vp)
|
||||
else
|
||||
guidm.centerViewportOn(self.df_layout,cursor)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function MechanismList:changeSelected(delta)
|
||||
self.selected = 1 + (self.selected + delta - 1) % #self.links
|
||||
self:zoomToLink(self.links[self.selected])
|
||||
end
|
||||
|
||||
function MechanismList:onInput(keys)
|
||||
if keys.STANDARDSCROLL_UP or keys.SECONDSCROLL_UP then
|
||||
self:changeSelected(-1)
|
||||
elseif keys.STANDARDSCROLL_DOWN or keys.SECONDSCROLL_DOWN then
|
||||
self:changeSelected(1)
|
||||
elseif keys.LEAVESCREEN then
|
||||
if self.selected ~= 1 then
|
||||
self:zoomToLink(self.links[1])
|
||||
end
|
||||
self:dismiss()
|
||||
elseif keys.SELECT then
|
||||
self:dismiss()
|
||||
end
|
||||
end
|
||||
|
||||
if not df.viewscreen_dwarfmodest:is_instance(dfhack.gui.getCurViewscreen()) then
|
||||
qerror("This script requires the main dwarfmode view")
|
||||
end
|
||||
if df.global.ui.main.mode ~= df.ui_sidebar_mode.QueryBuilding then
|
||||
qerror("This script requires the 'q' interface mode")
|
||||
end
|
||||
|
||||
MechanismList.new(df.global.world.selected_building):show()
|
||||
Loading…
Reference in New Issue