Add support for depth copying

Author: vanruesc

src/materials/CopyMaterial.js

@@ -1,4 +1,4 @@
-import { NoBlending, ShaderMaterial, Uniform } from "three";
+import { AlwaysDepth, Color, NoBlending, ShaderMaterial, Uniform } from "three";
 
 import fragmentShader from "./glsl/copy.frag";
 import vertexShader from "./glsl/common.vert";
@@ -17,8 +17,14 @@ export class CopyMaterial extends ShaderMaterial {
 
 		super({
 			name: "CopyMaterial",
+			defines: {
+				DEPTH_PACKING: "0",
+				COLOR_WRITE: "1"
+			},
 			uniforms: {
 				inputBuffer: new Uniform(null),
+				depthBuffer: new Uniform(null),
+				color: new Uniform(new Color()),
 				opacity: new Uniform(1.0)
 			},
 			blending: NoBlending,
@@ -29,6 +35,8 @@ export class CopyMaterial extends ShaderMaterial {
 			vertexShader
 		});
 
+		this.depthFunc = AlwaysDepth;
+
 	}
 
 	/**
@@ -37,12 +45,111 @@ export class CopyMaterial extends ShaderMaterial {
 	 * @type {Texture}
 	 */
 
+	get inputBuffer() {
+
+		return this.uniforms.inputBuffer.value;
+
+	}
+
 	set inputBuffer(value) {
 
+		const colorWrite = value !== null;
+
+		if(this.colorWrite !== colorWrite) {
+
+			if(colorWrite) {
+
+				this.defines.COLOR_WRITE = true;
+
+			} else {
+
+				delete this.defines.COLOR_WRITE;
+
+			}
+
+			this.colorWrite = colorWrite;
+			this.needsUpdate = true;
+
+		}
+
 		this.uniforms.inputBuffer.value = value;
 
 	}
 
+	/**
+	 * A depth buffer.
+	 *
+	 * @type {Texture}
+	 */
+
+	get depthBuffer() {
+
+		return this.uniforms.depthBuffer.value;
+
+	}
+
+	set depthBuffer(value) {
+
+		const depthWrite = value !== null;
+
+		if(this.depthWrite !== depthWrite) {
+
+			if(depthWrite) {
+
+				this.defines.DEPTH_WRITE = true;
+
+			} else {
+
+				delete this.defines.DEPTH_WRITE;
+
+			}
+
+			this.depthTest = depthWrite;
+			this.depthWrite = depthWrite;
+			this.needsUpdate = true;
+
+		}
+
+		this.uniforms.depthBuffer.value = value;
+
+	}
+
+	/**
+	 * The depth packing strategy of the depth buffer.
+	 *
+	 * @type {DepthPackingStrategies}
+	 */
+
+	set depthPacking(value) {
+
+		this.defines.DEPTH_PACKING = value.toFixed(0);
+		this.needsUpdate = true;
+
+	}
+
+	/**
+	 * A color that should be applied to the input buffer.
+	 *
+	 * @type {Color | Number}
+	 */
+
+	set color(value) {
+
+		if(value !== null) {
+
+			this.defines.USE_COLOR = "1";
+			this.uniforms.color.value.set(value);
+
+		} else {
+
+			delete this.defines.USE_COLOR;
+
+		}
+
+		this.needsUpdate = true;
+
+	}
+
 	/**
 	 * Sets the input buffer.
 	 *

src/materials/glsl/copy.frag

@@ -1,13 +1,53 @@
-#include <common>
-#include <dithering_pars_fragment>
+#ifdef COLOR_WRITE
 
-#ifdef FRAMEBUFFER_PRECISION_HIGH
+	#include <common>
+	#include <dithering_pars_fragment>
 
-	uniform mediump sampler2D inputBuffer;
+	#ifdef FRAMEBUFFER_PRECISION_HIGH
 
-#else
+		uniform mediump sampler2D inputBuffer;
 
-	uniform lowp sampler2D inputBuffer;
+	#else
+
+		uniform lowp sampler2D inputBuffer;
+
+	#endif
+
+#endif
+
+#ifdef DEPTH_WRITE
+
+	#include <packing>
+
+	#ifdef GL_FRAGMENT_PRECISION_HIGH
+
+		uniform highp sampler2D depthBuffer;
+
+	#else
+
+		uniform mediump sampler2D depthBuffer;
+
+	#endif
+
+	float readDepth(const in vec2 uv) {
+
+		#if DEPTH_PACKING == 3201
+
+			return unpackRGBAToDepth(texture2D(depthBuffer, uv));
+
+		#else
+
+			return texture2D(depthBuffer, uv).r;
+
+		#endif
+
+	}
+
+#endif
+
+#ifdef USE_COLOR
+
+	uniform vec3 color;
 
 #endif
 
@@ -17,10 +57,36 @@ varying vec2 vUv;
 
 void main() {
 
-	vec4 texel = texture2D(inputBuffer, vUv);
-	gl_FragColor = opacity * texel;
+	#ifdef COLOR_WRITE
+
+		vec4 texel = texture2D(inputBuffer, vUv);
+
+		#ifdef USE_COLOR
+
+			texel.rgb *= color;
+
+		#endif
+
+		gl_FragColor = opacity * texel;
+
+		#ifdef COLOR_SPACE_CONVERSION
+
+			#include <colorspace_fragment>
+
+		#endif
+
+		#include <dithering_fragment>
+
+	#else
+
+		gl_FragColor = vec4(0.0);
+
+	#endif
+
+	#ifdef DEPTH_WRITE
+
+		gl_FragDepth = readDepth(vUv);
 
-	#include <colorspace_fragment>
-	#include <dithering_fragment>
+	#endif
 
 }