Update Module Gui to 2.0 (#13)

* Update module.lua

* Update research.lua

* Update research.lua

* Update research.lua

* Update research.lua

* Update research.lua

* Update research.lua

* Update module.lua

* Update module.lua

* Update module.lua

* Update module.lua

* Update gui.cfg

* Update module.lua

* Fix use of created_entity in events

* Update module.lua

* Improvements using 'module_slots' property

* Add quality support

---------

Co-authored-by: Cooldude2606 <25043174+Cooldude2606@users.noreply.github.com>

exp_legacy/module/config/module.lua

@@ -1,16 +1,20 @@
 return {
     -- type of machine to handle together
     default_module_row_count = 9,
-    module_slot_max = 4,
+    module_slots_per_row = 4,
-    copy_paste_module = true,
+    module_slot_max = 8,
+    copy_paste_module = false,
     copy_paste_rotation = false,
     machine = {
+    },
+    machine_set = {
+        ["base"] = {
             ["electric-mining-drill"] = {
-            ["module"] = "effectivity-module",
+                ["module"] = "efficiency-module",
                 ["prod"] = true,
             },
             ["pumpjack"] = {
-            ["module"] = "effectivity-module",
+                ["module"] = "efficiency-module",
                 ["prod"] = true,
             },
             ["assembling-machine-2"] = {
@@ -48,51 +52,37 @@ return {
             ["rocket-silo"] = {
                 ["module"] = "productivity-module-3",
                 ["prod"] = true,
+            }
         },
+        ["space-age"] = {
+            ["big-mining-drill"] = {
+                ["module"] = "efficiency-module",
+                ["prod"] = true,
             },
-    module_allowed = {
+            ["biochamber"] = {
-        ["advanced-circuit"] = true,
+                ["module"] = "efficiency-module",
-        ["automation-science-pack"] = true,
+                ["prod"] = true,
-        ["battery"] = true,
-        ["chemical-science-pack"] = true,
-        ["copper-cable"] = true,
-        ["copper-plate"] = true,
-        ["electric-engine-unit"] = true,
-        ["electronic-circuit"] = true,
-        ["empty-barrel"] = true,
-        ["engine-unit"] = true,
-        ["explosives"] = true,
-        ["flying-robot-frame"] = true,
-        ["iron-gear-wheel"] = true,
-        ["iron-plate"] = true,
-        ["iron-stick"] = true,
-        ["logistic-science-pack"] = true,
-        ["low-density-structure"] = true,
-        ["lubricant"] = true,
-        ["military-science-pack"] = true,
-        ["nuclear-fuel"] = true,
-        ["plastic-bar"] = true,
-        ["processing-unit"] = true,
-        ["production-science-pack"] = true,
-        ["rocket-control-unit"] = true,
-        ["rocket-fuel"] = true,
-        ["rocket-part"] = true,
-        ["steel-plate"] = true,
-        ["stone-brick"] = true,
-        ["sulfur"] = true,
-        ["sulfuric-acid"] = true,
-        ["uranium-fuel-cell"] = true,
-        ["utility-science-pack"] = true,
-        ["basic-oil-processing"] = true,
-        ["advanced-oil-processing"] = true,
-        ["coal-liquefaction"] = true,
-        ["heavy-oil-cracking"] = true,
-        ["light-oil-cracking"] = true,
-        ["solid-fuel-from-light-oil"] = true,
-        ["solid-fuel-from-petroleum-gas"] = true,
-        ["solid-fuel-from-heavy-oil"] = true,
-        ["uranium-processing"] = true,
-        ["nuclear-fuel-reprocessing"] = true,
-        ["kovarex-enrichment-process"] = true,
             },
+            ["electromagnetic-plant"] = {
+                ["module"] = "productivity-module-3",
+                ["prod"] = true,
+            },
+            ["cryogenic-plant"] = {
+                ["module"] = "productivity-module-3",
+                ["prod"] = true,
+            },
+            ["biolab"] = {
+                ["module"] = "productivity-module-3",
+                ["prod"] = true,
+            },
+            ["foundry"] = {
+                ["module"] = "productivity-module-3",
+                ["prod"] = true,
+            },
+            ["recycler"] = {
+                ["module"] = "quality-module-3",
+                ["prod"] = true,
+            }
+        }
+    }
 }

exp_legacy/module/locale/en/gui.cfg

@@ -297,6 +297,7 @@ control-type-storage-output=Storage Output
 
 [module]
 main-tooltip=Module GUI
+apply=Apply
 
 [landfill]
 main-tooltip=Blueprint Landfill GUI

exp_legacy/module/modules/gui/module.lua

@@ -1,6 +1,7 @@
 ---- module inserter
 -- @gui Module
 
+local AABB = require("modules/exp_util/aabb")
 local Gui = require("modules.exp_legacy.expcore.gui") --- @dep expcore.gui
 local Event = require("modules/exp_legacy/utils/event") --- @dep utils.event
 local Roles = require("modules.exp_legacy.expcore.roles") --- @dep expcore.roles
@@ -8,43 +9,31 @@ local config = require("modules.exp_legacy.config.module") --- @dep config.modul
 local Selection = require("modules.exp_legacy.modules.control.selection") --- @dep modules.control.selection
 local SelectionModuleArea = "ModuleArea"
 
---- align an aabb to the grid by expanding it
+local module_container -- Container for this GUI
-local function aabb_align_expand(aabb)
-    return {
-        left_top = {
-            x = math.floor(aabb.left_top.x),
-            y = math.floor(aabb.left_top.y),
-        },
-        right_bottom = {
-            x = math.ceil(aabb.right_bottom.x),
-            y = math.ceil(aabb.right_bottom.y),
-        },
-    }
-end
 
-local module_container
+local machine_names = {}
-local machine_name = {}
+for mod_name, machine_set in pairs(config.machine_set) do
-
+    if script.active_mods[mod_name] then
-for k, _ in pairs(config.machine) do
+        for machine_name, v in pairs(machine_set) do
-    table.insert(machine_name, k)
+            config.machine[machine_name] = v
+            table.insert(machine_names, machine_name)
+        end
+    end
 end
 
 local prod_module_names = {}
-
+for name, item in pairs(prototypes.item) do
-local function get_module_name()
-    for name, item in pairs(prototypes.item) do
     if item.module_effects and item.module_effects.productivity and item.module_effects.productivity > 0 then
         prod_module_names[#prod_module_names + 1] = name
     end
-    end
 end
 
 local elem_filter = {
-    name = { {
+    machine_name = { {
         filter = "name",
-        name = machine_name,
+        name = machine_names,
     } },
-    normal = { {
+    no_prod = { {
         filter = "type",
         type = "module",
     }, {
@@ -53,141 +42,207 @@ local elem_filter = {
         mode = "and",
         invert = true,
     } },
-    prod = { {
+    with_prod = { {
         filter = "type",
         type = "module",
     } },
 }
 
-local function clear_module(player, area, machine)
+--- Apply module changes to a crafting machine
+--- @param player LuaPlayer
+--- @param area BoundingBox
+--- @param machine_name string
+--- @param planner_with_prod LuaItemStack
+--- @param planner_no_prod LuaItemStack
+local function apply_module_to_crafter(player, area, machine_name, planner_with_prod, planner_no_prod)
     local force = player.force
-    local surface = player.surface -- Allow remote view
+    local surface = player.surface
-    for _, entity in pairs(surface.find_entities_filtered{ area = area, name = machine, force = force }) do
+    local upgrade_area = surface.upgrade_area
-        for _, r in pairs(surface.find_entities_filtered{ position = entity.position, name = "item-request-proxy", force = force }) do
-            if r then
-                r.destroy{ raise_destroy = true }
-            end
-        end
 
-        local m_current_module = entity.get_module_inventory()
+    --- @type BoundingBox
+    local param_area = { left_top = {}, right_bottom = {} }
 
-        if m_current_module then
+    --- @type LuaSurface.upgrade_area_param
-            local m_current_module_content = m_current_module.get_contents()
+    local params = {
-
+        area = param_area,
-            if m_current_module_content then
+        item = planner_with_prod,
-                for _, item in pairs(m_current_module_content) do
+        player = player,
-                    surface.spill_item_stack{
-                        position = entity.bounding_box.left_top,
-                        stack = item --[[ @as ItemStackDefinition https://forums.factorio.com/viewtopic.php?f=233&t=122323]],
-                        enable_looted = true,
         force = force,
-                        allow_belts = false
     }
-                end
-            end
 
-            m_current_module.clear()
+    for _, entity in pairs(surface.find_entities_filtered{ area = area, name = machine_name, force = force }) do
-        end
+        local pos = entity.position
-    end
+        param_area.left_top = pos
-end
+        param_area.right_bottom = pos
 
-local function apply_module(player, area, machine, modules)
+        local m_current_recipe = entity.get_recipe()
-    -- Intentionally left as player.surface to allow use in remote view
+        local r_proto = m_current_recipe and m_current_recipe.prototype
-    for _, entity in pairs(player.surface.find_entities_filtered{ area = area, name = machine, force = player.force }) do
-        local m_current_recipe
 
-        if entity.prototype.get_crafting_speed() then
+        if r_proto and (r_proto.maximum_productivity or (r_proto.allowed_effects and r_proto.allowed_effects["productivity"])) then
-            m_current_recipe = entity.get_recipe()
+            params.item = planner_with_prod
-        end
+            upgrade_area(params)
-
-        if m_current_recipe then
-            if config.module_allowed[m_current_recipe.name] then
-                entity.surface.create_entity{ name = "item-request-proxy", target = entity, position = entity.position, force = entity.force, modules = modules["n"] }
-                entity.last_user = player
         else
-                entity.surface.create_entity{ name = "item-request-proxy", target = entity, position = entity.position, force = entity.force, modules = modules["p"] }
+            params.item = planner_no_prod
-                entity.last_user = player
+            upgrade_area(params)
-            end
-        else
-            entity.surface.create_entity{ name = "item-request-proxy", target = entity, position = entity.position, force = entity.force, modules = modules["n"] }
-            entity.last_user = player
         end
     end
 end
 
 --- when an area is selected to add protection to the area
+--- @param event EventData.on_player_selected_area
 Selection.on_selection(SelectionModuleArea, function(event)
-    local area = aabb_align_expand(event.area)
+    local area = AABB.expand(event.area)
     local player = game.players[event.player_index]
-    local frame = Gui.get_left_element(player, module_container)
+    local frame = Gui.get_left_element(player, module_container) --- @type LuaGuiElement
     local scroll_table = frame.container.scroll.table
 
-    for i = 1, config.default_module_row_count do
+    -- Create an inventory with three upgrade planners
-        local mma = scroll_table["module_mm_" .. i .. "_0"].elem_value
+    local inventory = game.create_inventory(3)
+    inventory.insert{ name = "upgrade-planner", count = 3 }
+    local planner_all = inventory[1]
+    local planner_with_prod = inventory[2]
+    local planner_no_prod = inventory[3]
+    local mapper_index = 1
 
-        if mma then
+    for row = 1, config.default_module_row_count do
-            local mm = {
+        local machine_name = scroll_table["module_mm_" .. row .. "_0"].elem_value --[[@as string]]
-                ["n"] = {},
+        local entity_proto = prototypes.entity[machine_name]
-                ["p"] = {},
-            }
 
-            for j = 1, prototypes.entity[mma].module_inventory_size, 1 do
+        if machine_name then
-                local mmo = scroll_table["module_mm_" .. i .. "_" .. j].elem_value
+            local is_prod_crafter = false
+            local module_index = 1
+            local modules = {}
+            local no_prod = {}
 
-                if mmo then
+            -- Add all the modules selected
-                    if mm["n"][mmo] then
+            for column = 1, entity_proto.module_inventory_size do
-                        mm["n"][mmo] = mm["n"][mmo] + 1
+                local module_name = scroll_table["module_mm_" .. row .. "_" .. column].elem_value --[[ @as {name:string, quality:string} ]]
-                        mm["p"][mmo] = mm["p"][mmo] + 1
+
+                if module_name then
+                    local not_prod = module_name.name:gsub("productivity", "efficiency")
+                    modules[module_index] = module_name
+                    no_prod[module_index] = { name = not_prod, quality = module_name.quality }
+                    module_index = module_index + 1
+                    if not is_prod_crafter and module_name ~= not_prod and entity_proto.get_crafting_speed() then
+                        is_prod_crafter = true
+                    end
                 else
-                        mm["n"][mmo] = 1
+                    modules[module_index] = {}
-                        mm["p"][mmo] = 1
+                    no_prod[module_index] = {}
+                    module_index = module_index + 1
+                end
+            end
+
+            if is_prod_crafter then
+                -- Crafting machines with prod need to be handled on a case by case biases
+                local i = 0
+                for quality_name in pairs(prototypes.quality) do
+                    i = i + 1
+                    planner_with_prod.set_mapper(i, "from", {
+                        type = "entity",
+                        name = machine_name,
+                        quality = quality_name,
+                        comparator = "=",
+                    })
+                    planner_no_prod.set_mapper(i, "from", {
+                        type = "entity",
+                        name = machine_name,
+                        quality = quality_name,
+                        comparator = "=",
+                    })
+                    planner_with_prod.set_mapper(i, "to", {
+                        type = "entity",
+                        name = machine_name,
+                        module_slots = modules,
+                        quality = quality_name,
+                        comparator = "=",
+                    })
+                    planner_no_prod.set_mapper(i, "to", {
+                        type = "entity",
+                        name = machine_name,
+                        module_slots = no_prod,
+                        quality = quality_name,
+                        comparator = "=",
+                    })
+                end
+                apply_module_to_crafter(player, area, machine_name, planner_with_prod, planner_no_prod)
+            else
+                -- All other machines can be applied in a single upgrade planner
+                for quality_name in pairs(prototypes.quality) do
+                    planner_all.set_mapper(mapper_index, "from", {
+                        type = "entity",
+                        name = machine_name,
+                        quality = quality_name,
+                        comparator = "=",
+                    })
+                    planner_all.set_mapper(mapper_index, "to", {
+                        type = "entity",
+                        name = machine_name,
+                        module_slots = modules,
+                        quality = quality_name,
+                        comparator = "=",
+                    })
+                    mapper_index = mapper_index + 1
+                end
             end
         end
     end
 
-            for k, v in pairs(mm["p"]) do
+    -- Apply module changes for non crafting (or without prod selected)
-                if k:find("productivity") then
+    if mapper_index > 1 then
-                    local module_name = k:gsub("productivity", "effectivity")
+        player.surface.upgrade_area{
-                    mm["p"][module_name] = (mm["p"][module_name] or 0) + v
+            area = area,
-                    mm["p"][k] = nil
+            item = planner_all,
-                end
+            force = player.force,
+            player = player,
+        }
     end
 
-            if mm then
+    inventory.destroy()
-                clear_module(player, area, mma)
-                apply_module(player, area, mma, mm)
-            end
-        end
-    end
 end)
 
-local function row_set(player, element)
+--- Set the state of all elem selectors on a row
-    local frame = Gui.get_left_element(player, module_container)
+--- @param player LuaPlayer
+--- @param element_name string
+local function row_set(player, element_name)
+    local frame = Gui.get_left_element(player, module_container) --[[ @as LuaGuiElement ]]
     local scroll_table = frame.container.scroll.table
+    local machine_name = scroll_table[element_name .. "0"].elem_value --[[ @as string ]]
 
-    if scroll_table[element .. "0"].elem_value then
+    if machine_name then
+        local active_to = prototypes.entity[machine_name].module_inventory_size
+        local row_count = math.ceil(active_to / config.module_slots_per_row)
+        local visible_to = row_count * config.module_slots_per_row
         for i = 1, config.module_slot_max do
-            if i <= prototypes.entity[scroll_table[element .. "0"].elem_value].module_inventory_size then
+            local element = scroll_table[element_name .. i]
-                if config.machine[scroll_table[element .. "0"].elem_value].prod then
+            if i <= active_to then
-                    scroll_table[element .. i].elem_filters = elem_filter.prod
+                if config.machine[machine_name].prod then
+                    element.elem_filters = elem_filter.with_prod
                 else
-                    scroll_table[element .. i].elem_filters = elem_filter.normal
+                    element.elem_filters = elem_filter.no_prod
                 end
 
-                scroll_table[element .. i].enabled = true
+                element.visible = true
-                scroll_table[element .. i].elem_value = config.machine[scroll_table[element .. "0"].elem_value].module
+                element.enabled = true
+                element.elem_value = { name = config.machine[machine_name].module }
             else
-                scroll_table[element .. i].enabled = false
+                element.visible = i <= visible_to
-                scroll_table[element .. i].elem_value = nil
+                element.enabled = false
+                element.elem_value = nil
+            end
+            if i % (config.module_slots_per_row + 1) == 0 then
+                scroll_table[element_name .. "pad" .. i].visible = element.visible
             end
         end
     else
-        local mf = elem_filter.normal
-
         for i = 1, config.module_slot_max do
-            scroll_table[element .. i].enabled = false
+            local element = scroll_table[element_name .. i]
-            scroll_table[element .. i].elem_filters = mf
+            element.visible = i <= config.module_slots_per_row
-            scroll_table[element .. i].elem_value = nil
+            element.enabled = false
+            element.elem_value = nil
+            if i % (config.module_slots_per_row + 1) == 0 then
+                scroll_table[element_name .. "pad" .. i].visible = false
+            end
         end
     end
 end
@@ -195,7 +250,7 @@ end
 local button_apply =
     Gui.element{
         type = "button",
-        caption = "Apply",
+        caption = { "module.apply" },
         style = "button",
     }:on_click(function(player)
         if Selection.is_selecting(player, SelectionModuleArea) then
@@ -207,28 +262,37 @@ local button_apply =
 
 module_container =
     Gui.element(function(definition, parent)
-        local container = Gui.container(parent, definition.name, (config.module_slot_max + 2) * 36)
+        local container = Gui.container(parent, definition.name, (config.module_slots_per_row + 2) * 36)
         Gui.header(container, "Module Inserter", "", true)
 
-        local scroll_table = Gui.scroll_table(container, (config.module_slot_max + 2) * 36, config.module_slot_max + 1)
+        local slots_per_row = config.module_slots_per_row + 1
+        local scroll_table = Gui.scroll_table(container, (config.module_slots_per_row + 2) * 36, slots_per_row)
 
         for i = 1, config.default_module_row_count do
             scroll_table.add{
                 name = "module_mm_" .. i .. "_0",
                 type = "choose-elem-button",
                 elem_type = "entity",
-                elem_filters = elem_filter.name,
+                elem_filters = elem_filter.machine_name,
                 style = "slot_button",
             }
 
             for j = 1, config.module_slot_max do
+                if j % slots_per_row == 0 then
+                    scroll_table.add{
+                        type = "flow",
+                        name = "module_mm_" .. i .. "_pad" .. j,
+                        visible = false,
+                    }
+                end
                 scroll_table.add{
                     name = "module_mm_" .. i .. "_" .. j,
                     type = "choose-elem-button",
-                    elem_type = "item",
+                    elem_type = "item-with-quality",
-                    elem_filters = elem_filter.normal,
+                    elem_filters = elem_filter.no_prod,
                     style = "slot_button",
                     enabled = false,
+                    visible = j <= config.module_slots_per_row,
                 }
             end
         end
@@ -244,6 +308,7 @@ Gui.left_toolbar_button("item/productivity-module-3", { "module.main-tooltip" },
     return Roles.player_allowed(player, "gui/module")
 end)
 
+--- @param event EventData.on_gui_elem_changed
 Event.add(defines.events.on_gui_elem_changed, function(event)
     if event.element.name:sub(1, 10) == "module_mm_" then
         if event.element.name:sub(-1) == "0" then
@@ -252,8 +317,7 @@ Event.add(defines.events.on_gui_elem_changed, function(event)
     end
 end)
 
-Event.add(defines.events.on_player_joined_game, get_module_name)
+--- @param event EventData.on_entity_settings_pasted
-
 Event.add(defines.events.on_entity_settings_pasted, function(event)
     local source = event.source
     local destination = event.destination
@@ -292,6 +356,9 @@ Event.add(defines.events.on_entity_settings_pasted, function(event)
         end
     end
 
+    --[[
+    TODO handle later as may need using global to reduce creation of upgrade plans
+
     if config.copy_paste_module then
         if source.name ~= destination.name then
             return
@@ -315,4 +382,5 @@ Event.add(defines.events.on_entity_settings_pasted, function(event)
             apply_module(player, destination.bounding_box, destination.name, { ["n"] = source_inventory_content, ["p"] = source_inventory_content })
         end
     end
+    ]]
 end)