MSAAPass.js 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217
  1. /**
  2. * @author bhouston / http://clara.io/ *
  3. */
  4. THREE.MSAAPass = function ( scene, camera, params ) {
  5. this.scene = scene;
  6. this.camera = camera;
  7. this.sampleLevel = 4; // specified as n, where the number of samples is 2^n, so sampleLevel = 4, is 2^4 samples, 16.
  8. this.params = params || { minFilter: THREE.NearestFilter, magFilter: THREE.NearestFilter, format: THREE.RGBAFormat };
  9. this.params.minFilter = THREE.NearestFilter;
  10. this.params.maxFilter = THREE.NearestFilter;
  11. console.log( 'this.params', this.params );
  12. this.enabled = true;
  13. this.needsSwap = true;
  14. if ( THREE.CompositeShader === undefined ) {
  15. console.error( "THREE.MSAAPass relies on THREE.CompositeShader" );
  16. }
  17. var compositeShader = THREE.CompositeShader;
  18. this.uniforms = THREE.UniformsUtils.clone( compositeShader.uniforms );
  19. this.materialComposite = new THREE.ShaderMaterial( {
  20. uniforms: this.uniforms,
  21. vertexShader: compositeShader.vertexShader,
  22. fragmentShader: compositeShader.fragmentShader,
  23. transparent: true,
  24. blending: THREE.CustomBlending,
  25. blendSrc: THREE.OneFactor,
  26. blendDst: THREE.OneMinusSrcAlphaFactor,
  27. blendEquation: THREE.AddEquation,
  28. depthTest: false,
  29. depthWrite: false
  30. } );
  31. this.camera2 = new THREE.OrthographicCamera( -1, 1, 1, -1, 0, 1 );
  32. this.scene2 = new THREE.Scene();
  33. this.quad2 = new THREE.Mesh( new THREE.PlaneGeometry( 2, 2 ), this.materialComposite );
  34. this.scene2.add( this.quad2 );
  35. };
  36. THREE.MSAAPass.prototype = {
  37. dispose: function() {
  38. if( this.sampleRenderTarget ) {
  39. this.sampleRenderTarget.dispose();
  40. this.sampleRenderTarget = null;
  41. }
  42. },
  43. setSize: function ( width, height ) {
  44. this.sampleRenderTarget.setSize( width, height );
  45. },
  46. render: function ( renderer, writeBuffer, readBuffer, delta ) {
  47. if( ! this.sampleRenderTarget ) {
  48. this.sampleRenderTarget = new THREE.WebGLRenderTarget( readBuffer.width, readBuffer.height, this.params, "msaa.renderTarget0" );
  49. }
  50. var camera = ( this.camera || this.scene.camera );
  51. var jitterOffsets = THREE.MSAAPass.JitterVectors[ Math.max( 0, Math.min( this.sampleLevel, 5 ) ) ];
  52. this.uniforms[ "tForeground" ].value = this.sampleRenderTarget;
  53. var weight = 1.0;
  54. // render the scene multiple times, each slightly jitter offset from the last and accumulate the results.
  55. for( var i = 0; i < jitterOffsets.length; i ++ ) {
  56. // only jitters perspective cameras. TODO: add support for jittering orthogonal cameras
  57. if( camera.setViewOffset ) camera.setViewOffset( readBuffer.width, readBuffer.height, jitterOffsets[i].x, jitterOffsets[i].y, readBuffer.width, readBuffer.height );
  58. renderer.render( this.scene, camera, this.sampleRenderTarget, true );
  59. // this accumulation strategy is used to prevent decimation at low bit depths with lots of samples.
  60. this.uniforms[ "scale" ].value = weight;
  61. weight = 1.0 / (i+1);
  62. // clear on the first render, accumulate the others
  63. var autoClear = renderer.autoClear;
  64. renderer.autoClear = false;
  65. renderer.render( this.scene2, this.camera2, writeBuffer, i === 0 );
  66. renderer.autoClear = true;
  67. }
  68. // reset jitter to nothing. TODO: add support for orthogonal cameras
  69. if( camera.setViewOffset ) camera.setViewOffset( undefined, undefined, undefined, undefined, undefined, undefined );
  70. }
  71. };
  72. THREE.MSAAPass.normalizedJitterOffsets = function( jitterVectors ) {
  73. var scaledJitterOffsets = [];
  74. for( var i = 0; i < jitterVectors.length; i ++ ) {
  75. scaledJitterOffsets.push( new THREE.Vector2( jitterVectors[i][0], jitterVectors[i][1] ).multiplyScalar( 1.0 / 16.0 ) );
  76. }
  77. return scaledJitterOffsets;
  78. },
  79. // These jitter vectors are specified in integers because it is easier.
  80. // I am assuming a [-8,8) integer grid, but it needs to be mapped onto [-0.5,0.5)
  81. // before being used, thus these integers need to be scaled by 1/16.
  82. //
  83. // Sample patterns reference: https://msdn.microsoft.com/en-us/library/windows/desktop/ff476218%28v=vs.85%29.aspx?f=255&MSPPError=-2147217396
  84. THREE.MSAAPass.JitterVectors = [
  85. THREE.MSAAPass.normalizedJitterOffsets( [
  86. [ 0, 0 ]
  87. ] ),
  88. THREE.MSAAPass.normalizedJitterOffsets( [
  89. [ 4, 4 ],
  90. [ -4, -4 ]
  91. ] ),
  92. THREE.MSAAPass.normalizedJitterOffsets( [
  93. [ -2, -6 ],
  94. [ 6, -2 ],
  95. [ -6, 2 ],
  96. [ 2, 6 ]
  97. ] ),
  98. THREE.MSAAPass.normalizedJitterOffsets( [
  99. [ 1, -3 ],
  100. [ -1, 3 ],
  101. [ 5, 1 ],
  102. [ -3, -5 ],
  103. [ -5, 5 ],
  104. [ -7, -1 ],
  105. [ 3, 7 ],
  106. [ 7, -7 ]
  107. ] ),
  108. THREE.MSAAPass.normalizedJitterOffsets( [
  109. [ 1, 1 ],
  110. [ -1, -3 ],
  111. [ -3, 2 ],
  112. [ 4, -1 ],
  113. [ -5, -2 ],
  114. [ 2, 5 ],
  115. [ 5, 3 ],
  116. [ 3, -5 ],
  117. [ -2, 6 ],
  118. [ 0, -7 ],
  119. [ -4, -6 ],
  120. [ -6, 4 ],
  121. [ -8, 0 ],
  122. [ 7, -4 ],
  123. [ 6, 7 ],
  124. [ -7, -8 ]
  125. ] ),
  126. THREE.MSAAPass.normalizedJitterOffsets( [
  127. [ -4, -7 ],
  128. [ -7, -5 ],
  129. [ -3, -5 ],
  130. [ -5, -4 ],
  131. [ -1, -4 ],
  132. [ -2, -2 ],
  133. [ -6, -1 ],
  134. [ -4, 0 ],
  135. [ -7, 1 ],
  136. [ -1, 2 ],
  137. [ -6, 3 ],
  138. [ -3, 3 ],
  139. [ -7, 6 ],
  140. [ -3, 6 ],
  141. [ -5, 7 ],
  142. [ -1, 7 ],
  143. [ 5, -7 ],
  144. [ 1, -6 ],
  145. [ 6, -5 ],
  146. [ 4, -4 ],
  147. [ 2, -3 ],
  148. [ 7, -2 ],
  149. [ 1, -1 ],
  150. [ 4, -1 ],
  151. [ 2, 1 ],
  152. [ 6, 2 ],
  153. [ 0, 4 ],
  154. [ 4, 4 ],
  155. [ 2, 5 ],
  156. [ 7, 5 ],
  157. [ 5, 6 ],
  158. [ 3, 7 ]
  159. ] )
  160. ];