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

Branch : - branch - main - tag -

• Show log

  Commit

• Author : Stuart Morgan
  Date : 2019-08-14 12:25:12
  Hash : 9d737966
  Message : Standardize copyright notices to project style For all "ANGLE Project" copyrights, standardize to the format specified by the style guide. Changes: - "Copyright (c)" and "Copyright(c)" changed to just "Copyright". - Removed the second half of date ranges ("Y1Y1-Y2Y2"->"Y1Y1"). - Fixed a small number of files that had no copyright date using the initial commit year from the version control history. - Fixed one instance of copyright being "The ANGLE Project" rather than "The ANGLE Project Authors" These changes are applied both to the copyright of source file, and where applicable to copyright statements that are generated by templates. BUG=angleproject:3811 Change-Id: I973dd65e4ef9deeba232d5be74c768256a0eb2e5 Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/1754397 Commit-Queue: Jamie Madill <jmadill@chromium.org> Reviewed-by: Jamie Madill <jmadill@chromium.org>

• src/compiler/translator/tree_ops/RewriteDoWhile.cpp

• //
  // Copyright 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/tree_ops/RewriteDoWhile.h"
  
  #include "compiler/translator/StaticType.h"
  #include "compiler/translator/tree_util/IntermNode_util.h"
  #include "compiler/translator/tree_util/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.
              const TType *boolType = StaticType::Get<EbtBool, EbpUndefined, EvqTemporary, 1, 1>();
              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(), nullptr);
  
                  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