kc3-lang/angle/src/compiler/translator/tree_ops/vulkan/RewriteInterpolateAtOffset.cpp

• Show log

  Commit

• Hash : 382bf288
  Author :
  Date : 2020-12-24T23:56:39
  

  Organize AST transforms per backend
  
  Most of the AST transforms are written as a workaround to an issue that
  affects a single backend.  This change identifies such transforms and
  organizes them by backend.  They are then only built if the respective
  backend is.
  
  Additionally, about half of the GL transforms are due to mac
  workarounds, including the large RewriteRowMajorMatrices transform.
  Mac-specific workarounds are additionally only built on said platform.
  
  This change reduces the ANGLE binary size:
  
  - 106KB in a Vulkan-only build on Linux
  - 27KB in a GL-only build on Android (60KB on Linux)
  
  Bug: chromium:1084580
  Bug: chromium:1161513
  Change-Id: I64b334332c0d4f848756c6538af0d8d96864c7e9
  Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/2601346
  Commit-Queue: Shahbaz Youssefi <syoussefi@chromium.org>
  Reviewed-by: Jamie Madill <jmadill@chromium.org>
  Reviewed-by: Tim Van Patten <timvp@google.com>
  

Download

• src/compiler/translator/tree_ops/vulkan/RewriteInterpolateAtOffset.cpp

• //
  // Copyright 2020 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.
  //
  // Implementation of InterpolateAtOffset viewport transformation.
  // See header for more info.
  
  #include "compiler/translator/tree_ops/vulkan/RewriteInterpolateAtOffset.h"
  
  #include "common/angleutils.h"
  #include "compiler/translator/StaticType.h"
  #include "compiler/translator/SymbolTable.h"
  #include "compiler/translator/TranslatorVulkan.h"
  #include "compiler/translator/tree_util/DriverUniform.h"
  #include "compiler/translator/tree_util/IntermNode_util.h"
  #include "compiler/translator/tree_util/IntermTraverse.h"
  #include "compiler/translator/tree_util/SpecializationConstant.h"
  
  namespace sh
  {
  
  namespace
  {
  
  class Traverser : public TIntermTraverser
  {
    public:
      ANGLE_NO_DISCARD static bool Apply(TCompiler *compiler,
                                         ShCompileOptions compileOptions,
                                         TIntermNode *root,
                                         const TSymbolTable &symbolTable,
                                         int ShaderVersion,
                                         SpecConst *specConst,
                                         const DriverUniform *driverUniforms);
  
    private:
      Traverser(TSymbolTable *symbolTable,
                ShCompileOptions compileOptions,
                int shaderVersion,
                SpecConst *specConst,
                const DriverUniform *driverUniforms);
      bool visitAggregate(Visit visit, TIntermAggregate *node) override;
  
      const TSymbolTable *symbolTable = nullptr;
      const int shaderVersion;
      SpecConst *mRotationSpecConst        = nullptr;
      const DriverUniform *mDriverUniforms = nullptr;
      bool mUsePreRotation                 = false;
  };
  
  Traverser::Traverser(TSymbolTable *symbolTable,
                       ShCompileOptions compileOptions,
                       int shaderVersion,
                       SpecConst *specConst,
                       const DriverUniform *driverUniforms)
      : TIntermTraverser(true, false, false, symbolTable),
        symbolTable(symbolTable),
        shaderVersion(shaderVersion),
        mRotationSpecConst(specConst),
        mDriverUniforms(driverUniforms),
        mUsePreRotation((compileOptions & SH_ADD_PRE_ROTATION) != 0)
  {}
  
  // static
  bool Traverser::Apply(TCompiler *compiler,
                        ShCompileOptions compileOptions,
                        TIntermNode *root,
                        const TSymbolTable &symbolTable,
                        int shaderVersion,
                        SpecConst *specConst,
                        const DriverUniform *driverUniforms)
  {
      TSymbolTable *pSymbolTable = const_cast<TSymbolTable *>(&symbolTable);
      Traverser traverser(pSymbolTable, compileOptions, shaderVersion, specConst, driverUniforms);
      root->traverse(&traverser);
      return traverser.updateTree(compiler, root);
  }
  
  bool Traverser::visitAggregate(Visit visit, TIntermAggregate *node)
  {
      // Decide if the node represents the call of texelFetchOffset.
      if (node->getOp() != EOpCallBuiltInFunction)
      {
          return true;
      }
  
      ASSERT(node->getFunction()->symbolType() == SymbolType::BuiltIn);
      if (node->getFunction()->name() != "interpolateAtOffset")
      {
          return true;
      }
  
      const TIntermSequence *sequence = node->getSequence();
      ASSERT(sequence->size() == 2u);
  
      TIntermSequence *interpolateAtOffsetArguments = new TIntermSequence();
      // interpolant node
      interpolateAtOffsetArguments->push_back(sequence->at(0));
      // offset
      TIntermTyped *offsetNode = sequence->at(1)->getAsTyped();
      ASSERT(offsetNode->getType() == *(StaticType::GetBasic<EbtFloat, 2>()));
  
      // If pre-rotation is enabled apply the transformation else just flip the Y-coordinate
      TIntermTyped *rotatedXY;
      if (mUsePreRotation)
      {
          rotatedXY = mRotationSpecConst->getFragRotationMultiplyFlipXY();
          if (!rotatedXY)
          {
              TIntermTyped *flipXY       = mDriverUniforms->getFlipXYRef();
              TIntermTyped *fragRotation = mDriverUniforms->getFragRotationMatrixRef();
              rotatedXY = new TIntermBinary(EOpMatrixTimesVector, fragRotation, flipXY);
          }
      }
      else
      {
          rotatedXY = mRotationSpecConst->getFlipXY();
          if (!rotatedXY)
          {
              rotatedXY = mDriverUniforms->getFlipXYRef();
          }
      }
  
      TIntermBinary *correctedOffset = new TIntermBinary(EOpMul, offsetNode, rotatedXY);
      correctedOffset->setLine(offsetNode->getLine());
      interpolateAtOffsetArguments->push_back(correctedOffset);
  
      TIntermTyped *interpolateAtOffsetNode = CreateBuiltInFunctionCallNode(
          "interpolateAtOffset", interpolateAtOffsetArguments, *symbolTable, shaderVersion);
      interpolateAtOffsetNode->setLine(node->getLine());
  
      // Replace the old node by this new node.
      queueReplacement(interpolateAtOffsetNode, OriginalNode::IS_DROPPED);
  
      return true;
  }
  
  }  // anonymous namespace
  
  bool RewriteInterpolateAtOffset(TCompiler *compiler,
                                  ShCompileOptions compileOptions,
                                  TIntermNode *root,
                                  const TSymbolTable &symbolTable,
                                  int shaderVersion,
                                  SpecConst *specConst,
                                  const DriverUniform *driverUniforms)
  {
      // interpolateAtOffset is only valid in GLSL 3.0 and later.
      if (shaderVersion < 300)
      {
          return true;
      }
  
      return Traverser::Apply(compiler, compileOptions, root, symbolTable, shaderVersion, specConst,
                              driverUniforms);
  }
  
  }  // namespace sh
  

Download