Common.MatDefs.Terrain.PBRTerrain.frag Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jme3-terrain Show documentation
Show all versions of jme3-terrain Show documentation
jMonkeyEngine is a 3-D game engine for adventurous Java developers
#import "Common/ShaderLib/GLSLCompat.glsllib"
#import "Common/ShaderLib/PBR.glsllib"
#import "Common/ShaderLib/Parallax.glsllib"
#import "Common/ShaderLib/Lighting.glsllib"
#import "Common/MatDefs/Terrain/AfflictionLib.glsllib"
varying vec3 wPosition;
varying vec3 vNormal;
varying vec2 texCoord;
uniform vec3 g_CameraPosition;
varying vec3 vPosition;
varying vec3 vnPosition;
varying vec3 vViewDir;
varying vec4 vLightDir;
varying vec4 vnLightDir;
varying vec3 lightVec;
varying vec3 inNormal;
varying vec3 wNormal;
// Specular-AA
#ifdef SPECULAR_AA_SCREEN_SPACE_VARIANCE
uniform float m_SpecularAASigma;
#endif
#ifdef SPECULAR_AA_THRESHOLD
uniform float m_SpecularAAKappa;
#endif
#ifdef DEBUG_VALUES_MODE
uniform int m_DebugValuesMode;
#endif
uniform vec4 g_LightData[NB_LIGHTS];
uniform vec4 g_AmbientLightColor;
#if NB_PROBES >= 1
uniform samplerCube g_PrefEnvMap;
uniform vec3 g_ShCoeffs[9];
uniform mat4 g_LightProbeData;
#endif
#if NB_PROBES >= 2
uniform samplerCube g_PrefEnvMap2;
uniform vec3 g_ShCoeffs2[9];
uniform mat4 g_LightProbeData2;
#endif
#if NB_PROBES == 3
uniform samplerCube g_PrefEnvMap3;
uniform vec3 g_ShCoeffs3[9];
uniform mat4 g_LightProbeData3;
#endif
#ifdef TRI_PLANAR_MAPPING
varying vec4 wVertex;
#endif
//texture-slot params for 12 unique texture slots (0-11) :
#for i=0..12 ( $0 )
uniform int m_AfflictionMode_$i;
uniform float m_Roughness_$i;
uniform float m_Metallic_$i;
#ifdef ALBEDOMAP_$i
uniform sampler2D m_AlbedoMap_$i;
#endif
#ifdef ALBEDOMAP_$i_SCALE
uniform float m_AlbedoMap_$i_scale;
#endif
#ifdef NORMALMAP_$i
uniform sampler2D m_NormalMap_$i;
#endif
#endfor
//3 alpha maps :
#ifdef ALPHAMAP
uniform sampler2D m_AlphaMap;
#endif
#ifdef ALPHAMAP_1
uniform sampler2D m_AlphaMap_1;
#endif
#ifdef ALPHAMAP_2
uniform sampler2D m_AlphaMap_2;
#endif
#ifdef DISCARD_ALPHA
uniform float m_AlphaDiscardThreshold;
#endif
//fog vars for basic fog :
#ifdef USE_FOG
#import "Common/ShaderLib/MaterialFog.glsllib"
uniform vec4 m_FogColor;
float fogDistance;
uniform vec2 m_LinearFog;
#endif
#ifdef FOG_EXP
uniform float m_ExpFog;
#endif
#ifdef FOG_EXPSQ
uniform float m_ExpSqFog;
#endif
//sun intensity is a secondary AO value that can be painted per-vertex in the red channel of the
// vertex colors, or it can be set as a static value for an entire material with the StaticSunIntensity float param
#if defined(USE_VERTEX_COLORS_AS_SUN_INTENSITY)
varying vec4 vertColors;
#endif
#ifdef STATIC_SUN_INTENSITY
uniform float m_StaticSunIntensity;
#endif
//sun intensity AO value is only applied to the directional light, not to point lights, so it is important to track if the
//sun is more/less bright than the brightest point light for each fragment to determine how the light probe's ambient light should be scaled later on in light calculation code
float brightestPointLight = 0.0;
//optional affliction paramaters that use the AfflictionAlphaMap's green channel for splatting m_SplatAlbedoMap and the red channel for splatting desaturation :
#ifdef AFFLICTIONTEXTURE
uniform sampler2D m_AfflictionAlphaMap;
#endif
#ifdef USE_SPLAT_NOISE
uniform float m_SplatNoiseVar;
#endif
//only defined for non-terrain geoemtries and terrains that are not positioned nor sized in correlation to the 2d array of AfflictionAlphaMaps used for splatting accross large tile based scenes in a grid
#ifdef TILELOCATION
uniform float m_TileWidth;
uniform vec3 m_TileLocation;
#endif
#ifdef AFFLICTIONALBEDOMAP
uniform sampler2D m_SplatAlbedoMap;
#endif
#ifdef AFFLICTIONNORMALMAP
uniform sampler2D m_SplatNormalMap;
#endif
#ifdef AFFLICTIONROUGHNESSMETALLICMAP
uniform sampler2D m_SplatRoughnessMetallicMap;
#endif
#ifdef AFFLICTIONEMISSIVEMAP
uniform sampler2D m_SplatEmissiveMap;
#endif
uniform int m_AfflictionSplatScale;
uniform float m_AfflictionRoughnessValue;
uniform float m_AfflictionMetallicValue;
uniform float m_AfflictionEmissiveValue;
uniform vec4 m_AfflictionEmissiveColor;
vec4 afflictionVector;
float noiseHash;
float livelinessValue;
float afflictionValue;
int afflictionMode = 1;
//general temp vars :
vec4 tempAlbedo, tempNormal, tempEmissiveColor;
float tempParallax, tempMetallic, tempRoughness, tempAo, tempEmissiveIntensity;
vec3 viewDir;
vec2 coord;
vec4 albedo = vec4(1.0);
vec3 normal = vec3(0.5,0.5,1);
vec3 norm;
float Metallic;
float Roughness;
float packedAoValue = 1.0;
vec4 emissive;
float emissiveIntensity = 1.0;
float indoorSunLightExposure = 1.0;
vec4 packedMetallicRoughnessAoEiVec;
vec4 packedNormalParallaxVec;
void main(){
#ifdef USE_FOG
fogDistance = distance(g_CameraPosition, wPosition.xyz);
#endif
indoorSunLightExposure = 1.0;
viewDir = normalize(g_CameraPosition - wPosition);
norm = normalize(wNormal);
normal = norm;
afflictionVector = vec4(1.0, 0.0, 1.0, 0.0); //r channel is sturation, g channel is affliction splat texture intensity, b and a unused (might use b channel for wetness eventually)
#ifdef AFFLICTIONTEXTURE
#ifdef TILELOCATION
//subterrains that are not centred in tile or equal to tile width in total size need to have m_TileWidth pre-set. (tileWidth is the x,z dimesnions that the AfflictionAlphaMap represents)
vec2 tileCoords;
float xPos, zPos;
vec3 locInTile = (wPosition - m_TileLocation);
locInTile += vec3(m_TileWidth/2, 0, m_TileWidth/2);
xPos = (locInTile.x / m_TileWidth);
zPos = 1 - (locInTile.z / m_TileWidth);
tileCoords = vec2(xPos, zPos);
afflictionVector = texture2D(m_AfflictionAlphaMap, tileCoords).rgba;
#else
// ..othrewise when terrain size matches tileWidth and location matches tileLocation, the terrain's texCoords can be used for simple texel fetching of the AfflictionAlphaMap
afflictionVector = texture2D(m_AfflictionAlphaMap, texCoord.xy).rgba;
#endif
#endif
livelinessValue = afflictionVector.r;
afflictionValue = afflictionVector.g;
#ifdef ALBEDOMAP_0
#ifdef ALPHAMAP
vec4 alphaBlend;
vec4 alphaBlend_0, alphaBlend_1, alphaBlend_2;
int texChannelForAlphaBlending;
alphaBlend_0 = texture2D( m_AlphaMap, texCoord.xy );
#ifdef ALPHAMAP_1
alphaBlend_1 = texture2D( m_AlphaMap_1, texCoord.xy );
#endif
#ifdef ALPHAMAP_2
alphaBlend_2 = texture2D( m_AlphaMap_2, texCoord.xy );
#endif
vec2 texSlotCoords;
float finalAlphaBlendForLayer = 1.0;
vec3 blending = abs( norm );
blending = (blending -0.2) * 0.7;
blending = normalize(max(blending, 0.00001)); // Force weights to sum to 1.0 (very important!)
float b = (blending.x + blending.y + blending.z);
blending /= vec3(b, b, b);
#for i=0..12 (#ifdef ALBEDOMAP_$i $0 #endif)
//assign texture slot's blending from index's correct alpha map
if($i <= 3){
alphaBlend = alphaBlend_0;
}else if($i <= 7){
alphaBlend = alphaBlend_1;
}else if($i <= 11){
alphaBlend = alphaBlend_2;
}
texChannelForAlphaBlending = int(mod($i, 4.0)); //pick the correct channel (r g b or a) based on the layer's index
switch(texChannelForAlphaBlending) {
case 0:
finalAlphaBlendForLayer = alphaBlend.r;
break;
case 1:
finalAlphaBlendForLayer = alphaBlend.g;
break;
case 2:
finalAlphaBlendForLayer = alphaBlend.b;
break;
case 3:
finalAlphaBlendForLayer = alphaBlend.a;
break;
}
afflictionMode = m_AfflictionMode_$i;
#ifdef TRI_PLANAR_MAPPING
//tri planar
tempAlbedo = getTriPlanarBlend(wVertex, blending, m_AlbedoMap_$i, m_AlbedoMap_$i_scale);
#ifdef NORMALMAP_$i
tempNormal.rgb = getTriPlanarBlend(wVertex, blending, m_NormalMap_$i, m_AlbedoMap_$i_scale).rgb;
tempNormal.rgb = calculateTangentsAndApplyToNormals(tempNormal.rgb, wNormal);// this gets rid of the need for pre-generating tangents for TerrainPatches, since doing so doesn't seem to work (tbnMat is always blank for terrains even with tangents pre-generated, not sure why...)
#else
tempNormal.rgb = wNormal.rgb;
#endif
#else
// non triplanar
texSlotCoords = texCoord * m_AlbedoMap_$i_scale;
tempAlbedo.rgb = texture2D(m_AlbedoMap_$i, texSlotCoords).rgb;
#ifdef NORMALMAP_$i
tempNormal.xyz = texture2D(m_NormalMap_$i, texSlotCoords).xyz;
tempNormal.rgb = calculateTangentsAndApplyToNormals(tempNormal.rgb, wNormal);
#else
tempNormal.rgb = wNormal.rgb;
#endif
#endif
//note: most of these functions can be found in AfflictionLib.glslib
tempAlbedo.rgb = alterLiveliness(tempAlbedo.rgb, livelinessValue, afflictionMode); //changes saturation of albedo for this layer; does nothing if not using AfflictionAlphaMap for affliction splatting
//mix values from this index layer to final output values based on finalAlphaBlendForLayer
albedo.rgb = mix(albedo.rgb, tempAlbedo.rgb , finalAlphaBlendForLayer);
normal.rgb = mix(normal.rgb, tempNormal.rgb, finalAlphaBlendForLayer);
Metallic = mix(Metallic, m_Metallic_$i, finalAlphaBlendForLayer);
Roughness = mix(Roughness, m_Roughness_$i, finalAlphaBlendForLayer);
#endfor
#endif
#endif
float alpha = albedo.a;
#ifdef DISCARD_ALPHA
if(alpha < m_AlphaDiscardThreshold){
discard;
}
#endif
//APPLY AFFLICTIONN TO THE PIXEL
#ifdef AFFLICTIONTEXTURE
vec4 afflictionAlbedo;
float newAfflictionScale = m_AfflictionSplatScale;
vec2 newScaledCoords;
#ifdef AFFLICTIONALBEDOMAP
#ifdef TRI_PLANAR_MAPPING
newAfflictionScale = newAfflictionScale / 256;
afflictionAlbedo = getTriPlanarBlend(wVertex, blending, m_SplatAlbedoMap , newAfflictionScale);
#else
newScaledCoords = mod(wPosition.xz / m_AfflictionSplatScale, 0.985);
afflictionAlbedo = texture2D(m_SplatAlbedoMap , newScaledCoords);
#endif
#else
afflictionAlbedo = vec4(1.0, 1.0, 1.0, 1.0);
#endif
vec3 afflictionNormal;
#ifdef AFFLICTIONNORMALMAP
#ifdef TRI_PLANAR_MAPPING
afflictionNormal = getTriPlanarBlend(wVertex, blending, m_SplatNormalMap , newAfflictionScale).rgb;
#else
afflictionNormal = texture2D(m_SplatNormalMap , newScaledCoords).rgb;
#endif
#else
afflictionNormal = norm;
#endif
float afflictionMetallic = m_AfflictionMetallicValue;
float afflictionRoughness = m_AfflictionRoughnessValue;
float afflictionAo = 1.0;
vec4 afflictionEmissive = m_AfflictionEmissiveColor;
float afflictionEmissiveIntensity = m_AfflictionEmissiveValue;
#ifdef AFFLICTIONROUGHNESSMETALLICMAP
vec4 metallicRoughnessAoEiVec = texture2D(m_SplatRoughnessMetallicMap, newScaledCoords);
afflictionRoughness *= metallicRoughnessAoEiVec.g;
afflictionMetallic *= metallicRoughnessAoEiVec.b;
afflictionAo = metallicRoughnessAoEiVec.r;
afflictionEmissiveIntensity *= metallicRoughnessAoEiVec.a; //important not to leave this channel all black by accident when creating the mraoei map if using affliction emissiveness
#endif
#ifdef AFFLICTIONEMISSIVEMAP
vec4 emissiveMapColor = texture2D(m_SplatEmissiveMap, newScaledCoords);
afflictionEmissive *= emissiveMapColor;
#endif
float adjustedAfflictionValue = afflictionValue;
#ifdef USE_SPLAT_NOISE
noiseHash = getStaticNoiseVar0(wPosition, afflictionValue * m_SplatNoiseVar);
adjustedAfflictionValue = getAdjustedAfflictionVar(afflictionValue);
if(afflictionValue >= 0.99){
adjustedAfflictionValue = afflictionValue;
}
#else
noiseHash = 1.0;
#endif
Roughness = alterAfflictionRoughness(adjustedAfflictionValue, Roughness, afflictionRoughness, noiseHash);
Metallic = alterAfflictionMetallic(adjustedAfflictionValue, Metallic, afflictionMetallic, noiseHash);
albedo = alterAfflictionColor(adjustedAfflictionValue, albedo, afflictionAlbedo, noiseHash );
normal = alterAfflictionNormalsForTerrain(adjustedAfflictionValue, normal, afflictionNormal, noiseHash , wNormal);
emissive = alterAfflictionGlow(adjustedAfflictionValue, emissive, afflictionEmissive, noiseHash);
emissiveIntensity = alterAfflictionEmissiveIntensity(adjustedAfflictionValue, emissiveIntensity, afflictionEmissiveIntensity, noiseHash);
emissiveIntensity *= afflictionEmissive.a;
//affliction ao value blended below after specular calculation
#endif
// spec gloss pipeline code would go here if supported, but likely will not be for terrain shaders as defines are limited and heavily used
float specular = 0.5;
float nonMetalSpec = 0.08 * specular;
vec4 specularColor = (nonMetalSpec - nonMetalSpec * Metallic) + albedo * Metallic;
vec4 diffuseColor = albedo - albedo * Metallic;
vec3 fZero = vec3(specular);
gl_FragColor.rgb = vec3(0.0);
//simple ao calculation, no support for lightmaps like stock pbr shader.. (probably could add lightmap support with another texture array, but
// that would add another texture read per slot and require removing 12 other defines to make room...)
vec3 ao = vec3(packedAoValue);
#ifdef AFFLICTIONTEXTURE
ao = alterAfflictionAo(afflictionValue, ao, vec3(afflictionAo), noiseHash); // alter the AO map for affliction values
#endif
ao.rgb = ao.rrr;
specularColor.rgb *= ao;
#ifdef STATIC_SUN_INTENSITY
indoorSunLightExposure = m_StaticSunIntensity; //single float value to indicate percentage of
//sunlight hitting the model (only works for small models or models with 100% consistent sunlighting accross every pixel)
#endif
#ifdef USE_VERTEX_COLORS_AS_SUN_INTENSITY
indoorSunLightExposure = vertColors.r * indoorSunLightExposure; //use R channel of vertexColors for..
#endif
// similar purpose as above...
//but uses r channel vert colors like an AO map specifically
//for sunlight (solution for scaling lighting for indoor
// and shadey/dimly lit models, especially big ones that
// span accross varying directionalLight exposure)
brightestPointLight = 0.0;
float ndotv = max( dot( normal, viewDir ),0.0);
#ifdef SPECULAR_AA
float sigma = 1.0;
float kappa = 0.18;
#ifdef SPECULAR_AA_SCREEN_SPACE_VARIANCE
sigma = m_SpecularAASigma;
#endif
#ifdef SPECULAR_AA_THRESHOLD
kappa = m_SpecularAAKappa;
#endif
#endif
for( int i = 0;i < NB_LIGHTS; i+=3){
vec4 lightColor = g_LightData[i];
vec4 lightData1 = g_LightData[i+1];
vec4 lightDir;
vec3 lightVec;
lightComputeDir(wPosition, lightColor.w, lightData1, lightDir, lightVec);
float fallOff = 1.0;
#if __VERSION__ >= 110
// allow use of control flow
if(lightColor.w > 1.0){
#endif
fallOff = computeSpotFalloff(g_LightData[i+2], lightVec);
#if __VERSION__ >= 110
}
#endif
//point light attenuation
fallOff *= lightDir.w;
lightDir.xyz = normalize(lightDir.xyz);
vec3 directDiffuse;
vec3 directSpecular;
#ifdef SPECULAR_AA
float hdotv = PBR_ComputeDirectLightWithSpecularAA(
normal, lightDir.xyz, viewDir,
lightColor.rgb, fZero, Roughness, sigma, kappa, ndotv,
directDiffuse, directSpecular);
#else
float hdotv = PBR_ComputeDirectLight(
normal, lightDir.xyz, viewDir,
lightColor.rgb, fZero, Roughness, ndotv,
directDiffuse, directSpecular);
#endif
vec3 directLighting = diffuseColor.rgb *directDiffuse + directSpecular;
#if defined(USE_VERTEX_COLORS_AS_SUN_INTENSITY) || defined(STATIC_SUN_INTENSITY)
if(fallOff == 1.0){
directLighting.rgb *= indoorSunLightExposure;// ... *^. to scale down how intense just the sun is (ambient and direct light are 1.0 fallOff)
}
else{
brightestPointLight = max(fallOff, brightestPointLight);
}
#endif
gl_FragColor.rgb += directLighting * fallOff;
}
float minVertLighting;
#ifdef BRIGHTEN_INDOOR_SHADOWS
minVertLighting = 0.0833; //brighten shadows so that caves/indoors which are naturally covered from the DL shadows are not way too dark compared to when shadows are off (mostly only necessary for naturally dark scenes, or dark areas when using the sun intensity code above)
#else
minVertLighting = 0.0533;
#endif
indoorSunLightExposure = max(indoorSunLightExposure, brightestPointLight);
indoorSunLightExposure = max(indoorSunLightExposure, minVertLighting); //scale the indoorSunLightExposure back up to account for the brightest point light nearby before scaling light probes by this value below
#if NB_PROBES >= 1
vec3 color1 = vec3(0.0);
vec3 color2 = vec3(0.0);
vec3 color3 = vec3(0.0);
float weight1 = 1.0;
float weight2 = 0.0;
float weight3 = 0.0;
float ndf = renderProbe(viewDir, wPosition, normal, norm, Roughness, diffuseColor, specularColor, ndotv, ao, g_LightProbeData, g_ShCoeffs, g_PrefEnvMap, color1);
#if NB_PROBES >= 2
float ndf2 = renderProbe(viewDir, wPosition, normal, norm, Roughness, diffuseColor, specularColor, ndotv, ao, g_LightProbeData2, g_ShCoeffs2, g_PrefEnvMap2, color2);
#endif
#if NB_PROBES == 3
float ndf3 = renderProbe(viewDir, wPosition, normal, norm, Roughness, diffuseColor, specularColor, ndotv, ao, g_LightProbeData3, g_ShCoeffs3, g_PrefEnvMap3, color3);
#endif
#if NB_PROBES >= 2
float invNdf = max(1.0 - ndf,0.0);
float invNdf2 = max(1.0 - ndf2,0.0);
float sumNdf = ndf + ndf2;
float sumInvNdf = invNdf + invNdf2;
#if NB_PROBES == 3
float invNdf3 = max(1.0 - ndf3,0.0);
sumNdf += ndf3;
sumInvNdf += invNdf3;
weight3 = ((1.0 - (ndf3 / sumNdf)) / (NB_PROBES - 1)) * (invNdf3 / sumInvNdf);
#endif
weight1 = ((1.0 - (ndf / sumNdf)) / (NB_PROBES - 1)) * (invNdf / sumInvNdf);
weight2 = ((1.0 - (ndf2 / sumNdf)) / (NB_PROBES - 1)) * (invNdf2 / sumInvNdf);
float weightSum = weight1 + weight2 + weight3;
weight1 /= weightSum;
weight2 /= weightSum;
weight3 /= weightSum;
#endif
#ifdef USE_AMBIENT_LIGHT
color1.rgb *= g_AmbientLightColor.rgb;
color2.rgb *= g_AmbientLightColor.rgb;
color3.rgb *= g_AmbientLightColor.rgb;
#endif
// multiply probes by the indoorSunLightExposure, as determined by pixel's sunlightExposure and adjusted for
// nearby point/spot lights ( will be multiplied by 1.0 and left unchanged if you are not defining any of the sunlight exposure variables for dimming indoors areas)
color1.rgb *= indoorSunLightExposure;
color2.rgb *= indoorSunLightExposure;
color3.rgb *= indoorSunLightExposure;
gl_FragColor.rgb += color1 * clamp(weight1,0.0,1.0) + color2 * clamp(weight2,0.0,1.0) + color3 * clamp(weight3,0.0,1.0);
#endif
if(emissive.a > 0){
emissive = emissive * pow(emissive.a * 5, emissiveIntensity) * emissiveIntensity * 20 * emissive.a;
}
// emissive = emissive * pow(emissiveIntensity * 2.3, emissive.a);
gl_FragColor += emissive;
// add fog after the lighting because shadows will cause the fog to darken
// which just results in the geometry looking like it's changed color
#ifdef USE_FOG
#ifdef FOG_LINEAR
gl_FragColor = getFogLinear(gl_FragColor, m_FogColor, m_LinearFog.x, m_LinearFog.y, fogDistance);
#endif
#ifdef FOG_EXP
gl_FragColor = getFogExp(gl_FragColor, m_FogColor, m_ExpFog, fogDistance);
#endif
#ifdef FOG_EXPSQ
gl_FragColor = getFogExpSquare(gl_FragColor, m_FogColor, m_ExpSqFog, fogDistance);
#endif
#endif
//outputs the final value of the selected layer as a color for debug purposes.
#ifdef DEBUG_VALUES_MODE
if(m_DebugValuesMode == 0){
gl_FragColor.rgb = vec3(albedo);
}
else if(m_DebugValuesMode == 1){
gl_FragColor.rgb = vec3(normal);
}
else if(m_DebugValuesMode == 2){
gl_FragColor.rgb = vec3(Roughness);
}
else if(m_DebugValuesMode == 3){
gl_FragColor.rgb = vec3(Metallic);
}
else if(m_DebugValuesMode == 4){
gl_FragColor.rgb = ao.rgb;
}
else if(m_DebugValuesMode == 5){
gl_FragColor.rgb = vec3(emissive.rgb);
}
#endif
gl_FragColor.a = albedo.a;
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy