SavePass.js 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. import {
  2. ShaderMaterial,
  3. UniformsUtils,
  4. WebGLRenderTarget
  5. } from 'three';
  6. import { Pass, FullScreenQuad } from './Pass.js';
  7. import { CopyShader } from '../shaders/CopyShader.js';
  8. class SavePass extends Pass {
  9. constructor( renderTarget ) {
  10. super();
  11. if ( CopyShader === undefined ) console.error( 'THREE.SavePass relies on CopyShader' );
  12. const shader = CopyShader;
  13. this.textureID = 'tDiffuse';
  14. this.uniforms = UniformsUtils.clone( shader.uniforms );
  15. this.material = new ShaderMaterial( {
  16. uniforms: this.uniforms,
  17. vertexShader: shader.vertexShader,
  18. fragmentShader: shader.fragmentShader
  19. } );
  20. this.renderTarget = renderTarget;
  21. if ( this.renderTarget === undefined ) {
  22. this.renderTarget = new WebGLRenderTarget(); // will be resized later
  23. this.renderTarget.texture.name = 'SavePass.rt';
  24. }
  25. this.needsSwap = false;
  26. this.fsQuad = new FullScreenQuad( this.material );
  27. }
  28. render( renderer, writeBuffer, readBuffer/*, deltaTime, maskActive */ ) {
  29. if ( this.uniforms[ this.textureID ] ) {
  30. this.uniforms[ this.textureID ].value = readBuffer.texture;
  31. }
  32. renderer.setRenderTarget( this.renderTarget );
  33. if ( this.clear ) renderer.clear();
  34. this.fsQuad.render( renderer );
  35. }
  36. setSize( width, height ) {
  37. this.renderTarget.setSize( width, height );
  38. }
  39. dispose() {
  40. this.renderTarget.dispose();
  41. this.material.dispose();
  42. this.fsQuad.dispose();
  43. }
  44. }
  45. export { SavePass };