58 lines
1.8 KiB
Lua
58 lines
1.8 KiB
Lua
local table = require("__flib__.table")
|
|
|
|
local util = require("scripts.util")
|
|
|
|
local lab_proc = {}
|
|
|
|
function lab_proc.build(database)
|
|
for name, prototype in pairs(global.prototypes.lab) do
|
|
local fuel_categories, fuel_filter = util.process_energy_source(prototype)
|
|
database.entity[name] = {
|
|
blueprintable = util.is_blueprintable(prototype),
|
|
can_burn = {},
|
|
class = "entity",
|
|
entity_type = { class = "entity_type", name = prototype.type },
|
|
fuel_categories = fuel_categories,
|
|
fuel_filter = fuel_filter,
|
|
hidden = prototype.has_flag("hidden"),
|
|
inputs = table.map(prototype.lab_inputs, function(v)
|
|
return { class = "item", name = v }
|
|
end),
|
|
module_slots = prototype.module_inventory_size
|
|
and prototype.module_inventory_size > 0
|
|
and prototype.module_inventory_size
|
|
or nil,
|
|
placed_by = util.process_placed_by(prototype),
|
|
prototype_name = name,
|
|
researching_speed = prototype.researching_speed,
|
|
science_packs = {},
|
|
size = util.get_size(prototype),
|
|
unlocked_by = {},
|
|
}
|
|
util.add_to_dictionary("entity", name, prototype.localised_name)
|
|
util.add_to_dictionary("entity_description", name, prototype.localised_description)
|
|
end
|
|
end
|
|
|
|
-- Store labs in science pack items' researched_in
|
|
function lab_proc.process_researched_in(database)
|
|
for name, prototype in pairs(global.prototypes.lab) do
|
|
-- Add to items
|
|
for _, item_name in ipairs(prototype.lab_inputs) do
|
|
local item_data = database.item[item_name]
|
|
if item_data then
|
|
item_data.researched_in[#item_data.researched_in + 1] = { class = "entity", name = name }
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
-- When calling the module directly, call lab_proc.build
|
|
setmetatable(lab_proc, {
|
|
__call = function(_, ...)
|
|
return lab_proc.build(...)
|
|
end,
|
|
})
|
|
|
|
return lab_proc
|