luboslenco
266 lines
14 KiB
TypeScript
266 lines
14 KiB
TypeScript
|
|
function make_paint_color_attachments(): string[] {
|
|
if (context_raw.tool == workspace_tool_t.PICKER) {
|
|
let res: string[] = ["RGBA32", "RGBA32", "RGBA32", "RGBA32"];
|
|
return res;
|
|
}
|
|
let res: string[] = ["RGBA32", "RGBA32", "RGBA32", "R8"];
|
|
return res;
|
|
}
|
|
|
|
function make_paint_run(data: material_t, matcon: material_context_t): node_shader_context_t {
|
|
let props: shader_context_t = {
|
|
name: "paint",
|
|
depth_write: false,
|
|
compare_mode: "always", // TODO: align texcoords winding order
|
|
// cull_mode: "counter_clockwise",
|
|
cull_mode: "none",
|
|
vertex_elements: [
|
|
{
|
|
name: "pos",
|
|
data: "short4norm"
|
|
},
|
|
{
|
|
name: "nor",
|
|
data: "short2norm"
|
|
},
|
|
{
|
|
name: "tex",
|
|
data: "short2norm"
|
|
}
|
|
],
|
|
color_attachments: make_paint_color_attachments()
|
|
};
|
|
let con_paint: node_shader_context_t = node_shader_context_create(data, props);
|
|
|
|
con_paint.data.color_writes_red = [true, true, true, true];
|
|
con_paint.data.color_writes_green = [true, true, true, true];
|
|
con_paint.data.color_writes_blue = [true, true, true, true];
|
|
con_paint.data.color_writes_alpha = [true, true, true, true];
|
|
con_paint.allow_vcols = mesh_data_get_vertex_array(context_raw.paint_object.data, "col") != null;
|
|
|
|
let vert: node_shader_t = node_shader_context_make_vert(con_paint);
|
|
let frag: node_shader_t = node_shader_context_make_frag(con_paint);
|
|
frag.ins = vert.outs;
|
|
|
|
if (context_raw.tool == workspace_tool_t.PICKER) {
|
|
// Mangle vertices to form full screen triangle
|
|
node_shader_write(vert, "gl_Position = vec4(-1.0 + float((gl_VertexID & 1) << 2), -1.0 + float((gl_VertexID & 2) << 1), 0.0, 1.0);");
|
|
|
|
node_shader_add_uniform(frag, "sampler2D gbuffer2");
|
|
node_shader_add_uniform(frag, "vec2 gbuffer_size", "_gbuffer_size");
|
|
node_shader_add_uniform(frag, "vec4 inp", "_input_brush");
|
|
|
|
node_shader_write(frag, "vec2 tex_coord_inp = texelFetch(gbuffer2, ivec2(inp.x * gbuffer_size.x, inp.y * gbuffer_size.y), 0).ba;");
|
|
|
|
node_shader_add_out(frag, "vec4 frag_color[4]");
|
|
node_shader_add_uniform(frag, "sampler2D texpaint");
|
|
node_shader_add_uniform(frag, "sampler2D texpaint_nor");
|
|
node_shader_add_uniform(frag, "sampler2D texpaint_pack");
|
|
node_shader_write(frag, "frag_color[0] = textureLod(texpaint, tex_coord_inp, 0.0);");
|
|
node_shader_write(frag, "frag_color[1] = textureLod(texpaint_nor, tex_coord_inp, 0.0);");
|
|
node_shader_write(frag, "frag_color[2] = textureLod(texpaint_pack, tex_coord_inp, 0.0);");
|
|
node_shader_write(frag, "frag_color[3].rg = tex_coord_inp.xy;");
|
|
con_paint.data.shader_from_source = true;
|
|
con_paint.data.vertex_shader = node_shader_get(vert);
|
|
con_paint.data.fragment_shader = node_shader_get(frag);
|
|
return con_paint;
|
|
}
|
|
|
|
node_shader_write(vert, "vec2 tpos = vec2(tex.x * 2.0 - 1.0, (1.0 - tex.y) * 2.0 - 1.0);");
|
|
// node_shader_write(vert, "vec2 tpos = vec2(frac(tex.x * tex_scale) * 2.0 - 1.0, (1.0 - frac(tex.y * tex_scale)) * 2.0 - 1.0);"); // 3D View
|
|
|
|
node_shader_write(vert, "gl_Position = vec4(tpos, 0.0, 1.0);");
|
|
|
|
node_shader_add_uniform(vert, "mat4 WVP", "_world_view_proj_matrix");
|
|
|
|
node_shader_add_out(vert, "vec4 ndc");
|
|
node_shader_write_attrib(vert, "ndc = mul(vec4(pos.xyz, 1.0), WVP);");
|
|
|
|
node_shader_write_attrib(frag, "vec3 sp = vec3((ndc.xyz / ndc.w) * 0.5 + 0.5);");
|
|
node_shader_write_attrib(frag, "sp.y = 1.0 - sp.y;");
|
|
node_shader_write_attrib(frag, "sp.z -= 0.0001;"); // small bias
|
|
|
|
node_shader_add_uniform(frag, "vec4 inp", "_input_brush");
|
|
node_shader_add_uniform(frag, "vec4 inplast", "_input_brush_last");
|
|
node_shader_add_uniform(frag, "float aspect_ratio", "_aspect_ratio_window");
|
|
node_shader_write(frag, "vec2 bsp = sp.xy * 2.0 - 1.0;");
|
|
node_shader_write(frag, "bsp.x *= aspect_ratio;");
|
|
node_shader_write(frag, "bsp = bsp * 0.5 + 0.5;");
|
|
|
|
node_shader_add_uniform(frag, "sampler2D gbufferD");
|
|
|
|
node_shader_add_out(frag, "vec4 frag_color[4]");
|
|
|
|
node_shader_add_uniform(frag, "float brush_radius", "_brush_radius");
|
|
node_shader_add_uniform(frag, "float brush_opacity", "_brush_opacity");
|
|
node_shader_add_uniform(frag, "float brush_hardness", "_brush_hardness");
|
|
|
|
if (context_raw.tool == workspace_tool_t.ERASER ||
|
|
context_raw.tool == workspace_tool_t.CLONE ||
|
|
context_raw.tool == workspace_tool_t.BLUR ||
|
|
context_raw.tool == workspace_tool_t.SMUDGE) {
|
|
|
|
node_shader_write(frag, "float dist = 0.0;");
|
|
|
|
node_shader_write(frag, "float depth = textureLod(gbufferD, inp.xy, 0.0).r;");
|
|
|
|
node_shader_add_uniform(frag, "mat4 invVP", "_inv_view_proj_matrix");
|
|
node_shader_write(frag, "vec4 winp = vec4(vec2(inp.x, 1.0 - inp.y) * 2.0 - 1.0, depth * 2.0 - 1.0, 1.0);");
|
|
node_shader_write(frag, "winp = mul(winp, invVP);");
|
|
node_shader_write(frag, "winp.xyz /= winp.w;");
|
|
frag.wposition = true;
|
|
|
|
node_shader_write(frag, "float depthlast = textureLod(gbufferD, inplast.xy, 0.0).r;");
|
|
|
|
node_shader_write(frag, "vec4 winplast = vec4(vec2(inplast.x, 1.0 - inplast.y) * 2.0 - 1.0, depthlast * 2.0 - 1.0, 1.0);");
|
|
node_shader_write(frag, "winplast = mul(winplast, invVP);");
|
|
node_shader_write(frag, "winplast.xyz /= winplast.w;");
|
|
|
|
node_shader_write(frag, "vec3 pa = wposition - winp.xyz;");
|
|
node_shader_write(frag, "vec3 ba = winplast.xyz - winp.xyz;");
|
|
|
|
// Capsule
|
|
node_shader_write(frag, "float h = clamp(dot(pa, ba) / dot(ba, ba), 0.0, 1.0);");
|
|
node_shader_write(frag, "dist = length(pa - ba * h);");
|
|
|
|
node_shader_write(frag, "if (dist > brush_radius) discard;");
|
|
}
|
|
|
|
// node_shader_add_uniform(vert, "float brush_scale", "_brush_scale");
|
|
// node_shader_add_uniform(vert, "float tex_scale", "_tex_unpack");
|
|
// node_shader_add_out(vert, "vec2 tex_coord");
|
|
// node_shader_write(vert, "tex_coord = tex * brush_scale * tex_scale;");
|
|
|
|
if (context_raw.tool == workspace_tool_t.CLONE || context_raw.tool == workspace_tool_t.BLUR || context_raw.tool == workspace_tool_t.SMUDGE) {
|
|
node_shader_add_uniform(frag, "sampler2D gbuffer2");
|
|
node_shader_add_uniform(frag, "vec2 gbuffer_size", "_gbuffer_size");
|
|
node_shader_add_uniform(frag, "sampler2D texpaint_undo", "_texpaint_undo");
|
|
node_shader_add_uniform(frag, "sampler2D texpaint_nor_undo", "_texpaint_nor_undo");
|
|
node_shader_add_uniform(frag, "sampler2D texpaint_pack_undo", "_texpaint_pack_undo");
|
|
|
|
if (context_raw.tool == workspace_tool_t.CLONE) {
|
|
// node_shader_add_uniform(frag, "vec2 clone_delta", "_clone_delta");
|
|
// node_shader_write(frag, "vec2 tex_coord_inp = texelFetch(gbuffer2, ivec2((sp.xy + clone_delta) * gbuffer_size), 0).ba;");
|
|
|
|
// node_shader_write(frag, "vec3 texpaint_pack_sample = textureLod(texpaint_pack_undo, tex_coord_inp, 0.0).rgb;");
|
|
// let base: string = "textureLod(texpaint_undo, tex_coord_inp, 0.0).rgb";
|
|
// let rough: string = "texpaint_pack_sample.g";
|
|
// let met: string = "texpaint_pack_sample.b";
|
|
// let occ: string = "texpaint_pack_sample.r";
|
|
// let nortan: string = "textureLod(texpaint_nor_undo, tex_coord_inp, 0.0).rgb";
|
|
// let height: string = "0.0";
|
|
// let opac: string = "1.0";
|
|
// node_shader_write(frag, "vec3 basecol = " + base + ";");
|
|
// node_shader_write(frag, "float roughness = " + rough + ";");
|
|
// node_shader_write(frag, "float metallic = " + met + ";");
|
|
// node_shader_write(frag, "float occlusion = " + occ + ";");
|
|
// node_shader_write(frag, "vec3 nortan = " + nortan + ";");
|
|
// node_shader_write(frag, "float height = " + height + ";");
|
|
// node_shader_write(frag, "float mat_opacity = " + opac + ";");
|
|
// node_shader_write(frag, "float opacity = mat_opacity * brush_opacity;");
|
|
}
|
|
else { // Blur
|
|
// node_shader_write(frag, "vec2 tex_coord_inp = texelFetch(gbuffer2, ivec2(sp.x * gbuffer_size.x, sp.y * gbuffer_size.y), 0).ba;");
|
|
|
|
// node_shader_write(frag, "vec3 basecol = vec3(0.0, 0.0, 0.0);");
|
|
// node_shader_write(frag, "float roughness = 0.0;");
|
|
// node_shader_write(frag, "float metallic = 0.0;");
|
|
// node_shader_write(frag, "float occlusion = 0.0;");
|
|
// node_shader_write(frag, "vec3 nortan = vec3(0.0, 0.0, 0.0);");
|
|
// node_shader_write(frag, "float height = 0.0;");
|
|
// node_shader_write(frag, "float mat_opacity = 1.0;");
|
|
// node_shader_write(frag, "float opacity = 0.0;");
|
|
|
|
// node_shader_add_uniform(frag, "vec2 texpaint_size", "_texpaint_size");
|
|
// node_shader_write(frag, "float blur_step = 1.0 / texpaint_size.x;");
|
|
// if (context_raw.blur_directional) {
|
|
// ///if (arm_direct3d12 || arm_metal)
|
|
// node_shader_write(frag, "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
|
|
// node_shader_write(frag, "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
|
|
// node_shader_add_uniform(frag, "vec3 brush_direction", "_brush_direction");
|
|
// node_shader_write(frag, "vec2 blur_direction = brush_direction.yx;");
|
|
// node_shader_write(frag, "for (int i = 0; i < 7; ++i) {");
|
|
// node_shader_write(frag, "vec2 tex_coord_inp2 = texelFetch(gbuffer2, ivec2((sp.x + blur_direction.x * blur_step * float(i)) * gbuffer_size.x, (sp.y + blur_direction.y * blur_step * float(i)) * gbuffer_size.y), 0).ba;");
|
|
// node_shader_write(frag, "vec4 texpaint_sample = texture(texpaint_undo, tex_coord_inp2);");
|
|
// node_shader_write(frag, "opacity += texpaint_sample.a * blur_weight[i];");
|
|
// node_shader_write(frag, "basecol += texpaint_sample.rgb * blur_weight[i];");
|
|
// node_shader_write(frag, "vec4 texpaint_pack_sample = texture(texpaint_pack_undo, tex_coord_inp2) * blur_weight[i];");
|
|
// node_shader_write(frag, "roughness += texpaint_pack_sample.g;");
|
|
// node_shader_write(frag, "metallic += texpaint_pack_sample.b;");
|
|
// node_shader_write(frag, "occlusion += texpaint_pack_sample.r;");
|
|
// node_shader_write(frag, "height += texpaint_pack_sample.a;");
|
|
// node_shader_write(frag, "nortan += texture(texpaint_nor_undo, tex_coord_inp2).rgb * blur_weight[i];");
|
|
// node_shader_write(frag, "}");
|
|
// }
|
|
// else {
|
|
// ///if (arm_direct3d12 || arm_metal)
|
|
// node_shader_write(frag, "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
|
|
// node_shader_write(frag, "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
|
|
// node_shader_write(frag, "for (int i = -7; i <= 7; ++i) {");
|
|
// node_shader_write(frag, "vec4 texpaint_sample = texture(texpaint_undo, tex_coord_inp + vec2(blur_step * float(i), 0.0));");
|
|
// node_shader_write(frag, "opacity += texpaint_sample.a * blur_weight[i + 7];");
|
|
// node_shader_write(frag, "basecol += texpaint_sample.rgb * blur_weight[i + 7];");
|
|
// node_shader_write(frag, "vec4 texpaint_pack_sample = texture(texpaint_pack_undo, tex_coord_inp + vec2(blur_step * float(i), 0.0)) * blur_weight[i + 7];");
|
|
// node_shader_write(frag, "roughness += texpaint_pack_sample.g;");
|
|
// node_shader_write(frag, "metallic += texpaint_pack_sample.b;");
|
|
// node_shader_write(frag, "occlusion += texpaint_pack_sample.r;");
|
|
// node_shader_write(frag, "height += texpaint_pack_sample.a;");
|
|
// node_shader_write(frag, "nortan += texture(texpaint_nor_undo, tex_coord_inp + vec2(blur_step * float(i), 0.0)).rgb * blur_weight[i + 7];");
|
|
// node_shader_write(frag, "}");
|
|
// // Y
|
|
// node_shader_write(frag, "for (int j = -7; j <= 7; ++j) {");
|
|
// node_shader_write(frag, "vec4 texpaint_sample = texture(texpaint_undo, tex_coord_inp + vec2(0.0, blur_step * float(j)));");
|
|
// node_shader_write(frag, "opacity += texpaint_sample.a * blur_weight[j + 7];");
|
|
// node_shader_write(frag, "basecol += texpaint_sample.rgb * blur_weight[j + 7];");
|
|
// node_shader_write(frag, "vec4 texpaint_pack_sample = texture(texpaint_pack_undo, tex_coord_inp + vec2(0.0, blur_step * float(j))) * blur_weight[j + 7];");
|
|
// node_shader_write(frag, "roughness += texpaint_pack_sample.g;");
|
|
// node_shader_write(frag, "metallic += texpaint_pack_sample.b;");
|
|
// node_shader_write(frag, "occlusion += texpaint_pack_sample.r;");
|
|
// node_shader_write(frag, "height += texpaint_pack_sample.a;");
|
|
// node_shader_write(frag, "nortan += texture(texpaint_nor_undo, tex_coord_inp + vec2(0.0, blur_step * float(j))).rgb * blur_weight[j + 7];");
|
|
// node_shader_write(frag, "}");
|
|
// }
|
|
// node_shader_write(frag, "opacity *= brush_opacity;");
|
|
}
|
|
}
|
|
|
|
node_shader_write(frag, "float opacity = 1.0;");
|
|
node_shader_write(frag, "if (opacity == 0.0) discard;");
|
|
|
|
node_shader_write(frag, "float str = clamp((brush_radius - dist) * brush_hardness * 400.0, 0.0, 1.0) * opacity;");
|
|
|
|
// Manual blending to preserve memory
|
|
frag.wvpposition = true;
|
|
node_shader_write(frag, "vec2 sample_tc = vec2(wvpposition.xy / wvpposition.w) * 0.5 + 0.5;");
|
|
node_shader_write(frag, "sample_tc.y = 1.0 - sample_tc.y;");
|
|
node_shader_add_uniform(frag, "sampler2D paintmask");
|
|
node_shader_write(frag, "float sample_mask = textureLod(paintmask, sample_tc, 0.0).r;");
|
|
node_shader_write(frag, "str = max(str, sample_mask);");
|
|
|
|
node_shader_add_uniform(frag, "sampler2D texpaint_undo", "_texpaint_undo");
|
|
node_shader_write(frag, "vec4 sample_undo = textureLod(texpaint_undo, sample_tc, 0.0);");
|
|
|
|
if (context_raw.tool == workspace_tool_t.ERASER) {
|
|
// node_shader_write(frag, "frag_color[0] = vec4(mix(sample_undo.rgb, vec3(0.0, 0.0, 0.0), str), sample_undo.a - str);");
|
|
node_shader_write(frag, "frag_color[0] = vec4(0.0, 0.0, 0.0, 0.0);");
|
|
node_shader_write(frag, "frag_color[1] = vec4(0.5, 0.5, 1.0, 0.0);");
|
|
node_shader_write(frag, "frag_color[2] = vec4(1.0, 0.0, 0.0, 0.0);");
|
|
}
|
|
|
|
node_shader_write(frag, "frag_color[3] = vec4(str, 0.0, 0.0, 1.0);");
|
|
|
|
parser_material_finalize(con_paint);
|
|
parser_material_sample_keep_aspect = false;
|
|
con_paint.data.shader_from_source = true;
|
|
con_paint.data.vertex_shader = node_shader_get(vert);
|
|
con_paint.data.fragment_shader = node_shader_get(frag);
|
|
|
|
return con_paint;
|
|
}
|