Add a script to reclassify 'individual choice' weapons based on unit skill.
Alexander Gavrilov
parent
48da06ec37
commit
62cde96724
@ -0,0 +1,154 @@
|
||||
-- Rewrite individual choice weapons into specific types.
|
||||
|
||||
local utils = require 'utils'
|
||||
local dlg = require 'gui.dialogs'
|
||||
|
||||
local defs = df.global.world.raws.itemdefs
|
||||
local entity = df.global.ui.main.fortress_entity
|
||||
local tasks = df.global.ui.tasks
|
||||
local equipment = df.global.ui.equipment
|
||||
|
||||
function find_best_weapon(unit,mode)
|
||||
local best = nil
|
||||
local skill = nil
|
||||
local skill_level = nil
|
||||
local count = 0
|
||||
local function try(id,iskill)
|
||||
local slevel = dfhack.units.getNominalSkill(unit,iskill)
|
||||
-- Choose most skill
|
||||
if (skill ~= nil and slevel > skill_level)
|
||||
or (skill == nil and slevel > 0) then
|
||||
best,skill,skill_level,count = id,iskill,slevel,0
|
||||
end
|
||||
-- Then most produced within same skill
|
||||
if skill == iskill then
|
||||
local cnt = tasks.created_weapons[id]
|
||||
if cnt > count then
|
||||
best,count = id,cnt
|
||||
end
|
||||
end
|
||||
end
|
||||
for _,id in ipairs(entity.resources.weapon_type) do
|
||||
local def = defs.weapons[id]
|
||||
if def.skill_ranged >= 0 then
|
||||
if mode == nil or mode == 'ranged' then
|
||||
try(id, def.skill_ranged)
|
||||
end
|
||||
else
|
||||
if mode == nil or mode == 'melee' then
|
||||
try(id, def.skill_melee)
|
||||
end
|
||||
end
|
||||
end
|
||||
return best
|
||||
end
|
||||
|
||||
function unassign_wrong_items(unit,position,spec,subtype)
|
||||
for i=#spec.assigned-1,0,-1 do
|
||||
local id = spec.assigned[i]
|
||||
local item = df.item.find(id)
|
||||
|
||||
if item.subtype.subtype ~= subtype then
|
||||
spec.assigned:erase(i)
|
||||
|
||||
-- TODO: somewhat unexplored area; maybe missing some steps
|
||||
utils.erase_sorted(position.assigned_items,id)
|
||||
if utils.erase_sorted(equipment.items_assigned.WEAPON,item,'id') then
|
||||
utils.insert_sorted(equipment.items_unassigned.WEAPON,item,'id')
|
||||
end
|
||||
equipment.update.weapon = true
|
||||
unit.military.pickup_flags.update = true
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local count = 0
|
||||
|
||||
function adjust_uniform_spec(unit,position,spec,force)
|
||||
if not unit then return end
|
||||
local best
|
||||
if spec.indiv_choice.melee then
|
||||
best = find_best_weapon(unit, 'melee')
|
||||
elseif spec.indiv_choice.ranged then
|
||||
best = find_best_weapon(unit, 'ranged')
|
||||
elseif spec.indiv_choice.any or force then
|
||||
best = find_best_weapon(unit, nil)
|
||||
end
|
||||
if best then
|
||||
count = count + 1
|
||||
spec.item_filter.item_subtype = best
|
||||
spec.indiv_choice.any = false
|
||||
spec.indiv_choice.melee = false
|
||||
spec.indiv_choice.ranged = false
|
||||
unassign_wrong_items(unit, position, spec, best)
|
||||
end
|
||||
end
|
||||
|
||||
function adjust_position(unit,position,force)
|
||||
if not unit then
|
||||
local fig = df.historical_figure.find(position.occupant)
|
||||
if not fig then return end
|
||||
unit = df.unit.find(fig.unit_id)
|
||||
end
|
||||
|
||||
for _,v in ipairs(position.uniform.weapon) do
|
||||
adjust_uniform_spec(unit, position, v, force)
|
||||
end
|
||||
end
|
||||
|
||||
function adjust_squad(squad, force)
|
||||
for _,pos in ipairs(squad.positions) do
|
||||
adjust_position(nil, pos, force)
|
||||
end
|
||||
end
|
||||
|
||||
local args = {...}
|
||||
local vs = dfhack.gui.getCurViewscreen()
|
||||
local vstype = df.viewscreen_layer_militaryst
|
||||
if not vstype:is_instance(vs) then
|
||||
qerror('Call this from the military screen')
|
||||
end
|
||||
|
||||
if vs.page == vstype.T_page.Equip
|
||||
and vs.equip.mode == vstype.T_equip.T_mode.Customize then
|
||||
local slist = vs.layer_objects[0]
|
||||
local squad = vs.equip.squads[slist:getListCursor()]
|
||||
|
||||
if slist.active then
|
||||
print('Adjusting squad.')
|
||||
adjust_squad(squad)
|
||||
else
|
||||
local plist = vs.layer_objects[1]
|
||||
local pidx = plist:getListCursor()
|
||||
local pos = squad.positions[pidx]
|
||||
local unit = vs.equip.units[pidx]
|
||||
|
||||
if plist.active then
|
||||
print('Adjusting position.')
|
||||
adjust_position(unit, pos)
|
||||
elseif unit and vs.equip.edit_mode < 0 then
|
||||
local wlist = vs.layer_objects[2]
|
||||
local idx = wlist:getListCursor()
|
||||
local cat = vs.equip.assigned.category[idx]
|
||||
|
||||
if wlist.active and cat == df.uniform_category.weapon then
|
||||
print('Adjusting spec.')
|
||||
adjust_uniform_spec(unit, pos, vs.equip.assigned.spec[idx], true)
|
||||
end
|
||||
end
|
||||
end
|
||||
else
|
||||
qerror('Call this from the Equip page of military screen')
|
||||
end
|
||||
|
||||
if count > 1 then
|
||||
dlg.showMessage(
|
||||
'Choose Weapons',
|
||||
'Updated '..count..' uniform entries.', COLOR_GREEN
|
||||
)
|
||||
elseif count == 0 then
|
||||
dlg.showMessage(
|
||||
'Choose Weapons',
|
||||
'Did not find any entries to update.', COLOR_YELLOW
|
||||
)
|
||||
end
|
||||
Loading…
Reference in New Issue