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

Branch : - branch - main - tag -

• Show log

  Commit

• Author : Kyle Piddington
  Date : 2021-04-26 16:56:15
  Hash : d7aa0130
  Message : Upstream Apple's direct-to-Metal backend: compile translator. This change is meant to merge the translator changes from Apple's direct-to-Metal backend. Taken from Kyle Piddington's CL: https://chromium-review.googlesource.com/c/angle/angle/+/2857366/ The goal of this CL is to merge the translator code in a state that compiles, but not to switch the Metal backend over to use this translator backend yet. Bug: angleproject:5505 Change-Id: I68a6354604498cd5fd1eb96c13fc56f3b38f2bd0 Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/2897536 Reviewed-by: Jonah Ryan-Davis <jonahr@google.com> Commit-Queue: Jonah Ryan-Davis <jonahr@google.com>

• 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/TranslatorMetalDirect.h"
  #include "compiler/translator/TranslatorMetalDirect/AstHelpers.h"
  #include "compiler/translator/TranslatorMetalDirect/IntermRebuild.h"
  #include "compiler/translator/TranslatorMetalDirect/ReduceInterfaceBlocks.h"
  #include "compiler/translator/tree_ops/SeparateDeclarations.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)
  {
      Reducer reducer(compiler, idGen);
      if (!reducer.rebuildRoot(root))
      {
          return false;
      }
  
      if (!SeparateDeclarations(&compiler, &root))
      {
          return false;
      }
  
      return true;
  }