From 20e98d49263c2e1d719a57fc3c138dfafb32fd8f Mon Sep 17 00:00:00 2001
From: Alexander Gavrilov <angavrilov@gmail.com>
Date: Sun, 4 Nov 2012 20:51:13 +0400
Subject: [PATCH] Add a script for viewing and poking at local populations.

---
 scripts/region-pops.lua | 141 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 141 insertions(+)
 create mode 100644 scripts/region-pops.lua

diff --git a/scripts/region-pops.lua b/scripts/region-pops.lua
new file mode 100644
index 000000000..779167134
--- /dev/null
+++ b/scripts/region-pops.lua
@@ -0,0 +1,141 @@
+-- Shows populations of animals in the region, and allows tweaking them.
+
+local utils = require 'utils'
+
+local function sort_keys(tab)
+    local kt = {}
+    for k,v in pairs(tab) do table.insert(kt,k) end
+    table.sort(kt)
+    return ipairs(kt)
+end
+
+local is_plant_map = {
+    Animal = false, Vermin = false, VerminInnumerable = false,
+    ColonyInsect = false, Tree = true, Grass = true, Bush = true
+}
+
+function enum_populations()
+    local stat_table = {
+        plants = {},
+        creatures = {},
+        any = {}
+    }
+
+    for i,v in ipairs(df.global.world.populations) do
+        local typeid = df.world_population_type[v.type]
+        local is_plant = is_plant_map[typeid]
+        local id, obj, otable, idtoken
+
+        if is_plant then
+            id = v.plant
+            obj = df.plant_raw.find(id)
+            otable = stat_table.plants
+            idtoken = obj.id
+        else
+            id = v.race
+            obj = df.creature_raw.find(id)
+            otable = stat_table.creatures
+            idtoken = obj.creature_id
+        end
+
+        local entry = otable[idtoken]
+        if not entry then
+            entry = {
+                obj = obj, token = idtoken, id = id, records = {},
+                count = 0, known_count = 0,
+                known = false, infinite = false
+            }
+            otable[idtoken] = entry
+            stat_table.any[idtoken] = entry
+        end
+
+        table.insert(entry.records, v)
+        entry.known = entry.known or v.known
+
+        if v.quantity < 10000001 then
+            entry.count = entry.count + v.quantity
+            if v.known then
+                entry.known_count = entry.known_count + v.quantity
+            end
+        else
+            entry.infinite = true
+        end
+    end
+
+    return stat_table
+end
+
+function list_poptable(entries, all, pattern)
+    for _,k in sort_keys(entries) do
+        local entry = entries[k]
+        if (all or entry.known) and (not pattern or string.match(k,pattern)) then
+            local count = entry.known_count
+            if all then
+                count = entry.count
+            end
+            if entry.infinite then
+                count = 'innumerable'
+            end
+            print(string.format('%-40s %s', entry.token, count))
+        end
+    end
+end
+
+function list_populations(stat_table, all, pattern)
+    print('Plants:')
+    list_poptable(stat_table.plants, true, pattern)
+    print('\nCreatures and vermin:')
+    list_poptable(stat_table.creatures, all, pattern)
+end
+
+
+function boost_population(entry, factor, boost_count)
+    for _,v in ipairs(entry.records) do
+        if v.quantity < 10000001 then
+            boost_count = boost_count + 1
+            v.quantity = math.floor(v.quantity * factor)
+        end
+    end
+    return boost_count
+end
+
+local args = {...}
+local pops = enum_populations()
+
+if args[1] == 'list' or args[1] == 'list-all' then
+    list_populations(pops, args[1] == 'list-all', args[2])
+elseif args[1] == 'boost' or args[1] == 'boost-all' then
+    local factor = tonumber(args[3])
+    if not factor or factor < 0 then
+        qerror('Invalid boost factor.')
+    end
+
+    local count = 0
+
+    if args[1] == 'boost' then
+        local entry = pops.any[args[2]] or qerror('Unknown population token.')
+        count = boost_population(entry, factor, count)
+    else
+        for k,entry in pairs(pops.any) do
+            if string.match(k, args[2]) then
+                count = boost_population(entry, factor, count)
+            end
+        end
+    end
+
+    print('Updated '..count..' populations.')
+else
+    print([[
+Usage:
+  region-pops list [pattern]
+    Lists encountered populations of the region, possibly restricted by pattern.
+  region-pops list-all [pattern]
+    Lists all populations of the region.
+  region-pops boost <TOKEN> <factor>
+    Multiply all populations of TOKEN by factor.
+    If the factor is greater than one, increases the
+    population, otherwise decreases it.
+  region-pops boost-all <pattern> <factor>
+    Same as above, but match using a pattern acceptable to list.
+]])
+end