webgl_postprocessing_ssr.html 9.0 KB

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