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

198 lines
8.4 KiB
Lua
Raw Normal View History

2018-09-08 18:29:27 +02:00
-- this
local Mask = {}
local gaussBlurKernel = {}
gaussBlurKernel[5] = {
2018-09-11 17:39:09 +02:00
{0.003765, 0.015019, 0.023792, 0.015019, 0.003765},
{0.015019, 0.059912, 0.094907, 0.059912, 0.015019},
{0.023792, 0.094907, 0.150342, 0.094907, 0.023792},
{0.015019, 0.059912, 0.094907, 0.059912, 0.015019},
{0.003765, 0.015019, 0.023792, 0.015019, 0.003765}
}
gaussBlurKernel[7] = {
2018-09-13 23:20:30 +02:00
{0.0015, 0.00438, 0.008328, 0.010317, 0.008328, 0.00438, 0.0015},
{0.00438, 0.012788, 0.024314, 0.03012, 0.024314, 0.012788, 0.00438},
{0.008328, 0.024314, 0.046228, 0.057266, 0.046228, 0.024314, 0.008328},
{0.010317, 0.03012, 0.057266, 0.07094, 0.057266, 0.03012, 0.010317},
{0.008328, 0.024314, 0.046228, 0.057266, 0.046228, 0.024314, 0.008328},
{0.00438, 0.012788, 0.024314, 0.03012, 0.024314, 0.012788, 0.00438},
{0.0015, 0.00438, 0.008328, 0.010317, 0.008328, 0.00438, 0.0015}
}
gaussBlurKernel[9] = {
2018-09-13 23:20:30 +02:00
{0.000814, 0.001918, 0.003538, 0.005108, 0.005774, 0.005108, 0.003538, 0.001918, 0.000814},
{0.001918, 0.00452, 0.008338, 0.012038, 0.013605, 0.012038, 0.008338, 0.00452, 0.001918},
{0.003538, 0.008338, 0.015378, 0.022203, 0.025094, 0.022203, 0.015378, 0.008338, 0.003538},
{0.005774, 0.013605, 0.025094, 0.036231, 0.04095, 0.036231, 0.025094, 0.013605, 0.005774},
{0.005108, 0.012038, 0.022203, 0.032057, 0.036231, 0.032057, 0.022203, 0.012038, 0.005108},
{0.005108, 0.012038, 0.022203, 0.032057, 0.036231, 0.032057, 0.022203, 0.012038, 0.005108},
{0.003538, 0.008338, 0.015378, 0.022203, 0.025094, 0.022203, 0.015378, 0.008338, 0.003538},
{0.001918, 0.00452, 0.008338, 0.012038, 0.013605, 0.012038, 0.008338, 0.00452, 0.001918},
{0.000814, 0.001918, 0.003538, 0.005108, 0.005774, 0.005108, 0.003538, 0.001918, 0.000814}
}
gaussBlurKernel[11] = {
2018-09-13 23:20:30 +02:00
{0.000395, 0.000853, 0.001552, 0.002381, 0.003078, 0.003353, 0.003078, 0.002381, 0.001552, 0.000853, 0.000395},
{0.000853, 0.001842, 0.003353, 0.005143, 0.006648, 0.007242, 0.006648, 0.005143, 0.003353, 0.001842, 0.000853},
{0.001552, 0.003353, 0.006103, 0.009361, 0.012101, 0.013182, 0.012101, 0.009361, 0.006103, 0.003353, 0.001552},
{0.002381, 0.005143, 0.009361, 0.014359, 0.018561, 0.020219, 0.018561, 0.014359, 0.009361, 0.005143, 0.002381},
{0.003078, 0.006648, 0.012101, 0.018561, 0.023992, 0.026136, 0.023992, 0.018561, 0.012101, 0.006648, 0.003078},
{0.003353, 0.007242, 0.013182, 0.020219, 0.026136, 0.02847, 0.026136, 0.020219, 0.013182, 0.007242, 0.003353},
{0.003078, 0.006648, 0.012101, 0.018561, 0.023992, 0.026136, 0.023992, 0.018561, 0.012101, 0.006648, 0.003078},
{0.002381, 0.005143, 0.009361, 0.014359, 0.018561, 0.020219, 0.018561, 0.014359, 0.009361, 0.005143, 0.002381},
{0.001552, 0.003353, 0.006103, 0.009361, 0.012101, 0.013182, 0.012101, 0.009361, 0.006103, 0.003353, 0.001552},
{0.000853, 0.001842, 0.003353, 0.005143, 0.006648, 0.007242, 0.006648, 0.005143, 0.003353, 0.001842, 0.000853},
{0.000395, 0.000853, 0.001552, 0.002381, 0.003078, 0.003353, 0.003078, 0.002381, 0.001552, 0.000853, 0.000395}
}
local n = 9;
local radius = math.floor(n / 2)
local middle = radius + 1
local circle_blur_sum = 0
local circleBlurKernel = {}
function init()
local sum = 0
local edge_middle_distance = math.sqrt(2) * (n - middle)
for x = 1, n do
circleBlurKernel[x] = {}
for y = 1, n do
local distance = math.sqrt((x - middle) * (x - middle) + (y - middle) * (y - middle))
if distance <= radius then
sum = sum + 1
circleBlurKernel[x][y] = 1
else
local edge_distance = edge_middle_distance - distance
local normalized_value = edge_distance / (edge_middle_distance - radius)
sum = sum + normalized_value
circleBlurKernel[x][y] = normalized_value
end
end
end
for x = 1, n do
for y = 1, n do
circleBlurKernel[x][y] = circleBlurKernel[x][y] / sum
end
end
sum = 0
for x = 1,n do
for y = 1, n do
local distance = math.sqrt((x - middle) * (x - middle) + (y - middle) * (y - middle))
if distance <= radius then
sum = sum + (radius - distance) / radius + 0.1
end
end
end
circle_blur_sum = sum
end
init()
2018-09-11 17:39:09 +02:00
--[[--
2018-09-11 20:09:06 +02:00
Applies a blur filter.
2018-09-11 17:39:09 +02:00
@param x_start number center point
@param y_start number center point
2018-09-19 18:55:07 +02:00
@param factor number relative strength of the entity to withstand the stress
2018-09-11 17:39:09 +02:00
factor < 0 if entity is placed
factor > 0 if entity is removed
2018-09-11 17:39:09 +02:00
@param callback function to execute on each tile within the mask callback(x, y, value)
]]
2018-09-11 20:09:06 +02:00
function Mask.blur(x_start, y_start, factor, callback)
x_start = math.floor(x_start)
y_start = math.floor(y_start)
local filter = gaussBlurKernel[n]
2018-09-11 17:39:09 +02:00
local offset = - math.floor(n / 2) - 1 --move matrix over x_start|y_start and adjust for 1 index
for x = 1, n do
for y = 1, n do
cell = filter[x][y]
value = factor * cell
2018-09-14 14:15:25 +02:00
if math.abs(value) > 0.001 then
callback(x_start + x + offset, y_start + y + offset, value)
end
2018-09-11 17:39:09 +02:00
end
end
end
--[[--
Applies a circular blur
All values outside the circle are proportional to the distance to the center.
The circle radius is math.floor(n / 2)
The sum of all values is 1
@param x_start number center point
@param y_start number center point
@param factor the factor to multiply the cell value with (value = cell_value * factor)
@param callback function to execute on each tile within the mask callback(x, y, value)
]]
function Mask.circle_blur(x_start, y_start, factor, callback)
x_start = math.floor(x_start)
y_start = math.floor(y_start)
local offset = - math.floor(n / 2) - 1 --move matrix over x_start|y_start and adjust for 1 index
for x = 1,n do
for y = 1, n do
local distance = math.sqrt((x - middle) * (x - middle) + (y - middle) * (y - middle))
if distance <= radius then
local value = (radius - distance) / radius / circle_blur_sum * factor
callback(x_start + x + offset, y_start + y + offset, value)
end
end
end
end
--[[--
Applies a circular box blur
All values withing the radius of the filters are equal. All values outside the circle are proportional to the distance to the circle.
The circle radius is math.floor(n / 2)
The sum of all values is 1
@param x_start number center point
@param y_start number center point
@param factor the factor to multiply the cell value with (value = cell_value * factor)
@param callback function to execute on each tile within the mask callback(x, y, value)
]]
function Mask.box_blur(x_start, y_start, factor, callback)
x_start = math.floor(x_start)
y_start = math.floor(y_start)
local filter = circleBlurKernel
local offset = - math.floor(n / 2) - 1 --move matrix over x_start|y_start and adjust for 1 index
for x = 1, n do
for y = 1, n do
cell = filter[x][y]
value = factor * cell
if math.abs(value) > 0.001 then
callback(x_start + x + offset, y_start + y + offset, value)
end
end
end
end
2018-09-11 17:39:09 +02:00
2018-09-08 18:29:27 +02:00
--[[--
Masks in the shape of a circle.
@param x_start number, center point
@param y_start number, center point
@param diameter number size of the circle
@param callback function to execute on each tile within the circle callback(x, y, tile_distance_to_center, diameter)
]]
function Mask.circle(x_start, y_start, diameter, callback)
for x = (diameter * -1), diameter, 1 do
for y = (diameter * -1), diameter, 1 do
local tile_distance_to_center = math.floor(math.sqrt(x^2 + y^2))
if (tile_distance_to_center < diameter) then
callback(x + x_start, y + y_start, tile_distance_to_center, diameter)
end
end
end
end
return Mask