webgl_postprocessing_gtao.html 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - postprocessing - GTAO</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: #ffffff;
  11. color: #000;
  12. }
  13. a {
  14. color: #2983ff;
  15. }
  16. </style>
  17. </head>
  18. <body>
  19. <div id="info">
  20. <a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> - Ground Truth Ambient Occlusion (GTAO) by <a href="https://github.com/Rabbid76" target="_blank" rel="noopener">Rabbid76</a><br/>
  21. </div>
  22. <script type="importmap">
  23. {
  24. "imports": {
  25. "three": "../build/three.module.js",
  26. "three/addons/": "./jsm/"
  27. }
  28. }
  29. </script>
  30. <script type="module">
  31. import * as THREE from 'three';
  32. import Stats from 'three/addons/libs/stats.module.js';
  33. import { GUI } from 'three/addons/libs/lil-gui.module.min.js';
  34. import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
  35. import { RoomEnvironment } from 'three/addons/environments/RoomEnvironment.js';
  36. import { GLTFLoader } from 'three/addons/loaders/GLTFLoader.js';
  37. import { DRACOLoader } from 'three/addons/loaders/DRACOLoader.js';
  38. import { EffectComposer } from 'three/addons/postprocessing/EffectComposer.js';
  39. import { RenderPass } from 'three/addons/postprocessing/RenderPass.js';
  40. import { GTAOPass } from 'three/addons/postprocessing/GTAOPass.js';
  41. import { OutputPass } from 'three/addons/postprocessing/OutputPass.js';
  42. let camera, scene, renderer, composer, controls, clock, stats, mixer;
  43. init();
  44. function init() {
  45. const dracoLoader = new DRACOLoader();
  46. dracoLoader.setDecoderPath( 'jsm/libs/draco/' );
  47. dracoLoader.setDecoderConfig( { type: 'js' } );
  48. const loader = new GLTFLoader();
  49. loader.setDRACOLoader( dracoLoader );
  50. loader.setPath( 'models/gltf/' );
  51. clock = new THREE.Clock();
  52. const container = document.createElement( 'div' );
  53. document.body.appendChild( container );
  54. stats = new Stats();
  55. container.appendChild( stats.dom );
  56. renderer = new THREE.WebGLRenderer();
  57. renderer.setSize( window.innerWidth, window.innerHeight );
  58. renderer.setAnimationLoop( animate );
  59. document.body.appendChild( renderer.domElement );
  60. const pmremGenerator = new THREE.PMREMGenerator( renderer );
  61. scene = new THREE.Scene();
  62. scene.background = new THREE.Color( 0xbfe3dd );
  63. scene.environment = pmremGenerator.fromScene( new RoomEnvironment(), 0.04 ).texture;
  64. camera = new THREE.PerspectiveCamera( 40, window.innerWidth / window.innerHeight, 1, 100 );
  65. camera.position.set( 5, 2, 8 );
  66. controls = new OrbitControls( camera, renderer.domElement );
  67. controls.target.set( 0, 0.5, 0 );
  68. controls.update();
  69. controls.enablePan = false;
  70. controls.enableDamping = true;
  71. const width = window.innerWidth;
  72. const height = window.innerHeight;
  73. composer = new EffectComposer( renderer );
  74. const renderPass = new RenderPass( scene, camera );
  75. composer.addPass( renderPass );
  76. const gtaoPass = new GTAOPass( scene, camera, width, height );
  77. gtaoPass.output = GTAOPass.OUTPUT.Denoise;
  78. composer.addPass( gtaoPass );
  79. const outputPass = new OutputPass();
  80. composer.addPass( outputPass );
  81. //
  82. loader.load( 'LittlestTokyo.glb', ( gltf ) => {
  83. const model = gltf.scene;
  84. model.position.set( 1, 1, 0 );
  85. model.scale.set( 0.01, 0.01, 0.01 );
  86. scene.add( model );
  87. mixer = new THREE.AnimationMixer( model );
  88. mixer.clipAction( gltf.animations[ 0 ] ).play();
  89. const box = new THREE.Box3().setFromObject( scene );
  90. gtaoPass.setSceneClipBox( box );
  91. }, undefined, ( e ) => console.error( e ) );
  92. // Init gui
  93. const gui = new GUI();
  94. gui.add( gtaoPass, 'output', {
  95. 'Default': GTAOPass.OUTPUT.Default,
  96. 'Diffuse': GTAOPass.OUTPUT.Diffuse,
  97. 'AO Only': GTAOPass.OUTPUT.AO,
  98. 'AO Only + Denoise': GTAOPass.OUTPUT.Denoise,
  99. 'Depth': GTAOPass.OUTPUT.Depth,
  100. 'Normal': GTAOPass.OUTPUT.Normal
  101. } ).onChange( function ( value ) {
  102. gtaoPass.output = value;
  103. } );
  104. const aoParameters = {
  105. radius: 0.25,
  106. distanceExponent: 1.,
  107. thickness: 1.,
  108. scale: 1.,
  109. samples: 16,
  110. distanceFallOff: 1.,
  111. screenSpaceRadius: false,
  112. };
  113. const pdParameters = {
  114. lumaPhi: 10.,
  115. depthPhi: 2.,
  116. normalPhi: 3.,
  117. radius: 4.,
  118. radiusExponent: 1.,
  119. rings: 2.,
  120. samples: 16,
  121. };
  122. gtaoPass.updateGtaoMaterial( aoParameters );
  123. gtaoPass.updatePdMaterial( pdParameters );
  124. gui.add( gtaoPass, 'blendIntensity' ).min( 0 ).max( 1 ).step( 0.01 );
  125. gui.add( aoParameters, 'radius' ).min( 0.01 ).max( 1 ).step( 0.01 ).onChange( () => gtaoPass.updateGtaoMaterial( aoParameters ) );
  126. gui.add( aoParameters, 'distanceExponent' ).min( 1 ).max( 4 ).step( 0.01 ).onChange( () => gtaoPass.updateGtaoMaterial( aoParameters ) );
  127. gui.add( aoParameters, 'thickness' ).min( 0.01 ).max( 10 ).step( 0.01 ).onChange( () => gtaoPass.updateGtaoMaterial( aoParameters ) );
  128. gui.add( aoParameters, 'distanceFallOff' ).min( 0 ).max( 1 ).step( 0.01 ).onChange( () => gtaoPass.updateGtaoMaterial( aoParameters ) );
  129. gui.add( aoParameters, 'scale' ).min( 0.01 ).max( 2.0 ).step( 0.01 ).onChange( () => gtaoPass.updateGtaoMaterial( aoParameters ) );
  130. gui.add( aoParameters, 'samples' ).min( 2 ).max( 32 ).step( 1 ).onChange( () => gtaoPass.updateGtaoMaterial( aoParameters ) );
  131. gui.add( aoParameters, 'screenSpaceRadius' ).onChange( () => gtaoPass.updateGtaoMaterial( aoParameters ) );
  132. gui.add( pdParameters, 'lumaPhi' ).min( 0 ).max( 20 ).step( 0.01 ).onChange( () => gtaoPass.updatePdMaterial( pdParameters ) );
  133. gui.add( pdParameters, 'depthPhi' ).min( 0.01 ).max( 20 ).step( 0.01 ).onChange( () => gtaoPass.updatePdMaterial( pdParameters ) );
  134. gui.add( pdParameters, 'normalPhi' ).min( 0.01 ).max( 20 ).step( 0.01 ).onChange( () => gtaoPass.updatePdMaterial( pdParameters ) );
  135. gui.add( pdParameters, 'radius' ).min( 0 ).max( 32 ).step( 1 ).onChange( () => gtaoPass.updatePdMaterial( pdParameters ) );
  136. gui.add( pdParameters, 'radiusExponent' ).min( 0.1 ).max( 4. ).step( 0.1 ).onChange( () => gtaoPass.updatePdMaterial( pdParameters ) );
  137. gui.add( pdParameters, 'rings' ).min( 1 ).max( 16 ).step( 0.125 ).onChange( () => gtaoPass.updatePdMaterial( pdParameters ) );
  138. gui.add( pdParameters, 'samples' ).min( 2 ).max( 32 ).step( 1 ).onChange( () => gtaoPass.updatePdMaterial( pdParameters ) );
  139. window.addEventListener( 'resize', onWindowResize );
  140. }
  141. function onWindowResize() {
  142. const width = window.innerWidth;
  143. const height = window.innerHeight;
  144. camera.aspect = width / height;
  145. camera.updateProjectionMatrix();
  146. renderer.setSize( width, height );
  147. composer.setSize( width, height );
  148. }
  149. function animate() {
  150. const delta = clock.getDelta();
  151. if ( mixer ) {
  152. mixer.update( delta );
  153. }
  154. controls.update();
  155. stats.begin();
  156. composer.render();
  157. stats.end();
  158. }
  159. </script>
  160. </body>
  161. </html>