Lubos Lenco
92 lines
5.2 KiB
Haxe
92 lines
5.2 KiB
Haxe
package arm.node;
|
|
|
|
import arm.shader.NodeShader;
|
|
|
|
class MakeBlur {
|
|
|
|
public static function run(vert: NodeShader, frag: NodeShader) {
|
|
frag.write('vec2 gbufferSizeLocal = gbufferSize;'); // TODO: spirv workaround
|
|
#if (kha_direct3d11 || kha_direct3d12 || kha_metal || kha_vulkan)
|
|
frag.write('vec2 texCoordInp = texelFetch(gbuffer2, ivec2(sp.x * gbufferSizeLocal.x, sp.y * gbufferSizeLocal.y), 0).ba;');
|
|
#else
|
|
frag.write('vec2 texCoordInp = texelFetch(gbuffer2, ivec2(sp.x * gbufferSizeLocal.x, (1.0 - sp.y) * gbufferSizeLocal.y), 0).ba;');
|
|
#end
|
|
|
|
frag.write('vec3 basecol = vec3(0.0, 0.0, 0.0);');
|
|
frag.write('float roughness = 0.0;');
|
|
frag.write('float metallic = 0.0;');
|
|
frag.write('float occlusion = 0.0;');
|
|
frag.write('vec3 nortan = vec3(0.0, 0.0, 0.0);');
|
|
frag.write('float height = 0.0;');
|
|
frag.write('float mat_opacity = 1.0;');
|
|
frag.write('float opacity = 0.0;');
|
|
if (Context.material.paintEmis) {
|
|
frag.write('float emis = 0.0;');
|
|
}
|
|
if (Context.material.paintSubs) {
|
|
frag.write('float subs = 0.0;');
|
|
}
|
|
|
|
frag.add_uniform('vec2 texpaintSize', '_texpaintSize');
|
|
frag.write('vec2 texpaintSizeLocal = texpaintSize;'); // TODO: spirv workaround
|
|
frag.write('float blur_step = 1.0 / texpaintSizeLocal.x;');
|
|
if (Context.blurDirectional) {
|
|
#if (kha_direct3d11 || kha_direct3d12 || kha_metal)
|
|
frag.write('const float blur_weight[7] = {1.0 / 28.0, 2.0 / 28.0, 3.0 / 28.0, 4.0 / 28.0, 5.0 / 28.0, 6.0 / 28.0, 7.0 / 28.0};');
|
|
#else
|
|
frag.write('const float blur_weight[7] = float[](1.0 / 28.0, 2.0 / 28.0, 3.0 / 28.0, 4.0 / 28.0, 5.0 / 28.0, 6.0 / 28.0, 7.0 / 28.0);');
|
|
#end
|
|
frag.add_uniform('vec3 brushDirection', '_brushDirection');
|
|
frag.write('vec2 blur_direction = brushDirection.yx;');
|
|
frag.write('for (int i = 0; i < 7; ++i) {');
|
|
#if (kha_direct3d11 || kha_direct3d12 || kha_metal || kha_vulkan)
|
|
frag.write('vec2 texCoordInp2 = texelFetch(gbuffer2, ivec2((sp.x + blur_direction.x * blur_step * float(i)) * gbufferSizeLocal.x, (sp.y + blur_direction.y * blur_step * float(i)) * gbufferSizeLocal.y), 0).ba;');
|
|
#else
|
|
frag.write('vec2 texCoordInp2 = texelFetch(gbuffer2, ivec2((sp.x + blur_direction.x * blur_step * float(i)) * gbufferSizeLocal.x, (1.0 - (sp.y + blur_direction.y * blur_step * float(i))) * gbufferSizeLocal.y), 0).ba;');
|
|
#end
|
|
frag.write('vec4 texpaint_sample = texture(texpaint_undo, texCoordInp2);');
|
|
frag.write('opacity += texpaint_sample.a * blur_weight[i];');
|
|
frag.write('basecol += texpaint_sample.rgb * blur_weight[i];');
|
|
frag.write('vec4 texpaint_pack_sample = texture(texpaint_pack_undo, texCoordInp2) * blur_weight[i];');
|
|
frag.write('roughness += texpaint_pack_sample.g;');
|
|
frag.write('metallic += texpaint_pack_sample.b;');
|
|
frag.write('occlusion += texpaint_pack_sample.r;');
|
|
frag.write('height += texpaint_pack_sample.a;');
|
|
frag.write('nortan += texture(texpaint_nor_undo, texCoordInp2).rgb * blur_weight[i];');
|
|
frag.write('}');
|
|
}
|
|
else {
|
|
#if (kha_direct3d11 || kha_direct3d12 || kha_metal)
|
|
frag.write('const float blur_weight[15] = {0.034619 / 2.0, 0.044859 / 2.0, 0.055857 / 2.0, 0.066833 / 2.0, 0.076841 / 2.0, 0.084894 / 2.0, 0.090126 / 2.0, 0.09194 / 2.0, 0.090126 / 2.0, 0.084894 / 2.0, 0.076841 / 2.0, 0.066833 / 2.0, 0.055857 / 2.0, 0.044859 / 2.0, 0.034619 / 2.0};');
|
|
#else
|
|
frag.write('const float blur_weight[15] = float[](0.034619 / 2.0, 0.044859 / 2.0, 0.055857 / 2.0, 0.066833 / 2.0, 0.076841 / 2.0, 0.084894 / 2.0, 0.090126 / 2.0, 0.09194 / 2.0, 0.090126 / 2.0, 0.084894 / 2.0, 0.076841 / 2.0, 0.066833 / 2.0, 0.055857 / 2.0, 0.044859 / 2.0, 0.034619 / 2.0);');
|
|
#end
|
|
// X
|
|
frag.write('for (int i = -7; i <= 7; ++i) {');
|
|
frag.write('vec4 texpaint_sample = texture(texpaint_undo, texCoordInp + vec2(blur_step * float(i), 0.0));');
|
|
frag.write('opacity += texpaint_sample.a * blur_weight[i + 7];');
|
|
frag.write('basecol += texpaint_sample.rgb * blur_weight[i + 7];');
|
|
frag.write('vec4 texpaint_pack_sample = texture(texpaint_pack_undo, texCoordInp + vec2(blur_step * float(i), 0.0)) * blur_weight[i + 7];');
|
|
frag.write('roughness += texpaint_pack_sample.g;');
|
|
frag.write('metallic += texpaint_pack_sample.b;');
|
|
frag.write('occlusion += texpaint_pack_sample.r;');
|
|
frag.write('height += texpaint_pack_sample.a;');
|
|
frag.write('nortan += texture(texpaint_nor_undo, texCoordInp + vec2(blur_step * float(i), 0.0)).rgb * blur_weight[i + 7];');
|
|
frag.write('}');
|
|
// Y
|
|
frag.write('for (int j = -7; j <= 7; ++j) {');
|
|
frag.write('vec4 texpaint_sample = texture(texpaint_undo, texCoordInp + vec2(0.0, blur_step * float(j)));');
|
|
frag.write('opacity += texpaint_sample.a * blur_weight[j + 7];');
|
|
frag.write('basecol += texpaint_sample.rgb * blur_weight[j + 7];');
|
|
frag.write('vec4 texpaint_pack_sample = texture(texpaint_pack_undo, texCoordInp + vec2(0.0, blur_step * float(j))) * blur_weight[j + 7];');
|
|
frag.write('roughness += texpaint_pack_sample.g;');
|
|
frag.write('metallic += texpaint_pack_sample.b;');
|
|
frag.write('occlusion += texpaint_pack_sample.r;');
|
|
frag.write('height += texpaint_pack_sample.a;');
|
|
frag.write('nortan += texture(texpaint_nor_undo, texCoordInp + vec2(0.0, blur_step * float(j))).rgb * blur_weight[j + 7];');
|
|
frag.write('}');
|
|
}
|
|
frag.write('opacity *= brushOpacity;');
|
|
}
|
|
}
|