forked from DFHack/scripts
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathallneeds.lua
More file actions
84 lines (70 loc) · 2.44 KB
/
allneeds.lua
File metadata and controls
84 lines (70 loc) · 2.44 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
-- Prints the sum of all citizens' needs.
local argparse = require('argparse')
local sorts = {
id=function(a,b) return a.id < b.id end,
strength=function(a,b) return a.strength > b.strength end,
focus=function(a,b) return a.focus < b.focus end,
freq=function(a,b) return a.freq > b.freq end,
}
local sort = 'focus'
argparse.processArgsGetopt({...}, {
{'s', 'sort', hasArg=true, handler=function(optarg) sort = optarg end}
})
if not sorts[sort] then
qerror(('unknown sort: "%s"'):format(sort))
end
local fulfillment_threshold =
{ 300, 200, 100, -999, -9999, -99999 }
local function getFulfillment(focus_level)
for i = 1, 6 do
if focus_level >= fulfillment_threshold[i] then
return i
end
end
return 7
end
local fort_needs = {}
local units = dfhack.gui.getSelectedUnit(true)
if units then
print(('Summarizing needs for %s:'):format(dfhack.units.getReadableName(units)))
units = {units}
else
print('Summarizing needs for all (sane) citizens and residents:')
units = dfhack.units.getCitizens()
end
print()
for _, unit in ipairs(units) do
local mind = unit.status.current_soul.personality.needs
-- sum need_level and focus_level for each need
for _,need in ipairs(mind) do
local needs = ensure_key(fort_needs, need.id)
needs.strength = (needs.strength or 0) + need.need_level
needs.focus = (needs.focus or 0) + need.focus_level
needs.freq = (needs.freq or 0) + 1
local level = getFulfillment(need.focus_level)
ensure_key(needs, 'fulfillment', {0, 0, 0, 0, 0, 0, 0})
needs.fulfillment[level] = needs.fulfillment[level] + 1
end
end
local sorted_fort_needs = {}
for id, need in pairs(fort_needs) do
table.insert(sorted_fort_needs, {
id=df.need_type[id],
strength=need.strength,
focus=need.focus,
freq=need.freq,
fulfillment=need.fulfillment
})
end
table.sort(sorted_fort_needs, sorts[sort])
-- Print sorted output
local fmt = '%20s %8s %12s %9s %35s'
print(fmt:format("Need", "Strength", "Focus Impact", "Frequency", "Num. Unfettered -> Badly distracted"))
print(fmt:format("----", "--------", "------------", "---------", "-----------------------------------"))
for _, need in ipairs(sorted_fort_needs) do
local res = ""
for i = 1, 7 do
res = res..(('%5d'):format(need.fulfillment[i]))
end
print(fmt:format(need.id, need.strength, need.focus, need.freq, res))
end