webgl_interactive_instances_gpu.html 23 KB

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