Pixel-Composer/scripts/node_flood_fill/node_flood_fill.gml

116 lines
3.4 KiB
Plaintext
Raw Normal View History

2023-02-28 09:43:01 +01:00
function Node_Flood_Fill(_x, _y, _group = noone) : Node_Processor(_x, _y, _group) constructor {
name = "Flood Fill";
inputs[| 0] = nodeValue("Surface in", self, JUNCTION_CONNECT.input, VALUE_TYPE.surface, 0)
.rejectArray();
inputs[| 1] = nodeValue("Mask", self, JUNCTION_CONNECT.input, VALUE_TYPE.surface, 0);
inputs[| 2] = nodeValue("Mix", self, JUNCTION_CONNECT.input, VALUE_TYPE.float, 1)
.setDisplay(VALUE_DISPLAY.slider, [0, 1, 0.01]);
inputs[| 3] = nodeValue("Active", self, JUNCTION_CONNECT.input, VALUE_TYPE.boolean, true);
active_index = 3;
inputs[| 4] = nodeValue("Position", self, JUNCTION_CONNECT.input, VALUE_TYPE.float, [ 1, 1 ])
.setDisplay(VALUE_DISPLAY.vector);
inputs[| 5] = nodeValue("Colors", self, JUNCTION_CONNECT.input, VALUE_TYPE.color, c_black )
inputs[| 6] = nodeValue("Threshold", self, JUNCTION_CONNECT.input, VALUE_TYPE.float, 0.1)
.setDisplay(VALUE_DISPLAY.slider, [0, 1, 0.01]);
inputs[| 7] = nodeValue("Diagonal", self, JUNCTION_CONNECT.input, VALUE_TYPE.boolean, false);
outputs[| 0] = nodeValue("Surface out", self, JUNCTION_CONNECT.output, VALUE_TYPE.surface, noone);
input_display_list = [ 3,
2023-05-03 21:42:17 +02:00
["Output", false], 0, 1, 2,
2023-02-28 09:43:01 +01:00
["Fill", false], 4, 6, 5, 7,
]
temp_surface = [ surface_create(1, 1), surface_create(1, 1) ];
2023-03-19 09:17:39 +01:00
attribute_surface_depth();
2023-06-13 14:42:06 +02:00
attributes.fill_iteration = -1;
2023-03-24 09:32:08 +01:00
array_push(attributeEditors, "Algorithm");
array_push(attributeEditors, ["Fill iteration", "fill_iteration",
2023-04-11 20:29:20 +02:00
new textBox(TEXTBOX_INPUT.number, function(val) {
2023-06-13 14:42:06 +02:00
attributes.fill_iteration = val;
2023-04-11 20:29:20 +02:00
triggerRender();
})]);
2023-03-24 09:32:08 +01:00
2023-02-28 09:43:01 +01:00
static drawOverlay = function(active, _x, _y, _s, _mx, _my, _snx, _sny) {
inputs[| 4].drawOverlay(active, _x, _y, _s, _mx, _my, _snx, _sny);
}
static process_data = function(_outSurf, _data, _output_index, _array_index) {
var inSurf = _data[0];
if(!is_surface(inSurf)) return _outSurf;
var _pos = _data[4];
var _col = _data[5];
var _thr = _data[6];
var _dia = _data[7];
2023-03-31 06:59:08 +02:00
var _filC = surface_get_pixel_ext(inSurf, _pos[0], _pos[1]);
2023-02-28 09:43:01 +01:00
var sw = surface_get_width(inSurf);
var sh = surface_get_height(inSurf);
for( var i = 0; i < array_length(temp_surface); i++ )
2023-03-19 09:17:39 +01:00
temp_surface[i] = surface_verify(temp_surface[i], sw, sh, attrDepth());
2023-02-28 09:43:01 +01:00
surface_set_target(temp_surface[0]);
2023-03-19 09:17:39 +01:00
DRAW_CLEAR
2023-02-28 09:43:01 +01:00
shader_set(sh_flood_fill_thres);
2023-03-25 12:27:04 +01:00
shader_set_f("color", colaToVec4(_filC));
shader_set_f("thres", _thr);
2023-02-28 09:43:01 +01:00
BLEND_OVERRIDE
2023-03-19 09:17:39 +01:00
draw_surface_safe(inSurf, 0, 0);
2023-02-28 09:43:01 +01:00
BLEND_NORMAL
shader_reset();
BLEND_OVERRIDE
draw_set_color(c_red);
draw_point(_pos[0] - 1, _pos[1] - 1);
BLEND_NORMAL
surface_reset_target();
var ind = 0;
2023-06-13 14:42:06 +02:00
var it = attributes.fill_iteration == -1? sw + sh : attributes.fill_iteration;
2023-02-28 09:43:01 +01:00
repeat(it) {
ind = !ind;
surface_set_target(temp_surface[ind]);
2023-03-19 09:17:39 +01:00
DRAW_CLEAR
2023-02-28 09:43:01 +01:00
shader_set(sh_flood_fill_it);
2023-03-25 12:27:04 +01:00
shader_set_f("dimension", [ sw, sh ]);
shader_set_i("diagonal", _dia);
2023-02-28 09:43:01 +01:00
BLEND_OVERRIDE
2023-03-19 09:17:39 +01:00
draw_surface_safe(temp_surface[!ind], 0, 0);
2023-02-28 09:43:01 +01:00
BLEND_NORMAL
shader_reset();
surface_reset_target();
}
surface_set_target(_outSurf);
2023-03-19 09:17:39 +01:00
DRAW_CLEAR
2023-02-28 09:43:01 +01:00
shader_set(sh_flood_fill_replace);
2023-03-25 12:27:04 +01:00
shader_set_f("color", colToVec4(_col));
shader_set_surface("mask", temp_surface[ind]);
2023-02-28 09:43:01 +01:00
BLEND_OVERRIDE
2023-03-19 09:17:39 +01:00
draw_surface_safe(inSurf, 0, 0);
2023-02-28 09:43:01 +01:00
BLEND_NORMAL
shader_reset();
surface_reset_target();
_outSurf = mask_apply(_data[0], _outSurf, _data[1], _data[2]);
return _outSurf;
}
}