MaterialCacheShaders.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "MeshMaterialShader.h"
#include "MaterialCache/MaterialCacheAttribute.h"
#include "ShaderParameterStruct.h"
 
class FMaterialCacheUnwrapVSBase : public FMeshMaterialShader
{
    typedef FMeshMaterialShader Super;
    DECLARE_INLINE_TYPE_LAYOUT(FMaterialCacheUnwrapVSBase, NonVirtual);
 
public:
    class FTagIndex          : SHADER_PERMUTATION_INT("TAG_INDEX", MaterialCacheMaxTagsPerPrimitive);
    using FPermutationDomain = TShaderPermutationDomain<FTagIndex>;
    
    FMaterialCacheUnwrapVSBase(const ShaderMetaType::CompiledShaderInitializerType& Initializer) : FMeshMaterialShader(Initializer)
    {
        
    }
 
    FMaterialCacheUnwrapVSBase() = default;
};
 
template<bool bSupportsViewportFromVS>
class FMaterialCacheUnwrapVS : public FMaterialCacheUnwrapVSBase
{
    DECLARE_SHADER_TYPE(FMaterialCacheUnwrapVS, MeshMaterial);
 
public: 
    FMaterialCacheUnwrapVS(const ShaderMetaType::CompiledShaderInitializerType& Initializer) : FMaterialCacheUnwrapVSBase(Initializer)
    {
        
    }
    
    FMaterialCacheUnwrapVS() = default;
    
    static bool ShouldCompilePermutation(const FMeshMaterialShaderPermutationParameters& Parameters);
    static void ModifyCompilationEnvironment(const FMaterialShaderPermutationParameters& Parameters, FShaderCompilerEnvironment& OutEnvironment);
};
 
class FMaterialCacheUnwrapPS : public FMeshMaterialShader
{
    DECLARE_SHADER_TYPE(FMaterialCacheUnwrapPS, MeshMaterial);
 
public:
    class FTagIndex          : SHADER_PERMUTATION_INT("TAG_INDEX", MaterialCacheMaxTagsPerPrimitive);
    using FPermutationDomain = TShaderPermutationDomain<FTagIndex>;
    
    FMaterialCacheUnwrapPS(const ShaderMetaType::CompiledShaderInitializerType& Initializer) : FMeshMaterialShader(Initializer)
    {
        
    }
 
    FMaterialCacheUnwrapPS() = default;
 
    static bool ShouldCompilePermutation(const FMeshMaterialShaderPermutationParameters& Parameters);
    static void ModifyCompilationEnvironment(const FMaterialShaderPermutationParameters& Parameters, FShaderCompilerEnvironment& OutEnvironment);
};
 
class FMaterialCacheNaniteShadeCS : public FMeshMaterialShader
{
    DECLARE_SHADER_TYPE(FMaterialCacheNaniteShadeCS, MeshMaterial);
 
public:
    class FTagIndex          : SHADER_PERMUTATION_INT("TAG_INDEX", MaterialCacheMaxTagsPerPrimitive);
    using FPermutationDomain = TShaderPermutationDomain<FTagIndex>;
    
    FMaterialCacheNaniteShadeCS(const ShaderMetaType::CompiledShaderInitializerType& Initializer);
    FMaterialCacheNaniteShadeCS() = default;
 
    static bool ShouldCompilePermutation(const FMeshMaterialShaderPermutationParameters& Parameters);
    static void ModifyCompilationEnvironment(const FMeshMaterialShaderPermutationParameters& Parameters, FShaderCompilerEnvironment& OutEnvironment);
 
    void SetPassParameters(FRHIBatchedShaderParameters& BatchedParameters, const FUintVector4& PassData, FRHIShaderResourceView* PageIndirections);
 
private:
    LAYOUT_FIELD(FShaderResourceParameter, PageIndirectionsParam);
    LAYOUT_FIELD(FShaderParameter, PassDataParam);
};
 
class FMaterialCacheShadeCS : public FMeshMaterialShader
{
    DECLARE_SHADER_TYPE(FMaterialCacheShadeCS, MeshMaterial);
 
public:
    FMaterialCacheShadeCS(const ShaderMetaType::CompiledShaderInitializerType& Initializer);
    FMaterialCacheShadeCS() = default;
 
    static bool ShouldCompilePermutation(const FMeshMaterialShaderPermutationParameters& Parameters);
    static void ModifyCompilationEnvironment(const FMeshMaterialShaderPermutationParameters& Parameters, FShaderCompilerEnvironment& OutEnvironment);
 
    void SetPassParameters(FRHIBatchedShaderParameters& BatchedParameters, const FUintVector4& PassData, FRHIShaderResourceView* PageIndirections);
 
private:
    LAYOUT_FIELD(FShaderResourceParameter, PageIndirectionsParam);
    LAYOUT_FIELD(FShaderParameter, PassDataParam);
};
 
class FMaterialCacheABufferWritePagesCS : public FGlobalShader
{
    DECLARE_GLOBAL_SHADER(FMaterialCacheABufferWritePagesCS);
    SHADER_USE_PARAMETER_STRUCT(FMaterialCacheABufferWritePagesCS, FGlobalShader);
 
public:
    BEGIN_SHADER_PARAMETER_STRUCT(FParameters, )
        SHADER_PARAMETER_RDG_BUFFER_SRV(StructuredBuffer<FMaterialCacheBinData>, PageWriteData)
        SHADER_PARAMETER_RDG_TEXTURE_UAV(RWTexture2D<uint4>, RWVTLayerCompressed)
        SHADER_PARAMETER_RDG_TEXTURE_UAV(RWTexture2D<float4>, RWVTLayerUncompressed)
        SHADER_PARAMETER_RDG_TEXTURE_SRV(Texture2DArray<float4>, ABuffer)
        SHADER_PARAMETER_SAMPLER(SamplerState, TextureSampler)
        SHADER_PARAMETER(uint32, BlockOrThreadCount)
        SHADER_PARAMETER(FVector2f, TexelSize)
        SHADER_PARAMETER(uint32, bSRGB)
    END_SHADER_PARAMETER_STRUCT()
    
    class FCompressMode        : SHADER_PERMUTATION_INT("COMPRESS_MODE", 7);
    using FPermutationDomain = TShaderPermutationDomain<FCompressMode>;
 
    static int32 GetCompressMode(EPixelFormat Format);
    
    static bool ShouldCompilePermutation(const FGlobalShaderPermutationParameters& Parameters);
    static void ModifyCompilationEnvironment(const FShaderPermutationParameters& Parameters, FShaderCompilerEnvironment& OutEnvironment);
};
 
struct FNaniteMaterialCacheData
{
    TShaderRef<FMaterialCacheNaniteShadeCS> TypedShader;
};