luboslenco
156 lines
6.7 KiB
TypeScript
156 lines
6.7 KiB
TypeScript
|
|
let make_mesh_preview_opacity_discard_decal: f32 = 0.05;
|
|
|
|
function make_mesh_preview_run(data: material_t, matcon: material_context_t): node_shader_context_t {
|
|
let context_id: string = "mesh";
|
|
let props: shader_context_t = {
|
|
name: context_id,
|
|
depth_write: true,
|
|
compare_mode: "less",
|
|
cull_mode: "clockwise",
|
|
vertex_elements: [
|
|
{
|
|
name: "pos",
|
|
data: "short4norm"
|
|
},
|
|
{
|
|
name: "nor",
|
|
data: "short2norm"
|
|
},
|
|
{
|
|
name: "tex",
|
|
data: "short2norm"
|
|
}
|
|
],
|
|
color_attachments: ["RGBA64", "RGBA64", "RGBA64"],
|
|
depth_attachment: "D32"
|
|
};
|
|
let con_mesh: node_shader_context_t = node_shader_context_create(data, props);
|
|
|
|
let kong: node_shader_t = node_shader_context_make_kong(con_mesh);
|
|
|
|
let pos: string = "input.pos";
|
|
|
|
///if arm_skin
|
|
let skin: bool = mesh_data_get_vertex_array(context_raw.paint_object.data, "bone") != null;
|
|
if (skin) {
|
|
pos = "spos";
|
|
node_shader_context_add_elem(con_mesh, "bone", "short4norm");
|
|
node_shader_context_add_elem(con_mesh, "weight", "short4norm");
|
|
node_shader_add_function(kong, str_get_skinning_dual_quat);
|
|
node_shader_add_constant(kong, "skin_bones float4[128 * 2]", "_skin_bones");
|
|
node_shader_add_constant(kong, "pos_unpack: float", "_pos_unpack");
|
|
node_shader_write_attrib_vert(kong, "var skin_a: float4;");
|
|
node_shader_write_attrib_vert(kong, "var skin_b: float4;");
|
|
node_shader_write_attrib_vert(kong, "get_skinning_dual_quat(int4(bone * 32767), weight, skin_a, skin_b);");
|
|
node_shader_write_attrib_vert(kong, "var spos: float3 = input.pos.xyz;");
|
|
node_shader_write_attrib_vert(kong, "spos.xyz *= constants.pos_unpack;");
|
|
node_shader_write_attrib_vert(kong, "spos.xyz += 2.0 * cross(skin_a.xyz, cross(skin_a.xyz, spos.xyz) + skin_a.w * spos.xyz);");
|
|
node_shader_write_attrib_vert(kong, "spos.xyz += 2.0 * (skin_a.w * skin_b.xyz - skin_b.w * skin_a.xyz + cross(skin_a.xyz, skin_b.xyz));");
|
|
node_shader_write_attrib_vert(kong, "spos.xyz /= constants.pos_unpack;");
|
|
}
|
|
///end
|
|
|
|
node_shader_add_constant(kong, "WVP: float4x4", "_world_view_proj_matrix");
|
|
node_shader_write_attrib_vert(kong, "output.pos = constants.WVP * float4(" + pos + ".xyz, 1.0);");
|
|
|
|
let sc: f32 = context_raw.brush_scale * context_raw.brush_nodes_scale;
|
|
let brush_scale: string = sc + "";
|
|
node_shader_add_out(kong, "tex_coord: float2");
|
|
node_shader_write_attrib_vert(kong, "output.tex_coord = input.tex * float(" + brush_scale + ");");
|
|
node_shader_write_attrib_frag(kong, "var tex_coord: float2 = input.tex_coord;");
|
|
|
|
let decal: bool = context_raw.decal_preview;
|
|
parser_material_sample_keep_aspect = decal;
|
|
parser_material_sample_uv_scale = brush_scale;
|
|
parser_material_parse_height = make_material_height_used;
|
|
parser_material_parse_height_as_channel = true;
|
|
let sout: shader_out_t = parser_material_parse(ui_nodes_get_canvas_material(), con_mesh, kong, matcon);
|
|
parser_material_parse_height = false;
|
|
parser_material_parse_height_as_channel = false;
|
|
parser_material_sample_keep_aspect = false;
|
|
let base: string = sout.out_basecol;
|
|
let rough: string = sout.out_roughness;
|
|
let met: string = sout.out_metallic;
|
|
let occ: string = sout.out_occlusion;
|
|
let opac: string = sout.out_opacity;
|
|
let height: string = sout.out_height;
|
|
let nortan: string = parser_material_out_normaltan;
|
|
node_shader_write_frag(kong, "var basecol: float3 = pow3(" + base + ", float3(2.2, 2.2, 2.2));");
|
|
node_shader_write_frag(kong, "var roughness: float = " + rough + ";");
|
|
node_shader_write_frag(kong, "var metallic: float = " + met + ";");
|
|
node_shader_write_frag(kong, "var occlusion: float = " + occ + ";");
|
|
node_shader_write_frag(kong, "var opacity: float = " + opac + ";");
|
|
node_shader_write_frag(kong, "var nortan: float3 = " + nortan + ";");
|
|
node_shader_write_frag(kong, "var height: float = " + height + ";");
|
|
|
|
if (decal) {
|
|
if (context_raw.tool == tool_type_t.TEXT) {
|
|
node_shader_add_texture(kong, "textexttool", "_textexttool");
|
|
node_shader_write_frag(kong, "opacity *= sample_lod(textexttool, sampler_linear, tex_coord / float(" + brush_scale + "), 0.0).r;");
|
|
}
|
|
}
|
|
if (decal) {
|
|
let opac: f32 = make_mesh_preview_opacity_discard_decal;
|
|
node_shader_write_frag(kong, "if (opacity < " + opac + ") { discard; }");
|
|
}
|
|
|
|
kong.frag_out = "float4[3]";
|
|
kong.frag_n = true;
|
|
|
|
node_shader_add_function(kong, str_pack_float_int16);
|
|
node_shader_add_function(kong, str_cotangent_frame);
|
|
node_shader_add_function(kong, str_octahedron_wrap);
|
|
|
|
if (make_material_height_used) {
|
|
node_shader_write_frag(kong, "if (height > 0.0) {");
|
|
node_shader_write_frag(kong, "var height_dx: float = ddx(height * 2.0);");
|
|
node_shader_write_frag(kong, "var height_dy: float = ddy(height * 2.0);");
|
|
// Whiteout blend
|
|
node_shader_write_frag(kong, "var n1: float3 = nortan * float3(2.0, 2.0, 2.0) - float3(1.0, 1.0, 1.0);");
|
|
node_shader_write_frag(kong, "var n2: float3 = normalize(float3(height_dx * 16.0, height_dy * 16.0, 1.0));");
|
|
node_shader_write_frag(kong, "nortan = normalize(float3(n1.xy + n2.xy, n1.z * n2.z)) * float3(0.5, 0.5, 0.5) + float3(0.5, 0.5, 0.5);");
|
|
node_shader_write_frag(kong, "}");
|
|
}
|
|
|
|
// Apply normal channel
|
|
if (decal) {
|
|
// TODO
|
|
}
|
|
else {
|
|
kong.frag_vvec = true;
|
|
node_shader_write_frag(kong, "var TBN: float3x3 = cotangent_frame(n, vvec, tex_coord);");
|
|
node_shader_write_frag(kong, "n = nortan * 2.0 - 1.0;");
|
|
node_shader_write_frag(kong, "n.y = -n.y;");
|
|
node_shader_write_frag(kong, "n = normalize(TBN * n);");
|
|
}
|
|
|
|
node_shader_write_frag(kong, "n = n / (abs(n.x) + abs(n.y) + abs(n.z));");
|
|
// node_shader_write_frag(kong, "n.xy = n.z >= 0.0 ? n.xy : octahedron_wrap(n.xy);");
|
|
node_shader_write_frag(kong, "if (n.z < 0.0) { n.xy = octahedron_wrap(n.xy); }");
|
|
// uint matid = uint(0);
|
|
|
|
if (decal) {
|
|
node_shader_write_frag(kong, "output[0] = float4(n.x, n.y, roughness, pack_f32_i16(metallic, uint(0)));"); // metallic/matid
|
|
node_shader_write_frag(kong, "output[1] = float4(basecol, occlusion);");
|
|
}
|
|
else {
|
|
node_shader_write_frag(kong, "output[0] = float4(n.x, n.y, lerp(1.0, roughness, opacity), pack_f32_i16(lerp(1.0, metallic, opacity), uint(0)));"); // metallic/matid
|
|
node_shader_write_frag(kong, "output[1] = float4(lerp3(float3(0.0, 0.0, 0.0), basecol, opacity), occlusion);");
|
|
}
|
|
node_shader_write_frag(kong, "output[2] = float4(0.0, 0.0, 0.0, 0.0);");
|
|
|
|
parser_material_finalize(con_mesh);
|
|
|
|
///if arm_skin
|
|
if (skin) {
|
|
node_shader_write_vert(kong, "wnormal = normalize(constants.N * float3(input.nor.xy, input.pos.w) + 2.0 * cross(skin_a.xyz, cross(skin_a.xyz, float3(input.nor.xy, input.pos.w)) + skin_a.w * float3(input.nor.xy, input.pos.w)));");
|
|
}
|
|
///end
|
|
|
|
con_mesh.data.shader_from_source = true;
|
|
gpu_create_shaders_from_kong(node_shader_get(kong), ADDRESS(con_mesh.data.vertex_shader), ADDRESS(con_mesh.data.fragment_shader), ADDRESS(con_mesh.data._.vertex_shader_size), ADDRESS(con_mesh.data._.fragment_shader_size));
|
|
|
|
return con_mesh;
|
|
}
|