kc3-lang/angle/src/compiler/translator/RunAtTheEndOfShader.cpp

• Show log

  Commit

• Hash : 029e8ca7
  Author :
  Date : 2018-02-16T14:06:49
  

  Add a constexpr constructor for TFunction
  
  Access to TFunction parameters is now handled through two new members:
  a pointer to a parameter array and a parameter count.
  
  There's still also a vector pointer in TFunction for adding function
  parameters one by one. This is used when parsing user-defined
  functions.
  
  TEST=angle_unittests
  BUG=angleproject:2267
  
  Change-Id: I86987ae56b7cf37f010d0651e9861789050aec2b
  Reviewed-on: https://chromium-review.googlesource.com/923987
  Reviewed-by: Corentin Wallez <cwallez@chromium.org>
  Reviewed-by: Jamie Madill <jmadill@chromium.org>
  Commit-Queue: Olli Etuaho <oetuaho@nvidia.com>
  

Download

• src/compiler/translator/RunAtTheEndOfShader.cpp

• //
  // Copyright (c) 2017 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.
  //
  // RunAtTheEndOfShader.cpp: Add code to be run at the end of the shader. In case main() contains a
  // return statement, this is done by replacing the main() function with another function that calls
  // the old main, like this:
  //
  // void main() { body }
  // =>
  // void main0() { body }
  // void main()
  // {
  //     main0();
  //     codeToRun
  // }
  //
  // This way the code will get run even if the return statement inside main is executed.
  //
  
  #include "compiler/translator/RunAtTheEndOfShader.h"
  
  #include "compiler/translator/FindMain.h"
  #include "compiler/translator/IntermNode.h"
  #include "compiler/translator/IntermNode_util.h"
  #include "compiler/translator/IntermTraverse.h"
  #include "compiler/translator/StaticType.h"
  #include "compiler/translator/SymbolTable.h"
  
  namespace sh
  {
  
  namespace
  {
  
  constexpr const ImmutableString kMainString("main");
  
  class ContainsReturnTraverser : public TIntermTraverser
  {
    public:
      ContainsReturnTraverser() : TIntermTraverser(true, false, false), mContainsReturn(false) {}
  
      bool visitBranch(Visit visit, TIntermBranch *node) override
      {
          if (node->getFlowOp() == EOpReturn)
          {
              mContainsReturn = true;
          }
          return false;
      }
  
      bool containsReturn() { return mContainsReturn; }
  
    private:
      bool mContainsReturn;
  };
  
  bool ContainsReturn(TIntermNode *node)
  {
      ContainsReturnTraverser traverser;
      node->traverse(&traverser);
      return traverser.containsReturn();
  }
  
  void WrapMainAndAppend(TIntermBlock *root,
                         TIntermFunctionDefinition *main,
                         TIntermNode *codeToRun,
                         TSymbolTable *symbolTable)
  {
      // Replace main() with main0() with the same body.
      TFunction *oldMain = new TFunction(symbolTable, ImmutableString(""), SymbolType::AngleInternal,
                                         StaticType::GetBasic<EbtVoid>(), false);
      TIntermFunctionDefinition *oldMainDefinition =
          CreateInternalFunctionDefinitionNode(*oldMain, main->getBody());
  
      bool replaced = root->replaceChildNode(main, oldMainDefinition);
      ASSERT(replaced);
  
      // void main()
      TFunction *newMain = new TFunction(symbolTable, kMainString, SymbolType::UserDefined,
                                         StaticType::GetBasic<EbtVoid>(), false);
      TIntermFunctionPrototype *newMainProto = new TIntermFunctionPrototype(newMain);
  
      // {
      //     main0();
      //     codeToRun
      // }
      TIntermBlock *newMainBody     = new TIntermBlock();
      TIntermAggregate *oldMainCall =
          TIntermAggregate::CreateFunctionCall(*oldMain, new TIntermSequence());
      newMainBody->appendStatement(oldMainCall);
      newMainBody->appendStatement(codeToRun);
  
      // Add the new main() to the root node.
      TIntermFunctionDefinition *newMainDefinition =
          new TIntermFunctionDefinition(newMainProto, newMainBody);
      root->appendStatement(newMainDefinition);
  }
  
  }  // anonymous namespace
  
  void RunAtTheEndOfShader(TIntermBlock *root, TIntermNode *codeToRun, TSymbolTable *symbolTable)
  {
      TIntermFunctionDefinition *main = FindMain(root);
      if (!ContainsReturn(main))
      {
          main->getBody()->appendStatement(codeToRun);
          return;
      }
  
      WrapMainAndAppend(root, main, codeToRun, symbolTable);
  }
  
  }  // namespace sh
  

Download