Start working on gui for workflow.
Alexander Gavrilov
parent
cfbdf47f6e
commit
09f8e8e419
@ -0,0 +1,14 @@
|
||||
local _ENV = mkmodule('plugins.workflow')
|
||||
|
||||
--[[
|
||||
|
||||
Native functions:
|
||||
|
||||
* isEnabled()
|
||||
* setEnabled(enable)
|
||||
* listConstraints([job]) -> {...}
|
||||
* setConstraint(token, by_count, goal[, gap]) -> {...}
|
||||
|
||||
--]]
|
||||
|
||||
return _ENV
|
||||
@ -0,0 +1,202 @@
|
||||
-- A GUI front-end for the workflow plugin.
|
||||
|
||||
local utils = require 'utils'
|
||||
local gui = require 'gui'
|
||||
local guidm = require 'gui.dwarfmode'
|
||||
local guimat = require 'gui.materials'
|
||||
local widgets = require 'gui.widgets'
|
||||
local dlg = require 'gui.dialogs'
|
||||
|
||||
local workflow = require 'plugins.workflow'
|
||||
|
||||
function check_enabled(cb,...)
|
||||
if workflow.isEnabled() then
|
||||
return cb(...)
|
||||
else
|
||||
dlg.showYesNoPrompt(
|
||||
'Enable Plugin',
|
||||
{ 'The workflow plugin is not enabled currently.', NEWLINE, NEWLINE
|
||||
'Press ', { key = 'MENU_CONFIRM' }, ' to enable it.' },
|
||||
COLOR_YELLOW,
|
||||
curry(function(...)
|
||||
workflow.setEnabled(true)
|
||||
return cb(...)
|
||||
end,...)
|
||||
)
|
||||
end
|
||||
end
|
||||
|
||||
JobConstraints = defclass(JobConstraints, guidm.MenuOverlay)
|
||||
|
||||
JobConstraints.focus_path = 'workflow-job'
|
||||
|
||||
JobConstraints.ATTRS {
|
||||
job = DEFAULT_NIL,
|
||||
frame_inset = 1,
|
||||
frame_background = COLOR_BLACK,
|
||||
}
|
||||
|
||||
function JobConstraints:init(args)
|
||||
self.building = dfhack.job.getHolder(self.job)
|
||||
|
||||
local status = { text = 'No worker', pen = COLOR_DARKGREY }
|
||||
local worker = dfhack.job.getWorker(self.job)
|
||||
if self.job.flags.suspend then
|
||||
status = { text = 'Suspended', pen = COLOR_RED }
|
||||
elseif worker then
|
||||
status = { text = dfhack.TranslateName(dfhack.units.getVisibleName(worker)), pen = COLOR_GREEN }
|
||||
end
|
||||
|
||||
self:addviews{
|
||||
widgets.Label{
|
||||
frame = { l = 0, t = 0 },
|
||||
text = {
|
||||
'Workflow Constraints'
|
||||
}
|
||||
},
|
||||
widgets.List{
|
||||
view_id = 'list',
|
||||
frame = { t = 2, b = 2 },
|
||||
row_height = 4,
|
||||
scroll_keys = widgets.SECONDSCROLL,
|
||||
},
|
||||
widgets.Label{
|
||||
frame = { l = 0, b = 0 },
|
||||
text = {
|
||||
{ key = 'LEAVESCREEN', text = ': Back',
|
||||
on_activate = self:callback('dismiss') }
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
self:initListChoices(args.clist)
|
||||
end
|
||||
|
||||
function JobConstraints:onGetSelectedBuilding()
|
||||
return self.building
|
||||
end
|
||||
|
||||
function JobConstraints:onGetSelectedJob()
|
||||
return self.job
|
||||
end
|
||||
|
||||
function describe_item_type(iobj)
|
||||
local itemline = 'any item'
|
||||
if iobj.item_type >= 0 then
|
||||
itemline = df.item_type.attrs[iobj.item_type].caption or iobj.item_type
|
||||
local def = dfhack.items.getSubtypeDef(iobj.item_type, iobj.item_subtype)
|
||||
local count = dfhack.items.getSubtypeCount(iobj.item_type, iobj.item_subtype)
|
||||
if def then
|
||||
itemline = def.name
|
||||
elseif count >= 0 then
|
||||
itemline = 'any '..itemline
|
||||
end
|
||||
end
|
||||
return itemline
|
||||
end
|
||||
|
||||
function is_caste_mat(iobj)
|
||||
return dfhack.items.isCasteMaterial(iobj.item_type)
|
||||
end
|
||||
|
||||
function describe_material(iobj)
|
||||
local matline = 'any material'
|
||||
if is_caste_mat(iobj) then
|
||||
matline = 'material not applicable'
|
||||
elseif iobj.mat_type >= 0 then
|
||||
local info = dfhack.matinfo.decode(iobj.mat_type, iobj.mat_index)
|
||||
if info then
|
||||
matline = info:toString()
|
||||
else
|
||||
matline = iobj.mat_type..':'..iobj.mat_index
|
||||
end
|
||||
end
|
||||
return matline
|
||||
end
|
||||
|
||||
function list_flags(list, bitfield)
|
||||
for name,val in pairs(bitfield) do
|
||||
if val then
|
||||
table.insert(list, name)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function JobConstraints:initListChoices(clist)
|
||||
clist = clist or workflow.listConstraints(self.job)
|
||||
|
||||
local choices = {}
|
||||
|
||||
for i,cons in ipairs(clist) do
|
||||
local goal = (cons.goal_value-cons.goal_gap)..'-'..cons.goal_value
|
||||
local curval
|
||||
if cons.goal_by_count then
|
||||
goal = goal .. ' stacks'
|
||||
curval = cons.cur_count
|
||||
else
|
||||
goal = goal .. ' items'
|
||||
curval = cons.cur_amount
|
||||
end
|
||||
local order_pen = COLOR_GREY
|
||||
if cons.request == 'resume' then
|
||||
order_pen = COLOR_GREEN
|
||||
elseif cons.request == 'suspend' then
|
||||
order_pen = COLOR_RED
|
||||
end
|
||||
local itemstr
|
||||
if cons.is_craft then
|
||||
itemstr = 'any craft'
|
||||
else
|
||||
itemstr = describe_item_type(cons)
|
||||
end
|
||||
if cons.min_quality > 0 then
|
||||
itemstr = itemstr .. ' ('..df.item_quality[cons.min_quality]..')'
|
||||
end
|
||||
local matstr = describe_material(cons)
|
||||
local matflagstr = ''
|
||||
local matflags = {}
|
||||
list_flags(matflags, cons.mat_mask)
|
||||
if #matflags > 0 then
|
||||
matflagstr = 'class: '..table.concat(matflags, ', ')
|
||||
end
|
||||
|
||||
table.insert(choices, {
|
||||
text = {
|
||||
goal, ' ', { text = '(now '..curval..')', pen = order_pen }, NEWLINE,
|
||||
' ', itemstr, NEWLINE, ' ', matstr, NEWLINE, ' ', matflagstr
|
||||
},
|
||||
obj = cons
|
||||
})
|
||||
end
|
||||
|
||||
self.subviews.list:setChoices(choices)
|
||||
end
|
||||
|
||||
function JobConstraints:onInput(keys)
|
||||
if self:propagateMoveKeys(keys) then
|
||||
if df.global.world.selected_building ~= self.building then
|
||||
self:dismiss()
|
||||
end
|
||||
else
|
||||
JobConstraints.super.onInput(self, keys)
|
||||
end
|
||||
end
|
||||
|
||||
if not string.match(dfhack.gui.getCurFocus(), '^dwarfmode/QueryBuilding/Some/Workshop/Job') then
|
||||
qerror("This script requires a workshop job selected in the 'q' mode")
|
||||
end
|
||||
|
||||
check_enabled(function()
|
||||
local job = dfhack.gui.getSelectedJob()
|
||||
if not job.flags['repeat'] then
|
||||
dlg.showMessage('Not Supported', 'Workflow only tracks repeat jobs.', COLOR_LIGHTRED)
|
||||
return
|
||||
end
|
||||
local clist = workflow.listConstraints(job)
|
||||
if not clist then
|
||||
dlg.showMessage('Not Supported', 'This type of job is not supported by workflow.', COLOR_LIGHTRED)
|
||||
return
|
||||
end
|
||||
JobConstraints{ job = job, clist = clist }:show()
|
||||
end)
|
||||
|
||||
Loading…
Reference in New Issue