Pixel-Composer/scripts/node_rd/node_rd.gml
2024-08-08 11:57:51 +07:00

142 lines
4.0 KiB
Plaintext

function Node_RD(_x, _y, _group = noone) : Node(_x, _y, _group) constructor {
name = "Reaction Diffusion";
inputs[0] = nodeValue_Surface("Seed", self);
inputs[1] = nodeValue_Float("Kill rate", self, 0.058)
.setDisplay(VALUE_DISPLAY.slider, { range: [ 0, 0.1, 0.001] })
.setMappable(8);
inputs[2] = nodeValue_Float("Feed rate", self, 0.043)
.setDisplay(VALUE_DISPLAY.slider, { range: [ 0, 0.1, 0.001] })
.setMappable(9);
inputs[3] = nodeValue_Float("Timestep", self, 1)
.setMappable(10);
inputs[4] = nodeValue_Int("Iteration", self, 16);
inputs[5] = nodeValue_Float("Diffusion A", self, 1.)
.setDisplay(VALUE_DISPLAY.slider)
.setMappable(11);
inputs[6] = nodeValue_Float("Diffusion B", self, .2)
.setDisplay(VALUE_DISPLAY.slider)
.setMappable(12);
inputs[7] = nodeValue_Surface("Add B", self);
//////////////////////////////////////////////////////////////////////////////////////////////////////////
inputs[8] = nodeValue_Surface("Kill map", self)
.setVisible(false, false);
inputs[9] = nodeValue_Surface("Feed map", self)
.setVisible(false, false);
inputs[10] = nodeValue_Surface("Time map", self)
.setVisible(false, false);
inputs[11] = nodeValue_Surface("DfA map", self)
.setVisible(false, false);
inputs[12] = nodeValue_Surface("DfB map", self)
.setVisible(false, false);
//////////////////////////////////////////////////////////////////////////////////////////////////////////
inputs[13] = nodeValue_Float("Diffusion", self, 1.)
.setDisplay(VALUE_DISPLAY.slider)
outputs[0] = nodeValue_Output("Reacted", self, VALUE_TYPE.surface, noone);
outputs[1] = nodeValue_Output("Rendered", self, VALUE_TYPE.surface, noone);
preview_channel = 1;
input_display_list = [
["Surfaces", false], 0, 7,
["Diffusion", false], 13, 5, 11, 6, 12,
["Paramaters", false], 1, 8, 2, 9,
["Simulation", false], 3, 10, 4,
];
attribute_surface_depth();
temp_surface = [ surface_create(1, 1), surface_create(1, 1) ];
static step = function() { #region
inputs[1].mappableStep();
inputs[2].mappableStep();
inputs[3].mappableStep();
inputs[5].mappableStep();
inputs[6].mappableStep();
} #endregion
static update = function() {
var _surf = getInputData(0);
var _k = getInputData(1);
var _f = getInputData(2);
var _dt = getInputData(3);
var _it = getInputData(4);
var _dd = getInputData(13);
var _da = getInputData(5);
var _db = getInputData(6);
var _b = getInputData(7);
var _outp = outputs[0].getValue();
var _rend = outputs[1].getValue();
var _sw = surface_get_width_safe(_surf);
var _sh = surface_get_height_safe(_surf);
_outp = surface_verify(_outp, _sw, _sh);
_rend = surface_verify(_rend, _sw, _sh);
temp_surface[0] = surface_verify(temp_surface[0], _sw, _sh);
temp_surface[1] = surface_verify(temp_surface[1], _sw, _sh);
surface_set_shader(temp_surface[0], sh_rd_convert);
draw_surface_safe(_surf);
surface_reset_shader();
if(is_surface(_b)) {
surface_set_target(temp_surface[0]);
gpu_set_colorwriteenable(0, 1, 0, 0);
BLEND_ADD
draw_surface_safe(_b);
BLEND_NORMAL
gpu_set_colorwriteenable(1, 1, 1, 1);
surface_reset_target();
}
var _ind = 0;
repeat(_it) {
surface_set_shader(temp_surface[!_ind], sh_rd_propagate);
shader_set_f("dimension", _sw, _sh);
shader_set_f_map("k", _k , getInputData( 8), inputs[1]);
shader_set_f_map("f", _f , getInputData( 9), inputs[2]);
shader_set_f_map("dt", _dt, getInputData(10), inputs[3]);
shader_set_f("dd", _dd);
shader_set_f_map("da", _da, getInputData(11), inputs[5]);
shader_set_f_map("db", _db, getInputData(12), inputs[6]);
draw_surface_safe(temp_surface[_ind]);
surface_reset_shader();
_ind = !_ind;
}
surface_set_shader(_outp);
draw_surface_safe(temp_surface[_ind]);
surface_reset_shader();
surface_set_shader(_rend, sh_rd_render);
draw_surface_safe(_surf);
surface_reset_shader();
outputs[0].setValue(_outp);
outputs[1].setValue(_rend);
}
}