1 files changed, 94 insertions, 0 deletions
diff --git a/gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockFunctionHLSL.h b/gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockFunctionHLSL.h
new file mode 100644
index 0000000000..0d3cd0291b
--- /dev/null
+++ b/gfx/angle/checkout/src/compiler/translator/ShaderStorageBlockFunctionHLSL.h
@@ -0,0 +1,94 @@
+//
+// 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 ssbo_access_chain to an offset of
+// RWByteAddressBuffer.
+//     //EOpIndexDirectInterfaceBlock
+//     ssbo_variable :=
+//       | the name of the SSBO
+//       | the name of a variable in an SSBO backed interface block
+
+//     // EOpIndexInDirect
+//     // EOpIndexDirect
+//     ssbo_array_indexing := ssbo_access_chain[expr_no_ssbo]
+
+//     // EOpIndexDirectStruct
+//     ssbo_structure_access := ssbo_access_chain.identifier
+
+//     ssbo_access_chain :=
+//       | ssbo_variable
+//       | ssbo_array_indexing
+//       | ssbo_structure_access
+//
+
+#ifndef COMPILER_TRANSLATOR_SHADERSTORAGEBLOCKFUNCTIONHLSL_H_
+#define COMPILER_TRANSLATOR_SHADERSTORAGEBLOCKFUNCTIONHLSL_H_
+
+#include <set>
+
+#include "compiler/translator/InfoSink.h"
+#include "compiler/translator/Types.h"
+
+namespace sh
+{
+
+class TIntermSwizzle;
+enum class SSBOMethod
+{
+    LOAD,
+    STORE,
+    LENGTH,
+    ATOMIC_ADD,
+    ATOMIC_MIN,
+    ATOMIC_MAX,
+    ATOMIC_AND,
+    ATOMIC_OR,
+    ATOMIC_XOR,
+    ATOMIC_EXCHANGE,
+    ATOMIC_COMPSWAP
+};
+
+class ShaderStorageBlockFunctionHLSL final : angle::NonCopyable
+{
+  public:
+    TString registerShaderStorageBlockFunction(const TType &type,
+                                               SSBOMethod method,
+                                               TLayoutBlockStorage storage,
+                                               bool rowMajor,
+                                               int matrixStride,
+                                               int unsizedArrayStride,
+                                               TIntermSwizzle *node);
+
+    void shaderStorageBlockFunctionHeader(TInfoSinkBase &out);
+
+  private:
+    struct ShaderStorageBlockFunction
+    {
+        bool operator<(const ShaderStorageBlockFunction &rhs) const;
+        TString functionName;
+        TString typeString;
+        SSBOMethod method;
+        TType type;
+        bool rowMajor;
+        int matrixStride;
+        int unsizedArrayStride;
+        TVector<int> swizzleOffsets;
+        bool isDefaultSwizzle;
+    };
+
+    static void OutputSSBOLoadFunctionBody(TInfoSinkBase &out,
+                                           const ShaderStorageBlockFunction &ssboFunction);
+    static void OutputSSBOStoreFunctionBody(TInfoSinkBase &out,
+                                            const ShaderStorageBlockFunction &ssboFunction);
+    static void OutputSSBOLengthFunctionBody(TInfoSinkBase &out, int unsizedArrayStride);
+    static void OutputSSBOAtomicMemoryFunctionBody(TInfoSinkBase &out,
+                                                   const ShaderStorageBlockFunction &ssboFunction);
+    using ShaderStorageBlockFunctionSet = std::set<ShaderStorageBlockFunction>;
+    ShaderStorageBlockFunctionSet mRegisteredShaderStorageBlockFunctions;
+};
+
+}  // namespace sh
+
+#endif  // COMPILER_TRANSLATOR_SHADERSTORAGEBLOCKFUNCTIONHLSL_H_