@meimu/mutracer 0.0.0-stage → 0.1.1

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

@@ -0,0 +1,400 @@
+import { LIGHT_CHOICE_ANALYTIC, LIGHT_CHOICE_EMISSIVE_TRIANGLE, LIGHT_CHOICE_ENVIRONMENT } from '../../lightSampling.js';
+export default /*wgsl */ `
+fn eJ(light: Light, fm: vec3f, iT: vec2f, lightSample: ptr<function, LightSampleRec>) {
+let iq = light.position - fm;
+let aj = length(iq);
+if (light.radiusOrSpread <= 0.0) {
+
+
+
+if (aj <= 1e-6) {
+(*lightSample).direction = vec3f(0.0, 0.0, 1.0);
+(*lightSample).normal = vec3f(0.0, 0.0, -1.0);
+(*lightSample).dist = 0.0;
+(*lightSample).pdf = 1.0;
+(*lightSample).emission = vec3f(0.0);
+(*lightSample).isDelta = true;
+return;
+}
+(*lightSample).direction = iq / aj;
+(*lightSample).normal = -(*lightSample).direction;
+(*lightSample).dist = aj;
+(*lightSample).pdf = 1.0;
+(*lightSample).emission = light.emission /
+max(aj * aj, 1e-8);
+(*lightSample).isDelta = true;
+return;
+}
+let radius = max(light.radiusOrSpread, 1e-6);
+(*lightSample).isDelta = false;
+var gC: vec3f;
+
+if (aj <= radius) {
+let z = 1.0 - 2.0 * iT.x;
+let fi = sqrt(max(0.0, 1.0 - z * z));
+let br = TWOPI * iT.y;
+gC = vec3f(
+fi * cos(br),
+fi * sin(br),
+z
+);
+(*lightSample).pdf = INV4PI;
+}
+else {
+let sF = iq / aj;
+let jn = min(
+1.0,
+radius * radius / (aj * aj)
+);
+let cosThetaMax = sqrt(max(0.0, 1.0 - jn));
+let cc = mix(1.0, cosThetaMax, iT.x);
+let sinTheta = sqrt(max(0.0, 1.0 - cc * cc));
+let br = TWOPI * iT.y;
+var tangent: vec3f;
+var bitangent: vec3f;
+aK(sF, &tangent, &bitangent);
+gC = sF * cc
++ tangent * (sinTheta * cos(br))
++ bitangent * (sinTheta * sin(br));
+(*lightSample).pdf = 1.0 / max(
+TWOPI * (1.0 - cosThetaMax),
+1e-8
+);
+}
+
+let ir = dot(gC, iq);
+let pQ = dot(iq, iq);
+let ej = ir * ir
+- pQ + radius * radius;
+let cN = ir - sqrt(max(0.0, ej));
+let cB = ir + sqrt(max(0.0, ej));
+(*lightSample).dist = select(cB, cN, cN > 1e-6);
+(*lightSample).direction = gC;
+let rQ = fm + gC * (*lightSample).dist;
+(*lightSample).normal = normalize(rQ - light.position);
+(*lightSample).emission = light.emission;
+}
+
+fn cZ(
+light: Light,
+surfacePoint: vec3f,
+lowerLeft: vec3f,
+result: ptr<function, LightSampleRec>
+) {
+if (light.textureIndex < 0.0) {
+return;
+}
+let width = length(light.u);
+let height = length(light.v);
+let hY = surfacePoint - lowerLeft;
+let uv = vec2f(
+dot(hY, light.u / width) / width,
+dot(hY, light.v / height) / height
+);
+(*result).emission *= sampleMap(i32(light.textureIndex), uv).rgb;
+}
+
+fn jJ(
+light: Light,
+shadingPoint: vec3f,
+samplePoint: vec2f,
+result: ptr<function, LightSampleRec>
+) {
+let normal = normalize(cross(light.u, light.v));
+let lowerLeft = light.position - 0.5 * (light.u + light.v);
+let spread = clamp(light.radiusOrSpread, 0.0, 1.0);
+
+if (spread <= 0.001) {
+let ga = dot(normal, shadingPoint - light.position);
+let surfacePoint = shadingPoint - normal * ga;
+let hY = surfacePoint - lowerLeft;
+let width = length(light.u);
+let height = length(light.v);
+let uv = vec2f(
+dot(hY, light.u / width) / width,
+dot(hY, light.v / height) / height
+);
+if (ga <= 0.0 || any(uv < vec2f(0.0)) || any(uv > vec2f(1.0))) {
+(*result).emission = vec3f(0.0);
+(*result).pdf = 0.0;
+return;
+}
+(*result).direction = -normal;
+(*result).dist = ga;
+(*result).normal = normal;
+(*result).emission = light.emission;
+(*result).pdf = 1.0;
+(*result).isDelta = true;
+cZ(light, surfacePoint, lowerLeft, result);
+return;
+}
+
+let surfacePoint = lowerLeft
++ light.u * samplePoint.x
++ light.v * samplePoint.y;
+let offset = surfacePoint - shadingPoint;
+let hl = dot(offset, offset);
+(*result).dist = sqrt(hl);
+(*result).direction = offset / (*result).dist;
+(*result).normal = normal;
+let fV = dot(-(*result).direction, normal);
+let oC = cos(0.5 * PI * spread);
+if (fV <= 0.0 || fV < oC) {
+(*result).emission = vec3f(0.0);
+(*result).pdf = 0.0;
+return;
+}
+(*result).emission = light.emission;
+(*result).pdf = hl / max(light.area * fV, 1e-8);
+cZ(light, surfacePoint, lowerLeft, result);
+}
+
+fn fX(
+light: Light,
+result: ptr<function, LightSampleRec>
+) {
+(*result).direction = normalize(-light.u);
+(*result).normal = -(*result).direction;
+(*result).emission = light.emission;
+(*result).dist = INF;
+(*result).pdf = 1.0;
+(*result).isDelta = true;
+}
+
+fn jK(
+light: Light,
+shadingPoint: vec3f,
+samplePoint: vec2f,
+result: ptr<function, LightSampleRec>
+) {
+eJ(light, shadingPoint, samplePoint, result);
+let surfacePoint = shadingPoint + (*result).direction * (*result).dist;
+let dJ = normalize(shadingPoint - surfacePoint);
+(*result).emission *= bL(light, dJ);
+}
+
+fn iK(
+light: Light,
+shadingPoint: vec3f,
+samplePoint: vec2f,
+result: ptr<function, LightSampleRec>
+) {
+let dO = i32(light.litType);
+if (dO == RECT_LIGHT) {
+jJ(light, shadingPoint, samplePoint, result);
+return;
+}
+if (dO == SPHERE_LIGHT) {
+eJ(light, shadingPoint, samplePoint, result);
+return;
+}
+if (dO == SPOT_LIGHT) {
+jK(light, shadingPoint, samplePoint, result);
+return;
+}
+if (dO == DIRECTIONAL_LIGHT) {
+fX(light, result);
+}
+}
+
+fn mu(iT: vec2f, lightSample: ptr<function, LightSampleRec>) {
+var uv: vec2f;
+(*lightSample).direction = oA(iT, &uv, &(*lightSample).pdf);
+(*lightSample).emission = textureLinear(envMap, uv).rgb;
+(*lightSample).dist = INF;
+}
+
+fn bV(ij: i32) -> vec4f {
+let width = rtUniform.lightSamplingTextureWidth;
+return textureLoad(
+lightSamplingData,
+vec2i(ij % width, ij / width),
+0
+);
+}
+
+fn pR(index: i32) -> EmissiveTriangle {
+let offset = index * 6;
+let a = bV(offset);
+let b = bV(offset + 1);
+let c = bV(offset + 2);
+let d = bV(offset + 3);
+let e = bV(offset + 4);
+let f = bV(offset + 5);
+return EmissiveTriangle(
+a.xyz, a.w,
+b.xyz, b.w,
+c.xyz, c.w,
+d.xy, d.zw, e.xy,
+e.zw, f.xy, f.zw
+);
+}
+
+fn sa(index: i32) -> oR {
+let data = bV(rtUniform.lightChoiceTexelOffset + index);
+return oR(
+data.x,
+bitcast<i32>(data.y),
+bitcast<i32>(data.z),
+bitcast<i32>(data.w)
+);
+}
+
+fn jt(mN: i32, index: i32) -> oR {
+return sa(mN + index);
+}
+
+fn mF(
+rnd: vec3f,
+ca: u32,
+aW: u32
+) -> vec3f {
+var sH = rnd;
+let count = max(aW, 1u);
+sH.z = (rnd.z + f32(min(ca, count - 1u))) /
+f32(count);
+return sH;
+}
+
+fn lK(
+triangle: EmissiveTriangle,
+fm: vec3f,
+iT: vec2f,
+lightSample: ptr<function, LightSampleRec>
+) {
+let vl = sqrt(iT.x);
+let barycentric = vec3f(
+1.0 - vl,
+vl * (1.0 - iT.y),
+vl * iT.y
+);
+let sN =
+barycentric.x * triangle.p0 +
+barycentric.y * triangle.p1 +
+barycentric.z * triangle.p2;
+let uv =
+barycentric.x * triangle.uv0 +
+barycentric.y * triangle.uv1 +
+barycentric.z * triangle.uv2;
+let cX =
+barycentric.x * triangle.secondaryUv0 +
+barycentric.y * triangle.secondaryUv1 +
+barycentric.z * triangle.secondaryUv2;
+let normal = normalize(cross(triangle.p1 - triangle.p0, triangle.p2 - triangle.p0));
+(*lightSample).direction = sN - fm;
+(*lightSample).dist = length((*lightSample).direction);
+if ((*lightSample).dist <= EPS) {
+(*lightSample).emission = vec3f(0.0);
+(*lightSample).pdf = 0.0;
+return;
+}
+(*lightSample).direction /= (*lightSample).dist;
+(*lightSample).normal = normal;
+let oZ = dot(normal, -(*lightSample).direction);
+if (oZ <= 0.0) {
+(*lightSample).emission = vec3f(0.0);
+(*lightSample).pdf = 0.0;
+return;
+}
+(*lightSample).emission = lB(
+i32(triangle.materialIndex), uv, cX
+);
+(*lightSample).selectionPmf = triangle.selectionPmf;
+(*lightSample).pdf = triangle.selectionPmf *
+((*lightSample).dist * (*lightSample).dist) /
+max(triangle.area * oZ, 1e-8);
+}
+
+fn jZ(
+fm: vec3f,
+rnd: vec3f,
+mN: i32,
+oY: i32,
+mM: f32,
+lightSample: ptr<function, LightSampleRec>
+) {
+(*lightSample).emission = vec3f(0.0);
+(*lightSample).pdf = 0.0;
+(*lightSample).selectionPmf = 0.0;
+(*lightSample).dist = INF;
+(*lightSample).isDelta = false;
+(*lightSample).sourceType = 0u;
+(*lightSample).sourceIndex = 0xffffffffu;
+if (oY <= 0) {
+return;
+}
+
+let qV = rnd.z * f32(oY);
+let column = min(i32(qV), oY - 1);
+let qF = jt(mN, column);
+let kQ = select(
+qF.aliasIndex,
+column,
+fract(qV) < qF.probability
+);
+let oX = jt(mN, kQ);
+if (oX.sourceType == ${LIGHT_CHOICE_ENVIRONMENT}) {
+(*lightSample).sourceType = 0u;
+(*lightSample).sourceIndex = u32(oX.sourceIndex);
+mu(rnd.xy, lightSample);
+(*lightSample).selectionPmf = rtUniform.envLightPmf;
+(*lightSample).pdf *= rtUniform.envLightPmf;
+}
+else if (oX.sourceType == ${LIGHT_CHOICE_ANALYTIC}) {
+(*lightSample).sourceType = 2u;
+(*lightSample).sourceIndex = u32(oX.sourceIndex);
+let light = analyticLights[oX.sourceIndex];
+iK(light, fm, rnd.xy, lightSample);
+(*lightSample).selectionPmf = light.selectionPmf;
+(*lightSample).pdf *= light.selectionPmf;
+}
+else if (oX.sourceType == ${LIGHT_CHOICE_EMISSIVE_TRIANGLE}) {
+(*lightSample).sourceType = 1u;
+(*lightSample).sourceIndex = u32(oX.sourceIndex);
+lK(
+pR(oX.sourceIndex),
+fm,
+rnd.xy,
+lightSample
+);
+}
+(*lightSample).selectionPmf *= mM;
+(*lightSample).pdf *= mM;
+}
+
+fn sampleLight(
+fm: vec3f,
+rnd: vec3f,
+lightSample: ptr<function, LightSampleRec>,
+rP: bool
+) {
+jZ(
+fm,
+rnd,
+0,
+rtUniform.lightChoiceCount,
+1.0,
+lightSample
+);
+}
+
+
+
+fn mv(
+fm: vec3f,
+rnd: vec3f,
+ca: u32,
+aW: u32,
+lightSample: ptr<function, LightSampleRec>
+) {
+jZ(
+fm,
+mF(rnd, ca, aW),
+0,
+rtUniform.lightChoiceCount,
+1.0,
+lightSample
+);
+}
+`;
+//# sourceMappingURL=sampleLight.wgsl.js.map

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

@@ -0,0 +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,wBAAwB,CAAC;AAEhC,eAAe,SAAS,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uBA8UI,wBAAwB;;;;;;;4BAOnB,qBAAqB;;;;;;;;4BAQrB,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgD/D,CAAC","sourcesContent":["import {\n  LIGHT_CHOICE_ANALYTIC,\n  LIGHT_CHOICE_EMISSIVE_TRIANGLE,\n  LIGHT_CHOICE_ENVIRONMENT\n} from '../../lightSampling.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 (lightType == RECT_LIGHT) {\n    light_sample_rect(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (lightType == SPHERE_LIGHT) {\n    light_sample_sphere(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (lightType == SPOT_LIGHT) {\n    light_sample_spot(light, shadingPoint, samplePoint, result);\n    return;\n  }\n  if (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  (*lightSample).direction = sampleEnvmap(randomV2, &uv, &(*lightSample).pdf);\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 (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 (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

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