1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-14 10:13:13 +02:00
RedMew/map_gen/presets/crash_site/blueprint_extractor.lua

198 lines
4.3 KiB
Lua
Raw Normal View History

2018-07-14 11:33:06 +02:00
local function get_mins(entities, tiles)
local min_x, min_y = math.huge, math.huge
for _, e in ipairs(entities) do
local p = e.position
local x, y = p.x, p.y
if x < min_x then
min_x = x
end
if y < min_y then
min_y = y
end
end
for _, e in ipairs(tiles) do
local p = e.position
local x, y = p.x, p.y
if x < min_x then
min_x = x
end
if y < min_y then
min_y = y
end
end
return min_x, min_y
end
local function output(result)
local str = {}
table.insert(str, 'return {\n')
for i, entry in pairs(result) do
table.insert(str, '[')
table.insert(str, i)
table.insert(str, '] = {')
local e = entry.entity
if e then
table.insert(str, 'entity = {')
table.insert(str, "name = '")
table.insert(str, e.name)
table.insert(str, "'")
local dir = e.direction
if dir then
table.insert(str, ', direction = ')
table.insert(str, dir)
end
local offset = e.offset
if offset then
table.insert(str, ', offset = ')
table.insert(str, offset)
end
table.insert(str, '}')
end
local t = entry.tile
if t then
if e then
table.insert(str, ', ')
end
table.insert(str, "tile = '")
table.insert(str, t.name)
table.insert(str, "'")
end
table.insert(str, '}')
table.insert(str, ',\n')
end
table.remove(str)
table.insert(str, '\n}')
str = table.concat(str)
game.write_file('bp.lua', str)
end
function extract1(size)
local cs = game.player.cursor_stack
if not (cs.valid_for_read and cs.name == 'blueprint' and cs.is_blueprint_setup()) then
game.print('invalid blueprint')
return
end
size = size or 6
local es = cs.get_blueprint_entities() or {}
local ts = cs.get_blueprint_tiles() or {}
local min_x, min_y = get_mins(es, ts)
min_x = 1 - math.ceil(min_x)
min_y = 1 - math.ceil(min_y)
local result = {}
for _, e in ipairs(es) do
local p = e.position
local x, y = p.x + min_x, p.y + min_y
local x2, y2 = math.ceil(x), math.ceil(y)
local i = (y2 - 1) * size + x2
local entry = result[i]
if not entry then
entry = {}
result[i] = entry
end
entry.entity = e
end
for _, e in ipairs(ts) do
local p = e.position
local x, y = p.x + min_x, p.y + min_y
x, y = math.ceil(x), math.ceil(y)
local i = (y - 1) * size + x
local entry = result[i]
if not entry then
entry = {}
result[i] = entry
end
entry.tile = e
end
output(result)
end
function extract4(size)
local cs = game.player.cursor_stack
if not (cs.valid_for_read and cs.name == 'blueprint' and cs.is_blueprint_setup()) then
game.print('invalid blueprint')
return
end
size = size or 6
local es = cs.get_blueprint_entities() or {}
local ts = cs.get_blueprint_tiles() or {}
local min_x, min_y = get_mins(es, ts)
min_x = 1 - math.floor(min_x)
min_y = 1 - math.floor(min_y)
local result = {}
for _, e in ipairs(es) do
local p = e.position
local x, y = p.x + min_x, p.y + min_y
local x2, y2 = math.floor(x), math.floor(y)
local i = (y2 - 1) * size + x2
local offset = 0
if x2 ~= x then
offset = offset + 1
end
if y2 ~= y then
offset = offset + 2
end
if offset ~= 0 then
e.offset = offset
end
local entry = result[i]
if not entry then
entry = {}
result[i] = entry
end
entry.entity = e
end
for _, e in ipairs(ts) do
local p = e.position
local x, y = p.x + min_x, p.y + min_y
x, y = math.ceil(x), math.ceil(y)
local i = (y - 1) * size + x
local entry = result[i]
if not entry then
entry = {}
result[i] = entry
end
entry.tile = e
end
output(result)
end