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

Branch : - branch - main - tag -

• Show log

  Commit

• Author : Zhenyao Mo
  Date : 2014-02-26 15:40:48
  Hash : 550c600b
  Message : Improvement on loop unrolling with loops indexing sampler arrays 1) Before this workaround is hardwired on mac, now we move it behind a compil 2) Fix the issue where "break" inside the loop isn't handled while unrolled. BUG=338474 TEST=webgl conformance test sampler-array-using-loop-index.html Change-Id: I4996a42c2dea39a8a5af772c256f8e3cb383f59a Reviewed-on: https://chromium-review.googlesource.com/188079 Reviewed-by: Zhenyao Mo <zmo@chromium.org> Tested-by: Zhenyao Mo <zmo@chromium.org> Conflicts: include/GLSLANG/ShaderLang.h src/compiler/translator/ValidateLimitations.cpp Change-Id: I546197bd7df1634ebccdd380be14c3250cd56151 Reviewed-on: https://chromium-review.googlesource.com/189061 Reviewed-by: Shannon Woods <shannonwoods@chromium.org> Tested-by: Zhenyao Mo <zmo@chromium.org>

• src/compiler/translator/ForLoopUnroll.cpp

• //
  // Copyright (c) 2002-2013 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/ForLoopUnroll.h"
  
  bool ForLoopUnrollMarker::visitBinary(Visit, TIntermBinary *node)
  {
      if (mUnrollCondition != kSamplerArrayIndex)
          return true;
  
      // If a sampler array index is also the loop index,
      //   1) if the index type is integer, mark the loop for unrolling;
      //   2) if the index type if float, set a flag to later fail compile.
      switch (node->getOp())
      {
        case EOpIndexIndirect:
          if (node->getLeft() != NULL && node->getRight() != NULL && node->getLeft()->getAsSymbolNode())
          {
              TIntermSymbol *symbol = node->getLeft()->getAsSymbolNode();
              if (IsSampler(symbol->getBasicType()) && symbol->isArray() && !mLoopStack.empty())
              {
                  mVisitSamplerArrayIndexNodeInsideLoop = true;
                  node->getRight()->traverse(this);
                  mVisitSamplerArrayIndexNodeInsideLoop = false;
                  // We have already visited all the children.
                  return false;
              }
          }
          break;
        default:
          break;
      }
      return true;
  }
  
  bool ForLoopUnrollMarker::visitLoop(Visit, TIntermLoop *node)
  {
      if (mUnrollCondition == kIntegerIndex)
      {
          // Check if loop index type is integer.
          // This is called after ValidateLimitations pass, so all the calls
          // should be valid. See ValidateLimitations::validateForLoopInit().
          TIntermSequence& declSeq = node->getInit()->getAsAggregate()->getSequence();
          TIntermSymbol* symbol = declSeq[0]->getAsBinaryNode()->getLeft()->getAsSymbolNode();
          if (symbol->getBasicType() == EbtInt)
              node->setUnrollFlag(true);
      }
  
      TIntermNode *body = node->getBody();
      if (body != NULL)
      {
          mLoopStack.push(node);
          body->traverse(this);
          mLoopStack.pop();
      }
      // The loop is fully processed - no need to visit children.
      return false;
  }
  
  void ForLoopUnrollMarker::visitSymbol(TIntermSymbol* symbol)
  {
      if (!mVisitSamplerArrayIndexNodeInsideLoop)
          return;
      TIntermLoop *loop = mLoopStack.findLoop(symbol);
      if (loop)
      {
          switch (symbol->getBasicType())
          {
            case EbtFloat:
              mSamplerArrayIndexIsFloatLoopIndex = true;
              break;
            case EbtInt:
              loop->setUnrollFlag(true);
              break;
            default:
              UNREACHABLE();
          }
      }
  }