MaskPass.js 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. import { Pass } from '../postprocessing/Pass.js';
  2. var MaskPass = function ( scene, camera ) {
  3. Pass.call( this );
  4. this.scene = scene;
  5. this.camera = camera;
  6. this.clear = true;
  7. this.needsSwap = false;
  8. this.inverse = false;
  9. };
  10. MaskPass.prototype = Object.assign( Object.create( Pass.prototype ), {
  11. constructor: MaskPass,
  12. render: function ( renderer, writeBuffer, readBuffer /*, deltaTime, maskActive */ ) {
  13. var context = renderer.getContext();
  14. var state = renderer.state;
  15. // don't update color or depth
  16. state.buffers.color.setMask( false );
  17. state.buffers.depth.setMask( false );
  18. // lock buffers
  19. state.buffers.color.setLocked( true );
  20. state.buffers.depth.setLocked( true );
  21. // set up stencil
  22. var writeValue, clearValue;
  23. if ( this.inverse ) {
  24. writeValue = 0;
  25. clearValue = 1;
  26. } else {
  27. writeValue = 1;
  28. clearValue = 0;
  29. }
  30. state.buffers.stencil.setTest( true );
  31. state.buffers.stencil.setOp( context.REPLACE, context.REPLACE, context.REPLACE );
  32. state.buffers.stencil.setFunc( context.ALWAYS, writeValue, 0xffffffff );
  33. state.buffers.stencil.setClear( clearValue );
  34. state.buffers.stencil.setLocked( true );
  35. // draw into the stencil buffer
  36. renderer.setRenderTarget( readBuffer );
  37. if ( this.clear ) renderer.clear();
  38. renderer.render( this.scene, this.camera );
  39. renderer.setRenderTarget( writeBuffer );
  40. if ( this.clear ) renderer.clear();
  41. renderer.render( this.scene, this.camera );
  42. // unlock color and depth buffer for subsequent rendering
  43. state.buffers.color.setLocked( false );
  44. state.buffers.depth.setLocked( false );
  45. // only render where stencil is set to 1
  46. state.buffers.stencil.setLocked( false );
  47. state.buffers.stencil.setFunc( context.EQUAL, 1, 0xffffffff ); // draw if == 1
  48. state.buffers.stencil.setOp( context.KEEP, context.KEEP, context.KEEP );
  49. state.buffers.stencil.setLocked( true );
  50. }
  51. } );
  52. var ClearMaskPass = function () {
  53. Pass.call( this );
  54. this.needsSwap = false;
  55. };
  56. ClearMaskPass.prototype = Object.create( Pass.prototype );
  57. Object.assign( ClearMaskPass.prototype, {
  58. render: function ( renderer /*, writeBuffer, readBuffer, deltaTime, maskActive */ ) {
  59. renderer.state.buffers.stencil.setLocked( false );
  60. renderer.state.buffers.stencil.setTest( false );
  61. }
  62. } );
  63. export { MaskPass, ClearMaskPass };