src/third_party/skia/src/effects/SkMergeImageFilter.cpp

   1 /*
   2  * Copyright 2012 The Android Open Source Project
   3  *
   4  * Use of this source code is governed by a BSD-style license that can be
   5  * found in the LICENSE file.
   6  */
   7
   8 #include "SkMergeImageFilter.h"
   9 #include "SkCanvas.h"
  10 #include "SkDevice.h"
  11 #include "SkReadBuffer.h"
  12 #include "SkWriteBuffer.h"
  13 #include "SkValidationUtils.h"
  14
  15 ///////////////////////////////////////////////////////////////////////////////
  16
  17 void SkMergeImageFilter::initAllocModes() {
  18     int inputCount = countInputs();
  19     if (inputCount) {
  20         size_t size = sizeof(uint8_t) * inputCount;
  21         if (size <= sizeof(fStorage)) {
  22             fModes = SkTCast<uint8_t*>(fStorage);
  23         } else {
  24             fModes = SkTCast<uint8_t*>(sk_malloc_throw(size));
  25         }
  26     } else {
  27         fModes = NULL;
  28     }
  29 }
  30
  31 void SkMergeImageFilter::initModes(const SkXfermode::Mode modes[]) {
  32     if (modes) {
  33         this->initAllocModes();
  34         int inputCount = countInputs();
  35         for (int i = 0; i < inputCount; ++i) {
  36             fModes[i] = SkToU8(modes[i]);
  37         }
  38     } else {
  39         fModes = NULL;
  40     }
  41 }
  42
  43 SkMergeImageFilter::SkMergeImageFilter(SkImageFilter* first, SkImageFilter* second,
  44                                        SkXfermode::Mode mode,
  45                                        const CropRect* cropRect) : INHERITED(first, second, cropRect) {
  46     if (SkXfermode::kSrcOver_Mode != mode) {
  47         SkXfermode::Mode modes[] = { mode, mode };
  48         this->initModes(modes);
  49     } else {
  50         fModes = NULL;
  51     }
  52 }
  53
  54 SkMergeImageFilter::SkMergeImageFilter(SkImageFilter* filters[], int count,
  55                                        const SkXfermode::Mode modes[],
  56                                        const CropRect* cropRect) : INHERITED(count, filters, cropRect) {
  57     SkASSERT(count >= 0);
  58     this->initModes(modes);
  59 }
  60
  61 SkMergeImageFilter::~SkMergeImageFilter() {
  62
  63     if (fModes != SkTCast<uint8_t*>(fStorage)) {
  64         sk_free(fModes);
  65     }
  66 }
  67
  68 bool SkMergeImageFilter::onFilterImage(Proxy* proxy, const SkBitmap& src,
  69                                        const SkMatrix& ctm,
  70                                        SkBitmap* result, SkIPoint* offset) const {
  71     if (countInputs() < 1) {
  72         return false;
  73     }
  74
  75     SkIRect bounds;
  76     src.getBounds(&bounds);
  77     if (!this->applyCropRect(&bounds, ctm)) {
  78         return false;
  79     }
  80
  81     const int x0 = bounds.left();
  82     const int y0 = bounds.top();
  83
  84     SkAutoTUnref<SkBaseDevice> dst(proxy->createDevice(bounds.width(), bounds.height()));
  85     if (NULL == dst) {
  86         return false;
  87     }
  88     SkCanvas canvas(dst);
  89     SkPaint paint;
  90
  91     int inputCount = countInputs();
  92     for (int i = 0; i < inputCount; ++i) {
  93         SkBitmap tmp;
  94         const SkBitmap* srcPtr;
  95         SkIPoint pos = SkIPoint::Make(0, 0);
  96         SkImageFilter* filter = getInput(i);
  97         if (filter) {
  98             if (!filter->filterImage(proxy, src, ctm, &tmp, &pos)) {
  99                 return false;
 100             }
 101             srcPtr = &tmp;
 102         } else {
 103             srcPtr = &src;
 104         }
 105
 106         if (fModes) {
 107             paint.setXfermodeMode((SkXfermode::Mode)fModes[i]);
 108         } else {
 109             paint.setXfermode(NULL);
 110         }
 111         canvas.drawSprite(*srcPtr, pos.x() - x0, pos.y() - y0, &paint);
 112     }
 113
 114     offset->fX = bounds.left();
 115     offset->fY = bounds.top();
 116     *result = dst->accessBitmap(false);
 117     return true;
 118 }
 119
 120 void SkMergeImageFilter::flatten(SkWriteBuffer& buffer) const {
 121     this->INHERITED::flatten(buffer);
 122
 123     buffer.writeBool(fModes != NULL);
 124     if (fModes) {
 125         buffer.writeByteArray(fModes, countInputs() * sizeof(fModes[0]));
 126     }
 127 }
 128
 129 SkMergeImageFilter::SkMergeImageFilter(SkReadBuffer& buffer)
 130   : INHERITED(-1, buffer) {
 131     bool hasModes = buffer.readBool();
 132     if (hasModes) {
 133         this->initAllocModes();
 134         int nbInputs = countInputs();
 135         size_t size = nbInputs * sizeof(fModes[0]);
 136         SkASSERT(buffer.getArrayCount() == size);
 137         if (buffer.validate(buffer.getArrayCount() == size) &&
 138             buffer.readByteArray(fModes, size)) {
 139             for (int i = 0; i < nbInputs; ++i) {
 140                 buffer.validate(SkIsValidMode((SkXfermode::Mode)fModes[i]));
 141             }
 142         }
 143     } else {
 144         fModes = 0;
 145     }
 146 }