kc3-lang/angle/src/compiler/translator/TranslatorMetalDirect/WrapMain.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/WrapMain.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/WrapMain.h"
  #include "compiler/translator/Compiler.h"
  #include "compiler/translator/TranslatorMetalDirect/AstHelpers.h"
  
  using namespace sh;
  
  ////////////////////////////////////////////////////////////////////////////////
  
  namespace
  {
  
  class Wrapper : public TIntermTraverser
  {
    private:
      IdGen &mIdGen;
  
    public:
      Wrapper(TSymbolTable &symbolTable, IdGen &idGen)
          : TIntermTraverser(false, false, true, &symbolTable), mIdGen(idGen)
      {}
  
      bool visitBlock(Visit, TIntermBlock *blockNode) override
      {
          if (blockNode != getRootNode())
          {
              return true;
          }
  
          for (TIntermNode *node : *blockNode->getSequence())
          {
              if (TIntermFunctionDefinition *funcDefNode = node->getAsFunctionDefinition())
              {
                  const TFunction &func = *funcDefNode->getFunction();
                  if (func.isMain())
                  {
                      visitMain(*blockNode, funcDefNode);
                      break;
                  }
              }
          }
  
          return true;
      }
  
    private:
      void visitMain(TIntermBlock &root, TIntermFunctionDefinition *funcDefNode)
      {
          const TFunction &func = *funcDefNode->getFunction();
          ASSERT(func.isMain());
          ASSERT(func.getReturnType().getBasicType() == TBasicType::EbtVoid);
          ASSERT(func.getParamCount() == 0);
  
          const TFunction &externalMainFunc = *funcDefNode->getFunction();
          const TFunction &internalMainFunc = CloneFunction(*mSymbolTable, mIdGen, externalMainFunc);
  
          TIntermFunctionPrototype *externalMainProto = funcDefNode->getFunctionPrototype();
          TIntermFunctionPrototype *internalMainProto =
              new TIntermFunctionPrototype(&internalMainFunc);
  
          TIntermBlock *externalMainBody = new TIntermBlock();
          externalMainBody->appendStatement(
              TIntermAggregate::CreateFunctionCall(internalMainFunc, new TIntermSequence()));
  
          TIntermBlock *internalMainBody = funcDefNode->getBody();
  
          TIntermFunctionDefinition *externalMainDef =
              new TIntermFunctionDefinition(externalMainProto, externalMainBody);
          TIntermFunctionDefinition *internalMainDef =
              new TIntermFunctionDefinition(internalMainProto, internalMainBody);
  
          mMultiReplacements.push_back(NodeReplaceWithMultipleEntry(
              &root, funcDefNode, TIntermSequence{internalMainDef, externalMainDef}));
      }
  };
  
  }  // namespace
  
  bool sh::WrapMain(TCompiler &compiler, IdGen &idGen, TIntermBlock &root)
  {
      TSymbolTable &symbolTable = compiler.getSymbolTable();
      Wrapper wrapper(symbolTable, idGen);
      root.traverse(&wrapper);
      if (!wrapper.updateTree(&compiler, &root))
      {
          return false;
      }
      return true;
  }