@luma.gl/shaders 10.0.0-alpha.10 → 10.0.0-alpha.12

package/src/modules/lighting/contact-shadow/contact-shadow.ts

@@ -0,0 +1,365 @@
+// luma.gl
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: Copyright (c) vis.gl contributors
+
+import type {NumberArray3, NumberArray16} from '@math.gl/core';
+import type {ShaderPass} from '../../../lib/shader-module/shader-pass';
+
+const IDENTITY_MATRIX: NumberArray16 = [1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1];
+
+export type ContactShadowBindings<TextureT = unknown> = {
+  depthTexture?: TextureT;
+  normalTexture?: TextureT;
+};
+
+export type ContactTraceUniforms = {
+  projectionMatrix: Readonly<NumberArray16>;
+  inverseProjectionMatrix: Readonly<NumberArray16>;
+  lightDirectionView: Readonly<NumberArray3>;
+  maxDistance: number;
+  thickness: number;
+  stepCount: number;
+  frameIndex: number;
+  normalBias: number;
+};
+
+/** Runtime controls accepted by the contact-shadow trace and composite passes. */
+export type ContactShadowProps<TextureT = unknown> = Partial<ContactTraceUniforms> & {
+  strength?: number;
+  debugMode?: number;
+  directionalDirectTexture?: TextureT;
+  shadowDebugTexture?: TextureT;
+};
+
+export const contactShadowTrace = {
+  name: 'contactShadowTrace',
+  source: /* wgsl */ `\
+struct contactShadowTraceUniforms {
+  projectionMatrix: mat4x4f,
+  inverseProjectionMatrix: mat4x4f,
+  lightDirectionView: vec3f,
+  maxDistance: f32,
+  thickness: f32,
+  stepCount: f32,
+  frameIndex: f32,
+  normalBias: f32,
+};
+@group(0) @binding(auto) var<uniform> contactShadowTrace: contactShadowTraceUniforms;
+@group(0) @binding(auto) var depthTexture: texture_depth_2d;
+@group(0) @binding(auto) var depthTextureSampler: sampler;
+@group(0) @binding(auto) var normalTexture: texture_2d<f32>;
+@group(0) @binding(auto) var normalTextureSampler: sampler;
+
+fn contactShadow_reconstructViewPosition(uv: vec2f, depth: f32) -> vec3f {
+  let clip = vec4f(uv.x * 2.0 - 1.0, 1.0 - uv.y * 2.0, depth, 1.0);
+  let view = contactShadowTrace.inverseProjectionMatrix * clip;
+  return view.xyz / max(abs(view.w), 0.00001);
+}
+
+fn contactShadow_projectViewPosition(position: vec3f) -> vec2f {
+  let clip = contactShadowTrace.projectionMatrix * vec4f(position, 1.0);
+  let ndc = clip.xy / max(abs(clip.w), 0.00001);
+  return vec2f(ndc.x * 0.5 + 0.5, 0.5 - ndc.y * 0.5);
+}
+
+fn contactShadowTrace_sampleColor(
+  sourceTexture: texture_2d<f32>, sourceTextureSampler: sampler, texSize: vec2f, texCoord: vec2f
+) -> vec4f {
+  let centerDepth = textureSampleLevel(depthTexture, depthTextureSampler, texCoord, 0);
+  if (centerDepth >= 0.99999) { return vec4f(1.0); }
+  let centerNormal = normalize(textureSampleLevel(normalTexture, normalTextureSampler, texCoord, 0.0).xyz * 2.0 - 1.0);
+  let centerPosition = contactShadow_reconstructViewPosition(texCoord, centerDepth) + centerNormal * contactShadowTrace.normalBias;
+  let rayDirection = normalize(contactShadowTrace.lightDirectionView);
+  let noise = fract(sin(dot(texCoord * texSize + contactShadowTrace.frameIndex, vec2f(12.9898, 78.233))) * 43758.5453);
+  for (var index: i32 = 0; index < 40; index++) {
+    if (f32(index) >= contactShadowTrace.stepCount) { break; }
+    let fraction = (f32(index) + 0.35 + noise * 0.5) / max(contactShadowTrace.stepCount, 1.0);
+    let travel = contactShadowTrace.maxDistance * fraction * fraction;
+    let rayPosition = centerPosition + rayDirection * travel;
+    let sampleUv = contactShadow_projectViewPosition(rayPosition);
+    if (any(sampleUv <= vec2f(0.0)) || any(sampleUv >= vec2f(1.0))) { break; }
+    let sampleDepth = textureSampleLevel(depthTexture, depthTextureSampler, sampleUv, 0);
+    if (sampleDepth >= 0.99999) { continue; }
+    let scenePosition = contactShadow_reconstructViewPosition(sampleUv, sampleDepth);
+    let depthDelta = (-rayPosition.z) - (-scenePosition.z);
+    let hitThickness = contactShadowTrace.thickness + travel * 0.02;
+    if (depthDelta >= 0.0 && depthDelta <= hitThickness) {
+      let distanceFade = 1.0 - clamp(travel / contactShadowTrace.maxDistance, 0.0, 1.0);
+      return vec4f(vec3f(1.0 - distanceFade), 1.0);
+    }
+    if (depthDelta > hitThickness * 8.0) { break; }
+  }
+  return vec4f(1.0);
+}`,
+  fs: /* glsl */ `
+layout(std140) uniform contactShadowTraceUniforms {
+  mat4 projectionMatrix;
+  mat4 inverseProjectionMatrix;
+  vec3 lightDirectionView;
+  float maxDistance;
+  float thickness;
+  float stepCount;
+  float frameIndex;
+  float normalBias;
+} contactShadowTrace;
+uniform sampler2D depthTexture;
+uniform sampler2D normalTexture;
+vec3 contactShadow_reconstructViewPosition(vec2 uv, float depth) {
+  vec4 view = contactShadowTrace.inverseProjectionMatrix * vec4(uv * 2.0 - 1.0, depth * 2.0 - 1.0, 1.0);
+  return view.xyz / max(abs(view.w), 0.00001);
+}
+vec2 contactShadow_projectViewPosition(vec3 position) {
+  vec4 clip = contactShadowTrace.projectionMatrix * vec4(position,1.0);
+  return clip.xy / max(abs(clip.w),0.00001) * 0.5 + 0.5;
+}
+vec4 contactShadowTrace_sampleColor(sampler2D sourceTexture,vec2 texSize,vec2 texCoord) {
+  float centerDepth = texture(depthTexture,texCoord).r;
+  if(centerDepth >= 0.99999) return vec4(1.0);
+  vec3 centerNormal = normalize(texture(normalTexture,texCoord).xyz * 2.0 - 1.0);
+  vec3 centerPosition = contactShadow_reconstructViewPosition(texCoord,centerDepth) + centerNormal * contactShadowTrace.normalBias;
+  vec3 rayDirection = normalize(contactShadowTrace.lightDirectionView);
+  float noise = fract(sin(dot(texCoord * texSize + contactShadowTrace.frameIndex,vec2(12.9898,78.233))) * 43758.5453);
+  for(int index=0;index<40;index++) {
+    if(float(index)>=contactShadowTrace.stepCount) break;
+    float fraction=(float(index)+0.35+noise*0.5)/max(contactShadowTrace.stepCount,1.0);
+    float travel=contactShadowTrace.maxDistance*fraction*fraction;
+    vec3 rayPosition=centerPosition+rayDirection*travel;
+    vec2 sampleUv=contactShadow_projectViewPosition(rayPosition);
+    if(any(lessThanEqual(sampleUv,vec2(0.0)))||any(greaterThanEqual(sampleUv,vec2(1.0))))break;
+    float sampleDepth=texture(depthTexture,sampleUv).r;
+    if(sampleDepth>=0.99999)continue;
+    vec3 scenePosition=contactShadow_reconstructViewPosition(sampleUv,sampleDepth);
+    float depthDelta=(-rayPosition.z)-(-scenePosition.z);
+    float hitThickness=contactShadowTrace.thickness+travel*0.02;
+    if(depthDelta>=0.0&&depthDelta<=hitThickness) {
+      float distanceFade=1.0-clamp(travel/contactShadowTrace.maxDistance,0.0,1.0);
+      return vec4(vec3(1.0-distanceFade),1.0);
+    }
+    if(depthDelta>hitThickness*8.0)break;
+  }
+  return vec4(1.0);
+}
+`,
+  bindingLayout: [
+    {name: 'depthTexture', group: 0},
+    {name: 'normalTexture', group: 0}
+  ],
+  props: {} as Partial<ContactTraceUniforms> & ContactShadowBindings,
+  uniforms: {} as ContactTraceUniforms,
+  bindings: {} as ContactShadowBindings,
+  uniformTypes: {
+    projectionMatrix: 'mat4x4<f32>',
+    inverseProjectionMatrix: 'mat4x4<f32>',
+    lightDirectionView: 'vec3<f32>',
+    maxDistance: 'f32',
+    thickness: 'f32',
+    stepCount: 'f32',
+    frameIndex: 'f32',
+    normalBias: 'f32'
+  },
+  propTypes: {
+    projectionMatrix: {value: IDENTITY_MATRIX, private: true},
+    inverseProjectionMatrix: {value: IDENTITY_MATRIX, private: true},
+    lightDirectionView: {value: [0, 1, 0], private: true},
+    maxDistance: {value: 4, min: 0.1, softMax: 16},
+    thickness: {value: 0.12, min: 0.005, softMax: 1},
+    stepCount: {value: 24, min: 1, max: 40},
+    frameIndex: {value: 0, private: true},
+    normalBias: {value: 0.035, min: 0, softMax: 0.2}
+  },
+  passes: [{sampler: true}]
+} as const satisfies ShaderPass<
+  Partial<ContactTraceUniforms> & ContactShadowBindings,
+  ContactTraceUniforms,
+  ContactShadowBindings
+>;
+
+export type ContactBlurUniforms = {
+  direction: [number, number];
+  radius: number;
+  depthSigma: number;
+  normalPower: number;
+};
+
+export const contactShadowBilateralBlur = {
+  name: 'contactShadowBilateralBlur',
+  source: /* wgsl */ `\
+struct contactShadowBilateralBlurUniforms {
+  direction: vec2f,
+  radius: f32,
+  depthSigma: f32,
+  normalPower: f32,
+};
+@group(0) @binding(auto) var<uniform> contactShadowBilateralBlur: contactShadowBilateralBlurUniforms;
+@group(0) @binding(auto) var depthTexture: texture_depth_2d;
+@group(0) @binding(auto) var depthTextureSampler: sampler;
+@group(0) @binding(auto) var normalTexture: texture_2d<f32>;
+@group(0) @binding(auto) var normalTextureSampler: sampler;
+
+fn contactShadowBilateralBlur_sampleColor(
+  sourceTexture: texture_2d<f32>, sourceTextureSampler: sampler, texSize: vec2f, texCoord: vec2f
+) -> vec4f {
+  let sourceDimensions = vec2f(textureDimensions(sourceTexture));
+  let texel = contactShadowBilateralBlur.direction / sourceDimensions;
+  let centerDepth = textureSampleLevel(depthTexture, depthTextureSampler, texCoord, 0);
+  let centerNormal = normalize(textureSampleLevel(normalTexture, normalTextureSampler, texCoord, 0.0).xyz * 2.0 - 1.0);
+  var result = textureSampleLevel(sourceTexture, sourceTextureSampler, texCoord, 0.0).r;
+  var totalWeight = 1.0;
+  for (var index: i32 = 1; index <= 6; index++) {
+    if (f32(index) > contactShadowBilateralBlur.radius) { break; }
+    for (var side: i32 = -1; side <= 1; side += 2) {
+      let sampleUv = clamp(texCoord + texel * f32(index * side), vec2f(0.0), vec2f(1.0));
+      let sampleDepth = textureSampleLevel(depthTexture, depthTextureSampler, sampleUv, 0);
+      let sampleNormal = normalize(textureSampleLevel(normalTexture, normalTextureSampler, sampleUv, 0.0).xyz * 2.0 - 1.0);
+      let depthDelta = abs(sampleDepth - centerDepth);
+      let depthWeight = exp(-(depthDelta * depthDelta) / max(2.0 * contactShadowBilateralBlur.depthSigma * contactShadowBilateralBlur.depthSigma, 0.000001));
+      let normalWeight = pow(max(dot(centerNormal, sampleNormal), 0.0), contactShadowBilateralBlur.normalPower);
+      let spatialWeight = exp(-f32(index * index) / 8.0);
+      let weight = depthWeight * normalWeight * spatialWeight;
+      result += textureSampleLevel(sourceTexture, sourceTextureSampler, sampleUv, 0.0).r * weight;
+      totalWeight += weight;
+    }
+  }
+  let visibility = result / totalWeight;
+  return vec4f(vec3f(visibility), 1.0);
+}`,
+  fs: /* glsl */ `
+layout(std140) uniform contactShadowBilateralBlurUniforms {
+  vec2 direction;
+  float radius;
+  float depthSigma;
+  float normalPower;
+} contactShadowBilateralBlur;
+uniform sampler2D depthTexture;
+uniform sampler2D normalTexture;
+vec4 contactShadowBilateralBlur_sampleColor(sampler2D sourceTexture,vec2 texSize,vec2 texCoord) {
+  vec2 texel=contactShadowBilateralBlur.direction/vec2(textureSize(sourceTexture,0));
+  float centerDepth=texture(depthTexture,texCoord).r;
+  vec3 centerNormal=normalize(texture(normalTexture,texCoord).xyz*2.0-1.0);
+  float result=texture(sourceTexture,texCoord).r;
+  float totalWeight=1.0;
+  for(int index=1;index<=6;index++) {
+    if(float(index)>contactShadowBilateralBlur.radius)break;
+    for(int side=-1;side<=1;side+=2) {
+      vec2 sampleUv=clamp(texCoord+texel*float(index*side),vec2(0.0),vec2(1.0));
+      float sampleDepth=texture(depthTexture,sampleUv).r;
+      vec3 sampleNormal=normalize(texture(normalTexture,sampleUv).xyz*2.0-1.0);
+      float depthDelta=abs(sampleDepth-centerDepth);
+      float depthWeight=exp(-(depthDelta*depthDelta)/max(2.0*contactShadowBilateralBlur.depthSigma*contactShadowBilateralBlur.depthSigma,0.000001));
+      float normalWeight=pow(max(dot(centerNormal,sampleNormal),0.0),contactShadowBilateralBlur.normalPower);
+      float spatialWeight=exp(-float(index*index)/8.0);
+      float weight=depthWeight*normalWeight*spatialWeight;
+      result+=texture(sourceTexture,sampleUv).r*weight;totalWeight+=weight;
+    }
+  }
+  return vec4(vec3(result/totalWeight),1.0);
+}
+`,
+  bindingLayout: [
+    {name: 'depthTexture', group: 0},
+    {name: 'normalTexture', group: 0}
+  ],
+  props: {} as Partial<ContactBlurUniforms> & ContactShadowBindings,
+  uniforms: {} as ContactBlurUniforms,
+  bindings: {} as ContactShadowBindings,
+  uniformTypes: {
+    direction: 'vec2<f32>',
+    radius: 'f32',
+    depthSigma: 'f32',
+    normalPower: 'f32'
+  },
+  propTypes: {
+    direction: {value: [1, 0], private: true},
+    radius: {value: 4, min: 1, max: 6},
+    depthSigma: {value: 0.006, min: 0.0001, softMax: 0.05},
+    normalPower: {value: 24, min: 1, softMax: 64}
+  },
+  passes: [{sampler: true}]
+} as const satisfies ShaderPass<
+  Partial<ContactBlurUniforms> & ContactShadowBindings,
+  ContactBlurUniforms,
+  ContactShadowBindings
+>;
+
+export type ContactCompositeBindings<TextureT = unknown> = {
+  contactShadowTexture?: TextureT;
+  directionalDirectTexture?: TextureT;
+  shadowDebugTexture?: TextureT;
+};
+
+type ContactCompositeUniforms = {strength: number; debugMode: number};
+
+export const contactShadowComposite = {
+  name: 'contactShadowComposite',
+  source: /* wgsl */ `\
+struct contactShadowCompositeUniforms {
+  strength: f32,
+  debugMode: f32,
+};
+@group(0) @binding(auto) var<uniform> contactShadowComposite: contactShadowCompositeUniforms;
+@group(0) @binding(auto) var contactShadowTexture: texture_2d<f32>;
+@group(0) @binding(auto) var contactShadowTextureSampler: sampler;
+@group(0) @binding(auto) var directionalDirectTexture: texture_2d<f32>;
+@group(0) @binding(auto) var directionalDirectTextureSampler: sampler;
+@group(0) @binding(auto) var shadowDebugTexture: texture_2d<f32>;
+@group(0) @binding(auto) var shadowDebugTextureSampler: sampler;
+
+fn contactShadowComposite_sampleColor(
+  sourceTexture: texture_2d<f32>, sourceTextureSampler: sampler, texSize: vec2f, texCoord: vec2f
+) -> vec4f {
+  let color = textureSampleLevel(sourceTexture, sourceTextureSampler, texCoord, 0.0);
+  let contact = textureSampleLevel(contactShadowTexture, contactShadowTextureSampler, texCoord, 0.0).r;
+  let appliedContact = mix(1.0, contact, contactShadowComposite.strength);
+  if (contactShadowComposite.debugMode > 0.5 && contactShadowComposite.debugMode < 1.5) {
+    return vec4f(vec3f(contact), 1.0);
+  }
+  if (contactShadowComposite.debugMode >= 1.5) {
+    let factors = textureSampleLevel(shadowDebugTexture, shadowDebugTextureSampler, texCoord, 0.0).rgb;
+    return vec4f(vec3f(factors.x * factors.y * factors.z * contact), 1.0);
+  }
+  let directionalDirect = textureSampleLevel(
+    directionalDirectTexture,
+    directionalDirectTextureSampler,
+    texCoord,
+    0.0
+  ).rgb;
+  return vec4f(max(color.rgb - directionalDirect * (1.0 - appliedContact), vec3f(0.0)), color.a);
+}`,
+  fs: /* glsl */ `
+layout(std140) uniform contactShadowCompositeUniforms {
+  float strength;
+  float debugMode;
+} contactShadowComposite;
+uniform sampler2D contactShadowTexture;
+uniform sampler2D directionalDirectTexture;
+uniform sampler2D shadowDebugTexture;
+vec4 contactShadowComposite_sampleColor(sampler2D sourceTexture,vec2 texSize,vec2 texCoord) {
+  vec4 color=texture(sourceTexture,texCoord);
+  float contact=texture(contactShadowTexture,texCoord).r;
+  float appliedContact=mix(1.0,contact,contactShadowComposite.strength);
+  if(contactShadowComposite.debugMode>0.5&&contactShadowComposite.debugMode<1.5)return vec4(vec3(contact),1.0);
+  if(contactShadowComposite.debugMode>=1.5){vec3 factors=texture(shadowDebugTexture,texCoord).rgb;return vec4(vec3(factors.x*factors.y*factors.z*contact),1.0);}
+  vec3 directionalDirect=texture(directionalDirectTexture,texCoord).rgb;
+  return vec4(max(color.rgb-directionalDirect*(1.0-appliedContact),vec3(0.0)),color.a);
+}
+`,
+  bindingLayout: [
+    {name: 'contactShadowTexture', group: 0},
+    {name: 'directionalDirectTexture', group: 0},
+    {name: 'shadowDebugTexture', group: 0}
+  ],
+  props: {} as Partial<ContactCompositeUniforms> & ContactCompositeBindings,
+  uniforms: {} as ContactCompositeUniforms,
+  bindings: {} as ContactCompositeBindings,
+  uniformTypes: {strength: 'f32', debugMode: 'f32'},
+  propTypes: {
+    strength: {value: 1, min: 0, max: 1},
+    debugMode: {value: 0, min: 0, max: 2, private: true}
+  },
+  passes: [{sampler: true}]
+} as const satisfies ShaderPass<
+  Partial<ContactCompositeUniforms> & ContactCompositeBindings,
+  ContactCompositeUniforms,
+  ContactCompositeBindings,
+  'contactShadowFiltered'
+>;

package/src/modules/lighting/lightmap/lightmap-filter.ts

@@ -0,0 +1,73 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: Copyright (c) vis.gl contributors
+
+import type {ShaderModule} from '../../../lib/shader-module/shader-module';
+
+/** Internal accumulation/copy/dilation shader shared by the engine baker. */
+export const lightmapFilter = {
+  name: 'lightmapFilter',
+  bindingLayout: [
+    {name: 'lightmapFilter', group: 0},
+    {name: 'currentTexture', group: 0},
+    {name: 'previousTexture', group: 0},
+    {name: 'chartTexture', group: 0}
+  ],
+  uniformTypes: {weight: 'f32', mode: 'i32'},
+  defaultUniforms: {weight: 1, mode: 0},
+  getUniforms: (props: {weight?: number; mode?: number}) => props,
+  source: `
+struct LightmapFilterUniforms {weight: f32, mode: i32};
+@group(0) @binding(auto) var<uniform> lightmapFilter: LightmapFilterUniforms;
+@group(0) @binding(auto) var currentTexture: texture_2d<f32>;
+@group(0) @binding(auto) var previousTexture: texture_2d<f32>;
+@group(0) @binding(auto) var chartTexture: texture_2d<f32>;
+fn lightmapFilter_getColor(coordinates: vec2i) -> vec4f {
+  let current = textureLoad(currentTexture, coordinates, 0);
+  if (lightmapFilter.mode == 0) {
+    let previous = textureLoad(previousTexture, coordinates, 0);
+    return vec4f(mix(previous.rgb, current.rgb, lightmapFilter.weight), current.a);
+  }
+  if (lightmapFilter.mode == 2 || current.a > 0.0) {return current;}
+  let chart = textureLoad(chartTexture, coordinates, 0).r;
+  if (chart <= 0.0) {return current;}
+  let size = vec2i(textureDimensions(currentTexture));
+  for (var y = -1; y <= 1; y++) {
+    for (var x = -1; x <= 1; x++) {
+      let neighbor = coordinates + vec2i(x, y);
+      if (all(neighbor >= vec2i(0)) && all(neighbor < size)) {
+        let candidate = textureLoad(currentTexture, neighbor, 0);
+        if (candidate.a == chart) {return candidate;}
+      }
+    }
+  }
+  return current;
+}`,
+  fs: `
+layout(std140) uniform lightmapFilterUniforms {
+ float weight;
+ int mode;
+} lightmapFilter;
+uniform sampler2D currentTexture;
+uniform sampler2D previousTexture;
+uniform sampler2D chartTexture;
+vec4 lightmapFilter_getColor(ivec2 coordinates) {
+  vec4 current = texelFetch(currentTexture, coordinates, 0);
+  if (lightmapFilter.mode == 0) {
+    return vec4(mix(texelFetch(previousTexture, coordinates, 0).rgb, current.rgb, lightmapFilter.weight), current.a);
+  }
+  if (lightmapFilter.mode == 2 || current.a > 0.0) {return current;}
+  float chart = texelFetch(chartTexture, coordinates, 0).r;
+  ivec2 size = textureSize(currentTexture, 0);
+  if (chart <= 0.0) return current;
+  for (int y = -1; y <= 1; y++) {
+    for (int x = -1; x <= 1; x++) {
+      ivec2 neighbor = coordinates + ivec2(x,y);
+      if (all(greaterThanEqual(neighbor, ivec2(0))) && all(lessThan(neighbor,size))) {
+        vec4 candidate = texelFetch(currentTexture, neighbor, 0);
+        if (candidate.a == chart) {return candidate;}
+      }
+    }
+  }
+  return current;
+}`
+} as const satisfies ShaderModule<{weight?: number; mode?: number}, {weight: number; mode: number}>;

package/src/modules/lighting/lightmap/lightmap.ts

@@ -0,0 +1,76 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: Copyright (c) vis.gl contributors
+
+import type {NumberArray9} from '@math.gl/core';
+import type {ShaderModule} from '../../../lib/shader-module/shader-module';
+import {pbrMaterial} from '../pbr-material/pbr-material';
+
+export type LightmapBakeUniforms = {
+  uvSet: number;
+  uvTransform: Readonly<NumberArray9>;
+  chartId: number;
+};
+
+/** Texture-space rasterization controls. Chart IDs are positive integers exactly representable in the target format. */
+export const lightmapBake = {
+  name: 'lightmapBake',
+  dependencies: [pbrMaterial],
+  bindingLayout: [{name: 'lightmapBake', group: 0}],
+  uniformTypes: {uvSet: 'i32', uvTransform: 'mat3x3<f32>', chartId: 'f32'},
+  defaultUniforms: {uvSet: 1, uvTransform: [1, 0, 0, 0, 1, 0, 0, 0, 1], chartId: 1},
+  getUniforms: (props: Partial<LightmapBakeUniforms>) => props,
+  source: `
+struct LightmapBakeUniforms {uvSet: i32, uvTransform: mat3x3f, chartId: f32};
+@group(0) @binding(auto) var<uniform> lightmapBake: LightmapBakeUniforms;
+
+fn lightmap_getIrradiance(position: vec3f, normal: vec3f, directionalVisibility: f32, pointVisibility: vec4f, spotVisibility: vec4f) -> vec3f {
+  if (lighting.enabled == 0) {return vec3f(0.0);}
+  var irradiance = lighting.ambientColor;
+  for (var index = 0; index < lighting.directionalLightCount; index++) {
+    let light = lighting_getDirectionalLight(index);
+    irradiance += light.color * max(dot(normal, -normalize(light.direction)), 0.0) * select(1.0, directionalVisibility, index == 0);
+  }
+  for (var index = 0; index < lighting.pointLightCount; index++) {
+    let light = lighting_getPointLight(index);
+    let offset = light.position - position;
+    let visibility = select(1.0, pointVisibility[min(index, 3)], index < 4);
+    irradiance += light.color * max(dot(normal, normalize(offset)), 0.0) * visibility / max(getPointLightAttenuation(light, length(offset)), 0.0001);
+  }
+  for (var index = 0; index < lighting.spotLightCount; index++) {
+    let light = lighting_getSpotLight(index);
+    irradiance += light.color * max(dot(normal, normalize(light.position - position)), 0.0) * select(1.0, spotVisibility[min(index, 3)], index < 4) / max(getSpotLightAttenuation(light, position), 0.0001);
+  }
+  return irradiance;
+}
+`,
+  vs: `layout(std140) uniform lightmapBakeUniforms {
+ int uvSet;
+ mat3 uvTransform;
+ float chartId;
+} lightmapBake;`,
+  fs: `layout(std140) uniform lightmapBakeUniforms {
+ int uvSet;
+ mat3 uvTransform;
+ float chartId;
+} lightmapBake;
+
+vec3 lightmap_getIrradiance(vec3 position, vec3 normal, float directionalVisibility, vec4 pointVisibility, vec4 spotVisibility) {
+  if (lighting.enabled == 0) {return vec3(0.0);}
+  vec3 irradiance = lighting.ambientColor;
+  for (int index = 0; index < lighting.directionalLightCount; index++) {
+    DirectionalLight light = lighting_getDirectionalLight(index);
+    irradiance += light.color * max(dot(normal, -normalize(light.direction)), 0.0) * (index == 0 ? directionalVisibility : 1.0);
+  }
+  for (int index = 0; index < lighting.pointLightCount; index++) {
+    PointLight light = lighting_getPointLight(index);
+    vec3 offset = light.position - position;
+    irradiance += light.color * max(dot(normal, normalize(offset)), 0.0) * (index < 4 ? pointVisibility[index] : 1.0) / max(getPointLightAttenuation(light, length(offset)), 0.0001);
+  }
+  for (int index = 0; index < lighting.spotLightCount; index++) {
+    SpotLight light = lighting_getSpotLight(index);
+    irradiance += light.color * max(dot(normal, normalize(light.position - position)), 0.0) * (index < 4 ? spotVisibility[index] : 1.0) / max(getSpotLightAttenuation(light, position), 0.0001);
+  }
+  return irradiance;
+}
+`
+} as const satisfies ShaderModule<Partial<LightmapBakeUniforms>, LightmapBakeUniforms>;

package/src/modules/lighting/pbr-ambient-occlusion/pbr-ambient-occlusion.ts

@@ -0,0 +1,78 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: Copyright (c) vis.gl contributors
+
+import type {NumberArray4} from '@math.gl/core';
+import type {ShaderModule} from '../../../lib/shader-module/shader-module';
+import {pbrMaterial} from '../pbr-material/pbr-material';
+
+export type PBRAmbientOcclusionProps<TextureT = unknown> = {
+  ambientOcclusionTexture?: TextureT;
+  /** Origin and dimensions of the receiver viewport, in physical pixels. */
+  viewport?: Readonly<NumberArray4>;
+  strength?: number;
+  enabled?: boolean;
+};
+type AmbientUniforms = {viewport: Readonly<NumberArray4>; strength: number; enabled: number};
+type AmbientBindings<TextureT = unknown> = {ambientOcclusionTexture?: TextureT};
+
+/** Sets ambient visibility for the canonical PBR material; direct lighting and emission remain independent. */
+export const pbrAmbientOcclusion: ShaderModule<
+  PBRAmbientOcclusionProps,
+  AmbientUniforms,
+  AmbientBindings
+> = {
+  name: 'pbrAmbientOcclusion',
+  dependencies: [pbrMaterial],
+  bindingLayout: [
+    {name: 'pbrAmbientOcclusion', group: 0},
+    {name: 'ambientOcclusionTexture', group: 2}
+  ],
+  uniformTypes: {viewport: 'vec4<f32>', strength: 'f32', enabled: 'i32'},
+  defaultUniforms: {viewport: [0, 0, 1, 1], strength: 1, enabled: 1},
+  getUniforms(properties = {}) {
+    return {
+      ...(properties.viewport ? {viewport: properties.viewport} : {}),
+      ...(properties.strength === undefined ? {} : {strength: properties.strength}),
+      ...(properties.enabled === undefined ? {} : {enabled: Number(properties.enabled)}),
+      ...(properties.ambientOcclusionTexture
+        ? {ambientOcclusionTexture: properties.ambientOcclusionTexture}
+        : {})
+    };
+  },
+  source: /* wgsl */ `
+struct PBRAmbientOcclusionUniforms {
+  viewport: vec4f,
+  strength: f32,
+  enabled: i32,
+};
+@group(0) @binding(auto) var<uniform> pbrAmbientOcclusion: PBRAmbientOcclusionUniforms;
+@group(2) @binding(auto) var ambientOcclusionTexture: texture_2d<f32>;
+@group(2) @binding(auto) var ambientOcclusionTextureSampler: sampler;
+fn pbrAmbientOcclusion_getFactor(pixelPosition:vec2f)->f32 {
+  if(pbrAmbientOcclusion.enabled==0){return 1.0;}
+  let coordinates=(pixelPosition-pbrAmbientOcclusion.viewport.xy)/pbrAmbientOcclusion.viewport.zw;
+  if(any(coordinates<vec2f(0.0))||any(coordinates>vec2f(1.0))){return 1.0;}
+  let visibility=textureSampleLevel(ambientOcclusionTexture,ambientOcclusionTextureSampler,coordinates,0.0).r;
+  return mix(1.0,clamp(visibility,0.0,1.0),clamp(pbrAmbientOcclusion.strength,0.0,1.0));
+}
+fn pbrAmbientOcclusion_setFactor(pixelPosition:vec2f) {
+  pbr_ambientVisibility = pbrAmbientOcclusion_getFactor(pixelPosition);
+}`,
+  fs: /* glsl */ `
+layout(std140) uniform pbrAmbientOcclusionUniforms {
+  vec4 viewport;
+  float strength;
+  int enabled;
+} pbrAmbientOcclusion;
+uniform sampler2D ambientOcclusionTexture;
+float pbrAmbientOcclusion_getFactor(vec2 pixelPosition) {
+  if(pbrAmbientOcclusion.enabled==0)return 1.0;
+  vec2 coordinates=(pixelPosition-pbrAmbientOcclusion.viewport.xy)/pbrAmbientOcclusion.viewport.zw;
+  if(any(lessThan(coordinates,vec2(0.0)))||any(greaterThan(coordinates,vec2(1.0))))return 1.0;
+  float visibility=texture(ambientOcclusionTexture,coordinates).r;
+  return mix(1.0,clamp(visibility,0.0,1.0),clamp(pbrAmbientOcclusion.strength,0.0,1.0));
+}
+void pbrAmbientOcclusion_setFactor(vec2 pixelPosition) {
+  pbr_ambientVisibility = pbrAmbientOcclusion_getFactor(pixelPosition);
+}`
+};

package/src/modules/lighting/pbr-contact-shadow/pbr-contact-shadow.ts

@@ -0,0 +1,78 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: Copyright (c) vis.gl contributors
+
+import type {NumberArray4} from '@math.gl/core';
+import type {ShaderModule} from '../../../lib/shader-module/shader-module';
+import {pbrMaterial} from '../pbr-material/pbr-material';
+
+export type PBRContactShadowProps<TextureT = unknown> = {
+  contactShadowTexture?: TextureT;
+  /** Origin and dimensions of the receiver viewport, in physical pixels. */
+  viewport?: Readonly<NumberArray4>;
+  strength?: number;
+  enabled?: boolean;
+};
+type ContactUniforms = {viewport: Readonly<NumberArray4>; strength: number; enabled: number};
+type ContactBindings<TextureT = unknown> = {contactShadowTexture?: TextureT};
+
+/** Applies a screen-space visibility mask to the first directional light only. */
+export const pbrContactShadow: ShaderModule<
+  PBRContactShadowProps,
+  ContactUniforms,
+  ContactBindings
+> = {
+  name: 'pbrContactShadow',
+  dependencies: [pbrMaterial],
+  bindingLayout: [
+    {name: 'pbrContactShadow', group: 0},
+    {name: 'contactShadowTexture', group: 2}
+  ],
+  uniformTypes: {viewport: 'vec4<f32>', strength: 'f32', enabled: 'i32'},
+  defaultUniforms: {viewport: [0, 0, 1, 1], strength: 1, enabled: 1},
+  getUniforms(properties = {}) {
+    return {
+      ...(properties.viewport ? {viewport: properties.viewport} : {}),
+      ...(properties.strength === undefined ? {} : {strength: properties.strength}),
+      ...(properties.enabled === undefined ? {} : {enabled: Number(properties.enabled)}),
+      ...(properties.contactShadowTexture
+        ? {contactShadowTexture: properties.contactShadowTexture}
+        : {})
+    };
+  },
+  source: /* wgsl */ `
+struct PBRContactShadowUniforms {
+  viewport: vec4f,
+  strength: f32,
+  enabled: i32,
+};
+@group(0) @binding(auto) var<uniform> pbrContactShadow: PBRContactShadowUniforms;
+@group(2) @binding(auto) var contactShadowTexture: texture_2d<f32>;
+@group(2) @binding(auto) var contactShadowTextureSampler: sampler;
+fn pbrContactShadow_getFactor(pixelPosition:vec2f)->f32 {
+  if(pbrContactShadow.enabled==0){return 1.0;}
+  let coordinates=(pixelPosition-pbrContactShadow.viewport.xy)/pbrContactShadow.viewport.zw;
+  if(any(coordinates<vec2f(0.0))||any(coordinates>vec2f(1.0))){return 1.0;}
+  let visibility=textureSampleLevel(contactShadowTexture,contactShadowTextureSampler,coordinates,0.0).r;
+  return mix(1.0,clamp(visibility,0.0,1.0),clamp(pbrContactShadow.strength,0.0,1.0));
+}
+fn pbrContactShadow_filterColor(vertexColor:vec4f,pixelPosition:vec2f)->vec4f {
+  return pbr_filterColorWithShadows(vertexColor,pbrContactShadow_getFactor(pixelPosition),vec4f(1.0),vec4f(1.0));
+}`,
+  fs: /* glsl */ `
+layout(std140) uniform pbrContactShadowUniforms {
+  vec4 viewport;
+  float strength;
+  int enabled;
+} pbrContactShadow;
+uniform sampler2D contactShadowTexture;
+float pbrContactShadow_getFactor(vec2 pixelPosition) {
+  if(pbrContactShadow.enabled==0)return 1.0;
+  vec2 coordinates=(pixelPosition-pbrContactShadow.viewport.xy)/pbrContactShadow.viewport.zw;
+  if(any(lessThan(coordinates,vec2(0.0)))||any(greaterThan(coordinates,vec2(1.0))))return 1.0;
+  float visibility=texture(contactShadowTexture,coordinates).r;
+  return mix(1.0,clamp(visibility,0.0,1.0),clamp(pbrContactShadow.strength,0.0,1.0));
+}
+vec4 pbrContactShadow_filterColor(vec4 vertexColor,vec2 pixelPosition) {
+  return pbr_filterColorWithShadows(vertexColor,pbrContactShadow_getFactor(pixelPosition),vec4(1.0),vec4(1.0));
+}`
+};