Slaughterhouse gui

control.lua

@@ -22,6 +22,7 @@ require "scripts.ocula.ocula"
 require "scripts.farming.farming"
 require "scripts.smart-farm.smart-farm"
 require "scripts.smart-farm.mega-farm"
+require "scripts.slaughterhouse.slaughterhouse"
 require "scripts.worm.worm"
 require "scripts.turd.turd"
 require "scripts.vatbrain.vatbrain"

locale/de/extras.cfg

@@ -2,7 +2,7 @@
 
 [gui-game-finished]
 
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=Wähle das nächste Opfer
 slaughterhouse-mk02=Wähle das nächste Opfer
 slaughterhouse-mk03=Wähle das nächste Opfer

locale/en/extras.cfg

@@ -1,6 +1,6 @@
 [gui-game-finished]
 
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=Choose your next victim
 slaughterhouse-mk02=Choose your next victim
 slaughterhouse-mk03=Choose your next victim

locale/es-ES/extras.cfg

@@ -1,6 +1,6 @@
 [gui-game-finished]
 
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=Elige a tu próxima víctima
 slaughterhouse-mk02=Elige a tu próxima víctima
 slaughterhouse-mk03=Elige a tu próxima víctima

locale/nl/extras.cfg

@@ -1,6 +1,6 @@
 [gui-game-finished]
 
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=Kies je volgende slachtoffer
 slaughterhouse-mk02=Kies je volgende slachtoffer
 slaughterhouse-mk03=Kies je volgende slachtoffer

locale/pl/extras.cfg

@@ -1,6 +1,6 @@
 [gui-game-finished]
 
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=Wybierz swoją następną ofiarę
 slaughterhouse-mk02=Wybierz swoją następną ofiarę
 slaughterhouse-mk03=Wybierz swoją następną ofiarę

locale/ru/extras.cfg

@@ -1,4 +1,4 @@
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=Выберите следующую жертву
 slaughterhouse-mk02=Выберите следующую жертву
 slaughterhouse-mk03=Выберите следующую жертву

locale/uk/extras.cfg

@@ -1,6 +1,6 @@
 [gui-game-finished]
 
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=Виберіть наступну жертву
 slaughterhouse-mk02=Виберіть наступну жертву
 slaughterhouse-mk03=Виберіть наступну жертву

locale/zh-CN/pyalienlife.cfg

@@ -161,7 +161,7 @@ requestor_info=需求箱信息
 provider=供应箱
 provider_info=供应箱信息
 
-[slaughterhouse-gui]
+[py-recipe-gui]
 slaughterhouse-mk01=有请下一位受害者
 slaughterhouse-mk02=有请下一位受害者
 slaughterhouse-mk03=有请下一位受害者

scripts/slaughterhouse/slaughterhouse.lua

@@ -0,0 +1,291 @@
+-- for full history reference: https://github.com/pyanodon/pyalienlife/commit/ed87228489c87e6c68993d78f09e76e21970302a
+
+local subgroups = {
+  "auog",
+  "ulric",
+  "mukmoux",
+  "arthurian",
+  "cottongut",
+  "dhilmos",
+  "scrondrix",
+  "phadai",
+  "fish",
+  "phagnot",
+  "kmauts",
+  "dingrits",
+  "xeno",
+  "arqad",
+  "cridren",
+  "antelope",
+  "trits",
+  "vonix",
+  "vrauks",
+  "xyhiphoe",
+  "zipir",
+  "korlex",
+  "simik",
+}
+if script.active_mods["pyalternativeenergy"] then
+  subgroups[#subgroups+1] = "zungror"
+  subgroups[#subgroups+1] = "numal"
+end
+if script.active_mods["pystellarexpedition"] then
+  -- subgroups[#subgroups+1] = "tuls"
+  -- subgroups[#subgroups+1] = "riga"
+  subgroups[#subgroups+1] = "kakkalakki"
+end
+
+local machines_with_gui = {
+  ["slaughterhouse-mk01"] = true,
+  ["slaughterhouse-mk02"] = true,
+  ["slaughterhouse-mk03"] = true,
+  ["slaughterhouse-mk04"] = true,
+  ["rc-mk01"] = true,
+  ["rc-mk02"] = true,
+  ["rc-mk03"] = true,
+  ["rc-mk04"] = true,
+}
+
+local alt_items = {
+  zipir = "zipir1",
+  kakkalakki = "kakkalakki-f"
+}
+
+local permitted_recipes = {}
+
+local function update_recipes()
+  for category in pairs(permitted_recipes) do
+    for r, recipe in pairs(prototypes.get_recipe_filtered{{filter = "category", category = category}}) do
+      for _, subgroup in pairs(subgroups) do
+        if recipe.subgroup.name:match(subgroup) then
+          permitted_recipes[category][r] = subgroup
+          break
+        end
+      end
+      if not permitted_recipes[category][r] then
+        error("Could not find associated subgroup for recipe: " .. r)
+      end
+    end
+  end
+end
+
+remote.add_interface("py-recipe-gui", {
+  ---add a machine to use the custom recipe viewer. requires subgroups to be registered
+  ---@param machine data.EntityID
+  add_machine = function (machine)
+    local update = not machines_with_gui
+    machines_with_gui[machine] = true
+    for category in pairs(update and prototypes.entity[machine].crafting_categories or {}) do
+      permitted_recipes[category] = {}
+    end
+    permitted_recipes.parameters = nil
+  end,
+  ---remove a machine from the whitelist for the custom recipe viewer
+  ---@param machine data.EntityID
+  remove_machine = function (machine)
+    machines_with_gui[machine] = nil
+  end,
+  ---add a subgroup to use the custom recipe viewer. requires a compatible crafting machine
+  ---@param subgroup data.EntityID
+  add_subgroup = function (subgroup)
+    local update = not subgroups[subgroup]
+    subgroups[subgroup] = true
+    if update then update_recipes() end
+  end,
+  ---add a subgroup to use the custom recipe viewer
+  ---@param subgroup data.EntityID
+  remove_subgroup = function (subgroup)
+    subgroups[subgroup] = nil
+  end,
+  ---use an alternative item for the subgroup header icon. set to nil to remove
+  ---@param subgroup data.ItemSubGroupID
+  ---@param item data.ItemID
+  set_alt_item = function (subgroup, item)
+    alt_items[subgroup] = item
+  end,
+})
+
+-- TODO have AE/SE use remote interface
+
+-- cache crafting categories
+for machine in pairs(machines_with_gui) do
+  for category in pairs(prototypes.entity[machine].crafting_categories) do
+    permitted_recipes[category] = {}
+  end
+end
+
+-- ignore parameters
+permitted_recipes.parameters = nil
+
+update_recipes()
+
+py.on_event(defines.events.on_object_destroyed, function(event)
+  local unit_number = event.useful_id
+  if not unit_number or not storage.opened_slaughterhouses[unit_number] then return end
+
+  storage.opened_slaughterhouses[unit_number] = nil
+  for _, player in pairs(game.players) do
+    local gui = player.gui.screen.slaughterhouse
+    if gui and gui.tags.entity == unit_number then gui.destroy() end
+  end
+end)
+
+local function get_item_from_subgroup(subgroup)
+  return alt_items[subgroup] or subgroup
+end
+
+local function build_subgroup_table(content_frame, player)
+  content_frame.clear()
+  local main_frame = content_frame.parent
+  main_frame.caption = main_frame.tags.caption
+  local subgroup_table = content_frame.add {type = "table", name = "s_table", column_count = 6}
+  for category in pairs(main_frame.tags.categories) do
+    for recipe, subgroup in pairs(permitted_recipes[category] or {}) do
+			local name = "py_recipe_gui_subgroup_" .. subgroup
+      if not subgroup_table[name] and player.force.recipes[recipe].enabled then
+				subgroup_table.add {
+					type = "choose-elem-button",
+					name = name,
+					elem_type = "item",
+					item = get_item_from_subgroup(subgroup),
+					style = "image_tab_slot",
+					tags = {subgroup = subgroup},
+					locked = true
+				}
+      end
+    end
+  end
+end
+
+local function create_gui(player_index)
+  local player = game.get_player(player_index)
+  if not player then return end
+  local entity = player.opened
+  if not entity then return end
+  local name = entity.name == "entity-ghost" and entity.ghost_name or entity.name
+  local control_behavior = entity.get_control_behavior()
+  if not machines_with_gui[name] or entity.get_recipe() or (control_behavior and control_behavior.circuit_set_recipe) then return end
+  local main_frame = player.gui.screen.add {
+    type = "frame",
+    name = "slaughterhouse",
+    direction = "vertical",
+    tags = {entity = entity.unit_number, categories = (entity.name == "entity-ghost" and entity.ghost_prototype or entity.prototype).crafting_categories, caption = {"py-recipe-gui." .. name}}
+  }
+  main_frame.force_auto_center()
+  player.opened = main_frame
+  local content_frame = main_frame.add {type = "frame", name = "content_frame", direction = "vertical", style = "inside_shallow_frame_with_padding"}
+  content_frame.style.vertically_stretchable = true
+  build_subgroup_table(content_frame, player)
+  storage.opened_slaughterhouses[entity.unit_number] = entity
+  script.register_on_object_destroyed(entity)
+  storage.watched_slaughterhouses[player_index] = nil
+  storage.watch_slaughterhouse = not not next(storage.watched_slaughterhouses)
+end
+
+py.on_event(py.events.on_gui_opened(), function(event)
+  local entity = event.entity
+  if not entity then return end
+  local name = entity.name == "entity-ghost" and entity.ghost_name or entity.name
+  if not machines_with_gui[name] then return end
+  local control_behavior = entity.get_control_behavior()
+
+  if entity.get_recipe() or control_behavior and control_behavior.circuit_set_recipe then
+    storage.watched_slaughterhouses[event.player_index] = entity
+    storage.watch_slaughterhouse = true
+  else
+    create_gui(event.player_index)
+  end
+end)
+
+py.on_event(defines.events.on_gui_closed, function(event)
+  -- if not storage.watched_slaughterhouses then return end
+  local player = game.get_player(event.player_index)
+  if event.gui_type == defines.gui_type.custom then
+    local gui = player.gui.screen.slaughterhouse
+    if gui then gui.destroy() end
+  end
+  storage.watched_slaughterhouses[event.player_index] = nil
+  storage.watch_slaughterhouse = not not next(storage.watched_slaughterhouses)
+end)
+
+local function set_recipe(player, entity, recipe)
+  for item, count in pairs(entity.set_recipe(recipe)) do
+    count = count - player.insert {name = item, count = count}
+    if count > 0 then
+      player.surface.spill_item_stack {position = player.position, stack = {name = item, count = count}, enable_looted = true}
+    end
+  end
+  player.opened = entity
+end
+
+gui_events[defines.events.on_gui_click]["py_recipe_gui_subgroup_.+"] = function(event)
+  local player = game.get_player(event.player_index)
+  local element = event.element
+  local content_frame = element.parent.parent
+  local main_frame = content_frame.parent
+  local subgroup = element.tags.subgroup
+  content_frame.clear()
+  main_frame.caption = {"py-recipe-gui.select-recipe"}
+  local recipe_flow = content_frame.add {type = "flow", direction = "horizontal"}
+  recipe_flow.add {type = "sprite-button", name = "py_recipe_gui_back", sprite = "utility/left_arrow"}
+  local recipe_table = recipe_flow.add {type = "table", column_count = 5}
+  local recipe_count, avalible_recipe = 0, nil
+  for category in pairs(main_frame.tags.categories) do
+    for recipe, recipe_subgroup in pairs(permitted_recipes[category] or {}) do
+      if recipe_subgroup == subgroup and player.force.recipes[recipe].enabled then
+        recipe_count, avalible_recipe = recipe_count + 1, recipe
+        recipe_table.add {
+          type = "choose-elem-button",
+          name = "py_recipe_gui_recipe_" .. recipe,
+          elem_type = "recipe",
+          recipe = recipe,
+          style = "slot_button",
+          tags = {recipe = recipe},
+          locked = true
+        }
+      end
+    end
+  end
+
+  if recipe_count == 1 then
+    local entity = storage.opened_slaughterhouses[main_frame.tags.entity]
+    if not entity or not entity.valid then return end
+    set_recipe(player, entity, avalible_recipe)
+  end
+end
+
+gui_events[defines.events.on_gui_click]["py_recipe_gui_back"] = function(event)
+  local player = game.players[event.player_index]
+  local content_frame = event.element.parent.parent
+  build_subgroup_table(content_frame, player)
+end
+
+gui_events[defines.events.on_gui_click]["py_recipe_gui_recipe_.+"] = function(event)
+  local player = game.get_player(event.player_index)
+  local element = event.element
+  local main_frame = element.parent.parent.parent.parent
+  local entity = storage.opened_slaughterhouses[main_frame.tags.entity]
+  local recipe = element.tags.recipe
+  if not entity or not entity.valid then return end
+  set_recipe(player, entity, recipe)
+end
+
+py.on_event(py.events.on_init(), function()
+  storage.watched_slaughterhouses = storage.watched_slaughterhouses or {}
+  storage.opened_slaughterhouses = storage.opened_slaughterhouses or {}
+end)
+
+py.on_event(defines.events.on_tick, function()
+  if not storage.watch_slaughterhouse then return end
+
+  for player_index, entity in pairs(storage.watched_slaughterhouses) do
+    if not entity.valid then
+      storage.watched_slaughterhouses[player_index] = nil
+      return
+    end
+
+    if not entity.get_recipe() then
+      create_gui(player_index, entity)
+    end
+  end
+end)