webgl_postprocessing_ssr.html 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <head>
  5. <title>three.js webgl - postprocessing - Screen Space Reflection</title>
  6. <meta charset="utf-8">
  7. <meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
  8. <link type="text/css" rel="stylesheet" href="main.css">
  9. </head>
  10. <body>
  11. <div id="container"></div>
  12. <div id="info">
  13. <a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> -
  14. SSRPass demo by <a href="https://github.com/gonnavis" target="_blank">Vis</a>.<br />
  15. </div>
  16. <script type="module">
  17. import * as THREE from '../build/three.module.js';
  18. import Stats from './jsm/libs/stats.module.js';
  19. import { OrbitControls } from './jsm/controls/OrbitControls.js';
  20. import { GUI } from './jsm/libs/lil-gui.module.min.js';
  21. import { EffectComposer } from './jsm/postprocessing/EffectComposer.js';
  22. import { SSRPass } from './jsm/postprocessing/SSRPass.js';
  23. import { ShaderPass } from './jsm/postprocessing/ShaderPass.js';
  24. import { GammaCorrectionShader } from './jsm/shaders/GammaCorrectionShader.js';
  25. import { ReflectorForSSRPass } from './jsm/objects/ReflectorForSSRPass.js';
  26. import { DRACOLoader } from './jsm/loaders/DRACOLoader.js';
  27. const params = {
  28. enableSSR: true,
  29. autoRotate: true,
  30. otherMeshes: true,
  31. groundReflector: true,
  32. };
  33. let composer;
  34. let ssrPass;
  35. let gui;
  36. let stats;
  37. let controls;
  38. let camera, scene, renderer;
  39. const otherMeshes = [];
  40. let groundReflector;
  41. const selects = [];
  42. const container = document.querySelector( '#container' );
  43. // Configure and create Draco decoder.
  44. const dracoLoader = new DRACOLoader();
  45. dracoLoader.setDecoderPath( 'js/libs/draco/' );
  46. dracoLoader.setDecoderConfig( { type: 'js' } );
  47. init();
  48. animate();
  49. function init() {
  50. camera = new THREE.PerspectiveCamera( 35, window.innerWidth / window.innerHeight, 0.1, 15 );
  51. camera.position.set( 0.13271600513224902, 0.3489546826045913, 0.43921296427927076 );
  52. scene = new THREE.Scene();
  53. scene.background = new THREE.Color( 0x443333 );
  54. scene.fog = new THREE.Fog( 0x443333, 1, 4 );
  55. // Ground
  56. const plane = new THREE.Mesh(
  57. new THREE.PlaneGeometry( 8, 8 ),
  58. new THREE.MeshPhongMaterial( { color: 0x999999, specular: 0x101010 } )
  59. );
  60. plane.rotation.x = - Math.PI / 2;
  61. plane.position.y = - 0.0001;
  62. // plane.receiveShadow = true;
  63. scene.add( plane );
  64. // Lights
  65. const hemiLight = new THREE.HemisphereLight( 0x443333, 0x111122 );
  66. scene.add( hemiLight );
  67. const spotLight = new THREE.SpotLight();
  68. spotLight.angle = Math.PI / 16;
  69. spotLight.penumbra = 0.5;
  70. // spotLight.castShadow = true;
  71. spotLight.position.set( - 1, 1, 1 );
  72. scene.add( spotLight );
  73. dracoLoader.load( 'models/draco/bunny.drc', function ( geometry ) {
  74. geometry.computeVertexNormals();
  75. const material = new THREE.MeshStandardMaterial( { color: 0x606060 } );
  76. const mesh = new THREE.Mesh( geometry, material );
  77. mesh.position.y = - 0.0365;
  78. scene.add( mesh );
  79. selects.push( mesh );
  80. // Release decoder resources.
  81. dracoLoader.dispose();
  82. } );
  83. let geometry, material, mesh;
  84. geometry = new THREE.BoxBufferGeometry( .05, .05, .05 );
  85. material = new THREE.MeshStandardMaterial( { color: 'green' } );
  86. mesh = new THREE.Mesh( geometry, material );
  87. mesh.position.set( - .12, .025, .015 );
  88. scene.add( mesh );
  89. otherMeshes.push( mesh );
  90. selects.push( mesh );
  91. geometry = new THREE.IcosahedronBufferGeometry( .025, 4 );
  92. material = new THREE.MeshStandardMaterial( { color: 'cyan' } );
  93. mesh = new THREE.Mesh( geometry, material );
  94. mesh.position.set( - .05, .025, .08 );
  95. scene.add( mesh );
  96. otherMeshes.push( mesh );
  97. selects.push( mesh );
  98. geometry = new THREE.ConeBufferGeometry( .025, .05, 64 );
  99. material = new THREE.MeshStandardMaterial( { color: 'yellow' } );
  100. mesh = new THREE.Mesh( geometry, material );
  101. mesh.position.set( - .05, .025, - .055 );
  102. scene.add( mesh );
  103. otherMeshes.push( mesh );
  104. selects.push( mesh );
  105. geometry = new THREE.PlaneBufferGeometry( 1, 1 );
  106. groundReflector = new ReflectorForSSRPass( geometry, {
  107. clipBias: 0.0003,
  108. textureWidth: window.innerWidth,
  109. textureHeight: window.innerHeight,
  110. color: 0x888888,
  111. useDepthTexture: true,
  112. } );
  113. groundReflector.material.depthWrite = false;
  114. groundReflector.rotation.x = - Math.PI / 2;
  115. groundReflector.visible = false;
  116. scene.add( groundReflector );
  117. // renderer
  118. renderer = new THREE.WebGLRenderer( { antialias: false } );
  119. renderer.setSize( window.innerWidth, window.innerHeight );
  120. container.appendChild( renderer.domElement );
  121. //
  122. controls = new OrbitControls( camera, renderer.domElement );
  123. controls.enableDamping = true;
  124. controls.target.set( 0, 0.0635, 0 );
  125. controls.update();
  126. controls.enabled = ! params.autoRotate;
  127. // STATS
  128. stats = new Stats();
  129. container.appendChild( stats.dom );
  130. window.addEventListener( 'resize', onWindowResize );
  131. // composer
  132. composer = new EffectComposer( renderer );
  133. ssrPass = new SSRPass( {
  134. renderer,
  135. scene,
  136. camera,
  137. width: innerWidth,
  138. height: innerHeight,
  139. groundReflector: params.groundReflector ? groundReflector : null,
  140. selects: params.groundReflector ? selects : null
  141. } );
  142. composer.addPass( ssrPass );
  143. composer.addPass( new ShaderPass( GammaCorrectionShader ) );
  144. // GUI
  145. gui = new GUI( { width: 260 } );
  146. gui.add( params, 'enableSSR' ).name( 'Enable SSR' );
  147. gui.add( params, 'groundReflector' ).onChange( () => {
  148. if ( params.groundReflector ) {
  149. ssrPass.groundReflector = groundReflector,
  150. ssrPass.selects = selects;
  151. } else {
  152. ssrPass.groundReflector = null,
  153. ssrPass.selects = null;
  154. }
  155. } );
  156. ssrPass.thickness = 0.018;
  157. gui.add( ssrPass, 'thickness' ).min( 0 ).max( .1 ).step( .0001 );
  158. ssrPass.infiniteThick = false;
  159. gui.add( ssrPass, 'infiniteThick' );
  160. gui.add( params, 'autoRotate' ).onChange( () => {
  161. controls.enabled = ! params.autoRotate;
  162. } );
  163. const folder = gui.addFolder( 'more settings' );
  164. folder.add( ssrPass, 'fresnel' ).onChange( ()=>{
  165. groundReflector.fresnel = ssrPass.fresnel;
  166. } );
  167. folder.add( ssrPass, 'distanceAttenuation' ).onChange( ()=>{
  168. groundReflector.distanceAttenuation = ssrPass.distanceAttenuation;
  169. } );
  170. ssrPass.maxDistance = .1;
  171. groundReflector.maxDistance = ssrPass.maxDistance;
  172. folder.add( ssrPass, 'maxDistance' ).min( 0 ).max( .5 ).step( .001 ).onChange( ()=>{
  173. groundReflector.maxDistance = ssrPass.maxDistance;
  174. } );
  175. folder.add( params, 'otherMeshes' ).onChange( () => {
  176. if ( params.otherMeshes ) {
  177. otherMeshes.forEach( mesh => mesh.visible = true );
  178. } else {
  179. otherMeshes.forEach( mesh => mesh.visible = false );
  180. }
  181. } );
  182. folder.add( ssrPass, 'bouncing' );
  183. folder.add( ssrPass, 'output', {
  184. 'Default': SSRPass.OUTPUT.Default,
  185. 'SSR Only': SSRPass.OUTPUT.SSR,
  186. 'Beauty': SSRPass.OUTPUT.Beauty,
  187. 'Depth': SSRPass.OUTPUT.Depth,
  188. 'Normal': SSRPass.OUTPUT.Normal,
  189. 'Metalness': SSRPass.OUTPUT.Metalness,
  190. } ).onChange( function ( value ) {
  191. ssrPass.output = parseInt( value );
  192. } );
  193. ssrPass.opacity = 1;
  194. groundReflector.opacity = ssrPass.opacity;
  195. folder.add( ssrPass, 'opacity' ).min( 0 ).max( 1 ).onChange( ()=>{
  196. groundReflector.opacity = ssrPass.opacity;
  197. } );
  198. folder.add( ssrPass, 'blur' );
  199. // folder.open()
  200. // gui.close()
  201. }
  202. function onWindowResize() {
  203. camera.aspect = window.innerWidth / window.innerHeight;
  204. camera.updateProjectionMatrix();
  205. renderer.setSize( window.innerWidth, window.innerHeight );
  206. composer.setSize( window.innerWidth, window.innerHeight );
  207. groundReflector.getRenderTarget().setSize( window.innerWidth, window.innerHeight );
  208. groundReflector.resolution.set( window.innerWidth, window.innerHeight );
  209. }
  210. function animate() {
  211. requestAnimationFrame( animate );
  212. stats.begin();
  213. render();
  214. stats.end();
  215. }
  216. function render() {
  217. if ( params.autoRotate ) {
  218. const timer = Date.now() * 0.0003;
  219. camera.position.x = Math.sin( timer ) * 0.5;
  220. camera.position.y = 0.2135;
  221. camera.position.z = Math.cos( timer ) * 0.5;
  222. camera.lookAt( 0, 0.0635, 0 );
  223. } else {
  224. controls.update();
  225. }
  226. if ( params.enableSSR ) {
  227. // TODO: groundReflector has full ground info, need use it to solve reflection gaps problem on objects when camera near ground.
  228. // TODO: the normal and depth info where groundReflector reflected need to be changed.
  229. composer.render();
  230. } else {
  231. renderer.render( scene, camera );
  232. }
  233. }
  234. </script>
  235. </body>
  236. </html>