webgl_depth_texture.html 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - Depth Texture</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. #error {
  10. margin: auto;
  11. margin-top: 40px;
  12. display: block;
  13. max-width: 400px;
  14. padding: 20px;
  15. background: #CE0808;
  16. }
  17. </style>
  18. <script id="post-vert" type="x-shader/x-vertex">
  19. varying vec2 vUv;
  20. void main() {
  21. vUv = uv;
  22. gl_Position = projectionMatrix * modelViewMatrix * vec4(position, 1.0);
  23. }
  24. </script>
  25. <script id="post-frag" type="x-shader/x-fragment">
  26. #include <packing>
  27. varying vec2 vUv;
  28. uniform sampler2D tDiffuse;
  29. uniform sampler2D tDepth;
  30. uniform float cameraNear;
  31. uniform float cameraFar;
  32. float readDepth( sampler2D depthSampler, vec2 coord ) {
  33. float fragCoordZ = texture2D( depthSampler, coord ).x;
  34. float viewZ = perspectiveDepthToViewZ( fragCoordZ, cameraNear, cameraFar );
  35. return viewZToOrthographicDepth( viewZ, cameraNear, cameraFar );
  36. }
  37. void main() {
  38. //vec3 diffuse = texture2D( tDiffuse, vUv ).rgb;
  39. float depth = readDepth( tDepth, vUv );
  40. gl_FragColor.rgb = 1.0 - vec3( depth );
  41. gl_FragColor.a = 1.0;
  42. }
  43. </script>
  44. </head>
  45. <body>
  46. <div id="info">
  47. <a href="https://threejs.org" target="_blank" rel="noopener">threejs</a> webgl - depth texture<br/>
  48. Stores render target depth in a texture attachment.<br/>
  49. Created by <a href="http://twitter.com/mattdesl" target="_blank" rel="noopener">@mattdesl</a>.
  50. <div id="error" style="display: none;">
  51. Your browser does not support <strong>WEBGL_depth_texture</strong>.<br/><br/>
  52. This demo will not work.
  53. </div>
  54. </div>
  55. <script type="importmap">
  56. {
  57. "imports": {
  58. "three": "../build/three.module.js",
  59. "three/addons/": "./jsm/"
  60. }
  61. }
  62. </script>
  63. <script type="module">
  64. import * as THREE from 'three';
  65. import Stats from 'three/addons/libs/stats.module.js';
  66. import { GUI } from 'three/addons/libs/lil-gui.module.min.js';
  67. import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
  68. let camera, scene, renderer, controls, stats;
  69. let target;
  70. let postScene, postCamera, postMaterial;
  71. const params = {
  72. format: THREE.DepthFormat,
  73. type: THREE.UnsignedShortType
  74. };
  75. const formats = { DepthFormat: THREE.DepthFormat, DepthStencilFormat: THREE.DepthStencilFormat };
  76. const types = { UnsignedShortType: THREE.UnsignedShortType, UnsignedIntType: THREE.UnsignedIntType, UnsignedInt248Type: THREE.UnsignedInt248Type };
  77. init();
  78. function init() {
  79. renderer = new THREE.WebGLRenderer();
  80. renderer.setPixelRatio( window.devicePixelRatio );
  81. renderer.setSize( window.innerWidth, window.innerHeight );
  82. renderer.setAnimationLoop( animate );
  83. document.body.appendChild( renderer.domElement );
  84. //
  85. stats = new Stats();
  86. document.body.appendChild( stats.dom );
  87. camera = new THREE.PerspectiveCamera( 70, window.innerWidth / window.innerHeight, 0.01, 50 );
  88. camera.position.z = 4;
  89. controls = new OrbitControls( camera, renderer.domElement );
  90. controls.enableDamping = true;
  91. // Create a render target with depth texture
  92. setupRenderTarget();
  93. // Our scene
  94. setupScene();
  95. // Setup post-processing step
  96. setupPost();
  97. onWindowResize();
  98. window.addEventListener( 'resize', onWindowResize );
  99. //
  100. const gui = new GUI( { width: 300 } );
  101. gui.add( params, 'format', formats ).onChange( setupRenderTarget );
  102. gui.add( params, 'type', types ).onChange( setupRenderTarget );
  103. gui.open();
  104. }
  105. function setupRenderTarget() {
  106. if ( target ) target.dispose();
  107. const format = parseFloat( params.format );
  108. const type = parseFloat( params.type );
  109. target = new THREE.WebGLRenderTarget( window.innerWidth, window.innerHeight );
  110. target.texture.minFilter = THREE.NearestFilter;
  111. target.texture.magFilter = THREE.NearestFilter;
  112. target.stencilBuffer = ( format === THREE.DepthStencilFormat ) ? true : false;
  113. target.depthTexture = new THREE.DepthTexture();
  114. target.depthTexture.format = format;
  115. target.depthTexture.type = type;
  116. }
  117. function setupPost() {
  118. // Setup post processing stage
  119. postCamera = new THREE.OrthographicCamera( - 1, 1, 1, - 1, 0, 1 );
  120. postMaterial = new THREE.ShaderMaterial( {
  121. vertexShader: document.querySelector( '#post-vert' ).textContent.trim(),
  122. fragmentShader: document.querySelector( '#post-frag' ).textContent.trim(),
  123. uniforms: {
  124. cameraNear: { value: camera.near },
  125. cameraFar: { value: camera.far },
  126. tDiffuse: { value: null },
  127. tDepth: { value: null }
  128. }
  129. } );
  130. const postPlane = new THREE.PlaneGeometry( 2, 2 );
  131. const postQuad = new THREE.Mesh( postPlane, postMaterial );
  132. postScene = new THREE.Scene();
  133. postScene.add( postQuad );
  134. }
  135. function setupScene() {
  136. scene = new THREE.Scene();
  137. const geometry = new THREE.TorusKnotGeometry( 1, 0.3, 128, 64 );
  138. const material = new THREE.MeshBasicMaterial( { color: 'blue' } );
  139. const count = 50;
  140. const scale = 5;
  141. for ( let i = 0; i < count; i ++ ) {
  142. const r = Math.random() * 2.0 * Math.PI;
  143. const z = ( Math.random() * 2.0 ) - 1.0;
  144. const zScale = Math.sqrt( 1.0 - z * z ) * scale;
  145. const mesh = new THREE.Mesh( geometry, material );
  146. mesh.position.set(
  147. Math.cos( r ) * zScale,
  148. Math.sin( r ) * zScale,
  149. z * scale
  150. );
  151. mesh.rotation.set( Math.random(), Math.random(), Math.random() );
  152. scene.add( mesh );
  153. }
  154. }
  155. function onWindowResize() {
  156. const aspect = window.innerWidth / window.innerHeight;
  157. camera.aspect = aspect;
  158. camera.updateProjectionMatrix();
  159. const dpr = renderer.getPixelRatio();
  160. target.setSize( window.innerWidth * dpr, window.innerHeight * dpr );
  161. renderer.setSize( window.innerWidth, window.innerHeight );
  162. }
  163. function animate() {
  164. // render scene into target
  165. renderer.setRenderTarget( target );
  166. renderer.render( scene, camera );
  167. // render post FX
  168. postMaterial.uniforms.tDiffuse.value = target.texture;
  169. postMaterial.uniforms.tDepth.value = target.depthTexture;
  170. renderer.setRenderTarget( null );
  171. renderer.render( postScene, postCamera );
  172. controls.update(); // required because damping is enabled
  173. stats.update();
  174. }
  175. </script>
  176. </body>
  177. </html>