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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
|
-- Counts monsters in a specify level, or level range.
local niters = 150
local output_file = "monster-report.out"
local excluded_things = util.set({}) -- "plant", "fungus", "bush" })
local function count_monsters_at(place, set)
debug.goto_place(place)
test.regenerate_level()
local monsters_here = set or { }
for mons in test.level_monster_iterator() do
local mname = mons.name
if not excluded_things[mname] then
local count = monsters_here[mname] or 0
monsters_here[mname] = count + 1
end
end
return monsters_here
end
local function report_monster_counts_at(place, mcount_map)
local text = ''
text = text .. "\n-------------------------------------------\n"
text = text .. place .. " monsters per-level\n"
text = text .. "-------------------------------------------\n"
local monster_counts = util.pairs(mcount_map)
table.sort(monster_counts, function (a, b)
return a[2].total > b[2].total
end)
local total = 0
for _, monster_pop in ipairs(monster_counts) do
total = total + monster_pop[2].total
end
for _, monster_pop in ipairs(monster_counts) do
text = text .. string.format("%5.2f (min: %2d, max: %2d, sd: %6.2f, %%: %6.2f%%) %s\n",
monster_pop[2].mean,
monster_pop[2].min,
monster_pop[2].max,
monster_pop[2].sigma,
monster_pop[2].total * 100.0 / total,
monster_pop[1])
end
return text
end
local function report_monster_counts(mcounts)
local places = util.keys(mcounts)
table.sort(places)
local text = ''
for _, place in ipairs(places) do
text = text .. report_monster_counts_at(place, mcounts[place])
end
file.writefile(output_file, text)
crawl.mpr("Dumped monster stats to " .. output_file)
end
local function calculate_monster_stats(iter_mpops)
local final_stats = { }
local n = #iter_mpops
for _, mpop in ipairs(iter_mpops) do
for mons, count in pairs(mpop) do
local mstats =
final_stats[mons] or { total = 0, max = 0, min = 0, pops = { } }
mstats.total = mstats.total + count
if count > mstats.max then
mstats.max = count
end
if count < mstats.min then
mstats.min = count
end
table.insert(mstats.pops, count)
final_stats[mons] = mstats
end
end
for mons, stat in pairs(final_stats) do
stat.mean = stat.total / n
local totaldelta2 = 0
local mean = stat.mean
for _, count in ipairs(stat.pops) do
local delta = count - mean
totaldelta2 = totaldelta2 + delta * delta
end
stat.sigma = math.sqrt(totaldelta2 / n)
end
return final_stats
end
local function count_monsters_from(start_place, end_place)
local place = start_place
local mcounts = { }
while place do
local mset = { }
local iter_mpops = { }
for i = 1, niters do
crawl.mesclr()
crawl.mpr("Counting monsters at " .. place .. " (" ..
i .. "/" .. niters .. ")")
local res = count_monsters_at(place)
table.insert(iter_mpops, res)
end
mcounts[place] = calculate_monster_stats(iter_mpops)
if place == end_place then
break
end
place = test.deeper_place_from(place)
end
report_monster_counts(mcounts)
end
local function parse_resets(resets)
local pieces = crawl.split(resets, ",")
local resets = { }
for _, p in ipairs(pieces) do
local _, _, place, depth = string.find(p, "^(.+)=(%d+)$")
table.insert(resets, { place, tonumber(depth) })
end
return resets
end
local function branch_resets()
local args = crawl.script_args()
local resets = { }
for _, arg in ipairs(args) do
local _, _, rawresets = string.find(arg, "^-reset=(.*)")
if rawresets then
util.append(resets, parse_resets(rawresets))
end
end
return resets
end
local function set_branch_depths()
for _, reset in ipairs(branch_resets()) do
debug.goto_place(reset[1], reset[2])
end
end
local function start_end_levels()
local args = script.simple_args()
if #args == 0 then
script.usage([[
Usage: place-population <start> [<end>]
For instance: place-population Shoal:1 Shoal:5
place-population Lair:3
You may optionally force branches to have entrances at specific places
with:
place-population -reset=Lair:1=8,Snake:1=3 Snake:5
With the general form:
-reset=<place>=<depth>[,<place>=<depth>,...]
where <place> is a valid place name as used in .des files, and <depth> is the
depth of the branch's entrance in its parent branch. Thus -reset=Snake:1=3
implies that the entrance of the Snake Pit is assumed to be on Lair:3.
]])
end
return args[1], args[2] or args[1]
end
set_branch_depths()
local lstart, lend = start_end_levels()
count_monsters_from(lstart, lend)
|