1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-14 10:13:13 +02:00
RedMew/features/gui/paint.lua

227 lines
5.9 KiB
Lua
Raw Normal View History

2018-06-08 19:22:23 +02:00
local Event = require 'utils.event'
local Gui = require 'utils.gui'
2018-09-23 00:25:13 +02:00
local Game = require 'utils.game'
2018-06-08 19:22:23 +02:00
local brush_tool = 'refined-hazard-concrete'
2018-06-08 19:22:23 +02:00
local valid_filters = {
'dirt-1',
'dirt-2',
'dirt-3',
'dirt-4',
'dirt-5',
'dirt-6',
'dirt-7',
'dry-dirt',
'grass-1',
'grass-2',
'grass-3',
'grass-4',
'lab-dark-1',
'lab-dark-2',
'lab-white',
'red-desert-0',
'red-desert-1',
'red-desert-2',
'red-desert-3',
'sand-1',
'sand-2',
'sand-3',
'tutorial-grid'
2018-06-08 19:22:23 +02:00
}
local main_button_name = Gui.uid_name()
local main_frame_name = Gui.uid_name()
local filter_button_name = Gui.uid_name()
local filter_clear_name = Gui.uid_name()
2018-06-08 19:22:23 +02:00
local filter_element_name = Gui.uid_name()
local filters_table_name = Gui.uid_name()
local filter_table_close_button_name = Gui.uid_name()
global.paint_brushes_by_player = {}
local function player_build_tile(event)
2018-11-20 12:46:32 +02:00
if not global.scenario.config.paint.enable then
return
2018-09-11 16:13:07 +02:00
end
if event.item.name ~= brush_tool then
return
end
local replace_tile = global.paint_brushes_by_player[event.player_index]
if not replace_tile then
return
end
local player = Game.get_player_by_index(event.player_index)
if not player.gui.left[main_frame_name] then
return
end
local tiles = event.tiles
local count = 0
for _, tile_data in ipairs(tiles) do
tile_data.name = replace_tile
if tile_data.old_tile.name == replace_tile then
count = count + 1
end
end
game.surfaces[event.surface_index].set_tiles(tiles)
if count > 0 then
player.insert {name = brush_tool, count = count}
end
2018-06-08 19:22:23 +02:00
end
local function player_joined(event)
2018-11-20 12:46:32 +02:00
if not global.scenario.config.paint.enable then
return
2018-09-11 16:13:07 +02:00
end
local player = Game.get_player_by_index(event.player_index)
2018-06-08 19:22:23 +02:00
if not player or not player.valid then
return
end
if player.gui.top[main_button_name] ~= nil then
return
end
player.gui.top.add {name = main_button_name, type = 'sprite-button', sprite = 'utility/spray_icon'}
end
local function draw_filters_table(event)
local center = event.player.gui.center
if center[filters_table_name] then
return
end
local frame = center.add {type = 'frame', name = filters_table_name, direction = 'vertical', caption = 'Palette'}
2018-06-08 19:22:23 +02:00
local t = frame.add {type = 'table', column_count = 10}
t.style.horizontal_spacing = 0
t.style.vertical_spacing = 0
for _, v in ipairs(valid_filters) do
local flow = t.add {type = 'flow'}
local b = flow.add {type = 'sprite-button', name = filter_element_name, sprite = 'tile/' .. v, tooltip = v}
Gui.set_data(b, frame)
b.style = 'slot_button'
2018-06-08 19:22:23 +02:00
end
local flow = frame.add {type = 'flow'}
local close = flow.add {type = 'button', name = filter_table_close_button_name, caption = 'Close'}
Gui.set_data(close, frame)
event.player.opened = frame
Gui.set_data(frame, event.element)
end
local function toggle(event)
local left = event.player.gui.left
local main_frame = left[main_frame_name]
if main_frame and main_frame.valid then
Gui.remove_data_recursivly(main_frame)
main_frame.destroy()
else
main_frame =
left.add {
type = 'frame',
name = main_frame_name,
direction = 'vertical',
caption = 'Paint Brush'
}
main_frame.add {
type = 'label',
caption = 'Choose a replacement tile for Refined hazard concrete'
2018-06-08 19:22:23 +02:00
}
local tooltip = global.paint_brushes_by_player[event.player_index] or ''
2018-06-08 19:22:23 +02:00
local brush =
main_frame.add({type = 'flow'}).add {
2018-06-08 19:22:23 +02:00
type = 'sprite-button',
name = filter_button_name,
tooltip = tooltip,
sprite = tooltip ~= '' and 'tile/' .. tooltip or nil
}
2018-06-08 19:22:23 +02:00
brush.style = 'slot_button'
local buttons_flow = main_frame.add {type = 'flow', direction = 'horizontal'}
buttons_flow.add {type = 'button', name = main_button_name, caption = 'Close'}
local clear_bursh = buttons_flow.add {type = 'button', name = filter_clear_name, caption = 'Clear Brush'}
Gui.set_data(clear_bursh, brush)
2018-06-08 19:22:23 +02:00
end
end
Gui.on_click(main_button_name, toggle)
Gui.on_click(
filter_button_name,
function(event)
if event.button == defines.mouse_button_type.right then
global.paint_brushes_by_player[event.player_index] = nil
2018-06-08 19:22:23 +02:00
local element = event.element
element.sprite = 'utility/pump_cannot_connect_icon'
element.tooltip = ''
else
draw_filters_table(event)
end
end
)
Gui.on_click(
filter_clear_name,
function(event)
local brush = Gui.get_data(event.element)
brush.sprite = 'utility/pump_cannot_connect_icon'
brush.tooltip = ''
global.paint_brushes_by_player[event.player_index] = nil
end
)
2018-06-08 19:22:23 +02:00
Gui.on_click(
filter_element_name,
function(event)
local element = event.element
local frame = Gui.get_data(element)
local filter_button = Gui.get_data(frame)
global.paint_brushes_by_player[event.player_index] = element.tooltip
2018-06-08 19:22:23 +02:00
filter_button.sprite = element.sprite
filter_button.tooltip = element.tooltip
Gui.remove_data_recursivly(frame)
frame.destroy()
end
)
Gui.on_click(
filter_table_close_button_name,
function(event)
local frame = Gui.get_data(event.element)
Gui.remove_data_recursivly(frame)
frame.destroy()
end
)
Gui.on_custom_close(
filters_table_name,
function(event)
local element = event.element
Gui.remove_data_recursivly(element)
element.destroy()
end
)
Event.add(defines.events.on_player_joined_game, player_joined)
Event.add(defines.events.on_player_built_tile, player_build_tile)