webgpu_volume_cloud.html 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgpu - volumetric cloud</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> webgpu - volumetric cloud
  12. </div>
  13. <script type="importmap">
  14. {
  15. "imports": {
  16. "three": "../build/three.webgpu.js",
  17. "three/webgpu": "../build/three.webgpu.js",
  18. "three/tsl": "../build/three.tsl.js",
  19. "three/addons/": "./jsm/"
  20. }
  21. }
  22. </script>
  23. <script type="module">
  24. import * as THREE from 'three';
  25. import { texture3D, uniform } from 'three/tsl';
  26. import { raymarchingTexture3D } from 'three/addons/tsl/utils/Raymarching.js';
  27. import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
  28. import { ImprovedNoise } from 'three/addons/math/ImprovedNoise.js';
  29. import { GUI } from 'three/addons/libs/lil-gui.module.min.js';
  30. let renderer, scene, camera;
  31. let mesh;
  32. init();
  33. function init() {
  34. renderer = new THREE.WebGPURenderer( { antialias: true } );
  35. renderer.setPixelRatio( window.devicePixelRatio );
  36. renderer.setSize( window.innerWidth, window.innerHeight );
  37. renderer.setAnimationLoop( animate );
  38. document.body.appendChild( renderer.domElement );
  39. scene = new THREE.Scene();
  40. camera = new THREE.PerspectiveCamera( 60, window.innerWidth / window.innerHeight, 0.1, 100 );
  41. camera.position.set( 0, 0, 1.5 );
  42. new OrbitControls( camera, renderer.domElement );
  43. // Sky
  44. const canvas = document.createElement( 'canvas' );
  45. canvas.width = 1;
  46. canvas.height = 32;
  47. const context = canvas.getContext( '2d' );
  48. const gradient = context.createLinearGradient( 0, 0, 0, 32 );
  49. gradient.addColorStop( 0.0, '#014a84' );
  50. gradient.addColorStop( 0.5, '#0561a0' );
  51. gradient.addColorStop( 1.0, '#437ab6' );
  52. context.fillStyle = gradient;
  53. context.fillRect( 0, 0, 1, 32 );
  54. const skyMap = new THREE.CanvasTexture( canvas );
  55. skyMap.colorSpace = THREE.SRGBColorSpace;
  56. const sky = new THREE.Mesh(
  57. new THREE.SphereGeometry( 10 ),
  58. new THREE.MeshBasicNodeMaterial( { map: skyMap, side: THREE.BackSide } )
  59. );
  60. scene.add( sky );
  61. // Texture
  62. const size = 128;
  63. const data = new Uint8Array( size * size * size );
  64. let i = 0;
  65. const scale = 0.05;
  66. const perlin = new ImprovedNoise();
  67. const vector = new THREE.Vector3();
  68. for ( let z = 0; z < size; z ++ ) {
  69. for ( let y = 0; y < size; y ++ ) {
  70. for ( let x = 0; x < size; x ++ ) {
  71. const d = 1.0 - vector.set( x, y, z ).subScalar( size / 2 ).divideScalar( size ).length();
  72. data[ i ] = ( 128 + 128 * perlin.noise( x * scale / 1.5, y * scale, z * scale / 1.5 ) ) * d * d;
  73. i ++;
  74. }
  75. }
  76. }
  77. const texture = new THREE.Data3DTexture( data, size, size, size );
  78. texture.format = THREE.RedFormat;
  79. texture.minFilter = THREE.LinearFilter;
  80. texture.magFilter = THREE.LinearFilter;
  81. texture.unpackAlignment = 1;
  82. texture.needsUpdate = true;
  83. // Cloud Shader
  84. const baseColor = uniform( new THREE.Color( 0x798aa0 ) );
  85. const range = uniform( 0.1 );
  86. const threshold = uniform( 0.25 );
  87. const opacity = uniform( 0.25 );
  88. const steps = uniform( 100 );
  89. const cloud3d = raymarchingTexture3D( {
  90. texture: texture3D( texture, null, 0 ),
  91. range: range,
  92. threshold: threshold,
  93. opacity: opacity,
  94. steps: steps
  95. } );
  96. const finalCloud = cloud3d.setRGB( cloud3d.rgb.add( baseColor ) );
  97. //
  98. const geometry = new THREE.BoxGeometry( 1, 1, 1 );
  99. const material = new THREE.NodeMaterial();
  100. material.colorNode = finalCloud;
  101. material.side = THREE.BackSide;
  102. material.transparent = true;
  103. mesh = new THREE.Mesh( geometry, material );
  104. scene.add( mesh );
  105. //
  106. const gui = new GUI();
  107. gui.add( threshold, 'value', 0, 1, 0.01 ).name( 'threshold' );
  108. gui.add( opacity, 'value', 0, 1, 0.01 ).name( 'opacity' );
  109. gui.add( range, 'value', 0, 1, 0.01 ).name( 'range' );
  110. gui.add( steps, 'value', 0, 200, 1 ).name( 'steps' );
  111. window.addEventListener( 'resize', onWindowResize );
  112. }
  113. function onWindowResize() {
  114. camera.aspect = window.innerWidth / window.innerHeight;
  115. camera.updateProjectionMatrix();
  116. renderer.setSize( window.innerWidth, window.innerHeight );
  117. }
  118. function animate() {
  119. mesh.rotation.y = - performance.now() / 7500;
  120. renderer.render( scene, camera );
  121. }
  122. </script>
  123. </body>
  124. </html>
粤ICP备19079148号