Skip to content

Commit

Permalink
Finish modularizing PBRTerrain.j3md
Browse files Browse the repository at this point in the history 
Previously I only modularized AdvnacedPBRTerrain.j3md

But now this PR applies the same concept to PBRTerrain.j3md

I ended up deleting the PBRTerrain.frag file, so now both of the terrain's .j3md files reference the same fragment shader, but the advanced version has a boolean called UseTextureArrays set to true to tell the fragment shader to read from texture arrays instead of normal textures.

I also added suppor for metallicRoughnessAoEi maps in the regular PRRTerrain shader (Ao being ambient occlusion that is packed in blue channel, and Ei being emissiv intensity which gets packed in the alpha channel). You will still be limited to 16 textures without using texture arrays, and as always I'd suggest most people juts use the advanced version of the shader primarily, but atleast the option is there.
  • Loading branch information
@yaRnMcDonuts
yaRnMcDonuts authored Feb 11, 2025
1 parent b5f225d commit 5089672
Showing 1 changed file with 102 additions and 52 deletions.
154 changes: 102 additions & 52 deletions jme3-terrain/src/main/resources/Common/MatDefs/Terrain/Modular/PBRTerrainUtils.glsllib
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,12 @@

#ifdef ENABLE_PBRTerrainUtils_readPBRTerrainLayers

//texture arrays:
uniform sampler2DArray m_AlbedoTextureArray;
uniform sampler2DArray m_NormalParallaxTextureArray;
uniform sampler2DArray m_MetallicRoughnessAoEiTextureArray;
#ifdef USE_TEXTURE_ARRAYS
//texture arrays:
uniform sampler2DArray m_AlbedoTextureArray;
uniform sampler2DArray m_NormalParallaxTextureArray;
uniform sampler2DArray m_MetallicRoughnessAoEiTextureArray;
#endif

//texture-slot params for 12 unique texture slots (0-11) where the integer value points to the desired texture's index in the corresponding texture array:
#for i=0..12 (#ifdef ALBEDOMAP_$i $0 #endif)
Expand All @@ -21,15 +23,25 @@
uniform float m_AlbedoMap_$i_scale;
uniform vec4 m_EmissiveColor_$i;

uniform int m_AlbedoMap_$i;

#ifdef USE_TEXTURE_ARRAYS
uniform int m_AlbedoMap_$i;
#ifdef NORMALMAP_$i
uniform int m_NormalMap_$i;
#endif
#ifdef METALLICROUGHNESSMAP_$i
uniform int m_MetallicRoughnessMap_$i;
#endif
#else
uniform sampler2D m_AlbedoMap_$i;
#ifdef NORMALMAP_$i
uniform sampler2D m_NormalMap_$i;
#endif
#ifdef METALLICROUGHNESSMAP_$i
uniform sampler2D m_MetallicRoughnessMap_$i;
#endif
#endif
#endfor
#for n=0..12 (#ifdef METALLICROUGHNESSMAP_$n $0 #endif)
uniform int m_MetallicRoughnessMap_$n;
#endfor
#for x=0..12 (#ifdef NORMALMAP_$x $0 #endif)
uniform int m_NormalMap_$x;
#endfor

//3 alpha maps :
#ifdef ALPHAMAP
Expand All @@ -41,9 +53,9 @@
#ifdef ALPHAMAP_2
uniform sampler2D m_AlphaMap_2;
#endif

vec4 alphaBlend_0, alphaBlend_1, alphaBlend_2;


void PBRTerrainUtils_readAlphaMaps(){

#ifdef ALPHAMAP
Expand All @@ -54,9 +66,7 @@
#endif
#ifdef ALPHAMAP_2
alphaBlend_2 = texture2D( m_AlphaMap_2, texCoord.xy );
#endif


#endif
}

float PBRTerrainUtils_getAlphaBlendFromChannel(int layer){
Expand Down Expand Up @@ -87,78 +97,118 @@

return finalAlphaBlendForLayer;
}


PBRTerrainTextureLayer PBRTerrainUtils_createAdvancedPBRTerrainLayer(int layerNum){
PBRTerrainTextureLayer PBRTerrainUtils_createAdvancedPBRTerrainLayer(int layerNum, vec3 geometryNormal){

PBRTerrainTextureLayer terrainTextureLayer;
terrainTextureLayer.blendValue = PBRTerrainUtils_getAlphaBlendFromChannel(layerNum);
terrainTextureLayer.albedo = vec4(1.0);
terrainTextureLayer.emission = vec4(0.0);
terrainTextureLayer.normal = geometryNormal;
terrainTextureLayer.alpha = 1.0;
terrainTextureLayer.ao = 1.0;
terrainTextureLayer.roughness = 1.0;
terrainTextureLayer.metallic = 0.0;
terrainTextureLayer.height = 0.0;

return terrainTextureLayer;
}
}

//________

//3 functinos to update layers from respective packed data vecs:
void updateLayerFromPackedAlbedoMap(inout vec4 packedAlbedoVec, inout PBRTerrainTextureLayer layer){
layer.albedo = packedAlbedoVec;
layer.alpha = packedAlbedoVec.a;
}
void updateLayerFromPackedNormalParallaxVec(inout vec4 packedNormalParallaxVec, inout PBRTerrainTextureLayer layer){
layer.normal = calculateTangentsAndApplyToNormals(packedNormalParallaxVec.rgb, PBRLightingUtils_getWorldNormal());
layer.normal = normalize(calculateTangentsAndApplyToNormals(packedNormalParallaxVec.rgb, PBRLightingUtils_getWorldNormal()));
layer.height = packedNormalParallaxVec.a;
}
void updateLayerFromPackedMRAoEiVec(inout vec4 packedMRAoEiVec, inout PBRTerrainTextureLayer layer){
layer.ao = packedMRAoEiVec.r;
layer.roughness = packedMRAoEiVec.g;
layer.metallic = packedMRAoEiVec.b;
layer.ao = packedMRAoEiVec.r; //ao only comes from texture (no float scalars) so no *= is done here
layer.roughness *= packedMRAoEiVec.g;
layer.metallic *= packedMRAoEiVec.b;
layer.emission *= packedMRAoEiVec.a * layer.emission.a;
}
//________________________________
// Basic Texture Reads:

// Albedo:
void PBRTerrainUtils_readAlbedoTexture(in sampler2D tex, in float scale, inout PBRTerrainTextureLayer layer){
vec4 packedAlbedoVec = texture2D(tex, texCoord * scale);
updateLayerFromPackedAlbedoMap(packedAlbedoVec, layer);
}
//________

// read Triplanar Albedo from TextureArray:
void PBRTerrainUtils_readTriPlanarAlbedoTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedAlbedoVec = getTriPlanarBlendFromTexArray(lPosition, indexInTexArray, scale, texArray);
// normal:
void PBRTerrainUtils_readNormalTexture(in sampler2D tex, in float scale, inout PBRTerrainTextureLayer layer){
vec4 packedNormalParallaxVec = texture2D(tex, texCoord * scale);
updateLayerFromPackedNormalParallaxVec(packedNormalParallaxVec, layer);
}
// metallicRoughnessAoEi:
void PBRTerrainUtils_readMetallicRoughnessAoEiTexture(in sampler2D tex, in float scale, inout PBRTerrainTextureLayer layer){
vec4 packedMRAoEi = texture2D(tex, texCoord * scale);
updateLayerFromPackedMRAoEiVec(packedMRAoEi, layer);
}
//________________________________
// Basic Triplanar Reads:

// Triplanar Albedo:
void PBRTerrainUtils_readTriPlanarAlbedoTexture(in sampler2D tex, in float scale, inout PBRTerrainTextureLayer layer){
vec4 packedAlbedoVec = getTriPlanarBlend(lPosition, tex, scale);
updateLayerFromPackedAlbedoMap(packedAlbedoVec, layer);
}
// read Triplanar normal from TextureArray:
void PBRTerrainUtils_readTriPlanarNormalTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedNormalParallaxVec = getTriPlanarBlendFromTexArray(lPosition, indexInTexArray, scale, texArray);
// Triplanar normal:
void PBRTerrainUtils_readTriPlanarNormalTexture(in sampler2D tex, in float scale, inout PBRTerrainTextureLayer layer){
vec4 packedNormalParallaxVec = getTriPlanarNormalBlend(lPosition, tex, scale);
updateLayerFromPackedNormalParallaxVec(packedNormalParallaxVec, layer);
}
// read TriPlanar metallicRoughnessAoEi from TextureArray:
void PBRTerrainUtils_readTriPlanarMetallicRoughnessAoEiTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedMRAoEi = getTriPlanarBlendFromTexArray(lPosition, indexInTexArray, scale, texArray);
// TriPlanar metallicRoughnessAoEi:
void PBRTerrainUtils_readTriPlanarMetallicRoughnessAoEiTexture(in sampler2D tex, in float scale, inout PBRTerrainTextureLayer layer){
vec4 packedMRAoEi = getTriPlanarBlend(lPosition, tex, scale);
updateLayerFromPackedMRAoEiVec(packedMRAoEi, layer);
}
//________

// read Albedo from TextureArray:
}
//________________________________
// Basic TexArray reads:

// Albedo TextureArray:
void PBRTerrainUtils_readAlbedoTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedAlbedoVec = texture2DArray(texArray, vec3(texCoord * scale, indexInTexArray));
updateLayerFromPackedAlbedoMap(packedAlbedoVec, layer);

updateLayerFromPackedAlbedoMap(packedAlbedoVec, layer);
}
// read Normal from TextureArray:
// Normal TextureArray:
void PBRTerrainUtils_readNormalTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedNormalParallaxVec = texture2DArray(texArray, vec3(texCoord * scale, indexInTexArray));
updateLayerFromPackedNormalParallaxVec(packedNormalParallaxVec, layer);
}
// read metallicRoughnessAoEi from TextureArray:
// metallicRoughnessAoEi TextureArray:
void PBRTerrainUtils_readMetallicRoughnessAoEiTexArray(in int indexInTexArray, float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedMRAoEi = texture2DArray(texArray, vec3(texCoord * scale, indexInTexArray));
updateLayerFromPackedMRAoEiVec(packedMRAoEi, layer);

updateLayerFromPackedMRAoEiVec(packedMRAoEi, layer);
}
//________



//________________________________
// Triplanar TexArray reads:

// Triplana Albedo TextureArray:
void PBRTerrainUtils_readTriPlanarAlbedoTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedAlbedoVec = getTriPlanarBlendFromTexArray(lPosition, indexInTexArray, scale, texArray);
updateLayerFromPackedAlbedoMap(packedAlbedoVec, layer);
}
// Triplanar normal TextureArray:
void PBRTerrainUtils_readTriPlanarNormalTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedNormalParallaxVec = getTriPlanarNormalBlendFromTexArray(lPosition, indexInTexArray, scale, texArray);
updateLayerFromPackedNormalParallaxVec(packedNormalParallaxVec, layer);
}
// TriPlanar metallicRoughnessAoEi TextureArray:
void PBRTerrainUtils_readTriPlanarMetallicRoughnessAoEiTexArray(in int indexInTexArray, in float scale, in sampler2DArray texArray, inout PBRTerrainTextureLayer layer){
vec4 packedMRAoEi = getTriPlanarBlendFromTexArray(lPosition, indexInTexArray, scale, texArray);
updateLayerFromPackedMRAoEiVec(packedMRAoEi, layer);
}
//_______________________________

//blend layer function:
void PBRTerrainUtils_blendPBRTerrainLayer(inout PBRSurface surface, inout PBRTerrainTextureLayer layer){


//mix values from this index layer to final output values based on finalAlphaBlendForLayer
//mixes this layer's pbr vars over top of the current surface values based on the layer's blendValue
surface.albedo = mix(surface.albedo, layer.albedo.rgb, layer.blendValue);
surface.normal = mix(surface.normal.rgb, layer.normal, layer.blendValue);
surface.normal = normalize(mix(surface.normal.rgb, layer.normal, layer.blendValue));
surface.metallic = mix(surface.metallic, layer.metallic, layer.blendValue);
surface.roughness = mix(surface.roughness, layer.roughness, layer.blendValue);
surface.ao = mix(surface.ao, vec3(layer.ao), layer.blendValue);
Expand Down

0 comments on commit 5089672

Please sign in to comment.