webgl_gpgpu_water.html 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - gpgpu - water</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. </head>
  9. <body>
  10. <div id="info">
  11. <a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> - <span id="waterSize"></span> webgl gpgpu water<br/>
  12. Move mouse to disturb water.<br>
  13. Press mouse button to orbit around. 'W' key toggles wireframe.
  14. </div>
  15. <!-- This is the 'compute shader' for the water heightmap: -->
  16. <script id="heightmapFragmentShader" type="x-shader/x-fragment">
  17. #include <common>
  18. uniform vec2 mousePos;
  19. uniform float mouseSize;
  20. uniform float viscosityConstant;
  21. uniform float heightCompensation;
  22. void main() {
  23. vec2 cellSize = 1.0 / resolution.xy;
  24. vec2 uv = gl_FragCoord.xy * cellSize;
  25. // heightmapValue.x == height from previous frame
  26. // heightmapValue.y == height from penultimate frame
  27. // heightmapValue.z, heightmapValue.w not used
  28. vec4 heightmapValue = texture2D( heightmap, uv );
  29. // Get neighbours
  30. vec4 north = texture2D( heightmap, uv + vec2( 0.0, cellSize.y ) );
  31. vec4 south = texture2D( heightmap, uv + vec2( 0.0, - cellSize.y ) );
  32. vec4 east = texture2D( heightmap, uv + vec2( cellSize.x, 0.0 ) );
  33. vec4 west = texture2D( heightmap, uv + vec2( - cellSize.x, 0.0 ) );
  34. // https://web.archive.org/web/20080618181901/http://freespace.virgin.net/hugo.elias/graphics/x_water.htm
  35. float newHeight = ( ( north.x + south.x + east.x + west.x ) * 0.5 - heightmapValue.y ) * viscosityConstant;
  36. // Mouse influence
  37. float mousePhase = clamp( length( ( uv - vec2( 0.5 ) ) * BOUNDS - vec2( mousePos.x, - mousePos.y ) ) * PI / mouseSize, 0.0, PI );
  38. newHeight += ( cos( mousePhase ) + 1.0 ) * 0.28;
  39. heightmapValue.y = heightmapValue.x;
  40. heightmapValue.x = newHeight;
  41. gl_FragColor = heightmapValue;
  42. }
  43. </script>
  44. <!-- This is just a smoothing 'compute shader' for using manually: -->
  45. <script id="smoothFragmentShader" type="x-shader/x-fragment">
  46. uniform sampler2D smoothTexture;
  47. void main() {
  48. vec2 cellSize = 1.0 / resolution.xy;
  49. vec2 uv = gl_FragCoord.xy * cellSize;
  50. // Computes the mean of texel and 4 neighbours
  51. vec4 textureValue = texture2D( smoothTexture, uv );
  52. textureValue += texture2D( smoothTexture, uv + vec2( 0.0, cellSize.y ) );
  53. textureValue += texture2D( smoothTexture, uv + vec2( 0.0, - cellSize.y ) );
  54. textureValue += texture2D( smoothTexture, uv + vec2( cellSize.x, 0.0 ) );
  55. textureValue += texture2D( smoothTexture, uv + vec2( - cellSize.x, 0.0 ) );
  56. textureValue /= 5.0;
  57. gl_FragColor = textureValue;
  58. }
  59. </script>
  60. <!-- This is a 'compute shader' to read the current level and normal of water at a point -->
  61. <!-- It is used with a variable of size 1x1 -->
  62. <script id="readWaterLevelFragmentShader" type="x-shader/x-fragment">
  63. uniform vec2 point1;
  64. uniform sampler2D levelTexture;
  65. // Integer to float conversion from https://stackoverflow.com/questions/17981163/webgl-read-pixels-from-floating-point-render-target
  66. float shift_right( float v, float amt ) {
  67. v = floor( v ) + 0.5;
  68. return floor( v / exp2( amt ) );
  69. }
  70. float shift_left( float v, float amt ) {
  71. return floor( v * exp2( amt ) + 0.5 );
  72. }
  73. float mask_last( float v, float bits ) {
  74. return mod( v, shift_left( 1.0, bits ) );
  75. }
  76. float extract_bits( float num, float from, float to ) {
  77. from = floor( from + 0.5 ); to = floor( to + 0.5 );
  78. return mask_last( shift_right( num, from ), to - from );
  79. }
  80. vec4 encode_float( float val ) {
  81. if ( val == 0.0 ) return vec4( 0, 0, 0, 0 );
  82. float sign = val > 0.0 ? 0.0 : 1.0;
  83. val = abs( val );
  84. float exponent = floor( log2( val ) );
  85. float biased_exponent = exponent + 127.0;
  86. float fraction = ( ( val / exp2( exponent ) ) - 1.0 ) * 8388608.0;
  87. float t = biased_exponent / 2.0;
  88. float last_bit_of_biased_exponent = fract( t ) * 2.0;
  89. float remaining_bits_of_biased_exponent = floor( t );
  90. float byte4 = extract_bits( fraction, 0.0, 8.0 ) / 255.0;
  91. float byte3 = extract_bits( fraction, 8.0, 16.0 ) / 255.0;
  92. float byte2 = ( last_bit_of_biased_exponent * 128.0 + extract_bits( fraction, 16.0, 23.0 ) ) / 255.0;
  93. float byte1 = ( sign * 128.0 + remaining_bits_of_biased_exponent ) / 255.0;
  94. return vec4( byte4, byte3, byte2, byte1 );
  95. }
  96. void main() {
  97. vec2 cellSize = 1.0 / resolution.xy;
  98. float waterLevel = texture2D( levelTexture, point1 ).x;
  99. vec2 normal = vec2(
  100. ( texture2D( levelTexture, point1 + vec2( - cellSize.x, 0 ) ).x - texture2D( levelTexture, point1 + vec2( cellSize.x, 0 ) ).x ) * WIDTH / BOUNDS,
  101. ( texture2D( levelTexture, point1 + vec2( 0, - cellSize.y ) ).x - texture2D( levelTexture, point1 + vec2( 0, cellSize.y ) ).x ) * WIDTH / BOUNDS );
  102. if ( gl_FragCoord.x < 1.5 ) {
  103. gl_FragColor = encode_float( waterLevel );
  104. } else if ( gl_FragCoord.x < 2.5 ) {
  105. gl_FragColor = encode_float( normal.x );
  106. } else if ( gl_FragCoord.x < 3.5 ) {
  107. gl_FragColor = encode_float( normal.y );
  108. } else {
  109. gl_FragColor = encode_float( 0.0 );
  110. }
  111. }
  112. </script>
  113. <!-- This is the water visualization shader, copied from the THREE.MeshPhongMaterial and modified: -->
  114. <script id="waterVertexShader" type="x-shader/x-vertex">
  115. uniform sampler2D heightmap;
  116. #define PHONG
  117. varying vec3 vViewPosition;
  118. #ifndef FLAT_SHADED
  119. varying vec3 vNormal;
  120. #endif
  121. #include <common>
  122. #include <uv_pars_vertex>
  123. #include <uv2_pars_vertex>
  124. #include <displacementmap_pars_vertex>
  125. #include <envmap_pars_vertex>
  126. #include <color_pars_vertex>
  127. #include <morphtarget_pars_vertex>
  128. #include <skinning_pars_vertex>
  129. #include <shadowmap_pars_vertex>
  130. #include <logdepthbuf_pars_vertex>
  131. #include <clipping_planes_pars_vertex>
  132. void main() {
  133. vec2 cellSize = vec2( 1.0 / WIDTH, 1.0 / WIDTH );
  134. #include <uv_vertex>
  135. #include <uv2_vertex>
  136. #include <color_vertex>
  137. // # include <beginnormal_vertex>
  138. // Compute normal from heightmap
  139. vec3 objectNormal = vec3(
  140. ( texture2D( heightmap, uv + vec2( - cellSize.x, 0 ) ).x - texture2D( heightmap, uv + vec2( cellSize.x, 0 ) ).x ) * WIDTH / BOUNDS,
  141. ( texture2D( heightmap, uv + vec2( 0, - cellSize.y ) ).x - texture2D( heightmap, uv + vec2( 0, cellSize.y ) ).x ) * WIDTH / BOUNDS,
  142. 1.0 );
  143. //<beginnormal_vertex>
  144. #include <morphnormal_vertex>
  145. #include <skinbase_vertex>
  146. #include <skinnormal_vertex>
  147. #include <defaultnormal_vertex>
  148. #ifndef FLAT_SHADED // Normal computed with derivatives when FLAT_SHADED
  149. vNormal = normalize( transformedNormal );
  150. #endif
  151. //# include <begin_vertex>
  152. float heightValue = texture2D( heightmap, uv ).x;
  153. vec3 transformed = vec3( position.x, position.y, heightValue );
  154. //<begin_vertex>
  155. #include <morphtarget_vertex>
  156. #include <skinning_vertex>
  157. #include <displacementmap_vertex>
  158. #include <project_vertex>
  159. #include <logdepthbuf_vertex>
  160. #include <clipping_planes_vertex>
  161. vViewPosition = - mvPosition.xyz;
  162. #include <worldpos_vertex>
  163. #include <envmap_vertex>
  164. #include <shadowmap_vertex>
  165. }
  166. </script>
  167. <script type="module">
  168. import * as THREE from '../build/three.module.js';
  169. import Stats from './jsm/libs/stats.module.js';
  170. import { GUI } from './jsm/libs/dat.gui.module.js';
  171. import { OrbitControls } from './jsm/controls/OrbitControls.js';
  172. import { GPUComputationRenderer } from './jsm/misc/GPUComputationRenderer.js';
  173. import { SimplexNoise } from './jsm/math/SimplexNoise.js';
  174. // Texture width for simulation
  175. var WIDTH = 128;
  176. // Water size in system units
  177. var BOUNDS = 512;
  178. var BOUNDS_HALF = BOUNDS * 0.5;
  179. var container, stats;
  180. var camera, scene, renderer;
  181. var mouseMoved = false;
  182. var mouseCoords = new THREE.Vector2();
  183. var raycaster = new THREE.Raycaster();
  184. var waterMesh;
  185. var meshRay;
  186. var gpuCompute;
  187. var heightmapVariable;
  188. var waterUniforms;
  189. var smoothShader;
  190. var readWaterLevelShader;
  191. var readWaterLevelRenderTarget;
  192. var readWaterLevelImage;
  193. var waterNormal = new THREE.Vector3();
  194. var NUM_SPHERES = 5;
  195. var spheres = [];
  196. var spheresEnabled = true;
  197. var simplex = new SimplexNoise();
  198. init();
  199. animate();
  200. function init() {
  201. container = document.createElement( 'div' );
  202. document.body.appendChild( container );
  203. camera = new THREE.PerspectiveCamera( 75, window.innerWidth / window.innerHeight, 1, 3000 );
  204. camera.position.set( 0, 200, 350 );
  205. scene = new THREE.Scene();
  206. var sun = new THREE.DirectionalLight( 0xFFFFFF, 1.0 );
  207. sun.position.set( 300, 400, 175 );
  208. scene.add( sun );
  209. var sun2 = new THREE.DirectionalLight( 0x40A040, 0.6 );
  210. sun2.position.set( - 100, 350, - 200 );
  211. scene.add( sun2 );
  212. renderer = new THREE.WebGLRenderer();
  213. renderer.setPixelRatio( window.devicePixelRatio );
  214. renderer.setSize( window.innerWidth, window.innerHeight );
  215. container.appendChild( renderer.domElement );
  216. var controls = new OrbitControls( camera, renderer.domElement );
  217. controls.minDistance = 100;
  218. controls.maxDistance = 1000;
  219. stats = new Stats();
  220. container.appendChild( stats.dom );
  221. document.addEventListener( 'mousemove', onDocumentMouseMove, false );
  222. document.addEventListener( 'touchstart', onDocumentTouchStart, false );
  223. document.addEventListener( 'touchmove', onDocumentTouchMove, false );
  224. document.addEventListener( 'keydown', function ( event ) {
  225. // W Pressed: Toggle wireframe
  226. if ( event.keyCode === 87 ) {
  227. waterMesh.material.wireframe = ! waterMesh.material.wireframe;
  228. waterMesh.material.needsUpdate = true;
  229. }
  230. }, false );
  231. window.addEventListener( 'resize', onWindowResize, false );
  232. var gui = new GUI();
  233. var effectController = {
  234. mouseSize: 20.0,
  235. viscosity: 0.98,
  236. spheresEnabled: spheresEnabled
  237. };
  238. var valuesChanger = function () {
  239. heightmapVariable.material.uniforms[ "mouseSize" ].value = effectController.mouseSize;
  240. heightmapVariable.material.uniforms[ "viscosityConstant" ].value = effectController.viscosity;
  241. spheresEnabled = effectController.spheresEnabled;
  242. for ( var i = 0; i < NUM_SPHERES; i ++ ) {
  243. if ( spheres[ i ] ) {
  244. spheres[ i ].visible = spheresEnabled;
  245. }
  246. }
  247. };
  248. gui.add( effectController, "mouseSize", 1.0, 100.0, 1.0 ).onChange( valuesChanger );
  249. gui.add( effectController, "viscosity", 0.9, 0.999, 0.001 ).onChange( valuesChanger );
  250. gui.add( effectController, "spheresEnabled", 0, 1, 1 ).onChange( valuesChanger );
  251. var buttonSmooth = {
  252. smoothWater: function () {
  253. smoothWater();
  254. }
  255. };
  256. gui.add( buttonSmooth, 'smoothWater' );
  257. initWater();
  258. createSpheres();
  259. valuesChanger();
  260. }
  261. function initWater() {
  262. var materialColor = 0x0040C0;
  263. var geometry = new THREE.PlaneBufferGeometry( BOUNDS, BOUNDS, WIDTH - 1, WIDTH - 1 );
  264. // material: make a THREE.ShaderMaterial clone of THREE.MeshPhongMaterial, with customized vertex shader
  265. var material = new THREE.ShaderMaterial( {
  266. uniforms: THREE.UniformsUtils.merge( [
  267. THREE.ShaderLib[ 'phong' ].uniforms,
  268. {
  269. "heightmap": { value: null }
  270. }
  271. ] ),
  272. vertexShader: document.getElementById( 'waterVertexShader' ).textContent,
  273. fragmentShader: THREE.ShaderChunk[ 'meshphong_frag' ]
  274. } );
  275. material.lights = true;
  276. // Material attributes from THREE.MeshPhongMaterial
  277. material.color = new THREE.Color( materialColor );
  278. material.specular = new THREE.Color( 0x111111 );
  279. material.shininess = 50;
  280. // Sets the uniforms with the material values
  281. material.uniforms[ "diffuse" ].value = material.color;
  282. material.uniforms[ "specular" ].value = material.specular;
  283. material.uniforms[ "shininess" ].value = Math.max( material.shininess, 1e-4 );
  284. material.uniforms[ "opacity" ].value = material.opacity;
  285. // Defines
  286. material.defines.WIDTH = WIDTH.toFixed( 1 );
  287. material.defines.BOUNDS = BOUNDS.toFixed( 1 );
  288. waterUniforms = material.uniforms;
  289. waterMesh = new THREE.Mesh( geometry, material );
  290. waterMesh.rotation.x = - Math.PI / 2;
  291. waterMesh.matrixAutoUpdate = false;
  292. waterMesh.updateMatrix();
  293. scene.add( waterMesh );
  294. // THREE.Mesh just for mouse raycasting
  295. var geometryRay = new THREE.PlaneBufferGeometry( BOUNDS, BOUNDS, 1, 1 );
  296. meshRay = new THREE.Mesh( geometryRay, new THREE.MeshBasicMaterial( { color: 0xFFFFFF, visible: false } ) );
  297. meshRay.rotation.x = - Math.PI / 2;
  298. meshRay.matrixAutoUpdate = false;
  299. meshRay.updateMatrix();
  300. scene.add( meshRay );
  301. // Creates the gpu computation class and sets it up
  302. gpuCompute = new GPUComputationRenderer( WIDTH, WIDTH, renderer );
  303. if ( isSafari() ) {
  304. gpuCompute.setDataType( THREE.HalfFloatType );
  305. }
  306. var heightmap0 = gpuCompute.createTexture();
  307. fillTexture( heightmap0 );
  308. heightmapVariable = gpuCompute.addVariable( "heightmap", document.getElementById( 'heightmapFragmentShader' ).textContent, heightmap0 );
  309. gpuCompute.setVariableDependencies( heightmapVariable, [ heightmapVariable ] );
  310. heightmapVariable.material.uniforms[ "mousePos" ] = { value: new THREE.Vector2( 10000, 10000 ) };
  311. heightmapVariable.material.uniforms[ "mouseSize" ] = { value: 20.0 };
  312. heightmapVariable.material.uniforms[ "viscosityConstant" ] = { value: 0.98 };
  313. heightmapVariable.material.uniforms[ "heightCompensation" ] = { value: 0 };
  314. heightmapVariable.material.defines.BOUNDS = BOUNDS.toFixed( 1 );
  315. var error = gpuCompute.init();
  316. if ( error !== null ) {
  317. console.error( error );
  318. }
  319. // Create compute shader to smooth the water surface and velocity
  320. smoothShader = gpuCompute.createShaderMaterial( document.getElementById( 'smoothFragmentShader' ).textContent, { smoothTexture: { value: null } } );
  321. // Create compute shader to read water level
  322. readWaterLevelShader = gpuCompute.createShaderMaterial( document.getElementById( 'readWaterLevelFragmentShader' ).textContent, {
  323. point1: { value: new THREE.Vector2() },
  324. levelTexture: { value: null }
  325. } );
  326. readWaterLevelShader.defines.WIDTH = WIDTH.toFixed( 1 );
  327. readWaterLevelShader.defines.BOUNDS = BOUNDS.toFixed( 1 );
  328. // Create a 4x1 pixel image and a render target (Uint8, 4 channels, 1 byte per channel) to read water height and orientation
  329. readWaterLevelImage = new Uint8Array( 4 * 1 * 4 );
  330. readWaterLevelRenderTarget = new THREE.WebGLRenderTarget( 4, 1, {
  331. wrapS: THREE.ClampToEdgeWrapping,
  332. wrapT: THREE.ClampToEdgeWrapping,
  333. minFilter: THREE.NearestFilter,
  334. magFilter: THREE.NearestFilter,
  335. format: THREE.RGBAFormat,
  336. type: THREE.UnsignedByteType,
  337. depthBuffer: false
  338. } );
  339. }
  340. function isSafari() {
  341. return !! navigator.userAgent.match( /Safari/i ) && ! navigator.userAgent.match( /Chrome/i );
  342. }
  343. function fillTexture( texture ) {
  344. var waterMaxHeight = 10;
  345. function noise( x, y ) {
  346. var multR = waterMaxHeight;
  347. var mult = 0.025;
  348. var r = 0;
  349. for ( var i = 0; i < 15; i ++ ) {
  350. r += multR * simplex.noise( x * mult, y * mult );
  351. multR *= 0.53 + 0.025 * i;
  352. mult *= 1.25;
  353. }
  354. return r;
  355. }
  356. var pixels = texture.image.data;
  357. var p = 0;
  358. for ( var j = 0; j < WIDTH; j ++ ) {
  359. for ( var i = 0; i < WIDTH; i ++ ) {
  360. var x = i * 128 / WIDTH;
  361. var y = j * 128 / WIDTH;
  362. pixels[ p + 0 ] = noise( x, y );
  363. pixels[ p + 1 ] = pixels[ p + 0 ];
  364. pixels[ p + 2 ] = 0;
  365. pixels[ p + 3 ] = 1;
  366. p += 4;
  367. }
  368. }
  369. }
  370. function smoothWater() {
  371. var currentRenderTarget = gpuCompute.getCurrentRenderTarget( heightmapVariable );
  372. var alternateRenderTarget = gpuCompute.getAlternateRenderTarget( heightmapVariable );
  373. for ( var i = 0; i < 10; i ++ ) {
  374. smoothShader.uniforms[ "smoothTexture" ].value = currentRenderTarget.texture;
  375. gpuCompute.doRenderTarget( smoothShader, alternateRenderTarget );
  376. smoothShader.uniforms[ "smoothTexture" ].value = alternateRenderTarget.texture;
  377. gpuCompute.doRenderTarget( smoothShader, currentRenderTarget );
  378. }
  379. }
  380. function createSpheres() {
  381. var sphereTemplate = new THREE.Mesh( new THREE.SphereBufferGeometry( 4, 24, 12 ), new THREE.MeshPhongMaterial( { color: 0xFFFF00 } ) );
  382. for ( var i = 0; i < NUM_SPHERES; i ++ ) {
  383. var sphere = sphereTemplate;
  384. if ( i < NUM_SPHERES - 1 ) {
  385. sphere = sphereTemplate.clone();
  386. }
  387. sphere.position.x = ( Math.random() - 0.5 ) * BOUNDS * 0.7;
  388. sphere.position.z = ( Math.random() - 0.5 ) * BOUNDS * 0.7;
  389. sphere.userData.velocity = new THREE.Vector3();
  390. scene.add( sphere );
  391. spheres[ i ] = sphere;
  392. }
  393. }
  394. function sphereDynamics() {
  395. var currentRenderTarget = gpuCompute.getCurrentRenderTarget( heightmapVariable );
  396. readWaterLevelShader.uniforms[ "levelTexture" ].value = currentRenderTarget.texture;
  397. for ( var i = 0; i < NUM_SPHERES; i ++ ) {
  398. var sphere = spheres[ i ];
  399. if ( sphere ) {
  400. // Read water level and orientation
  401. var u = 0.5 * sphere.position.x / BOUNDS_HALF + 0.5;
  402. var v = 1 - ( 0.5 * sphere.position.z / BOUNDS_HALF + 0.5 );
  403. readWaterLevelShader.uniforms[ "point1" ].value.set( u, v );
  404. gpuCompute.doRenderTarget( readWaterLevelShader, readWaterLevelRenderTarget );
  405. renderer.readRenderTargetPixels( readWaterLevelRenderTarget, 0, 0, 4, 1, readWaterLevelImage );
  406. var pixels = new Float32Array( readWaterLevelImage.buffer );
  407. // Get orientation
  408. waterNormal.set( pixels[ 1 ], 0, - pixels[ 2 ] );
  409. var pos = sphere.position;
  410. // Set height
  411. pos.y = pixels[ 0 ];
  412. // Move sphere
  413. waterNormal.multiplyScalar( 0.1 );
  414. sphere.userData.velocity.add( waterNormal );
  415. sphere.userData.velocity.multiplyScalar( 0.998 );
  416. pos.add( sphere.userData.velocity );
  417. if ( pos.x < - BOUNDS_HALF ) {
  418. pos.x = - BOUNDS_HALF + 0.001;
  419. sphere.userData.velocity.x *= - 0.3;
  420. } else if ( pos.x > BOUNDS_HALF ) {
  421. pos.x = BOUNDS_HALF - 0.001;
  422. sphere.userData.velocity.x *= - 0.3;
  423. }
  424. if ( pos.z < - BOUNDS_HALF ) {
  425. pos.z = - BOUNDS_HALF + 0.001;
  426. sphere.userData.velocity.z *= - 0.3;
  427. } else if ( pos.z > BOUNDS_HALF ) {
  428. pos.z = BOUNDS_HALF - 0.001;
  429. sphere.userData.velocity.z *= - 0.3;
  430. }
  431. }
  432. }
  433. }
  434. function onWindowResize() {
  435. camera.aspect = window.innerWidth / window.innerHeight;
  436. camera.updateProjectionMatrix();
  437. renderer.setSize( window.innerWidth, window.innerHeight );
  438. }
  439. function setMouseCoords( x, y ) {
  440. mouseCoords.set( ( x / renderer.domElement.clientWidth ) * 2 - 1, - ( y / renderer.domElement.clientHeight ) * 2 + 1 );
  441. mouseMoved = true;
  442. }
  443. function onDocumentMouseMove( event ) {
  444. setMouseCoords( event.clientX, event.clientY );
  445. }
  446. function onDocumentTouchStart( event ) {
  447. if ( event.touches.length === 1 ) {
  448. event.preventDefault();
  449. setMouseCoords( event.touches[ 0 ].pageX, event.touches[ 0 ].pageY );
  450. }
  451. }
  452. function onDocumentTouchMove( event ) {
  453. if ( event.touches.length === 1 ) {
  454. event.preventDefault();
  455. setMouseCoords( event.touches[ 0 ].pageX, event.touches[ 0 ].pageY );
  456. }
  457. }
  458. function animate() {
  459. requestAnimationFrame( animate );
  460. render();
  461. stats.update();
  462. }
  463. function render() {
  464. // Set uniforms: mouse interaction
  465. var uniforms = heightmapVariable.material.uniforms;
  466. if ( mouseMoved ) {
  467. raycaster.setFromCamera( mouseCoords, camera );
  468. var intersects = raycaster.intersectObject( meshRay );
  469. if ( intersects.length > 0 ) {
  470. var point = intersects[ 0 ].point;
  471. uniforms[ "mousePos" ].value.set( point.x, point.z );
  472. } else {
  473. uniforms[ "mousePos" ].value.set( 10000, 10000 );
  474. }
  475. mouseMoved = false;
  476. } else {
  477. uniforms[ "mousePos" ].value.set( 10000, 10000 );
  478. }
  479. // Do the gpu computation
  480. gpuCompute.compute();
  481. if ( spheresEnabled ) {
  482. sphereDynamics();
  483. }
  484. // Get compute output in custom uniform
  485. waterUniforms[ "heightmap" ].value = gpuCompute.getCurrentRenderTarget( heightmapVariable ).texture;
  486. // Render
  487. renderer.render( scene, camera );
  488. }
  489. </script>
  490. </body>
  491. </html>