kc3-lang/angle/src/compiler/translator/TranslatorMetalDirect/DiscoverDependentFunctions.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/DiscoverDependentFunctions.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 <cstring>
  #include <unordered_map>
  #include <unordered_set>
  
  #include "compiler/translator/TranslatorMetalDirect/DiscoverDependentFunctions.h"
  #include "compiler/translator/TranslatorMetalDirect/DiscoverEnclosingFunctionTraverser.h"
  #include "compiler/translator/TranslatorMetalDirect/MapFunctionsToDefinitions.h"
  
  using namespace sh;
  
  ////////////////////////////////////////////////////////////////////////////////
  
  namespace
  {
  
  class Discoverer : public DiscoverEnclosingFunctionTraverser
  {
    private:
      const std::function<bool(const TVariable &)> &mVars;
      const FunctionToDefinition &mFuncToDef;
      std::unordered_set<const TFunction *> mNonDepFunctions;
  
    public:
      std::unordered_set<const TFunction *> mDepFunctions;
  
    public:
      Discoverer(const std::function<bool(const TVariable &)> &vars,
                 const FunctionToDefinition &funcToDef)
          : DiscoverEnclosingFunctionTraverser(true, false, true), mVars(vars), mFuncToDef(funcToDef)
      {}
  
      void visitSymbol(TIntermSymbol *symbolNode) override
      {
          const TVariable &var = symbolNode->variable();
          if (!mVars(var))
          {
              return;
          }
          const TFunction *owner = discoverEnclosingFunction(symbolNode);
          ASSERT(owner);
          mDepFunctions.insert(owner);
      }
  
      bool visitAggregate(Visit visit, TIntermAggregate *aggregateNode) override
      {
          if (visit != Visit::PreVisit)
          {
              return true;
          }
  
          if (!aggregateNode->isConstructor())
          {
              const TFunction *func = aggregateNode->getFunction();
  
              if (mNonDepFunctions.find(func) != mNonDepFunctions.end())
              {
                  return true;
              }
  
              if (mDepFunctions.find(func) == mDepFunctions.end())
              {
                  auto it = mFuncToDef.find(func);
                  if (it == mFuncToDef.end())
                  {
                      return true;
                  }
  
                  // Recursion is banned in GLSL, so I believe AngleIR has this property too.
                  // This implementation assumes (direct and mutual) recursion is prohibited.
                  TIntermFunctionDefinition &funcDefNode = *it->second;
                  funcDefNode.traverse(this);
                  if (mNonDepFunctions.find(func) != mNonDepFunctions.end())
                  {
                      return true;
                  }
                  ASSERT(mDepFunctions.find(func) != mDepFunctions.end());
              }
  
              const TFunction *owner = discoverEnclosingFunction(aggregateNode);
              ASSERT(owner);
              mDepFunctions.insert(owner);
          }
  
          return true;
      }
  
      bool visitFunctionDefinition(Visit visit, TIntermFunctionDefinition *funcDefNode) override
      {
          const TFunction *func = funcDefNode->getFunction();
  
          if (visit != Visit::PostVisit)
          {
              if (mDepFunctions.find(func) != mDepFunctions.end())
              {
                  return false;
              }
  
              if (mNonDepFunctions.find(func) != mNonDepFunctions.end())
              {
                  return false;
              }
  
              return true;
          }
  
          if (mDepFunctions.find(func) == mDepFunctions.end())
          {
              mNonDepFunctions.insert(func);
          }
  
          return true;
      }
  };
  
  }  // namespace
  
  std::unordered_set<const TFunction *> sh::DiscoverDependentFunctions(
      TIntermBlock &root,
      const std::function<bool(const TVariable &)> &vars)
  {
      const FunctionToDefinition funcToDef = MapFunctionsToDefinitions(root);
      Discoverer discoverer(vars, funcToDef);
      root.traverse(&discoverer);
      return std::move(discoverer.mDepFunctions);
  }
  

Download