diff options
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_ |