1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-16 10:19:27 +02:00
RedMew/map_gen/maps/diggy/feature/experience.lua

595 lines
23 KiB
Lua
Raw Normal View History

-- dependencies
local Event = require 'utils.event'
local Game = require 'utils.game'
local Global = require 'utils.global'
local Toast = require 'features.gui.toast'
local ForceControl = require 'features.force_control'
local ScoreTable = require 'map_gen.maps.diggy.score_table'
local Retailer = require 'features.retailer'
local Gui = require 'utils.gui'
2018-12-29 19:39:20 +02:00
local Utils = require 'utils.core'
local Color = require 'resources.color_presets'
local format = string.format
local floor = math.floor
local log = math.log
local max = math.max
local insert = table.insert
local pairs = pairs
local add_experience = ForceControl.add_experience
local add_experience_percentage = ForceControl.add_experience_percentage
local remove_experience_percentage = ForceControl.remove_experience_percentage
local print_player_floating_text_position = Game.print_player_floating_text_position
local get_force_data = ForceControl.get_force_data
local get_player_by_index = Game.get_player_by_index
local set_item = Retailer.set_item
local disable_item = Retailer.disable_item
local enable_item = Retailer.enable_item
-- this
local Experience = {}
local mining_efficiency = {
active_modifier = 0,
research_modifier = 0,
level_modifier = 0,
}
local inventory_slots = {
active_modifier = 0,
research_modifier = 0,
level_modifier = 0,
}
local health_bonus = {
active_modifier = 0,
research_modifier = 0,
level_modifier = 0,
}
Global.register({
mining_efficiency = mining_efficiency,
inventory_slots = inventory_slots,
health_bonus = health_bonus
}, function(tbl)
mining_efficiency = tbl.mining_efficiency
inventory_slots = tbl.inventory_slots
health_bonus = tbl.health_bonus
end)
local config = {}
2018-12-29 19:39:20 +02:00
local gain_xp_color = Color.light_sky_blue
local lose_xp_color = Color.red
local unlocked_color = Color.black
local locked_color = Color.gray
local table_column_layout = { type = 'table', column_count = 2 }
2018-11-26 21:31:10 +02:00
local level_up_formula = (function(level_reached)
local difficulty_scale = floor(config.difficulty_scale)
local level_fine_tune = floor(config.xp_fine_tune)
local start_value = (floor(config.first_lvl_xp))
local precision = (floor(config.cost_precision))
local function formula(level)
return (floor(
(1.15 ^ (level * 0.1))
+ difficulty_scale * (level) ^ 3
+ level_fine_tune * (level) ^ 2
+ start_value * (level)
- difficulty_scale * (level)
- level_fine_tune * (level))
)
end
local value = formula(level_reached + 1)
local lower_value = formula(level_reached)
value = value - (value % (10 ^ (floor(log(value, 10)) - precision)))
if lower_value == 0 then
return value - lower_value
end
lower_value = lower_value - (lower_value % (10 ^ (floor(log(lower_value, 10)) - precision)))
return value - lower_value
end)
local level_table = {}
---Get experience requirement for a given level
---Primarily used for the Experience GUI to display total experience required to unlock a specific item
---@param level number a number specifying the level
---@return number required total experience to reach supplied level
local function calculate_level_xp(level)
if level_table[level] == nil then
local value
if level == 1 then
value = level_up_formula(level - 1)
else
value = level_up_formula(level - 1) + calculate_level_xp(level - 1)
end
insert(level_table, level, value)
end
return level_table[level]
end
---Get a percentage of required experience between a level and the next level
---@param level number a number specifying the current level
---@return number a percentage of the required experience to level up from one level to the other
local function percentage_of_level_req(level, percentage)
return level_up_formula(level) * percentage
end
---Updates the market contents based on the current level.
---@param force LuaForce the force which the unlocking requirement should be based of
function Experience.update_market_contents(force)
local current_level = get_force_data(force).current_level
local force_name = force.name
for _, prototype in pairs(config.unlockables) do
local prototype_level = prototype.level
if current_level < prototype_level then
disable_item(force_name, prototype.name, format('Unlocks at level %d', prototype_level))
else
enable_item(force_name, prototype.name)
end
end
end
---Updates a forces manual mining speed modifier. By removing active modifiers and re-adding
---@param force LuaForce the force of which will be updated
---@param level_up number a level if updating as part of a level up (optional)
function Experience.update_mining_speed(force, level_up)
local buff = config.buffs['mining_speed']
if buff.max == nil or force.manual_mining_speed_modifier < buff.max then
level_up = level_up ~= nil and level_up or 0
if level_up > 0 and buff ~= nil then
local level = get_force_data(force).current_level
local adjusted_value = floor(max(buff.value, 24 * 0.9 ^ level))
local value = (buff.double_level ~= nil and level_up % buff.double_level == 0) and adjusted_value * 2 or adjusted_value
mining_efficiency.level_modifier = mining_efficiency.level_modifier + (value * 0.01)
end
-- remove the current buff
2019-02-21 01:47:16 +02:00
local old_modifier = (force.manual_mining_speed_modifier == 0) and 0 or force.manual_mining_speed_modifier - mining_efficiency.active_modifier
old_modifier = old_modifier >= 0 and old_modifier or 0
-- update the active modifier
mining_efficiency.active_modifier = mining_efficiency.research_modifier + mining_efficiency.level_modifier
-- add the new active modifier to the non-buffed modifier
force.manual_mining_speed_modifier = old_modifier + mining_efficiency.active_modifier
end
end
---Updates a forces inventory slots. By removing active modifiers and re-adding
---@param force LuaForce the force of which will be updated
---@param level_up number a level if updating as part of a level up (optional)
function Experience.update_inventory_slots(force, level_up)
local buff = config.buffs['inventory_slot']
if buff.max == nil or force.character_inventory_slots_bonus < buff.max then
level_up = level_up ~= nil and level_up or 0
if level_up > 0 and buff ~= nil then
local value = (buff.double_level ~= nil and level_up % buff.double_level == 0) and buff.value * 2 or buff.value
inventory_slots.level_modifier = inventory_slots.level_modifier + value
end
-- remove the current buff
2019-02-21 01:47:16 +02:00
local old_modifier = (force.character_inventory_slots_bonus == 0) and 0 or force.character_inventory_slots_bonus - inventory_slots.active_modifier
old_modifier = old_modifier >= 0 and old_modifier or 0
-- update the active modifier
inventory_slots.active_modifier = inventory_slots.research_modifier + inventory_slots.level_modifier
-- add the new active modifier to the non-buffed modifier
force.character_inventory_slots_bonus = old_modifier + inventory_slots.active_modifier
end
end
---Updates a forces health bonus. By removing active modifiers and re-adding
---@param force LuaForce the force of which will be updated
---@param level_up number a level if updating as part of a level up (optional)
function Experience.update_health_bonus(force, level_up)
local buff = config.buffs['health_bonus']
if buff.max == nil or force.character_health_bonus < buff.max then
level_up = level_up ~= nil and level_up or 0
if level_up > 0 and buff ~= nil then
local value = (buff.double_level ~= nil and level_up % buff.double_level == 0) and buff.value * 2 or buff.value
health_bonus.level_modifier = health_bonus.level_modifier + value
end
-- remove the current buff
2019-02-21 01:47:16 +02:00
local old_modifier = (force.character_health_bonus == 0) and 0 or force.character_health_bonus - health_bonus.active_modifier
old_modifier = old_modifier >= 0 and old_modifier or 0
-- update the active modifier
health_bonus.active_modifier = health_bonus.research_modifier + health_bonus.level_modifier
-- add the new active modifier to the non-buffed modifier
force.character_health_bonus = old_modifier + health_bonus.active_modifier
end
end
-- declaration of variables to prevent table look ups @see Experience.register
local sand_rock_xp
2018-11-30 21:30:50 +02:00
local rock_big_xp
local rock_huge_xp
---Awards experience when a rock has been mined (increases by 1 XP every 5th level)
---@param event LuaEvent
local function on_player_mined_entity(event)
local entity = event.entity
local name = entity.name
local player_index = event.player_index
local force = get_player_by_index(player_index).force
local level = get_force_data(force).current_level
local exp = 0
if name == 'sand-rock-big' then
2018-11-25 13:53:05 +02:00
exp = sand_rock_xp + floor(level / 5)
elseif name == 'rock-big' then
exp = rock_big_xp + floor(level / 5)
elseif name == 'rock-huge' then
2018-11-25 13:53:05 +02:00
exp = rock_huge_xp + floor(level / 5)
end
if exp == 0 then
return
end
print_player_floating_text_position(player_index, format('+%s XP', exp), gain_xp_color, 0, -0.5)
add_experience(force, exp)
end
---Awards experience when a research has finished, based on ingredient cost of research
---@param event LuaEvent
local function on_research_finished(event)
local research = event.research
local force = research.force
local exp
if research.research_unit_count_formula ~= nil then
local force_data = get_force_data(force)
exp = percentage_of_level_req(force_data.current_level, config.XP['infinity-research'])
else
local award_xp = 0
for _, ingredient in pairs(research.research_unit_ingredients) do
local name = ingredient.name
local reward = config.XP[name]
award_xp = award_xp + reward
end
exp = award_xp * research.research_unit_count
end
local text = format('Research completed! +%s XP', exp)
for _, p in pairs(game.connected_players) do
local player_index = p.index
print_player_floating_text_position(player_index, text, gain_xp_color, -1, -0.5)
end
add_experience(force, exp)
local current_modifier = mining_efficiency.research_modifier
local new_modifier = force.mining_drill_productivity_bonus * config.mining_speed_productivity_multiplier * 0.5
if (current_modifier == new_modifier) then
-- something else was researched
return
end
mining_efficiency.research_modifier = new_modifier
inventory_slots.research_modifier = force.mining_drill_productivity_bonus * 50 -- 1 per level
Experience.update_inventory_slots(force, 0)
Experience.update_mining_speed(force, 0)
2019-02-21 01:47:16 +02:00
Experience.update_health_bonus(force, 0)
game.forces.player.technologies['landfill'].enabled = false
end
---Awards experience when a rocket has been launched based on percentage of total experience
---@param event LuaEvent
local function on_rocket_launched(event)
local force = event.rocket.force
local exp = add_experience_percentage(force, config.XP['rocket_launch'])
local text = format('Rocket launched! +%s XP', exp)
for _, p in pairs(game.connected_players) do
local player_index = p.index
print_player_floating_text_position(player_index, text, gain_xp_color, -1, -0.5)
end
end
---Awards experience when a player kills an enemy, based on type of enemy
---@param event LuaEvent
2018-12-22 01:37:44 +02:00
local function on_entity_died(event)
local entity = event.entity
local force = event.force
local cause = event.cause
--For bot mining and turrets
if not cause or not cause.valid or cause.type ~= 'player' then
local exp = 0
local floating_text_position
-- stuff killed by the player force, but not the player
if force and force.name == 'player' then
2018-11-30 21:30:50 +02:00
local entity_name = entity.name
if cause and (cause.name == 'artillery-turret' or cause.name == 'gun-turret' or cause.name == 'laser-turret' or cause.name == 'flamethrower-turret') then
exp = config.XP['enemy_killed'] * (config.alien_experience_modifiers[entity_name] or 1)
floating_text_position = cause.position
else
local level = get_force_data(force).current_level
2018-11-30 21:30:50 +02:00
if entity_name == 'sand-rock-big' then
exp = floor((sand_rock_xp + level * 0.2) * 0.5)
elseif entity_name == 'rock-big' then
exp = floor((rock_big_xp + level * 0.2) * 0.5)
elseif entity_name == 'rock-huge' then
exp = floor((rock_huge_xp + level * 0.2) * 0.5)
end
2018-12-01 22:44:01 +02:00
floating_text_position = entity.position
end
end
if exp > 0 then
Game.print_floating_text(entity.surface, floating_text_position, format('+%s XP', exp), gain_xp_color)
add_experience(force, exp)
end
2018-11-26 21:23:53 +02:00
return
end
if entity.force.name ~= 'enemy' then
return
end
local exp = config.XP['enemy_killed'] * (config.alien_experience_modifiers[entity.name] or 1)
print_player_floating_text_position(cause.player.index, format('+%d XP', exp), gain_xp_color, -1, -0.5)
add_experience(force, exp)
end
---Deducts experience when a player respawns, based on a percentage of total experience
---@param event LuaEvent
local function on_player_respawned(event)
local player = get_player_by_index(event.player_index)
local exp = remove_experience_percentage(player.force, config.XP['death-penalty'], 50)
local text = format('-%s XP', exp)
game.print(format('%s drained %s experience.', player.name, exp), lose_xp_color)
for _, p in pairs(game.connected_players) do
print_player_floating_text_position(p.index, text, lose_xp_color, -1, -0.5)
end
2018-12-22 01:37:44 +02:00
ScoreTable.add('Experience lost', exp)
end
local function redraw_title(data)
local force_data = get_force_data('player')
2018-12-29 19:39:20 +02:00
data.frame.caption = Utils.comma_value(force_data.total_experience) .. ' total experience earned!'
end
local function apply_heading_style(style, width)
style.font = 'default-bold'
style.width = width
end
local function redraw_heading(data, header)
local head_condition = (header == 1)
local frame = (head_condition) and data.experience_list_heading or data.buff_list_heading
local header_caption = (head_condition) and 'Reward Item' or 'Reward Buff'
Gui.clear(frame)
local heading_table = frame.add(table_column_layout)
apply_heading_style(heading_table.add({ type = 'label', caption = 'Requirement' }).style, 100)
apply_heading_style(heading_table.add({ type = 'label', caption = header_caption }).style, 220)
end
local function redraw_progressbar(data)
local force_data = get_force_data('player')
local flow = data.experience_progressbars
Gui.clear(flow)
apply_heading_style(flow.add({ type = 'label', tooltip = 'Currently at level: ' .. force_data.current_level .. '\nNext level at: ' .. Utils.comma_value((force_data.total_experience - force_data.current_experience) + force_data.experience_level_up_cap) .. ' xp\nRemaining xp: ' .. Utils.comma_value(force_data.experience_level_up_cap - force_data.current_experience), name = 'Diggy.Experience.Frame.Progress.Level', caption = 'Progress to next level:' }).style)
local level_progressbar = flow.add({ type = 'progressbar', tooltip = floor(force_data.experience_percentage * 100) * 0.01 .. '% xp to next level' })
level_progressbar.style.width = 350
level_progressbar.value = force_data.experience_percentage * 0.01
end
local function redraw_table(data)
local experience_scroll_pane = data.experience_scroll_pane
Gui.clear(experience_scroll_pane)
redraw_progressbar(data)
redraw_heading(data, 1)
local last_level = 0
local current_force_level = get_force_data('player').current_level
for _, prototype in pairs(config.unlockables) do
local current_item_level = prototype.level
local first_item_for_level = current_item_level ~= last_level
local color
if current_force_level >= current_item_level then
2018-11-26 21:31:10 +02:00
color = unlocked_color
else
2018-11-26 21:31:10 +02:00
color = locked_color
end
local list = experience_scroll_pane.add(table_column_layout)
local level_caption = ''
if first_item_for_level then
level_caption = 'level ' .. current_item_level
end
local level_column = list.add({
type = 'label',
caption = level_caption,
2018-12-29 19:39:20 +02:00
tooltip = 'XP: ' .. Utils.comma_value(calculate_level_xp(current_item_level)),
})
level_column.style.minimal_width = 100
level_column.style.font_color = color
local item_column = list.add({
type = 'label',
caption = prototype.name
})
item_column.style.minimal_width = 22
item_column.style.font_color = color
last_level = current_item_level
end
end
local function redraw_buff(data)
local buff_scroll_pane = data.buff_scroll_pane
Gui.clear(buff_scroll_pane)
local all_levels_shown = false
for name, effects in pairs(config.buffs) do
local list = buff_scroll_pane.add(table_column_layout)
list.style.horizontal_spacing = 16
local level_caption = ''
if not all_levels_shown then
all_levels_shown = true
level_caption = 'All levels'
end
local level_label = list.add({ type = 'label', caption = level_caption })
level_label.style.minimal_width = 100
2018-11-26 21:31:10 +02:00
level_label.style.font_color = unlocked_color
local buff_caption
local effect_value = effects.value
if name == 'mining_speed' then
buff_caption = format('+%d mining speed', effect_value)
elseif name == 'inventory_slot' then
buff_caption = format('+%d inventory slot%s', effect_value, effect_value > 1 and 's' or '')
elseif name == 'health_bonus' then
buff_caption = format('+%d max health', effect_value)
else
buff_caption = format('+%d %s', effect_value, name)
end
local buffs_label = list.add({ type = 'label', caption = buff_caption })
buffs_label.style.minimal_width = 220
2018-11-26 21:31:10 +02:00
buffs_label.style.font_color = unlocked_color
end
end
local function toggle(event)
local player = event.player
local left = player.gui.left
local frame = left['Diggy.Experience.Frame']
if (frame and event.trigger == nil) then
Gui.destroy(frame)
return
elseif (frame) then
local data = Gui.get_data(frame)
redraw_title(data)
redraw_progressbar(data)
redraw_table(data)
return
end
frame = left.add({ name = 'Diggy.Experience.Frame', type = 'frame', direction = 'vertical' })
local experience_progressbars = frame.add({ type = 'flow', direction = 'vertical' })
local experience_list_heading = frame.add({ type = 'flow', direction = 'horizontal' })
local experience_scroll_pane = frame.add({ type = 'scroll-pane' })
experience_scroll_pane.style.maximal_height = 300
local buff_list_heading = frame.add({ type = 'flow', direction = 'horizontal' })
local buff_scroll_pane = frame.add({ type = 'scroll-pane' })
buff_scroll_pane.style.maximal_height = 100
frame.add({ type = 'button', name = 'Diggy.Experience.Button', caption = 'Close' })
local data = {
frame = frame,
experience_progressbars = experience_progressbars,
experience_list_heading = experience_list_heading,
experience_scroll_pane = experience_scroll_pane,
buff_list_heading = buff_list_heading,
buff_scroll_pane = buff_scroll_pane,
}
redraw_title(data)
redraw_table(data)
redraw_heading(data, 2)
redraw_buff(data)
Gui.set_data(frame, data)
end
local function on_player_created(event)
get_player_by_index(event.player_index).gui.top.add({
name = 'Diggy.Experience.Button',
type = 'sprite-button',
sprite = 'entity/market',
})
end
Gui.allow_player_to_toggle_top_element_visibility('Diggy.Experience.Button')
Gui.on_click('Diggy.Experience.Button', toggle)
Gui.on_custom_close('Diggy.Experience.Frame', function(event)
event.element.destroy()
end)
---Updates the experience progress gui for every player that has it open
local function update_gui()
local players = game.connected_players
for i = #players, 1, -1 do
local p = players[i]
local frame = p.gui.left['Diggy.Experience.Frame']
if frame and frame.valid then
local data = { player = p, trigger = 'update_gui' }
toggle(data)
end
end
2019-02-21 01:47:16 +02:00
--Resets buffs if they have been set to 0
local force = game.forces.player
Experience.update_inventory_slots(force, 0)
Experience.update_mining_speed(force, 0)
Experience.update_health_bonus(force, 0)
end
function Experience.register(cfg)
config = cfg
2018-12-22 01:37:44 +02:00
ScoreTable.reset('Experience lost')
--Adds the function on how to calculate level caps (When to level up)
local ForceControlBuilder = ForceControl.register(level_up_formula)
--Adds a function that'll be executed at every level up
ForceControlBuilder.register_on_every_level(function(level_reached, force)
Toast.toast_force(force, 10, format('Your team has reached level %d!', level_reached))
Experience.update_inventory_slots(force, level_reached)
Experience.update_mining_speed(force, level_reached)
Experience.update_health_bonus(force, level_reached)
Experience.update_market_contents(force)
end)
-- Events
Event.add(defines.events.on_player_mined_entity, on_player_mined_entity)
Event.add(defines.events.on_research_finished, on_research_finished)
Event.add(defines.events.on_rocket_launched, on_rocket_launched)
Event.add(defines.events.on_player_respawned, on_player_respawned)
Event.add(defines.events.on_entity_died, on_entity_died)
Event.add(defines.events.on_player_created, on_player_created)
Event.on_nth_tick(61, update_gui)
-- Prevents table lookup thousands of times
sand_rock_xp = config.XP['sand-rock-big']
2018-11-30 21:30:50 +02:00
rock_big_xp = config.XP['rock-big']
rock_huge_xp = config.XP['rock-huge']
end
function Experience.on_init()
--Adds the 'player' force to participate in the force control system.
local force = game.forces.player
ForceControl.register_force(force)
local force_name = force.name
for _, prototype in pairs(config.unlockables) do
set_item(force_name, prototype)
end
Experience.update_market_contents(force)
end
return Experience