Make filters use SkImage instead of SkBitmap

src/effects/SkDisplacementMapEffect.cpp

 /*
  * Copyright 2013 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "SkDisplacementMapEffect.h"
+#include "SkColorPriv.h"
+#include "SkImagePriv.h"
+#include "SkImage_Base.h"
 #include "SkReadBuffer.h"
+#include "SkUnPreMultiply.h"
 #include "SkWriteBuffer.h"
-#include "SkUnPreMultiply.h"
-#include "SkColorPriv.h"
 #if SK_SUPPORT_GPU
 #include "GrContext.h"
 #include "GrCoordTransform.h"
 #include "GrInvariantOutput.h"
 #include "gl/GrGLProcessor.h"
 #include "gl/builders/GrGLProgramBuilder.h"
 #endif
 
 namespace {
 
@@ skipping 177 matching lines @@
 }
 
 void SkDisplacementMapEffect::flatten(SkWriteBuffer& buffer) const {
     this->INHERITED::flatten(buffer);
     buffer.writeInt((int) fXChannelSelector);
     buffer.writeInt((int) fYChannelSelector);
     buffer.writeScalar(fScale);
 }
 
 bool SkDisplacementMapEffect::onFilterImage(Proxy* proxy,
-                                            const SkBitmap& src,
+                                            SkImage& src,
                                             const Context& ctx,
-                                            SkBitmap* dst,
+                                            SkAutoTUnref<SkImage>& dst,
                                             SkIPoint* offset) const {
-    SkBitmap displ = src, color = src;
-    const SkImageFilter* colorInput = getColorInput();
-    const SkImageFilter* displInput = getDisplacementInput();
+    SkBitmap dstBitmap;
+    SkBitmap displacementBitmap;
+    SkBitmap colorBitmap;
+    SkIRect bounds;
     SkIPoint colorOffset = SkIPoint::Make(0, 0), displOffset = SkIPoint::Make(0, 0);
-    if ((colorInput && !colorInput->filterImage(proxy, src, ctx, &color, &colorOffset)) ||
-        (displInput && !displInput->filterImage(proxy, src, ctx, &displ, &displOffset))) {
+    {
+        SkAutoTUnref<SkImage> displ(SkRef(&src));
+        SkAutoTUnref<SkImage> color(SkRef(&src));
+        const SkImageFilter* colorInput = getColorInput();
+        const SkImageFilter* displInput = getDisplacementInput();
+
+        if ((colorInput && !colorInput->filterImage(proxy, src, ctx, color, &colorOffset)) ||
+            (displInput && !displInput->filterImage(proxy, src, ctx, displ, &displOffset))) {
+            return false;
+        }
+#if 0 // TODO COLORTYPE
+        if ((displ.colorType() != kN32_SkColorType) ||
+            (color.colorType() != kN32_SkColorType)) {
+            return false;
+        }
+#endif
+        // Since computeDisplacement does bounds checking on color pixel access, we don't need to pad
+        // the color bitmap to bounds here.
+        if (!this->applyCropRect(ctx, *color, colorOffset, &bounds)) {
+            return false;
+        }
+        SkIRect displBounds;
+        if (!this->applyCropRect(ctx, proxy, *displ, &displOffset, &displBounds, displ)) {
+            return false;
+        }
+        if (!bounds.intersect(displBounds)) {
+            return false;
+        }
+
+
+        if (!as_IB(displ)->getROPixels(&displacementBitmap)) {
+            return false;
+        }
+
+        if (!as_IB(color)->getROPixels(&colorBitmap)) {
+            return false;
+        }
+    }
+    {
+        SkAutoLockPixels alp_displacement(displacementBitmap);
+        SkAutoLockPixels alp_color(colorBitmap);
+        if (!displacementBitmap.getPixels() || !colorBitmap.getPixels()) {
+            return false;
+        }
+
+        if (!dstBitmap.tryAllocPixels(colorBitmap.info().makeWH(bounds.width(), bounds.height()))) {
+            return false;
+        }
+
+        SkVector scale = SkVector::Make(fScale, fScale);
+        ctx.ctm().mapVectors(&scale, 1);
+        SkIRect colorBounds = bounds;
+        colorBounds.offset(-colorOffset);
+
+        computeDisplacement(fXChannelSelector, fYChannelSelector, scale, &dstBitmap,
+                            &displacementBitmap, colorOffset - displOffset,
+                            &colorBitmap, colorBounds);
+    }
+
+    displacementBitmap = SkBitmap();
+    colorBitmap = SkBitmap();
+
+    SkImage* image = SkNewImageFromBitmap(dstBitmap, true, NULL);
+    if (NULL == image) {
         return false;
     }
-    if ((displ.colorType() != kN32_SkColorType) ||
-        (color.colorType() != kN32_SkColorType)) {
-        return false;
-    }
-    SkIRect bounds;
-    // Since computeDisplacement does bounds checking on color pixel access, we don't need to pad
-    // the color bitmap to bounds here.
-    if (!this->applyCropRect(ctx, color, colorOffset, &bounds)) {
-        return false;
-    }
-    SkIRect displBounds;
-    if (!this->applyCropRect(ctx, proxy, displ, &displOffset, &displBounds, &displ)) {
-        return false;
-    }
-    if (!bounds.intersect(displBounds)) {
-        return false;
-    }
-    SkAutoLockPixels alp_displacement(displ), alp_color(color);
-    if (!displ.getPixels() || !color.getPixels()) {
-        return false;
-    }
-
-    if (!dst->tryAllocPixels(color.info().makeWH(bounds.width(), bounds.height()))) {
-        return false;
-    }
-
-    SkVector scale = SkVector::Make(fScale, fScale);
-    ctx.ctm().mapVectors(&scale, 1);
-    SkIRect colorBounds = bounds;
-    colorBounds.offset(-colorOffset);
-
-    computeDisplacement(fXChannelSelector, fYChannelSelector, scale, dst,
-                        &displ, colorOffset - displOffset, &color, colorBounds);
-
+    dst.reset(image);
     offset->fX = bounds.left();
     offset->fY = bounds.top();
     return true;
 }
 
 void SkDisplacementMapEffect::computeFastBounds(const SkRect& src, SkRect* dst) const {
     if (getColorInput()) {
         getColorInput()->computeFastBounds(src, dst);
     } else {
         *dst = src;
@@ skipping 110 matching lines @@
     GrTextureAccess             fDisplacementAccess;
     GrCoordTransform            fColorTransform;
     GrTextureAccess             fColorAccess;
     SkDisplacementMapEffect::ChannelSelectorType fXChannelSelector;
     SkDisplacementMapEffect::ChannelSelectorType fYChannelSelector;
     SkVector fScale;
 
     typedef GrFragmentProcessor INHERITED;
 };
 
-bool SkDisplacementMapEffect::filterImageGPU(Proxy* proxy, const SkBitmap& src, const Context& ctx,
-                                             SkBitmap* result, SkIPoint* offset) const {
-    SkBitmap colorBM = src;
+bool SkDisplacementMapEffect::filterImageGPU(Proxy* proxy, SkImage& src, const Context& ctx,
+                                             SkAutoTUnref<SkImage>& result, SkIPoint* offset) const {
+    SkAutoTUnref<SkImage> colorBM(SkRef(&src));
     SkIPoint colorOffset = SkIPoint::Make(0, 0);
-    if (getColorInput() && !getColorInput()->getInputResultGPU(proxy, src, ctx, &colorBM,
+    if (getColorInput() && !getColorInput()->getInputResultGPU(proxy, src, ctx, colorBM,
                                                                &colorOffset)) {
         return false;
     }
-    SkBitmap displacementBM = src;
+    SkAutoTUnref<SkImage> displacementBM(SkRef(&src));
     SkIPoint displacementOffset = SkIPoint::Make(0, 0);
     if (getDisplacementInput() &&
-        !getDisplacementInput()->getInputResultGPU(proxy, src, ctx, &displacementBM,
+        !getDisplacementInput()->getInputResultGPU(proxy, src, ctx, displacementBM,
                                                    &displacementOffset)) {
         return false;
     }
     SkIRect bounds;
     // Since GrDisplacementMapEffect does bounds checking on color pixel access, we don't need to
     // pad the color bitmap to bounds here.
-    if (!this->applyCropRect(ctx, colorBM, colorOffset, &bounds)) {
+    if (!this->applyCropRect(ctx, *colorBM, colorOffset, &bounds)) {
         return false;
     }
     SkIRect displBounds;
-    if (!this->applyCropRect(ctx, proxy, displacementBM,
-                             &displacementOffset, &displBounds, &displacementBM)) {
+    if (!this->applyCropRect(ctx, proxy, *displacementBM,
+                             &displacementOffset, &displBounds, displacementBM)) {
         return false;
     }
     if (!bounds.intersect(displBounds)) {
         return false;
     }
-    GrTexture* color = colorBM.getTexture();
-    GrTexture* displacement = displacementBM.getTexture();
+    GrTexture* color = colorBM->getTexture();
+    GrTexture* displacement = displacementBM->getTexture();
     GrContext* context = color->getContext();
 
     GrSurfaceDesc desc;
     desc.fFlags = kRenderTarget_GrSurfaceFlag | kNoStencil_GrSurfaceFlag;
     desc.fWidth = bounds.width();
     desc.fHeight = bounds.height();
     desc.fConfig = kSkia8888_GrPixelConfig;
 
     SkAutoTUnref<GrTexture> dst(
         context->refScratchTexture(desc, GrContext::kApprox_ScratchTexMatch));
@@ skipping 18 matching lines @@
                                         scale,
                                         displacement,
                                         offsetMatrix,
                                         color))->unref();
     SkIRect colorBounds = bounds;
     colorBounds.offset(-colorOffset);
     SkMatrix matrix;
     matrix.setTranslate(-SkIntToScalar(colorBounds.x()),
                         -SkIntToScalar(colorBounds.y()));
     context->drawRect(paint, matrix, SkRect::Make(colorBounds));
+    if (!WrapTexture(dst, bounds.width(), bounds.height(), result)) {
+        return false;
+    }
     offset->fX = bounds.left();
     offset->fY = bounds.top();
-    WrapTexture(dst, bounds.width(), bounds.height(), result);
     return true;
 }
 
 ///////////////////////////////////////////////////////////////////////////////
 
 GrDisplacementMapEffect::GrDisplacementMapEffect(
                              SkDisplacementMapEffect::ChannelSelectorType xChannelSelector,
                              SkDisplacementMapEffect::ChannelSelectorType yChannelSelector,
                              const SkVector& scale,
                              GrTexture* displacement,
@@ skipping 164 matching lines @@
                                        const GrGLCaps&, GrProcessorKeyBuilder* b) {
     const GrDisplacementMapEffect& displacementMap = proc.cast<GrDisplacementMapEffect>();
 
     uint32_t xKey = displacementMap.xChannelSelector();
     uint32_t yKey = displacementMap.yChannelSelector() << kChannelSelectorKeyBits;
 
     b->add32(xKey | yKey);
 }
 #endif