Pixel-Composer/scripts/node_array_convolute/node_array_convolute.gml

89 lines
2 KiB
Text
Raw Normal View History

function Node_Array_Convolute(_x, _y, _group = noone) : Node(_x, _y, _group) constructor {
2024-03-28 14:18:02 +01:00
name = "Array Convolute";
setDimension(96, 32 + 24);
2024-10-01 05:46:28 +02:00
newInput(0, nodeValue_Float("Array", self, []))
.setArrayDepth(1)
.setVisible(true, true);
2024-08-18 09:13:41 +02:00
newInput(1, nodeValue_Float("Kernel", self, []))
.setArrayDepth(1)
.setVisible(true, true);
2024-10-01 05:46:28 +02:00
newInput(2, nodeValue_Enum_Scroll("Boundary", self, 0, [ "Zero", "Wrap", "Skip" ]))
.setArrayDepth(1);
2024-09-04 03:57:11 +02:00
newOutput(0, nodeValue_Output("Array", self, VALUE_TYPE.float, 0))
.setArrayDepth(1);
static convolute = function(arr, ker) {
2024-10-01 05:46:28 +02:00
var _bnd = getInputData(2);
var _len = array_length(ker);
var _arn = array_length(arr);
var _st = floor((_len - 1) / 2);
var r, _a;
if(_bnd == 2) {
var _ll = _arn - _len + 1;
_a = array_create(_ll);
2024-10-01 05:46:28 +02:00
for(var i = 0; i < _ll; i++ ) {
r = 0;
for(var j = 0; j < _len; j++) {
var _ind = i + j;
if(_ind < 0 || _ind >= _arn) continue;
r += arr[_ind] * ker[j];
}
2024-10-01 05:46:28 +02:00
_a[i] = r;
}
2024-10-01 05:46:28 +02:00
} else {
_a = array_create(_arn);
for( var i = 0; i < _arn; i++ ) {
r = 0;
for(var j = 0; j < _len; j++) {
var _ind = i + j - _st;
if(_ind < 0 || _ind >= _arn) {
if(_bnd == 0) continue;
_ind = safe_mod(_ind + _arn, _arn);
}
r += arr[_ind] * ker[j];
}
_a[i] = r;
}
}
return _a;
}
2023-10-09 16:07:33 +02:00
static update = function(frame = CURRENT_FRAME) {
var _arr = getInputData(0);
var _ker = getInputData(1);
2024-10-01 05:46:28 +02:00
if(!is_array(_arr) || !is_array(_ker)) return;
if(array_empty(_arr) || array_empty(_ker)) return;
var res;
if(is_array(_arr[0])) {
for( var i = 0, n = array_length(_arr); i < n; i++ )
res[i] = convolute(_arr[i], _ker);
} else
res = convolute(_arr, _ker);
2024-08-08 06:57:51 +02:00
outputs[0].setValue(res);
}
static onDrawNode = function(xx, yy, _mx, _my, _s, _hover, _focus) {
var bbox = drawGetBbox(xx, yy, _s);
draw_sprite_fit(s_node_array_convolute, 0, bbox.xc, bbox.yc, bbox.w, bbox.h);
}
}