AfterimagePass.js 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. import {
  2. MeshBasicMaterial,
  3. NearestFilter,
  4. ShaderMaterial,
  5. UniformsUtils,
  6. WebGLRenderTarget
  7. } from 'three';
  8. import { Pass, FullScreenQuad } from './Pass.js';
  9. import { AfterimageShader } from '../shaders/AfterimageShader.js';
  10. class AfterimagePass extends Pass {
  11. constructor( damp = 0.96 ) {
  12. super();
  13. this.shader = AfterimageShader;
  14. this.uniforms = UniformsUtils.clone( this.shader.uniforms );
  15. this.uniforms[ 'damp' ].value = damp;
  16. this.textureComp = new WebGLRenderTarget( window.innerWidth, window.innerHeight, {
  17. magFilter: NearestFilter,
  18. } );
  19. this.textureOld = new WebGLRenderTarget( window.innerWidth, window.innerHeight, {
  20. magFilter: NearestFilter,
  21. } );
  22. this.compFsMaterial = new ShaderMaterial( {
  23. uniforms: this.uniforms,
  24. vertexShader: this.shader.vertexShader,
  25. fragmentShader: this.shader.fragmentShader
  26. } );
  27. this.compFsQuad = new FullScreenQuad( this.compFsMaterial );
  28. this.copyFsMaterial = new MeshBasicMaterial();
  29. this.copyFsQuad = new FullScreenQuad( this.copyFsMaterial );
  30. }
  31. render( renderer, writeBuffer, readBuffer/*, deltaTime, maskActive*/ ) {
  32. this.uniforms[ 'tOld' ].value = this.textureOld.texture;
  33. this.uniforms[ 'tNew' ].value = readBuffer.texture;
  34. renderer.setRenderTarget( this.textureComp );
  35. this.compFsQuad.render( renderer );
  36. this.copyFsQuad.material.map = this.textureComp.texture;
  37. if ( this.renderToScreen ) {
  38. renderer.setRenderTarget( null );
  39. this.copyFsQuad.render( renderer );
  40. } else {
  41. renderer.setRenderTarget( writeBuffer );
  42. if ( this.clear ) renderer.clear();
  43. this.copyFsQuad.render( renderer );
  44. }
  45. // Swap buffers.
  46. const temp = this.textureOld;
  47. this.textureOld = this.textureComp;
  48. this.textureComp = temp;
  49. // Now textureOld contains the latest image, ready for the next frame.
  50. }
  51. setSize( width, height ) {
  52. this.textureComp.setSize( width, height );
  53. this.textureOld.setSize( width, height );
  54. }
  55. dispose() {
  56. this.textureComp.dispose();
  57. this.textureOld.dispose();
  58. this.compFsMaterial.dispose();
  59. this.copyFsMaterial.dispose();
  60. this.compFsQuad.dispose();
  61. this.copyFsQuad.dispose();
  62. }
  63. }
  64. export { AfterimagePass };