New Error Diffuse Dither node.

This commit is contained in:
Tanasart 2024-12-12 15:31:39 +07:00
parent efd13eb5fc
commit b0f80c5572

View file

@ -1,3 +1,54 @@
// This code is bad.
//!#mfunc __error_diffuse {"args":["i"],"order":[0,0,0,0]}
#macro __error_diffuse_mf0 buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e0 *
#macro __error_diffuse_mf1 ); _pl += 2; \
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e1 *
#macro __error_diffuse_mf2 ); _pl += 2; \
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e2 *
#macro __error_diffuse_mf3 ); _pl += 2; \
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e3 *
#macro __error_diffuse_mf4 ); _pl += 2;
#macro __err_diffuse_write buffer_seek(_bi, buffer_seek_start, _p); \
var _o0 = buffer_read(_bi, buffer_s16); \
var _o1 = buffer_read(_bi, buffer_s16); \
var _o2 = buffer_read(_bi, buffer_s16); \
var _o3 = buffer_read(_bi, buffer_s16); \
\
var _d0 = _o0 > 128? 255 : 0; \
var _d1 = _o1 > 128? 255 : 0; \
var _d2 = _o2 > 128? 255 : 0; \
var _d3 = _o3 > 128? 255 : 0; \
\
var _e0 = _o0 - _d0; \
var _e1 = _o1 - _d1; \
var _e2 = _o2 - _d2; \
var _e3 = _o3 - _d3; \
\
buffer_write(_bo, buffer_u8, _d0); \
buffer_write(_bo, buffer_u8, _d1); \
buffer_write(_bo, buffer_u8, _d2); \
buffer_write(_bo, buffer_u8, _d3);
//!#mfunc __error_diffuse_grey {"args":["i"],"order":[0]}
#macro __error_diffuse_grey_mf0 buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e0 *
#macro __error_diffuse_grey_mf1 ); _pl += 2;
#macro __err_diffuse_write_grey buffer_seek(_bi, buffer_seek_start, _p); \
var _o0 = buffer_read(_bi, buffer_s16); \
var _o1 = buffer_read(_bi, buffer_s16); \
var _o2 = buffer_read(_bi, buffer_s16); \
var _o3 = buffer_read(_bi, buffer_s16); \
\
var _d0 = _o0 > 128? 255 : 0; \
var _e0 = _o0 - _d0; \
\
buffer_write(_bo, buffer_u8, _d0); \
buffer_write(_bo, buffer_u8, _d0); \
buffer_write(_bo, buffer_u8, _d0); \
buffer_write(_bo, buffer_u8, 255);
function Node_Dither_Diffuse(_x, _y, _group = noone) : Node_Processor(_x, _y, _group) constructor {
name = "Error Diffuse Dither";
@ -15,15 +66,17 @@ function Node_Dither_Diffuse(_x, _y, _group = noone) : Node_Processor(_x, _y, _g
__init_mask_modifier(1); // inputs 5, 6,
newInput(7, nodeValue_Enum_Scroll("Type", self, 0, [ "Floyd-Steinberg" ]));
newInput(7, nodeValue_Enum_Scroll("Type", self, 0, { data: [ "Floyd-Steinberg", "Jarvis, Judice, and Ninke", "Atkinson" ], update_hover: false }));
newInput(8, nodeValueSeed(self));
newInput(9, nodeValue_Bool("Greyscale", self, false));
newOutput(0, nodeValue_Output("Surface out", self, VALUE_TYPE.surface, noone));
input_display_list = [ 3, 4, 8,
["Surfaces", true], 0, 1, 2, 5, 6,
["Dither", false], 7,
["Dither", false], 7, 9,
];
static step = function() {
@ -34,106 +87,190 @@ function Node_Dither_Diffuse(_x, _y, _group = noone) : Node_Processor(_x, _y, _g
var _surf = _data[0];
var _type = _data[7];
var _seed = _data[8];
var _bw = _data[9];
if(!is_surface(_surf)) return _outSurf;
var _sw = surface_get_width(_surf);
var _sh = surface_get_height(_surf);
var _a = _sw * _sh;
_outSurf = surface_verify(_outSurf, _sw, _sh);
#region buffer preparation
var _sw = surface_get_width(_surf);
var _sh = surface_get_height(_surf);
var _a = _sw * _sh;
_outSurf = surface_verify(_outSurf, _sw, _sh);
var _b = buffer_from_surface(_surf, false); buffer_to_start(_b);
var _bi = buffer_create(_a * 8, buffer_fixed, 1); buffer_to_start(_bi);
var _bo = buffer_create(_a * 4, buffer_fixed, 1); buffer_to_start(_bo);
repeat(_a) {
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
}
buffer_to_start(_b);
buffer_to_start(_bi);
buffer_to_start(_bo);
#endregion
var _p = 0, _pl;
var _x = 0;
var _y = 0;
var _s = _sw * 8;
var _b = buffer_from_surface(_surf, false); buffer_to_start(_b);
var _bi = buffer_create(_a * 8, buffer_fixed, 1); buffer_to_start(_bi);
var _bo = buffer_create(_a * 4, buffer_fixed, 1); buffer_to_start(_bo);
repeat(_a) {
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
buffer_write(_bi, buffer_s16, buffer_read(_b, buffer_u8));
}
buffer_to_start(_b);
buffer_to_start(_bi);
buffer_to_start(_bo);
var _i = 0;
var _p = 0;
var _x = 0;
var _y = 0;
var _k1 = 7 / 16;
var _k2 = 3 / 16;
var _k3 = 5 / 16;
var _k4 = 1 / 16;
repeat(_a) {
var _prop_l = _x > 0;
var _prop_r = _x < _sw - 1;
var _prop_d = _y < _sh - 1;
buffer_seek(_bi, buffer_seek_start, _p * 2);
var _o0 = buffer_read(_bi, buffer_s16);
var _o1 = buffer_read(_bi, buffer_s16);
var _o2 = buffer_read(_bi, buffer_s16);
var _o3 = buffer_read(_bi, buffer_s16);
var _d0 = _o0 > 128? 255 : 0;
var _d1 = _o1 > 128? 255 : 0;
var _d2 = _o2 > 128? 255 : 0;
var _d3 = _o3 > 128? 255 : 0;
var _e0 = _o0 - _d0;
var _e1 = _o1 - _d1;
var _e2 = _o2 - _d2;
var _e3 = _o3 - _d3;
buffer_write(_bo, buffer_u8, _d0);
buffer_write(_bo, buffer_u8, _d1);
buffer_write(_bo, buffer_u8, _d2);
buffer_write(_bo, buffer_u8, _d3);
if(_prop_r) {
var _pl = _p * 2 + 8;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e0 * _k1); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e1 * _k1); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e2 * _k1); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e3 * _k1); _pl += 2;
}
if(_prop_d) {
var _pl = _p * 2 + _sw * 8;
if(_prop_l) {
_pl -= 8;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e0 * _k2); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e1 * _k2); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e2 * _k2); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e3 * _k2); _pl += 2;
}
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e0 * _k3); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e1 * _k3); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e2 * _k3); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e3 * _k3); _pl += 2;
if(_prop_r) {
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e0 * _k4); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e1 * _k4); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e2 * _k4); _pl += 2;
buffer_poke(_bi, _pl, buffer_s16, buffer_peek(_bi, _pl, buffer_s16) + _e3 * _k4); _pl += 2;
}
}
_p += 4;
_x++;
if(_x >= _sw) {
_x = 0;
_y++;
}
if(_bw) {
if(_type == 0) { // Floyd-Steinberg
var _k1 = 7 / 16, _k2 = 3 / 16, _k3 = 5 / 16, _k4 = 1 / 16;
repeat(_a) {
__err_diffuse_write_grey
if(_x < _sw - 1) { _pl = _p + 8; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1; }
if(_y < _sh - 1) {
if(_x > 0) { _pl = _p + _s - 8; __error_diffuse_grey_mf0 _k2 __error_diffuse_grey_mf1; }
_pl = _p + _s; __error_diffuse_grey_mf0 _k3 __error_diffuse_grey_mf1;
if(_x < _sw - 1) { _pl = _p + _s + 8; __error_diffuse_grey_mf0 _k4 __error_diffuse_grey_mf1; }
}
_p += 8; _x++; if(_x >= _sw) { _x = 0; _y++; }
}
} else if(_type == 1) { // Jarvis, Judice, and Ninke
var _k1 = 7 / 48, _k2 = 5 / 48;
var _k3 = 3 / 48, _k4 = 1 / 48, _k5 = 7 / 48, _k6 = 5 / 48, _k7 = 3 / 48;
var _k8 = 1 / 48, _k9 = 3 / 48, _k10 = 5 / 48, _k11 = 3 / 48, _k12 = 1 / 48;
repeat(_a) {
__err_diffuse_write_grey
if(_x < _sw - 1) {
_pl = _p + 8; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1;
if(_x < _sw - 2) { _pl = _p + 16; __error_diffuse_grey_mf0 _k2 __error_diffuse_grey_mf1; }
}
if(_y < _sh - 1) {
if(_x > 1) { _pl = _p + _s - 16; __error_diffuse_grey_mf0 _k3 __error_diffuse_grey_mf1; }
if(_x > 0) { _pl = _p + _s - 8; __error_diffuse_grey_mf0 _k4 __error_diffuse_grey_mf1; }
_pl = _p + _s; __error_diffuse_grey_mf0 _k5 __error_diffuse_grey_mf1;
if(_x < _sw - 1) { _pl = _p + _s + 8; __error_diffuse_grey_mf0 _k6 __error_diffuse_grey_mf1; }
if(_x < _sw - 2) { _pl = _p + _s + 16; __error_diffuse_grey_mf0 _k7 __error_diffuse_grey_mf1; }
}
if(_y < _sh - 2) {
if(_x > 1) { _pl = _p + _s * 2 - 16; __error_diffuse_grey_mf0 _k8 __error_diffuse_grey_mf1; }
if(_x > 0) { _pl = _p + _s * 2 - 8; __error_diffuse_grey_mf0 _k9 __error_diffuse_grey_mf1; }
_pl = _p + _s * 2; __error_diffuse_grey_mf0 _k10 __error_diffuse_grey_mf1;
if(_x < _sw - 1) { _pl = _p + _s * 2 + 8; __error_diffuse_grey_mf0 _k11 __error_diffuse_grey_mf1; }
if(_x < _sw - 2) { _pl = _p + _s * 2 + 16; __error_diffuse_grey_mf0 _k12 __error_diffuse_grey_mf1; }
}
_p += 8; _x++; if(_x >= _sw) { _x = 0; _y++; }
}
} else if(_type == 2) { // Atkinson
var _k1 = 1 / 8;
repeat(_a) {
__err_diffuse_write_grey
if(_x < _sw - 1) {
_pl = _p + 8; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1;
if(_x < _sw - 2) { _pl = _p + 16; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1; }
}
if(_y < _sh - 1) {
if(_x > 0) { _pl = _p + _s - 8; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1; }
_pl = _p + _s; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1;
if(_x < _sw - 1) { _pl = _p + _s + 8; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1; }
}
if(_y < _sh - 2) { _pl = _p + _s * 2; __error_diffuse_grey_mf0 _k1 __error_diffuse_grey_mf1; }
_p += 8; _x++; if(_x >= _sw) { _x = 0; _y++; }
}
}
} else {
if(_type == 0) { // Floyd-Steinberg
var _k1 = 7 / 16, _k2 = 3 / 16, _k3 = 5 / 16, _k4 = 1 / 16;
repeat(_a) {
__err_diffuse_write
if(_x < _sw - 1) { _pl = _p + 8; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4; }
if(_y < _sh - 1) {
if(_x > 0) { _pl = _p + _s - 8; __error_diffuse_mf0 _k2 __error_diffuse_mf1 _k2 __error_diffuse_mf2 _k2 __error_diffuse_mf3 _k2 __error_diffuse_mf4; }
_pl = _p + _s; __error_diffuse_mf0 _k3 __error_diffuse_mf1 _k3 __error_diffuse_mf2 _k3 __error_diffuse_mf3 _k3 __error_diffuse_mf4;
if(_x < _sw - 1) { _pl = _p + _s + 8; __error_diffuse_mf0 _k4 __error_diffuse_mf1 _k4 __error_diffuse_mf2 _k4 __error_diffuse_mf3 _k4 __error_diffuse_mf4; }
}
_p += 8; _x++; if(_x >= _sw) { _x = 0; _y++; }
}
} else if(_type == 1) { // Jarvis, Judice, and Ninke
var _k1 = 7 / 48, _k2 = 5 / 48;
var _k3 = 3 / 48, _k4 = 1 / 48, _k5 = 7 / 48, _k6 = 5 / 48, _k7 = 3 / 48;
var _k8 = 1 / 48, _k9 = 3 / 48, _k10 = 5 / 48, _k11 = 3 / 48, _k12 = 1 / 48;
repeat(_a) {
__err_diffuse_write
if(_x < _sw - 1) {
_pl = _p + 8; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4;
if(_x < _sw - 2) { _pl = _p + 16; __error_diffuse_mf0 _k2 __error_diffuse_mf1 _k2 __error_diffuse_mf2 _k2 __error_diffuse_mf3 _k2 __error_diffuse_mf4; }
}
if(_y < _sh - 1) {
if(_x > 1) { _pl = _p + _s - 16; __error_diffuse_mf0 _k3 __error_diffuse_mf1 _k3 __error_diffuse_mf2 _k3 __error_diffuse_mf3 _k3 __error_diffuse_mf4; }
if(_x > 0) { _pl = _p + _s - 8; __error_diffuse_mf0 _k4 __error_diffuse_mf1 _k4 __error_diffuse_mf2 _k4 __error_diffuse_mf3 _k4 __error_diffuse_mf4; }
_pl = _p + _s; __error_diffuse_mf0 _k5 __error_diffuse_mf1 _k5 __error_diffuse_mf2 _k5 __error_diffuse_mf3 _k5 __error_diffuse_mf4;
if(_x < _sw - 1) { _pl = _p + _s + 8; __error_diffuse_mf0 _k6 __error_diffuse_mf1 _k6 __error_diffuse_mf2 _k6 __error_diffuse_mf3 _k6 __error_diffuse_mf4; }
if(_x < _sw - 2) { _pl = _p + _s + 16; __error_diffuse_mf0 _k7 __error_diffuse_mf1 _k7 __error_diffuse_mf2 _k7 __error_diffuse_mf3 _k7 __error_diffuse_mf4; }
}
if(_y < _sh - 2) {
if(_x > 1) { _pl = _p + _s * 2 - 16; __error_diffuse_mf0 _k8 __error_diffuse_mf1 _k8 __error_diffuse_mf2 _k8 __error_diffuse_mf3 _k8 __error_diffuse_mf4; }
if(_x > 0) { _pl = _p + _s * 2 - 8; __error_diffuse_mf0 _k9 __error_diffuse_mf1 _k9 __error_diffuse_mf2 _k9 __error_diffuse_mf3 _k9 __error_diffuse_mf4; }
_pl = _p + _s * 2; __error_diffuse_mf0 _k10 __error_diffuse_mf1 _k10 __error_diffuse_mf2 _k10 __error_diffuse_mf3 _k10 __error_diffuse_mf4;
if(_x < _sw - 1) { _pl = _p + _s * 2 + 8; __error_diffuse_mf0 _k11 __error_diffuse_mf1 _k11 __error_diffuse_mf2 _k11 __error_diffuse_mf3 _k11 __error_diffuse_mf4; }
if(_x < _sw - 2) { _pl = _p + _s * 2 + 16; __error_diffuse_mf0 _k12 __error_diffuse_mf1 _k12 __error_diffuse_mf2 _k12 __error_diffuse_mf3 _k12 __error_diffuse_mf4; }
}
_p += 8; _x++; if(_x >= _sw) { _x = 0; _y++; }
}
} else if(_type == 2) { // Atkinson
var _k1 = 1 / 8;
repeat(_a) {
__err_diffuse_write
if(_x < _sw - 1) {
_pl = _p + 8; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4;
if(_x < _sw - 2) { _pl = _p + 16; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4; }
}
if(_y < _sh - 1) {
if(_x > 0) { _pl = _p + _s - 8; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4; }
_pl = _p + _s; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4;
if(_x < _sw - 1) { _pl = _p + _s + 8; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4; }
}
if(_y < _sh - 2) { _pl = _p + _s * 2; __error_diffuse_mf0 _k1 __error_diffuse_mf1 _k1 __error_diffuse_mf2 _k1 __error_diffuse_mf3 _k1 __error_diffuse_mf4; }
_p += 8; _x++; if(_x >= _sw) { _x = 0; _y++; }
}
}
}
buffer_set_surface(_bo, _outSurf, 0);