root.hlsl

#pragma pack_matrix(column_major)

struct VIn {
    float4 position : POSITION;
#if defined(ATTRIB_TEXCOORD0)
    float2 texCoord0 : TEXCOORD;
#endif
#if defined (ATTRIB_JOINT_INDEX)
    float3 jointIndex : BLENDINDICES;
#endif
#if defined (ATTRIB_JOINT_WEIGHT)
    float3 jointWeight : BLENDWEIGHT;
#endif
};

struct VOut
{
    float4 position : SV_POSITION;
#if defined (ATTRIB_TEXCOORD0)
    float2 texCoord0 : TEXCOORD;
#endif
};

cbuffer VS_CONSTANT_BUFFER : register(b0) {
    float4x4 objectModelMatrix;
    float4x4 objectNormalMatrix;
    float2 uvScale;
    float2 uvOffset;
    uint layer;
};

cbuffer VS_CONSTANT_BUFFER : register(b1) {
    float3 cameraPosition;
    uint2 cameraScreenSize;
    float4x4 cameraViewMatrix;
    float4x4 cameraProjectionMatrix;
};

#if defined(CONSTANT_BUFFER_SKINNING_MATRICES)
cbuffer VS_CONSTANT_BUFFER : register(b2) {
    float4x4 skinningMatrices[70];
};
#endif

VOut VShader(VIn input) {
    VOut output;

    float4x4 modelMatrix;
#if defined(CAP_SKINNING)
    modelMatrix = float4x4(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
    modelMatrix += skinningMatrices[int(input.jointIndex.x)] * input.jointWeight.x;
    modelMatrix += skinningMatrices[int(input.jointIndex.y)] * input.jointWeight.y;
    modelMatrix += skinningMatrices[int(input.jointIndex.z)] * input.jointWeight.z;
#else
    modelMatrix = objectModelMatrix;
#endif

    float4 position_worldspace = mul(modelMatrix, input.position);
    float4 position_cameraspace = mul(cameraViewMatrix, position_worldspace);
    output.position = mul(cameraProjectionMatrix, position_cameraspace);

#if defined(ATTRIB_TEXCOORD0)
    output.texCoord0 = input.texCoord0;
#endif
    
    return output;
}

#if defined (IS_PIXEL)

#if defined (RESOURCE_LIGHT_GRID)
struct LightGrid {
    uint offset;
    uint lightsCount;
    uint projectorsCount;
}

StructuredBuffer<LightGrid> lightGridBuffer : register(t3);
#endif

Texture2D texture0 : register(t0);
Texture2D shaderTexture : register(t0);
SamplerState SampleType;

float4 PShader(VOut input) : SV_TARGET
{
    float4 result = float4(1, 1, 1, 1);

#if defined(RESOURCE_TEXTURE0)
    float4 textureColor = shaderTexture.Sample(SampleType, input.texCoord0);
    //textureColor *= float4(3, 1, 1, 1);
    result = textureColor;
#endif

#if defined (RESOURCE_LIGHT_GRID)
    float TILE_SIZE = 32.0;
    float2 screenSize = float2(cameraScreenSize);
    int2 tilesCount = int2(ceil(screenSize / TILE_SIZE));


    int tileX = int(floor(SV_Position.x / TILE_SIZE));
    int tileY = int(floor(SV_Position.y / TILE_SIZE));
    vec2 tileCount = ceil(vec2(screenSize / TILE_SIZE));

    int tileIndex = tileX + tilesCount.x * tileY;
    LightGrid gridItem = lightGridBuffer[tileIndex];

    uint lightOffset = gridItem.offset;
    uint pointLightCount = gridItem.lightsCount & 0x000fffu;
    uint spotLightCount = gridItem.lightsCount >> 16;

    result *= 0.2;
    result += float4(0.2 * pointLightCount, 0.2 * spotLightCount, 0, 0);
#endif

    float gamma = 2.2;
    result.rgb = pow(result.rgb, 1.0/gamma);
    return result;
}

#endif