Pixel-Composer/scripts/node_rgb_channel/node_rgb_channel.gml

83 lines
2.3 KiB
Plaintext
Raw Normal View History

2023-02-28 09:43:01 +01:00
function Node_RGB_Channel(_x, _y, _group = noone) : Node_Processor(_x, _y, _group) constructor {
2023-02-14 05:32:32 +01:00
name = "RGBA Extract";
batch_output = false;
2022-01-13 05:24:03 +01:00
2024-08-18 06:16:20 +02:00
newInput(0, nodeValue_Surface("Surface In", self));
2022-01-13 05:24:03 +01:00
2024-08-18 06:16:20 +02:00
newInput(1, nodeValue_Enum_Scroll("Output Type", self, 0, [ "Channel value", "Greyscale" ]));
2023-08-04 13:12:32 +02:00
2024-08-18 06:16:20 +02:00
newInput(2, nodeValue_Bool("Keep Alpha", self, false));
2023-01-17 08:11:55 +01:00
2024-08-18 06:16:20 +02:00
newInput(3, nodeValue_Bool("Output Array", self, false));
2024-03-19 09:49:29 +01:00
2024-09-04 03:57:11 +02:00
newOutput(0, nodeValue_Output("Red", self, VALUE_TYPE.surface, noone));
newOutput(1, nodeValue_Output("Green", self, VALUE_TYPE.surface, noone));
newOutput(2, nodeValue_Output("Blue", self, VALUE_TYPE.surface, noone));
newOutput(3, nodeValue_Output("Alpha", self, VALUE_TYPE.surface, noone));
2022-01-13 05:24:03 +01:00
2023-03-19 09:17:39 +01:00
attribute_surface_depth();
2024-03-19 09:49:29 +01:00
static step = function() { #region
var _arr = getInputData(3);
2024-08-08 06:57:51 +02:00
outputs[0].name = _arr? "RGBA" : "Red";
outputs[0].setArrayDepth(_arr);
2024-03-19 09:49:29 +01:00
2024-08-08 06:57:51 +02:00
outputs[1].setVisible(!_arr, !_arr);
outputs[2].setVisible(!_arr, !_arr);
outputs[3].setVisible(!_arr, !_arr);
2024-03-19 09:49:29 +01:00
} #endregion
static setShader = function(index, grey, _alp) {
DRAW_CLEAR
BLEND_OVERRIDE
switch(index) {
case 0 : shader_set(grey? sh_channel_R_grey : sh_channel_R); break;
case 1 : shader_set(grey? sh_channel_G_grey : sh_channel_G); break;
case 2 : shader_set(grey? sh_channel_B_grey : sh_channel_B); break;
case 3 : shader_set(grey? sh_channel_A_grey : sh_channel_A); break;
}
shader_set_i("keepAlpha", _alp);
}
static resetShader = function() {
shader_reset();
BLEND_NORMAL
}
2023-09-14 16:29:39 +02:00
static processData = function(_outSurf, _data, output_index) { #region
2023-01-17 08:11:55 +01:00
var _out = _data[1];
2023-08-04 13:12:32 +02:00
var _alp = _data[2];
2024-03-19 09:49:29 +01:00
var _arr = _data[3];
2023-01-17 08:11:55 +01:00
2024-03-19 09:49:29 +01:00
if(_arr && output_index) return _outSurf;
var _ww = surface_get_width_safe(_data[0]);
var _hh = surface_get_height_safe(_data[0]);
if(_arr) {
for( var i = 0; i < 4; i++ ) {
2024-03-31 05:36:11 +02:00
var _surf = array_safe_get_fast(_outSurf, i);
2024-03-19 09:49:29 +01:00
_surf = surface_verify(_surf, _ww, _hh);
_outSurf[i] = _surf;
surface_set_target(_surf);
setShader(i, _out, _alp);
draw_surface_safe(_data[0]);
resetShader();
surface_reset_target();
2022-01-13 05:24:03 +01:00
}
2023-08-04 13:12:32 +02:00
2024-03-19 09:49:29 +01:00
} else {
surface_set_target(_outSurf);
setShader(output_index, _out, _alp);
draw_surface_safe(_data[0]);
resetShader();
surface_reset_target();
}
2022-01-13 05:24:03 +01:00
return _outSurf;
2023-09-14 16:29:39 +02:00
} #endregion
2022-01-13 05:24:03 +01:00
}