webgl_gpgpu_birds_gltf.html 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - gpgpu - flocking</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. <link type="text/css" rel="stylesheet" href="main.css">
  8. <style>
  9. body {
  10. background-color: #fff;
  11. color: #444;
  12. }
  13. a {
  14. color:#08f;
  15. }
  16. </style>
  17. </head>
  18. <body>
  19. <div id="info">
  20. <a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> - webgl gpgpu birds + GLTF mesh<br/>
  21. Flamingo by <a href="http://mirada.com/">mirada</a> from <a href="http://ro.me">rome</a><br/>
  22. Move mouse to disturb birds.
  23. </div>
  24. <!-- shader for bird's position -->
  25. <script id="fragmentShaderPosition" type="x-shader/x-fragment">
  26. uniform float time;
  27. uniform float delta;
  28. void main() {
  29. vec2 uv = gl_FragCoord.xy / resolution.xy;
  30. vec4 tmpPos = texture2D( texturePosition, uv );
  31. vec3 position = tmpPos.xyz;
  32. vec3 velocity = texture2D( textureVelocity, uv ).xyz;
  33. float phase = tmpPos.w;
  34. phase = mod( ( phase + delta +
  35. length( velocity.xz ) * delta * 3. +
  36. max( velocity.y, 0.0 ) * delta * 6. ), 62.83 );
  37. gl_FragColor = vec4( position + velocity * delta * 15. , phase );
  38. }
  39. </script>
  40. <!-- shader for bird's velocity -->
  41. <script id="fragmentShaderVelocity" type="x-shader/x-fragment">
  42. uniform float time;
  43. uniform float testing;
  44. uniform float delta; // about 0.016
  45. uniform float separationDistance; // 20
  46. uniform float alignmentDistance; // 40
  47. uniform float cohesionDistance; //
  48. uniform float freedomFactor;
  49. uniform vec3 predator;
  50. const float width = resolution.x;
  51. const float height = resolution.y;
  52. const float PI = 3.141592653589793;
  53. const float PI_2 = PI * 2.0;
  54. // const float VISION = PI * 0.55;
  55. float zoneRadius = 40.0;
  56. float zoneRadiusSquared = 1600.0;
  57. float separationThresh = 0.45;
  58. float alignmentThresh = 0.65;
  59. const float UPPER_BOUNDS = BOUNDS;
  60. const float LOWER_BOUNDS = -UPPER_BOUNDS;
  61. const float SPEED_LIMIT = 9.0;
  62. float rand( vec2 co ){
  63. return fract( sin( dot( co.xy, vec2(12.9898,78.233) ) ) * 43758.5453 );
  64. }
  65. void main() {
  66. zoneRadius = separationDistance + alignmentDistance + cohesionDistance;
  67. separationThresh = separationDistance / zoneRadius;
  68. alignmentThresh = ( separationDistance + alignmentDistance ) / zoneRadius;
  69. zoneRadiusSquared = zoneRadius * zoneRadius;
  70. vec2 uv = gl_FragCoord.xy / resolution.xy;
  71. vec3 birdPosition, birdVelocity;
  72. vec3 selfPosition = texture2D( texturePosition, uv ).xyz;
  73. vec3 selfVelocity = texture2D( textureVelocity, uv ).xyz;
  74. float dist;
  75. vec3 dir; // direction
  76. float distSquared;
  77. float separationSquared = separationDistance * separationDistance;
  78. float cohesionSquared = cohesionDistance * cohesionDistance;
  79. float f;
  80. float percent;
  81. vec3 velocity = selfVelocity;
  82. float limit = SPEED_LIMIT;
  83. dir = predator * UPPER_BOUNDS - selfPosition;
  84. dir.z = 0.;
  85. // dir.z *= 0.6;
  86. dist = length( dir );
  87. distSquared = dist * dist;
  88. float preyRadius = 150.0;
  89. float preyRadiusSq = preyRadius * preyRadius;
  90. // move birds away from predator
  91. if ( dist < preyRadius ) {
  92. f = ( distSquared / preyRadiusSq - 1.0 ) * delta * 100.;
  93. velocity += normalize( dir ) * f;
  94. limit += 5.0;
  95. }
  96. // if (testing == 0.0) {}
  97. // if ( rand( uv + time ) < freedomFactor ) {}
  98. // Attract flocks to the center
  99. vec3 central = vec3( 0., 0., 0. );
  100. dir = selfPosition - central;
  101. dist = length( dir );
  102. dir.y *= 2.5;
  103. velocity -= normalize( dir ) * delta * 5.;
  104. for ( float y = 0.0; y < height; y++ ) {
  105. for ( float x = 0.0; x < width; x++ ) {
  106. vec2 ref = vec2( x + 0.5, y + 0.5 ) / resolution.xy;
  107. birdPosition = texture2D( texturePosition, ref ).xyz;
  108. dir = birdPosition - selfPosition;
  109. dist = length( dir );
  110. if ( dist < 0.0001 ) continue;
  111. distSquared = dist * dist;
  112. if ( distSquared > zoneRadiusSquared ) continue;
  113. percent = distSquared / zoneRadiusSquared;
  114. if ( percent < separationThresh ) { // low
  115. // Separation - Move apart for comfort
  116. f = ( separationThresh / percent - 1.0 ) * delta;
  117. velocity -= normalize( dir ) * f;
  118. } else if ( percent < alignmentThresh ) { // high
  119. // Alignment - fly the same direction
  120. float threshDelta = alignmentThresh - separationThresh;
  121. float adjustedPercent = ( percent - separationThresh ) / threshDelta;
  122. birdVelocity = texture2D( textureVelocity, ref ).xyz;
  123. f = ( 0.5 - cos( adjustedPercent * PI_2 ) * 0.5 + 0.5 ) * delta;
  124. velocity += normalize( birdVelocity ) * f;
  125. } else {
  126. // Attraction / Cohesion - move closer
  127. float threshDelta = 1.0 - alignmentThresh;
  128. float adjustedPercent = ( percent - alignmentThresh ) / threshDelta;
  129. f = ( 0.5 - ( cos( adjustedPercent * PI_2 ) * -0.5 + 0.5 ) ) * delta;
  130. velocity += normalize( dir ) * f;
  131. }
  132. }
  133. }
  134. // this make tends to fly around than down or up
  135. // if (velocity.y > 0.) velocity.y *= (1. - 0.2 * delta);
  136. // Speed Limits
  137. if ( length( velocity ) > limit ) {
  138. velocity = normalize( velocity ) * limit;
  139. }
  140. gl_FragColor = vec4( velocity, 1.0 );
  141. }
  142. </script>
  143. <script type="module">
  144. import * as THREE from '../build/three.module.js';
  145. import Stats from './jsm/libs/stats.module.js';
  146. import { GUI } from './jsm/libs/dat.gui.module.js';
  147. import { GLTFLoader } from './jsm/loaders/GLTFLoader.js';
  148. import { GPUComputationRenderer } from './jsm/misc/GPUComputationRenderer.js';
  149. /* TEXTURE WIDTH FOR SIMULATION */
  150. var WIDTH = 64;
  151. var BIRDS = WIDTH * WIDTH;
  152. /* BAKE ANIMATION INTO TEXTURE and CREATE GEOMETRY FROM BASE MODEL */
  153. var BirdGeometry = new THREE.BufferGeometry();
  154. var textureAnimation, durationAnimation, birdMesh, materialShader, vertexPerBird;
  155. function nextPowerOf2( n ) {
  156. return Math.pow( 2, Math.ceil( Math.log( n ) / Math.log( 2 ) ) );
  157. }
  158. Math.lerp = function ( value1, value2, amount ) {
  159. amount = Math.max( Math.min( amount, 1 ), 0 );
  160. return value1 + ( value2 - value1 ) * amount;
  161. };
  162. var gltfs = [ 'models/gltf/Parrot.glb', 'models/gltf/Flamingo.glb' ];
  163. var colors = [ 0xccFFFF, 0xffdeff ];
  164. var sizes = [ 0.2, 0.1 ];
  165. var selectModel = Math.floor( Math.random() * gltfs.length );
  166. new GLTFLoader().load( gltfs[ selectModel ], function ( gltf ) {
  167. var animations = gltf.animations;
  168. durationAnimation = Math.round( animations[ 0 ].duration * 60 );
  169. var birdGeo = gltf.scene.children[ 0 ].geometry;
  170. var morphAttributes = birdGeo.morphAttributes.position;
  171. var tHeight = nextPowerOf2( durationAnimation );
  172. var tWidth = nextPowerOf2( birdGeo.getAttribute( 'position' ).count );
  173. vertexPerBird = birdGeo.getAttribute( 'position' ).count;
  174. var tData = new Float32Array( 3 * tWidth * tHeight );
  175. for ( var i = 0; i < tWidth; i ++ ) {
  176. for ( var j = 0; j < tHeight; j ++ ) {
  177. var offset = j * tWidth * 3;
  178. var curMorph = Math.floor( j / durationAnimation * morphAttributes.length );
  179. var nextMorph = ( Math.floor( j / durationAnimation * morphAttributes.length ) + 1 ) % morphAttributes.length;
  180. var lerpAmount = j / durationAnimation * morphAttributes.length % 1;
  181. if ( j < durationAnimation ) {
  182. var d0, d1;
  183. d0 = morphAttributes[ curMorph ].array[ i * 3 ];
  184. d1 = morphAttributes[ nextMorph ].array[ i * 3 ];
  185. if ( d0 !== undefined && d1 !== undefined ) tData[ offset + i * 3 ] = Math.lerp( d0, d1, lerpAmount );
  186. d0 = morphAttributes[ curMorph ].array[ i * 3 + 1 ];
  187. d1 = morphAttributes[ nextMorph ].array[ i * 3 + 1 ];
  188. if ( d0 !== undefined && d1 !== undefined ) tData[ offset + i * 3 + 1 ] = Math.lerp( d0, d1, lerpAmount );
  189. d0 = morphAttributes[ curMorph ].array[ i * 3 + 2 ];
  190. d1 = morphAttributes[ nextMorph ].array[ i * 3 + 2 ];
  191. if ( d0 !== undefined && d1 !== undefined ) tData[ offset + i * 3 + 2 ] = Math.lerp( d0, d1, lerpAmount );
  192. }
  193. }
  194. }
  195. textureAnimation = new THREE.DataTexture( tData, tWidth, tHeight, THREE.RGBFormat, THREE.FloatType );
  196. textureAnimation.needsUpdate = true;
  197. var vertices = [], color = [], reference = [], seeds = [], indices = [];
  198. var totalVertices = birdGeo.getAttribute( 'position' ).count * 3 * BIRDS;
  199. for ( var i = 0; i < totalVertices; i ++ ) {
  200. var bIndex = i % ( birdGeo.getAttribute( 'position' ).count * 3 );
  201. vertices.push( birdGeo.getAttribute( 'position' ).array[ bIndex ] );
  202. color.push( birdGeo.getAttribute( 'color' ).array[ bIndex ] );
  203. }
  204. var r = Math.random();
  205. for ( var i = 0; i < birdGeo.getAttribute( 'position' ).count * BIRDS; i ++ ) {
  206. var bIndex = i % ( birdGeo.getAttribute( 'position' ).count );
  207. var bird = Math.floor( i / birdGeo.getAttribute( 'position' ).count );
  208. if ( bIndex == 0 ) r = Math.random();
  209. var j = ~ ~ bird;
  210. var x = ( j % WIDTH ) / WIDTH;
  211. var y = ~ ~ ( j / WIDTH ) / WIDTH;
  212. reference.push( x, y, bIndex / tWidth, durationAnimation / tHeight );
  213. seeds.push( bird, r, Math.random(), Math.random() );
  214. }
  215. for ( var i = 0; i < birdGeo.index.array.length * BIRDS; i ++ ) {
  216. var offset = Math.floor( i / birdGeo.index.array.length ) * ( birdGeo.getAttribute( 'position' ).count );
  217. indices.push( birdGeo.index.array[ i % birdGeo.index.array.length ] + offset );
  218. }
  219. BirdGeometry.setAttribute( 'position', new THREE.BufferAttribute( new Float32Array( vertices ), 3 ) );
  220. BirdGeometry.setAttribute( 'birdColor', new THREE.BufferAttribute( new Float32Array( color ), 3 ) );
  221. BirdGeometry.setAttribute( 'color', new THREE.BufferAttribute( new Float32Array( color ), 3 ) );
  222. BirdGeometry.setAttribute( 'reference', new THREE.BufferAttribute( new Float32Array( reference ), 4 ) );
  223. BirdGeometry.setAttribute( 'seeds', new THREE.BufferAttribute( new Float32Array( seeds ), 4 ) );
  224. BirdGeometry.setIndex( indices );
  225. init();
  226. animate();
  227. } );
  228. var container, stats;
  229. var camera, scene, renderer;
  230. var mouseX = 0, mouseY = 0;
  231. var windowHalfX = window.innerWidth / 2;
  232. var windowHalfY = window.innerHeight / 2;
  233. var BOUNDS = 800, BOUNDS_HALF = BOUNDS / 2;
  234. var last = performance.now();
  235. var gpuCompute;
  236. var velocityVariable;
  237. var positionVariable;
  238. var positionUniforms;
  239. var velocityUniforms;
  240. function init() {
  241. container = document.createElement( 'div' );
  242. document.body.appendChild( container );
  243. camera = new THREE.PerspectiveCamera( 75, window.innerWidth / window.innerHeight, 1, 3000 );
  244. camera.position.z = 350;
  245. scene = new THREE.Scene();
  246. scene.background = new THREE.Color( colors[ selectModel ] );
  247. scene.fog = new THREE.Fog( colors[ selectModel ], 100, 1000 );
  248. // LIGHTS
  249. var hemiLight = new THREE.HemisphereLight( colors[ selectModel ], 0xffffff, 1.6 );
  250. hemiLight.color.setHSL( 0.6, 1, 0.6 );
  251. hemiLight.groundColor.setHSL( 0.095, 1, 0.75 );
  252. hemiLight.position.set( 0, 50, 0 );
  253. scene.add( hemiLight );
  254. var dirLight = new THREE.DirectionalLight( 0x00CED1, 0.6 );
  255. dirLight.color.setHSL( 0.1, 1, 0.95 );
  256. dirLight.position.set( - 1, 1.75, 1 );
  257. dirLight.position.multiplyScalar( 30 );
  258. scene.add( dirLight );
  259. renderer = new THREE.WebGLRenderer( { antialias: true } );
  260. renderer.setPixelRatio( window.devicePixelRatio );
  261. renderer.setSize( window.innerWidth, window.innerHeight );
  262. container.appendChild( renderer.domElement );
  263. initComputeRenderer();
  264. stats = new Stats();
  265. container.appendChild( stats.dom );
  266. document.addEventListener( 'mousemove', onDocumentMouseMove, false );
  267. document.addEventListener( 'touchstart', onDocumentTouchStart, false );
  268. document.addEventListener( 'touchmove', onDocumentTouchMove, false );
  269. window.addEventListener( 'resize', onWindowResize, false );
  270. var gui = new GUI();
  271. var effectController = {
  272. separation: 20.0,
  273. alignment: 20.0,
  274. cohesion: 20.0,
  275. freedom: 0.75,
  276. size: sizes[ selectModel ],
  277. count: BIRDS
  278. };
  279. var valuesChanger = function () {
  280. velocityUniforms[ "separationDistance" ].value = effectController.separation;
  281. velocityUniforms[ "alignmentDistance" ].value = effectController.alignment;
  282. velocityUniforms[ "cohesionDistance" ].value = effectController.cohesion;
  283. velocityUniforms[ "freedomFactor" ].value = effectController.freedom;
  284. if ( materialShader ) materialShader.uniforms[ "size" ].value = effectController.size;
  285. BirdGeometry.setDrawRange( 0, vertexPerBird * effectController.count );
  286. };
  287. valuesChanger();
  288. gui.add( effectController, "separation", 0.0, 100.0, 1.0 ).onChange( valuesChanger );
  289. gui.add( effectController, "alignment", 0.0, 100, 0.001 ).onChange( valuesChanger );
  290. gui.add( effectController, "cohesion", 0.0, 100, 0.025 ).onChange( valuesChanger );
  291. gui.add( effectController, "size", 0, 1, 0.01 ).onChange( valuesChanger );
  292. gui.add( effectController, "count", 0, BIRDS, 1 ).onChange( valuesChanger );
  293. gui.close();
  294. initBirds();
  295. }
  296. function initComputeRenderer() {
  297. gpuCompute = new GPUComputationRenderer( WIDTH, WIDTH, renderer );
  298. if ( isSafari() ) {
  299. gpuCompute.setDataType( THREE.HalfFloatType );
  300. }
  301. var dtPosition = gpuCompute.createTexture();
  302. var dtVelocity = gpuCompute.createTexture();
  303. fillPositionTexture( dtPosition );
  304. fillVelocityTexture( dtVelocity );
  305. velocityVariable = gpuCompute.addVariable( "textureVelocity", document.getElementById( 'fragmentShaderVelocity' ).textContent, dtVelocity );
  306. positionVariable = gpuCompute.addVariable( "texturePosition", document.getElementById( 'fragmentShaderPosition' ).textContent, dtPosition );
  307. gpuCompute.setVariableDependencies( velocityVariable, [ positionVariable, velocityVariable ] );
  308. gpuCompute.setVariableDependencies( positionVariable, [ positionVariable, velocityVariable ] );
  309. positionUniforms = positionVariable.material.uniforms;
  310. velocityUniforms = velocityVariable.material.uniforms;
  311. positionUniforms[ "time" ] = { value: 0.0 };
  312. positionUniforms[ "delta" ] = { value: 0.0 };
  313. velocityUniforms[ "time" ] = { value: 1.0 };
  314. velocityUniforms[ "delta" ] = { value: 0.0 };
  315. velocityUniforms[ "testing" ] = { value: 1.0 };
  316. velocityUniforms[ "separationDistance" ] = { value: 1.0 };
  317. velocityUniforms[ "alignmentDistance" ] = { value: 1.0 };
  318. velocityUniforms[ "cohesionDistance" ] = { value: 1.0 };
  319. velocityUniforms[ "freedomFactor" ] = { value: 1.0 };
  320. velocityUniforms[ "predator" ] = { value: new THREE.Vector3() };
  321. velocityVariable.material.defines.BOUNDS = BOUNDS.toFixed( 2 );
  322. velocityVariable.wrapS = THREE.RepeatWrapping;
  323. velocityVariable.wrapT = THREE.RepeatWrapping;
  324. positionVariable.wrapS = THREE.RepeatWrapping;
  325. positionVariable.wrapT = THREE.RepeatWrapping;
  326. var error = gpuCompute.init();
  327. if ( error !== null ) {
  328. console.error( error );
  329. }
  330. }
  331. function isSafari() {
  332. return !! navigator.userAgent.match( /Safari/i ) && ! navigator.userAgent.match( /Chrome/i );
  333. }
  334. function initBirds() {
  335. var geometry = BirdGeometry;
  336. var m = new THREE.MeshStandardMaterial( {
  337. vertexColors: true,
  338. flatShading: true,
  339. roughness: 1,
  340. metalness: 0
  341. } );
  342. m.onBeforeCompile = ( shader ) => {
  343. shader.uniforms.texturePosition = { value: null };
  344. shader.uniforms.textureVelocity = { value: null };
  345. shader.uniforms.textureAnimation = { value: textureAnimation };
  346. shader.uniforms.time = { value: 1.0 };
  347. shader.uniforms.size = { value: 0.1 };
  348. shader.uniforms.delta = { value: 0.0 };
  349. var token = '#define STANDARD';
  350. var insert = /* glsl */`
  351. attribute vec4 reference;
  352. attribute vec4 seeds;
  353. attribute vec3 birdColor;
  354. uniform sampler2D texturePosition;
  355. uniform sampler2D textureVelocity;
  356. uniform sampler2D textureAnimation;
  357. uniform float size;
  358. uniform float time;
  359. `;
  360. shader.vertexShader = shader.vertexShader.replace( token, token + insert );
  361. var token = '#include <begin_vertex>';
  362. var insert = /* glsl */`
  363. vec4 tmpPos = texture2D( texturePosition, reference.xy );
  364. vec3 pos = tmpPos.xyz;
  365. vec3 velocity = normalize(texture2D( textureVelocity, reference.xy ).xyz);
  366. vec3 aniPos = texture2D( textureAnimation, vec2( reference.z, mod( ( time + seeds.x ) * ( ( 0.0004 + seeds.y / 10000.0) + normalize( velocity ) / 20000.0 ), reference.w ) ) ).xyz;
  367. vec3 newPosition = position;
  368. newPosition = mat3( modelMatrix ) * ( newPosition + aniPos );
  369. newPosition *= size + seeds.y * size * 0.2;
  370. velocity.z *= -1.;
  371. float xz = length( velocity.xz );
  372. float xyz = 1.;
  373. float x = sqrt( 1. - velocity.y * velocity.y );
  374. float cosry = velocity.x / xz;
  375. float sinry = velocity.z / xz;
  376. float cosrz = x / xyz;
  377. float sinrz = velocity.y / xyz;
  378. mat3 maty = mat3( cosry, 0, -sinry, 0 , 1, 0 , sinry, 0, cosry );
  379. mat3 matz = mat3( cosrz , sinrz, 0, -sinrz, cosrz, 0, 0 , 0 , 1 );
  380. newPosition = maty * matz * newPosition;
  381. newPosition += pos;
  382. vec3 transformed = vec3( newPosition );
  383. `;
  384. shader.vertexShader = shader.vertexShader.replace( token, insert );
  385. materialShader = shader;
  386. };
  387. birdMesh = new THREE.Mesh( geometry, m );
  388. birdMesh.rotation.y = Math.PI / 2;
  389. birdMesh.castShadow = true;
  390. birdMesh.receiveShadow = true;
  391. scene.add( birdMesh );
  392. }
  393. function fillPositionTexture( texture ) {
  394. var theArray = texture.image.data;
  395. for ( var k = 0, kl = theArray.length; k < kl; k += 4 ) {
  396. var x = Math.random() * BOUNDS - BOUNDS_HALF;
  397. var y = Math.random() * BOUNDS - BOUNDS_HALF;
  398. var z = Math.random() * BOUNDS - BOUNDS_HALF;
  399. theArray[ k + 0 ] = x;
  400. theArray[ k + 1 ] = y;
  401. theArray[ k + 2 ] = z;
  402. theArray[ k + 3 ] = 1;
  403. }
  404. }
  405. function fillVelocityTexture( texture ) {
  406. var theArray = texture.image.data;
  407. for ( var k = 0, kl = theArray.length; k < kl; k += 4 ) {
  408. var x = Math.random() - 0.5;
  409. var y = Math.random() - 0.5;
  410. var z = Math.random() - 0.5;
  411. theArray[ k + 0 ] = x * 10;
  412. theArray[ k + 1 ] = y * 10;
  413. theArray[ k + 2 ] = z * 10;
  414. theArray[ k + 3 ] = 1;
  415. }
  416. }
  417. function onWindowResize() {
  418. windowHalfX = window.innerWidth / 2;
  419. windowHalfY = window.innerHeight / 2;
  420. camera.aspect = window.innerWidth / window.innerHeight;
  421. camera.updateProjectionMatrix();
  422. renderer.setSize( window.innerWidth, window.innerHeight );
  423. }
  424. function onDocumentMouseMove( event ) {
  425. mouseX = event.clientX - windowHalfX;
  426. mouseY = event.clientY - windowHalfY;
  427. }
  428. function onDocumentTouchStart( event ) {
  429. if ( event.touches.length === 1 ) {
  430. event.preventDefault();
  431. mouseX = event.touches[ 0 ].pageX - windowHalfX;
  432. mouseY = event.touches[ 0 ].pageY - windowHalfY;
  433. }
  434. }
  435. function onDocumentTouchMove( event ) {
  436. if ( event.touches.length === 1 ) {
  437. event.preventDefault();
  438. mouseX = event.touches[ 0 ].pageX - windowHalfX;
  439. mouseY = event.touches[ 0 ].pageY - windowHalfY;
  440. }
  441. }
  442. //
  443. function animate() {
  444. requestAnimationFrame( animate );
  445. render();
  446. stats.update();
  447. }
  448. function render() {
  449. var now = performance.now();
  450. var delta = ( now - last ) / 1000;
  451. if ( delta > 1 ) delta = 1; // safety cap on large deltas
  452. last = now;
  453. positionUniforms[ "time" ].value = now;
  454. positionUniforms[ "delta" ].value = delta;
  455. velocityUniforms[ "time" ].value = now;
  456. velocityUniforms[ "delta" ].value = delta;
  457. if ( materialShader ) materialShader.uniforms[ "time" ].value = now;
  458. if ( materialShader ) materialShader.uniforms[ "delta" ].value = delta;
  459. velocityUniforms[ "predator" ].value.set( 0.5 * mouseX / windowHalfX, - 0.5 * mouseY / windowHalfY, 0 );
  460. mouseX = 10000;
  461. mouseY = 10000;
  462. gpuCompute.compute();
  463. if ( materialShader ) materialShader.uniforms[ "texturePosition" ].value = gpuCompute.getCurrentRenderTarget( positionVariable ).texture;
  464. if ( materialShader ) materialShader.uniforms[ "textureVelocity" ].value = gpuCompute.getCurrentRenderTarget( velocityVariable ).texture;
  465. renderer.render( scene, camera );
  466. }
  467. </script>
  468. </body>
  469. </html>