diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockOutputHLSL.h | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockOutputHLSL.h')
-rw-r--r-- | gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockOutputHLSL.h | 86 |
1 files changed, 86 insertions, 0 deletions
diff --git a/gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockOutputHLSL.h b/gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockOutputHLSL.h new file mode 100644 index 0000000000..2acf06a0ea --- /dev/null +++ b/gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockOutputHLSL.h @@ -0,0 +1,86 @@ +// +// Copyright 2018 The ANGLE Project Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. +// +// ShaderStorageBlockOutputHLSL: A traverser to translate a buffer variable of shader storage block +// to an offset of RWByteAddressBuffer. +// + +#ifndef COMPILER_TRANSLATOR_SHADERSTORAGEBLOCKOUTPUTHLSL_H_ +#define COMPILER_TRANSLATOR_SHADERSTORAGEBLOCKOUTPUTHLSL_H_ + +#include "compiler/translator/IntermNode.h" +#include "compiler/translator/ShaderStorageBlockFunctionHLSL.h" +#include "compiler/translator/blocklayout.h" + +namespace sh +{ +class ResourcesHLSL; +class OutputHLSL; +class TSymbolTable; + +struct TReferencedBlock : angle::NonCopyable +{ + POOL_ALLOCATOR_NEW_DELETE + TReferencedBlock(const TInterfaceBlock *block, const TVariable *instanceVariable); + const TInterfaceBlock *block; + const TVariable *instanceVariable; // May be nullptr if the block is not instanced. +}; + +// Maps from uniqueId to a variable. +using ReferencedInterfaceBlocks = std::map<int, const TReferencedBlock *>; + +// Used to save shader storage block field member information. +using BlockMemberInfoMap = std::map<const TField *, BlockMemberInfo>; + +using ShaderVarToFieldMap = std::map<std::string, const TField *>; + +class ShaderStorageBlockOutputHLSL +{ + public: + ShaderStorageBlockOutputHLSL(OutputHLSL *outputHLSL, + ResourcesHLSL *resourcesHLSL, + const std::vector<InterfaceBlock> &shaderStorageBlocks); + + ~ShaderStorageBlockOutputHLSL(); + + // This writes part of the function call to store a value to a SSBO to the output stream. After + // calling this, ", <stored value>)" should be written to the output stream to complete the + // function call. + void outputStoreFunctionCallPrefix(TIntermTyped *node); + // This writes the function call to load a SSBO value to the output stream. + void outputLoadFunctionCall(TIntermTyped *node); + // This writes the function call to get the lengh of unsized array member of SSBO. + void outputLengthFunctionCall(TIntermTyped *node); + // Writes the atomic memory function calls for SSBO. + void outputAtomicMemoryFunctionCallPrefix(TIntermTyped *node, TOperator op); + + void writeShaderStorageBlocksHeader(GLenum shaderType, TInfoSinkBase &out) const; + + private: + void traverseSSBOAccess(TIntermTyped *node, SSBOMethod method); + TIntermTyped *traverseNode(TInfoSinkBase &out, + TIntermTyped *node, + BlockMemberInfo *blockMemberInfo); + int getMatrixStride(TIntermTyped *node, + TLayoutBlockStorage storage, + bool rowMajor, + bool *isRowMajor) const; + TIntermTyped *writeEOpIndexDirectOrIndirectOutput(TInfoSinkBase &out, + TIntermBinary *node, + BlockMemberInfo *blockMemberInfo); + // Common part in dot operations. + TIntermTyped *createFieldOffset(const TField *field, BlockMemberInfo *blockMemberInfo); + void collectShaderStorageBlocks(TIntermTyped *node); + OutputHLSL *mOutputHLSL; + ShaderStorageBlockFunctionHLSL *mSSBOFunctionHLSL; + ResourcesHLSL *mResourcesHLSL; + ReferencedInterfaceBlocks mReferencedShaderStorageBlocks; + + BlockMemberInfoMap mBlockMemberInfoMap; + const std::vector<InterfaceBlock> &mShaderStorageBlocks; +}; +} // namespace sh + +#endif // COMPILER_TRANSLATOR_SHADERSTORAGEBLOCKOUTPUTHLSL_H_ |