Pixel-Composer/scripts/node_find_pixel/node_find_pixel.gml
2024-03-14 20:35:19 +07:00

90 lines
No EOL
2.7 KiB
Text

function Node_Find_Pixel(_x, _y, _group = noone) : Node_Processor(_x, _y, _group) constructor {
name = "Find pixel";
w = 96;
inputs[| 0] = nodeValue("Surface in", self, JUNCTION_CONNECT.input, VALUE_TYPE.surface, noone);
inputs[| 1] = nodeValue("Search color", self, JUNCTION_CONNECT.input, VALUE_TYPE.color, c_black);
inputs[| 2] = nodeValue("Tolerance", self, JUNCTION_CONNECT.input, VALUE_TYPE.float, 0)
.setDisplay(VALUE_DISPLAY.slider);
inputs[| 3] = nodeValue("Find all", self, JUNCTION_CONNECT.input, VALUE_TYPE.boolean, false);
inputs[| 4] = nodeValue("Include alpha", self, JUNCTION_CONNECT.input, VALUE_TYPE.boolean, false);
inputs[| 5] = nodeValue("Alpha tolerance", self, JUNCTION_CONNECT.input, VALUE_TYPE.float, 0.2)
.setDisplay(VALUE_DISPLAY.slider);
outputs[| 0] = nodeValue("Position", self, JUNCTION_CONNECT.output, VALUE_TYPE.integer, [ 0, 0 ])
.setDisplay(VALUE_DISPLAY.vector);
input_display_list = [ 0,
["Search", false], 1, 2, 3,
["Alpha", true, 4], 5,
]
static getPreviewValues = function() { return getInputData(0); }
temp_surface = [ surface_create(1, 1) ];
static processData = function(_output, _data, _output_index, _array_index = 0) { #region
var _surf = _data[0];
var _col = _data[1];
var _tol = _data[2];
var _all = _data[3];
var _alp = _data[4];
var _alpT = _data[5];
if(!is_surface(_surf)) return [0, 0];
var _buff = buffer_from_surface(_surf, false);
var _sw = surface_get_width_safe(_surf);
var _sh = surface_get_height_safe(_surf);
buffer_seek(_buff, buffer_seek_start, 0);
var res = [];
var r = _color_get_red(_col);
var g = _color_get_green(_col);
var b = _color_get_blue(_col);
var a = _color_get_alpha(_col);
for( var i = 0; i < _sh; i++ )
for( var j = 0; j < _sw; j++ ) {
var _c = buffer_read(_buff, buffer_u32);
var _r = ((_c & 0x000000FF) >> 0) / 255;
var _g = ((_c & 0x0000FF00) >> 8) / 255;
var _b = ((_c & 0x00FF0000) >> 16) / 255;
var _a = ((_c & 0xFF000000) >> 24) / 255;
if(!_alp && _a == 0) continue;
var colMatch = (abs(r - _r) + abs(g - _g) + abs(b - _b)) / 3 <= _tol;
if(!colMatch) continue;
if(!_alp || abs(a - _a) <= _alpT) {
if(_all) array_push(res, [ j, i ]);
else return [ j, i ];
}
}
buffer_delete(_buff);
return _all? res : [ -1, -1 ];
} #endregion
static onDrawNode = function(xx, yy, _mx, _my, _s, _hover, _focus) { #region
var bbox = drawGetBbox(xx, yy, _s);
var col = getInputData(1);
if(bbox.h <= 0) return;
if(is_array(col)) {
drawPalette(col, bbox.x0, bbox.y0, bbox.w, bbox.h);
return;
}
drawColor(col, bbox.x0, bbox.y0, bbox.w, bbox.h);
} #endregion
}