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
|
-- RST
-- calculations.lua
-- -------------------------------------
--
-- Functions used for the ingame help windows.
include "tribes/scripting/help/format_help.lua"
-- RST
-- .. function:: programs_wares_count(tribe, building, ware_description)
--
-- Examines the :ref:`production site programs <productionsite_programs>`
-- and returns three tables.
--
-- :arg tribe: The :class:`wl.map.TribeDescription` for the tribe that consumes the ware
-- :arg building: The :class:`wl.map.BuildingDescription`
-- :arg ware_description: A :class:`wl.map.WareDecription`
--
-- :returns (in this order):
-- * An :class:`array` of program_names this production site has
-- * A :class:`table` of ``{program_name,wares_count}``
-- * A :class:`table` of ``{program_name,formatted_string(s)}``
--
function programs_wares_count(tribe, building, ware_description)
local producing_programs = {}
if (building.is_mine) then
-- Find out which programs in the building produce this ware for the mines we skip the real production and take a dummy "encyclopedia" program
-- TODO(GunChleoc): Reimplement production programs as a proper tree, so that we can get rid of this hack and of the special case for mines.
for j, program_name in ipairs(building.production_programs) do
for ware, amount in pairs(building:produced_wares(program_name)) do
local consumes = building:consumed_wares_workers(program_name)
if (ware_description.name == ware and #consumes > 0) then
table.insert(producing_programs, program_name)
end
end
end
elseif (building.type_name == "trainingsite") then
-- Find out which programs in the building produce this ware
for j, program_name in ipairs(building.production_programs) do
local consumes = building:consumed_wares_workers(program_name)
if (#consumes > 0) then
table.insert(producing_programs, program_name)
end
end
else
-- Find out which programs in the building produce this ware
for j, program_name in ipairs(building.production_programs) do
for ware, amount in pairs(building:produced_wares(program_name)) do
if (ware_description.name == ware) then
table.insert(producing_programs, program_name)
end
end
end
end
-- Now collect all wares produced by the filtered programs
local produced_wares_strings = {}
local produced_wares_counters = {}
for j, program_name in ipairs(producing_programs) do
local produced_wares_amount = {}
produced_wares_counters[program_name] = 0
for ware, amount in pairs(building:produced_wares(program_name)) do
if (produced_wares_amount[ware] == nil) then
produced_wares_amount[ware] = 0
end
produced_wares_amount[ware] = produced_wares_amount[ware] + amount
produced_wares_counters[program_name] = produced_wares_counters[program_name] + amount
end
local produced_wares_string = ""
for ware, amount in pairs(produced_wares_amount) do
local ware_descr = wl.Game():get_ware_description(ware)
produced_wares_string = produced_wares_string
.. help_ware_amount_line(ware_descr, amount)
end
produced_wares_strings[program_name] = produced_wares_string
end
-- check for doubled entries (identical consumed and produced wares)
local deduplicated_programs = {}
for j, prog1_name in ipairs(producing_programs) do
local duplicate = false
for i, prog2_name in ipairs(deduplicated_programs) do
if (building.type_name == "trainingsite") then
break
end
if produced_wares_strings[prog1_name] == produced_wares_strings[prog2_name] and
help_consumed_wares_workers(tribe, building, prog1_name) ==
help_consumed_wares_workers(tribe, building, prog2_name) then
duplicate = true
break
end
end
if not duplicate then
table.insert(deduplicated_programs, prog1_name)
end
end
return deduplicated_programs, produced_wares_counters, produced_wares_strings
end
-- RST
-- .. function:: programs_workers_count(tribe, building, worker_description)
--
-- Examines the :ref:`production site programs <productionsite_programs>`
-- and returns three tables.
--
-- :arg tribe: The :class:`wl.map.TribeDescription` for the tribe that consumes the ware
-- :arg building: The :class:`wl.map.BuildingDescription`
-- :arg worker_description: A :class:`wl.map.WorkerDecription`
--
-- :returns (in this order):
-- * An :class:`array` of program_names this production site has
-- * A :class:`table` of ``{program_name,workers_count}``
-- * A :class:`table` of ``{program_name,formatted_string(s)}``
--
function programs_workers_count(tribe, building, worker_description)
local producing_programs = {}
-- Find out which programs in the building produce this worker
for j, program_name in ipairs(building.production_programs) do
for worker, amount in pairs(building:recruited_workers(program_name)) do
if (worker_description.name == worker) then
table.insert(producing_programs, program_name)
end
end
end
-- Now collect all workers recruited by the filtered programs
local recruited_workers_strings = {}
local recruited_workers_counters = {}
for j, program_name in ipairs(producing_programs) do
local recruited_workers_amount = {}
recruited_workers_counters[program_name] = 0
for worker, amount in pairs(building:recruited_workers(program_name)) do
if (recruited_workers_amount[worker] == nil) then
recruited_workers_amount[worker] = 0
end
recruited_workers_amount[worker] = recruited_workers_amount[worker] + amount
recruited_workers_counters[program_name] = recruited_workers_counters[program_name] + amount
end
local recruited_workers_string = ""
for worker, amount in pairs(recruited_workers_amount) do
local worker_descr = wl.Game():get_worker_description(worker)
recruited_workers_string = recruited_workers_string
.. help_ware_amount_line(worker_descr, amount)
end
recruited_workers_strings[program_name] = recruited_workers_string
end
-- check for doubled entries (identical consumed and produced workers)
local deduplicated_programs = {}
for j, prog1_name in ipairs(producing_programs) do
local duplicate = false
for i, prog2_name in ipairs(deduplicated_programs) do
if recruited_workers_strings[prog1_name] == recruited_workers_strings[prog2_name] and
help_consumed_wares_workers(tribe, building, prog1_name) ==
help_consumed_wares_workers(tribe, building, prog2_name) then
duplicate = true
break
end
end
if not duplicate then
table.insert(deduplicated_programs, prog1_name)
end
end
return deduplicated_programs, recruited_workers_counters, recruited_workers_strings
end
|
