kc3-lang/angle/src/compiler/translator/tree_ops/glsl/RewriteRepeatedAssignToSwizzled.cpp

Branch - branch -main

• Show log

  Commit

• Hash : 9b63700b
  Author :
  Date : 2023-07-11T10:25:40
  

  Translator: Reorganize files
  
  This change makes the translator files more organized by:
  
  - Grouping files that are specific to a certain output under their own
    directory,
  - Moving transformations under tree_ops
  - Removing Direct from metal translation now that that SPIR-V Cross path
    is removed
  
  Bug: angleproject:6394
  Change-Id: Iaf5bb8d5604b84748dece029821b1f77b2813967
  Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/4678780
  Commit-Queue: Shahbaz Youssefi <syoussefi@chromium.org>
  Reviewed-by: Geoff Lang <geofflang@chromium.org>
  

Download

• src/compiler/translator/tree_ops/glsl/RewriteRepeatedAssignToSwizzled.cpp

• //
  // Copyright 2018 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.
  //
  // RewriteRepeatedAssignToSwizzled.cpp: Rewrite expressions that assign an assignment to a swizzled
  // vector, like:
  //     v.x = z = expression;
  // to:
  //     z = expression;
  //     v.x = z;
  //
  // Note that this doesn't handle some corner cases: expressions nested inside other expressions,
  // inside loop headers, or inside if conditions.
  
  #include "compiler/translator/tree_ops/glsl/RewriteRepeatedAssignToSwizzled.h"
  
  #include "compiler/translator/tree_util/IntermNode_util.h"
  #include "compiler/translator/tree_util/IntermTraverse.h"
  
  namespace sh
  {
  
  namespace
  {
  
  class RewriteAssignToSwizzledTraverser : public TIntermTraverser
  {
    public:
      [[nodiscard]] static bool rewrite(TCompiler *compiler, TIntermBlock *root);
  
    private:
      RewriteAssignToSwizzledTraverser();
  
      bool visitBinary(Visit, TIntermBinary *node) override;
  
      void nextIteration();
  
      bool didRewrite() { return mDidRewrite; }
  
      bool mDidRewrite;
  };
  
  // static
  bool RewriteAssignToSwizzledTraverser::rewrite(TCompiler *compiler, TIntermBlock *root)
  {
      RewriteAssignToSwizzledTraverser rewrite;
      do
      {
          rewrite.nextIteration();
          root->traverse(&rewrite);
          if (!rewrite.updateTree(compiler, root))
          {
              return false;
          }
      } while (rewrite.didRewrite());
  
      return true;
  }
  
  RewriteAssignToSwizzledTraverser::RewriteAssignToSwizzledTraverser()
      : TIntermTraverser(true, false, false), mDidRewrite(false)
  {}
  
  void RewriteAssignToSwizzledTraverser::nextIteration()
  {
      mDidRewrite = false;
  }
  
  bool RewriteAssignToSwizzledTraverser::visitBinary(Visit, TIntermBinary *node)
  {
      TIntermBinary *rightBinary = node->getRight()->getAsBinaryNode();
      TIntermBlock *parentBlock  = getParentNode()->getAsBlock();
      if (parentBlock && node->isAssignment() && node->getLeft()->getAsSwizzleNode() && rightBinary &&
          rightBinary->isAssignment())
      {
          TIntermSequence replacements;
          replacements.push_back(rightBinary);
          TIntermTyped *rightAssignmentTargetCopy = rightBinary->getLeft()->deepCopy();
          TIntermBinary *lastAssign =
              new TIntermBinary(node->getOp(), node->getLeft(), rightAssignmentTargetCopy);
          replacements.push_back(lastAssign);
          mMultiReplacements.emplace_back(parentBlock, node, std::move(replacements));
          mDidRewrite = true;
          return false;
      }
      return true;
  }
  
  }  // anonymous namespace
  
  bool RewriteRepeatedAssignToSwizzled(TCompiler *compiler, TIntermBlock *root)
  {
      return RewriteAssignToSwizzledTraverser::rewrite(compiler, root);
  }
  
  }  // namespace sh
  

Download