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

• Show log

  Commit

• Hash : d7aa0130
  Author :
  Date : 2021-04-26T16:56:15
  

  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>
  

Download

• src/compiler/translator/TranslatorMetalDirect/HoistConstants.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 "compiler/translator/TranslatorMetalDirect/HoistConstants.h"
  #include "compiler/translator/TranslatorMetalDirect/AstHelpers.h"
  #include "compiler/translator/TranslatorMetalDirect/IntermRebuild.h"
  #include "compiler/translator/TranslatorMetalDirect/Layout.h"
  #include "compiler/translator/tree_util/FindFunction.h"
  #include "compiler/translator/tree_util/ReplaceVariable.h"
  
  using namespace sh;
  
  ////////////////////////////////////////////////////////////////////////////////
  
  namespace
  {
  
  class Rewriter : private TIntermRebuild
  {
    private:
      const size_t mMinRequiredSize;
      TIntermSequence mHoistedDeclNodes;
  
    public:
      Rewriter(TCompiler &compiler, size_t minRequiredSize)
          : TIntermRebuild(compiler, true, false), mMinRequiredSize(minRequiredSize)
      {}
  
      PreResult visitDeclarationPre(TIntermDeclaration &declNode) override
      {
          if (getParentFunction())
          {
              Declaration decl  = ViewDeclaration(declNode);
              const TType &type = decl.symbol.getType();
              if (type.getQualifier() == TQualifier::EvqConst)
              {
                  if (decl.initExpr && decl.initExpr->hasConstantValue())
                  {
                      const size_t size = MetalLayoutOf(type).sizeOf;
                      if (size >= mMinRequiredSize)
                      {
                          mHoistedDeclNodes.push_back(&declNode);
                          return nullptr;
                      }
                  }
              }
          }
          return {declNode, VisitBits::Neither};
      }
  
      bool rewrite(TIntermBlock &root, IdGen &idGen)
      {
          if (!rebuildRoot(root))
          {
              return false;
          }
  
          if (mHoistedDeclNodes.empty())
          {
              return true;
          }
  
          root.insertChildNodes(FindFirstFunctionDefinitionIndex(&root), mHoistedDeclNodes);
  
          for (TIntermNode *opaqueDeclNode : mHoistedDeclNodes)
          {
              TIntermDeclaration *declNode = opaqueDeclNode->getAsDeclarationNode();
              ASSERT(declNode);
              const TVariable &oldVar = ViewDeclaration(*declNode).symbol.variable();
              const Name newName      = idGen.createNewName(oldVar.name());
              auto *newVar = new TVariable(&mSymbolTable, newName.rawName(), &oldVar.getType(),
                                           newName.symbolType());
              if (!ReplaceVariable(&mCompiler, &root, &oldVar, newVar))
              {
                  return false;
              }
          }
  
          return true;
      }
  };
  
  }  // anonymous namespace
  
  ////////////////////////////////////////////////////////////////////////////////
  
  bool sh::HoistConstants(TCompiler &compiler,
                          TIntermBlock &root,
                          IdGen &idGen,
                          size_t minRequiredSize)
  {
      return Rewriter(compiler, minRequiredSize).rewrite(root, idGen);
  }
  

Download