Potentially optimize some GrGLEffects for known input color values

src/core/SkXfermode.cpp

 
 /*
  * Copyright 2006 The Android Open Source Project
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 
 #include "SkXfermode.h"
@@ skipping 867 matching lines @@
 //////////////////////////////////////////////////////////////////////////////
 
 #if SK_SUPPORT_GPU
 
 #include "GrEffect.h"
 #include "GrCoordTransform.h"
 #include "GrEffectUnitTest.h"
 #include "GrTBackendEffectFactory.h"
 #include "gl/GrGLEffect.h"
 
+template GrGLSLExpr3 GrGLSL::min<GrGLSLExpr3>(const GrGLSLExpr3&, const GrGLSLExpr3&);
+template GrGLSLExpr3 GrGLSL::max<GrGLSLExpr3>(const GrGLSLExpr3&, const GrGLSLExpr3&);
+
 /**
  * GrEffect that implements the all the separable xfer modes that cannot be expressed as Coeffs.
  */
 class XferEffect : public GrEffect {
 public:
     static bool IsSupportedMode(SkXfermode::Mode mode) {
         return mode > SkXfermode::kLastCoeffMode && mode <= SkXfermode::kLastMode;
     }
 
     static GrEffectRef* Create(SkXfermode::Mode mode, GrTexture* background) {
@@ skipping 21 matching lines @@
 
     class GLEffect : public GrGLEffect {
     public:
         GLEffect(const GrBackendEffectFactory& factory, const GrDrawEffect&)
             : GrGLEffect(factory) {
         }
         virtual void emitCode(GrGLShaderBuilder* builder,
                               const GrDrawEffect& drawEffect,
                               EffectKey key,
                               const char* outputColor,
-                              const char* inputColor,
+                              const GrGLSLExpr4& inputColorExpr,
                               const TransformedCoordsArray& coords,
                               const TextureSamplerArray& samplers) SK_OVERRIDE {
             SkXfermode::Mode mode = drawEffect.castEffect<XferEffect>().mode();
             const GrTexture* backgroundTex = drawEffect.castEffect<XferEffect>().backgroundAccess().getTexture();
             const char* dstColor;
             if (backgroundTex) {
                 dstColor = "bgColor";
                 builder->fsCodeAppendf("\t\tvec4 %s = ", dstColor);
                 builder->fsAppendTextureLookup(samplers[0], coords[0].c_str(), coords[0].type());
                 builder->fsCodeAppendf(";\n");
             } else {
                 dstColor = builder->dstColor();
             }
             SkASSERT(NULL != dstColor);
 
-            // We don't try to optimize for this case at all
-            if (NULL == inputColor) {
-                builder->fsCodeAppendf("\t\tconst vec4 ones = vec4(1);\n");
-                inputColor = "ones";
-            }
             builder->fsCodeAppendf("\t\t// SkXfermode::Mode: %s\n", SkXfermode::ModeName(mode));
 
+            GrGLSLExpr1 one(1);
+            GrGLSLExpr4 dstColorExpr(dstColor);
             // These all perform src-over on the alpha channel.
-            builder->fsCodeAppendf("\t\t%s.a = %s.a + (1.0 - %s.a) * %s.a;\n",
-                                    outputColor, inputColor, inputColor, dstColor);
+            builder->fsCodeAppendf("\t\t%s.a = %s;\n",
+                                   outputColor,
+                                   (inputColorExpr.a() + (one - inputColorExpr.a()) * dstColorExpr.a()).c_str());
+
+            const char* inputColor = inputColorExpr.c_str(); // Shorthand for below.
 
             switch (mode) {
                 case SkXfermode::kOverlay_Mode:
                     // Overlay is Hard-Light with the src and dst reversed
                     HardLight(builder, outputColor, dstColor, inputColor);
                     break;
                 case SkXfermode::kDarken_Mode:
-                    builder->fsCodeAppendf("\t\t%s.rgb = min((1.0 - %s.a) * %s.rgb + %s.rgb, "
-                                                            "(1.0 - %s.a) * %s.rgb + %s.rgb);\n",
-                                            outputColor,
-                                            inputColor, dstColor, inputColor,
-                                            dstColor, inputColor, dstColor);
+                case SkXfermode::kLighten_Mode: {
+                    GrGLSLExpr3 (*minOrMax)(const GrGLSLExpr3&, const GrGLSLExpr3&);
+                    minOrMax = mode == SkXfermode::kDarken_Mode ? &GrGLSL::min<GrGLSLExpr3> : &GrGLSL::max<GrGLSLExpr3>;
+
+                    builder->fsCodeAppendf("\t\t%s.rgb = %s;\n", outputColor,
+                                           minOrMax((one - inputColorExpr.a()) * dstColorExpr.rgb() + inputColorExpr.rgb(),
+                                                (one - dstColorExpr.a()) *  inputColorExpr.rgb() + dstColorExpr.rgb()).c_str());
                     break;
-                case SkXfermode::kLighten_Mode:
-                    builder->fsCodeAppendf("\t\t%s.rgb = max((1.0 - %s.a) * %s.rgb + %s.rgb, "
-                                                            "(1.0 - %s.a) * %s.rgb + %s.rgb);\n",
-                                            outputColor,
-                                            inputColor, dstColor, inputColor,
-                                            dstColor, inputColor, dstColor);
-                    break;
+                }
                 case SkXfermode::kColorDodge_Mode:
                     ColorDodgeComponent(builder, outputColor, inputColor, dstColor, 'r');
                     ColorDodgeComponent(builder, outputColor, inputColor, dstColor, 'g');
                     ColorDodgeComponent(builder, outputColor, inputColor, dstColor, 'b');
                     break;
                 case SkXfermode::kColorBurn_Mode:
                     ColorBurnComponent(builder, outputColor, inputColor, dstColor, 'r');
                     ColorBurnComponent(builder, outputColor, inputColor, dstColor, 'g');
                     ColorBurnComponent(builder, outputColor, inputColor, dstColor, 'b');
                     break;
@@ skipping 959 matching lines @@
     return proc16;
 }
 
 SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_START(SkXfermode)
     SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkProcCoeffXfermode)
     SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkClearXfermode)
     SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkSrcXfermode)
     SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkDstInXfermode)
     SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkDstOutXfermode)
 SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_END