webgl_postprocessing_ssr.html 8.8 KB

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