urho3d/SourceAssets/HLSLShaders/ForwardLit.hlsl

#include "Uniforms.hlsl"
#include "Samplers.hlsl"
#include "Transform.hlsl"
#include "Lighting.hlsl"
#include "Fog.hlsl"

void VS(float4 iPos : POSITION,
    float3 iNormal : NORMAL,
    #ifdef NORMALMAP
        float4 iTangent : TANGENT0,
    #endif
    float2 iTexCoord : TEXCOORD0,
    #ifdef VERTEXCOLOR
        float4 iColor : COLOR0,
    #endif
    #ifdef SKINNED
        float4 iBlendWeights : BLENDWEIGHT,
        int4 iBlendIndices : BLENDINDICES,
    #endif
    #ifdef INSTANCED
        float4x3 iModelInstance : TEXCOORD2,
    #endif
    #ifdef BILLBOARD
        float2 iSize : TEXCOORD1,
    #endif
    out float2 oTexCoord : TEXCOORD0,
    out float4 oLightVec : TEXCOORD1,
    #ifndef NORMALMAP
        out float3 oNormal : TEXCOORD2,
    #endif
    #ifdef SPECULAR
        out float3 oEyeVec : TEXCOORD3,
    #endif
    #ifdef SHADOW
        #if defined(DIRLIGHT)
            out float4 oShadowPos[4] : TEXCOORD4,
        #elif defined(SPOTLIGHT)
            out float4 oShadowPos : TEXCOORD4,
        #else
            out float3 oShadowPos : TEXCOORD4,
        #endif
    #endif
    #ifdef SPOTLIGHT
        out float4 oSpotPos : TEXCOORD5,
    #endif
    #ifdef POINTLIGHT
        out float3 oCubeMaskVec : TEXCOORD5,
    #endif
    #ifdef VERTEXCOLOR
        out float4 oColor : COLOR0,
    #endif
    out float4 oPos : POSITION)
{
    float4x3 modelMatrix = iModelMatrix;
    float3 worldPos = GetWorldPos(modelMatrix);
    oPos = GetClipPos(worldPos);
    oTexCoord = GetTexCoord(iTexCoord);

    #ifdef VERTEXCOLOR
        oColor = iColor;
    #endif

    #ifdef NORMALMAP
        float3 oNormal;
        float3 oTangent;
        float3 oBitangent;
    #endif

    oNormal = GetWorldNormal(modelMatrix);
    float4 projWorldPos = float4(worldPos, 1.0);

    #ifdef DIRLIGHT
        oLightVec = float4(cLightDir, GetDepth(oPos));
    #else
        oLightVec = float4((cLightPos.xyz - worldPos) * cLightPos.w, GetDepth(oPos));
    #endif

    #ifdef SHADOW
        // Shadow projection: transform from world space to shadow space
        #if defined(DIRLIGHT)
            oShadowPos[0] = mul(projWorldPos, cLightMatrices[0]);
            oShadowPos[1] = mul(projWorldPos, cLightMatrices[1]);
            oShadowPos[2] = mul(projWorldPos, cLightMatrices[2]);
            oShadowPos[3] = mul(projWorldPos, cLightMatrices[3]);
        #elif defined(SPOTLIGHT)
            oShadowPos = mul(projWorldPos, cLightMatrices[1]);
        #else
            oShadowPos = worldPos - cLightPos.xyz;
        #endif
    #endif

    #ifdef SPOTLIGHT
        // Spotlight projection: transform from world space to projector texture coordinates
        oSpotPos = mul(projWorldPos, cLightMatrices[0]);
    #endif

    #ifdef POINTLIGHT
        oCubeMaskVec = mul(oLightVec.xyz, (float3x3)cLightMatrices[0]);
    #endif

    #ifdef NORMALMAP
        oTangent = GetWorldTangent(modelMatrix);
        oBitangent = cross(oTangent, oNormal) * iTangent.w;
        float3x3 tbn = float3x3(oTangent, oBitangent, oNormal);
        oLightVec.xyz = mul(tbn, oLightVec.xyz);
        #ifdef SPECULAR
            oEyeVec = mul(tbn, cCameraPos - worldPos);
        #endif
    #elif defined(SPECULAR)
        oEyeVec = cCameraPos - worldPos;
    #endif
}

void PS(float2 iTexCoord : TEXCOORD0,
    float4 iLightVec : TEXCOORD1,
    #ifndef NORMALMAP
        float3 iNormal : TEXCOORD2,
    #endif
    #ifdef SPECULAR
        float3 iEyeVec : TEXCOORD3,
    #endif
    #ifdef SHADOW
        #if defined(DIRLIGHT)
            float4 iShadowPos[4] : TEXCOORD4,
        #elif defined(SPOTLIGHT)
            float4 iShadowPos : TEXCOORD4,
        #else
            float3 iShadowPos : TEXCOORD4,
        #endif
    #endif
    #ifdef SPOTLIGHT
        float4 iSpotPos : TEXCOORD5,
    #endif
    #ifdef CUBEMASK
        float3 iCubeMaskVec : TEXCOORD5,
    #endif
    #ifdef VERTEXCOLOR
        float4 iColor : COLOR0,
    #endif
    out float4 oColor : COLOR0)
{
    #ifdef DIFFMAP
        float4 diffInput = tex2D(sDiffMap, iTexCoord);
        #ifdef ALPHAMASK
            if (diffInput.a < 0.5)
                discard;
        #endif
        float4 diffColor = cMatDiffColor * diffInput;
    #else
        float4 diffColor = cMatDiffColor;
    #endif

    #ifdef VERTEXCOLOR
        diffColor *= iColor;
    #endif

    #ifdef NORMALMAP
        float3 normal = DecodeNormal(tex2D(sNormalMap, iTexCoord));
    #else
        float3 normal = normalize(iNormal);
    #endif

    float3 lightDir;
    float3 lightColor;
    float3 finalColor;
    float diff;

    #ifdef DIRLIGHT
        #ifdef NORMALMAP
            lightDir = normalize(iLightVec.xyz);
        #else
            lightDir = iLightVec.xyz;
        #endif
        diff = GetDiffuseDir(normal, lightDir);
    #else
        diff = GetDiffusePointOrSpot(normal, iLightVec.xyz, lightDir);
    #endif

    #ifdef SHADOW
        #if defined(DIRLIGHT)
            float4 shadowPos = GetDirShadowPos(iShadowPos, iLightVec.w);
            diff *= saturate(GetShadow(shadowPos) + GetShadowFade(iLightVec.w));
        #elif defined(SPOTLIGHT)
            diff *= GetShadow(iShadowPos);
        #else
            diff *= GetCubeShadow(iShadowPos);
        #endif
    #endif

    #if defined(SPOTLIGHT)
        lightColor = iSpotPos.w > 0.0 ? tex2Dproj(sLightSpotMap, iSpotPos).rgb * cLightColor.rgb : 0.0;
    #elif defined(CUBEMASK)
        lightColor = texCUBE(sLightCubeMap, iCubeMaskVec).rgb * cLightColor.rgb;
    #else
        lightColor = cLightColor.rgb;
    #endif

    #ifdef SPECULAR
        #ifdef SPECMAP
            float3 specColor = cMatSpecColor.rgb * tex2D(sSpecMap, iTexCoord).g;
        #else
            float3 specColor = cMatSpecColor.rgb;
        #endif
        float spec = GetSpecular(normal, iEyeVec, lightDir, cMatSpecColor.a);
        finalColor = diff * lightColor * (diffColor.rgb + spec * specColor * cLightColor.a);
    #else
        finalColor = diff * lightColor * diffColor.rgb;
    #endif

    #ifdef AMBIENT
        finalColor += cAmbientColor * diffColor.rgb;
        oColor = float4(GetFog(finalColor, iLightVec.w), diffColor.a);
    #else
        oColor = float4(GetLitFog(finalColor, iLightVec.w), diffColor.a);
    #endif
}