kc3-lang/angle/src/compiler/translator/TranslatorMetalDirect/ReduceInterfaceBlocks.cpp

• Show log

  Commit

• Hash : 958dce52
  Author :
  Date : 2021-09-09T14:36:37
  

  Move IntermRebuild.h/cpp to tree_util
  
  Bug: angleproject:5505
  Change-Id: I7b68057fff0a0eb0d86c1aed01599fa9dbb7db3b
  Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/3152167
  Commit-Queue: Gregg Tavares <gman@chromium.org>
  Reviewed-by: Kenneth Russell <kbr@chromium.org>
  Reviewed-by: Shahbaz Youssefi <syoussefi@chromium.org>
  

Download

• src/compiler/translator/TranslatorMetalDirect/ReduceInterfaceBlocks.cpp

• //
  // Copyright 2020 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.
  //
  
  #include <algorithm>
  #include <unordered_map>
  
  #include "compiler/translator/SymbolTable.h"
  #include "compiler/translator/TranslatorMetalDirect.h"
  #include "compiler/translator/TranslatorMetalDirect/AstHelpers.h"
  #include "compiler/translator/TranslatorMetalDirect/ReduceInterfaceBlocks.h"
  #include "compiler/translator/tree_ops/SeparateDeclarations.h"
  #include "compiler/translator/tree_util/IntermRebuild.h"
  
  using namespace sh;
  
  ////////////////////////////////////////////////////////////////////////////////
  
  namespace
  {
  
  class Reducer : public TIntermRebuild
  {
      std::unordered_map<const TInterfaceBlock *, const TVariable *> mLiftedMap;
      std::unordered_map<const TVariable *, const TVariable *> mInstanceMap;
      IdGen &mIdGen;
  
    public:
      Reducer(TCompiler &compiler, IdGen &idGen)
          : TIntermRebuild(compiler, true, false), mIdGen(idGen)
      {}
  
      PreResult visitDeclarationPre(TIntermDeclaration &declNode) override
      {
          ASSERT(declNode.getChildCount() == 1);
          TIntermNode &node = *declNode.getChildNode(0);
  
          if (TIntermSymbol *symbolNode = node.getAsSymbolNode())
          {
              const TVariable &var        = symbolNode->variable();
              const TType &type           = var.getType();
              const SymbolType symbolType = var.symbolType();
              if (const TInterfaceBlock *interfaceBlock = type.getInterfaceBlock())
              {
                  if (symbolType == SymbolType::Empty)
                  {
                      // Create instance variable
                      auto &structure =
                          *new TStructure(&mSymbolTable, interfaceBlock->name(),
                                          &interfaceBlock->fields(), interfaceBlock->symbolType());
                      auto &structVar = CreateStructTypeVariable(mSymbolTable, structure);
  
                      auto &instanceVar = CreateInstanceVariable(
                          mSymbolTable, structure, mIdGen.createNewName(interfaceBlock->name()),
                          TQualifier::EvqBuffer, &type.getArraySizes());
                      mLiftedMap[interfaceBlock] = &instanceVar;
  
                      TIntermNode *replacements[] = {
                          new TIntermDeclaration{new TIntermSymbol(&structVar)},
                          new TIntermDeclaration{new TIntermSymbol(&instanceVar)}};
                      return PreResult::Multi(std::begin(replacements), std::end(replacements));
                  }
                  else
                  {
                      ASSERT(type.getQualifier() == TQualifier::EvqUniform);
  
                      auto &structure =
                          *new TStructure(&mSymbolTable, interfaceBlock->name(),
                                          &interfaceBlock->fields(), interfaceBlock->symbolType());
                      auto &structVar = CreateStructTypeVariable(mSymbolTable, structure);
                      auto &instanceVar =
                          CreateInstanceVariable(mSymbolTable, structure, Name(var),
                                                 TQualifier::EvqBuffer, &type.getArraySizes());
  
                      mInstanceMap[&var] = &instanceVar;
  
                      TIntermNode *replacements[] = {
                          new TIntermDeclaration{new TIntermSymbol(&structVar)},
                          new TIntermDeclaration{new TIntermSymbol(&instanceVar)}};
                      return PreResult::Multi(std::begin(replacements), std::end(replacements));
                  }
              }
          }
  
          return {declNode, VisitBits::Both};
      }
  
      PreResult visitSymbolPre(TIntermSymbol &symbolNode) override
      {
          const TVariable &var = symbolNode.variable();
          {
              auto it = mInstanceMap.find(&var);
              if (it != mInstanceMap.end())
              {
                  return *new TIntermSymbol(it->second);
              }
          }
          if (const TInterfaceBlock *ib = var.getType().getInterfaceBlock())
          {
              auto it = mLiftedMap.find(ib);
              if (it != mLiftedMap.end())
              {
                  return AccessField(*(it->second), var.name());
              }
          }
          return symbolNode;
      }
  };
  
  }  // anonymous namespace
  
  ////////////////////////////////////////////////////////////////////////////////
  
  bool sh::ReduceInterfaceBlocks(TCompiler &compiler,
                                 TIntermBlock &root,
                                 IdGen &idGen,
                                 TSymbolTable *symbolTable)
  {
      Reducer reducer(compiler, idGen);
      if (!reducer.rebuildRoot(root))
      {
          return false;
      }
  
      if (!SeparateDeclarations(&compiler, &root, symbolTable))
      {
          return false;
      }
  
      return true;
  }
  

Download