webgl_gpgpu_water.html 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749
  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. stats = new Stats();
  218. container.appendChild( stats.dom );
  219. document.addEventListener( 'mousemove', onDocumentMouseMove, false );
  220. document.addEventListener( 'touchstart', onDocumentTouchStart, false );
  221. document.addEventListener( 'touchmove', onDocumentTouchMove, false );
  222. document.addEventListener( 'keydown', function ( event ) {
  223. // W Pressed: Toggle wireframe
  224. if ( event.keyCode === 87 ) {
  225. waterMesh.material.wireframe = ! waterMesh.material.wireframe;
  226. waterMesh.material.needsUpdate = true;
  227. }
  228. }, false );
  229. window.addEventListener( 'resize', onWindowResize, false );
  230. var gui = new GUI();
  231. var effectController = {
  232. mouseSize: 20.0,
  233. viscosity: 0.98,
  234. spheresEnabled: spheresEnabled
  235. };
  236. var valuesChanger = function () {
  237. heightmapVariable.material.uniforms[ "mouseSize" ].value = effectController.mouseSize;
  238. heightmapVariable.material.uniforms[ "viscosityConstant" ].value = effectController.viscosity;
  239. spheresEnabled = effectController.spheresEnabled;
  240. for ( var i = 0; i < NUM_SPHERES; i ++ ) {
  241. if ( spheres[ i ] ) {
  242. spheres[ i ].visible = spheresEnabled;
  243. }
  244. }
  245. };
  246. gui.add( effectController, "mouseSize", 1.0, 100.0, 1.0 ).onChange( valuesChanger );
  247. gui.add( effectController, "viscosity", 0.9, 0.999, 0.001 ).onChange( valuesChanger );
  248. gui.add( effectController, "spheresEnabled", 0, 1, 1 ).onChange( valuesChanger );
  249. var buttonSmooth = {
  250. smoothWater: function () {
  251. smoothWater();
  252. }
  253. };
  254. gui.add( buttonSmooth, 'smoothWater' );
  255. initWater();
  256. createSpheres();
  257. valuesChanger();
  258. }
  259. function initWater() {
  260. var materialColor = 0x0040C0;
  261. var geometry = new THREE.PlaneBufferGeometry( BOUNDS, BOUNDS, WIDTH - 1, WIDTH - 1 );
  262. // material: make a THREE.ShaderMaterial clone of THREE.MeshPhongMaterial, with customized vertex shader
  263. var material = new THREE.ShaderMaterial( {
  264. uniforms: THREE.UniformsUtils.merge( [
  265. THREE.ShaderLib[ 'phong' ].uniforms,
  266. {
  267. "heightmap": { value: null }
  268. }
  269. ] ),
  270. vertexShader: document.getElementById( 'waterVertexShader' ).textContent,
  271. fragmentShader: THREE.ShaderChunk[ 'meshphong_frag' ]
  272. } );
  273. material.lights = true;
  274. // Material attributes from THREE.MeshPhongMaterial
  275. material.color = new THREE.Color( materialColor );
  276. material.specular = new THREE.Color( 0x111111 );
  277. material.shininess = 50;
  278. // Sets the uniforms with the material values
  279. material.uniforms[ "diffuse" ].value = material.color;
  280. material.uniforms[ "specular" ].value = material.specular;
  281. material.uniforms[ "shininess" ].value = Math.max( material.shininess, 1e-4 );
  282. material.uniforms[ "opacity" ].value = material.opacity;
  283. // Defines
  284. material.defines.WIDTH = WIDTH.toFixed( 1 );
  285. material.defines.BOUNDS = BOUNDS.toFixed( 1 );
  286. waterUniforms = material.uniforms;
  287. waterMesh = new THREE.Mesh( geometry, material );
  288. waterMesh.rotation.x = - Math.PI / 2;
  289. waterMesh.matrixAutoUpdate = false;
  290. waterMesh.updateMatrix();
  291. scene.add( waterMesh );
  292. // THREE.Mesh just for mouse raycasting
  293. var geometryRay = new THREE.PlaneBufferGeometry( BOUNDS, BOUNDS, 1, 1 );
  294. meshRay = new THREE.Mesh( geometryRay, new THREE.MeshBasicMaterial( { color: 0xFFFFFF, visible: false } ) );
  295. meshRay.rotation.x = - Math.PI / 2;
  296. meshRay.matrixAutoUpdate = false;
  297. meshRay.updateMatrix();
  298. scene.add( meshRay );
  299. // Creates the gpu computation class and sets it up
  300. gpuCompute = new GPUComputationRenderer( WIDTH, WIDTH, renderer );
  301. var heightmap0 = gpuCompute.createTexture();
  302. fillTexture( heightmap0 );
  303. heightmapVariable = gpuCompute.addVariable( "heightmap", document.getElementById( 'heightmapFragmentShader' ).textContent, heightmap0 );
  304. gpuCompute.setVariableDependencies( heightmapVariable, [ heightmapVariable ] );
  305. heightmapVariable.material.uniforms[ "mousePos" ] = { value: new THREE.Vector2( 10000, 10000 ) };
  306. heightmapVariable.material.uniforms[ "mouseSize" ] = { value: 20.0 };
  307. heightmapVariable.material.uniforms[ "viscosityConstant" ] = { value: 0.98 };
  308. heightmapVariable.material.uniforms[ "heightCompensation" ] = { value: 0 };
  309. heightmapVariable.material.defines.BOUNDS = BOUNDS.toFixed( 1 );
  310. var error = gpuCompute.init();
  311. if ( error !== null ) {
  312. console.error( error );
  313. }
  314. // Create compute shader to smooth the water surface and velocity
  315. smoothShader = gpuCompute.createShaderMaterial( document.getElementById( 'smoothFragmentShader' ).textContent, { smoothTexture: { value: null } } );
  316. // Create compute shader to read water level
  317. readWaterLevelShader = gpuCompute.createShaderMaterial( document.getElementById( 'readWaterLevelFragmentShader' ).textContent, {
  318. point1: { value: new THREE.Vector2() },
  319. levelTexture: { value: null }
  320. } );
  321. readWaterLevelShader.defines.WIDTH = WIDTH.toFixed( 1 );
  322. readWaterLevelShader.defines.BOUNDS = BOUNDS.toFixed( 1 );
  323. // Create a 4x1 pixel image and a render target (Uint8, 4 channels, 1 byte per channel) to read water height and orientation
  324. readWaterLevelImage = new Uint8Array( 4 * 1 * 4 );
  325. readWaterLevelRenderTarget = new THREE.WebGLRenderTarget( 4, 1, {
  326. wrapS: THREE.ClampToEdgeWrapping,
  327. wrapT: THREE.ClampToEdgeWrapping,
  328. minFilter: THREE.NearestFilter,
  329. magFilter: THREE.NearestFilter,
  330. format: THREE.RGBAFormat,
  331. type: THREE.UnsignedByteType,
  332. stencilBuffer: false,
  333. depthBuffer: false
  334. } );
  335. }
  336. function fillTexture( texture ) {
  337. var waterMaxHeight = 10;
  338. function noise( x, y ) {
  339. var multR = waterMaxHeight;
  340. var mult = 0.025;
  341. var r = 0;
  342. for ( var i = 0; i < 15; i ++ ) {
  343. r += multR * simplex.noise( x * mult, y * mult );
  344. multR *= 0.53 + 0.025 * i;
  345. mult *= 1.25;
  346. }
  347. return r;
  348. }
  349. var pixels = texture.image.data;
  350. var p = 0;
  351. for ( var j = 0; j < WIDTH; j ++ ) {
  352. for ( var i = 0; i < WIDTH; i ++ ) {
  353. var x = i * 128 / WIDTH;
  354. var y = j * 128 / WIDTH;
  355. pixels[ p + 0 ] = noise( x, y );
  356. pixels[ p + 1 ] = pixels[ p + 0 ];
  357. pixels[ p + 2 ] = 0;
  358. pixels[ p + 3 ] = 1;
  359. p += 4;
  360. }
  361. }
  362. }
  363. function smoothWater() {
  364. var currentRenderTarget = gpuCompute.getCurrentRenderTarget( heightmapVariable );
  365. var alternateRenderTarget = gpuCompute.getAlternateRenderTarget( heightmapVariable );
  366. for ( var i = 0; i < 10; i ++ ) {
  367. smoothShader.uniforms[ "smoothTexture" ].value = currentRenderTarget.texture;
  368. gpuCompute.doRenderTarget( smoothShader, alternateRenderTarget );
  369. smoothShader.uniforms[ "smoothTexture" ].value = alternateRenderTarget.texture;
  370. gpuCompute.doRenderTarget( smoothShader, currentRenderTarget );
  371. }
  372. }
  373. function createSpheres() {
  374. var sphereTemplate = new THREE.Mesh( new THREE.SphereBufferGeometry( 4, 24, 12 ), new THREE.MeshPhongMaterial( { color: 0xFFFF00 } ) );
  375. for ( var i = 0; i < NUM_SPHERES; i ++ ) {
  376. var sphere = sphereTemplate;
  377. if ( i < NUM_SPHERES - 1 ) {
  378. sphere = sphereTemplate.clone();
  379. }
  380. sphere.position.x = ( Math.random() - 0.5 ) * BOUNDS * 0.7;
  381. sphere.position.z = ( Math.random() - 0.5 ) * BOUNDS * 0.7;
  382. sphere.userData.velocity = new THREE.Vector3();
  383. scene.add( sphere );
  384. spheres[ i ] = sphere;
  385. }
  386. }
  387. function sphereDynamics() {
  388. var currentRenderTarget = gpuCompute.getCurrentRenderTarget( heightmapVariable );
  389. readWaterLevelShader.uniforms[ "levelTexture" ].value = currentRenderTarget.texture;
  390. for ( var i = 0; i < NUM_SPHERES; i ++ ) {
  391. var sphere = spheres[ i ];
  392. if ( sphere ) {
  393. // Read water level and orientation
  394. var u = 0.5 * sphere.position.x / BOUNDS_HALF + 0.5;
  395. var v = 1 - ( 0.5 * sphere.position.z / BOUNDS_HALF + 0.5 );
  396. readWaterLevelShader.uniforms[ "point1" ].value.set( u, v );
  397. gpuCompute.doRenderTarget( readWaterLevelShader, readWaterLevelRenderTarget );
  398. renderer.readRenderTargetPixels( readWaterLevelRenderTarget, 0, 0, 4, 1, readWaterLevelImage );
  399. var pixels = new Float32Array( readWaterLevelImage.buffer );
  400. // Get orientation
  401. waterNormal.set( pixels[ 1 ], 0, - pixels[ 2 ] );
  402. var pos = sphere.position;
  403. // Set height
  404. pos.y = pixels[ 0 ];
  405. // Move sphere
  406. waterNormal.multiplyScalar( 0.1 );
  407. sphere.userData.velocity.add( waterNormal );
  408. sphere.userData.velocity.multiplyScalar( 0.998 );
  409. pos.add( sphere.userData.velocity );
  410. if ( pos.x < - BOUNDS_HALF ) {
  411. pos.x = - BOUNDS_HALF + 0.001;
  412. sphere.userData.velocity.x *= - 0.3;
  413. } else if ( pos.x > BOUNDS_HALF ) {
  414. pos.x = BOUNDS_HALF - 0.001;
  415. sphere.userData.velocity.x *= - 0.3;
  416. }
  417. if ( pos.z < - BOUNDS_HALF ) {
  418. pos.z = - BOUNDS_HALF + 0.001;
  419. sphere.userData.velocity.z *= - 0.3;
  420. } else if ( pos.z > BOUNDS_HALF ) {
  421. pos.z = BOUNDS_HALF - 0.001;
  422. sphere.userData.velocity.z *= - 0.3;
  423. }
  424. }
  425. }
  426. }
  427. function onWindowResize() {
  428. camera.aspect = window.innerWidth / window.innerHeight;
  429. camera.updateProjectionMatrix();
  430. renderer.setSize( window.innerWidth, window.innerHeight );
  431. }
  432. function setMouseCoords( x, y ) {
  433. mouseCoords.set( ( x / renderer.domElement.clientWidth ) * 2 - 1, - ( y / renderer.domElement.clientHeight ) * 2 + 1 );
  434. mouseMoved = true;
  435. }
  436. function onDocumentMouseMove( event ) {
  437. setMouseCoords( event.clientX, event.clientY );
  438. }
  439. function onDocumentTouchStart( event ) {
  440. if ( event.touches.length === 1 ) {
  441. event.preventDefault();
  442. setMouseCoords( event.touches[ 0 ].pageX, event.touches[ 0 ].pageY );
  443. }
  444. }
  445. function onDocumentTouchMove( event ) {
  446. if ( event.touches.length === 1 ) {
  447. event.preventDefault();
  448. setMouseCoords( event.touches[ 0 ].pageX, event.touches[ 0 ].pageY );
  449. }
  450. }
  451. function animate() {
  452. requestAnimationFrame( animate );
  453. render();
  454. stats.update();
  455. }
  456. function render() {
  457. // Set uniforms: mouse interaction
  458. var uniforms = heightmapVariable.material.uniforms;
  459. if ( mouseMoved ) {
  460. raycaster.setFromCamera( mouseCoords, camera );
  461. var intersects = raycaster.intersectObject( meshRay );
  462. if ( intersects.length > 0 ) {
  463. var point = intersects[ 0 ].point;
  464. uniforms[ "mousePos" ].value.set( point.x, point.z );
  465. } else {
  466. uniforms[ "mousePos" ].value.set( 10000, 10000 );
  467. }
  468. mouseMoved = false;
  469. } else {
  470. uniforms[ "mousePos" ].value.set( 10000, 10000 );
  471. }
  472. // Do the gpu computation
  473. gpuCompute.compute();
  474. if ( spheresEnabled ) {
  475. sphereDynamics();
  476. }
  477. // Get compute output in custom uniform
  478. waterUniforms[ "heightmap" ].value = gpuCompute.getCurrentRenderTarget( heightmapVariable ).texture;
  479. // Render
  480. renderer.render( scene, camera );
  481. }
  482. </script>
  483. </body>
  484. </html>