92 lines
4.2 KiB
C++
92 lines
4.2 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Containers/Array.h"
|
|
#include "Containers/Map.h"
|
|
#include "Containers/UnrealString.h"
|
|
#include "CoreMinimal.h"
|
|
#include "CrossCompilerCommon.h"
|
|
#include "HAL/Platform.h"
|
|
#include "Serialization/Archive.h"
|
|
|
|
#define UE_API SHADERCOMPILERCOMMON_API
|
|
|
|
// Forward declaration from <spirv_reflect.h>
|
|
struct SpvReflectBlockVariable;
|
|
struct SpvReflectInterfaceVariable;
|
|
struct SpvReflectTypeDescription;
|
|
|
|
// Cross compiler support/common functionality
|
|
namespace CrossCompiler
|
|
{
|
|
class FHlslccHeaderWriter
|
|
{
|
|
public:
|
|
|
|
UE_API void WriteSourceInfo(const TCHAR* VirtualSourceFilePath, const TCHAR* EntryPointName);
|
|
UE_API void WriteCompilerInfo(const TCHAR* CompilerName = TEXT("ShaderConductor"));
|
|
|
|
UE_API void WriteInputAttribute(const SpvReflectInterfaceVariable& Attribute);
|
|
UE_API void WriteInputAttribute(const TCHAR* AttributeName, const TCHAR* TypeSpecifier, int32 Location, bool bLocationPrefix, bool bLocationSuffix);
|
|
UE_API void WriteOutputAttribute(const SpvReflectInterfaceVariable& Attribute);
|
|
UE_API void WriteOutputAttribute(const TCHAR* AttributeName, const TCHAR* TypeSpecifier, int32 Location, bool bLocationPrefix, bool bLocationSuffix);
|
|
UE_API void WriteUniformBlock(const TCHAR* ResourceName, uint32 BindingIndex);
|
|
UE_API void WritePackedGlobal(const TCHAR* ResourceName, EPackedTypeName PackedType, uint32 ByteOffset, uint32 ByteSize);
|
|
UE_API void WritePackedGlobal(const SpvReflectBlockVariable& Variable);
|
|
UE_API void WritePackedUB(uint32 BindingIndex);
|
|
UE_API void WritePackedUBField(const TCHAR* ResourceName, uint32 ByteOffset, uint32 ByteSize);
|
|
UE_API void WritePackedUB(const FString& UBName, uint32 BindingIndex);
|
|
UE_API void WritePackedUBField(const FString& UBName, const TCHAR* ResourceName, uint32 ByteOffset, uint32 ByteSize);
|
|
UE_API void WritePackedUBCopy(uint32 SourceCB, uint32 SourceOffset, uint32 DestCBIndex, uint32 DestCBPrecision, uint32 DestOffset, uint32 Size, bool bGroupFlattenedUBs = false);
|
|
UE_API void WritePackedUBGlobalCopy(uint32 SourceCB, uint32 SourceOffset, uint32 DestCBIndex, uint32 DestCBPrecision, uint32 DestOffset, uint32 Size, bool bGroupFlattenedUBs = false);
|
|
UE_API void WriteSRV(const TCHAR* ResourceName, uint32 BindingIndex, uint32 Count = 1);
|
|
UE_API void WriteSRV(const TCHAR* ResourceName, uint32 BindingIndex, uint32 Count, const TArray<FString>& AssociatedResourceNames);
|
|
UE_API void WriteUAV(const TCHAR* ResourceName, uint32 BindingIndex, uint32 Count = 1);
|
|
UE_API void WriteSamplerState(const TCHAR* ResourceName, uint32 BindingIndex);
|
|
UE_API void WriteNumThreads(uint32 NumThreadsX, uint32 NumThreadsY, uint32 NumThreadsZ);
|
|
UE_API void WriteAccelerationStructures(const TCHAR* ResourceName, uint32 BindingIndex);
|
|
|
|
UE_API void WriteSideTable(const TCHAR* ResourceName, uint32 SideTableIndex);
|
|
UE_API void WriteArgumentBuffers(uint32 BindingIndex, const TArray<uint32>& ResourceIndices);
|
|
|
|
/** Returns the finalized meta data. */
|
|
UE_API FString ToString() const;
|
|
static UE_API EPackedTypeName EncodePackedGlobalType(const SpvReflectTypeDescription& TypeDescription, bool bHalfPrecision = false);
|
|
|
|
static UE_API void WriteIOAttribute(FString& OutMetaData, const SpvReflectInterfaceVariable& Attribute, bool bIsInput);
|
|
|
|
private:
|
|
void WriteIOAttribute(FString& OutMetaData, const TCHAR* AttributeName, const TCHAR* TypeSpecifier, int32 Location, bool bLocationPrefix, bool bLocationSuffix);
|
|
|
|
private:
|
|
struct FMetaDataStrings
|
|
{
|
|
FString SourceInfo;
|
|
FString CompilerInfo;
|
|
FString InputAttributes;
|
|
FString OutputAttributes;
|
|
FString UniformBlocks;
|
|
FString PackedGlobals;
|
|
TMap<FString, FString> PackedUBs;
|
|
TMap<FString, FString> PackedUBFields;
|
|
FString PackedUBCopies;
|
|
FString PackedUBGlobalCopies;
|
|
FString SRVs; // Shader resource views (SRV) and samplers
|
|
FString UAVs; // Unordered access views (UAV)
|
|
FString SamplerStates;
|
|
FString NumThreads;
|
|
FString ExternalTextures; // External texture resources (Vulkan ES3.1 profile only)
|
|
FString SideTable; // Side table for additional indices, e.,g. "spvBufferSizeConstants(31)" (Metal only)
|
|
FString ArgumentBuffers; // Indirect argument buffers (Metal only)
|
|
FString AccelerationStructures;
|
|
};
|
|
|
|
private:
|
|
FMetaDataStrings Strings;
|
|
|
|
};
|
|
}
|
|
|
|
#undef UE_API
|