kc3-lang/angle/src/compiler/translator/TranslatorMetalDirect/RewriteGlobalQualifierDecls.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/RewriteGlobalQualifierDecls.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/RewriteGlobalQualifierDecls.h"
  #include "compiler/translator/tree_util/IntermRebuild.h"
  
  using namespace sh;
  
  namespace
  {
  
  class FindDeclaredGlobals : public TIntermRebuild
  {
    public:
      std::unordered_set<const TVariable *> mDeclaredGlobals;
  
      FindDeclaredGlobals(TCompiler &compiler) : TIntermRebuild(compiler, true, false) {}
  
      PreResult visitDeclarationPre(TIntermDeclaration &declNode) override
      {
          TIntermNode *declaratorNode = declNode.getChildNode(0);
          TIntermSymbol *symbolNode   = nullptr;
  
          if (TIntermBinary *initNode = declaratorNode->getAsBinaryNode())
          {
              symbolNode = initNode->getLeft()->getAsSymbolNode();
          }
          else
          {
              symbolNode = declaratorNode->getAsSymbolNode();
          }
  
          ASSERT(symbolNode);
          const TVariable &var = symbolNode->variable();
  
          mDeclaredGlobals.insert(&var);
  
          return {declNode, VisitBits::Neither};
      }
  
      PreResult visitFunctionDefinitionPre(TIntermFunctionDefinition &node) override
      {
          return {node, VisitBits::Neither};
      }
  };
  
  class Rewriter : public TIntermRebuild
  {
      const std::unordered_set<const TVariable *> &mDeclaredGlobals;
      Invariants &mOutInvariants;
  
    public:
      Rewriter(TCompiler &compiler,
               const std::unordered_set<const TVariable *> &declaredGlobals,
               Invariants &outInvariants)
          : TIntermRebuild(compiler, true, false),
            mDeclaredGlobals(declaredGlobals),
            mOutInvariants(outInvariants)
      {}
  
      PreResult visitGlobalQualifierDeclarationPre(
          TIntermGlobalQualifierDeclaration &gqDeclNode) override
      {
          TIntermSymbol &symbolNode = *gqDeclNode.getSymbol();
          const TVariable &var      = symbolNode.variable();
  
          if (gqDeclNode.isInvariant())
          {
              mOutInvariants.insert(var);
          }
  
          if (mDeclaredGlobals.find(&var) == mDeclaredGlobals.end())
          {
              return *new TIntermDeclaration{&symbolNode};
          }
          return nullptr;
      }
  
      PreResult visitDeclarationPre(TIntermDeclaration &node) override
      {
          return {node, VisitBits::Neither};
      }
  
      PreResult visitFunctionDefinitionPre(TIntermFunctionDefinition &node) override
      {
          return {node, VisitBits::Neither};
      }
  };
  
  }  // anonymous namespace
  
  bool sh::RewriteGlobalQualifierDecls(TCompiler &compiler,
                                       TIntermBlock &root,
                                       Invariants &outInvariants)
  {
      FindDeclaredGlobals fdg(compiler);
      if (!fdg.rebuildRoot(root))
      {
          UNREACHABLE();
          return false;
      }
  
      Rewriter rewriter(compiler, fdg.mDeclaredGlobals, outInvariants);
      if (!rewriter.rebuildRoot(root))
      {
          return false;
      }
  
      return true;
  }
  

Download