senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 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 "SkImageFilter.h" |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 9 | |
| 10 | #include "SkBitmap.h" |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame^] | 11 | #include "SkReadBuffer.h" |
| 12 | #include "SkWriteBuffer.h" |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 13 | #include "SkRect.h" |
commit-bot@chromium.org | c0b7e10 | 2013-10-23 17:06:21 +0000 | [diff] [blame] | 14 | #include "SkValidationUtils.h" |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 15 | #if SK_SUPPORT_GPU |
| 16 | #include "GrContext.h" |
| 17 | #include "GrTexture.h" |
| 18 | #include "SkImageFilterUtils.h" |
| 19 | #endif |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 20 | |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 21 | SkImageFilter::SkImageFilter(int inputCount, SkImageFilter** inputs, const CropRect* cropRect) |
senorblanco@chromium.org | 194d775 | 2013-07-24 22:19:24 +0000 | [diff] [blame] | 22 | : fInputCount(inputCount), |
| 23 | fInputs(new SkImageFilter*[inputCount]), |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 24 | fCropRect(cropRect ? *cropRect : CropRect(SkRect(), 0x0)) { |
senorblanco@chromium.org | 8d21f6c | 2012-10-12 19:14:06 +0000 | [diff] [blame] | 25 | for (int i = 0; i < inputCount; ++i) { |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 26 | fInputs[i] = inputs[i]; |
| 27 | SkSafeRef(fInputs[i]); |
| 28 | } |
| 29 | } |
| 30 | |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 31 | SkImageFilter::SkImageFilter(SkImageFilter* input, const CropRect* cropRect) |
senorblanco@chromium.org | 194d775 | 2013-07-24 22:19:24 +0000 | [diff] [blame] | 32 | : fInputCount(1), |
| 33 | fInputs(new SkImageFilter*[1]), |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 34 | fCropRect(cropRect ? *cropRect : CropRect(SkRect(), 0x0)) { |
senorblanco@chromium.org | c2e8cef | 2012-10-22 15:07:14 +0000 | [diff] [blame] | 35 | fInputs[0] = input; |
| 36 | SkSafeRef(fInputs[0]); |
| 37 | } |
| 38 | |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 39 | SkImageFilter::SkImageFilter(SkImageFilter* input1, SkImageFilter* input2, const CropRect* cropRect) |
senorblanco@chromium.org | 194d775 | 2013-07-24 22:19:24 +0000 | [diff] [blame] | 40 | : fInputCount(2), fInputs(new SkImageFilter*[2]), |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 41 | fCropRect(cropRect ? *cropRect : CropRect(SkRect(), 0x0)) { |
senorblanco@chromium.org | c2e8cef | 2012-10-22 15:07:14 +0000 | [diff] [blame] | 42 | fInputs[0] = input1; |
| 43 | fInputs[1] = input2; |
| 44 | SkSafeRef(fInputs[0]); |
| 45 | SkSafeRef(fInputs[1]); |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 46 | } |
| 47 | |
| 48 | SkImageFilter::~SkImageFilter() { |
senorblanco@chromium.org | 8d21f6c | 2012-10-12 19:14:06 +0000 | [diff] [blame] | 49 | for (int i = 0; i < fInputCount; i++) { |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 50 | SkSafeUnref(fInputs[i]); |
| 51 | } |
| 52 | delete[] fInputs; |
| 53 | } |
| 54 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame^] | 55 | SkImageFilter::SkImageFilter(int inputCount, SkReadBuffer& buffer) { |
commit-bot@chromium.org | ce33d60 | 2013-11-25 21:46:31 +0000 | [diff] [blame] | 56 | fInputCount = buffer.readInt(); |
commit-bot@chromium.org | c84728d | 2013-12-04 20:07:47 +0000 | [diff] [blame] | 57 | if (buffer.validate((fInputCount >= 0) && ((inputCount < 0) || (fInputCount == inputCount)))) { |
commit-bot@chromium.org | ce33d60 | 2013-11-25 21:46:31 +0000 | [diff] [blame] | 58 | fInputs = new SkImageFilter*[fInputCount]; |
| 59 | for (int i = 0; i < fInputCount; i++) { |
| 60 | if (buffer.readBool()) { |
| 61 | fInputs[i] = buffer.readImageFilter(); |
| 62 | } else { |
| 63 | fInputs[i] = NULL; |
| 64 | } |
commit-bot@chromium.org | c2e9db3 | 2013-12-06 20:14:46 +0000 | [diff] [blame] | 65 | if (!buffer.isValid()) { |
| 66 | fInputCount = i; // Do not use fInputs past that point in the destructor |
| 67 | break; |
| 68 | } |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 69 | } |
commit-bot@chromium.org | ce33d60 | 2013-11-25 21:46:31 +0000 | [diff] [blame] | 70 | SkRect rect; |
| 71 | buffer.readRect(&rect); |
commit-bot@chromium.org | c2e9db3 | 2013-12-06 20:14:46 +0000 | [diff] [blame] | 72 | if (buffer.isValid() && buffer.validate(SkIsValidRect(rect))) { |
commit-bot@chromium.org | ce33d60 | 2013-11-25 21:46:31 +0000 | [diff] [blame] | 73 | uint32_t flags = buffer.readUInt(); |
| 74 | fCropRect = CropRect(rect, flags); |
| 75 | } |
| 76 | } else { |
| 77 | fInputCount = 0; |
| 78 | fInputs = NULL; |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 79 | } |
| 80 | } |
| 81 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame^] | 82 | void SkImageFilter::flatten(SkWriteBuffer& buffer) const { |
senorblanco@chromium.org | 8d21f6c | 2012-10-12 19:14:06 +0000 | [diff] [blame] | 83 | buffer.writeInt(fInputCount); |
| 84 | for (int i = 0; i < fInputCount; i++) { |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 85 | SkImageFilter* input = getInput(i); |
| 86 | buffer.writeBool(input != NULL); |
| 87 | if (input != NULL) { |
| 88 | buffer.writeFlattenable(input); |
| 89 | } |
| 90 | } |
senorblanco@chromium.org | 3f1f2a3 | 2013-10-16 18:07:48 +0000 | [diff] [blame] | 91 | buffer.writeRect(fCropRect.rect()); |
| 92 | buffer.writeUInt(fCropRect.flags()); |
senorblanco@chromium.org | 9f25de7 | 2012-10-10 20:36:13 +0000 | [diff] [blame] | 93 | } |
| 94 | |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 95 | bool SkImageFilter::filterImage(Proxy* proxy, const SkBitmap& src, |
| 96 | const SkMatrix& ctm, |
senorblanco@chromium.org | 6776b82 | 2014-01-03 21:48:22 +0000 | [diff] [blame] | 97 | SkBitmap* result, SkIPoint* offset) { |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 98 | SkASSERT(result); |
senorblanco@chromium.org | 6776b82 | 2014-01-03 21:48:22 +0000 | [diff] [blame] | 99 | SkASSERT(offset); |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 100 | /* |
| 101 | * Give the proxy first shot at the filter. If it returns false, ask |
| 102 | * the filter to do it. |
| 103 | */ |
senorblanco@chromium.org | 6776b82 | 2014-01-03 21:48:22 +0000 | [diff] [blame] | 104 | return (proxy && proxy->filterImage(this, src, ctm, result, offset)) || |
| 105 | this->onFilterImage(proxy, src, ctm, result, offset); |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 106 | } |
| 107 | |
| 108 | bool SkImageFilter::filterBounds(const SkIRect& src, const SkMatrix& ctm, |
| 109 | SkIRect* dst) { |
| 110 | SkASSERT(&src); |
| 111 | SkASSERT(dst); |
| 112 | return this->onFilterBounds(src, ctm, dst); |
| 113 | } |
| 114 | |
senorblanco@chromium.org | 336d1d7 | 2014-01-27 21:03:17 +0000 | [diff] [blame] | 115 | void SkImageFilter::computeFastBounds(const SkRect& src, SkRect* dst) const { |
| 116 | if (0 == fInputCount) { |
| 117 | *dst = src; |
| 118 | return; |
| 119 | } |
| 120 | if (this->getInput(0)) { |
| 121 | this->getInput(0)->computeFastBounds(src, dst); |
| 122 | } else { |
| 123 | *dst = src; |
| 124 | } |
| 125 | for (int i = 1; i < fInputCount; i++) { |
| 126 | SkImageFilter* input = this->getInput(i); |
| 127 | if (input) { |
| 128 | SkRect bounds; |
| 129 | input->computeFastBounds(src, &bounds); |
| 130 | dst->join(bounds); |
| 131 | } else { |
| 132 | dst->join(src); |
| 133 | } |
| 134 | } |
| 135 | } |
| 136 | |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 137 | bool SkImageFilter::onFilterImage(Proxy*, const SkBitmap&, const SkMatrix&, |
| 138 | SkBitmap*, SkIPoint*) { |
| 139 | return false; |
| 140 | } |
| 141 | |
| 142 | bool SkImageFilter::canFilterImageGPU() const { |
senorblanco@chromium.org | 7938bae | 2013-10-18 20:08:14 +0000 | [diff] [blame] | 143 | return this->asNewEffect(NULL, NULL, SkMatrix::I(), SkIRect()); |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 144 | } |
| 145 | |
commit-bot@chromium.org | 1aa54bf | 2013-08-05 16:53:50 +0000 | [diff] [blame] | 146 | bool SkImageFilter::filterImageGPU(Proxy* proxy, const SkBitmap& src, const SkMatrix& ctm, |
| 147 | SkBitmap* result, SkIPoint* offset) { |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 148 | #if SK_SUPPORT_GPU |
| 149 | SkBitmap input; |
| 150 | SkASSERT(fInputCount == 1); |
commit-bot@chromium.org | 1aa54bf | 2013-08-05 16:53:50 +0000 | [diff] [blame] | 151 | if (!SkImageFilterUtils::GetInputResultGPU(this->getInput(0), proxy, src, ctm, &input, offset)) { |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 152 | return false; |
| 153 | } |
commit-bot@chromium.org | b8d00db | 2013-06-26 19:18:23 +0000 | [diff] [blame] | 154 | GrTexture* srcTexture = input.getTexture(); |
senorblanco@chromium.org | 4e16bb2 | 2013-07-26 00:10:07 +0000 | [diff] [blame] | 155 | SkIRect bounds; |
| 156 | src.getBounds(&bounds); |
senorblanco@chromium.org | fbaea53 | 2013-08-27 21:37:01 +0000 | [diff] [blame] | 157 | if (!this->applyCropRect(&bounds, ctm)) { |
senorblanco@chromium.org | 4e16bb2 | 2013-07-26 00:10:07 +0000 | [diff] [blame] | 158 | return false; |
| 159 | } |
| 160 | SkRect srcRect = SkRect::Make(bounds); |
| 161 | SkRect dstRect = SkRect::MakeWH(srcRect.width(), srcRect.height()); |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 162 | GrContext* context = srcTexture->getContext(); |
| 163 | |
| 164 | GrTextureDesc desc; |
| 165 | desc.fFlags = kRenderTarget_GrTextureFlagBit, |
senorblanco@chromium.org | 4e16bb2 | 2013-07-26 00:10:07 +0000 | [diff] [blame] | 166 | desc.fWidth = bounds.width(); |
| 167 | desc.fHeight = bounds.height(); |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 168 | desc.fConfig = kRGBA_8888_GrPixelConfig; |
| 169 | |
| 170 | GrAutoScratchTexture dst(context, desc); |
| 171 | GrContext::AutoMatrix am; |
| 172 | am.setIdentity(context); |
| 173 | GrContext::AutoRenderTarget art(context, dst.texture()->asRenderTarget()); |
senorblanco@chromium.org | 4e16bb2 | 2013-07-26 00:10:07 +0000 | [diff] [blame] | 174 | GrContext::AutoClip acs(context, dstRect); |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 175 | GrEffectRef* effect; |
senorblanco@chromium.org | fbaea53 | 2013-08-27 21:37:01 +0000 | [diff] [blame] | 176 | SkMatrix matrix(ctm); |
| 177 | matrix.postTranslate(SkIntToScalar(-bounds.left()), SkIntToScalar(-bounds.top())); |
senorblanco@chromium.org | 7938bae | 2013-10-18 20:08:14 +0000 | [diff] [blame] | 178 | this->asNewEffect(&effect, srcTexture, matrix, bounds); |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 179 | SkASSERT(effect); |
| 180 | SkAutoUnref effectRef(effect); |
| 181 | GrPaint paint; |
commit-bot@chromium.org | 42dacab | 2013-07-13 17:24:24 +0000 | [diff] [blame] | 182 | paint.addColorEffect(effect); |
senorblanco@chromium.org | 4e16bb2 | 2013-07-26 00:10:07 +0000 | [diff] [blame] | 183 | context->drawRectToRect(paint, dstRect, srcRect); |
| 184 | |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 185 | SkAutoTUnref<GrTexture> resultTex(dst.detach()); |
senorblanco@chromium.org | 4e16bb2 | 2013-07-26 00:10:07 +0000 | [diff] [blame] | 186 | SkImageFilterUtils::WrapTexture(resultTex, bounds.width(), bounds.height(), result); |
| 187 | offset->fX += bounds.left(); |
| 188 | offset->fY += bounds.top(); |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 189 | return true; |
| 190 | #else |
senorblanco@chromium.org | c2594f4 | 2013-01-30 19:08:47 +0000 | [diff] [blame] | 191 | return false; |
senorblanco@chromium.org | d043cce | 2013-04-08 19:43:22 +0000 | [diff] [blame] | 192 | #endif |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 193 | } |
| 194 | |
senorblanco@chromium.org | fbaea53 | 2013-08-27 21:37:01 +0000 | [diff] [blame] | 195 | bool SkImageFilter::applyCropRect(SkIRect* rect, const SkMatrix& matrix) const { |
| 196 | SkRect cropRect; |
senorblanco@chromium.org | 3f1f2a3 | 2013-10-16 18:07:48 +0000 | [diff] [blame] | 197 | matrix.mapRect(&cropRect, fCropRect.rect()); |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 198 | SkIRect cropRectI; |
| 199 | cropRect.roundOut(&cropRectI); |
senorblanco@chromium.org | 3f1f2a3 | 2013-10-16 18:07:48 +0000 | [diff] [blame] | 200 | uint32_t flags = fCropRect.flags(); |
senorblanco@chromium.org | b295fb6 | 2013-10-10 13:51:19 +0000 | [diff] [blame] | 201 | // If the original crop rect edges were unset, max out the new crop edges |
senorblanco@chromium.org | 3f1f2a3 | 2013-10-16 18:07:48 +0000 | [diff] [blame] | 202 | if (!(flags & CropRect::kHasLeft_CropEdge)) cropRectI.fLeft = SK_MinS32; |
| 203 | if (!(flags & CropRect::kHasTop_CropEdge)) cropRectI.fTop = SK_MinS32; |
| 204 | if (!(flags & CropRect::kHasRight_CropEdge)) cropRectI.fRight = SK_MaxS32; |
| 205 | if (!(flags & CropRect::kHasBottom_CropEdge)) cropRectI.fBottom = SK_MaxS32; |
senorblanco@chromium.org | fbaea53 | 2013-08-27 21:37:01 +0000 | [diff] [blame] | 206 | return rect->intersect(cropRectI); |
senorblanco@chromium.org | 194d775 | 2013-07-24 22:19:24 +0000 | [diff] [blame] | 207 | } |
| 208 | |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 209 | bool SkImageFilter::onFilterBounds(const SkIRect& src, const SkMatrix& ctm, |
| 210 | SkIRect* dst) { |
| 211 | *dst = src; |
| 212 | return true; |
| 213 | } |
| 214 | |
senorblanco@chromium.org | 7938bae | 2013-10-18 20:08:14 +0000 | [diff] [blame] | 215 | bool SkImageFilter::asNewEffect(GrEffectRef**, GrTexture*, const SkMatrix&, const SkIRect&) const { |
senorblanco@chromium.org | bf2768b | 2012-08-20 15:43:14 +0000 | [diff] [blame] | 216 | return false; |
| 217 | } |
senorblanco@chromium.org | 8d21f6c | 2012-10-12 19:14:06 +0000 | [diff] [blame] | 218 | |
sugoi@google.com | a1c511b | 2013-02-21 15:02:28 +0000 | [diff] [blame] | 219 | bool SkImageFilter::asColorFilter(SkColorFilter**) const { |
| 220 | return false; |
senorblanco@chromium.org | 8d21f6c | 2012-10-12 19:14:06 +0000 | [diff] [blame] | 221 | } |