@woosh/meep-engine 3.11.2 → 3.12.0

package/src/shade/renderer/rasterize/native/oit/shader_oit_generate_moments.js

@@ -1,218 +1,218 @@
-import { RenderPipelineDescriptor } from "../../../../descriptor/pipeline/render/RenderPipelineDescriptor.js";
-import { CAMERA_UNIFORM_STRUCT } from "../../../camera/CAMERA_UNIFORM_STRUCT.js";
-import { DEPTH_BACKGROUND_VALUE } from "../../../depth/DEPTH_BACKGROUND_VALUE.js";
-import { SHADING_MATERIAL_RESOURCE_GROUP } from "../../../material/standard/SHADING_MATERIAL_RESOURCE_GROUP.js";
-import { chunk_F_Hauber } from "../../../shader/chunk/brdf/fresnel/chunk_F_Hauber.js";
-import { chunk_mat4_extract_position } from "../../../shader/chunk/geometry/mat4/chunk_mat4_extract_position.js";
-import { chunk_dielectric_specular_color } from "../../../shader/chunk/material/chunk_dielectric_specular_color.js";
-import { chunk_unjitter_uv } from "../../../shader/chunk/texture/chunk_unjitter_uv.js";
-import { CodeChunk } from "../../../shader/compiler/CodeChunk.js";
-import { RenderPassColorAttachmentDescriptor } from "../../../shader/pass/RenderPassColorAttachmentDescriptor.js";
-import { RenderPassDescriptor } from "../../../shader/pass/RenderPassDescriptor.js";
-import { ShaderResourceSetDescriptor } from "../../../shader/resource/ShaderResourceSetDescriptor.js";
-import { ShaderDescriptor } from "../../../shader/ShaderDescriptor.js";
-import { VIEW_UNIFORM_STRUCT } from "../../../view/VIEW_UNIFORM_STRUCT.js";
-import { rasterization_primitive_state } from "../rasterization_primitive_state.js";
-import {
-    viz_rasterization_alpha_tested_pass_descriptor
-} from "../viz/viz_rasterization_alpha_tested_pass_descriptor.js";
-import { chunk_oit_linearize_depth } from "./chunk_oit_linearize_depth.js";
-import { FORWARD_VERTEX_SHADER_STRUCT } from "./FORWARD_VERTEX_SHADER_STRUCT.js";
-import { OIT_COVERAGE_MAX } from "./OIT_CONSTANTS.js";
-import { SHADING_SCENE_RESOURCE_GROUP } from "./SHADING_SCENE_RESOURCE_GROUP.js";
-import { vertex_shader_chunk } from "./vertex_shader_chunk.js";
-
-
-const resources = new ShaderResourceSetDescriptor();
-
-resources.addGroup(SHADING_MATERIAL_RESOURCE_GROUP);
-resources.addGroup(SHADING_SCENE_RESOURCE_GROUP);
-
-resources.createGroup()
-    .addUniform("camera", CAMERA_UNIFORM_STRUCT)
-    .addUniform("view", VIEW_UNIFORM_STRUCT)
-
-const shader_vertex = ShaderDescriptor.from({
-    label: 'OIT / vertex shader',
-    body: vertex_shader_chunk,
-    resources,
-});
-
-const fragment_body = CodeChunk.from(
-    //language=WGSL
-    `
-struct Output{
-    @location(0) b_0 : f32,
-    @location(1) moments : vec4f,
-}
-
-fn generate_moments(
-    depth: f32,
-    transmittance: f32,
-    b_0: ptr<function , f32>,
-    b: ptr<function , vec4f>,
-){
-    let absorbance = -log(transmittance);
-    
-    // Stored Total Transimittance.
-    *b_0 = absorbance; // 1 * absorbance
-    
-    let depth2 = depth * depth;
-    let depth4 = depth2 * depth2;
-    
-    // Moments Vector. Stored Transimittance.
-    *b = vec4(depth, depth2, depth2 * depth, depth4) * absorbance;
-}
-
-@fragment
-fn main(
-    @builtin(front_facing) front_facing : bool,
-    vert: ${FORWARD_VERTEX_SHADER_STRUCT.wgsl_ref},
-) -> Output {
-
-    // compensate for TAA jitter
-    var uv = unjitter_uv(vert.uv, view.jitter);
-
-    let t_diffuse = textureSample(texture_diffuse, texture_diffuse_sampler, uv);
-    let surface_alpha = t_diffuse.a * material_info.albedo_color.a;
-
-    // MaterialX/OpenPBR transmission: when transmission_factor > 0, coverage
-    // is driven by view-angle Fresnel instead of the diffuse-texture alpha.
-    // This is the rasterizer's only material-PBR coupling in the moments-gen
-    // pass — the price of doing glass right under scalar MBOIT, but it's a
-    // per-fragment constant cost (one ORM sample + one Fresnel call).
-    var opacity = surface_alpha;
-    let f_transmission = material_info.transmission_factor;
-    if (f_transmission > 0.0) {
-        let t_pbr = textureSample(texture_orm, texture_orm_sampler, uv);
-        var diffuse_rgb = t_diffuse.rgb;
-        diffuse_rgb /= max(t_diffuse.a, 1e-7);
-        let f_metalness = t_pbr.b * material_info.metallic_factor;
-        let specular_f0 = dielectric_specular_color(material_info.ior_factor, f_metalness, diffuse_rgb);
-
-        // vert.normal is in world space already; abs(NoV) handles back-facing
-        // without needing the tangent-frame perturbation done in the full
-        // forward shader.
-        let view_position = mat4_extract_position(camera.transform);
-        let view_direction = normalize(view_position - vert.position_ws);
-        let nov = saturate(abs(dot(normalize(vert.normal), view_direction)));
-        let f_view = F_Hauber(specular_f0, 1.0, nov);
-        let f_scalar = max(f_view.r, max(f_view.g, f_view.b));
-        // No per-channel transmission tint in the material today — scalar
-        // MBOIT can't filter the background per-channel. For tinted/dark
-        // glass use diffuse_color.a (legacy alpha-blend hack).
-        opacity = mix(surface_alpha, f_scalar, f_transmission);
-    }
-
-    let coverage = min(opacity, ${OIT_COVERAGE_MAX});
-
-    let transmittance = 1.0 - coverage;
-
-    // linear distance to fragment (vert.position is @builtin(position); .z is device depth)
-    let depth = oit_linearize_depth(vert.position.z, camera);
-
-    var b_0: f32;
-
-    var moments:vec4f;
-
-    generate_moments(
-        depth,
-        transmittance,
-        &b_0,
-        &moments,
-    );
-
-    var out: Output;
-
-    out.b_0 = b_0;
-    out.moments = moments;
-
-    return out;
-}
-    `, [
-        FORWARD_VERTEX_SHADER_STRUCT.declaration_chunk,
-        chunk_oit_linearize_depth,
-        chunk_unjitter_uv,
-        chunk_mat4_extract_position,
-        chunk_dielectric_specular_color,
-        chunk_F_Hauber,
-    ]
-);
-
-const shader_fragment = ShaderDescriptor.from({
-    label: "OIT / generate moments",
-    resources,
-    body: fragment_body
-});
-
-const pipeline = RenderPipelineDescriptor.from({
-    label: "OIT / generate moments",
-    layout: resources.generatePipelineLayout(GPUShaderStage.VERTEX | GPUShaderStage.FRAGMENT),
-    vertex: {
-        module: {
-            code: shader_vertex.compile()
-        }
-    },
-    fragment: {
-        module: {
-            code: shader_fragment.compile()
-        },
-        targets: [
-            // optical depth (moment 0)
-            {
-                format: "r32float",
-                blend: {
-                    color: {
-                        operation: "add",
-                        srcFactor: "one",
-                        dstFactor: "one",
-                    },
-                    alpha: {
-                        operation: "add",
-                        srcFactor: "one",
-                        dstFactor: "one",
-                    }
-                }
-            },
-            // 4 moments (1,2,3,4)
-            {
-                format: "rgba32float",
-                blend: {
-                    color: {
-                        operation: "add",
-                        srcFactor: "one",
-                        dstFactor: "one",
-                    },
-                    alpha: {
-                        operation: "add",
-                        srcFactor: "one",
-                        dstFactor: "one",
-                    }
-                }
-            }
-        ]
-    },
-    depthStencil: {
-        // reversed Z
-        depthCompare: "greater",
-        format: "depth32float",
-        depthWriteEnabled: false
-    },
-    primitive: rasterization_primitive_state,
-});
-
-export const shader_oit_generate_moments = new RenderPassDescriptor();
-shader_oit_generate_moments.pipeline = pipeline;
-shader_oit_generate_moments.resources = resources;
-shader_oit_generate_moments.colorAttachments = [
-    RenderPassColorAttachmentDescriptor.from({
-        clearValue: [0, 0, 0, 0],
-    }),
-    RenderPassColorAttachmentDescriptor.from({
-        clearValue: [0, 0, 0, 0],
-    }),
-];
-/**
- * Clear depth to 0 for reverse Z technique
- */
+import { RenderPipelineDescriptor } from "../../../../descriptor/pipeline/render/RenderPipelineDescriptor.js";
+import { CAMERA_UNIFORM_STRUCT } from "../../../camera/CAMERA_UNIFORM_STRUCT.js";
+import { DEPTH_BACKGROUND_VALUE } from "../../../depth/DEPTH_BACKGROUND_VALUE.js";
+import { SHADING_MATERIAL_RESOURCE_GROUP } from "../../../material/standard/SHADING_MATERIAL_RESOURCE_GROUP.js";
+import { chunk_F_Hauber } from "../../../shader/chunk/brdf/fresnel/chunk_F_Hauber.js";
+import { chunk_mat4_extract_position } from "../../../shader/chunk/geometry/mat4/chunk_mat4_extract_position.js";
+import { chunk_dielectric_specular_color } from "../../../shader/chunk/material/chunk_dielectric_specular_color.js";
+import { chunk_unjitter_uv } from "../../../shader/chunk/texture/chunk_unjitter_uv.js";
+import { CodeChunk } from "../../../shader/compiler/CodeChunk.js";
+import { RenderPassColorAttachmentDescriptor } from "../../../shader/pass/RenderPassColorAttachmentDescriptor.js";
+import { RenderPassDescriptor } from "../../../shader/pass/RenderPassDescriptor.js";
+import { ShaderResourceSetDescriptor } from "../../../shader/resource/ShaderResourceSetDescriptor.js";
+import { ShaderDescriptor } from "../../../shader/ShaderDescriptor.js";
+import { VIEW_UNIFORM_STRUCT } from "../../../view/VIEW_UNIFORM_STRUCT.js";
+import { rasterization_primitive_state } from "../rasterization_primitive_state.js";
+import {
+    viz_rasterization_alpha_tested_pass_descriptor
+} from "../viz/viz_rasterization_alpha_tested_pass_descriptor.js";
+import { chunk_oit_linearize_depth } from "./chunk_oit_linearize_depth.js";
+import { FORWARD_VERTEX_SHADER_STRUCT } from "./FORWARD_VERTEX_SHADER_STRUCT.js";
+import { OIT_COVERAGE_MAX } from "./OIT_CONSTANTS.js";
+import { SHADING_SCENE_RESOURCE_GROUP } from "./SHADING_SCENE_RESOURCE_GROUP.js";
+import { vertex_shader_chunk } from "./vertex_shader_chunk.js";
+
+
+const resources = new ShaderResourceSetDescriptor();
+
+resources.addGroup(SHADING_MATERIAL_RESOURCE_GROUP);
+resources.addGroup(SHADING_SCENE_RESOURCE_GROUP);
+
+resources.createGroup()
+    .addUniform("camera", CAMERA_UNIFORM_STRUCT)
+    .addUniform("view", VIEW_UNIFORM_STRUCT)
+
+const shader_vertex = ShaderDescriptor.from({
+    label: 'OIT / vertex shader',
+    body: vertex_shader_chunk,
+    resources,
+});
+
+const fragment_body = CodeChunk.from(
+    //language=WGSL
+    `
+struct Output{
+    @location(0) b_0 : f32,
+    @location(1) moments : vec4f,
+}
+
+fn generate_moments(
+    depth: f32,
+    transmittance: f32,
+    b_0: ptr<function , f32>,
+    b: ptr<function , vec4f>,
+){
+    let absorbance = -log(transmittance);
+    
+    // Stored Total Transimittance.
+    *b_0 = absorbance; // 1 * absorbance
+    
+    let depth2 = depth * depth;
+    let depth4 = depth2 * depth2;
+    
+    // Moments Vector. Stored Transimittance.
+    *b = vec4(depth, depth2, depth2 * depth, depth4) * absorbance;
+}
+
+@fragment
+fn main(
+    @builtin(front_facing) front_facing : bool,
+    vert: ${FORWARD_VERTEX_SHADER_STRUCT.wgsl_ref},
+) -> Output {
+
+    // compensate for TAA jitter
+    var uv = unjitter_uv(vert.uv, view.jitter);
+
+    let t_diffuse = textureSample(texture_diffuse, texture_diffuse_sampler, uv);
+    let surface_alpha = t_diffuse.a * material_info.albedo_color.a;
+
+    // MaterialX/OpenPBR transmission: when transmission_factor > 0, coverage
+    // is driven by view-angle Fresnel instead of the diffuse-texture alpha.
+    // This is the rasterizer's only material-PBR coupling in the moments-gen
+    // pass — the price of doing glass right under scalar MBOIT, but it's a
+    // per-fragment constant cost (one ORM sample + one Fresnel call).
+    var opacity = surface_alpha;
+    let f_transmission = material_info.transmission_factor;
+    if (f_transmission > 0.0) {
+        let t_pbr = textureSample(texture_orm, texture_orm_sampler, uv);
+        var diffuse_rgb = t_diffuse.rgb;
+        diffuse_rgb /= max(t_diffuse.a, 1e-7);
+        let f_metalness = t_pbr.b * material_info.metallic_factor;
+        let specular_f0 = dielectric_specular_color(material_info.ior_factor, f_metalness, diffuse_rgb);
+
+        // vert.normal is in world space already; abs(NoV) handles back-facing
+        // without needing the tangent-frame perturbation done in the full
+        // forward shader.
+        let view_position = mat4_extract_position(camera.transform);
+        let view_direction = normalize(view_position - vert.position_ws);
+        let nov = saturate(abs(dot(normalize(vert.normal), view_direction)));
+        let f_view = F_Hauber(specular_f0, 1.0, nov);
+        let f_scalar = max(f_view.r, max(f_view.g, f_view.b));
+        // No per-channel transmission tint in the material today — scalar
+        // MBOIT can't filter the background per-channel. For tinted/dark
+        // glass use diffuse_color.a (legacy alpha-blend hack).
+        opacity = mix(surface_alpha, f_scalar, f_transmission);
+    }
+
+    let coverage = min(opacity, ${OIT_COVERAGE_MAX});
+
+    let transmittance = 1.0 - coverage;
+
+    // linear distance to fragment (vert.position is @builtin(position); .z is device depth)
+    let depth = oit_linearize_depth(vert.position.z, camera);
+
+    var b_0: f32;
+
+    var moments:vec4f;
+
+    generate_moments(
+        depth,
+        transmittance,
+        &b_0,
+        &moments,
+    );
+
+    var out: Output;
+
+    out.b_0 = b_0;
+    out.moments = moments;
+
+    return out;
+}
+    `, [
+        FORWARD_VERTEX_SHADER_STRUCT.declaration_chunk,
+        chunk_oit_linearize_depth,
+        chunk_unjitter_uv,
+        chunk_mat4_extract_position,
+        chunk_dielectric_specular_color,
+        chunk_F_Hauber,
+    ]
+);
+
+const shader_fragment = ShaderDescriptor.from({
+    label: "OIT / generate moments",
+    resources,
+    body: fragment_body
+});
+
+const pipeline = RenderPipelineDescriptor.from({
+    label: "OIT / generate moments",
+    layout: resources.generatePipelineLayout(GPUShaderStage.VERTEX | GPUShaderStage.FRAGMENT),
+    vertex: {
+        module: {
+            code: shader_vertex.source
+        }
+    },
+    fragment: {
+        module: {
+            code: shader_fragment.source
+        },
+        targets: [
+            // optical depth (moment 0)
+            {
+                format: "r32float",
+                blend: {
+                    color: {
+                        operation: "add",
+                        srcFactor: "one",
+                        dstFactor: "one",
+                    },
+                    alpha: {
+                        operation: "add",
+                        srcFactor: "one",
+                        dstFactor: "one",
+                    }
+                }
+            },
+            // 4 moments (1,2,3,4)
+            {
+                format: "rgba32float",
+                blend: {
+                    color: {
+                        operation: "add",
+                        srcFactor: "one",
+                        dstFactor: "one",
+                    },
+                    alpha: {
+                        operation: "add",
+                        srcFactor: "one",
+                        dstFactor: "one",
+                    }
+                }
+            }
+        ]
+    },
+    depthStencil: {
+        // reversed Z
+        depthCompare: "greater",
+        format: "depth32float",
+        depthWriteEnabled: false
+    },
+    primitive: rasterization_primitive_state,
+});
+
+export const shader_oit_generate_moments = new RenderPassDescriptor();
+shader_oit_generate_moments.pipeline = pipeline;
+shader_oit_generate_moments.resources = resources;
+shader_oit_generate_moments.colorAttachments = [
+    RenderPassColorAttachmentDescriptor.from({
+        clearValue: [0, 0, 0, 0],
+    }),
+    RenderPassColorAttachmentDescriptor.from({
+        clearValue: [0, 0, 0, 0],
+    }),
+];
+/**
+ * Clear depth to 0 for reverse Z technique
+ */
 viz_rasterization_alpha_tested_pass_descriptor.depthStencilAttachment.depthClearValue = DEPTH_BACKGROUND_VALUE;