2
0

AfterimagePass.js 2.2 KB

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