three-gpu-pathtracer 0.0.5 → 0.0.7

package/src/uniforms/MaterialsTexture.js

@@ -1,319 +1,426 @@
-import { DataTexture, RGBAFormat, ClampToEdgeWrapping, FloatType, FrontSide, BackSide, DoubleSide } from 'three';
-
-const MATERIAL_PIXELS = 26;
-const MATERIAL_STRIDE = MATERIAL_PIXELS * 4;
-
-const SIDE_OFFSET = 7 * 4 + 1;
-const MATTE_OFFSET = 7 * 4 + 2;
-const SHADOW_OFFSET = 7 * 4 + 3;
-
-export class MaterialsTexture extends DataTexture {
-
-	constructor() {
-
-		super( new Float32Array( 4 ), 1, 1 );
-
-		this.format = RGBAFormat;
-		this.type = FloatType;
-		this.wrapS = ClampToEdgeWrapping;
-		this.wrapT = ClampToEdgeWrapping;
-		this.generateMipmaps = false;
-		this.threeCompatibilityTransforms = false;
-
-	}
-
-	setCastShadow( materialIndex, cast ) {
-
-		// invert the shadow value so we default to "true" when initializing a material
-		const array = this.image.data;
-		const index = materialIndex * MATERIAL_STRIDE + SHADOW_OFFSET;
-		array[ index ] = ! cast ? 1 : 0;
-
-	}
-
-	getCastShadow( materialIndex ) {
-
-		const array = this.image.data;
-		const index = materialIndex * MATERIAL_STRIDE + SHADOW_OFFSET;
-		return ! Boolean( array[ index ] );
-
-	}
-
-	setSide( materialIndex, side ) {
-
-		const array = this.image.data;
-		const index = materialIndex * MATERIAL_STRIDE + SIDE_OFFSET;
-		switch ( side ) {
-
-		case FrontSide:
-			array[ index ] = 1;
-			break;
-		case BackSide:
-			array[ index ] = - 1;
-			break;
-		case DoubleSide:
-			array[ index ] = 0;
-			break;
-
-		}
-
-	}
-
-	getSide( materialIndex ) {
-
-		const array = this.image.data;
-		const index = materialIndex * MATERIAL_STRIDE + SIDE_OFFSET;
-		switch ( array[ index ] ) {
-
-		case 0:
-			return DoubleSide;
-		case 1:
-			return FrontSide;
-		case - 1:
-			return BackSide;
-
-		}
-
-		return 0;
-
-	}
-
-	setMatte( materialIndex, matte ) {
-
-		const array = this.image.data;
-		const index = materialIndex * MATERIAL_STRIDE + MATTE_OFFSET;
-		array[ index ] = matte ? 1 : 0;
-
-	}
-
-	getMatte( materialIndex ) {
-
-		const array = this.image.data;
-		const index = materialIndex * MATERIAL_STRIDE + MATTE_OFFSET;
-		return Boolean( array[ index ] );
-
-	}
-
-	updateFrom( materials, textures ) {
-
-		function getTexture( material, key, def = - 1 ) {
-
-			return key in material ? textures.indexOf( material[ key ] ) : def;
-
-		}
-
-		function getField( material, key, def ) {
-
-			return key in material ? material[ key ] : def;
-
-		}
-
-		function getUVTransformTexture( material ) {
-
-			// https://github.com/mrdoob/three.js/blob/f3a832e637c98a404c64dae8174625958455e038/src/renderers/webgl/WebGLMaterials.js#L204-L306
-			// https://threejs.org/docs/#api/en/textures/Texture.offset
-			// fallback order of textures to use as a common uv transform
-			return material.map ||
-				material.specularMap ||
-				material.displacementMap ||
-				material.normalMap ||
-				material.bumpMap ||
-				material.roughnessMap ||
-				material.metalnessMap ||
-				material.alphaMap ||
-				material.emissiveMap ||
-				material.clearcoatMap ||
-				material.clearcoatNormalMap ||
-				material.clearcoatRoughnessMap ||
-				material.iridescenceMap ||
-				material.iridescenceThicknessMap ||
-				material.specularIntensityMap ||
-				material.specularColorMap ||
-				material.transmissionMap ||
-				material.thicknessMap ||
-				material.sheenColorMap ||
-				material.sheenRoughnessMap ||
-				null;
-
-		}
-
-		function writeTextureMatrixToArray( material, textureKey, array, offset ) {
-
-			let texture;
-			if ( threeCompatibilityTransforms ) {
-
-				texture = getUVTransformTexture( material );
-
-			} else {
-
-				texture = material[ textureKey ] && material[ textureKey ].isTexture ? material[ textureKey ] : null;
-
-			}
-
-			// check if texture exists
-			if ( texture ) {
-
-				const elements = texture.matrix.elements;
-
-				let i = 0;
-
-				// first row
-				array[ offset + i ++ ] = elements[ 0 ];
-				array[ offset + i ++ ] = elements[ 3 ];
-				array[ offset + i ++ ] = elements[ 6 ];
-				i ++;
-
-				// second row
-				array[ offset + i ++ ] = elements[ 1 ];
-				array[ offset + i ++ ] = elements[ 4 ];
-				array[ offset + i ++ ] = elements[ 7 ];
-				i ++;
-
-			}
-
-			return 8;
-
-		}
-
-		let index = 0;
-		const pixelCount = materials.length * MATERIAL_PIXELS;
-		const dimension = Math.ceil( Math.sqrt( pixelCount ) );
-		const { threeCompatibilityTransforms, image } = this;
-
-		if ( image.width !== dimension ) {
-
-			this.dispose();
-
-			image.data = new Float32Array( dimension * dimension * 4 );
-			image.width = dimension;
-			image.height = dimension;
-
-		}
-
-		const floatArray = image.data;
-
-		// on some devices (Google Pixel 6) the "floatBitsToInt" function does not work correctly so we
-		// can't encode texture ids that way.
-		// const intArray = new Int32Array( floatArray.buffer );
-
-		for ( let i = 0, l = materials.length; i < l; i ++ ) {
-
-			const m = materials[ i ];
-
-			// color
-			floatArray[ index ++ ] = m.color.r;
-			floatArray[ index ++ ] = m.color.g;
-			floatArray[ index ++ ] = m.color.b;
-			floatArray[ index ++ ] = getTexture( m, 'map' );
-
-			// metalness & roughness
-			floatArray[ index ++ ] = getField( m, 'metalness', 0.0 );
-			floatArray[ index ++ ] = textures.indexOf( m.metalnessMap );
-			floatArray[ index ++ ] = getField( m, 'roughness', 0.0 );
-			floatArray[ index ++ ] = textures.indexOf( m.roughnessMap );
-
-			// transmission & emissiveIntensity
-			floatArray[ index ++ ] = getField( m, 'ior', 1.0 );
-			floatArray[ index ++ ] = getField( m, 'transmission', 0.0 );
-			floatArray[ index ++ ] = getTexture( m, 'transmissionMap' );
-			floatArray[ index ++ ] = getField( m, 'emissiveIntensity', 0.0 );
-
-			// emission
-			if ( 'emissive' in m ) {
-
-				floatArray[ index ++ ] = m.emissive.r;
-				floatArray[ index ++ ] = m.emissive.g;
-				floatArray[ index ++ ] = m.emissive.b;
-
-			} else {
-
-				floatArray[ index ++ ] = 0.0;
-				floatArray[ index ++ ] = 0.0;
-				floatArray[ index ++ ] = 0.0;
-
-			}
-
-			floatArray[ index ++ ] = getTexture( m, 'emissiveMap' );
-
-			// normals
-			floatArray[ index ++ ] = getTexture( m, 'normalMap' );
-			if ( 'normalScale' in m ) {
-
-				floatArray[ index ++ ] = m.normalScale.x;
-				floatArray[ index ++ ] = m.normalScale.y;
-
- 			} else {
-
- 				floatArray[ index ++ ] = 1;
- 				floatArray[ index ++ ] = 1;
-
- 			}
-
-			// clearcoat
-			floatArray[ index ++ ] = getField( m, 'clearcoat', 0.0 );
-			floatArray[ index ++ ] = getTexture( m, 'clearcoatMap' );
-
-			// clearcoatRoughness
-			floatArray[ index ++ ] = getField( m, 'clearcoatRoughness', 0.0 );
-			floatArray[ index ++ ] = getTexture( m, 'clearcoatRoughnessMap' );
-
-			// clearcoatNormal
-			floatArray[ index ++ ] = getTexture( m, 'clearcoatNormalMap' );
-
-			if ( 'clearcoatNormalScale' in m ) {
-
-				floatArray[ index ++ ] = m.clearcoatNormalScale.x;
-				floatArray[ index ++ ] = m.clearcoatNormalScale.y;
-
-			} else {
-
-				floatArray[ index ++ ] = 1;
-				floatArray[ index ++ ] = 1;
-
-			}
-
-			// alphaMap
-			floatArray[ index ++ ] = getTexture( m, 'alphaMap' );
-
-			// side & matte
-			floatArray[ index ++ ] = m.opacity;
-			floatArray[ index ++ ] = m.alphaTest;
-			index ++; // side
-			index ++; // matte
-
-			index ++; // shadow
-
-			// map transform
-			index += writeTextureMatrixToArray( m, 'map', floatArray, index );
-
-			// metalnessMap transform
-			index += writeTextureMatrixToArray( m, 'metalnessMap', floatArray, index );
-
-			// roughnessMap transform
-			index += writeTextureMatrixToArray( m, 'roughnessMap', floatArray, index );
-
-			// transmissionMap transform
-			index += writeTextureMatrixToArray( m, 'transmissionMap', floatArray, index );
-
-			// emissiveMap transform
-			index += writeTextureMatrixToArray( m, 'emissiveMap', floatArray, index );
-
-			// normalMap transform
-			index += writeTextureMatrixToArray( m, 'normalMap', floatArray, index );
-
-			// clearcoatMap transform
-			index += writeTextureMatrixToArray( m, 'clearcoatMap', floatArray, index );
-
-			// clearcoatNormalMap transform
-			index += writeTextureMatrixToArray( m, 'clearcoatNormalMap', floatArray, index );
-
-			// clearcoatRoughnessMap transform
-			index += writeTextureMatrixToArray( m, 'clearcoatRoughnessMap', floatArray, index );
-
-		}
-
-		this.needsUpdate = true;
-
-	}
-
-}
+import { DataTexture, RGBAFormat, ClampToEdgeWrapping, FloatType, FrontSide, BackSide, DoubleSide } from 'three';
+
+const MATERIAL_PIXELS = 45;
+const MATERIAL_STRIDE = MATERIAL_PIXELS * 4;
+
+const SIDE_OFFSET = 13 * 4 + 3; // s12.a
+const MATTE_OFFSET = 14 * 4 + 0; // s14.r
+const SHADOW_OFFSET = 14 * 4 + 1; // s14.g
+
+export class MaterialsTexture extends DataTexture {
+
+	constructor() {
+
+		super( new Float32Array( 4 ), 1, 1 );
+
+		this.format = RGBAFormat;
+		this.type = FloatType;
+		this.wrapS = ClampToEdgeWrapping;
+		this.wrapT = ClampToEdgeWrapping;
+		this.generateMipmaps = false;
+		this.threeCompatibilityTransforms = false;
+
+	}
+
+	setCastShadow( materialIndex, cast ) {
+
+		// invert the shadow value so we default to "true" when initializing a material
+		const array = this.image.data;
+		const index = materialIndex * MATERIAL_STRIDE + SHADOW_OFFSET;
+		array[ index ] = ! cast ? 1 : 0;
+
+	}
+
+	getCastShadow( materialIndex ) {
+
+		const array = this.image.data;
+		const index = materialIndex * MATERIAL_STRIDE + SHADOW_OFFSET;
+		return ! Boolean( array[ index ] );
+
+	}
+
+	setSide( materialIndex, side ) {
+
+		const array = this.image.data;
+		const index = materialIndex * MATERIAL_STRIDE + SIDE_OFFSET;
+		switch ( side ) {
+
+		case FrontSide:
+			array[ index ] = 1;
+			break;
+		case BackSide:
+			array[ index ] = - 1;
+			break;
+		case DoubleSide:
+			array[ index ] = 0;
+			break;
+
+		}
+
+	}
+
+	getSide( materialIndex ) {
+
+		const array = this.image.data;
+		const index = materialIndex * MATERIAL_STRIDE + SIDE_OFFSET;
+		switch ( array[ index ] ) {
+
+		case 0:
+			return DoubleSide;
+		case 1:
+			return FrontSide;
+		case - 1:
+			return BackSide;
+
+		}
+
+		return 0;
+
+	}
+
+	setMatte( materialIndex, matte ) {
+
+		const array = this.image.data;
+		const index = materialIndex * MATERIAL_STRIDE + MATTE_OFFSET;
+		array[ index ] = matte ? 1 : 0;
+
+	}
+
+	getMatte( materialIndex ) {
+
+		const array = this.image.data;
+		const index = materialIndex * MATERIAL_STRIDE + MATTE_OFFSET;
+		return Boolean( array[ index ] );
+
+	}
+
+	updateFrom( materials, textures ) {
+
+		function getTexture( material, key, def = - 1 ) {
+
+			return key in material ? textures.indexOf( material[ key ] ) : def;
+
+		}
+
+		function getField( material, key, def ) {
+
+			return key in material ? material[ key ] : def;
+
+		}
+
+		function getUVTransformTexture( material ) {
+
+			// https://github.com/mrdoob/three.js/blob/f3a832e637c98a404c64dae8174625958455e038/src/renderers/webgl/WebGLMaterials.js#L204-L306
+			// https://threejs.org/docs/#api/en/textures/Texture.offset
+			// fallback order of textures to use as a common uv transform
+			return material.map ||
+				material.specularMap ||
+				material.displacementMap ||
+				material.normalMap ||
+				material.bumpMap ||
+				material.roughnessMap ||
+				material.metalnessMap ||
+				material.alphaMap ||
+				material.emissiveMap ||
+				material.clearcoatMap ||
+				material.clearcoatNormalMap ||
+				material.clearcoatRoughnessMap ||
+				material.iridescenceMap ||
+				material.iridescenceThicknessMap ||
+				material.specularIntensityMap ||
+				material.specularColorMap ||
+				material.transmissionMap ||
+				material.thicknessMap ||
+				material.sheenColorMap ||
+				material.sheenRoughnessMap ||
+				null;
+
+		}
+
+		function writeTextureMatrixToArray( material, textureKey, array, offset ) {
+
+			let texture;
+			if ( threeCompatibilityTransforms ) {
+
+				texture = getUVTransformTexture( material );
+
+			} else {
+
+				texture = material[ textureKey ] && material[ textureKey ].isTexture ? material[ textureKey ] : null;
+
+			}
+
+			// check if texture exists
+			if ( texture ) {
+
+				const elements = texture.matrix.elements;
+
+				let i = 0;
+
+				// first row
+				array[ offset + i ++ ] = elements[ 0 ];
+				array[ offset + i ++ ] = elements[ 3 ];
+				array[ offset + i ++ ] = elements[ 6 ];
+				i ++;
+
+				// second row
+				array[ offset + i ++ ] = elements[ 1 ];
+				array[ offset + i ++ ] = elements[ 4 ];
+				array[ offset + i ++ ] = elements[ 7 ];
+				i ++;
+
+			}
+
+			return 8;
+
+		}
+
+		let index = 0;
+		const pixelCount = materials.length * MATERIAL_PIXELS;
+		const dimension = Math.ceil( Math.sqrt( pixelCount ) );
+		const { threeCompatibilityTransforms, image } = this;
+
+		if ( image.width !== dimension ) {
+
+			this.dispose();
+
+			image.data = new Float32Array( dimension * dimension * 4 );
+			image.width = dimension;
+			image.height = dimension;
+
+		}
+
+		const floatArray = image.data;
+
+		// on some devices (Google Pixel 6) the "floatBitsToInt" function does not work correctly so we
+		// can't encode texture ids that way.
+		// const intArray = new Int32Array( floatArray.buffer );
+
+		for ( let i = 0, l = materials.length; i < l; i ++ ) {
+
+			const m = materials[ i ];
+
+			// sample 0
+			// color
+			floatArray[ index ++ ] = m.color.r;
+			floatArray[ index ++ ] = m.color.g;
+			floatArray[ index ++ ] = m.color.b;
+			floatArray[ index ++ ] = getTexture( m, 'map' );
+
+			// sample 1
+			// metalness & roughness
+			floatArray[ index ++ ] = getField( m, 'metalness', 0.0 );
+			floatArray[ index ++ ] = textures.indexOf( m.metalnessMap );
+			floatArray[ index ++ ] = getField( m, 'roughness', 0.0 );
+			floatArray[ index ++ ] = textures.indexOf( m.roughnessMap );
+
+			// sample 2
+			// transmission & emissiveIntensity
+			// three.js assumes a default f0 of 0.04 if no ior is provided which equates to an ior of 1.5
+			floatArray[ index ++ ] = getField( m, 'ior', 1.5 );
+			floatArray[ index ++ ] = getField( m, 'transmission', 0.0 );
+			floatArray[ index ++ ] = getTexture( m, 'transmissionMap' );
+			floatArray[ index ++ ] = getField( m, 'emissiveIntensity', 0.0 );
+
+			// sample 3
+			// emission
+			if ( 'emissive' in m ) {
+
+				floatArray[ index ++ ] = m.emissive.r;
+				floatArray[ index ++ ] = m.emissive.g;
+				floatArray[ index ++ ] = m.emissive.b;
+
+			} else {
+
+				floatArray[ index ++ ] = 0.0;
+				floatArray[ index ++ ] = 0.0;
+				floatArray[ index ++ ] = 0.0;
+
+			}
+
+			floatArray[ index ++ ] = getTexture( m, 'emissiveMap' );
+
+			// sample 4
+			// normals
+			floatArray[ index ++ ] = getTexture( m, 'normalMap' );
+			if ( 'normalScale' in m ) {
+
+				floatArray[ index ++ ] = m.normalScale.x;
+				floatArray[ index ++ ] = m.normalScale.y;
+
+ 			} else {
+
+ 				floatArray[ index ++ ] = 1;
+ 				floatArray[ index ++ ] = 1;
+
+ 			}
+
+			// clearcoat
+			floatArray[ index ++ ] = getField( m, 'clearcoat', 0.0 );
+			floatArray[ index ++ ] = getTexture( m, 'clearcoatMap' ); // sample 5
+
+			floatArray[ index ++ ] = getField( m, 'clearcoatRoughness', 0.0 );
+			floatArray[ index ++ ] = getTexture( m, 'clearcoatRoughnessMap' );
+
+			floatArray[ index ++ ] = getTexture( m, 'clearcoatNormalMap' );
+
+			// sample 6
+			if ( 'clearcoatNormalScale' in m ) {
+
+				floatArray[ index ++ ] = m.clearcoatNormalScale.x;
+				floatArray[ index ++ ] = m.clearcoatNormalScale.y;
+
+			} else {
+
+				floatArray[ index ++ ] = 1;
+				floatArray[ index ++ ] = 1;
+
+			}
+
+			index ++;
+			index ++;
+
+			// sample 7
+			// sheen
+			if ( 'sheenColor' in m ) {
+
+				floatArray[ index ++ ] = m.sheenColor.r;
+				floatArray[ index ++ ] = m.sheenColor.g;
+				floatArray[ index ++ ] = m.sheenColor.b;
+
+			} else {
+
+				floatArray[ index ++ ] = 0.0;
+				floatArray[ index ++ ] = 0.0;
+				floatArray[ index ++ ] = 0.0;
+
+			}
+
+			floatArray[ index ++ ] = getTexture( m, 'sheenColorMap' );
+
+			// sample 8
+			floatArray[ index ++ ] = getField( m, 'sheenRoughness', 0.0 );
+			floatArray[ index ++ ] = getTexture( m, 'sheenRoughnessMap' );
+
+			// iridescence
+			floatArray[ index ++ ] = getTexture( m, 'iridescenceMap' );
+			floatArray[ index ++ ] = getTexture( m, 'iridescenceThicknessMap' );
+
+			floatArray[ index ++ ] = getField( m, 'iridescence', 0.0 ); // sample 9
+			floatArray[ index ++ ] = getField( m, 'iridescenceIOR', 1.3 );
+
+			const iridescenceThicknessRange = getField( m, 'iridescenceThicknessRange', [ 100, 400 ] );
+			floatArray[ index ++ ] = iridescenceThicknessRange[ 0 ];
+			floatArray[ index ++ ] = iridescenceThicknessRange[ 1 ];
+
+			// sample 10
+			// specular color
+			if ( 'specularColor' in m ) {
+
+				floatArray[ index ++ ] = m.specularColor.r;
+				floatArray[ index ++ ] = m.specularColor.g;
+				floatArray[ index ++ ] = m.specularColor.b;
+
+			} else {
+
+				floatArray[ index ++ ] = 1.0;
+				floatArray[ index ++ ] = 1.0;
+				floatArray[ index ++ ] = 1.0;
+
+			}
+
+			floatArray[ index ++ ] = getTexture( m, 'specularColorMap' );
+
+			// sample 11
+			// specular intensity
+			floatArray[ index ++ ] = getField( m, 'specularIntensity', 1.0 );
+			floatArray[ index ++ ] = getTexture( m, 'specularIntensityMap' );
+
+			// thickness
+			floatArray[ index ++ ] = getField( m, 'thickness', 0.0 ) === 0.0 && getField( m, 'attenuationDistance', Infinity ) === Infinity;
+			index ++;
+
+			// sample 12
+			if ( 'attenuationColor' in m ) {
+
+				floatArray[ index ++ ] = m.attenuationColor.r;
+				floatArray[ index ++ ] = m.attenuationColor.g;
+				floatArray[ index ++ ] = m.attenuationColor.b;
+
+			} else {
+
+				floatArray[ index ++ ] = 1.0;
+				floatArray[ index ++ ] = 1.0;
+				floatArray[ index ++ ] = 1.0;
+
+			}
+
+			floatArray[ index ++ ] = getField( m, 'attenuationDistance', Infinity );
+
+			// sample 13
+			// alphaMap
+			floatArray[ index ++ ] = getTexture( m, 'alphaMap' );
+
+			// side & matte
+			floatArray[ index ++ ] = m.opacity;
+			floatArray[ index ++ ] = m.alphaTest;
+			index ++; // side
+
+			// sample 14
+			index ++; // matte
+			index ++; // shadow
+			floatArray[ index ++ ] = Number( m.vertexColors ); // vertexColors
+			floatArray[ index ++ ] = Number( m.transparent ); // transparent
+
+			// map transform 15
+			index += writeTextureMatrixToArray( m, 'map', floatArray, index );
+
+			// metalnessMap transform 17
+			index += writeTextureMatrixToArray( m, 'metalnessMap', floatArray, index );
+
+			// roughnessMap transform 19
+			index += writeTextureMatrixToArray( m, 'roughnessMap', floatArray, index );
+
+			// transmissionMap transform 21
+			index += writeTextureMatrixToArray( m, 'transmissionMap', floatArray, index );
+
+			// emissiveMap transform 22
+			index += writeTextureMatrixToArray( m, 'emissiveMap', floatArray, index );
+
+			// normalMap transform 25
+			index += writeTextureMatrixToArray( m, 'normalMap', floatArray, index );
+
+			// clearcoatMap transform 27
+			index += writeTextureMatrixToArray( m, 'clearcoatMap', floatArray, index );
+
+			// clearcoatNormalMap transform 29
+			index += writeTextureMatrixToArray( m, 'clearcoatNormalMap', floatArray, index );
+
+			// clearcoatRoughnessMap transform 31
+			index += writeTextureMatrixToArray( m, 'clearcoatRoughnessMap', floatArray, index );
+
+			// sheenColorMap transform 33
+			index += writeTextureMatrixToArray( m, 'sheenColorMap', floatArray, index );
+
+			// sheenRoughnessMap transform 35
+			index += writeTextureMatrixToArray( m, 'sheenRoughnessMap', floatArray, index );
+
+			// iridescenceMap transform 37
+			index += writeTextureMatrixToArray( m, 'iridescenceMap', floatArray, index );
+
+			// iridescenceThicknessMap transform 39
+			index += writeTextureMatrixToArray( m, 'iridescenceThicknessMap', floatArray, index );
+
+			// specularColorMap transform 41
+			index += writeTextureMatrixToArray( m, 'specularColorMap', floatArray, index );
+
+			// specularIntensityMap transform 43
+			index += writeTextureMatrixToArray( m, 'specularIntensityMap', floatArray, index );
+
+		}
+
+		this.needsUpdate = true;
+
+	}
+
+}