The Tor Browser: gfx/skia/trunk/src/effects/SkArithmeticMode.cpp@6474c204b198

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

     1 /*

     2  * Copyright 2013 Google Inc.

     3  *

     4  * Use of this source code is governed by a BSD-style license that can be

     5  * found in the LICENSE file.

     6  */

     8 #include "SkArithmeticMode.h"

     9 #include "SkColorPriv.h"

    10 #include "SkReadBuffer.h"

    11 #include "SkWriteBuffer.h"

    12 #include "SkString.h"

    13 #include "SkUnPreMultiply.h"

    14 #if SK_SUPPORT_GPU

    15 #include "GrContext.h"

    16 #include "GrCoordTransform.h"

    17 #include "gl/GrGLEffect.h"

    18 #include "GrTBackendEffectFactory.h"

    19 #endif

    21 static const bool gUseUnpremul = false;

    23 class SkArithmeticMode_scalar : public SkXfermode {

    24 public:

    25     static SkArithmeticMode_scalar* Create(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4) {

    26         return SkNEW_ARGS(SkArithmeticMode_scalar, (k1, k2, k3, k4));

    27     }

    29     virtual void xfer32(SkPMColor dst[], const SkPMColor src[], int count,

    30                         const SkAlpha aa[]) const SK_OVERRIDE;

    32     SK_TO_STRING_OVERRIDE()

    33     SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkArithmeticMode_scalar)

    35 #if SK_SUPPORT_GPU

    36     virtual bool asNewEffect(GrEffectRef** effect, GrTexture* background) const SK_OVERRIDE;

    37 #endif

    39 private:

    40     SkArithmeticMode_scalar(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4) {

    41         fK[0] = k1;

    42         fK[1] = k2;

    43         fK[2] = k3;

    44         fK[3] = k4;

    45     }

    47     SkArithmeticMode_scalar(SkReadBuffer& buffer) : INHERITED(buffer) {

    48         fK[0] = buffer.readScalar();

    49         fK[1] = buffer.readScalar();

    50         fK[2] = buffer.readScalar();

    51         fK[3] = buffer.readScalar();

    52     }

    54     virtual void flatten(SkWriteBuffer& buffer) const SK_OVERRIDE {

    55         INHERITED::flatten(buffer);

    56         buffer.writeScalar(fK[0]);

    57         buffer.writeScalar(fK[1]);

    58         buffer.writeScalar(fK[2]);

    59         buffer.writeScalar(fK[3]);

    60     }

    61     SkScalar fK[4];

    63     typedef SkXfermode INHERITED;

    64 };

    66 static int pinToByte(int value) {

    67     if (value < 0) {

    68         value = 0;

    69     } else if (value > 255) {

    70         value = 255;

    71     }

    72     return value;

    73 }

    75 static int arith(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4,

    76                  int src, int dst) {

    77     SkScalar result = SkScalarMul(k1, src * dst) +

    78                       SkScalarMul(k2, src) +

    79                       SkScalarMul(k3, dst) +

    80                       k4;

    81     int res = SkScalarRoundToInt(result);

    82     return pinToByte(res);

    83 }

    85 static int blend(int src, int dst, int scale) {

    86     return dst + ((src - dst) * scale >> 8);

    87 }

    89 static bool needsUnpremul(int alpha) {

    90     return 0 != alpha && 0xFF != alpha;

    91 }

    93 void SkArithmeticMode_scalar::xfer32(SkPMColor dst[], const SkPMColor src[],

    94                                  int count, const SkAlpha aaCoverage[]) const {

    95     SkScalar k1 = fK[0] / 255;

    96     SkScalar k2 = fK[1];

    97     SkScalar k3 = fK[2];

    98     SkScalar k4 = fK[3] * 255;

   100     for (int i = 0; i < count; ++i) {

   101         if ((NULL == aaCoverage) || aaCoverage[i]) {

   102             SkPMColor sc = src[i];

   103             SkPMColor dc = dst[i];

   105             int a, r, g, b;

   107             if (gUseUnpremul) {

   108                 int sa = SkGetPackedA32(sc);

   109                 int da = SkGetPackedA32(dc);

   111                 int srcNeedsUnpremul = needsUnpremul(sa);

   112                 int dstNeedsUnpremul = needsUnpremul(da);

   114                 if (!srcNeedsUnpremul && !dstNeedsUnpremul) {

   115                     a = arith(k1, k2, k3, k4, sa, da);

   116                     r = arith(k1, k2, k3, k4, SkGetPackedR32(sc), SkGetPackedR32(dc));

   117                     g = arith(k1, k2, k3, k4, SkGetPackedG32(sc), SkGetPackedG32(dc));

   118                     b = arith(k1, k2, k3, k4, SkGetPackedB32(sc), SkGetPackedB32(dc));

   119                 } else {

   120                     int sr = SkGetPackedR32(sc);

   121                     int sg = SkGetPackedG32(sc);

   122                     int sb = SkGetPackedB32(sc);

   123                     if (srcNeedsUnpremul) {

   124                         SkUnPreMultiply::Scale scale = SkUnPreMultiply::GetScale(sa);

   125                         sr = SkUnPreMultiply::ApplyScale(scale, sr);

   126                         sg = SkUnPreMultiply::ApplyScale(scale, sg);

   127                         sb = SkUnPreMultiply::ApplyScale(scale, sb);

   128                     }

   130                     int dr = SkGetPackedR32(dc);

   131                     int dg = SkGetPackedG32(dc);

   132                     int db = SkGetPackedB32(dc);

   133                     if (dstNeedsUnpremul) {

   134                         SkUnPreMultiply::Scale scale = SkUnPreMultiply::GetScale(da);

   135                         dr = SkUnPreMultiply::ApplyScale(scale, dr);

   136                         dg = SkUnPreMultiply::ApplyScale(scale, dg);

   137                         db = SkUnPreMultiply::ApplyScale(scale, db);

   138                     }

   140                     a = arith(k1, k2, k3, k4, sa, da);

   141                     r = arith(k1, k2, k3, k4, sr, dr);

   142                     g = arith(k1, k2, k3, k4, sg, dg);

   143                     b = arith(k1, k2, k3, k4, sb, db);

   144                 }

   145             } else {

   146                 a = arith(k1, k2, k3, k4, SkGetPackedA32(sc), SkGetPackedA32(dc));

   147                 r = arith(k1, k2, k3, k4, SkGetPackedR32(sc), SkGetPackedR32(dc));

   148                 r = SkMin32(r, a);

   149                 g = arith(k1, k2, k3, k4, SkGetPackedG32(sc), SkGetPackedG32(dc));

   150                 g = SkMin32(g, a);

   151                 b = arith(k1, k2, k3, k4, SkGetPackedB32(sc), SkGetPackedB32(dc));

   152                 b = SkMin32(b, a);

   153             }

   155             // apply antialias coverage if necessary

   156             if (aaCoverage && 0xFF != aaCoverage[i]) {

   157                 int scale = aaCoverage[i] + (aaCoverage[i] >> 7);

   158                 a = blend(a, SkGetPackedA32(sc), scale);

   159                 r = blend(r, SkGetPackedR32(sc), scale);

   160                 g = blend(g, SkGetPackedG32(sc), scale);

   161                 b = blend(b, SkGetPackedB32(sc), scale);

   162             }

   164             // turn the result back into premul

   165             if (gUseUnpremul && (0xFF != a)) {

   166                 int scale = a + (a >> 7);

   167                 r = SkAlphaMul(r, scale);

   168                 g = SkAlphaMul(g, scale);

   169                 b = SkAlphaMul(b, scale);

   170             }

   171             dst[i] = SkPackARGB32(a, r, g, b);

   172         }

   173     }

   174 }

   176 #ifndef SK_IGNORE_TO_STRING

   177 void SkArithmeticMode_scalar::toString(SkString* str) const {

   178     str->append("SkArithmeticMode_scalar: ");

   179     for (int i = 0; i < 4; ++i) {

   180         str->appendScalar(fK[i]);

   181         if (i < 3) {

   182             str->append(" ");

   183         }

   184     }

   185 }

   186 #endif

   188 ///////////////////////////////////////////////////////////////////////////////

   190 static bool fitsInBits(SkScalar x, int bits) {

   191     return SkScalarAbs(x) < (1 << (bits - 1));

   192 }

   194 #if 0 // UNUSED

   195 static int32_t toDot8(SkScalar x) {

   196     return (int32_t)(x * 256);

   197 }

   198 #endif

   200 SkXfermode* SkArithmeticMode::Create(SkScalar k1, SkScalar k2,

   201                                      SkScalar k3, SkScalar k4) {

   202     if (fitsInBits(k1, 8) && fitsInBits(k2, 16) &&

   203         fitsInBits(k2, 16) && fitsInBits(k2, 24)) {

   205 #if 0 // UNUSED

   206         int32_t i1 = toDot8(k1);

   207         int32_t i2 = toDot8(k2);

   208         int32_t i3 = toDot8(k3);

   209         int32_t i4 = toDot8(k4);

   210         if (i1) {

   211             return SkNEW_ARGS(SkArithmeticMode_quad, (i1, i2, i3, i4));

   212         }

   213         if (0 == i2) {

   214             return SkNEW_ARGS(SkArithmeticMode_dst, (i3, i4));

   215         }

   216         if (0 == i3) {

   217             return SkNEW_ARGS(SkArithmeticMode_src, (i2, i4));

   218         }

   219         return SkNEW_ARGS(SkArithmeticMode_linear, (i2, i3, i4));

   220 #endif

   221     }

   222     return SkArithmeticMode_scalar::Create(k1, k2, k3, k4);

   223 }

   226 //////////////////////////////////////////////////////////////////////////////

   228 #if SK_SUPPORT_GPU

   230 class GrGLArithmeticEffect : public GrGLEffect {

   231 public:

   232     GrGLArithmeticEffect(const GrBackendEffectFactory&, const GrDrawEffect&);

   233     virtual ~GrGLArithmeticEffect();

   235     virtual void emitCode(GrGLShaderBuilder*,

   236                           const GrDrawEffect&,

   237                           EffectKey,

   238                           const char* outputColor,

   239                           const char* inputColor,

   240                           const TransformedCoordsArray&,

   241                           const TextureSamplerArray&) SK_OVERRIDE;

   243     virtual void setData(const GrGLUniformManager&, const GrDrawEffect&) SK_OVERRIDE;

   245 private:

   246     GrGLUniformManager::UniformHandle fKUni;

   248     typedef GrGLEffect INHERITED;

   249 };

   251 ///////////////////////////////////////////////////////////////////////////////

   253 class GrArithmeticEffect : public GrEffect {

   254 public:

   255     static GrEffectRef* Create(float k1, float k2, float k3, float k4, GrTexture* background) {

   256         AutoEffectUnref effect(SkNEW_ARGS(GrArithmeticEffect, (k1, k2, k3, k4, background)));

   257         return CreateEffectRef(effect);

   258     }

   260     virtual ~GrArithmeticEffect();

   262     virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE;

   264     typedef GrGLArithmeticEffect GLEffect;

   265     static const char* Name() { return "Arithmetic"; }

   266     GrTexture* backgroundTexture() const { return fBackgroundAccess.getTexture(); }

   268     virtual void getConstantColorComponents(GrColor* color, uint32_t* validFlags) const SK_OVERRIDE;

   270     float k1() const { return fK1; }

   271     float k2() const { return fK2; }

   272     float k3() const { return fK3; }

   273     float k4() const { return fK4; }

   275 private:

   276     virtual bool onIsEqual(const GrEffect&) const SK_OVERRIDE;

   278     GrArithmeticEffect(float k1, float k2, float k3, float k4, GrTexture* background);

   279     float                       fK1, fK2, fK3, fK4;

   280     GrCoordTransform            fBackgroundTransform;

   281     GrTextureAccess             fBackgroundAccess;

   283     GR_DECLARE_EFFECT_TEST;

   284     typedef GrEffect INHERITED;

   286 };

   288 ///////////////////////////////////////////////////////////////////////////////

   290 GrArithmeticEffect::GrArithmeticEffect(float k1, float k2, float k3, float k4,

   291                                        GrTexture* background)

   292   : fK1(k1), fK2(k2), fK3(k3), fK4(k4) {

   293     if (background) {

   294         fBackgroundTransform.reset(kLocal_GrCoordSet, background);

   295         this->addCoordTransform(&fBackgroundTransform);

   296         fBackgroundAccess.reset(background);

   297         this->addTextureAccess(&fBackgroundAccess);

   298     } else {

   299         this->setWillReadDstColor();

   300     }

   301 }

   303 GrArithmeticEffect::~GrArithmeticEffect() {

   304 }

   306 bool GrArithmeticEffect::onIsEqual(const GrEffect& sBase) const {

   307     const GrArithmeticEffect& s = CastEffect<GrArithmeticEffect>(sBase);

   308     return fK1 == s.fK1 &&

   309            fK2 == s.fK2 &&

   310            fK3 == s.fK3 &&

   311            fK4 == s.fK4 &&

   312            backgroundTexture() == s.backgroundTexture();

   313 }

   315 const GrBackendEffectFactory& GrArithmeticEffect::getFactory() const {

   316     return GrTBackendEffectFactory<GrArithmeticEffect>::getInstance();

   317 }

   319 void GrArithmeticEffect::getConstantColorComponents(GrColor* color, uint32_t* validFlags) const {

   320     // TODO: optimize this

   321     *validFlags = 0;

   322 }

   324 ///////////////////////////////////////////////////////////////////////////////

   326 GrGLArithmeticEffect::GrGLArithmeticEffect(const GrBackendEffectFactory& factory,

   327                                            const GrDrawEffect& drawEffect)

   328    : INHERITED(factory) {

   329 }

   331 GrGLArithmeticEffect::~GrGLArithmeticEffect() {

   332 }

   334 void GrGLArithmeticEffect::emitCode(GrGLShaderBuilder* builder,

   335                                     const GrDrawEffect& drawEffect,

   336                                     EffectKey key,

   337                                     const char* outputColor,

   338                                     const char* inputColor,

   339                                     const TransformedCoordsArray& coords,

   340                                     const TextureSamplerArray& samplers) {

   342     GrTexture* backgroundTex = drawEffect.castEffect<GrArithmeticEffect>().backgroundTexture();

   343     const char* dstColor;

   344     if (backgroundTex) {

   345         builder->fsCodeAppend("\t\tvec4 bgColor = ");

   346         builder->fsAppendTextureLookup(samplers[0], coords[0].c_str(), coords[0].type());

   347         builder->fsCodeAppendf(";\n");

   348         dstColor = "bgColor";

   349     } else {

   350         dstColor = builder->dstColor();

   351     }

   353     SkASSERT(NULL != dstColor);

   354     fKUni = builder->addUniform(GrGLShaderBuilder::kFragment_Visibility,

   355                                 kVec4f_GrSLType, "k");

   356     const char* kUni = builder->getUniformCStr(fKUni);

   358     // We don't try to optimize for this case at all

   359     if (NULL == inputColor) {

   360         builder->fsCodeAppendf("\t\tconst vec4 src = vec4(1);\n");

   361     } else {

   362         builder->fsCodeAppendf("\t\tvec4 src = %s;\n", inputColor);

   363         if (gUseUnpremul) {

   364             builder->fsCodeAppendf("\t\tsrc.rgb = clamp(src.rgb / src.a, 0.0, 1.0);\n");

   365         }

   366     }

   368     builder->fsCodeAppendf("\t\tvec4 dst = %s;\n", dstColor);

   369     if (gUseUnpremul) {

   370         builder->fsCodeAppendf("\t\tdst.rgb = clamp(dst.rgb / dst.a, 0.0, 1.0);\n");

   371     }

   373     builder->fsCodeAppendf("\t\t%s = %s.x * src * dst + %s.y * src + %s.z * dst + %s.w;\n", outputColor, kUni, kUni, kUni, kUni);

   374     builder->fsCodeAppendf("\t\t%s = clamp(%s, 0.0, 1.0);\n", outputColor, outputColor);

   375     if (gUseUnpremul) {

   376         builder->fsCodeAppendf("\t\t%s.rgb *= %s.a;\n", outputColor, outputColor);

   377     } else {

   378         builder->fsCodeAppendf("\t\t%s.rgb = min(%s.rgb, %s.a);\n", outputColor, outputColor, outputColor);

   379     }

   380 }

   382 void GrGLArithmeticEffect::setData(const GrGLUniformManager& uman, const GrDrawEffect& drawEffect) {

   383     const GrArithmeticEffect& arith = drawEffect.castEffect<GrArithmeticEffect>();

   384     uman.set4f(fKUni, arith.k1(), arith.k2(), arith.k3(), arith.k4());

   385 }

   387 GrEffectRef* GrArithmeticEffect::TestCreate(SkRandom* rand,

   388                                             GrContext*,

   389                                             const GrDrawTargetCaps&,

   390                                             GrTexture*[]) {

   391     float k1 = rand->nextF();

   392     float k2 = rand->nextF();

   393     float k3 = rand->nextF();

   394     float k4 = rand->nextF();

   396     AutoEffectUnref gEffect(SkNEW_ARGS(GrArithmeticEffect, (k1, k2, k3, k4, NULL)));

   397     return CreateEffectRef(gEffect);

   398 }

   400 GR_DEFINE_EFFECT_TEST(GrArithmeticEffect);

   402 bool SkArithmeticMode_scalar::asNewEffect(GrEffectRef** effect, GrTexture* background) const {

   403     if (effect) {

   404         *effect = GrArithmeticEffect::Create(SkScalarToFloat(fK[0]),

   405                                              SkScalarToFloat(fK[1]),

   406                                              SkScalarToFloat(fK[2]),

   407                                              SkScalarToFloat(fK[3]),

   408                                              background);

   409     }

   410     return true;

   411 }

   413 #endif

   415 SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_START(SkArithmeticMode)

   416     SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkArithmeticMode_scalar)

   417 SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_END

The Tor Browser / file revision

gfx/skia/trunk/src/effects/SkArithmeticMode.cpp@6474c204b198

gfx/skia/trunk/src/effects/SkArithmeticMode.cpp