mirror of
https://github.com/Refactorio/RedMew.git
synced 2025-01-18 03:21:47 +02:00
1029 lines
27 KiB
Lua
1029 lines
27 KiB
Lua
-- helpers
|
|
tau = 2 * math.pi
|
|
deg_to_rad = tau / 360
|
|
function degrees(angle)
|
|
return angle * deg_to_rad
|
|
end
|
|
|
|
function get_tile(tile)
|
|
if type(tile) == "table" then
|
|
return tile.tile
|
|
else
|
|
return tile
|
|
end
|
|
end
|
|
|
|
function add_entity(tile, entity)
|
|
if tile.entities then
|
|
table.insert(tile.entities, entity)
|
|
else
|
|
tile.entities = {entity}
|
|
end
|
|
end
|
|
|
|
-- shape builders
|
|
function empty_builder()
|
|
return nil
|
|
end
|
|
|
|
function full_builder()
|
|
return true
|
|
end
|
|
|
|
function tile_builder(tile)
|
|
return function()
|
|
return tile
|
|
end
|
|
end
|
|
|
|
function path_builder(thickness, optional_thickness_height)
|
|
local width = thickness / 2
|
|
local thickness2 = optional_thickness_height or thickness
|
|
local height = thickness2 / 2
|
|
return function(x, y)
|
|
return (x > -width and x <= width) or (y > -height and y <= height)
|
|
end
|
|
end
|
|
|
|
function rectangle_builder(width, height)
|
|
width = width / 2
|
|
if height then
|
|
height = height / 2
|
|
else
|
|
height = width
|
|
end
|
|
return function(x, y)
|
|
return x > -width and x <= width and y > -height and y <= height
|
|
end
|
|
end
|
|
|
|
function line_x_builder(thickness)
|
|
thickness = thickness / 2
|
|
return function(x, y)
|
|
return y > -thickness and y <= thickness
|
|
end
|
|
end
|
|
|
|
function line_y_builder(thickness)
|
|
thickness = thickness / 2
|
|
return function(x, y)
|
|
return x > -thickness and x <= thickness
|
|
end
|
|
end
|
|
|
|
function square_diamond_builder(size)
|
|
size = size / 2
|
|
return function(x, y)
|
|
return math.abs(x) + math.abs(y) <= size
|
|
end
|
|
end
|
|
|
|
local rot = math.sqrt(2) / 2 -- 45 degree rotation.
|
|
function rectangle_diamond_builder(width, height)
|
|
width = width / 2
|
|
height = height / 2
|
|
return function(x, y)
|
|
local rot_x = rot * (x - y)
|
|
local rot_y = rot * (x + y)
|
|
return math.abs(rot_x) < width and math.abs(rot_y) < height
|
|
end
|
|
end
|
|
|
|
function circle_builder(radius)
|
|
local rr = radius * radius
|
|
return function(x, y)
|
|
return x * x + y * y < rr
|
|
end
|
|
end
|
|
|
|
function oval_builder(x_radius, y_radius)
|
|
local x_rr = x_radius * x_radius
|
|
local y_rr = y_radius * y_radius
|
|
return function(x, y)
|
|
return ((x * x) / x_rr + (y * y) / y_rr) < 1
|
|
end
|
|
end
|
|
|
|
function sine_fill_builder(width, height)
|
|
width_inv = tau / width
|
|
height_inv = -2 / height
|
|
return function(x, y)
|
|
local x2 = x * width_inv
|
|
local y2 = y * height_inv
|
|
if y <= 0 then
|
|
return y2 < math.sin(x2)
|
|
else
|
|
return y2 > math.sin(x2)
|
|
end
|
|
end
|
|
end
|
|
|
|
local tile_map = {
|
|
[1] = false,
|
|
[2] = true,
|
|
[3] = "concrete",
|
|
[4] = "deepwater-green",
|
|
[5] = "deepwater",
|
|
[6] = "dirt-1",
|
|
[7] = "dirt-2",
|
|
[8] = "dirt-3",
|
|
[9] = "dirt-4",
|
|
[10] = "dirt-5",
|
|
[11] = "dirt-6",
|
|
[12] = "dirt-7",
|
|
[13] = "dry-dirt",
|
|
[14] = "grass-1",
|
|
[15] = "grass-2",
|
|
[16] = "grass-3",
|
|
[17] = "grass-4",
|
|
[18] = "hazard-concrete-left",
|
|
[19] = "hazard-concrete-right",
|
|
[20] = "lab-dark-1",
|
|
[21] = "lab-dark-2",
|
|
[22] = "lab-white",
|
|
[23] = "out-of-map",
|
|
[24] = "red-desert-0",
|
|
[25] = "red-desert-1",
|
|
[26] = "red-desert-2",
|
|
[27] = "red-desert-3",
|
|
[28] = "sand-1",
|
|
[29] = "sand-2",
|
|
[30] = "sand-3",
|
|
[31] = "stone-path",
|
|
[32] = "water-green",
|
|
[33] = "water"
|
|
}
|
|
|
|
function decompress(pic)
|
|
local data = pic.data
|
|
local width = pic.width
|
|
local height = pic.height
|
|
|
|
local uncompressed = {}
|
|
|
|
for y = 1, height do
|
|
local row = data[y]
|
|
local u_row = {}
|
|
uncompressed[y] = u_row
|
|
local x = 1
|
|
for index = 1, #row, 2 do
|
|
local pixel = tile_map[row[index]]
|
|
local count = row[index + 1]
|
|
|
|
for i = 1, count do
|
|
u_row[x] = pixel
|
|
x = x + 1
|
|
end
|
|
end
|
|
end
|
|
|
|
return {width = width, height = height, data = uncompressed}
|
|
end
|
|
|
|
function picture_builder(pic)
|
|
local data = pic.data
|
|
local width = pic.width
|
|
local height = pic.height
|
|
|
|
-- the plus one is because lua tables are one based.
|
|
local half_width = math.floor(width / 2) + 1
|
|
local half_height = math.floor(height / 2) + 1
|
|
return function(x, y)
|
|
x = math.floor(x)
|
|
y = math.floor(y)
|
|
local x2 = x + half_width
|
|
local y2 = y + half_height
|
|
|
|
if y2 > 0 and y2 <= height and x2 > 0 and x2 <= width then
|
|
return data[y2][x2]
|
|
else
|
|
return false
|
|
end
|
|
end
|
|
end
|
|
|
|
-- transforms and shape helpers
|
|
function translate(builder, x_offset, y_offset)
|
|
return function(x, y, world)
|
|
return builder(x - x_offset, y - y_offset, world)
|
|
end
|
|
end
|
|
|
|
function scale(builder, x_scale, y_scale)
|
|
x_scale = 1 / x_scale
|
|
y_scale = 1 / y_scale
|
|
return function(x, y, world)
|
|
return builder(x * x_scale, y * y_scale, world)
|
|
end
|
|
end
|
|
|
|
function rotate(builder, angle)
|
|
local qx = math.cos(angle)
|
|
local qy = math.sin(angle)
|
|
return function(x, y, world)
|
|
local rot_x = qx * x - qy * y
|
|
local rot_y = qy * x + qx * y
|
|
return builder(rot_x, rot_y, world)
|
|
end
|
|
end
|
|
|
|
function flip_x(builder)
|
|
return function(x, y, world)
|
|
return builder(-x, y, world)
|
|
end
|
|
end
|
|
|
|
function flip_y(builder)
|
|
return function(x, y, world)
|
|
return builder(x, -y, world)
|
|
end
|
|
end
|
|
|
|
function flip_xy(builder)
|
|
return function(x, y, world)
|
|
return builder(-x, -y, world)
|
|
end
|
|
end
|
|
|
|
-- For resource_module_builder it will return the first success.
|
|
function compound_or(builders)
|
|
return function(x, y, world)
|
|
for _, v in ipairs(builders) do
|
|
local tile = v(x, y, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
end
|
|
|
|
-- Wont work correctly with resource_module_builder becasues I don't know which one to return.
|
|
function compound_and(builders)
|
|
return function(x, y, world)
|
|
local tile
|
|
for _, v in ipairs(builders) do
|
|
tile = v(x, y, world)
|
|
if not tile then
|
|
return false
|
|
end
|
|
end
|
|
return tile
|
|
end
|
|
end
|
|
|
|
function invert(builder)
|
|
return function(x, y, world)
|
|
local tile = builder(x, y, world)
|
|
return not tile
|
|
end
|
|
end
|
|
|
|
function throttle_x(builder, x_in, x_size)
|
|
return function(x, y, world)
|
|
if x % x_size < x_in then
|
|
return builder(x, y, world)
|
|
else
|
|
return false
|
|
end
|
|
end
|
|
end
|
|
|
|
function throttle_y(builder, y_in, y_size)
|
|
return function(x, y, world)
|
|
if y % y_size < y_in then
|
|
return builder(x, y, world)
|
|
else
|
|
return false
|
|
end
|
|
end
|
|
end
|
|
|
|
function throttle_xy(builder, x_in, x_size, y_in, y_size)
|
|
return function(x, y, world)
|
|
if x % x_size < x_in and y % y_size < y_in then
|
|
return builder(x, y, world)
|
|
else
|
|
return false
|
|
end
|
|
end
|
|
end
|
|
|
|
function throttle_xy(builder, x_in, x_size, y_in, y_size)
|
|
return function(x, y, world)
|
|
if x % x_size < x_in and y % y_size < y_in then
|
|
return builder(x, y, world)
|
|
else
|
|
return false
|
|
end
|
|
end
|
|
end
|
|
|
|
function throttle_world_xy(builder, x_in, x_size, y_in, y_size)
|
|
return function(x, y, world)
|
|
if world.x % x_size < x_in and world.y % y_size < y_in then
|
|
return builder(x, y, world)
|
|
else
|
|
return false
|
|
end
|
|
end
|
|
end
|
|
|
|
function choose(condition, true_shape, false_shape)
|
|
return function(x, y, world)
|
|
if condition(x, y, world) then
|
|
return true_shape(x, y, world)
|
|
else
|
|
return false_shape(x, y, world)
|
|
end
|
|
end
|
|
end
|
|
|
|
function shape_or_else(shape, else_shape)
|
|
return function(x, y, world)
|
|
return shape(x, y, world) or else_shape(x, y, world)
|
|
end
|
|
end
|
|
|
|
function linear_grow(shape, size)
|
|
local half_size = size / 2
|
|
return function(x, y, world)
|
|
local t = math.ceil((y / size) + 0.5)
|
|
local n = math.ceil((math.sqrt(8 * t + 1) - 1) / 2)
|
|
local t_upper = n * (n + 1) * 0.5
|
|
local t_lower = t_upper - n
|
|
|
|
local y = (y - size * (t_lower + n / 2 - 0.5)) / n
|
|
local x = x / n
|
|
|
|
return shape(x, y, world)
|
|
end
|
|
end
|
|
|
|
function grow(in_shape, out_shape, size, offset)
|
|
local half_size = size / 2
|
|
return function(x, y, world)
|
|
local tx = math.ceil(math.abs(x) / half_size)
|
|
local ty = math.ceil(math.abs(y) / half_size)
|
|
local t = math.max(tx, ty)
|
|
|
|
for i = t, 2.5 * t, 1 do
|
|
local out_t = 1 / (i - offset)
|
|
local in_t = 1 / i
|
|
|
|
if out_shape(out_t * x, out_t * y, world) then
|
|
return nil
|
|
end
|
|
|
|
local tile = in_shape(in_t * x, in_t * y, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
end
|
|
|
|
return nil
|
|
end
|
|
end
|
|
|
|
function project(shape, size, r)
|
|
local ln_r = math.log(r)
|
|
local r2 = 1 / (r - 1)
|
|
local a = 1 / size
|
|
|
|
return function(x, y, world)
|
|
local offset = 0.5 * size
|
|
local sn = math.ceil(y + offset)
|
|
|
|
local n = math.ceil(math.log((r - 1) * sn * a + 1) / ln_r - 1)
|
|
local rn = r ^ n
|
|
local rn2 = 1 / rn
|
|
local c = size * rn
|
|
|
|
local sn_upper = size * (r ^ (n + 1) - 1) * r2
|
|
local x = x * rn2
|
|
local y = (y - (sn_upper - 0.5 * c) + offset) * rn2
|
|
|
|
return shape(x, y, world)
|
|
end
|
|
end
|
|
|
|
function project_overlap(shape, size, r)
|
|
local ln_r = math.log(r)
|
|
local r2 = 1 / (r - 1)
|
|
local a = 1 / size
|
|
local offset = 0.5 * size
|
|
|
|
return function(x, y, world)
|
|
local sn = math.ceil(y + offset)
|
|
|
|
local n = math.ceil(math.log((r - 1) * sn * a + 1) / ln_r - 1)
|
|
local rn = r ^ n
|
|
local rn2 = 1 / rn
|
|
local c = size * rn
|
|
|
|
local sn_upper = size * (r ^ (n + 1) - 1) * r2
|
|
local x = x * rn2
|
|
local y = (y - (sn_upper - 0.5 * c) + offset) * rn2
|
|
|
|
local tile
|
|
|
|
tile = shape(x, y, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
local n_above = n - 1
|
|
local rn_above = rn / r
|
|
local rn2_above = 1 / rn_above
|
|
local c_above = size * rn_above
|
|
|
|
local sn_upper_above = sn_upper - c
|
|
local x_above = x * rn2_above
|
|
local y_above = (y - (sn_upper_above - 0.5 * c_above) + offset) * rn2_above
|
|
|
|
tile = shape(x_above, y_above, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
local n_below = n + 1
|
|
local rn_below = rn * r
|
|
local rn2_below = 1 / rn_below
|
|
local c_below = size * rn_below
|
|
|
|
local sn_upper_below = sn_upper + c_below
|
|
local x_below = x * rn2_below
|
|
local y_below = (y - (sn_upper_below - 0.5 * c_below) + offset) * rn2_below
|
|
|
|
return shape(x_below, y_below, world)
|
|
end
|
|
end
|
|
|
|
-- ore generation.
|
|
-- builder is the shape of the ore patch.
|
|
function resource_module_builder(builder, resource_type, amount_function)
|
|
amount_function = amount_function or function(a, b)
|
|
return 404
|
|
end
|
|
return function(x, y, world)
|
|
if builder(x, y, world) then
|
|
return {
|
|
name = resource_type,
|
|
position = {world.x, world.y},
|
|
amount = amount_function(world.x, world.y)
|
|
}
|
|
end
|
|
end
|
|
end
|
|
|
|
function builder_with_resource(land_builder, resource_module)
|
|
return function(x, y, world)
|
|
local tile = land_builder(x, y, world)
|
|
if tile then
|
|
local entity = resource_module(x, y, world)
|
|
if entity then
|
|
if type(tile) == "table" then
|
|
add_entity(tile, entity)
|
|
else
|
|
tile = {
|
|
tile = tile,
|
|
entities = {entity}
|
|
}
|
|
end
|
|
end
|
|
end
|
|
|
|
return tile
|
|
end
|
|
end
|
|
|
|
-- pattern builders.
|
|
function single_pattern_builder(shape, width, height)
|
|
shape = shape or empty_builder
|
|
local half_width = width / 2
|
|
local half_height
|
|
if height then
|
|
half_height = height / 2
|
|
else
|
|
half_height = half_width
|
|
end
|
|
|
|
return function(x, y, world)
|
|
y = ((y + half_height) % height) - half_height
|
|
x = ((x + half_width) % width) - half_width
|
|
|
|
return shape(x, y, world)
|
|
end
|
|
end
|
|
|
|
function single_pattern_overlap_builder(shape, width, height)
|
|
shape = shape or empty_builder
|
|
local half_width = width / 2
|
|
local half_height
|
|
if height then
|
|
half_height = height / 2
|
|
else
|
|
half_height = half_width
|
|
end
|
|
|
|
return function(x, y, world)
|
|
y = ((y + half_height) % height) - half_height
|
|
x = ((x + half_width) % width) - half_width
|
|
|
|
return shape(x, y, world) or shape(x + width, y, world) or shape(x - width, y, world) or
|
|
shape(x, y + height, world) or
|
|
shape(x, y - height, world)
|
|
end
|
|
end
|
|
|
|
function single_x_pattern_builder(shape, width)
|
|
shape = shape or empty_builder
|
|
local half_width = width / 2
|
|
|
|
return function(x, y, world)
|
|
x = ((x + half_width) % width) - half_width
|
|
|
|
return shape(x, y, world)
|
|
end
|
|
end
|
|
|
|
function single_y_pattern_builder(shape, height)
|
|
shape = shape or empty_builder
|
|
local half_height = height / 2
|
|
|
|
return function(x, y, world)
|
|
y = ((y + half_height) % height) - half_height
|
|
|
|
return shape(x, y, world)
|
|
end
|
|
end
|
|
|
|
function grid_pattern_builder(pattern, columns, rows, width, height)
|
|
local half_width = width / 2
|
|
local half_height = height / 2
|
|
|
|
return function(x, y, world)
|
|
local y2 = ((y + half_height) % height) - half_height
|
|
local row_pos = math.floor(y / height + 0.5)
|
|
local row_i = row_pos % rows + 1
|
|
local row = pattern[row_i] or {}
|
|
|
|
local x2 = ((x + half_width) % width) - half_width
|
|
local col_pos = math.floor(x / width + 0.5)
|
|
local col_i = col_pos % columns + 1
|
|
|
|
local shape = row[col_i] or empty_builder
|
|
return shape(x2, y2, world)
|
|
end
|
|
end
|
|
|
|
function grid_pattern_overlap_builder(pattern, columns, rows, width, height)
|
|
local half_width = width / 2
|
|
local half_height = height / 2
|
|
|
|
return function(x, y, world)
|
|
local y2 = ((y + half_height) % height) - half_height
|
|
local row_pos = math.floor(y / height + 0.5)
|
|
local row_i = row_pos % rows + 1
|
|
local row = pattern[row_i] or {}
|
|
|
|
local x2 = ((x + half_width) % width) - half_width
|
|
local col_pos = math.floor(x / width + 0.5)
|
|
local col_i = col_pos % columns + 1
|
|
|
|
local shape = row[col_i] or empty_builder
|
|
|
|
local tile = shape(x2, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
-- edges
|
|
local col_i_left = (col_pos - 1) % columns + 1
|
|
shape = row[col_i_left] or empty_builder
|
|
tile = shape(x2 + width, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
local col_i_right = (col_pos + 1) % columns + 1
|
|
shape = row[col_i_right] or empty_builder
|
|
tile = shape(x2 - width, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
local row_i_up = (row_pos - 1) % rows + 1
|
|
local row_up = pattern[row_i_up] or {}
|
|
shape = row_up[col_i] or empty_builder
|
|
tile = shape(x2, y2 + height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
local row_i_down = (row_pos + 1) % rows + 1
|
|
local row_down = pattern[row_i_down] or {}
|
|
shape = row_down[col_i] or empty_builder
|
|
return shape(x2, y2 - height, world)
|
|
end
|
|
end
|
|
|
|
function grid_pattern_full_overlap_builder(pattern, columns, rows, width, height)
|
|
local half_width = width / 2
|
|
local half_height = height / 2
|
|
|
|
return function(x, y, world)
|
|
local y2 = ((y + half_height) % height) - half_height
|
|
local row_pos = math.floor(y / height + 0.5)
|
|
local row_i = row_pos % rows + 1
|
|
local row = pattern[row_i] or {}
|
|
|
|
local x2 = ((x + half_width) % width) - half_width
|
|
local col_pos = math.floor(x / width + 0.5)
|
|
local col_i = col_pos % columns + 1
|
|
|
|
local row_i_up = (row_pos - 1) % rows + 1
|
|
local row_up = pattern[row_i_up] or {}
|
|
local row_i_down = (row_pos + 1) % rows + 1
|
|
local row_down = pattern[row_i_down] or {}
|
|
|
|
local col_i_left = (col_pos - 1) % columns + 1
|
|
local col_i_right = (col_pos + 1) % columns + 1
|
|
|
|
-- start from top left, move left to right then down
|
|
local shape = row_up[col_i_left] or empty_builder
|
|
local tile = shape(x2 + width, y2 + height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_up[col_i] or empty_builder
|
|
tile = shape(x2, y2 + height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_up[col_i_right] or empty_builder
|
|
tile = shape(x2 - width, y2 + height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row[col_i_left] or empty_builder
|
|
tile = shape(x2 + width, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
local shape = row[col_i] or empty_builder
|
|
tile = shape(x2, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row[col_i_right] or empty_builder
|
|
tile = shape(x2 - width, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_down[col_i_left] or empty_builder
|
|
tile = shape(x2 + width, y2 - height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_down[col_i] or empty_builder
|
|
tile = shape(x2, y2 - height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_down[col_i_right] or empty_builder
|
|
return shape(x2 - width, y2 - height, world)
|
|
end
|
|
end
|
|
|
|
function segment_pattern_builder(pattern)
|
|
local count = #pattern
|
|
|
|
return function(x, y, world)
|
|
local angle = math.atan2(-y, x)
|
|
local index = math.floor(angle / tau * count) % count + 1
|
|
local shape = pattern[index] or empty_builder
|
|
return shape(x, y, world)
|
|
end
|
|
end
|
|
|
|
function pyramid_builder(pattern, columns, rows, width, height)
|
|
local half_width = width / 2
|
|
local half_height = height / 2
|
|
|
|
return function(x, y, world)
|
|
local y2 = ((y + half_height) % height) - half_height
|
|
local row_pos = math.floor(y / height + 0.5)
|
|
local row_i = row_pos % rows + 1
|
|
local row = pattern[row_i] or {}
|
|
|
|
if row_pos % 2 ~= 0 then
|
|
x = x - half_width
|
|
end
|
|
|
|
local x2 = ((x + half_width) % width) - half_width
|
|
local col_pos = math.floor(x / width + 0.5)
|
|
local col_i = col_pos % columns + 1
|
|
|
|
if col_pos > row_pos / 2 or -col_pos > (row_pos + 1) / 2 then
|
|
return false
|
|
end
|
|
|
|
local shape = row[col_i] or empty_builder
|
|
return shape(x2, y2, world)
|
|
end
|
|
end
|
|
|
|
function pyramid_inner_overlap_builder(pattern, columns, rows, width, height)
|
|
local half_width = width / 2
|
|
local half_height = height / 2
|
|
|
|
return function(x, y, world)
|
|
local y2 = ((y + half_height) % height) - half_height
|
|
local row_pos = math.floor(y / height + 0.5)
|
|
local row_i = row_pos % rows + 1
|
|
local row = pattern[row_i] or {}
|
|
|
|
local x_odd
|
|
local x_even
|
|
if row_pos % 2 == 0 then
|
|
x_even = x
|
|
x_odd = x - half_width
|
|
else
|
|
x_even = x - half_width
|
|
x_odd = x
|
|
x = x - half_width
|
|
end
|
|
|
|
x_even = ((x_even + half_width) % width) - half_width
|
|
x_odd = ((x_odd + half_width) % width) - half_width
|
|
|
|
local col_pos = math.floor(x / width + 0.5)
|
|
|
|
local offset = 1
|
|
local offset_odd = 0
|
|
if (col_pos % 2) == (row_pos % 2) then
|
|
offset = 0
|
|
offset_odd = 1
|
|
end
|
|
|
|
local col_i = (col_pos - offset) % columns + 1
|
|
local col_i_odd = (col_pos - offset_odd) % columns + 1
|
|
|
|
if col_pos > row_pos / 2 or -col_pos > (row_pos + 1) / 2 then
|
|
return false
|
|
end
|
|
|
|
local row_i_up = (row_pos - 1) % rows + 1
|
|
local row_up = pattern[row_i_up] or {}
|
|
local row_i_down = (row_pos + 1) % rows + 1
|
|
local row_down = pattern[row_i_down] or {}
|
|
|
|
local col_i_left = (col_pos - 1) % columns + 1
|
|
local col_i_right = (col_pos + 1) % columns + 1
|
|
|
|
local col_i_left2 = (col_pos - 2) % columns + 1
|
|
local col_i_right2 = (col_pos + 0) % columns + 1
|
|
|
|
-- start from top left, move left to right then down
|
|
local shape = row_up[col_i_left] or empty_builder
|
|
local tile = shape(x_even + width, y2 + height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_up[col_i_odd] or empty_builder
|
|
tile = shape(x_odd, y2 + height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_up[col_i_right] or empty_builder
|
|
tile = shape(x_even - width, y2 + height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row[col_i_left] or empty_builder
|
|
tile = shape(x_even + width, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
local shape = row[col_i] or empty_builder
|
|
tile = shape(x_even, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row[col_i_right] or empty_builder
|
|
tile = shape(x_even - width, y2, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_down[col_i_left] or empty_builder
|
|
tile = shape(x_even + width, y2 - height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_down[col_i_odd] or empty_builder
|
|
tile = shape(x_odd, y2 - height, world)
|
|
if tile then
|
|
return tile
|
|
end
|
|
|
|
shape = row_down[col_i_right] or empty_builder
|
|
return shape(x_even - width, y2 - height, world)
|
|
end
|
|
end
|
|
|
|
function grid_pattern_offset_builder(pattern, columns, rows, width, height)
|
|
local half_width = width / 2
|
|
local half_height = height / 2
|
|
|
|
return function(x, y, world)
|
|
local y2 = ((y + half_height) % height) - half_height
|
|
local row_pos = math.floor(y / height + 0.5)
|
|
local row_i = row_pos % rows + 1
|
|
local row = pattern[row_i] or {}
|
|
|
|
local x2 = ((x + half_width) % width) - half_width
|
|
local col_pos = math.floor(x / width + 0.5)
|
|
local col_i = col_pos % columns + 1
|
|
|
|
local y2 = y2 + height * math.floor((row_pos + 1) / rows)
|
|
local x2 = x2 + width * math.floor((col_pos + 1) / columns)
|
|
|
|
local shape = row[col_i] or empty_builder
|
|
return shape(x2, y2, world)
|
|
end
|
|
end
|
|
|
|
-- tile converters
|
|
function change_tile(builder, old_tile, new_tile)
|
|
return function(x, y, world)
|
|
local tile = builder(x, y, world)
|
|
|
|
if type(tile) == "table" then
|
|
if tile.tile == old_tile then
|
|
tile.tile = new_tile
|
|
end
|
|
else
|
|
if tile == old_tile then
|
|
tile = new_tile
|
|
end
|
|
end
|
|
|
|
return tile
|
|
end
|
|
end
|
|
|
|
function change_collision_tile(builder, collides, new_tile)
|
|
return function(x, y, world)
|
|
local tile = builder(x, y, world)
|
|
|
|
if type(tile) == "table" then
|
|
if tile.tile.collides_with(collides) then
|
|
tile.tile = new_tile
|
|
return tile
|
|
end
|
|
else
|
|
if tile.collides_with(collides) then
|
|
return new_tile
|
|
end
|
|
end
|
|
|
|
return tile
|
|
end
|
|
end
|
|
|
|
-- only changes tiles made by the factorio map generator.
|
|
function change_map_gen_tile(builder, old_tile, new_tile)
|
|
return function(x, y, world)
|
|
local function handle_tile(tile)
|
|
if type(tile) == "boolean" and tile then
|
|
local gen_tile = world.surface.get_tile(world.x, world.y).name
|
|
if gen_tile == old_tile then
|
|
return new_tile
|
|
end
|
|
end
|
|
return tile
|
|
end
|
|
|
|
local tile = builder(x, y, world)
|
|
|
|
if type(tile) == "table" then
|
|
tile.tile = handle_tile(tile.tile)
|
|
else
|
|
tile = handle_tile(tile)
|
|
end
|
|
|
|
return tile
|
|
end
|
|
end
|
|
|
|
-- only changes tiles made by the factorio map generator.
|
|
function change_map_gen_collision_tile(builder, collides, new_tile)
|
|
return function(x, y, world)
|
|
local function handle_tile(tile)
|
|
if type(tile) == "boolean" and tile then
|
|
local gen_tile = world.surface.get_tile(world.x, world.y)
|
|
if gen_tile.collides_with(collides) then
|
|
return new_tile
|
|
end
|
|
end
|
|
return tile
|
|
end
|
|
|
|
local tile = builder(x, y, world)
|
|
|
|
if type(tile) == "table" then
|
|
tile.tile = handle_tile(tile.tile)
|
|
else
|
|
tile = handle_tile(tile)
|
|
end
|
|
|
|
return tile
|
|
end
|
|
end
|
|
|
|
local water_tiles = {
|
|
["water"] = true,
|
|
["deepwater"] = true,
|
|
["water-green"] = true,
|
|
["deepwater-green"] = true
|
|
}
|
|
|
|
function spawn_fish(builder, spawn_rate)
|
|
return function(x, y, world)
|
|
local function handle_tile(tile)
|
|
if type(tile) == "string" then
|
|
if water_tiles[tile] and spawn_rate >= math.random() then
|
|
return {name = "fish", position = {world.x, world.y}}
|
|
end
|
|
elseif tile then
|
|
if world.surface.get_tile(world.x, world.y).collides_with("water-tile") and spawn_rate >= math.random() then
|
|
return {name = "fish", position = {world.x, world.y}}
|
|
end
|
|
end
|
|
end
|
|
|
|
local tile = builder(x, y, world)
|
|
|
|
if type(tile) == "table" then
|
|
local entity = handle_tile(tile.tile)
|
|
if entity then
|
|
add_entity(tile, entity)
|
|
end
|
|
else
|
|
local entity = handle_tile(tile)
|
|
if entity then
|
|
tile = {
|
|
tile = tile,
|
|
entities = {entity}
|
|
}
|
|
end
|
|
end
|
|
|
|
return tile
|
|
end
|
|
end
|
|
|
|
--todo change to entiy_module_builder
|
|
function spawn_entity(builder, name)
|
|
return function(x, y, world)
|
|
if builder(x, y, world) then
|
|
return {name = name, position = {world.x, world.y}}
|
|
end
|
|
end
|
|
end
|
|
|
|
function apply_effect(builder, func)
|
|
return function(x, y, world)
|
|
local tile = builder(x, y, world)
|
|
return func(x, y, world, tile)
|
|
end
|
|
end
|
|
|
|
function manhattan_ore_value(base, mult)
|
|
return function(x, y)
|
|
return mult * (math.abs(x) + math.abs(y)) + base
|
|
end
|
|
end
|
|
|
|
function euclidean_ore_value(base, mult)
|
|
return function(x, y)
|
|
return mult * math.sqrt(x * x + y * y) + base
|
|
end
|
|
end
|