kc3-lang/angle/src/compiler/translator/ShaderStorageBlockFunctionHLSL.h

Branch : - branch - main - tag -

• Show log

  Commit

• Author : Qin Jiajia
  Date : 2018-12-10 13:31:00
  Hash : 4622905b
  Message : ES31: Add atomic memory functions for SSBO Due to SSBO is translated to RWByteAddressBuffer in HLSL, the corresponding atomic memory functions atomic* will be translated to RWByteAddressBuffer.Interlocked*. The translation is like below: atomicAdd(instanceName.data[0], 5u); // becomes uint _ssbo_atomicAdd_uint(RWByteAddressBuffer buffer, uint loc, uint value) { uint original_value; buffer.InterlockedAdd(loc, value, original_value); return original_value; } _ssbo_atomicAdd_uint(_instanceName, 0 + 16 * 0, 5); Bug: angleproject:1951 Change-Id: If2af8bedb67a4135b443d2512d43c6058a78888d Reviewed-on: https://chromium-review.googlesource.com/c/1370676 Reviewed-by: Corentin Wallez <cwallez@chromium.org> Commit-Queue: Jiajia Qin <jiajia.qin@intel.com>

• src/compiler/translator/ShaderStorageBlockFunctionHLSL.h

• //
  // 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_