src/third_party/skia/src/effects/SkRectShaderImageFilter.cpp

   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  */
   7
   8 #include "SkRectShaderImageFilter.h"
   9 #include "SkBitmap.h"
  10 #include "SkCanvas.h"
  11 #include "SkDevice.h"
  12 #include "SkReadBuffer.h"
  13 #include "SkWriteBuffer.h"
  14 #include "SkShader.h"
  15
  16 SkRectShaderImageFilter* SkRectShaderImageFilter::Create(SkShader* s, const SkRect& rect) {
  17     SkASSERT(s);
  18     uint32_t flags = CropRect::kHasAll_CropEdge;
  19     if (rect.width() == 0 || rect.height() == 0) {
  20         flags = 0x0;
  21     }
  22     CropRect cropRect(rect, flags);
  23     return SkNEW_ARGS(SkRectShaderImageFilter, (s, &cropRect));
  24 }
  25
  26 SkRectShaderImageFilter* SkRectShaderImageFilter::Create(SkShader* s, const CropRect* cropRect) {
  27     SkASSERT(s);
  28     return SkNEW_ARGS(SkRectShaderImageFilter, (s, cropRect));
  29 }
  30
  31 SkRectShaderImageFilter::SkRectShaderImageFilter(SkShader* s, const CropRect* cropRect)
  32   : INHERITED(NULL, cropRect)
  33   , fShader(s) {
  34     SkASSERT(s);
  35     s->ref();
  36 }
  37
  38 SkRectShaderImageFilter::SkRectShaderImageFilter(SkReadBuffer& buffer)
  39   : INHERITED(1, buffer) {
  40     fShader = buffer.readShader();
  41 }
  42
  43 void SkRectShaderImageFilter::flatten(SkWriteBuffer& buffer) const {
  44     this->INHERITED::flatten(buffer);
  45
  46     buffer.writeFlattenable(fShader);
  47 }
  48
  49 SkRectShaderImageFilter::~SkRectShaderImageFilter() {
  50     SkSafeUnref(fShader);
  51 }
  52
  53 bool SkRectShaderImageFilter::onFilterImage(Proxy* proxy,
  54                                             const SkBitmap& source,
  55                                             const SkMatrix& ctm,
  56                                             SkBitmap* result,
  57                                             SkIPoint* offset) const {
  58     SkIRect bounds;
  59     source.getBounds(&bounds);
  60     if (!this->applyCropRect(&bounds, ctm)) {
  61         return false;
  62     }
  63
  64     SkAutoTUnref<SkBaseDevice> device(proxy->createDevice(bounds.width(),
  65                                                           bounds.height()));
  66     if (NULL == device.get()) {
  67         return false;
  68     }
  69     SkCanvas canvas(device.get());
  70     SkPaint paint;
  71     paint.setShader(fShader);
  72     SkMatrix matrix;
  73     matrix.setTranslate(-SkIntToScalar(bounds.fLeft), -SkIntToScalar(bounds.fTop));
  74     fShader->setLocalMatrix(matrix);
  75     SkRect rect = SkRect::MakeWH(SkIntToScalar(bounds.width()), SkIntToScalar(bounds.height()));
  76     canvas.drawRect(rect, paint);
  77     *result = device.get()->accessBitmap(false);
  78     offset->fX = bounds.fLeft;
  79     offset->fY = bounds.fTop;
  80     return true;
  81 }