@meimu/mutracer 0.2.0 → 0.3.0

package/lib/renderer/wgsl/chunks/sampleLight.wgsl.js.map

@@ -1 +1 @@
-{"version":3,"file":"sampleLight.wgsl.js","sourceRoot":"","sources":["../../../../src/renderer/wgsl/chunks/sampleLight.wgsl.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,8BAA8B,EAC9B,wBAAwB,EACzB,MAAM,yBAAyB,CAAC;AAEjC,eAAe,SAAS,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCAkVkB,wBAAwB;;;;;;;4BAOjC,qBAAqB;;;;;;;;wDAQO,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgD3F,CAAC","sourcesContent":["import {\n  LIGHT_CHOICE_ANALYTIC,\n  LIGHT_CHOICE_EMISSIVE_TRIANGLE,\n  LIGHT_CHOICE_ENVIRONMENT\n} from '../../lightConstants.js';\n\nexport default /*wgsl */ `\nfn light_sample_sphere(light: Light, scatterPos: vec3f, randomV2: vec2f, lightSample: ptr<function, LightSampleRec>) {\n  let centerOffset = light.position - scatterPos;\n  let centerDistance = length(centerOffset);\n  if (light.radiusOrSpread <= 0.0) {\n    // A zero-radius sphere is a point emitter. The stored emission is\n    // luminous intensity, so convert it to incident radiance here rather\n    // than approximating the delta source as a tiny area sphere.\n    if (centerDistance <= 1e-6) {\n      (*lightSample).direction = vec3f(0.0, 0.0, 1.0);\n      (*lightSample).normal = vec3f(0.0, 0.0, -1.0);\n      (*lightSample).dist = 0.0;\n      (*lightSample).pdf = 1.0;\n      (*lightSample).emission = vec3f(0.0);\n      (*lightSample).isDelta = true;\n      return;\n    }\n    (*lightSample).direction = centerOffset / centerDistance;\n    (*lightSample).normal = -(*lightSample).direction;\n    (*lightSample).dist = centerDistance;\n    (*lightSample).pdf = 1.0;\n    (*lightSample).emission = light.emission /\n      max(centerDistance * centerDistance, 1e-8);\n    (*lightSample).isDelta = true;\n    return;\n  }\n  let radius = max(light.radiusOrSpread, 1e-6);\n  (*lightSample).isDelta = false;\n  var sampledDirection: vec3f;\n\n  if (centerDistance <= radius) {\n    let z = 1.0 - 2.0 * randomV2.x;\n    let radial = sqrt(max(0.0, 1.0 - z * z));\n    let azimuth = TWOPI * randomV2.y;\n    sampledDirection = vec3f(\n      radial * cos(azimuth),\n      radial * sin(azimuth),\n      z\n    );\n    (*lightSample).pdf = INV4PI;\n  }\n  else {\n    let sphereAxis = centerOffset / centerDistance;\n    let sinThetaMaxSquared = min(\n      1.0,\n      radius * radius / (centerDistance * centerDistance)\n    );\n    let cosThetaMax = sqrt(max(0.0, 1.0 - sinThetaMaxSquared));\n    let cosTheta = mix(1.0, cosThetaMax, randomV2.x);\n    let sinTheta = sqrt(max(0.0, 1.0 - cosTheta * cosTheta));\n    let azimuth = TWOPI * randomV2.y;\n    var tangent: vec3f;\n    var bitangent: vec3f;\n    sampling_orthonormal_basis(sphereAxis, &tangent, &bitangent);\n    sampledDirection = sphereAxis * cosTheta\n      + tangent * (sinTheta * cos(azimuth))\n      + bitangent * (sinTheta * sin(azimuth));\n    (*lightSample).pdf = 1.0 / max(\n      TWOPI * (1.0 - cosThetaMax),\n      1e-8\n    );\n  }\n\n  let centerProjection = dot(sampledDirection, centerOffset);\n  let centerOffsetSquared = dot(centerOffset, centerOffset);\n  let discriminant = centerProjection * centerProjection\n    - centerOffsetSquared + radius * radius;\n  let nearDistance = centerProjection - sqrt(max(0.0, discriminant));\n  let farDistance = centerProjection + sqrt(max(0.0, discriminant));\n  (*lightSample).dist = select(farDistance, nearDistance, nearDistance > 1e-6);\n  (*lightSample).direction = sampledDirection;\n  let lightSurfacePos = scatterPos + sampledDirection * (*lightSample).dist;\n  (*lightSample).normal = normalize(lightSurfacePos - light.position);\n  (*lightSample).emission = light.emission;\n}\n\nfn light_apply_rect_texture(\n  light: Light,\n  surfacePoint: vec3f,\n  lowerLeft: vec3f,\n  result: ptr<function, LightSampleRec>\n) {\n  if (light.textureIndex < 0.0) {\n    return;\n  }\n  let width = length(light.u);\n  let height = length(light.v);\n  let relative = surfacePoint - lowerLeft;\n  let uv = vec2f(\n    dot(relative, light.u / width) / width,\n    dot(relative, light.v / height) / height\n  );\n  (*result).emission *= sampleMap(i32(light.textureIndex), uv).rgb;\n}\n\nfn light_sample_rect(\n  light: Light,\n  shadingPoint: vec3f,\n  samplePoint: vec2f,\n  result: ptr<function, LightSampleRec>\n) {\n  let normal = normalize(cross(light.u, light.v));\n  let lowerLeft = light.position - 0.5 * (light.u + light.v);\n  let spread = clamp(light.radiusOrSpread, 0.0, 1.0);\n\n  if (spread <= 0.001) {\n    let planeDistance = dot(normal, shadingPoint - light.position);\n    let surfacePoint = shadingPoint - normal * planeDistance;\n    let relative = surfacePoint - lowerLeft;\n    let width = length(light.u);\n    let height = length(light.v);\n    let uv = vec2f(\n      dot(relative, light.u / width) / width,\n      dot(relative, light.v / height) / height\n    );\n    if (planeDistance <= 0.0 || any(uv < vec2f(0.0)) || any(uv > vec2f(1.0))) {\n      (*result).emission = vec3f(0.0);\n      (*result).pdf = 0.0;\n      return;\n    }\n    (*result).direction = -normal;\n    (*result).dist = planeDistance;\n    (*result).normal = normal;\n    (*result).emission = light.emission;\n    (*result).pdf = 1.0;\n    (*result).isDelta = true;\n    light_apply_rect_texture(light, surfacePoint, lowerLeft, result);\n    return;\n  }\n\n  let surfacePoint = lowerLeft\n    + light.u * samplePoint.x\n    + light.v * samplePoint.y;\n  let offset = surfacePoint - shadingPoint;\n  let distanceSquared = dot(offset, offset);\n  (*result).dist = sqrt(distanceSquared);\n  (*result).direction = offset / (*result).dist;\n  (*result).normal = normal;\n  let emitterCosine = dot(-(*result).direction, normal);\n  let spreadCosine = cos(0.5 * PI * spread);\n  if (emitterCosine <= 0.0 || emitterCosine < spreadCosine) {\n    (*result).emission = vec3f(0.0);\n    (*result).pdf = 0.0;\n    return;\n  }\n  (*result).emission = light.emission;\n  (*result).pdf = distanceSquared / max(light.area * emitterCosine, 1e-8);\n  light_apply_rect_texture(light, surfacePoint, lowerLeft, result);\n}\n\nfn light_sample_directional(\n  light: Light,\n  result: ptr<function, LightSampleRec>\n) {\n  (*result).direction = normalize(-light.u);\n  (*result).normal = -(*result).direction;\n  (*result).emission = light.emission;\n  (*result).dist = INF;\n  (*result).pdf = 1.0;\n  (*result).isDelta = true;\n}\n\nfn light_sample_spot(\n  light: Light,\n  shadingPoint: vec3f,\n  samplePoint: vec2f,\n  result: ptr<function, LightSampleRec>\n) {\n  light_sample_sphere(light, shadingPoint, samplePoint, result);\n  let surfacePoint = shadingPoint + (*result).direction * (*result).dist;\n  let lightToSurface = normalize(shadingPoint - surfacePoint);\n  (*result).emission *= evaluateSpotEmissionMultiplier(light, lightToSurface);\n}\n\nfn sampleAnalyticLight(\n  light: Light,\n  shadingPoint: vec3f,\n  samplePoint: vec2f,\n  result: ptr<function, LightSampleRec>\n) {\n  let lightType = i32(light.litType);\n  if (HAS_SCENE_RECT_LIGHT && lightType == RECT_LIGHT) {\n    light_sample_rect(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (HAS_SCENE_SPHERE_LIGHT && lightType == SPHERE_LIGHT) {\n    light_sample_sphere(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (HAS_SCENE_SPOT_LIGHT && lightType == SPOT_LIGHT) {\n    light_sample_spot(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (HAS_SCENE_DIRECTIONAL_LIGHT && lightType == DIRECTIONAL_LIGHT) {\n    light_sample_directional(light, result);\n  }\n}\n\nfn sampleEnvLight(randomV2: vec2f, lightSample: ptr<function, LightSampleRec>) {\n  var uv: vec2f;\n  // Naga's SPIR-V writer cannot pass a struct-member pointer as a call argument.\n  var pdf: f32;\n  let direction = sampleEnvmap(randomV2, &uv, &pdf);\n  (*lightSample).pdf = pdf;\n  (*lightSample).direction = direction;\n  (*lightSample).emission = textureLinear(envMap, uv).rgb;\n  (*lightSample).dist = INF;\n}\n\nfn loadLightSamplingTexel(texelIndex: i32) -> vec4f {\n  let width = rtUniform.lightSamplingTextureWidth;\n  return textureLoad(\n    lightSamplingData,\n    vec2i(texelIndex % width, texelIndex / width),\n    0\n  );\n}\n\nfn getEmissiveTriangle(index: i32) -> EmissiveTriangle {\n  let offset = index * 6;\n  let a = loadLightSamplingTexel(offset);\n  let b = loadLightSamplingTexel(offset + 1);\n  let c = loadLightSamplingTexel(offset + 2);\n  let d = loadLightSamplingTexel(offset + 3);\n  let e = loadLightSamplingTexel(offset + 4);\n  let f = loadLightSamplingTexel(offset + 5);\n  return EmissiveTriangle(\n    a.xyz, a.w,\n    b.xyz, b.w,\n    c.xyz, c.w,\n    d.xy, d.zw, e.xy,\n    e.zw, f.xy, f.zw\n  );\n}\n\nfn getLightChoice(index: i32) -> LightChoice {\n  let data = loadLightSamplingTexel(rtUniform.lightChoiceTexelOffset + index);\n  return LightChoice(\n    data.x,\n    bitcast<i32>(data.y),\n    bitcast<i32>(data.z),\n    bitcast<i32>(data.w)\n  );\n}\n\nfn getLightChoiceFromTable(tableOffset: i32, index: i32) -> LightChoice {\n  return getLightChoice(tableOffset + index);\n}\n\nfn stratifyLightChoiceRandom(\n  rnd: vec3f,\n  candidateIndex: u32,\n  candidateCount: u32\n) -> vec3f {\n  var stratified = rnd;\n  let count = max(candidateCount, 1u);\n  stratified.z = (rnd.z + f32(min(candidateIndex, count - 1u))) /\n    f32(count);\n  return stratified;\n}\n\nfn sampleEmissiveTriangleLight(\n  triangle: EmissiveTriangle,\n  scatterPos: vec3f,\n  randomV2: vec2f,\n  lightSample: ptr<function, LightSampleRec>\n) {\n  let sqrtU = sqrt(randomV2.x);\n  let barycentric = vec3f(\n    1.0 - sqrtU,\n    sqrtU * (1.0 - randomV2.y),\n    sqrtU * randomV2.y\n  );\n  let lightPosition =\n    barycentric.x * triangle.p0 +\n    barycentric.y * triangle.p1 +\n    barycentric.z * triangle.p2;\n  let uv =\n    barycentric.x * triangle.uv0 +\n    barycentric.y * triangle.uv1 +\n    barycentric.z * triangle.uv2;\n  let secondaryUv =\n    barycentric.x * triangle.secondaryUv0 +\n    barycentric.y * triangle.secondaryUv1 +\n    barycentric.z * triangle.secondaryUv2;\n  let normal = normalize(cross(triangle.p1 - triangle.p0, triangle.p2 - triangle.p0));\n  (*lightSample).direction = lightPosition - scatterPos;\n  (*lightSample).dist = length((*lightSample).direction);\n  if ((*lightSample).dist <= EPS) {\n    (*lightSample).emission = vec3f(0.0);\n    (*lightSample).pdf = 0.0;\n    return;\n  }\n  (*lightSample).direction /= (*lightSample).dist;\n  (*lightSample).normal = normal;\n  let cosLight = dot(normal, -(*lightSample).direction);\n  if (cosLight <= 0.0) {\n    (*lightSample).emission = vec3f(0.0);\n    (*lightSample).pdf = 0.0;\n    return;\n  }\n  (*lightSample).emission = getMaterialEmission(\n    i32(triangle.materialIndex), uv, secondaryUv\n  );\n  (*lightSample).selectionPmf = triangle.selectionPmf;\n  (*lightSample).pdf = triangle.selectionPmf *\n    ((*lightSample).dist * (*lightSample).dist) /\n    max(triangle.area * cosLight, 1e-8);\n}\n\nfn sampleLightChoiceTable(\n  scatterPos: vec3f,\n  rnd: vec3f,\n  tableOffset: i32,\n  choiceCount: i32,\n  selectionPmfScale: f32,\n  lightSample: ptr<function, LightSampleRec>\n) {\n  (*lightSample).emission = vec3f(0.0);\n  (*lightSample).pdf = 0.0;\n  (*lightSample).selectionPmf = 0.0;\n  (*lightSample).dist = INF;\n  (*lightSample).isDelta = false;\n  (*lightSample).sourceType = 0u;\n  (*lightSample).sourceIndex = 0xffffffffu;\n  if (choiceCount <= 0) {\n    return;\n  }\n\n  let scaledChoice = rnd.z * f32(choiceCount);\n  let column = min(i32(scaledChoice), choiceCount - 1);\n  let columnChoice = getLightChoiceFromTable(tableOffset, column);\n  let selectedIndex = select(\n    columnChoice.aliasIndex,\n    column,\n    fract(scaledChoice) < columnChoice.probability\n  );\n  let choice = getLightChoiceFromTable(tableOffset, selectedIndex);\n  if (USE_ENVMAP && choice.sourceType == ${LIGHT_CHOICE_ENVIRONMENT}) {\n    (*lightSample).sourceType = 0u;\n    (*lightSample).sourceIndex = u32(choice.sourceIndex);\n    sampleEnvLight(rnd.xy, lightSample);\n    (*lightSample).selectionPmf = rtUniform.envLightPmf;\n    (*lightSample).pdf *= rtUniform.envLightPmf;\n  }\n  else if (choice.sourceType == ${LIGHT_CHOICE_ANALYTIC}) {\n    (*lightSample).sourceType = 2u;\n    (*lightSample).sourceIndex = u32(choice.sourceIndex);\n    let light = analyticLights[choice.sourceIndex];\n    sampleAnalyticLight(light, scatterPos, rnd.xy, lightSample);\n    (*lightSample).selectionPmf = light.selectionPmf;\n    (*lightSample).pdf *= light.selectionPmf;\n  }\n  else if (HAS_SCENE_EMISSIVE_LIGHT && choice.sourceType == ${LIGHT_CHOICE_EMISSIVE_TRIANGLE}) {\n    (*lightSample).sourceType = 1u;\n    (*lightSample).sourceIndex = u32(choice.sourceIndex);\n    sampleEmissiveTriangleLight(\n      getEmissiveTriangle(choice.sourceIndex),\n      scatterPos,\n      rnd.xy,\n      lightSample\n    );\n  }\n  (*lightSample).selectionPmf *= selectionPmfScale;\n  (*lightSample).pdf *= selectionPmfScale;\n}\n\nfn sampleLight(\n  scatterPos: vec3f,\n  rnd: vec3f,\n  lightSample: ptr<function, LightSampleRec>,\n  distByIntensity: bool\n) {\n  sampleLightChoiceTable(\n    scatterPos,\n    rnd,\n    0,\n    rtUniform.lightChoiceCount,\n    1.0,\n    lightSample\n  );\n}\n\n// Light choice for one of the resampled NEE candidates. The selection random\n// number is stratified across the candidates.\nfn sampleLightForNEECandidate(\n  scatterPos: vec3f,\n  rnd: vec3f,\n  candidateIndex: u32,\n  candidateCount: u32,\n  lightSample: ptr<function, LightSampleRec>\n) {\n  sampleLightChoiceTable(\n    scatterPos,\n    stratifyLightChoiceRandom(rnd, candidateIndex, candidateCount),\n    0,\n    rtUniform.lightChoiceCount,\n    1.0,\n    lightSample\n  );\n}\n`;\n"]}
+{"version":3,"file":"sampleLight.wgsl.js","sourceRoot":"","sources":["../../../../src/renderer/wgsl/chunks/sampleLight.wgsl.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,8BAA8B,EAC9B,wBAAwB,EACzB,MAAM,yBAAyB,CAAC;AAEjC,eAAe,SAAS,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCAsVkB,wBAAwB;;;;;;;gCAOjC,qBAAqB;;;;;;;;4DAQO,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgD3F,CAAC","sourcesContent":["import {\n  LIGHT_CHOICE_ANALYTIC,\n  LIGHT_CHOICE_EMISSIVE_TRIANGLE,\n  LIGHT_CHOICE_ENVIRONMENT\n} from '../../lightConstants.js';\n\nexport default /*wgsl */ `\nfn light_sample_sphere(light: Light, scatterPos: vec3f, randomV2: vec2f, lightSample: ptr<function, LightSampleRec>) {\n  let centerOffset = light.position - scatterPos;\n  let centerDistance = length(centerOffset);\n  if (light.radiusOrSpread <= 0.0) {\n    // A zero-radius sphere is a point emitter. The stored emission is\n    // luminous intensity, so convert it to incident radiance here rather\n    // than approximating the delta source as a tiny area sphere.\n    if (centerDistance <= 1e-6) {\n      (*lightSample).direction = vec3f(0.0, 0.0, 1.0);\n      (*lightSample).normal = vec3f(0.0, 0.0, -1.0);\n      (*lightSample).dist = 0.0;\n      (*lightSample).pdf = 1.0;\n      (*lightSample).emission = vec3f(0.0);\n      (*lightSample).isDelta = true;\n      return;\n    }\n    (*lightSample).direction = centerOffset / centerDistance;\n    (*lightSample).normal = -(*lightSample).direction;\n    (*lightSample).dist = centerDistance;\n    (*lightSample).pdf = 1.0;\n    (*lightSample).emission = light.emission /\n      max(centerDistance * centerDistance, 1e-8);\n    (*lightSample).isDelta = true;\n    return;\n  }\n  let radius = max(light.radiusOrSpread, 1e-6);\n  (*lightSample).isDelta = false;\n  var sampledDirection: vec3f;\n\n  if (centerDistance <= radius) {\n    let z = 1.0 - 2.0 * randomV2.x;\n    let radial = sqrt(max(0.0, 1.0 - z * z));\n    let azimuth = TWOPI * randomV2.y;\n    sampledDirection = vec3f(\n      radial * cos(azimuth),\n      radial * sin(azimuth),\n      z\n    );\n    (*lightSample).pdf = INV4PI;\n  }\n  else {\n    let sphereAxis = centerOffset / centerDistance;\n    let sinThetaMaxSquared = min(\n      1.0,\n      radius * radius / (centerDistance * centerDistance)\n    );\n    let cosThetaMax = sqrt(max(0.0, 1.0 - sinThetaMaxSquared));\n    // Rationalize 1 - cos(thetaMax): subtracting nearly equal f32 values\n    // quantizes small emitters' solid angles into visible irradiance bands.\n    let oneMinusCosThetaMax = sinThetaMaxSquared / (1.0 + cosThetaMax);\n    let oneMinusCosTheta = randomV2.x * oneMinusCosThetaMax;\n    let cosTheta = 1.0 - oneMinusCosTheta;\n    let sinTheta = sqrt(max(0.0, oneMinusCosTheta * (2.0 - oneMinusCosTheta)));\n    let azimuth = TWOPI * randomV2.y;\n    var tangent: vec3f;\n    var bitangent: vec3f;\n    sampling_orthonormal_basis(sphereAxis, &tangent, &bitangent);\n    sampledDirection = sphereAxis * cosTheta\n      + tangent * (sinTheta * cos(azimuth))\n      + bitangent * (sinTheta * sin(azimuth));\n    (*lightSample).pdf = 1.0 / max(\n      TWOPI * oneMinusCosThetaMax,\n      1e-8\n    );\n  }\n\n  let centerProjection = dot(sampledDirection, centerOffset);\n  let centerOffsetSquared = dot(centerOffset, centerOffset);\n  let discriminant = centerProjection * centerProjection\n    - centerOffsetSquared + radius * radius;\n  let nearDistance = centerProjection - sqrt(max(0.0, discriminant));\n  let farDistance = centerProjection + sqrt(max(0.0, discriminant));\n  (*lightSample).dist = select(farDistance, nearDistance, nearDistance > 1e-6);\n  (*lightSample).direction = sampledDirection;\n  let lightSurfacePos = scatterPos + sampledDirection * (*lightSample).dist;\n  (*lightSample).normal = normalize(lightSurfacePos - light.position);\n  (*lightSample).emission = light.emission;\n}\n\nfn light_apply_rect_texture(\n  light: Light,\n  surfacePoint: vec3f,\n  lowerLeft: vec3f,\n  result: ptr<function, LightSampleRec>\n) {\n  if (light.textureIndex < 0.0) {\n    return;\n  }\n  let width = length(light.u);\n  let height = length(light.v);\n  let relative = surfacePoint - lowerLeft;\n  let uv = vec2f(\n    dot(relative, light.u / width) / width,\n    dot(relative, light.v / height) / height\n  );\n  (*result).emission *= sampleMap(i32(light.textureIndex), uv).rgb;\n}\n\nfn light_sample_rect(\n  light: Light,\n  shadingPoint: vec3f,\n  samplePoint: vec2f,\n  result: ptr<function, LightSampleRec>\n) {\n  let normal = normalize(cross(light.u, light.v));\n  let lowerLeft = light.position - 0.5 * (light.u + light.v);\n  let spread = clamp(light.radiusOrSpread, 0.0, 1.0);\n\n  if (spread <= 0.001) {\n    let planeDistance = dot(normal, shadingPoint - light.position);\n    let surfacePoint = shadingPoint - normal * planeDistance;\n    let relative = surfacePoint - lowerLeft;\n    let width = length(light.u);\n    let height = length(light.v);\n    let uv = vec2f(\n      dot(relative, light.u / width) / width,\n      dot(relative, light.v / height) / height\n    );\n    if (planeDistance <= 0.0 || any(uv < vec2f(0.0)) || any(uv > vec2f(1.0))) {\n      (*result).emission = vec3f(0.0);\n      (*result).pdf = 0.0;\n      return;\n    }\n    (*result).direction = -normal;\n    (*result).dist = planeDistance;\n    (*result).normal = normal;\n    (*result).emission = light.emission;\n    (*result).pdf = 1.0;\n    (*result).isDelta = true;\n    light_apply_rect_texture(light, surfacePoint, lowerLeft, result);\n    return;\n  }\n\n  let surfacePoint = lowerLeft\n    + light.u * samplePoint.x\n    + light.v * samplePoint.y;\n  let offset = surfacePoint - shadingPoint;\n  let distanceSquared = dot(offset, offset);\n  (*result).dist = sqrt(distanceSquared);\n  (*result).direction = offset / (*result).dist;\n  (*result).normal = normal;\n  let emitterCosine = dot(-(*result).direction, normal);\n  let spreadCosine = cos(0.5 * PI * spread);\n  if (emitterCosine <= 0.0 || emitterCosine < spreadCosine) {\n    (*result).emission = vec3f(0.0);\n    (*result).pdf = 0.0;\n    return;\n  }\n  (*result).emission = light.emission;\n  (*result).pdf = distanceSquared / max(light.area * emitterCosine, 1e-8);\n  light_apply_rect_texture(light, surfacePoint, lowerLeft, result);\n}\n\nfn light_sample_directional(\n  light: Light,\n  result: ptr<function, LightSampleRec>\n) {\n  (*result).direction = normalize(-light.u);\n  (*result).normal = -(*result).direction;\n  (*result).emission = light.emission;\n  (*result).dist = INF;\n  (*result).pdf = 1.0;\n  (*result).isDelta = true;\n}\n\nfn light_sample_spot(\n  light: Light,\n  shadingPoint: vec3f,\n  samplePoint: vec2f,\n  result: ptr<function, LightSampleRec>\n) {\n  light_sample_sphere(light, shadingPoint, samplePoint, result);\n  let surfacePoint = shadingPoint + (*result).direction * (*result).dist;\n  let lightToSurface = normalize(shadingPoint - surfacePoint);\n  (*result).emission *= evaluateSpotEmissionMultiplier(light, lightToSurface);\n}\n\nfn sampleAnalyticLight(\n  light: Light,\n  shadingPoint: vec3f,\n  samplePoint: vec2f,\n  result: ptr<function, LightSampleRec>\n) {\n  let lightType = i32(light.litType);\n  if (HAS_SCENE_RECT_LIGHT && lightType == RECT_LIGHT) {\n    light_sample_rect(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (HAS_SCENE_SPHERE_LIGHT && lightType == SPHERE_LIGHT) {\n    light_sample_sphere(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (HAS_SCENE_SPOT_LIGHT && lightType == SPOT_LIGHT) {\n    light_sample_spot(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (HAS_SCENE_DIRECTIONAL_LIGHT && lightType == DIRECTIONAL_LIGHT) {\n    light_sample_directional(light, result);\n  }\n}\n\nfn sampleEnvLight(randomV2: vec2f, lightSample: ptr<function, LightSampleRec>) {\n  var uv: vec2f;\n  // Naga's SPIR-V writer cannot pass a struct-member pointer as a call argument.\n  var pdf: f32;\n  let direction = sampleEnvmap(randomV2, &uv, &pdf);\n  (*lightSample).pdf = pdf;\n  (*lightSample).direction = direction;\n  (*lightSample).emission = textureLinear(envMap, uv).rgb;\n  (*lightSample).dist = INF;\n}\n\nfn loadLightSamplingTexel(texelIndex: i32) -> vec4f {\n  let width = rtUniform.lightSamplingTextureWidth;\n  return textureLoad(\n    lightSamplingData,\n    vec2i(texelIndex % width, texelIndex / width),\n    0\n  );\n}\n\nfn getEmissiveTriangle(index: i32) -> EmissiveTriangle {\n  let offset = index * 6;\n  let a = loadLightSamplingTexel(offset);\n  let b = loadLightSamplingTexel(offset + 1);\n  let c = loadLightSamplingTexel(offset + 2);\n  let d = loadLightSamplingTexel(offset + 3);\n  let e = loadLightSamplingTexel(offset + 4);\n  let f = loadLightSamplingTexel(offset + 5);\n  return EmissiveTriangle(\n    a.xyz, a.w,\n    b.xyz, b.w,\n    c.xyz, c.w,\n    d.xy, d.zw, e.xy,\n    e.zw, f.xy, f.zw\n  );\n}\n\nfn getLightChoice(index: i32) -> LightChoice {\n  let data = loadLightSamplingTexel(rtUniform.lightChoiceTexelOffset + index);\n  return LightChoice(\n    data.x,\n    bitcast<i32>(data.y),\n    bitcast<i32>(data.z),\n    bitcast<i32>(data.w)\n  );\n}\n\nfn getLightChoiceFromTable(tableOffset: i32, index: i32) -> LightChoice {\n  return getLightChoice(tableOffset + index);\n}\n\nfn stratifyLightChoiceRandom(\n  rnd: vec3f,\n  candidateIndex: u32,\n  candidateCount: u32\n) -> vec3f {\n  var stratified = rnd;\n  let count = max(candidateCount, 1u);\n  stratified.z = (rnd.z + f32(min(candidateIndex, count - 1u))) /\n    f32(count);\n  return stratified;\n}\n\nfn sampleEmissiveTriangleLight(\n  triangle: EmissiveTriangle,\n  scatterPos: vec3f,\n  randomV2: vec2f,\n  lightSample: ptr<function, LightSampleRec>\n) {\n  let sqrtU = sqrt(randomV2.x);\n  let barycentric = vec3f(\n    1.0 - sqrtU,\n    sqrtU * (1.0 - randomV2.y),\n    sqrtU * randomV2.y\n  );\n  let lightPosition =\n    barycentric.x * triangle.p0 +\n    barycentric.y * triangle.p1 +\n    barycentric.z * triangle.p2;\n  let uv =\n    barycentric.x * triangle.uv0 +\n    barycentric.y * triangle.uv1 +\n    barycentric.z * triangle.uv2;\n  let secondaryUv =\n    barycentric.x * triangle.secondaryUv0 +\n    barycentric.y * triangle.secondaryUv1 +\n    barycentric.z * triangle.secondaryUv2;\n  let normal = normalize(cross(triangle.p1 - triangle.p0, triangle.p2 - triangle.p0));\n  (*lightSample).direction = lightPosition - scatterPos;\n  (*lightSample).dist = length((*lightSample).direction);\n  if ((*lightSample).dist <= EPS) {\n    (*lightSample).emission = vec3f(0.0);\n    (*lightSample).pdf = 0.0;\n    return;\n  }\n  (*lightSample).direction /= (*lightSample).dist;\n  (*lightSample).normal = normal;\n  let cosLight = dot(normal, -(*lightSample).direction);\n  if (cosLight <= 0.0) {\n    (*lightSample).emission = vec3f(0.0);\n    (*lightSample).pdf = 0.0;\n    return;\n  }\n  (*lightSample).emission = getMaterialEmission(\n    i32(triangle.materialIndex), uv, secondaryUv\n  );\n  (*lightSample).selectionPmf = triangle.selectionPmf;\n  (*lightSample).pdf = triangle.selectionPmf *\n    ((*lightSample).dist * (*lightSample).dist) /\n    max(triangle.area * cosLight, 1e-8);\n}\n\nfn sampleLightChoiceTable(\n  scatterPos: vec3f,\n  rnd: vec3f,\n  tableOffset: i32,\n  choiceCount: i32,\n  selectionPmfScale: f32,\n  lightSample: ptr<function, LightSampleRec>\n) {\n  (*lightSample).emission = vec3f(0.0);\n  (*lightSample).pdf = 0.0;\n  (*lightSample).selectionPmf = 0.0;\n  (*lightSample).dist = INF;\n  (*lightSample).isDelta = false;\n  (*lightSample).sourceType = 0u;\n  (*lightSample).sourceIndex = 0xffffffffu;\n  if (choiceCount <= 0) {\n    return;\n  }\n\n  let scaledChoice = rnd.z * f32(choiceCount);\n  let column = min(i32(scaledChoice), choiceCount - 1);\n  let columnChoice = getLightChoiceFromTable(tableOffset, column);\n  let selectedIndex = select(\n    columnChoice.aliasIndex,\n    column,\n    fract(scaledChoice) < columnChoice.probability\n  );\n  let choice = getLightChoiceFromTable(tableOffset, selectedIndex);\n  if (USE_ENVMAP && choice.sourceType == ${LIGHT_CHOICE_ENVIRONMENT}) {\n    (*lightSample).sourceType = 0u;\n    (*lightSample).sourceIndex = u32(choice.sourceIndex);\n    sampleEnvLight(rnd.xy, lightSample);\n    (*lightSample).selectionPmf = rtUniform.envLightPmf;\n    (*lightSample).pdf *= rtUniform.envLightPmf;\n  }\n  else if (choice.sourceType == ${LIGHT_CHOICE_ANALYTIC}) {\n    (*lightSample).sourceType = 2u;\n    (*lightSample).sourceIndex = u32(choice.sourceIndex);\n    let light = analyticLights[choice.sourceIndex];\n    sampleAnalyticLight(light, scatterPos, rnd.xy, lightSample);\n    (*lightSample).selectionPmf = light.selectionPmf;\n    (*lightSample).pdf *= light.selectionPmf;\n  }\n  else if (HAS_SCENE_EMISSIVE_LIGHT && choice.sourceType == ${LIGHT_CHOICE_EMISSIVE_TRIANGLE}) {\n    (*lightSample).sourceType = 1u;\n    (*lightSample).sourceIndex = u32(choice.sourceIndex);\n    sampleEmissiveTriangleLight(\n      getEmissiveTriangle(choice.sourceIndex),\n      scatterPos,\n      rnd.xy,\n      lightSample\n    );\n  }\n  (*lightSample).selectionPmf *= selectionPmfScale;\n  (*lightSample).pdf *= selectionPmfScale;\n}\n\nfn sampleLight(\n  scatterPos: vec3f,\n  rnd: vec3f,\n  lightSample: ptr<function, LightSampleRec>,\n  distByIntensity: bool\n) {\n  sampleLightChoiceTable(\n    scatterPos,\n    rnd,\n    0,\n    rtUniform.lightChoiceCount,\n    1.0,\n    lightSample\n  );\n}\n\n// Light choice for one of the resampled NEE candidates. The selection random\n// number is stratified across the candidates.\nfn sampleLightForNEECandidate(\n  scatterPos: vec3f,\n  rnd: vec3f,\n  candidateIndex: u32,\n  candidateCount: u32,\n  lightSample: ptr<function, LightSampleRec>\n) {\n  sampleLightChoiceTable(\n    scatterPos,\n    stratifyLightChoiceRandom(rnd, candidateIndex, candidateCount),\n    0,\n    rtUniform.lightChoiceCount,\n    1.0,\n    lightSample\n  );\n}\n`;\n"]}

package/lib/renderer/wgsl/chunks/sampleMaterial.wgsl.d.ts

@@ -1,3 +1,3 @@
-export declare function createSampleMaterialWgsl(hasAnisotropyDirections?: boolean, hasCoatNormals?: boolean): string;
+export declare function createSampleMaterialWgsl(hasAnisotropyDirections?: boolean, hasCoatNormals?: boolean, caustics?: boolean, researchContribution?: string, photonMapping?: boolean): string;
 declare const _default: string;
 export default _default;