webgl_nearestneighbour.html 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - nearest neighbour</title>
  5. <meta charset="utf-8">
  6. <meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
  7. <style>
  8. html, body {
  9. width: 100%;
  10. height: 100%;
  11. }
  12. body {
  13. background-color: #ffffff;
  14. margin: 0;
  15. overflow: hidden;
  16. font-family: arial;
  17. }
  18. #info {
  19. text-align: center;
  20. padding: 5px;
  21. position: absolute;
  22. width: 100%;
  23. color: white;
  24. }
  25. </style>
  26. </head>
  27. <body>
  28. <div id="info"><a href="http://threejs.org" target="_blank" rel="noopener">three.js</a> webgl - typed arrays - nearest neighbour for 500,000 sprites</div>
  29. <script src="../build/three.js"></script>
  30. <script src="js/TypedArrayUtils.js"></script>
  31. <script src="js/controls/FirstPersonControls.js"></script>
  32. <script type="x-shader/x-vertex" id="vertexshader">
  33. //uniform float zoom;
  34. attribute float alpha;
  35. varying float vAlpha;
  36. void main() {
  37. vAlpha = 1.0 - alpha;
  38. vec4 mvPosition = modelViewMatrix * vec4( position, 1.0 );
  39. gl_PointSize = 4.0 * ( 300.0 / -mvPosition.z );
  40. gl_Position = projectionMatrix * mvPosition;
  41. }
  42. </script>
  43. <script type="x-shader/x-fragment" id="fragmentshader">
  44. uniform sampler2D tex1;
  45. varying float vAlpha;
  46. void main() {
  47. gl_FragColor = texture2D( tex1, gl_PointCoord );
  48. gl_FragColor.r = ( 1.0 - gl_FragColor.r ) * vAlpha + gl_FragColor.r;
  49. }
  50. </script>
  51. <script>
  52. var camera, scene, renderer;
  53. var controls;
  54. var amountOfParticles = 500000, maxDistance = Math.pow( 120, 2 );
  55. var positions, alphas, particles, _particleGeom;
  56. var kdtree;
  57. var clock = new THREE.Clock();
  58. var blocker = document.getElementById( 'blocker' );
  59. var instructions = document.getElementById( 'instructions' );
  60. function init() {
  61. camera = new THREE.PerspectiveCamera( 75, window.innerWidth / window.innerHeight, 1, 1000000 );
  62. scene = new THREE.Scene();
  63. controls = new THREE.FirstPersonControls( camera );
  64. controls.movementSpeed = 100;
  65. controls.lookSpeed = 0.1;
  66. // add a skybox background
  67. var cubeTextureLoader = new THREE.CubeTextureLoader();
  68. cubeTextureLoader.setPath( 'textures/cube/skybox/' );
  69. var cubeTexture = cubeTextureLoader.load( [
  70. 'px.jpg', 'nx.jpg',
  71. 'py.jpg', 'ny.jpg',
  72. 'pz.jpg', 'nz.jpg',
  73. ] );
  74. scene.background = cubeTexture;
  75. //
  76. renderer = new THREE.WebGLRenderer(); // Detector.webgl? new THREE.WebGLRenderer(): new THREE.CanvasRenderer()
  77. renderer.setPixelRatio( window.devicePixelRatio );
  78. renderer.setSize( window.innerWidth, window.innerHeight );
  79. document.body.appendChild( renderer.domElement );
  80. // create the custom shader
  81. var textureLoader = new THREE.TextureLoader();
  82. var imagePreviewTexture = textureLoader.load( 'textures/crate.gif' );
  83. imagePreviewTexture.minFilter = THREE.LinearMipMapLinearFilter;
  84. imagePreviewTexture.magFilter = THREE.LinearFilter;
  85. var pointShaderMaterial = new THREE.ShaderMaterial( {
  86. uniforms: {
  87. tex1: { value: imagePreviewTexture },
  88. zoom: { value: 9.0 }
  89. },
  90. vertexShader: document.getElementById( 'vertexshader' ).textContent,
  91. fragmentShader: document.getElementById( 'fragmentshader' ).textContent,
  92. transparent: true
  93. } );
  94. //create particles with buffer geometry
  95. var distanceFunction = function ( a, b ) {
  96. return Math.pow( a[ 0 ] - b [0 ], 2 ) + Math.pow( a[ 1 ] - b[ 1 ], 2 ) + Math.pow( a[ 2 ] - b[ 2 ], 2 );
  97. };
  98. positions = new Float32Array( amountOfParticles * 3 );
  99. alphas = new Float32Array( amountOfParticles );
  100. _particleGeom = new THREE.BufferGeometry();
  101. _particleGeom.addAttribute( 'position', new THREE.BufferAttribute( positions, 3 ) );
  102. _particleGeom.addAttribute( 'alpha', new THREE.BufferAttribute( alphas, 1 ) );
  103. particles = new THREE.Points( _particleGeom, pointShaderMaterial );
  104. for ( var x = 0; x < amountOfParticles; x ++ ) {
  105. positions[ x * 3 + 0 ] = Math.random() * 1000;
  106. positions[ x * 3 + 1 ] = Math.random() * 1000;
  107. positions[ x * 3 + 2 ] = Math.random() * 1000;
  108. alphas[ x ] = 1.0;
  109. }
  110. var measureStart = new Date().getTime();
  111. // creating the kdtree takes a lot of time to execute, in turn the nearest neighbour search will be much faster
  112. kdtree = new THREE.TypedArrayUtils.Kdtree( positions, distanceFunction, 3 );
  113. console.log( 'TIME building kdtree', new Date().getTime() - measureStart );
  114. // display particles after the kd-tree was generated and the sorting of the positions-array is done
  115. scene.add( particles );
  116. window.addEventListener( 'resize', onWindowResize, false );
  117. }
  118. function onWindowResize() {
  119. camera.aspect = window.innerWidth / window.innerHeight;
  120. camera.updateProjectionMatrix();
  121. renderer.setSize( window.innerWidth, window.innerHeight );
  122. controls.handleResize();
  123. }
  124. function animate() {
  125. requestAnimationFrame( animate );
  126. //
  127. displayNearest( camera.position );
  128. controls.update( clock.getDelta() );
  129. renderer.render( scene, camera );
  130. }
  131. function displayNearest( position ) {
  132. // take the nearest 200 around him. distance^2 'cause we use the manhattan distance and no square is applied in the distance function
  133. var imagePositionsInRange = kdtree.nearest( [ position.x, position.y, position.z ], 100, maxDistance );
  134. // We combine the nearest neighbour with a view frustum. Doesn't make sense if we change the sprites not in our view... well maybe it does. Whatever you want.
  135. var _frustum = new THREE.Frustum();
  136. var _projScreenMatrix = new THREE.Matrix4();
  137. _projScreenMatrix.multiplyMatrices( camera.projectionMatrix, camera.matrixWorldInverse );
  138. _frustum.setFromMatrix( _projScreenMatrix );
  139. for ( var i = 0, il = imagePositionsInRange.length; i < il; i ++ ) {
  140. var object = imagePositionsInRange[ i ];
  141. var objectPoint = new THREE.Vector3().fromArray( object[ 0 ].obj );
  142. if ( _frustum.containsPoint( objectPoint ) ) {
  143. var objectIndex = object[ 0 ].pos;
  144. // set the alpha according to distance
  145. alphas[ objectIndex ] = 1.0 / maxDistance * object[ 1 ];
  146. // update the attribute
  147. _particleGeom.attributes.alpha.needsUpdate = true;
  148. }
  149. }
  150. }
  151. init();
  152. animate();
  153. </script>
  154. </body>
  155. </html>