edgevoid/armorpaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity
Files
ae67dc52ecf203e638e08b9c7274e024e8bc39ff
armorpaint/armorsculpt/Sources/MakeMesh.ts
T
luboslenco 5b340fc755 Prepare iron to c port
2024-01-31 15:12:28 +01:00

339 lines
16 KiB
TypeScript
Raw Blame History

class MakeMesh {
static layerPassCount = 1;
static run = (data: TMaterial, layerPass = 0): NodeShaderContextRaw => {
let context_id = layerPass == 0 ? "mesh" : "mesh" + layerPass;
let con_mesh = NodeShaderContext.create(data, {
name: context_id,
depth_write: layerPass == 0 ? true : false,
compare_mode: layerPass == 0 ? "less" : "equal",
cull_mode: (Context.raw.cullBackfaces || layerPass > 0) ? "clockwise" : "none",
vertex_elements: [{name: "pos", data: "short4norm"}],
color_attachments: ["RGBA64", "RGBA64", "RGBA64"],
depth_attachment: "DEPTH32"
});
let vert = NodeShaderContext.make_vert(con_mesh);
let frag = NodeShaderContext.make_frag(con_mesh);
frag.ins = vert.outs;
NodeShader.add_out(vert, 'vec2 texCoord');
frag.wvpposition = true;
NodeShader.add_out(vert, 'vec4 prevwvpposition');
NodeShader.add_uniform(vert, 'mat4 VP', '_viewProjectionMatrix');
NodeShader.add_uniform(vert, 'mat4 prevWVP', '_prevWorldViewProjectionMatrix');
vert.wposition = true;
let textureCount = 0;
NodeShader.add_uniform(vert, 'mat4 WVP', '_worldViewProjectionMatrix');
NodeShader.add_uniform(vert, 'sampler2D texpaint_vert', '_texpaint_vert' + Project.layers[0].id);
NodeShader.write(vert, 'vec3 meshpos = texelFetch(texpaint_vert, ivec2(gl_VertexID % textureSize(texpaint_vert, 0).x, gl_VertexID / textureSize(texpaint_vert, 0).y), 0).xyz;');
// + pos.xyz * 0.000001
NodeShader.write(vert, 'gl_Position = mul(vec4(meshpos.xyz, 1.0), WVP);');
NodeShader.write(vert, 'texCoord = vec2(0.0, 0.0);');
NodeShader.write(vert, 'prevwvpposition = mul(vec4(pos.xyz, 1.0), prevWVP);');
NodeShader.add_out(frag, 'vec4 fragColor[3]');
NodeShader.add_uniform(vert, "mat3 N", "_normalMatrix");
NodeShader.add_out(vert, "vec3 wnormal");
NodeShader.write_attrib(vert, 'int baseVertex0 = gl_VertexID - (gl_VertexID % 3);');
NodeShader.write_attrib(vert, 'int baseVertex1 = baseVertex0 + 1;');
NodeShader.write_attrib(vert, 'int baseVertex2 = baseVertex0 + 2;');
NodeShader.write_attrib(vert, 'vec3 meshpos0 = texelFetch(texpaint_vert, ivec2(baseVertex0 % textureSize(texpaint_vert, 0).x, baseVertex0 / textureSize(texpaint_vert, 0).y), 0).xyz;');
NodeShader.write_attrib(vert, 'vec3 meshpos1 = texelFetch(texpaint_vert, ivec2(baseVertex1 % textureSize(texpaint_vert, 0).x, baseVertex1 / textureSize(texpaint_vert, 0).y), 0).xyz;');
NodeShader.write_attrib(vert, 'vec3 meshpos2 = texelFetch(texpaint_vert, ivec2(baseVertex2 % textureSize(texpaint_vert, 0).x, baseVertex2 / textureSize(texpaint_vert, 0).y), 0).xyz;');
NodeShader.write_attrib(vert, 'vec3 meshnor = normalize(cross(meshpos2 - meshpos1, meshpos0 - meshpos1));');
NodeShader.write_attrib(vert, 'wnormal = mul(meshnor, N);');
NodeShader.write_attrib(frag, 'vec3 n = normalize(wnormal);');
NodeShader.add_function(frag, ShaderFunctions.str_packFloatInt16);
NodeShader.add_function(frag, ShaderFunctions.str_octahedronWrap);
NodeShader.add_function(frag, ShaderFunctions.str_cotangentFrame);
if (layerPass > 0) {
NodeShader.add_uniform(frag, 'sampler2D gbuffer0');
NodeShader.add_uniform(frag, 'sampler2D gbuffer1');
NodeShader.add_uniform(frag, 'sampler2D gbuffer2');
NodeShader.write(frag, 'vec2 fragcoord = (wvpposition.xy / wvpposition.w) * 0.5 + 0.5;');
///if (krom_direct3d11 || krom_direct3d12 || krom_metal || krom_vulkan)
NodeShader.write(frag, 'fragcoord.y = 1.0 - fragcoord.y;');
///end
NodeShader.write(frag, 'vec4 gbuffer0_sample = textureLod(gbuffer0, fragcoord, 0.0);');
NodeShader.write(frag, 'vec4 gbuffer1_sample = textureLod(gbuffer1, fragcoord, 0.0);');
NodeShader.write(frag, 'vec4 gbuffer2_sample = textureLod(gbuffer2, fragcoord, 0.0);');
NodeShader.write(frag, 'vec3 basecol = gbuffer0_sample.rgb;');
NodeShader.write(frag, 'float roughness = gbuffer2_sample.g;');
NodeShader.write(frag, 'float metallic = gbuffer2_sample.b;');
NodeShader.write(frag, 'float occlusion = gbuffer2_sample.r;');
NodeShader.write(frag, 'float opacity = 1.0;//gbuffer0_sample.a;');
NodeShader.write(frag, 'float matid = gbuffer1_sample.a;');
NodeShader.write(frag, 'vec3 ntex = gbuffer1_sample.rgb;');
NodeShader.write(frag, 'float height = gbuffer2_sample.a;');
}
else {
NodeShader.write(frag, 'vec3 basecol = vec3(0.0, 0.0, 0.0);');
NodeShader.write(frag, 'float roughness = 0.3;');
NodeShader.write(frag, 'float metallic = 0.0;');
NodeShader.write(frag, 'float occlusion = 1.0;');
NodeShader.write(frag, 'float opacity = 1.0;');
NodeShader.write(frag, 'float matid = 0.0;');
NodeShader.write(frag, 'vec3 ntex = vec3(0.5, 0.5, 1.0);');
NodeShader.write(frag, 'float height = 0.0;');
}
NodeShader.write(frag, 'vec4 texpaint_sample = vec4(0.0, 0.0, 0.0, 1.0);');
NodeShader.write(frag, 'vec4 texpaint_nor_sample;');
NodeShader.write(frag, 'vec4 texpaint_pack_sample;');
NodeShader.write(frag, 'float texpaint_opac;');
if (MakeMaterial.heightUsed) {
NodeShader.write(frag, 'float height0 = 0.0;');
NodeShader.write(frag, 'float height1 = 0.0;');
NodeShader.write(frag, 'float height2 = 0.0;');
NodeShader.write(frag, 'float height3 = 0.0;');
}
if (Context.raw.drawWireframe) {
textureCount++;
NodeShader.add_uniform(frag, 'sampler2D texuvmap', '_texuvmap');
}
if (Context.raw.viewportMode == ViewportMode.ViewLit && Context.raw.renderMode == RenderMode.RenderForward) {
textureCount += 4;
NodeShader.add_uniform(frag, 'sampler2D senvmapBrdf', "$brdf.k");
NodeShader.add_uniform(frag, 'sampler2D senvmapRadiance', '_envmapRadiance');
NodeShader.add_uniform(frag, 'sampler2D sltcMat', '_ltcMat');
NodeShader.add_uniform(frag, 'sampler2D sltcMag', '_ltcMag');
}
// Get layers for this pass
MakeMesh.layerPassCount = 1;
let layers: SlotLayerRaw[] = [];
let startCount = textureCount;
for (let l of Project.layers) {
if (!SlotLayer.isLayer(l) || !SlotLayer.isVisible(l)) continue;
let count = 3;
let masks = SlotLayer.getMasks(l);
if (masks != null) count += masks.length;
textureCount += count;
if (textureCount >= MakeMesh.getMaxTextures()) {
textureCount = startCount + count + 3; // gbuffer0_copy, gbuffer1_copy, gbuffer2_copy
MakeMesh.layerPassCount++;
}
if (layerPass == MakeMesh.layerPassCount - 1) {
layers.push(l);
}
}
let lastPass = layerPass == MakeMesh.layerPassCount - 1;
for (let l of layers) {
if (SlotLayer.getObjectMask(l) > 0) {
NodeShader.add_uniform(frag, 'int uid', '_uid');
if (SlotLayer.getObjectMask(l) > Project.paintObjects.length) { // Atlas
let visibles = Project.getAtlasObjects(SlotLayer.getObjectMask(l));
NodeShader.write(frag, 'if (');
for (let i = 0; i < visibles.length; ++i) {
if (i > 0) NodeShader.write(frag, ' || ');
NodeShader.write(frag, `${visibles[i].base.uid} == uid`);
}
NodeShader.write(frag, ') {');
}
else { // Object mask
let uid = Project.paintObjects[SlotLayer.getObjectMask(l) - 1].base.uid;
NodeShader.write(frag, `if (${uid} == uid) {`);
}
}
NodeShader.add_shared_sampler(frag, 'sampler2D texpaint' + l.id);
NodeShader.write(frag, 'texpaint_sample = vec4(0.8, 0.8, 0.8, 1.0);');
NodeShader.write(frag, 'texpaint_opac = texpaint_sample.a;');
let masks = SlotLayer.getMasks(l);
if (masks != null) {
let hasVisible = false;
for (let m of masks) {
if (SlotLayer.isVisible(m)) {
hasVisible = true;
break;
}
}
if (hasVisible) {
let texpaint_mask = 'texpaint_mask' + l.id;
NodeShader.write(frag, `float ${texpaint_mask} = 0.0;`);
for (let m of masks) {
if (!SlotLayer.isVisible(m)) continue;
NodeShader.add_shared_sampler(frag, 'sampler2D texpaint' + m.id);
NodeShader.write(frag, '{'); // Group mask is sampled across multiple layers
NodeShader.write(frag, 'float texpaint_mask_sample' + m.id + ' = textureLodShared(texpaint' + m.id + ', texCoord, 0.0).r;');
NodeShader.write(frag, '}');
}
NodeShader.write(frag, `texpaint_opac *= clamp(${texpaint_mask}, 0.0, 1.0);`);
}
}
if (SlotLayer.getOpacity(l) < 1) {
NodeShader.write(frag, `texpaint_opac *= ${SlotLayer.getOpacity(l)};`);
}
if (l == Project.layers[0]) {
NodeShader.write(frag, 'basecol = vec3(0.8, 0.8, 0.8);// texpaint_sample.rgb * texpaint_opac;');
}
if (SlotLayer.getObjectMask(l) > 0) {
NodeShader.write(frag, '}');
}
if (lastPass && Context.raw.drawTexels) {
NodeShader.add_uniform(frag, 'vec2 texpaintSize', '_texpaintSize');
NodeShader.write(frag, 'vec2 texel0 = texCoord * texpaintSize * 0.01;');
NodeShader.write(frag, 'vec2 texel1 = texCoord * texpaintSize * 0.1;');
NodeShader.write(frag, 'vec2 texel2 = texCoord * texpaintSize;');
NodeShader.write(frag, 'basecol *= max(float(mod(int(texel0.x), 2.0) == mod(int(texel0.y), 2.0)), 0.9);');
NodeShader.write(frag, 'basecol *= max(float(mod(int(texel1.x), 2.0) == mod(int(texel1.y), 2.0)), 0.9);');
NodeShader.write(frag, 'basecol *= max(float(mod(int(texel2.x), 2.0) == mod(int(texel2.y), 2.0)), 0.9);');
}
if (lastPass && Context.raw.drawWireframe) {
NodeShader.write(frag, 'basecol *= 1.0 - textureLod(texuvmap, texCoord, 0.0).r;');
}
if (MakeMaterial.heightUsed) {
NodeShader.write(frag, 'if (height > 0.0) {');
NodeShader.write(frag, 'float height_dx = height0 - height1;');
NodeShader.write(frag, 'float height_dy = height2 - height3;');
// Whiteout blend
NodeShader.write(frag, 'vec3 n1 = ntex * vec3(2.0, 2.0, 2.0) - vec3(1.0, 1.0, 1.0);');
NodeShader.write(frag, 'vec3 n2 = normalize(vec3(height_dx * 16.0, height_dy * 16.0, 1.0));');
NodeShader.write(frag, 'ntex = normalize(vec3(n1.xy + n2.xy, n1.z * n2.z)) * vec3(0.5, 0.5, 0.5) + vec3(0.5, 0.5, 0.5);');
NodeShader.write(frag, '}');
}
if (!lastPass) {
NodeShader.write(frag, 'fragColor[0] = vec4(basecol, opacity);');
NodeShader.write(frag, 'fragColor[1] = vec4(ntex, matid);');
NodeShader.write(frag, 'fragColor[2] = vec4(occlusion, roughness, metallic, height);');
ParserMaterial.finalize(con_mesh);
con_mesh.data.shader_from_source = true;
con_mesh.data.vertex_shader = NodeShader.get(vert);
con_mesh.data.fragment_shader = NodeShader.get(frag);
return con_mesh;
}
frag.vVec = true;
///if (krom_direct3d11 || krom_direct3d12 || krom_metal || krom_vulkan)
NodeShader.write(frag, 'mat3 TBN = cotangentFrame(n, vVec, texCoord);');
///else
NodeShader.write(frag, 'mat3 TBN = cotangentFrame(n, -vVec, texCoord);');
///end
NodeShader.write(frag, 'n = ntex * 2.0 - 1.0;');
NodeShader.write(frag, 'n.y = -n.y;');
NodeShader.write(frag, 'n = normalize(mul(n, TBN));');
if (Context.raw.viewportMode == ViewportMode.ViewLit) {
NodeShader.write(frag, 'basecol = pow(basecol, vec3(2.2, 2.2, 2.2));');
if (Context.raw.viewportShader != null) {
let color = Context.raw.viewportShader(frag);
NodeShader.write(frag, `fragColor[1] = vec4(${color}, 1.0);`);
}
else if (Context.raw.renderMode == RenderMode.RenderForward) {
frag.wposition = true;
NodeShader.write(frag, 'vec3 albedo = mix(basecol, vec3(0.0, 0.0, 0.0), metallic);');
NodeShader.write(frag, 'vec3 f0 = mix(vec3(0.04, 0.04, 0.04), basecol, metallic);');
frag.vVec = true;
NodeShader.write(frag, 'float dotNV = max(0.0, dot(n, vVec));');
NodeShader.write(frag, 'vec2 envBRDF = texelFetch(senvmapBrdf, ivec2(vec2(roughness, 1.0 - dotNV) * 256.0), 0).xy;');
NodeShader.add_uniform(frag, 'int envmapNumMipmaps', '_envmapNumMipmaps');
NodeShader.add_uniform(frag, 'vec4 envmapData', '_envmapData'); // angle, sin(angle), cos(angle), strength
NodeShader.write(frag, 'vec3 wreflect = reflect(-vVec, n);');
NodeShader.write(frag, 'float envlod = roughness * float(envmapNumMipmaps);');
NodeShader.add_function(frag, ShaderFunctions.str_envMapEquirect);
NodeShader.write(frag, 'vec3 prefilteredColor = textureLod(senvmapRadiance, envMapEquirect(wreflect, envmapData.x), envlod).rgb;');
NodeShader.add_uniform(frag, 'vec3 lightArea0', '_lightArea0');
NodeShader.add_uniform(frag, 'vec3 lightArea1', '_lightArea1');
NodeShader.add_uniform(frag, 'vec3 lightArea2', '_lightArea2');
NodeShader.add_uniform(frag, 'vec3 lightArea3', '_lightArea3');
NodeShader.add_function(frag, ShaderFunctions.str_ltcEvaluate);
NodeShader.add_uniform(frag, 'vec3 lightPos', '_pointPosition');
NodeShader.add_uniform(frag, 'vec3 lightColor', '_pointColor');
NodeShader.write(frag, 'float ldist = distance(wposition, lightPos);');
NodeShader.write(frag, 'const float LUT_SIZE = 64.0;');
NodeShader.write(frag, 'const float LUT_SCALE = (LUT_SIZE - 1.0) / LUT_SIZE;');
NodeShader.write(frag, 'const float LUT_BIAS = 0.5 / LUT_SIZE;');
NodeShader.write(frag, 'float theta = acos(dotNV);');
NodeShader.write(frag, 'vec2 tuv = vec2(roughness, theta / (0.5 * 3.14159265));');
NodeShader.write(frag, 'tuv = tuv * LUT_SCALE + LUT_BIAS;');
NodeShader.write(frag, 'vec4 t = textureLod(sltcMat, tuv, 0.0);');
NodeShader.write(frag, 'mat3 minv = mat3(vec3(1.0, 0.0, t.y), vec3(0.0, t.z, 0.0), vec3(t.w, 0.0, t.x));');
NodeShader.write(frag, 'float ltcspec = ltcEvaluate(n, vVec, dotNV, wposition, minv, lightArea0, lightArea1, lightArea2, lightArea3);');
NodeShader.write(frag, 'ltcspec *= textureLod(sltcMag, tuv, 0.0).a;');
NodeShader.write(frag, 'mat3 mident = mat3(1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0);');
NodeShader.write(frag, 'float ltcdiff = ltcEvaluate(n, vVec, dotNV, wposition, mident, lightArea0, lightArea1, lightArea2, lightArea3);');
NodeShader.write(frag, 'vec3 direct = albedo * ltcdiff + ltcspec * 0.05;');
NodeShader.write(frag, 'direct *= lightColor * (1.0 / (ldist * ldist));');
NodeShader.add_uniform(frag, 'vec4 shirr[7]', '_envmapIrradiance');
NodeShader.add_function(frag, ShaderFunctions.str_shIrradiance);
NodeShader.write(frag, 'vec3 indirect = albedo * (shIrradiance(vec3(n.x * envmapData.z - n.y * envmapData.y, n.x * envmapData.y + n.y * envmapData.z, n.z), shirr) / 3.14159265);');
NodeShader.write(frag, 'indirect += prefilteredColor * (f0 * envBRDF.x + envBRDF.y) * 1.5;');
NodeShader.write(frag, 'indirect *= envmapData.w * occlusion;');
NodeShader.write(frag, 'fragColor[1] = vec4(direct + indirect, 1.0);');
}
else { // Deferred, Pathtraced
if (MakeMaterial.emisUsed) NodeShader.write(frag, 'if (int(matid * 255.0) % 3 == 1) basecol *= 10.0;'); // Boost for bloom
NodeShader.write(frag, 'fragColor[1] = vec4(basecol, occlusion);');
}
}
else if (Context.raw.viewportMode == ViewportMode.ViewObjectNormal) {
frag.nAttr = true;
NodeShader.write(frag, 'fragColor[1] = vec4(nAttr, 1.0);');
}
else if (Context.raw.viewportMode == ViewportMode.ViewObjectID) {
NodeShader.add_uniform(frag, 'float objectId', '_objectId');
NodeShader.write(frag, 'float obid = objectId + 1.0 / 255.0;');
NodeShader.write(frag, 'float id_r = fract(sin(dot(vec2(obid, obid * 20.0), vec2(12.9898, 78.233))) * 43758.5453);');
NodeShader.write(frag, 'float id_g = fract(sin(dot(vec2(obid * 20.0, obid), vec2(12.9898, 78.233))) * 43758.5453);');
NodeShader.write(frag, 'float id_b = fract(sin(dot(vec2(obid, obid * 40.0), vec2(12.9898, 78.233))) * 43758.5453);');
NodeShader.write(frag, 'fragColor[1] = vec4(id_r, id_g, id_b, 1.0);');
}
else {
NodeShader.write(frag, 'fragColor[1] = vec4(1.0, 0.0, 1.0, 1.0);'); // Pink
}
if (Context.raw.viewportMode != ViewportMode.ViewLit && Context.raw.viewportMode != ViewportMode.ViewPathTrace) {
NodeShader.write(frag, 'fragColor[1].rgb = pow(fragColor[1].rgb, vec3(2.2, 2.2, 2.2));');
}
NodeShader.write(frag, 'n /= (abs(n.x) + abs(n.y) + abs(n.z));');
NodeShader.write(frag, 'n.xy = n.z >= 0.0 ? n.xy : octahedronWrap(n.xy);');
NodeShader.write(frag, 'fragColor[0] = vec4(n.xy, roughness, packFloatInt16(metallic, uint(int(matid * 255.0) % 3)));');
}
NodeShader.write(frag, 'vec2 posa = (wvpposition.xy / wvpposition.w) * 0.5 + 0.5;');
NodeShader.write(frag, 'vec2 posb = (prevwvpposition.xy / prevwvpposition.w) * 0.5 + 0.5;');
NodeShader.write(frag, 'fragColor[2] = vec4(posa - posb, texCoord.xy);');
ParserMaterial.finalize(con_mesh);
con_mesh.data.shader_from_source = true;
con_mesh.data.vertex_shader = NodeShader.get(vert);
con_mesh.data.fragment_shader = NodeShader.get(frag);
return con_mesh;
}
static getMaxTextures = (): i32 => {
///if krom_direct3d11
return 128 - 66;
///else
return 16 - 3; // G4onG5/G4.c.h MAX_TEXTURES
///end
}
}
Reference in New Issue View Git Blame Copy Permalink