123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363 |
- import {
- Clock,
- HalfFloatType,
- NoBlending,
- Vector2,
- WebGLRenderTarget
- } from 'three';
- import { CopyShader } from '../shaders/CopyShader.js';
- import { ShaderPass } from './ShaderPass.js';
- import { ClearMaskPass, MaskPass } from './MaskPass.js';
- class EffectComposer {
-
- constructor( renderer, renderTarget ) {
-
- this.renderer = renderer;
- this._pixelRatio = renderer.getPixelRatio();
- if ( renderTarget === undefined ) {
- const size = renderer.getSize( new Vector2() );
- this._width = size.width;
- this._height = size.height;
- renderTarget = new WebGLRenderTarget( this._width * this._pixelRatio, this._height * this._pixelRatio, { type: HalfFloatType } );
- renderTarget.texture.name = 'EffectComposer.rt1';
- } else {
- this._width = renderTarget.width;
- this._height = renderTarget.height;
- }
- this.renderTarget1 = renderTarget;
- this.renderTarget2 = renderTarget.clone();
- this.renderTarget2.texture.name = 'EffectComposer.rt2';
-
- this.writeBuffer = this.renderTarget1;
-
- this.readBuffer = this.renderTarget2;
-
- this.renderToScreen = true;
-
- this.passes = [];
-
- this.copyPass = new ShaderPass( CopyShader );
- this.copyPass.material.blending = NoBlending;
-
- this.clock = new Clock();
- }
-
- swapBuffers() {
- const tmp = this.readBuffer;
- this.readBuffer = this.writeBuffer;
- this.writeBuffer = tmp;
- }
-
- addPass( pass ) {
- this.passes.push( pass );
- pass.setSize( this._width * this._pixelRatio, this._height * this._pixelRatio );
- }
-
- insertPass( pass, index ) {
- this.passes.splice( index, 0, pass );
- pass.setSize( this._width * this._pixelRatio, this._height * this._pixelRatio );
- }
-
- removePass( pass ) {
- const index = this.passes.indexOf( pass );
- if ( index !== - 1 ) {
- this.passes.splice( index, 1 );
- }
- }
-
- isLastEnabledPass( passIndex ) {
- for ( let i = passIndex + 1; i < this.passes.length; i ++ ) {
- if ( this.passes[ i ].enabled ) {
- return false;
- }
- }
- return true;
- }
-
- render( deltaTime ) {
-
- if ( deltaTime === undefined ) {
- deltaTime = this.clock.getDelta();
- }
- const currentRenderTarget = this.renderer.getRenderTarget();
- let maskActive = false;
- for ( let i = 0, il = this.passes.length; i < il; i ++ ) {
- const pass = this.passes[ i ];
- if ( pass.enabled === false ) continue;
- pass.renderToScreen = ( this.renderToScreen && this.isLastEnabledPass( i ) );
- pass.render( this.renderer, this.writeBuffer, this.readBuffer, deltaTime, maskActive );
- if ( pass.needsSwap ) {
- if ( maskActive ) {
- const context = this.renderer.getContext();
- const stencil = this.renderer.state.buffers.stencil;
-
- stencil.setFunc( context.NOTEQUAL, 1, 0xffffffff );
- this.copyPass.render( this.renderer, this.writeBuffer, this.readBuffer, deltaTime );
-
- stencil.setFunc( context.EQUAL, 1, 0xffffffff );
- }
- this.swapBuffers();
- }
- if ( MaskPass !== undefined ) {
- if ( pass instanceof MaskPass ) {
- maskActive = true;
- } else if ( pass instanceof ClearMaskPass ) {
- maskActive = false;
- }
- }
- }
- this.renderer.setRenderTarget( currentRenderTarget );
- }
-
- reset( renderTarget ) {
- if ( renderTarget === undefined ) {
- const size = this.renderer.getSize( new Vector2() );
- this._pixelRatio = this.renderer.getPixelRatio();
- this._width = size.width;
- this._height = size.height;
- renderTarget = this.renderTarget1.clone();
- renderTarget.setSize( this._width * this._pixelRatio, this._height * this._pixelRatio );
- }
- this.renderTarget1.dispose();
- this.renderTarget2.dispose();
- this.renderTarget1 = renderTarget;
- this.renderTarget2 = renderTarget.clone();
- this.writeBuffer = this.renderTarget1;
- this.readBuffer = this.renderTarget2;
- }
-
- setSize( width, height ) {
- this._width = width;
- this._height = height;
- const effectiveWidth = this._width * this._pixelRatio;
- const effectiveHeight = this._height * this._pixelRatio;
- this.renderTarget1.setSize( effectiveWidth, effectiveHeight );
- this.renderTarget2.setSize( effectiveWidth, effectiveHeight );
- for ( let i = 0; i < this.passes.length; i ++ ) {
- this.passes[ i ].setSize( effectiveWidth, effectiveHeight );
- }
- }
-
- setPixelRatio( pixelRatio ) {
- this._pixelRatio = pixelRatio;
- this.setSize( this._width, this._height );
- }
-
- dispose() {
- this.renderTarget1.dispose();
- this.renderTarget2.dispose();
- this.copyPass.dispose();
- }
- }
- export { EffectComposer };
|