Files
factorio-scenario-ExpCluster/exp_legacy/module/modules/gui/research.lua
2024-11-19 22:16:34 +00:00

329 lines
9.0 KiB
Lua

--- research gui
-- @gui Research
local Gui = require("modules.exp_legacy.expcore.gui") --- @dep expcore.gui
local Storage = require("modules/exp_util/storage")
local Event = require("modules/exp_legacy/utils/event") --- @dep utils.event
local Roles = require("modules.exp_legacy.expcore.roles") --- @dep expcore.roles
local config = require("modules.exp_legacy.config.research") --- @dep config.research
local format_time = _C.format_time --- @dep expcore.common
local research = {}
Storage.register(research, function(tbl)
research = tbl
end)
research.time = {}
research.res_queue_enable = false
local research_time_format = {
hours=true,
minutes=true,
seconds=true,
time=true,
string=true
}
local empty_time = format_time(0, {
hours=true,
minutes=true,
seconds=true,
time=true,
string=true,
null=true
})
local font_color = {
-- positive
[1] = {r = 0.3, g = 1, b = 0.3},
-- negative
[2] = {r = 1, g = 0.3, b = 0.3}
}
local res = {
['lookup_name'] = {},
['disp'] = {}
}
do
local res_total = 0
local i = 1
for k, v in pairs(config.milestone) do
research.time[i] = 0
res['lookup_name'][k] = i
res_total = res_total + v * 60
res['disp'][i] = {
raw_name = k,
target = res_total,
target_disp = format_time(res_total, research_time_format),
}
i = i + 1
end
end
local function research_add_log()
local result_data = {}
for i=1, #research.time, 1 do
result_data[res['disp'][i]['raw_name']] = research.time[i]
end
game.write_file(config.file_name, game.table_to_json(result_data) .. '\n', true, 0)
end
local function research_res_n(res_)
local res_n = 1
for k, _ in pairs(res_) do
if research.time[k] == 0 then
res_n = k - 1
break
end
end
if research.time[#res_] and research.time[#res_] > 0 then
if res_n == 1 then
res_n = #res_
end
end
if res_n < 3 then
res_n = 3
elseif res_n > (#research.time - 5) then
res_n = #research.time - 5
end
return res_n
end
local function research_notification(event)
if config.inf_res[event.research.name] then
if event.research.name == 'mining-productivity-4' then
if event.research.level == 5 then
-- Add run result to log
research_add_log()
end
if config.bonus_inventory.enabled then
if (event.research.level - 1) <= math.ceil(config.bonus_inventory.limit / config.bonus_inventory.rate) then
event.research.force[config.bonus_inventory.name] = math.max((event.research.level - 1) * config.bonus_inventory.rate, config.bonus_inventory.limit)
end
end
if config.pollution_ageing_by_research then
game.map_settings.pollution.ageing = math.min(10, event.research.level / 5)
end
else
if not (event.by_script) then
game.print{'expcom-res.inf', format_time(game.tick, research_time_format), event.research.name, event.research.level - 1}
end
end
else
if not (event.by_script) then
game.print{'expcom-res.msg', format_time(game.tick, research_time_format), event.research.name}
end
if config.bonus_inventory.enabled then
if event.research.name == 'mining-productivity-1' or event.research.name == 'mining-productivity-2' or event.research.name == 'mining-productivity-3' then
event.research.force[config.bonus_inventory.name] = event.research.level * config.bonus_inventory.rate
end
end
end
end
local function research_gui_update()
local res_disp = {}
local res_n = research_res_n(res['disp'])
for i=1, 8, 1 do
res_disp[i] = {
['name'] = '',
['target'] = '',
['attempt'] = '',
['difference'] = '',
['difference_color'] = font_color[1]
}
local res_i = res_n + i - 3
if res['disp'][res_i] then
res_disp[i]['name'] = {'expcom-res.res-name', res['disp'][res_i]['raw_name'], prototypes.technology[res['disp'][res_i]['raw_name']].localised_name}
if research.time[res_i] == 0 then
res_disp[i]['target'] = res['disp'][res_i].target_disp
res_disp[i]['attempt'] = empty_time
res_disp[i]['difference'] = empty_time
res_disp[i]['difference_color'] = font_color[1]
else
res_disp[i]['target'] = res['disp'][res_i].target_disp
res_disp[i]['attempt'] = format_time(research.time[res_i], research_time_format)
if research.time[res_i] < res['disp'][res_i].target then
res_disp[i]['difference'] = '-' .. format_time(res['disp'][res_i].target - research.time[res_i], research_time_format)
res_disp[i]['difference_color'] = font_color[1]
else
res_disp[i]['difference'] = format_time(research.time[res_i] - res['disp'][res_i].target, research_time_format)
res_disp[i]['difference_color'] = font_color[2]
end
end
end
end
return res_disp
end
--- Display label for the clock display
-- @element research_gui_clock_display
local research_gui_clock =
Gui.element{
type = 'label',
name = Gui.unique_static_name,
caption = empty_time,
style = 'heading_2_label'
}
--- A vertical flow containing the clock
-- @element research_clock_set
local research_clock_set =
Gui.element(function(_, parent, name)
local research_set = parent.add{type='flow', direction='vertical', name=name}
local disp = Gui.scroll_table(research_set, 390, 1, 'disp')
research_gui_clock(disp)
return research_set
end)
--- Display group
-- @element research_data_group
local research_data_group =
Gui.element(function(_definition, parent, i)
local name = parent.add{
type = 'label',
name = 'research_' .. i .. '_name',
caption = '',
style = 'heading_2_label'
}
name.style.width = 180
name.style.horizontal_align = 'left'
local target = parent.add{
type = 'label',
name = 'research_' .. i .. '_target',
caption = '',
style = 'heading_2_label'
}
target.style.width = 70
target.style.horizontal_align = 'right'
local attempt = parent.add{
type = 'label',
name = 'research_' .. i .. '_attempt',
caption = '',
style = 'heading_2_label'
}
attempt.style.width = 70
attempt.style.horizontal_align = 'right'
local difference = parent.add{
type = 'label',
name = 'research_' .. i .. '_difference',
caption = '',
style = 'heading_2_label'
}
difference.style.width = 70
difference.style.horizontal_align = 'right'
difference.style.font_color = font_color[1]
end)
--- A vertical flow containing the data
-- @element research_data_set
local research_data_set =
Gui.element(function(_, parent, name)
local research_set = parent.add{type='flow', direction='vertical', name=name}
local disp = Gui.scroll_table(research_set, 390, 4, 'disp')
local res_disp = research_gui_update()
research_data_group(disp, 0)
disp['research_0_name'].caption = {'expcom-res.name'}
disp['research_0_target'].caption = {'expcom-res.target'}
disp['research_0_attempt'].caption = {'expcom-res.attempt'}
disp['research_0_difference'].caption = {'expcom-res.difference'}
for i=1, 8, 1 do
research_data_group(disp, i)
local research_name_i = 'research_' .. i
disp[research_name_i .. '_name'].caption = res_disp[i]['name']
disp[research_name_i .. '_target'].caption = res_disp[i]['target']
disp[research_name_i .. '_attempt'].caption = res_disp[i]['attempt']
disp[research_name_i .. '_difference'].caption = res_disp[i]['difference']
disp[research_name_i .. '_difference'].style.font_color = res_disp[i]['difference_color']
end
return research_set
end)
local research_container =
Gui.element(function(definition, parent)
local container = Gui.container(parent, definition.name, 390)
research_clock_set(container, 'research_st_1')
research_data_set(container, 'research_st_2')
return container.parent
end)
:static_name(Gui.unique_static_name)
:add_to_left_flow()
Gui.left_toolbar_button('item/space-science-pack', {'expcom-res.main-tooltip'}, research_container, function(player)
return Roles.player_allowed(player, 'gui/research')
end)
Event.add(defines.events.on_research_finished, function(event)
research_notification(event)
if res['lookup_name'][event.research.name] == nil then
return
end
local n_i = res['lookup_name'][event.research.name]
research.time[n_i] = game.tick
local res_disp = research_gui_update()
for _, player in pairs(game.connected_players) do
local frame = Gui.get_left_element(player, research_container)
local disp = frame.container['research_st_2'].disp.table
for i=1, 8, 1 do
local research_name_i = 'research_' .. i
disp[research_name_i .. '_name'].caption = res_disp[i]['name']
disp[research_name_i .. '_target'].caption = res_disp[i]['target']
disp[research_name_i .. '_attempt'].caption = res_disp[i]['attempt']
disp[research_name_i .. '_difference'].caption = res_disp[i]['difference']
disp[research_name_i .. '_difference'].style.font_color = res_disp[i]['difference_color']
end
end
end)
Event.on_nth_tick(60, function()
local current_time = format_time(game.tick, research_time_format)
for _, player in pairs(game.connected_players) do
local frame = Gui.get_left_element(player, research_container)
local disp = frame.container['research_st_1'].disp.table
disp[research_gui_clock.name].caption = current_time
end
end)