BokehPass.js 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. /**
  2. * Depth-of-field post-process with bokeh shader
  3. */
  4. THREE.BokehPass = function ( scene, camera, params ) {
  5. this.scene = scene;
  6. this.camera = camera;
  7. var focus = ( params.focus !== undefined ) ? params.focus : 1.0;
  8. var aspect = ( params.aspect !== undefined ) ? params.aspect : camera.aspect;
  9. var aperture = ( params.aperture !== undefined ) ? params.aperture : 0.025;
  10. var maxblur = ( params.maxblur !== undefined ) ? params.maxblur : 1.0;
  11. // render targets
  12. var width = params.width || window.innerWidth || 1;
  13. var height = params.height || window.innerHeight || 1;
  14. this.renderTargetColor = new THREE.WebGLRenderTarget( width, height, {
  15. minFilter: THREE.LinearFilter,
  16. magFilter: THREE.LinearFilter,
  17. format: THREE.RGBFormat
  18. } );
  19. this.renderTargetDepth = this.renderTargetColor.clone();
  20. // depth material
  21. this.materialDepth = new THREE.MeshDepthMaterial();
  22. // bokeh material
  23. if ( THREE.BokehShader === undefined ) {
  24. console.error( "THREE.BokehPass relies on THREE.BokehShader" );
  25. }
  26. var bokehShader = THREE.BokehShader;
  27. var bokehUniforms = THREE.UniformsUtils.clone( bokehShader.uniforms );
  28. bokehUniforms[ "tDepth" ].value = this.renderTargetDepth;
  29. bokehUniforms[ "focus" ].value = focus;
  30. bokehUniforms[ "aspect" ].value = aspect;
  31. bokehUniforms[ "aperture" ].value = aperture;
  32. bokehUniforms[ "maxblur" ].value = maxblur;
  33. this.materialBokeh = new THREE.ShaderMaterial({
  34. uniforms: bokehUniforms,
  35. vertexShader: bokehShader.vertexShader,
  36. fragmentShader: bokehShader.fragmentShader
  37. });
  38. this.uniforms = bokehUniforms;
  39. this.enabled = true;
  40. this.needsSwap = false;
  41. this.renderToScreen = false;
  42. this.clear = false;
  43. };
  44. THREE.BokehPass.prototype = {
  45. render: function ( renderer, writeBuffer, readBuffer, delta, maskActive ) {
  46. var composer = THREE.EffectComposer;
  47. composer.quad.material = this.materialBokeh;
  48. // Render depth into texture
  49. this.scene.overrideMaterial = this.materialDepth;
  50. renderer.render( this.scene, this.camera, this.renderTargetDepth, true );
  51. // Render bokeh composite
  52. this.uniforms[ "tColor" ].value = readBuffer;
  53. if ( this.renderToScreen ) {
  54. renderer.render( composer.scene, composer.camera );
  55. } else {
  56. renderer.render( composer.scene, composer.camera, writeBuffer, this.clear );
  57. }
  58. this.scene.overrideMaterial = null;
  59. }
  60. };