three-gpu-pathtracer 0.0.13 → 0.0.14

package/src/shader/shaderBvhAnyHit.js

@@ -1,76 +0,0 @@
-export const shaderBvhAnyHit = /* glsl */`
-
-bool bvhIntersectAnyHit(
-	BVH bvh, vec3 rayOrigin, vec3 rayDirection,
-
-	// output variables
-	out float side, out float dist
-) {
-
-	uvec4 faceIndices;
-	vec3 faceNormal;
-	vec3 barycoord;
-
-	// stack needs to be twice as long as the deepest tree we expect because
-	// we push both the left and right child onto the stack every traversal
-	int ptr = 0;
-	uint stack[ 60 ];
-	stack[ 0 ] = 0u;
-
-	float triangleDistance = 1e20;
-	while ( ptr > - 1 && ptr < 60 ) {
-
-		uint currNodeIndex = stack[ ptr ];
-		ptr --;
-
-		// check if we intersect the current bounds
-		float boundsHitDistance = intersectsBVHNodeBounds( rayOrigin, rayDirection, bvh, currNodeIndex );
-		if ( boundsHitDistance == INFINITY ) {
-
-			continue;
-
-		}
-
-		uvec2 boundsInfo = uTexelFetch1D( bvh.bvhContents, currNodeIndex ).xy;
-		bool isLeaf = bool( boundsInfo.x & 0xffff0000u );
-
-		if ( isLeaf ) {
-
-			uint count = boundsInfo.x & 0x0000ffffu;
-			uint offset = boundsInfo.y;
-
-			bool found = intersectTriangles(
-				bvh, rayOrigin, rayDirection, offset, count, triangleDistance,
-				faceIndices, faceNormal, barycoord, side, dist
-			);
-
-			if ( found ) {
-
-				return true;
-
-			}
-
-		} else {
-
-			uint leftIndex = currNodeIndex + 1u;
-			uint splitAxis = boundsInfo.x & 0x0000ffffu;
-			uint rightIndex = boundsInfo.y;
-
-			// set c2 in the stack so we traverse it later. We need to keep track of a pointer in
-			// the stack while we traverse. The second pointer added is the one that will be
-			// traversed first
-			ptr ++;
-			stack[ ptr ] = leftIndex;
-
-			ptr ++;
-			stack[ ptr ] = rightIndex;
-
-		}
-
-	}
-
-	return false;
-
-}
-
-`;

package/src/shader/shaderEnvMapSampling.js

@@ -1,58 +0,0 @@
-export const shaderEnvMapSampling = /* glsl */`
-
-vec3 sampleEquirectEnvMapColor( vec3 direction, sampler2D map ) {
-
-	return texture2D( map, equirectDirectionToUv( direction ) ).rgb;
-
-}
-
-float envMapDirectionPdf( vec3 direction ) {
-
-	vec2 uv = equirectDirectionToUv( direction );
-	float theta = uv.y * PI;
-	float sinTheta = sin( theta );
-	if ( sinTheta == 0.0 ) {
-
-		return 0.0;
-
-	}
-
-	return 1.0 / ( 2.0 * PI * PI * sinTheta );
-
-}
-
-float sampleEnvMap( EquirectHdrInfo info, vec3 direction, out vec3 color ) {
-
-	vec2 uv = equirectDirectionToUv( direction );
-	color = texture2D( info.map, uv ).rgb;
-
-	float totalSum = info.totalSumWhole + info.totalSumDecimal;
-	float lum = luminance( color );
-	ivec2 resolution = textureSize( info.map, 0 );
-	float pdf = lum / totalSum;
-
-	return float( resolution.x * resolution.y ) * pdf * envMapDirectionPdf( direction );
-
-}
-
-float sampleEnvMapProbability( EquirectHdrInfo info, vec2 r, out vec3 color, out vec3 direction ) {
-
-	// sample env map cdf
-	float v = texture2D( info.marginalWeights, vec2( r.x, 0.0 ) ).x;
-	float u = texture2D( info.conditionalWeights, vec2( r.y, v ) ).x;
-	vec2 uv = vec2( u, v );
-
-	vec3 derivedDirection = equirectUvToDirection( uv );
-	direction = derivedDirection;
-	color = texture2D( info.map, uv ).rgb;
-
-	float totalSum = info.totalSumWhole + info.totalSumDecimal;
-	float lum = luminance( color );
-	ivec2 resolution = textureSize( info.map, 0 );
-	float pdf = lum / totalSum;
-
-	return float( resolution.x * resolution.y ) * pdf * envMapDirectionPdf( direction );
-
-}
-
-`;

package/src/shader/shaderGGXFunctions.js

@@ -1,100 +0,0 @@
-export const shaderGGXFunctions = /* glsl */`
-// The GGX functions provide sampling and distribution information for normals as output so
-// in order to get probability of scatter direction the half vector must be computed and provided.
-// [0] https://www.cs.cornell.edu/~srm/publications/EGSR07-btdf.pdf
-// [1] https://hal.archives-ouvertes.fr/hal-01509746/document
-// [2] http://jcgt.org/published/0007/04/01/
-// [4] http://jcgt.org/published/0003/02/03/
-
-// trowbridge-reitz === GGX === GTR
-
-vec3 ggxDirection( vec3 incidentDir, vec2 roughness, vec2 uv ) {
-
-	// TODO: try GGXVNDF implementation from reference [2], here. Needs to update ggxDistribution
-	// function below, as well
-
-	// Implementation from reference [1]
-	// stretch view
-	vec3 V = normalize( vec3( roughness * incidentDir.xy, incidentDir.z ) );
-
-	// orthonormal basis
-	vec3 T1 = ( V.z < 0.9999 ) ? normalize( cross( V, vec3( 0.0, 0.0, 1.0 ) ) ) : vec3( 1.0, 0.0, 0.0 );
-	vec3 T2 = cross( T1, V );
-
-	// sample point with polar coordinates (r, phi)
-	float a = 1.0 / ( 1.0 + V.z );
-	float r = sqrt( uv.x );
-	float phi = ( uv.y < a ) ? uv.y / a * PI : PI + ( uv.y - a ) / ( 1.0 - a ) * PI;
-	float P1 = r * cos( phi );
-	float P2 = r * sin( phi ) * ( ( uv.y < a ) ? 1.0 : V.z );
-
-	// compute normal
-	vec3 N = P1 * T1 + P2 * T2 + V * sqrt( max( 0.0, 1.0 - P1 * P1 - P2 * P2 ) );
-
-	// unstretch
-	N = normalize( vec3( roughness * N.xy, max( 0.0, N.z ) ) );
-
-	return N;
-
-}
-
-// Below are PDF and related functions for use in a Monte Carlo path tracer
-// as specified in Appendix B of the following paper
-// See equation (34) from reference [0]
-float ggxLamda( float theta, float roughness ) {
-
-	float tanTheta = tan( theta );
-	float tanTheta2 = tanTheta * tanTheta;
-	float alpha2 = roughness * roughness;
-
-	float numerator = - 1.0 + sqrt( 1.0 + alpha2 * tanTheta2 );
-	return numerator / 2.0;
-
-}
-
-// See equation (34) from reference [0]
-float ggxShadowMaskG1( float theta, float roughness ) {
-
-	return 1.0 / ( 1.0 + ggxLamda( theta, roughness ) );
-
-}
-
-// See equation (125) from reference [4]
-float ggxShadowMaskG2( vec3 wi, vec3 wo, float roughness ) {
-
-	float incidentTheta = acos( wi.z );
-	float scatterTheta = acos( wo.z );
-	return 1.0 / ( 1.0 + ggxLamda( incidentTheta, roughness ) + ggxLamda( scatterTheta, roughness ) );
-
-}
-
-// See equation (33) from reference [0]
-float ggxDistribution( vec3 halfVector, float roughness ) {
-
-	float a2 = roughness * roughness;
-	a2 = max( EPSILON, a2 );
-	float cosTheta = halfVector.z;
-	float cosTheta4 = pow( cosTheta, 4.0 );
-
-	if ( cosTheta == 0.0 ) return 0.0;
-
-	float theta = acosSafe( halfVector.z );
-	float tanTheta = tan( theta );
-	float tanTheta2 = pow( tanTheta, 2.0 );
-
-	float denom = PI * cosTheta4 * pow( a2 + tanTheta2, 2.0 );
-	return ( a2 / denom );
-
-}
-
-// See equation (3) from reference [2]
-float ggxPDF( vec3 wi, vec3 halfVector, float roughness ) {
-
-	float incidentTheta = acos( wi.z );
-	float D = ggxDistribution( halfVector, roughness );
-	float G1 = ggxShadowMaskG1( incidentTheta, roughness );
-
-	return D * G1 * max( 0.0, dot( wi, halfVector ) ) / wi.z;
-
-}
-`;

package/src/shader/shaderIridescenceFunctions.js

@@ -1,135 +0,0 @@
-export const shaderIridescenceFunctions = /* glsl */`
-
-// XYZ to sRGB color space
-const mat3 XYZ_TO_REC709 = mat3(
-	 3.2404542, -0.9692660,  0.0556434,
-	-1.5371385,  1.8760108, -0.2040259,
-	-0.4985314,  0.0415560,  1.0572252
-);
-
-vec3 fresnel0ToIor( vec3 fresnel0 ) {
-
-	vec3 sqrtF0 = sqrt( fresnel0 );
-	return ( vec3( 1.0 ) + sqrtF0 ) / ( vec3( 1.0 ) - sqrtF0 );
-
-}
-
-// Conversion FO/IOR
-vec3 iorToFresnel0( vec3 transmittedIor, float incidentIor ) {
-
-	return square( ( transmittedIor - vec3( incidentIor ) ) / ( transmittedIor + vec3( incidentIor ) ) );
-
-}
-
-// ior is a value between 1.0 and 3.0. 1.0 is air interface
-float iorToFresnel0( float transmittedIor, float incidentIor ) {
-
-	return square( ( transmittedIor - incidentIor ) / ( transmittedIor + incidentIor ) );
-
-}
-
-// Fresnel equations for dielectric/dielectric interfaces. See https://belcour.github.io/blog/research/2017/05/01/brdf-thin-film.html
-vec3 evalSensitivity( float OPD, vec3 shift ) {
-
-	float phase = 2.0 * PI * OPD * 1.0e-9;
-
-	vec3 val = vec3( 5.4856e-13, 4.4201e-13, 5.2481e-13 );
-	vec3 pos = vec3( 1.6810e+06, 1.7953e+06, 2.2084e+06 );
-	vec3 var = vec3( 4.3278e+09, 9.3046e+09, 6.6121e+09 );
-
-	vec3 xyz = val * sqrt( 2.0 * PI * var ) * cos( pos * phase + shift ) * exp( - square( phase ) * var );
-	xyz.x += 9.7470e-14 * sqrt( 2.0 * PI * 4.5282e+09 ) * cos( 2.2399e+06 * phase + shift[ 0 ] ) * exp( - 4.5282e+09 * square( phase ) );
-	xyz /= 1.0685e-7;
-
-	vec3 srgb = XYZ_TO_REC709 * xyz;
-	return srgb;
-
-}
-
-// See Section 4. Analytic Spectral Integration, A Practical Extension to Microfacet Theory for the Modeling of Varying Iridescence, https://hal.archives-ouvertes.fr/hal-01518344/document
-vec3 evalIridescence( float outsideIOR, float eta2, float cosTheta1, float thinFilmThickness, vec3 baseF0 ) {
-
-	vec3 I;
-
-	// Force iridescenceIor -> outsideIOR when thinFilmThickness -> 0.0
-	float iridescenceIor = mix( outsideIOR, eta2, smoothstep( 0.0, 0.03, thinFilmThickness ) );
-
-	// Evaluate the cosTheta on the base layer (Snell law)
-	float sinTheta2Sq = square( outsideIOR / iridescenceIor ) * ( 1.0 - square( cosTheta1 ) );
-
-	// Handle TIR:
-	float cosTheta2Sq = 1.0 - sinTheta2Sq;
-	if ( cosTheta2Sq < 0.0 ) {
-
-		return vec3( 1.0 );
-
-	}
-
-	float cosTheta2 = sqrt( cosTheta2Sq );
-
-	// First interface
-	float R0 = iorToFresnel0( iridescenceIor, outsideIOR );
-	float R12 = schlickFresnel( cosTheta1, R0 );
-	float R21 = R12;
-	float T121 = 1.0 - R12;
-	float phi12 = 0.0;
-	if ( iridescenceIor < outsideIOR ) {
-
-		phi12 = PI;
-
-	}
-
-	float phi21 = PI - phi12;
-
-	// Second interface
-	vec3 baseIOR = fresnel0ToIor( clamp( baseF0, 0.0, 0.9999 ) ); // guard against 1.0
-	vec3 R1 = iorToFresnel0( baseIOR, iridescenceIor );
-	vec3 R23 = schlickFresnel( cosTheta2, R1 );
-	vec3 phi23 = vec3( 0.0 );
-	if ( baseIOR[0] < iridescenceIor ) {
-
-		phi23[ 0 ] = PI;
-
-	}
-
-	if ( baseIOR[1] < iridescenceIor ) {
-
-		phi23[ 1 ] = PI;
-
-	}
-
-	if ( baseIOR[2] < iridescenceIor ) {
-
-		phi23[ 2 ] = PI;
-
-	}
-
-	// Phase shift
-	float OPD = 2.0 * iridescenceIor * thinFilmThickness * cosTheta2;
-	vec3 phi = vec3( phi21 ) + phi23;
-
-	// Compound terms
-	vec3 R123 = clamp( R12 * R23, 1e-5, 0.9999 );
-	vec3 r123 = sqrt( R123 );
-	vec3 Rs = square( T121 ) * R23 / ( vec3( 1.0 ) - R123 );
-
-	// Reflectance term for m = 0 (DC term amplitude)
-	vec3 C0 = R12 + Rs;
-	I = C0;
-
-	// Reflectance term for m > 0 (pairs of diracs)
-	vec3 Cm = Rs - T121;
-	for ( int m = 1; m <= 2; ++ m ) {
-
-		Cm *= r123;
-		vec3 Sm = 2.0 * evalSensitivity( float( m ) * OPD, float( m ) * phi );
-		I += Cm * Sm;
-
-	}
-
-	// Since out of gamut colors might be produced, negative color values are clamped to 0.
-	return max( I, vec3( 0.0 ) );
-
-}
-
-`;

package/src/shader/shaderLightSampling.js

@@ -1,229 +0,0 @@
-export const shaderLightSampling = /* glsl */`
-
-float getSpotAttenuation( const in float coneCosine, const in float penumbraCosine, const in float angleCosine ) {
-
-	return smoothstep( coneCosine, penumbraCosine, angleCosine );
-
-}
-
-float getDistanceAttenuation( const in float lightDistance, const in float cutoffDistance, const in float decayExponent ) {
-
-	// based upon Frostbite 3 Moving to Physically-based Rendering
-	// page 32, equation 26: E[window1]
-	// https://seblagarde.files.wordpress.com/2015/07/course_notes_moving_frostbite_to_pbr_v32.pdf
-	float distanceFalloff = 1.0 / max( pow( lightDistance, decayExponent ), EPSILON );
-
-	if ( cutoffDistance > 0.0 ) {
-
-		distanceFalloff *= pow2( saturate( 1.0 - pow4( lightDistance / cutoffDistance ) ) );
-
-	}
-
-	return distanceFalloff;
-
-}
-
-float getPhotometricAttenuation( sampler2DArray iesProfiles, int iesProfile, vec3 posToLight, vec3 lightDir, vec3 u, vec3 v ) {
-
-    float cosTheta = dot( posToLight, lightDir );
-    float angle = acos( cosTheta ) * ( 1.0 / PI );
-
-    return texture2D( iesProfiles, vec3( 0.0, angle, iesProfile ) ).r;
-
-}
-
-struct LightSampleRec {
-
-	bool hit;
-	float dist;
-	vec3 direction;
-	float pdf;
-	vec3 emission;
-	int type;
-
-};
-
-LightSampleRec lightsClosestHit( sampler2D lights, uint lightCount, vec3 rayOrigin, vec3 rayDirection ) {
-
-	LightSampleRec lightSampleRec;
-	lightSampleRec.hit = false;
-
-	uint l;
-	for ( l = 0u; l < lightCount; l ++ ) {
-
-		Light light = readLightInfo( lights, l );
-
-		vec3 u = light.u;
-		vec3 v = light.v;
-
-		// check for backface
-		vec3 normal = normalize( cross( u, v ) );
-		if ( dot( normal, rayDirection ) < 0.0 ) {
-			continue;
-		}
-
-		u *= 1.0 / dot( u, u );
-		v *= 1.0 / dot( v, v );
-
-		float dist;
-
-		// MIS / light intersection is not supported for punctual lights.
-		if(
-			( light.type == RECT_AREA_LIGHT_TYPE && intersectsRectangle( light.position, normal, u, v, rayOrigin, rayDirection, dist ) ) ||
-			( light.type == CIRC_AREA_LIGHT_TYPE && intersectsCircle( light.position, normal, u, v, rayOrigin, rayDirection, dist ) )
-		) {
-
-			if ( dist < lightSampleRec.dist || ! lightSampleRec.hit ) {
-
-				float cosTheta = dot( rayDirection, normal );
-
-				lightSampleRec.hit = true;
-				lightSampleRec.dist = dist;
-				lightSampleRec.pdf = ( dist * dist ) / ( light.area * cosTheta );
-				lightSampleRec.emission = light.color * light.intensity;
-				lightSampleRec.direction = rayDirection;
-				lightSampleRec.type = light.type;
-
-			}
-
-		}
-
-	}
-
-	return lightSampleRec;
-
-}
-
-LightSampleRec randomAreaLightSample( Light light, vec3 rayOrigin, vec2 ruv ) {
-
-	LightSampleRec lightSampleRec;
-	lightSampleRec.hit = true;
-	lightSampleRec.type = light.type;
-
-	lightSampleRec.emission = light.color * light.intensity;
-
-	vec3 randomPos;
-	if( light.type == RECT_AREA_LIGHT_TYPE ) {
-
-		// rectangular area light
-		randomPos = light.position + light.u * ( ruv.x - 0.5 ) + light.v * ( ruv.y - 0.5 );
-
-	} else if( light.type == CIRC_AREA_LIGHT_TYPE ) {
-
-		// circular area light
-		float r = 0.5 * sqrt( ruv.x );
-		float theta = ruv.y * 2.0 * PI;
-		float x = r * cos( theta );
-		float y = r * sin( theta );
-
-		randomPos = light.position + light.u * x + light.v * y;
-
-	}
-
-	vec3 toLight = randomPos - rayOrigin;
-	float lightDistSq = dot( toLight, toLight );
-	lightSampleRec.dist = sqrt( lightDistSq );
-
-	vec3 direction = toLight / lightSampleRec.dist;
-	lightSampleRec.direction = direction;
-
-	vec3 lightNormal = normalize( cross( light.u, light.v ) );
-	lightSampleRec.pdf = lightDistSq / ( light.area * dot( direction, lightNormal ) );
-
-	return lightSampleRec;
-
-}
-
-LightSampleRec randomSpotLightSample( Light light, sampler2DArray iesProfiles, vec3 rayOrigin, vec2 ruv ) {
-
-	float radius = light.radius * sqrt( ruv.x );
-	float theta = ruv.y * 2.0 * PI;
-	float x = radius * cos( theta );
-	float y = radius * sin( theta );
-
-	vec3 u = light.u;
-	vec3 v = light.v;
-	vec3 normal = normalize( cross( u, v ) );
-
-	float angle = acos( light.coneCos );
-	float angleTan = tan( angle );
-	float startDistance = light.radius / max( angleTan, EPSILON );
-
-	vec3 randomPos = light.position - normal * startDistance + u * x + v * y;
-	vec3 toLight = randomPos - rayOrigin;
-	float lightDistSq = dot( toLight, toLight );
-	float dist = sqrt( lightDistSq );
-
-	vec3 direction = toLight / max( dist, EPSILON );
-	float cosTheta = dot( direction, normal );
-
-	float spotAttenuation = light.iesProfile != - 1 ?
-		getPhotometricAttenuation( iesProfiles, light.iesProfile, direction, normal, u, v ) :
-		getSpotAttenuation( light.coneCos, light.penumbraCos, cosTheta );
-
-	float distanceAttenuation = getDistanceAttenuation( dist, light.distance, light.decay );
-	LightSampleRec lightSampleRec;
-	lightSampleRec.hit = true;
-	lightSampleRec.type = light.type;
-	lightSampleRec.dist = dist;
-	lightSampleRec.direction = direction;
-	lightSampleRec.emission = light.color * light.intensity * distanceAttenuation * spotAttenuation;
-	lightSampleRec.pdf = 1.0;
-
-	return lightSampleRec;
-
-}
-
-LightSampleRec randomLightSample( sampler2D lights, sampler2DArray iesProfiles, uint lightCount, vec3 rayOrigin, vec3 ruv ) {
-
-	// pick a random light
-	uint l = uint( ruv.x * float( lightCount ) );
-	Light light = readLightInfo( lights, l );
-
-	if ( light.type == SPOT_LIGHT_TYPE ) {
-
-		return randomSpotLightSample( light, iesProfiles, rayOrigin, ruv.yz );
-
-	} else if ( light.type == POINT_LIGHT_TYPE ) {
-
-		vec3 lightRay = light.u - rayOrigin;
-		float lightDist = length( lightRay );
-		float cutoffDistance = light.distance;
-		float distanceFalloff = 1.0 / max( pow( lightDist, light.decay ), 0.01 );
-		if ( cutoffDistance > 0.0 ) {
-
-			distanceFalloff *= pow2( saturate( 1.0 - pow4( lightDist / cutoffDistance ) ) );
-
-		}
-
-		LightSampleRec rec;
-		rec.hit = true;
-		rec.direction = normalize( lightRay );
-		rec.dist = length( lightRay );
-		rec.pdf = 1.0;
-		rec.emission = light.color * light.intensity * distanceFalloff;
-		rec.type = light.type;
-		return rec;
-
-	} else if ( light.type == DIR_LIGHT_TYPE ) {
-
-		LightSampleRec rec;
-		rec.hit = true;
-		rec.dist = 1e10;
-		rec.direction = light.u;
-		rec.pdf = 1.0;
-		rec.emission = light.color * light.intensity;
-		rec.type = light.type;
-
-		return rec;
-
-	} else {
-
-		// sample the light
-		return randomAreaLightSample( light, rayOrigin, ruv.yz );
-
-	}
-
-}
-
-`;