class DefaultForward

package h3d.shader.pbr

extends Shader

@:directlyUsed@:src({ @global var camera:{ var view : Mat4; var viewProj : Mat4; var position : Vec3; var inverseViewProj : Mat4}; @const(4) var CASCADE_COUNT:Int; @const(2) var DIR_SHADOW_COUNT:Int; @const(16) var POINT_SHADOW_COUNT:Int; @const(16) var SPOT_SHADOW_COUNT:Int; @:import h3d.shader.pbr.Light.LightEvaluation; @:import h3d.shader.pbr.BDRF; var output:{ var color : Vec4; var metalness : Float; var roughness : Float; var occlusion : Float; var emissive : Float; var depth : Float}; @const(256) var BUFFER_SIZE:Int = 1; @param var lightInfos:Buffer<Vec4,BUFFER_SIZE>; @param var dirLightCount:Int; @param var pointLightCount:Int; @param var spotLightCount:Int; @param var dirLightStride:Int; @param var pointLightStride:Int; @param var spotLightStride:Int; @param var cascadeShadowMaps:Array<Sampler2D,CASCADE_COUNT>; @param var dirShadowMaps:Array<Sampler2D,DIR_SHADOW_COUNT>; @param var pointShadowMaps:Array<SamplerCube,POINT_SHADOW_COUNT>; @param var spotShadowMaps:Array<Sampler2D,SPOT_SHADOW_COUNT>; @param var cameraPosition:Vec3; @param var emissivePower:Float; var albedoGamma:Vec3; var view:Vec3; var NdV:Float; var pbrSpecularColor:Vec3; var metalness:Float; var roughness:Float; var occlusion:Float; var emissive:Float; var F0:Vec3; @const var USE_INDIRECT = false; @param var irrLut:Sampler2D; @param var irrDiffuse:SamplerCube; @param var irrSpecular:SamplerCube; @param var irrSpecularLevels:Float; @param var irrPower:Float; @param var irrRotation:Vec2; var transformedNormal:Vec3; var transformedPosition:Vec3; var pixelColor:Vec4; var depth:Float; function rotateNormal(n:Vec3):Vec3 { return vec3(n.x * irrRotation.x - n.y * irrRotation.y, n.x * irrRotation.y + n.y * irrRotation.x, n.z); }; function indirectLighting():Vec3 { var F = F0 + (max(vec3(1 - roughness), F0) - F0) * exp2((-5.55473 * NdV - 6.98316) * NdV); var rotatedNormal = rotateNormal(transformedNormal); var diffuse = irrDiffuse.get(rotatedNormal).rgb * albedoGamma; var reflectVec = reflect(-view, transformedNormal); var rotatedReflecVec = rotateNormal(reflectVec); var envSpec = textureLod(irrSpecular, rotatedReflecVec, roughness * irrSpecularLevels).rgb; var envBRDF = irrLut.get(vec2(roughness, NdV)); var specular = envSpec * (F * envBRDF.x + envBRDF.y); var indirect = (diffuse * (1 - metalness) * (1 - F) + specular) * irrPower; return indirect * occlusion; }; function directLighting(lightColor:Vec3, lightDirection:Vec3):Vec3 { var result = vec3(0); var NdL = clamp(transformedNormal.dot(lightDirection), 0.0, 1.0); if (lightColor.dot(lightColor) > 0.0001 && NdL > 0) { var half = (lightDirection + view).normalize(); var NdH = clamp(transformedNormal.dot(half), 0.0, 1.0); var VdH = clamp(view.dot(half), 0.0, 1.0); var diffuse = albedoGamma / PI; var D = normalDistributionGGX(NdH, roughness); var F = fresnelSchlick(VdH, F0); var G = geometrySchlickGGX(NdV, NdL, roughness); var specular = (D * F * G).max(0.); result = (diffuse * (1 - metalness) * (1 - F) + specular) * lightColor * NdL; }; return result; }; function __init__fragment() { pbrSpecularColor = vec3(0.04); albedoGamma = pixelColor.rgb * pixelColor.rgb; }; function init() { view = (cameraPosition - transformedPosition).normalize(); NdV = transformedNormal.dot(view).max(0.); }; function evaluateDirShadow(index:Int):Float { var i = index * 5; var shadow = 1.0; if (lightInfos[i].a > 0) { var shadowBias = lightInfos[i + 1].a; var shadowProj = mat3x4(lightInfos[i + 2], lightInfos[i + 3], lightInfos[i + 4]); var shadowPos = transformedPosition * shadowProj; var shadowUv = screenToUv(shadowPos.xy); var depth = dirShadowMaps[index].get(shadowUv.xy).r; shadow = (shadowPos.z - shadowBias > depth) ? 0.0 : 1.0; }; return shadow; }; function evaluateDirLight(index:Int):Vec3 { var i = index * 5; var lightColor = lightInfos[i].rgb; var lightDir = lightInfos[i + 1].xyz; return directLighting(lightColor, lightDir); }; function evaluatePointShadow(index:Int):Float { var i = index * 3 + dirLightStride; var shadow = 1.0; if (lightInfos[i + 2].g > 0) { var lightPos = lightInfos[i + 1].rgb; var range = lightInfos[i + 2].r; var shadowBias = lightInfos[i + 2].b; var posToLight = transformedPosition.xyz - lightPos; var dir = normalize(posToLight.xyz); var depth = pointShadowMaps[index].getLod(dir, 0).r * range; var zMax = length(posToLight); shadow = (zMax - shadowBias > depth) ? 0.0 : 1.0; }; return shadow; }; function evaluatePointLight(index:Int):Vec3 { var i = index * 3 + dirLightStride; var lightColor = lightInfos[i].rgb; var size = lightInfos[i].a; var lightPos = lightInfos[i + 1].rgb; var invRange4 = lightInfos[i + 1].a; var delta = lightPos - transformedPosition; return directLighting(pointLightIntensity(delta, size, invRange4) * lightColor, delta.normalize()); }; function evaluateSpotShadow(index:Int):Float { var i = index * 8 + dirLightStride + pointLightStride; var shadow = 1.0; if (lightInfos[i + 3].b > 0) { var shadowBias = lightInfos[i + 3].a; var shadowProj = mat4(lightInfos[i + 4], lightInfos[i + 5], lightInfos[i + 6], lightInfos[i + 7]); var shadowPos = vec4(transformedPosition, 1.0) * shadowProj; shadowPos.xyz /= shadowPos.w; var shadowUv = screenToUv(shadowPos.xy); var depth = spotShadowMaps[index].get(shadowUv.xy).r; shadow = (shadowPos.z - shadowBias > depth) ? 0.0 : 1.0; }; return shadow; }; function evaluateSpotLight(index:Int):Vec3 { var i = index * 8 + dirLightStride + pointLightStride; var lightColor = lightInfos[i].rgb; var range = lightInfos[i].a; var lightPos = lightInfos[i + 1].xyz; var invRange4 = lightInfos[i + 1].a; var lightDir = lightInfos[i + 2].xyz; var angle = lightInfos[i + 3].r; var fallOff = lightInfos[i + 3].g; var delta = lightPos - transformedPosition; var fallOffInfo = spotLightIntensity(delta, lightDir, range, invRange4, fallOff, angle); var fallOff = fallOffInfo.x; var fallOffInfoAngle = fallOffInfo.y; return directLighting(fallOff * lightColor * fallOffInfoAngle, delta.normalize()); }; function evaluateCascadeLight():Vec3 { var i = dirLightStride + pointLightStride + spotLightStride; var lightColor = lightInfos[i].rgb; var lightDir = lightInfos[i + 1].xyz; return directLighting(lightColor, lightDir); }; function inside(pos:Vec3):Bool { if (abs(pos.x) < 1.0 && abs(pos.y) < 1.0 && abs(pos.z) < 1.0) return true else return false; }; function evaluateCascadeShadow():Float { var i = dirLightStride + pointLightStride + spotLightStride; var shadow = 1.0; var shadowProj = mat3x4(lightInfos[i + 2], lightInfos[i + 3], lightInfos[i + 4]); @unroll for (c in 0 ... CASCADE_COUNT) { var cascadeScale = lightInfos[i + 5 + 2 * c]; var shadowPos0 = transformedPosition * shadowProj; var shadowPos = c == 0 ? shadowPos0 : shadowPos0 * cascadeScale.xyz + lightInfos[i + 6 + 2 * c].xyz; if (inside(shadowPos)) { var zMax = saturate(shadowPos.z); var shadowUv = shadowPos.xy; shadowUv.y = 1.0 - shadowUv.y; var depth = cascadeShadowMaps[c].get(shadowUv.xy).r; shadow -= zMax > depth ? 1.0 : 0.0; }; }; return saturate(shadow); }; function evaluateLighting():Vec3 { var lightAccumulation = vec3(0); F0 = mix(pbrSpecularColor, albedoGamma, metalness); @unroll for (l in 0 ... DIR_SHADOW_COUNT) { var c = evaluateDirLight(l); if (dot(c, c) > 1e-6) c *= evaluateDirShadow(l); lightAccumulation += c; }; var start = DIR_SHADOW_COUNT; if (CASCADE_COUNT > 0) start++; @unroll for (l in start ... dirLightCount + DIR_SHADOW_COUNT) lightAccumulation += evaluateDirLight(l); @unroll for (l in 0 ... POINT_SHADOW_COUNT) { var c = evaluatePointLight(l); if (dot(c, c) > 1e-6) c *= evaluatePointShadow(l); lightAccumulation += c; }; @unroll for (l in POINT_SHADOW_COUNT ... pointLightCount + POINT_SHADOW_COUNT) lightAccumulation += evaluatePointLight(l); @unroll for (l in 0 ... SPOT_SHADOW_COUNT) { var c = evaluateSpotLight(l); if (dot(c, c) > 1e-6) c *= evaluateSpotShadow(l); lightAccumulation += c; }; @unroll for (l in SPOT_SHADOW_COUNT ... spotLightCount + SPOT_SHADOW_COUNT) lightAccumulation += evaluateSpotLight(l); if (CASCADE_COUNT > 0) { var c = evaluateCascadeLight(); if (dot(c, c) > 1e-6) c *= evaluateCascadeShadow(); lightAccumulation += c; }; if (USE_INDIRECT) lightAccumulation += indirectLighting(); lightAccumulation += emissive * emissivePower * pixelColor.rgb; return lightAccumulation; }; function fragment() { init(); output.color = vec4(evaluateLighting(), pixelColor.a); output.depth = depth; }; })@:build(hxsl.Macros.buildShader())@:autoBuild(hxsl.Macros.buildShader())