SavePass.js 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  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( window.innerWidth, window.innerHeight );
  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. }
  37. export { SavePass };