Added DoF setting to BokehPass.

This lets the user define an area in front of and behind the focus point that remains sharp.

src/materials/BokehMaterial.js

@@ -17,14 +17,16 @@ export class BokehMaterial extends ShaderMaterial {
 	 *
 	 * @param {PerspectiveCamera} [camera] - A camera.
 	 * @param {Object} [options] - The options.
-	 * @param {Number} [options.focus=1.0] - Focus distance.
-	 * @param {Number} [options.aperture=0.025] - Camera aperture scale. Bigger values for shallower depth of field.
+	 * @param {Number} [options.focus=1.0] - The focus distance, corresponds directly with the scene depth.
+	 * @param {Number} [options.dof=0.02] - Depth of field. An area in front of and behind the focus point that still appears sharp.
+	 * @param {Number} [options.aperture=0.025] - Camera aperture scale. Bigger values for stronger blur and shallower depth of field.
 	 * @param {Number} [options.maxBlur=1.0] - Maximum blur strength.
 	 */
 
 	constructor(camera = null, options = {}) {
 
 		if(options.focus === undefined) { options.focus = 1.0; }
+		if(options.dof === undefined) { options.dof = 0.02; }
 		if(options.aperture === undefined) { options.aperture = 0.025; }
 		if(options.maxBlur === undefined) { options.maxBlur = 1.0; }
 
@@ -42,6 +44,7 @@ export class BokehMaterial extends ShaderMaterial {
 				tDepth: new Uniform(null),
 
 				focus: new Uniform(options.focus),
+				dof: new Uniform(options.dof),
 				aperture: new Uniform(options.aperture),
 				maxBlur: new Uniform(options.maxBlur)
 

src/materials/glsl/bokeh/shader.frag

@@ -2,6 +2,7 @@ uniform sampler2D tDiffuse;
 uniform sampler2D tDepth;
 
 uniform float focus;
+uniform float dof;
 uniform float aspect;
 uniform float aperture;
 uniform float maxBlur;
@@ -40,7 +41,14 @@ void main() {
 
 	#endif
 
-	float factor = depth - focus;
+	float focusNear = clamp(focus - dof, 0.0, 1.0);
+	float focusFar = clamp(focus + dof, 0.0, 1.0);
+
+	// Calculate a DoF mask.
+	float low = step(depth, focusNear);
+	float high = step(focusFar, depth);
+
+	float factor = (depth - focusNear) * low + (depth - focusFar) * high;
 
 	vec2 dofBlur = vec2(clamp(factor * aperture, -maxBlur, maxBlur));
 

src/passes/BokehPass.js

@@ -15,6 +15,7 @@ export class BokehPass extends Pass {
 	 * @param {PerspectiveCamera} camera - The main camera. Used to obtain the aspect ratio and the near and far plane settings.
 	 * @param {Object} [options] - Additional parameters.
 	 * @param {Number} [options.focus=1.0] - Focus distance.
+	 * @param {Number} [options.dof=0.02] - Depth of field. An area in front of and behind the focus point that still appears sharp.
 	 * @param {Number} [options.aperture=0.025] - Camera aperture scale. Bigger values for shallower depth of field.
 	 * @param {Number} [options.maxBlur=1.0] - Maximum blur strength.
 	 */