@woosh/meep-engine 3.11.3 → 3.12.0

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

@@ -1,180 +1,180 @@
-import { RenderPipelineDescriptor } from "../../../../descriptor/pipeline/render/RenderPipelineDescriptor.js";
-import { CAMERA_UNIFORM_STRUCT } from "../../../camera/CAMERA_UNIFORM_STRUCT.js";
-import { SHADING_LIGHT_RESOURCE_GROUP } from "../../../deferred/SHADING_LIGHT_RESOURCE_GROUP.js";
-import { DEPTH_BACKGROUND_VALUE } from "../../../depth/DEPTH_BACKGROUND_VALUE.js";
-import { BRICK4_DATA_STRUCT } from "../../../global_illumination/brick4/gpu/BRICK4_DATA_STRUCT.js";
-import { VOLUMETRICS_METADATA_STRUCT } from "../../../volumetrics/VOLUMETRICS_METADATA_STRUCT.js";
-import { chunk_sample_volumetrics } from "../../../deferred/chunk_sample_volumetrics.js";
-import { SHADING_MATERIAL_RESOURCE_GROUP } from "../../../material/standard/SHADING_MATERIAL_RESOURCE_GROUP.js";
-import {
-    chunk_resolve_power_moments_4
-} from "../../../shader/chunk/mboit/moments/power/chunk_resolve_power_moments_4.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 { SAMPLER_LINEAR_UPSCALE } from "../../../texture/SAMPLER_LINEAR_UPSCALE.js";
-import { VIEW_UNIFORM_STRUCT } from "../../../view/VIEW_UNIFORM_STRUCT.js";
-import { rasterization_primitive_state } from "../rasterization_primitive_state.js";
-import { chunk_forward_shade_standard_fragment_brick4 } from "./chunk_forward_shade_standard_fragment_brick4.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)
-    .addTexture('moments', 'unfilterable-float')
-    .addTexture('optical_depth', 'unfilterable-float')
-    .addStorageBuffer("brick4",BRICK4_DATA_STRUCT, false)
-    .addTexture("tSplitSum", "float")
-    .addSampler("sLinear","filtering", SAMPLER_LINEAR_UPSCALE)
-    // volumetrics (composited inline below)
-    .addUniform("volumetrics_metadata", VOLUMETRICS_METADATA_STRUCT)
-    .addTexture("tVolumetrics", "float", "3d")
-    .addTexture("tSTBN_vec3", "float", "3d")
-
-resources.addGroup(SHADING_LIGHT_RESOURCE_GROUP);
-
-const shader_vertex = ShaderDescriptor.from({
-    label: 'OIT / vertex shader',
-    body: vertex_shader_chunk,
-    resources,
-});
-
-
-const shader_fragment = ShaderDescriptor.from({
-    label: "OIT / resolve moments",
-    resources,
-    body: CodeChunk.from(
-        //language=WGSL
-        `
-@fragment
-fn main(
-    @builtin(front_facing) front_facing : bool,
-    vert: ${FORWARD_VERTEX_SHADER_STRUCT.wgsl_ref},
-) -> @location(0) vec4<f32>{
-
-    let coord_u = vec2u(vert.position.xy);
-    
-    // this is total absorbance
-    let b_0 = textureLoad(optical_depth, coord_u, 0).r;
-        
-    if(b_0 < 0.00100050033f){
-        // indicates that pixel has no opacity
-        // would result in division by 0
-        discard;
-    }
-    
-    // need to unjitter UV to account for TAA
-    var vert_adjusted = vert;
-    vert_adjusted.uv = unjitter_uv(vert.uv, view.jitter);
-    
-    let shaded = forward_shade_standard_fragment(vert_adjusted, front_facing);
-
-    // composite volumetric fog between the camera and this transparent fragment. Use the linear
-    // view-space depth from the fragment's device depth (vert.position is @builtin(position), so
-    // .z is the device depth — the same metric the opaque path feeds the froxel lookup). NOT
-    // oit_linearize_depth, which returns a normalized [0,1] log-depth for the moment warp and would
-    // collapse every fragment onto a near slice (no fog).
-    let frag_view_depth = get_view_space_depth(vert_adjusted.position.z, camera);
-    let frag_screen_uv = vert_adjusted.position.xy / vec2<f32>(f32(view.width), f32(view.height));
-    let volumetrics = sample_volumetrics(frag_view_depth, frag_screen_uv, vert_adjusted.position.xy);
-
-    let color = shaded.rgb * volumetrics.w + volumetrics.xyz;
-    let opacity = shaded.a;
-    
-    // if opacity is 1 - we end up with 0 transmittance, which breaks the algorithm
-    let coverage = min(opacity, ${OIT_COVERAGE_MAX});
-
-    let total_moments1234 = textureLoad(moments, coord_u, 0);
-    
-    // linear distance to fragment (vert_adjusted.position is @builtin(position); .z is device depth)
-    let depth = oit_linearize_depth(vert_adjusted.position.z, camera);
-    
-    let transmittance = resolve_power_moments_4( 
-        depth,
-        b_0,
-        total_moments1234,
-    );
-        
-    return vec4(
-        color * coverage, // premultiply
-        coverage
-    ) * transmittance;
-
-}
-        `, [
-            FORWARD_VERTEX_SHADER_STRUCT.declaration_chunk,
-            chunk_forward_shade_standard_fragment_brick4,
-            chunk_resolve_power_moments_4,
-            chunk_oit_linearize_depth,
-            chunk_unjitter_uv,
-            chunk_sample_volumetrics,
-        ]
-    )
-});
-
-
-const pipeline = RenderPipelineDescriptor.from({
-    label: "OIT / resolve moments",
-    layout: resources.generatePipelineLayout(GPUShaderStage.VERTEX | GPUShaderStage.FRAGMENT),
-    vertex: {
-        module: {
-            code: shader_vertex.compile()
-        }
-    },
-    fragment: {
-        module: {
-            code: shader_fragment.compile()
-        },
-        targets: [
-            // HDR color output
-            {
-                format: "rgba16float",
-                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_resolve_moments = new RenderPassDescriptor();
-shader_oit_resolve_moments.pipeline = pipeline;
-shader_oit_resolve_moments.resources = resources;
-shader_oit_resolve_moments.colorAttachments = [
-    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 { SHADING_LIGHT_RESOURCE_GROUP } from "../../../deferred/SHADING_LIGHT_RESOURCE_GROUP.js";
+import { DEPTH_BACKGROUND_VALUE } from "../../../depth/DEPTH_BACKGROUND_VALUE.js";
+import { BRICK4_DATA_STRUCT } from "../../../global_illumination/brick4/gpu/BRICK4_DATA_STRUCT.js";
+import { VOLUMETRICS_METADATA_STRUCT } from "../../../volumetrics/VOLUMETRICS_METADATA_STRUCT.js";
+import { chunk_sample_volumetrics } from "../../../deferred/chunk_sample_volumetrics.js";
+import { SHADING_MATERIAL_RESOURCE_GROUP } from "../../../material/standard/SHADING_MATERIAL_RESOURCE_GROUP.js";
+import {
+    chunk_resolve_power_moments_4
+} from "../../../shader/chunk/mboit/moments/power/chunk_resolve_power_moments_4.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 { SAMPLER_LINEAR_UPSCALE } from "../../../texture/SAMPLER_LINEAR_UPSCALE.js";
+import { VIEW_UNIFORM_STRUCT } from "../../../view/VIEW_UNIFORM_STRUCT.js";
+import { rasterization_primitive_state } from "../rasterization_primitive_state.js";
+import { chunk_forward_shade_standard_fragment_brick4 } from "./chunk_forward_shade_standard_fragment_brick4.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)
+    .addTexture('moments', 'unfilterable-float')
+    .addTexture('optical_depth', 'unfilterable-float')
+    .addStorageBuffer("brick4",BRICK4_DATA_STRUCT, false)
+    .addTexture("tSplitSum", "float")
+    .addSampler("sLinear","filtering", SAMPLER_LINEAR_UPSCALE)
+    // volumetrics (composited inline below)
+    .addUniform("volumetrics_metadata", VOLUMETRICS_METADATA_STRUCT)
+    .addTexture("tVolumetrics", "float", "3d")
+    .addTexture("tSTBN_vec3", "float", "3d")
+
+resources.addGroup(SHADING_LIGHT_RESOURCE_GROUP);
+
+const shader_vertex = ShaderDescriptor.from({
+    label: 'OIT / vertex shader',
+    body: vertex_shader_chunk,
+    resources,
+});
+
+
+const shader_fragment = ShaderDescriptor.from({
+    label: "OIT / resolve moments",
+    resources,
+    body: CodeChunk.from(
+        //language=WGSL
+        `
+@fragment
+fn main(
+    @builtin(front_facing) front_facing : bool,
+    vert: ${FORWARD_VERTEX_SHADER_STRUCT.wgsl_ref},
+) -> @location(0) vec4<f32>{
+
+    let coord_u = vec2u(vert.position.xy);
+    
+    // this is total absorbance
+    let b_0 = textureLoad(optical_depth, coord_u, 0).r;
+        
+    if(b_0 < 0.00100050033f){
+        // indicates that pixel has no opacity
+        // would result in division by 0
+        discard;
+    }
+    
+    // need to unjitter UV to account for TAA
+    var vert_adjusted = vert;
+    vert_adjusted.uv = unjitter_uv(vert.uv, view.jitter);
+    
+    let shaded = forward_shade_standard_fragment(vert_adjusted, front_facing);
+
+    // composite volumetric fog between the camera and this transparent fragment. Use the linear
+    // view-space depth from the fragment's device depth (vert.position is @builtin(position), so
+    // .z is the device depth — the same metric the opaque path feeds the froxel lookup). NOT
+    // oit_linearize_depth, which returns a normalized [0,1] log-depth for the moment warp and would
+    // collapse every fragment onto a near slice (no fog).
+    let frag_view_depth = get_view_space_depth(vert_adjusted.position.z, camera);
+    let frag_screen_uv = vert_adjusted.position.xy / vec2<f32>(f32(view.width), f32(view.height));
+    let volumetrics = sample_volumetrics(frag_view_depth, frag_screen_uv, vert_adjusted.position.xy);
+
+    let color = shaded.rgb * volumetrics.w + volumetrics.xyz;
+    let opacity = shaded.a;
+    
+    // if opacity is 1 - we end up with 0 transmittance, which breaks the algorithm
+    let coverage = min(opacity, ${OIT_COVERAGE_MAX});
+
+    let total_moments1234 = textureLoad(moments, coord_u, 0);
+    
+    // linear distance to fragment (vert_adjusted.position is @builtin(position); .z is device depth)
+    let depth = oit_linearize_depth(vert_adjusted.position.z, camera);
+    
+    let transmittance = resolve_power_moments_4( 
+        depth,
+        b_0,
+        total_moments1234,
+    );
+        
+    return vec4(
+        color * coverage, // premultiply
+        coverage
+    ) * transmittance;
+
+}
+        `, [
+            FORWARD_VERTEX_SHADER_STRUCT.declaration_chunk,
+            chunk_forward_shade_standard_fragment_brick4,
+            chunk_resolve_power_moments_4,
+            chunk_oit_linearize_depth,
+            chunk_unjitter_uv,
+            chunk_sample_volumetrics,
+        ]
+    )
+});
+
+
+const pipeline = RenderPipelineDescriptor.from({
+    label: "OIT / resolve moments",
+    layout: resources.generatePipelineLayout(GPUShaderStage.VERTEX | GPUShaderStage.FRAGMENT),
+    vertex: {
+        module: {
+            code: shader_vertex.source
+        }
+    },
+    fragment: {
+        module: {
+            code: shader_fragment.source
+        },
+        targets: [
+            // HDR color output
+            {
+                format: "rgba16float",
+                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_resolve_moments = new RenderPassDescriptor();
+shader_oit_resolve_moments.pipeline = pipeline;
+shader_oit_resolve_moments.resources = resources;
+shader_oit_resolve_moments.colorAttachments = [
+    RenderPassColorAttachmentDescriptor.from({
+        clearValue: [0, 0, 0, 0],
+    })
+];
+/**
+ * Clear depth to 0 for reverse Z technique
+ */
 shader_oit_resolve_moments.depthStencilAttachment.depthClearValue = DEPTH_BACKGROUND_VALUE;