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

• Show log

  Commit

• Hash : 195be942
  Author :
  Date : 2017-12-04T23:40:14
  

  Always create TVariables for TIntermSymbol nodes
  
  TIntermSymbol nodes are now constructed based on a specific TVariable.
  This makes sure that all TIntermSymbol nodes that are created to refer
  to a specific temporary in an AST transform will have consistent data.
  The TVariable objects are not necessarily added to the symbol table
  levels - just those variables that can be referred to by their name
  during parsing need to be reachable through there.
  
  In the future this can be taken a step further so that TIntermSymbol
  nodes just to point to a TVariable instead of duplicating the
  information.
  
  BUG=angleproject:2267
  TEST=angle_unittests
  
  Change-Id: I4e7bcdb0637cd3b588d3c202ef02f4b7bd7954a1
  Reviewed-on: https://chromium-review.googlesource.com/811925
  Commit-Queue: Olli Etuaho <oetuaho@nvidia.com>
  Reviewed-by: Corentin Wallez <cwallez@chromium.org>
  

Download

• src/compiler/translator/RewriteDoWhile.cpp

• //
  // Copyright (c) 2015 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.
  //
  
  // RewriteDoWhile.cpp: rewrites do-while loops using another equivalent
  // construct.
  
  #include "compiler/translator/RewriteDoWhile.h"
  
  #include "compiler/translator/IntermNode_util.h"
  #include "compiler/translator/IntermTraverse.h"
  
  namespace sh
  {
  
  namespace
  {
  
  // An AST traverser that rewrites loops of the form
  //   do {
  //     CODE;
  //   } while (CONDITION)
  //
  // to loops of the form
  //   bool temp = false;
  //   while (true) {
  //     if (temp) {
  //       if (!CONDITION) {
  //         break;
  //       }
  //     }
  //     temp = true;
  //     CODE;
  //   }
  //
  // The reason we don't use a simpler form, with for example just (temp && !CONDITION) in the
  // while condition, is that short-circuit is often badly supported by driver shader compiler.
  // The double if has the same effect, but forces shader compilers to behave.
  //
  // TODO(cwallez) when UnfoldShortCircuitIntoIf handles loops correctly, revisit this as we might
  // be able to use while (temp || CONDITION) with temp initially set to true then run
  // UnfoldShortCircuitIntoIf
  class DoWhileRewriter : public TIntermTraverser
  {
    public:
      DoWhileRewriter(TSymbolTable *symbolTable) : TIntermTraverser(true, false, false, symbolTable)
      {
      }
  
      bool visitBlock(Visit, TIntermBlock *node) override
      {
          // A well-formed AST can only have do-while inside TIntermBlock. By doing a prefix traversal
          // we are able to replace the do-while in the sequence directly as the content of the
          // do-while will be traversed later.
  
          TIntermSequence *statements = node->getSequence();
  
          // The statements vector will have new statements inserted when we encounter a do-while,
          // which prevents us from using a range-based for loop. Using the usual i++ works, as
          // the (two) new statements inserted replace the statement at the current position.
          for (size_t i = 0; i < statements->size(); i++)
          {
              TIntermNode *statement = (*statements)[i];
              TIntermLoop *loop      = statement->getAsLoopNode();
  
              if (loop == nullptr || loop->getType() != ELoopDoWhile)
              {
                  continue;
              }
  
              // Found a loop to change.
              TType boolType(EbtBool);
              TVariable *conditionVariable = CreateTempVariable(mSymbolTable, boolType);
  
              // bool temp = false;
              TIntermDeclaration *tempDeclaration =
                  CreateTempInitDeclarationNode(conditionVariable, CreateBoolNode(false));
  
              // temp = true;
              TIntermBinary *assignTrue =
                  CreateTempAssignmentNode(conditionVariable, CreateBoolNode(true));
  
              // if (temp) {
              //   if (!CONDITION) {
              //     break;
              //   }
              // }
              TIntermIfElse *breakIf = nullptr;
              {
                  TIntermBranch *breakStatement = new TIntermBranch(EOpBreak, nullptr);
  
                  TIntermBlock *breakBlock = new TIntermBlock();
                  breakBlock->getSequence()->push_back(breakStatement);
  
                  TIntermUnary *negatedCondition =
                      new TIntermUnary(EOpLogicalNot, loop->getCondition());
  
                  TIntermIfElse *innerIf = new TIntermIfElse(negatedCondition, breakBlock, nullptr);
  
                  TIntermBlock *innerIfBlock = new TIntermBlock();
                  innerIfBlock->getSequence()->push_back(innerIf);
  
                  breakIf = new TIntermIfElse(CreateTempSymbolNode(conditionVariable), innerIfBlock,
                                              nullptr);
              }
  
              // Assemble the replacement loops, reusing the do-while loop's body and inserting our
              // statements at the front.
              TIntermLoop *newLoop = nullptr;
              {
                  TIntermBlock *body = loop->getBody();
                  if (body == nullptr)
                  {
                      body = new TIntermBlock();
                  }
                  auto sequence = body->getSequence();
                  sequence->insert(sequence->begin(), assignTrue);
                  sequence->insert(sequence->begin(), breakIf);
  
                  newLoop = new TIntermLoop(ELoopWhile, nullptr, CreateBoolNode(true), nullptr, body);
              }
  
              TIntermSequence replacement;
              replacement.push_back(tempDeclaration);
              replacement.push_back(newLoop);
  
              node->replaceChildNodeWithMultiple(loop, replacement);
          }
          return true;
      }
  };
  
  }  // anonymous namespace
  
  void RewriteDoWhile(TIntermNode *root, TSymbolTable *symbolTable)
  {
      DoWhileRewriter rewriter(symbolTable);
  
      root->traverse(&rewriter);
  }
  
  }  // namespace sh
  

Download