examples/webgl_custom_attributes_points2.html

<!DOCTYPE html>
<html lang="en">
	<head>
		<title>three.js webgl - custom attributes [particles][billboards]</title>
		<meta charset="utf-8">
		<meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
		<link type="text/css" rel="stylesheet" href="main.css">
	</head>

	<body>
		<div id="info"><a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> - custom attributes example - particles - billboards</div>
		<div id="container"></div>

		<script type="x-shader/x-vertex" id="vertexshader">

			attribute float size;
			attribute vec3 ca;

			varying vec3 vColor;

			void main() {

				vColor = ca;

				vec4 mvPosition = modelViewMatrix * vec4( position, 1.0 );

				gl_PointSize = size * ( 300.0 / -mvPosition.z );

				gl_Position = projectionMatrix * mvPosition;

			}

		</script>

		<script type="x-shader/x-fragment" id="fragmentshader">

			uniform vec3 color;
			uniform sampler2D pointTexture;

			varying vec3 vColor;

			void main() {

				vec4 color = vec4( color * vColor, 1.0 ) * texture2D( pointTexture, gl_PointCoord );

				gl_FragColor = color;

			}

		</script>

		<!-- Import maps polyfill -->
		<!-- Remove this when import maps will be widely supported -->
		<script async src="https://unpkg.com/es-module-shims@1.3.6/dist/es-module-shims.js"></script>

		<script type="importmap">
			{
				"imports": {
					"three": "../build/three.module.js",
					"three/addons/": "./jsm/"
				}
			}
		</script>

		<script type="module">

			import * as THREE from 'three';

			import Stats from 'three/addons/libs/stats.module.js';

			import * as BufferGeometryUtils from 'three/addons/utils/BufferGeometryUtils.js';

			let renderer, scene, camera, stats;
			let sphere, length1;

			const WIDTH = window.innerWidth;
			const HEIGHT = window.innerHeight;

			init();
			animate();

			function init() {

				camera = new THREE.PerspectiveCamera( 45, WIDTH / HEIGHT, 1, 10000 );
				camera.position.z = 300;

				scene = new THREE.Scene();

				const radius = 100, segments = 68, rings = 38;

				let sphereGeometry = new THREE.SphereGeometry( radius, segments, rings );
				let boxGeometry = new THREE.BoxGeometry( 0.8 * radius, 0.8 * radius, 0.8 * radius, 10, 10, 10 );

				// if normal and uv attributes are not removed, mergeVertices() can't consolidate identical vertices with different normal/uv data

				sphereGeometry.deleteAttribute( 'normal' );
				sphereGeometry.deleteAttribute( 'uv' );

				boxGeometry.deleteAttribute( 'normal' );
				boxGeometry.deleteAttribute( 'uv' );

				sphereGeometry = BufferGeometryUtils.mergeVertices( sphereGeometry );
				boxGeometry = BufferGeometryUtils.mergeVertices( boxGeometry );

				const combinedGeometry = BufferGeometryUtils.mergeBufferGeometries( [ sphereGeometry, boxGeometry ] );
				const positionAttribute = combinedGeometry.getAttribute( 'position' );

				const colors = [];
				const sizes = [];

				const color = new THREE.Color();
				const vertex = new THREE.Vector3();

				length1 = sphereGeometry.getAttribute( 'position' ).count;

				for ( let i = 0, l = positionAttribute.count; i < l; i ++ ) {

					vertex.fromBufferAttribute( positionAttribute, i );

					if ( i < length1 ) {

						color.setHSL( 0.01 + 0.1 * ( i / length1 ), 0.99, ( vertex.y + radius ) / ( 4 * radius ) );

					} else {

						color.setHSL( 0.6, 0.75, 0.25 + vertex.y / ( 2 * radius ) );

					}

					color.toArray( colors, i * 3 );

					sizes[ i ] = i < length1 ? 10 : 40;

				}

				const geometry = new THREE.BufferGeometry();
				geometry.setAttribute( 'position', positionAttribute );
				geometry.setAttribute( 'size', new THREE.Float32BufferAttribute( sizes, 1 ) );
				geometry.setAttribute( 'ca', new THREE.Float32BufferAttribute( colors, 3 ) );

				//

				const texture = new THREE.TextureLoader().load( 'textures/sprites/disc.png' );
				texture.wrapS = THREE.RepeatWrapping;
				texture.wrapT = THREE.RepeatWrapping;

				const material = new THREE.ShaderMaterial( {

					uniforms: {
						color: { value: new THREE.Color( 0xffffff ) },
						pointTexture: { value: texture }
					},
					vertexShader: document.getElementById( 'vertexshader' ).textContent,
					fragmentShader: document.getElementById( 'fragmentshader' ).textContent,
					transparent: true

				} );

				//

				sphere = new THREE.Points( geometry, material );
				scene.add( sphere );

				//

				renderer = new THREE.WebGLRenderer();
				renderer.setPixelRatio( window.devicePixelRatio );
				renderer.setSize( WIDTH, HEIGHT );

				const container = document.getElementById( 'container' );
				container.appendChild( renderer.domElement );

				stats = new Stats();
				container.appendChild( stats.dom );

				//

				window.addEventListener( 'resize', onWindowResize );

			}

			function onWindowResize() {

				camera.aspect = window.innerWidth / window.innerHeight;
				camera.updateProjectionMatrix();

				renderer.setSize( window.innerWidth, window.innerHeight );

			}

			function sortPoints() {

				const vector = new THREE.Vector3();

				// Model View Projection matrix

				const matrix = new THREE.Matrix4();
				matrix.multiplyMatrices( camera.projectionMatrix, camera.matrixWorldInverse );
				matrix.multiply( sphere.matrixWorld );

				//

				const geometry = sphere.geometry;

				let index = geometry.getIndex();
				const positions = geometry.getAttribute( 'position' ).array;
				const length = positions.length / 3;

				if ( index === null ) {

					const array = new Uint16Array( length );

					for ( let i = 0; i < length; i ++ ) {

						array[ i ] = i;

					}

					index = new THREE.BufferAttribute( array, 1 );

					geometry.setIndex( index );

				}

				const sortArray = [];

				for ( let i = 0; i < length; i ++ ) {

					vector.fromArray( positions, i * 3 );
					vector.applyMatrix4( matrix );

					sortArray.push( [ vector.z, i ] );

				}

				function numericalSort( a, b ) {

					return b[ 0 ] - a[ 0 ];

				}

				sortArray.sort( numericalSort );

				const indices = index.array;

				for ( let i = 0; i < length; i ++ ) {

					indices[ i ] = sortArray[ i ][ 1 ];

				}

				geometry.index.needsUpdate = true;

			}

			function animate() {

				requestAnimationFrame( animate );

				render();
				stats.update();

			}

			function render() {

				const time = Date.now() * 0.005;

				sphere.rotation.y = 0.02 * time;
				sphere.rotation.z = 0.02 * time;

				const geometry = sphere.geometry;
				const attributes = geometry.attributes;

				for ( let i = 0; i < attributes.size.array.length; i ++ ) {

					if ( i < length1 ) {

						attributes.size.array[ i ] = 16 + 12 * Math.sin( 0.1 * i + time );

					}

				}

				attributes.size.needsUpdate = true;

				sortPoints();

				renderer.render( scene, camera );

			}

		</script>

</body>

</html>