webgl_interactive_instances_gpu.html 23 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - interactive instances (gpu)</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. body {
  9. font-family: Monospace;
  10. background-color: #f0f0f0;
  11. margin: 0px;
  12. overflow: hidden;
  13. }
  14. .info {
  15. position: absolute;
  16. background-color: black;
  17. opacity: 0.8;
  18. color: white;
  19. text-align: center;
  20. top: 0px;
  21. width: 100%;
  22. }
  23. .info a {
  24. color: #00ffff;
  25. }
  26. #notSupported {
  27. width: 50%;
  28. margin: auto;
  29. border: 2px red solid;
  30. margin-top: 20px;
  31. padding: 10px;
  32. }
  33. </style>
  34. </head>
  35. <body>
  36. <div class="info">
  37. <a href="http://threejs.org" target="_blank">three.js</a> webgl - gpu picking of geometry instances
  38. <div id="notSupported" style="display:none">Sorry your graphics card + browser does not support hardware instancing</div>
  39. <br/><br/>
  40. <div>This demo compares different methods of constructing and rendering many instances of a single geometry.</div>
  41. <br/>
  42. <div>
  43. <div style="display:inline-block;">
  44. <span>number of<br/>geometry instances</span>
  45. <br/>
  46. <select id="instanceCount">
  47. <option>100</option>
  48. <option>500</option>
  49. <option selected>1000</option>
  50. <option>2000</option>
  51. <option>3000</option>
  52. <option>5000</option>
  53. <option>10000</option>
  54. <option>20000</option>
  55. <option>30000</option>
  56. <option>50000</option>
  57. <option>100000</option>
  58. </select>
  59. </div>
  60. &nbsp;&nbsp;&nbsp;
  61. <div style="display:inline-block;">
  62. <span>method of<br/>construction/rendering</span>
  63. <br/>
  64. <select id="method">
  65. <option>instanced</option>
  66. <option>merged</option>
  67. <option selected>singleMaterial</option>
  68. <option>multiMaterial</option>
  69. </select>
  70. </div>
  71. &nbsp;&nbsp;&nbsp;
  72. <div style="display:inline-block;">
  73. <span>render continuously<br/>(to get fps reading)</span>
  74. <br/>
  75. <input id="animate" type="checkbox" />
  76. </div>
  77. &nbsp;&nbsp;&nbsp;
  78. <div style="display:inline-block;">
  79. <span>use override material<br/>(only effects singleMaterial method)</span>
  80. <br/>
  81. <input id="override" type="checkbox" checked/>
  82. </div>
  83. &nbsp;&nbsp;&nbsp;
  84. <div style="display:inline-block;">
  85. <span>construct anew<br/>(to get additional timings)</span>
  86. <br/>
  87. <button id="construct" type="button">do it</button>
  88. </div>
  89. </div>
  90. <br/>
  91. <div>
  92. <span>Materials: #<span id="materialCount"></span></span>
  93. &nbsp;&nbsp;&nbsp;
  94. <span>Objects: #<span id="objectCount"></span></span>
  95. &nbsp;&nbsp;&nbsp;
  96. <span>Drawcalls: #<span id="drawcalls"></span></span>
  97. &nbsp;&nbsp;&nbsp;
  98. <span>Construction time: <span id="initTime"></span>&nbsp;ms</span>
  99. &nbsp;&nbsp;&nbsp;
  100. </div>
  101. </div>
  102. <div id="container"></div>
  103. <script src="../build/three.js"></script>
  104. <script src="js/controls/TrackballControls.js"></script>
  105. <script src="js/libs/stats.min.js"></script>
  106. <script id="vertMerged" type="x-shader/x-vertex">
  107. #define SHADER_NAME vertMerged
  108. precision highp float;
  109. uniform mat4 modelViewMatrix;
  110. uniform mat4 projectionMatrix;
  111. attribute vec3 position;
  112. #ifdef PICKING
  113. attribute vec3 pickingColor;
  114. #else
  115. attribute vec3 color;
  116. varying vec3 vPosition;
  117. #endif
  118. varying vec3 vColor;
  119. void main() {
  120. vec3 positionEye = ( modelViewMatrix * vec4( position, 1.0 ) ).xyz;
  121. #ifdef PICKING
  122. vColor = pickingColor;
  123. #else
  124. vColor = color;
  125. vPosition = positionEye;
  126. #endif
  127. gl_Position = projectionMatrix * vec4( positionEye, 1.0 );
  128. }
  129. </script>
  130. <script id="fragMerged" type="x-shader/x-fragment">
  131. #define SHADER_NAME fragMerged
  132. #extension GL_OES_standard_derivatives : enable
  133. precision highp float;
  134. varying vec3 vColor;
  135. #ifndef PICKING
  136. varying vec3 vPosition;
  137. #endif
  138. void main() {
  139. #ifdef PICKING
  140. gl_FragColor = vec4( vColor, 1.0 );
  141. #else
  142. vec3 fdx = dFdx( vPosition );
  143. vec3 fdy = dFdy( vPosition );
  144. vec3 normal = normalize( cross( fdx, fdy ) );
  145. float diffuse = dot( normal, vec3( 0.0, 0.0, 1.0 ) );
  146. gl_FragColor = vec4( diffuse * vColor, 1.0 );
  147. #endif
  148. }
  149. </script>
  150. <script id="vertInstanced" type="x-shader/x-vertex">
  151. #define SHADER_NAME vertInstanced
  152. precision highp float;
  153. uniform mat4 modelViewMatrix;
  154. uniform mat4 projectionMatrix;
  155. attribute vec3 position;
  156. attribute vec3 mcol0;
  157. attribute vec3 mcol1;
  158. attribute vec3 mcol2;
  159. attribute vec3 mcol3;
  160. #ifdef PICKING
  161. attribute vec3 pickingColor;
  162. #else
  163. attribute vec3 color;
  164. varying vec3 vPosition;
  165. #endif
  166. varying vec3 vColor;
  167. void main() {
  168. mat4 matrix = mat4(
  169. vec4( mcol0, 0 ),
  170. vec4( mcol1, 0 ),
  171. vec4( mcol2, 0 ),
  172. vec4( mcol3, 1 )
  173. );
  174. vec3 positionEye = ( modelViewMatrix * matrix * vec4( position, 1.0 ) ).xyz;
  175. #ifdef PICKING
  176. vColor = pickingColor;
  177. #else
  178. vColor = color;
  179. vPosition = positionEye;
  180. #endif
  181. gl_Position = projectionMatrix * vec4( positionEye, 1.0 );
  182. }
  183. </script>
  184. <script id="fragInstanced" type="x-shader/x-fragment">
  185. #define SHADER_NAME fragInstanced
  186. #extension GL_OES_standard_derivatives : enable
  187. precision highp float;
  188. varying vec3 vColor;
  189. #ifndef PICKING
  190. varying vec3 vPosition;
  191. #endif
  192. void main() {
  193. #ifdef PICKING
  194. gl_FragColor = vec4( vColor, 1.0 );
  195. #else
  196. vec3 fdx = dFdx( vPosition );
  197. vec3 fdy = dFdy( vPosition );
  198. vec3 normal = normalize( cross( fdx, fdy ) );
  199. float diffuse = dot( normal, vec3( 0.0, 0.0, 1.0 ) );
  200. gl_FragColor = vec4( diffuse * vColor, 1.0 );
  201. #endif
  202. }
  203. </script>
  204. <script id="vertMaterial" type="x-shader/x-vertex">
  205. #define SHADER_NAME vertMaterial
  206. precision highp float;
  207. uniform mat4 modelViewMatrix;
  208. uniform mat4 projectionMatrix;
  209. attribute vec3 position;
  210. #ifndef PICKING
  211. varying vec3 vPosition;
  212. #endif
  213. void main() {
  214. vec3 positionEye = ( modelViewMatrix * vec4( position, 1.0 ) ).xyz;
  215. #ifndef PICKING
  216. vPosition = positionEye;
  217. #endif
  218. gl_Position = projectionMatrix * vec4( positionEye, 1.0 );
  219. }
  220. </script>
  221. <script id="fragMaterial" type="x-shader/x-fragment">
  222. #define SHADER_NAME fragMaterial
  223. #extension GL_OES_standard_derivatives : enable
  224. precision highp float;
  225. #ifdef PICKING
  226. uniform vec3 pickingColor;
  227. #else
  228. uniform vec3 color;
  229. varying vec3 vPosition;
  230. #endif
  231. void main() {
  232. #ifdef PICKING
  233. gl_FragColor = vec4( pickingColor, 1.0 );
  234. #else
  235. vec3 fdx = dFdx( vPosition );
  236. vec3 fdy = dFdy( vPosition );
  237. vec3 normal = normalize( cross( fdx, fdy ) );
  238. float diffuse = dot( normal, vec3( 0.0, 0.0, 1.0 ) );
  239. gl_FragColor = vec4( diffuse * color, 1.0 );
  240. #endif
  241. }
  242. </script>
  243. <script>
  244. var container, stats;
  245. var camera, controls, scene, renderer;
  246. var pickingData, pickingRenderTarget, pickingScene;
  247. var useOverrideMaterial = true;
  248. var singleMaterial, singlePickingMaterial;
  249. var highlightBox;
  250. var materialList = [];
  251. var geometryList = [];
  252. var objectCount = 0;
  253. var geometrySize;
  254. var mouse = new THREE.Vector2();
  255. var scale = 1.03;
  256. var loader = new THREE.JSONLoader();
  257. //create buffer for reading a single pixel
  258. var pixelBuffer = new Uint8Array( 4 );
  259. // gui
  260. var instanceCount, method, doAnimate;
  261. //
  262. gui();
  263. init();
  264. initMesh();
  265. if ( doAnimate ) animate();
  266. //
  267. function gui() {
  268. var instanceCountElm = document.getElementById( 'instanceCount' );
  269. instanceCount = parseInt( instanceCountElm.value );
  270. instanceCountElm.addEventListener( "change", function() {
  271. instanceCount = parseInt( instanceCountElm.value );
  272. initMesh();
  273. } );
  274. //
  275. var methodElm = document.getElementById( 'method' );
  276. method = methodElm.value;
  277. methodElm.addEventListener( "change", function() {
  278. method = methodElm.value;
  279. initMesh();
  280. } );
  281. //
  282. var animateElm = document.getElementById( 'animate' );
  283. doAnimate = animateElm.checked;
  284. animateElm.addEventListener( "click", function() {
  285. doAnimate = animateElm.checked;
  286. animate();
  287. } );
  288. //
  289. var overrideElm = document.getElementById( 'override' );
  290. useOverrideMaterial = overrideElm.checked;
  291. overrideElm.addEventListener( "click", function() {
  292. useOverrideMaterial = overrideElm.checked;
  293. initMesh();
  294. } );
  295. //
  296. var constructElm = document.getElementById( 'construct' );
  297. constructElm.addEventListener( "click", function() {
  298. initMesh();
  299. } );
  300. }
  301. function clean() {
  302. THREE.Cache.clear();
  303. materialList.forEach( function( m ) {
  304. m.dispose();
  305. } );
  306. geometryList.forEach( function( g ) {
  307. g.dispose();
  308. } );
  309. scene = new THREE.Scene();
  310. scene.add( camera );
  311. scene.add( highlightBox );
  312. pickingScene = new THREE.Scene();
  313. pickingData = {};
  314. materialList = [];
  315. geometryList = [];
  316. objectCount = 0;
  317. singleMaterial = undefined;
  318. singlePickingMaterial = undefined;
  319. }
  320. var randomizeMatrix = function() {
  321. var position = new THREE.Vector3();
  322. var rotation = new THREE.Euler();
  323. var quaternion = new THREE.Quaternion();
  324. var scale = new THREE.Vector3();
  325. return function( matrix ) {
  326. position.x = Math.random() * 40 - 20;
  327. position.y = Math.random() * 40 - 20;
  328. position.z = Math.random() * 40 - 20;
  329. rotation.x = Math.random() * 2 * Math.PI;
  330. rotation.y = Math.random() * 2 * Math.PI;
  331. rotation.z = Math.random() * 2 * Math.PI;
  332. quaternion.setFromEuler( rotation, false );
  333. scale.x = scale.y = scale.z = Math.random() * 1;
  334. matrix.compose( position, quaternion, scale );
  335. };
  336. }();
  337. function initMesh() {
  338. clean();
  339. // make instances
  340. loader.load( 'obj/Suzanne.js', function ( geo ) {
  341. geo.computeBoundingBox();
  342. geometrySize = geo.boundingBox.getSize();
  343. geometryList.push( geo );
  344. var start = window.performance.now();
  345. switch ( method ){
  346. case "merged":
  347. makeMerged( geo );
  348. break;
  349. case "instanced":
  350. makeInstanced( geo );
  351. break;
  352. case "singleMaterial":
  353. makeSingleMaterial( geo );
  354. break;
  355. case "multiMaterial":
  356. makeMultiMaterial( geo );
  357. break;
  358. }
  359. render();
  360. var end = window.performance.now();
  361. document.getElementById( 'materialCount' ).innerText = materialList.length;
  362. document.getElementById( 'objectCount' ).innerText = objectCount;
  363. document.getElementById( 'drawcalls' ).innerText = renderer.info.render.calls;
  364. document.getElementById( 'initTime' ).innerText = ( end - start ).toFixed( 2 );
  365. } );
  366. }
  367. function makeMultiMaterial( geo ) {
  368. // material
  369. var vert = document.getElementById( 'vertMaterial' ).textContent;
  370. var frag = document.getElementById( 'fragMaterial' ).textContent;
  371. var material = new THREE.RawShaderMaterial( {
  372. vertexShader: vert,
  373. fragmentShader: frag,
  374. uniforms: {
  375. color: {
  376. value: new THREE.Color()
  377. }
  378. }
  379. } );
  380. var pickingMaterial = new THREE.RawShaderMaterial( {
  381. vertexShader: "#define PICKING\n" + vert,
  382. fragmentShader: "#define PICKING\n" + frag,
  383. uniforms: {
  384. pickingColor: {
  385. value: new THREE.Color()
  386. }
  387. }
  388. } );
  389. // geometry / mesh
  390. var matrix = new THREE.Matrix4();
  391. for ( var i = 0; i < instanceCount; i ++ ) {
  392. var object = new THREE.Mesh( geo, material );
  393. objectCount ++;
  394. randomizeMatrix( matrix );
  395. object.applyMatrix( matrix );
  396. var pickingObject = object.clone();
  397. objectCount ++;
  398. object.material = material.clone();
  399. object.material.uniforms.color.value.setHex( Math.random() * 0xffffff );
  400. materialList.push( object.material );
  401. pickingObject.material = pickingMaterial.clone();
  402. pickingObject.material.uniforms.pickingColor.value.setHex( i + 1 );
  403. materialList.push( pickingObject.material );
  404. pickingData[ i + 1 ] = object;
  405. scene.add( object );
  406. pickingScene.add( pickingObject );
  407. }
  408. material.dispose();
  409. pickingMaterial.dispose();
  410. }
  411. function makeSingleMaterial( geo ) {
  412. // material
  413. var vert = document.getElementById( 'vertMaterial' ).textContent;
  414. var frag = document.getElementById( 'fragMaterial' ).textContent;
  415. function updateColor( object, material, camera ) {
  416. this.value.setHex( object.userData.color );
  417. }
  418. var material = new THREE.RawShaderMaterial( {
  419. vertexShader: vert,
  420. fragmentShader: frag,
  421. uniforms: {
  422. color: new THREE.Uniform( new THREE.Color() ).onUpdate( updateColor )
  423. }
  424. } );
  425. materialList.push( material );
  426. function updatePickingColor( object, camera ) {
  427. this.value.setHex( object.userData.pickingColor );
  428. }
  429. var pickingMaterial = new THREE.RawShaderMaterial( {
  430. vertexShader: "#define PICKING\n" + vert,
  431. fragmentShader: "#define PICKING\n" + frag,
  432. uniforms: {
  433. pickingColor: new THREE.Uniform( new THREE.Color() ).onUpdate( updatePickingColor )
  434. }
  435. } );
  436. materialList.push( pickingMaterial );
  437. if ( useOverrideMaterial ) {
  438. // make globally available
  439. singleMaterial = material;
  440. singlePickingMaterial = pickingMaterial;
  441. }
  442. // geometry / mesh
  443. var matrix = new THREE.Matrix4();
  444. for ( var i = 0; i < instanceCount; i ++ ) {
  445. var object = new THREE.Mesh( geo, material );
  446. objectCount ++;
  447. randomizeMatrix( matrix );
  448. object.applyMatrix( matrix );
  449. var pickingObject;
  450. if ( ! useOverrideMaterial ) {
  451. pickingObject = object.clone();
  452. objectCount ++;
  453. }
  454. object.material = material;
  455. object.userData[ "color" ] = Math.random() * 0xffffff;
  456. if ( useOverrideMaterial ) {
  457. object.userData[ "pickingColor" ] = i + 1;
  458. }else {
  459. pickingObject.material = pickingMaterial;
  460. pickingObject.userData[ "pickingColor" ] = i + 1;
  461. }
  462. pickingData[ i + 1 ] = object;
  463. scene.add( object );
  464. if ( ! useOverrideMaterial ) pickingScene.add( pickingObject );
  465. }
  466. }
  467. function makeMerged( geo ) {
  468. // material
  469. var vert = document.getElementById( 'vertMerged' ).textContent;
  470. var frag = document.getElementById( 'fragMerged' ).textContent;
  471. var material = new THREE.RawShaderMaterial( {
  472. vertexShader: vert,
  473. fragmentShader: frag
  474. } );
  475. materialList.push( material );
  476. var pickingMaterial = new THREE.RawShaderMaterial( {
  477. vertexShader: "#define PICKING\n" + vert,
  478. fragmentShader: "#define PICKING\n" + frag
  479. } );
  480. materialList.push( pickingMaterial );
  481. // geometry
  482. var bgeo = new THREE.BufferGeometry().fromGeometry( geo );
  483. geometryList.push( bgeo );
  484. var mgeo = new THREE.BufferGeometry();
  485. geometryList.push( mgeo );
  486. var pos = bgeo.attributes.position;
  487. var posLen = bgeo.attributes.position.count * 3;
  488. var vertices = new THREE.BufferAttribute(
  489. new Float32Array( instanceCount * posLen ), 3
  490. );
  491. var vertex = new THREE.Vector3();
  492. var matrix = new THREE.Matrix4();
  493. for ( var i = 0, ul = instanceCount; i < ul; i ++ ) {
  494. var offset = i * posLen;
  495. randomizeMatrix( matrix );
  496. var object = new THREE.Object3D();
  497. objectCount ++;
  498. object.applyMatrix( matrix );
  499. pickingData[ i + 1 ] = object;
  500. vertices.set( pos.array, offset );
  501. for ( var k = 0, l = offset; k < posLen; k += 3, l += 3 ) {
  502. vertex.fromArray( vertices.array, l );
  503. vertex.applyMatrix4( matrix );
  504. vertex.toArray( vertices.array, l );
  505. }
  506. }
  507. mgeo.addAttribute( 'position', vertices );
  508. var colCount = posLen / 3;
  509. var colors = new THREE.BufferAttribute(
  510. new Float32Array( instanceCount * colCount * 3 ), 3
  511. );
  512. var randCol = function() {
  513. return Math.random();
  514. };
  515. for ( var i = 0, ul = instanceCount; i < ul; i ++ ) {
  516. var r = randCol(), g = randCol(), b = randCol();
  517. for ( var j = i * colCount, jl = ( i + 1 ) * colCount; j < jl; j ++ ) {
  518. colors.setXYZ( j, r, g, b );
  519. }
  520. }
  521. mgeo.addAttribute( 'color', colors );
  522. var col = new THREE.Color();
  523. var pickingColors = new THREE.BufferAttribute(
  524. new Float32Array( instanceCount * colCount * 3 ), 3
  525. );
  526. for ( var i = 0, ul = instanceCount; i < ul; i ++ ) {
  527. col.setHex( i + 1 );
  528. for ( var j = i * colCount, jl = ( i + 1 ) * colCount; j < jl; j ++ ) {
  529. pickingColors.setXYZ( j, col.r, col.g, col.b );
  530. }
  531. }
  532. mgeo.addAttribute( 'pickingColor', pickingColors );
  533. // mesh
  534. var mesh = new THREE.Mesh( mgeo, material );
  535. scene.add( mesh );
  536. var pickingMesh = new THREE.Mesh( mgeo, pickingMaterial );
  537. pickingScene.add( pickingMesh );
  538. }
  539. function makeInstanced( geo ) {
  540. // material
  541. var vert = document.getElementById( 'vertInstanced' ).textContent;
  542. var frag = document.getElementById( 'fragInstanced' ).textContent;
  543. var material = new THREE.RawShaderMaterial( {
  544. vertexShader: vert,
  545. fragmentShader: frag,
  546. } );
  547. materialList.push( material );
  548. var pickingMaterial = new THREE.RawShaderMaterial( {
  549. vertexShader: "#define PICKING\n" + vert,
  550. fragmentShader: "#define PICKING\n" + frag
  551. } );
  552. materialList.push( pickingMaterial );
  553. // geometry
  554. var bgeo = new THREE.BufferGeometry().fromGeometry( geo );
  555. geometryList.push( bgeo );
  556. var igeo = new THREE.InstancedBufferGeometry();
  557. geometryList.push( igeo );
  558. var vertices = bgeo.attributes.position.clone();
  559. igeo.addAttribute( 'position', vertices );
  560. // var matrices = new THREE.InstancedBufferAttribute(
  561. // new Float32Array( instanceCount * 16 ), 16, 1
  562. // );
  563. var mcol0 = new THREE.InstancedBufferAttribute(
  564. new Float32Array( instanceCount * 3 ), 3, 1
  565. );
  566. var mcol1 = new THREE.InstancedBufferAttribute(
  567. new Float32Array( instanceCount * 3 ), 3, 1
  568. );
  569. var mcol2 = new THREE.InstancedBufferAttribute(
  570. new Float32Array( instanceCount * 3 ), 3, 1
  571. );
  572. var mcol3 = new THREE.InstancedBufferAttribute(
  573. new Float32Array( instanceCount * 3 ), 3, 1
  574. );
  575. var matrix = new THREE.Matrix4();
  576. var me = matrix.elements;
  577. for ( var i = 0, ul = mcol0.count; i < ul; i ++ ) {
  578. randomizeMatrix( matrix );
  579. var object = new THREE.Object3D();
  580. objectCount ++;
  581. object.applyMatrix( matrix );
  582. pickingData[ i + 1 ] = object;
  583. // matrices.set( matrix.elements, i * 16 );
  584. mcol0.setXYZ( i, me[ 0 ], me[ 1 ], me[ 2 ] );
  585. mcol1.setXYZ( i, me[ 4 ], me[ 5 ], me[ 6 ] );
  586. mcol2.setXYZ( i, me[ 8 ], me[ 9 ], me[ 10 ] );
  587. mcol3.setXYZ( i, me[ 12 ], me[ 13 ], me[ 14 ] );
  588. }
  589. // igeo.addAttribute( 'matrix', matrices );
  590. igeo.addAttribute( 'mcol0', mcol0 );
  591. igeo.addAttribute( 'mcol1', mcol1 );
  592. igeo.addAttribute( 'mcol2', mcol2 );
  593. igeo.addAttribute( 'mcol3', mcol3 );
  594. var randCol = function() {
  595. return Math.random();
  596. };
  597. var colors = new THREE.InstancedBufferAttribute(
  598. new Float32Array( instanceCount * 3 ), 3, 1
  599. );
  600. for ( var i = 0, ul = colors.count; i < ul; i ++ ) {
  601. colors.setXYZ( i, randCol(), randCol(), randCol() );
  602. }
  603. igeo.addAttribute( 'color', colors );
  604. var col = new THREE.Color();
  605. var pickingColors = new THREE.InstancedBufferAttribute(
  606. new Float32Array( instanceCount * 3 ), 3, 1
  607. );
  608. for ( var i = 0, ul = pickingColors.count; i < ul; i ++ ) {
  609. col.setHex( i + 1 );
  610. pickingColors.setXYZ( i, col.r, col.g, col.b );
  611. }
  612. igeo.addAttribute( 'pickingColor', pickingColors );
  613. // mesh
  614. var mesh = new THREE.Mesh( igeo, material );
  615. scene.add( mesh );
  616. var pickingMesh = new THREE.Mesh( igeo, pickingMaterial );
  617. pickingScene.add( pickingMesh );
  618. }
  619. function init() {
  620. // camera
  621. camera = new THREE.PerspectiveCamera(
  622. 70, window.innerWidth / window.innerHeight, 1, 100
  623. );
  624. camera.position.z = 40;
  625. // picking render target
  626. pickingRenderTarget = new THREE.WebGLRenderTarget(
  627. window.innerWidth, window.innerHeight
  628. );
  629. pickingRenderTarget.texture.generateMipmaps = false;
  630. pickingRenderTarget.texture.minFilter = THREE.NearestFilter;
  631. // highlight box
  632. highlightBox = new THREE.Mesh(
  633. new THREE.BoxGeometry( 1, 1, 1 ),
  634. new THREE.MeshLambertMaterial( {
  635. emissive: 0xffff00,
  636. transparent: true,
  637. opacity: 0.5,
  638. side: THREE.FrontSide
  639. } )
  640. );
  641. // renderer
  642. container = document.getElementById( "container" );
  643. renderer = new THREE.WebGLRenderer( {
  644. antialias: true,
  645. alpha: true
  646. } );
  647. if ( renderer.extensions.get( 'ANGLE_instanced_arrays' ) === false ) {
  648. document.getElementById( "notSupported" ).style.display = "";
  649. return;
  650. }
  651. renderer.setClearColor( 0xffffff );
  652. renderer.setPixelRatio( window.devicePixelRatio );
  653. renderer.setSize( window.innerWidth, window.innerHeight );
  654. renderer.sortObjects = false;
  655. container.appendChild( renderer.domElement );
  656. if ( renderer.extensions.get( 'ANGLE_instanced_arrays' ) === false ) {
  657. throw 'ANGLE_instanced_arrays not supported';
  658. }
  659. // controls
  660. controls = new THREE.TrackballControls(
  661. camera, renderer.domElement
  662. );
  663. controls.staticMoving = true;
  664. // stats
  665. stats = new Stats();
  666. container.appendChild( stats.dom );
  667. // listeners
  668. renderer.domElement.addEventListener( 'mousemove', onMouseMove );
  669. window.addEventListener( 'resize', onWindowResize, false );
  670. }
  671. //
  672. function onMouseMove( e ) {
  673. mouse.x = e.clientX;
  674. mouse.y = e.clientY;
  675. controls.update();
  676. requestAnimationFrame( render );
  677. }
  678. function onWindowResize( event ) {
  679. camera.aspect = window.innerWidth / window.innerHeight;
  680. camera.updateProjectionMatrix();
  681. renderer.setSize( window.innerWidth, window.innerHeight );
  682. pickingRenderTarget.setSize( window.innerWidth, window.innerHeight );
  683. }
  684. function animate() {
  685. if ( doAnimate ) {
  686. requestAnimationFrame( animate );
  687. }
  688. controls.update();
  689. stats.update();
  690. render();
  691. }
  692. function pick() {
  693. // render the picking scene off-screen
  694. highlightBox.visible = false;
  695. if ( singlePickingMaterial ) {
  696. scene.overrideMaterial = singlePickingMaterial;
  697. renderer.render( scene, camera, pickingRenderTarget );
  698. scene.overrideMaterial = null;
  699. }else {
  700. renderer.render( pickingScene, camera, pickingRenderTarget );
  701. }
  702. // read the pixel under the mouse from the texture
  703. renderer.readRenderTargetPixels(
  704. pickingRenderTarget,
  705. mouse.x,
  706. pickingRenderTarget.height - mouse.y,
  707. 1,
  708. 1,
  709. pixelBuffer
  710. );
  711. // interpret the pixel as an ID
  712. var id =
  713. ( pixelBuffer[ 0 ] << 16 ) |
  714. ( pixelBuffer[ 1 ] << 8 ) |
  715. ( pixelBuffer[ 2 ] );
  716. var object = pickingData[ id ];
  717. if ( object ) {
  718. // move the highlightBox so that it surrounds the picked object
  719. if ( object.position && object.rotation && object.scale ) {
  720. highlightBox.position.copy( object.position );
  721. highlightBox.rotation.copy( object.rotation );
  722. highlightBox.scale.copy( object.scale )
  723. .multiply( geometrySize )
  724. .multiplyScalar( scale );
  725. highlightBox.visible = true;
  726. }
  727. } else {
  728. highlightBox.visible = false;
  729. }
  730. }
  731. function render() {
  732. pick();
  733. renderer.render( scene, camera );
  734. }
  735. </script>
  736. </body>
  737. </html>