Newer
Older
HuangJiPC / public / static / three / examples / jsm / shaders / ToneMapShader.js
@zhangdeliang zhangdeliang on 21 Jun 1 KB update
/**
 * @author miibond
 *
 * Full-screen tone-mapping shader based on http://www.cis.rit.edu/people/faculty/ferwerda/publications/sig02_paper.pdf
 */



var ToneMapShader = {

	uniforms: {

		"tDiffuse": { value: null },
		"averageLuminance": { value: 1.0 },
		"luminanceMap": { value: null },
		"maxLuminance": { value: 16.0 },
		"minLuminance": { value: 0.01 },
		"middleGrey": { value: 0.6 }
	},

	vertexShader: [

		"varying vec2 vUv;",

		"void main() {",

		"	vUv = uv;",
		"	gl_Position = projectionMatrix * modelViewMatrix * vec4( position, 1.0 );",

		"}"

	].join( "\n" ),

	fragmentShader: [

		"#include <common>",

		"uniform sampler2D tDiffuse;",

		"varying vec2 vUv;",

		"uniform float middleGrey;",
		"uniform float minLuminance;",
		"uniform float maxLuminance;",
		"#ifdef ADAPTED_LUMINANCE",
		"	uniform sampler2D luminanceMap;",
		"#else",
		"	uniform float averageLuminance;",
		"#endif",

		"vec3 ToneMap( vec3 vColor ) {",
		"	#ifdef ADAPTED_LUMINANCE",
		// Get the calculated average luminance
		"		float fLumAvg = texture2D(luminanceMap, vec2(0.5, 0.5)).r;",
		"	#else",
		"		float fLumAvg = averageLuminance;",
		"	#endif",

		// Calculate the luminance of the current pixel
		"	float fLumPixel = linearToRelativeLuminance( vColor );",

		// Apply the modified operator (Eq. 4)
		"	float fLumScaled = (fLumPixel * middleGrey) / max( minLuminance, fLumAvg );",

		"	float fLumCompressed = (fLumScaled * (1.0 + (fLumScaled / (maxLuminance * maxLuminance)))) / (1.0 + fLumScaled);",
		"	return fLumCompressed * vColor;",
		"}",

		"void main() {",

		"	vec4 texel = texture2D( tDiffuse, vUv );",

		"	gl_FragColor = vec4( ToneMap( texel.xyz ), texel.w );",

		"}"

	].join( "\n" )

};

export { ToneMapShader };