blob: 09d9f5de4ae1762a01229e19532034c777bb61ab [file] [log] [blame]
senorblanco@chromium.org05054f12012-03-02 21:05:45 +00001/*
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 "SkMorphologyImageFilter.h"
djsollen@google.com64a0ec32012-06-12 15:17:27 +00009#include "SkBitmap.h"
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000010#include "SkColorPriv.h"
djsollen@google.comc73dd5c2012-08-07 15:54:32 +000011#include "SkFlattenableBuffers.h"
12#include "SkRect.h"
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +000013#include "SkMorphology_opts.h"
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +000014#if SK_SUPPORT_GPU
senorblanco@chromium.org302cffb2012-08-01 20:16:34 +000015#include "GrContext.h"
16#include "GrTexture.h"
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +000017#include "GrTBackendEffectFactory.h"
bsalomon@google.comd698f772012-10-25 13:22:00 +000018#include "gl/GrGLEffect.h"
senorblanco@chromium.org84207c42012-08-22 20:51:19 +000019#include "effects/Gr1DKernelEffect.h"
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +000020#include "SkImageFilterUtils.h"
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +000021#endif
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000022
23SkMorphologyImageFilter::SkMorphologyImageFilter(SkFlattenableReadBuffer& buffer)
commit-bot@chromium.orgce33d602013-11-25 21:46:31 +000024 : INHERITED(1, buffer) {
tomhudson@google.com75589252012-04-10 17:42:21 +000025 fRadius.fWidth = buffer.readInt();
26 fRadius.fHeight = buffer.readInt();
commit-bot@chromium.orgc0b7e102013-10-23 17:06:21 +000027 buffer.validate((fRadius.fWidth >= 0) &&
28 (fRadius.fHeight >= 0));
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000029}
30
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +000031SkMorphologyImageFilter::SkMorphologyImageFilter(int radiusX,
32 int radiusY,
33 SkImageFilter* input,
34 const CropRect* cropRect)
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +000035 : INHERITED(input, cropRect), fRadius(SkISize::Make(radiusX, radiusY)) {
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000036}
37
38
djsollen@google.com54924242012-03-29 15:18:04 +000039void SkMorphologyImageFilter::flatten(SkFlattenableWriteBuffer& buffer) const {
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000040 this->INHERITED::flatten(buffer);
tomhudson@google.com75589252012-04-10 17:42:21 +000041 buffer.writeInt(fRadius.fWidth);
42 buffer.writeInt(fRadius.fHeight);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000043}
44
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +000045enum MorphDirection {
46 kX, kY
47};
48
49template<MorphDirection direction>
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000050static void erode(const SkPMColor* src, SkPMColor* dst,
51 int radius, int width, int height,
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +000052 int srcStride, int dstStride)
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000053{
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +000054 const int srcStrideX = direction == kX ? 1 : srcStride;
55 const int dstStrideX = direction == kX ? 1 : dstStride;
56 const int srcStrideY = direction == kX ? srcStride : 1;
57 const int dstStrideY = direction == kX ? dstStride : 1;
senorblanco@chromium.org56dd6302012-04-10 17:25:44 +000058 radius = SkMin32(radius, width - 1);
59 const SkPMColor* upperSrc = src + radius * srcStrideX;
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000060 for (int x = 0; x < width; ++x) {
61 const SkPMColor* lp = src;
62 const SkPMColor* up = upperSrc;
63 SkPMColor* dptr = dst;
64 for (int y = 0; y < height; ++y) {
65 int minB = 255, minG = 255, minR = 255, minA = 255;
66 for (const SkPMColor* p = lp; p <= up; p += srcStrideX) {
67 int b = SkGetPackedB32(*p);
68 int g = SkGetPackedG32(*p);
69 int r = SkGetPackedR32(*p);
70 int a = SkGetPackedA32(*p);
71 if (b < minB) minB = b;
72 if (g < minG) minG = g;
73 if (r < minR) minR = r;
74 if (a < minA) minA = a;
75 }
76 *dptr = SkPackARGB32(minA, minR, minG, minB);
77 dptr += dstStrideY;
78 lp += srcStrideY;
79 up += srcStrideY;
80 }
81 if (x >= radius) src += srcStrideX;
82 if (x + radius < width - 1) upperSrc += srcStrideX;
83 dst += dstStrideX;
84 }
85}
86
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +000087static void erodeX(const SkBitmap& src, SkBitmap* dst, int radiusX, const SkIRect& bounds)
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000088{
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +000089 SkMorphologyProc erodeXProc = SkMorphologyGetPlatformProc(kErodeX_SkMorphologyProcType);
90 if (!erodeXProc) {
91 erodeXProc = erode<kX>;
92 }
93 erodeXProc(src.getAddr32(bounds.left(), bounds.top()), dst->getAddr32(0, 0),
94 radiusX, bounds.width(), bounds.height(),
95 src.rowBytesAsPixels(), dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000096}
97
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +000098static void erodeY(const SkBitmap& src, SkBitmap* dst, int radiusY, const SkIRect& bounds)
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000099{
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +0000100 SkMorphologyProc erodeYProc = SkMorphologyGetPlatformProc(kErodeY_SkMorphologyProcType);
101 if (!erodeYProc) {
102 erodeYProc = erode<kY>;
103 }
104 erodeYProc(src.getAddr32(bounds.left(), bounds.top()), dst->getAddr32(0, 0),
105 radiusY, bounds.height(), bounds.width(),
106 src.rowBytesAsPixels(), dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000107}
108
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +0000109template<MorphDirection direction>
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000110static void dilate(const SkPMColor* src, SkPMColor* dst,
111 int radius, int width, int height,
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +0000112 int srcStride, int dstStride)
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000113{
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +0000114 const int srcStrideX = direction == kX ? 1 : srcStride;
115 const int dstStrideX = direction == kX ? 1 : dstStride;
116 const int srcStrideY = direction == kX ? srcStride : 1;
117 const int dstStrideY = direction == kX ? dstStride : 1;
senorblanco@chromium.org56dd6302012-04-10 17:25:44 +0000118 radius = SkMin32(radius, width - 1);
119 const SkPMColor* upperSrc = src + radius * srcStrideX;
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000120 for (int x = 0; x < width; ++x) {
121 const SkPMColor* lp = src;
122 const SkPMColor* up = upperSrc;
123 SkPMColor* dptr = dst;
124 for (int y = 0; y < height; ++y) {
125 int maxB = 0, maxG = 0, maxR = 0, maxA = 0;
126 for (const SkPMColor* p = lp; p <= up; p += srcStrideX) {
127 int b = SkGetPackedB32(*p);
128 int g = SkGetPackedG32(*p);
129 int r = SkGetPackedR32(*p);
130 int a = SkGetPackedA32(*p);
131 if (b > maxB) maxB = b;
132 if (g > maxG) maxG = g;
133 if (r > maxR) maxR = r;
134 if (a > maxA) maxA = a;
135 }
136 *dptr = SkPackARGB32(maxA, maxR, maxG, maxB);
137 dptr += dstStrideY;
138 lp += srcStrideY;
139 up += srcStrideY;
140 }
141 if (x >= radius) src += srcStrideX;
142 if (x + radius < width - 1) upperSrc += srcStrideX;
143 dst += dstStrideX;
144 }
145}
146
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000147static void dilateX(const SkBitmap& src, SkBitmap* dst, int radiusX, const SkIRect& bounds)
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000148{
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +0000149 SkMorphologyProc dilateXProc = SkMorphologyGetPlatformProc(kDilateX_SkMorphologyProcType);
150 if (!dilateXProc) {
151 dilateXProc = dilate<kX>;
152 }
153 dilateXProc(src.getAddr32(bounds.left(), bounds.top()), dst->getAddr32(0, 0),
154 radiusX, bounds.width(), bounds.height(),
155 src.rowBytesAsPixels(), dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000156}
157
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000158static void dilateY(const SkBitmap& src, SkBitmap* dst, int radiusY, const SkIRect& bounds)
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000159{
senorblanco@chromium.org7a47ad32013-10-30 21:57:04 +0000160 SkMorphologyProc dilateYProc = SkMorphologyGetPlatformProc(kDilateY_SkMorphologyProcType);
161 if (!dilateYProc) {
162 dilateYProc = dilate<kY>;
163 }
164 dilateYProc(src.getAddr32(bounds.left(), bounds.top()), dst->getAddr32(0, 0),
165 radiusY, bounds.height(), bounds.width(),
166 src.rowBytesAsPixels(), dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000167}
168
senorblanco@chromium.orgf1369ce2012-08-20 14:53:21 +0000169bool SkErodeImageFilter::onFilterImage(Proxy* proxy,
170 const SkBitmap& source, const SkMatrix& ctm,
171 SkBitmap* dst, SkIPoint* offset) {
senorblanco@chromium.org68400762013-05-24 15:04:07 +0000172 SkBitmap src = source;
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000173 SkIPoint srcOffset = SkIPoint::Make(0, 0);
174 if (getInput(0) && !getInput(0)->filterImage(proxy, source, ctm, &src, &srcOffset)) {
senorblanco@chromium.org68400762013-05-24 15:04:07 +0000175 return false;
176 }
177
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000178 if (src.config() != SkBitmap::kARGB_8888_Config) {
179 return false;
180 }
181
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000182 SkIRect bounds;
183 src.getBounds(&bounds);
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000184 bounds.offset(srcOffset);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000185 if (!this->applyCropRect(&bounds, ctm)) {
186 return false;
187 }
188
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000189 SkAutoLockPixels alp(src);
190 if (!src.getPixels()) {
191 return false;
192 }
193
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000194 dst->setConfig(src.config(), bounds.width(), bounds.height());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000195 dst->allocPixels();
commit-bot@chromium.orgcd3b15c2013-12-04 17:06:49 +0000196 if (!dst->getPixels()) {
197 return false;
198 }
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000199
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000200 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
201 SkIntToScalar(this->radius().height()));
202 ctm.mapVectors(&radius, 1);
203 int width = SkScalarFloorToInt(radius.fX);
204 int height = SkScalarFloorToInt(radius.fY);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000205
206 if (width < 0 || height < 0) {
207 return false;
208 }
209
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000210 SkIRect srcBounds = bounds;
211 srcBounds.offset(-srcOffset);
212
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000213 if (width == 0 && height == 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000214 src.extractSubset(dst, srcBounds);
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000215 offset->fX = bounds.left();
216 offset->fY = bounds.top();
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000217 return true;
218 }
219
220 SkBitmap temp;
221 temp.setConfig(dst->config(), dst->width(), dst->height());
222 if (!temp.allocPixels()) {
223 return false;
224 }
225
226 if (width > 0 && height > 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000227 erodeX(src, &temp, width, srcBounds);
228 SkIRect tmpBounds = SkIRect::MakeWH(srcBounds.width(), srcBounds.height());
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000229 erodeY(temp, dst, height, tmpBounds);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000230 } else if (width > 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000231 erodeX(src, dst, width, srcBounds);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000232 } else if (height > 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000233 erodeY(src, dst, height, srcBounds);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000234 }
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000235 offset->fX = bounds.left();
236 offset->fY = bounds.top();
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000237 return true;
238}
239
senorblanco@chromium.orgf1369ce2012-08-20 14:53:21 +0000240bool SkDilateImageFilter::onFilterImage(Proxy* proxy,
241 const SkBitmap& source, const SkMatrix& ctm,
242 SkBitmap* dst, SkIPoint* offset) {
senorblanco@chromium.org68400762013-05-24 15:04:07 +0000243 SkBitmap src = source;
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000244 SkIPoint srcOffset = SkIPoint::Make(0, 0);
245 if (getInput(0) && !getInput(0)->filterImage(proxy, source, ctm, &src, &srcOffset)) {
senorblanco@chromium.org68400762013-05-24 15:04:07 +0000246 return false;
247 }
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000248 if (src.config() != SkBitmap::kARGB_8888_Config) {
249 return false;
250 }
251
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000252 SkIRect bounds;
253 src.getBounds(&bounds);
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000254 bounds.offset(srcOffset);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000255 if (!this->applyCropRect(&bounds, ctm)) {
256 return false;
257 }
258
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000259 SkAutoLockPixels alp(src);
260 if (!src.getPixels()) {
261 return false;
262 }
263
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000264 dst->setConfig(src.config(), bounds.width(), bounds.height());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000265 dst->allocPixels();
commit-bot@chromium.orgcd3b15c2013-12-04 17:06:49 +0000266 if (!dst->getPixels()) {
267 return false;
268 }
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000269
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000270 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
271 SkIntToScalar(this->radius().height()));
272 ctm.mapVectors(&radius, 1);
273 int width = SkScalarFloorToInt(radius.fX);
274 int height = SkScalarFloorToInt(radius.fY);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000275
276 if (width < 0 || height < 0) {
277 return false;
278 }
279
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000280 SkIRect srcBounds = bounds;
281 srcBounds.offset(-srcOffset);
282
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000283 if (width == 0 && height == 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000284 src.extractSubset(dst, srcBounds);
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000285 offset->fX = bounds.left();
286 offset->fY = bounds.top();
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000287 return true;
288 }
289
290 SkBitmap temp;
291 temp.setConfig(dst->config(), dst->width(), dst->height());
292 if (!temp.allocPixels()) {
293 return false;
294 }
295
296 if (width > 0 && height > 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000297 dilateX(src, &temp, width, srcBounds);
298 SkIRect tmpBounds = SkIRect::MakeWH(srcBounds.width(), srcBounds.height());
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000299 dilateY(temp, dst, height, tmpBounds);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000300 } else if (width > 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000301 dilateX(src, dst, width, srcBounds);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000302 } else if (height > 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000303 dilateY(src, dst, height, srcBounds);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000304 }
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000305 offset->fX = bounds.left();
306 offset->fY = bounds.top();
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000307 return true;
308}
309
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000310#if SK_SUPPORT_GPU
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000311
312///////////////////////////////////////////////////////////////////////////////
313
314class GrGLMorphologyEffect;
315
316/**
317 * Morphology effects. Depending upon the type of morphology, either the
318 * component-wise min (Erode_Type) or max (Dilate_Type) of all pixels in the
319 * kernel is selected as the new color. The new color is modulated by the input
320 * color.
321 */
322class GrMorphologyEffect : public Gr1DKernelEffect {
323
324public:
325
326 enum MorphologyType {
327 kErode_MorphologyType,
328 kDilate_MorphologyType,
329 };
330
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000331 static GrEffectRef* Create(GrTexture* tex, Direction dir, int radius, MorphologyType type) {
bsalomon@google.com6340a412013-01-22 19:55:59 +0000332 AutoEffectUnref effect(SkNEW_ARGS(GrMorphologyEffect, (tex, dir, radius, type)));
bsalomon@google.coma1ebbe42013-01-16 15:51:47 +0000333 return CreateEffectRef(effect);
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000334 }
335
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000336 virtual ~GrMorphologyEffect();
337
338 MorphologyType type() const { return fType; }
339
340 static const char* Name() { return "Morphology"; }
341
bsalomon@google.com422e81a2012-10-25 14:11:03 +0000342 typedef GrGLMorphologyEffect GLEffect;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000343
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000344 virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE;
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000345 virtual void getConstantColorComponents(GrColor* color, uint32_t* validFlags) const SK_OVERRIDE;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000346
347protected:
348
349 MorphologyType fType;
350
351private:
bsalomon@google.com8a252f72013-01-22 20:35:13 +0000352 virtual bool onIsEqual(const GrEffect&) const SK_OVERRIDE;
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000353
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000354 GrMorphologyEffect(GrTexture*, Direction, int radius, MorphologyType);
355
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000356 GR_DECLARE_EFFECT_TEST;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000357
358 typedef Gr1DKernelEffect INHERITED;
359};
360
361///////////////////////////////////////////////////////////////////////////////
362
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000363class GrGLMorphologyEffect : public GrGLEffect {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000364public:
bsalomon@google.comc7818882013-03-20 19:19:53 +0000365 GrGLMorphologyEffect (const GrBackendEffectFactory&, const GrDrawEffect&);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000366
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000367 virtual void emitCode(GrGLShaderBuilder*,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000368 const GrDrawEffect&,
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000369 EffectKey,
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000370 const char* outputColor,
371 const char* inputColor,
bsalomon@google.com77af6802013-10-02 13:04:56 +0000372 const TransformedCoordsArray&,
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000373 const TextureSamplerArray&) SK_OVERRIDE;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000374
bsalomon@google.comc7818882013-03-20 19:19:53 +0000375 static inline EffectKey GenKey(const GrDrawEffect&, const GrGLCaps&);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000376
bsalomon@google.comc7818882013-03-20 19:19:53 +0000377 virtual void setData(const GrGLUniformManager&, const GrDrawEffect&) SK_OVERRIDE;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000378
379private:
380 int width() const { return GrMorphologyEffect::WidthFromRadius(fRadius); }
381
382 int fRadius;
383 GrMorphologyEffect::MorphologyType fType;
384 GrGLUniformManager::UniformHandle fImageIncrementUni;
385
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000386 typedef GrGLEffect INHERITED;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000387};
388
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000389GrGLMorphologyEffect::GrGLMorphologyEffect(const GrBackendEffectFactory& factory,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000390 const GrDrawEffect& drawEffect)
bsalomon@google.com77af6802013-10-02 13:04:56 +0000391 : INHERITED(factory) {
bsalomon@google.comc7818882013-03-20 19:19:53 +0000392 const GrMorphologyEffect& m = drawEffect.castEffect<GrMorphologyEffect>();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000393 fRadius = m.radius();
394 fType = m.type();
395}
396
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000397void GrGLMorphologyEffect::emitCode(GrGLShaderBuilder* builder,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000398 const GrDrawEffect&,
bsalomon@google.comb4a55b72012-11-02 20:45:37 +0000399 EffectKey key,
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000400 const char* outputColor,
401 const char* inputColor,
bsalomon@google.com77af6802013-10-02 13:04:56 +0000402 const TransformedCoordsArray& coords,
bsalomon@google.com47d7a882012-10-29 12:47:51 +0000403 const TextureSamplerArray& samplers) {
bsalomon@google.com77af6802013-10-02 13:04:56 +0000404 SkString coords2D = builder->ensureFSCoords2D(coords, 0);
commit-bot@chromium.org74a3a212013-08-30 19:43:59 +0000405 fImageIncrementUni = builder->addUniform(GrGLShaderBuilder::kFragment_Visibility,
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000406 kVec2f_GrSLType, "ImageIncrement");
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000407
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000408 const char* func;
409 switch (fType) {
410 case GrMorphologyEffect::kErode_MorphologyType:
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000411 builder->fsCodeAppendf("\t\t%s = vec4(1, 1, 1, 1);\n", outputColor);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000412 func = "min";
413 break;
414 case GrMorphologyEffect::kDilate_MorphologyType:
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000415 builder->fsCodeAppendf("\t\t%s = vec4(0, 0, 0, 0);\n", outputColor);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000416 func = "max";
417 break;
418 default:
419 GrCrash("Unexpected type");
420 func = ""; // suppress warning
421 break;
422 }
423 const char* imgInc = builder->getUniformCStr(fImageIncrementUni);
424
bsalomon@google.com77af6802013-10-02 13:04:56 +0000425 builder->fsCodeAppendf("\t\tvec2 coord = %s - %d.0 * %s;\n", coords2D.c_str(), fRadius, imgInc);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000426 builder->fsCodeAppendf("\t\tfor (int i = 0; i < %d; i++) {\n", this->width());
427 builder->fsCodeAppendf("\t\t\t%s = %s(%s, ", outputColor, func, outputColor);
commit-bot@chromium.org74a3a212013-08-30 19:43:59 +0000428 builder->fsAppendTextureLookup(samplers[0], "coord");
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000429 builder->fsCodeAppend(");\n");
430 builder->fsCodeAppendf("\t\t\tcoord += %s;\n", imgInc);
431 builder->fsCodeAppend("\t\t}\n");
432 SkString modulate;
433 GrGLSLMulVarBy4f(&modulate, 2, outputColor, inputColor);
434 builder->fsCodeAppend(modulate.c_str());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000435}
436
bsalomon@google.comc7818882013-03-20 19:19:53 +0000437GrGLEffect::EffectKey GrGLMorphologyEffect::GenKey(const GrDrawEffect& drawEffect,
438 const GrGLCaps&) {
439 const GrMorphologyEffect& m = drawEffect.castEffect<GrMorphologyEffect>();
bsalomon@google.com46fba0d2012-10-25 21:42:05 +0000440 EffectKey key = static_cast<EffectKey>(m.radius());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000441 key |= (m.type() << 8);
bsalomon@google.com77af6802013-10-02 13:04:56 +0000442 return key;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000443}
444
bsalomon@google.comc7818882013-03-20 19:19:53 +0000445void GrGLMorphologyEffect::setData(const GrGLUniformManager& uman,
446 const GrDrawEffect& drawEffect) {
447 const Gr1DKernelEffect& kern = drawEffect.castEffect<Gr1DKernelEffect>();
bsalomon@google.com2d0bade2012-10-26 19:01:17 +0000448 GrTexture& texture = *kern.texture(0);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000449 // the code we generated was for a specific kernel radius
commit-bot@chromium.org96ae6882013-08-14 12:09:00 +0000450 SkASSERT(kern.radius() == fRadius);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000451 float imageIncrement[2] = { 0 };
452 switch (kern.direction()) {
453 case Gr1DKernelEffect::kX_Direction:
454 imageIncrement[0] = 1.0f / texture.width();
455 break;
456 case Gr1DKernelEffect::kY_Direction:
457 imageIncrement[1] = 1.0f / texture.height();
458 break;
459 default:
460 GrCrash("Unknown filter direction.");
461 }
commit-bot@chromium.orgd3baf202013-11-07 22:06:08 +0000462 uman.set2fv(fImageIncrementUni, 1, imageIncrement);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000463}
464
465///////////////////////////////////////////////////////////////////////////////
466
467GrMorphologyEffect::GrMorphologyEffect(GrTexture* texture,
468 Direction direction,
469 int radius,
470 MorphologyType type)
471 : Gr1DKernelEffect(texture, direction, radius)
472 , fType(type) {
473}
474
475GrMorphologyEffect::~GrMorphologyEffect() {
476}
477
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000478const GrBackendEffectFactory& GrMorphologyEffect::getFactory() const {
479 return GrTBackendEffectFactory<GrMorphologyEffect>::getInstance();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000480}
481
bsalomon@google.com8a252f72013-01-22 20:35:13 +0000482bool GrMorphologyEffect::onIsEqual(const GrEffect& sBase) const {
bsalomon@google.com6340a412013-01-22 19:55:59 +0000483 const GrMorphologyEffect& s = CastEffect<GrMorphologyEffect>(sBase);
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000484 return (this->texture(0) == s.texture(0) &&
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000485 this->radius() == s.radius() &&
486 this->direction() == s.direction() &&
487 this->type() == s.type());
488}
489
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000490void GrMorphologyEffect::getConstantColorComponents(GrColor* color, uint32_t* validFlags) const {
491 // This is valid because the color components of the result of the kernel all come
492 // exactly from existing values in the source texture.
493 this->updateConstantColorComponentsForModulation(color, validFlags);
494}
495
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000496///////////////////////////////////////////////////////////////////////////////
497
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000498GR_DEFINE_EFFECT_TEST(GrMorphologyEffect);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000499
commit-bot@chromium.orge0e7cfe2013-09-09 20:09:12 +0000500GrEffectRef* GrMorphologyEffect::TestCreate(SkRandom* random,
sugoi@google.come0e385c2013-03-11 18:50:03 +0000501 GrContext*,
bsalomon@google.comc26d94f2013-03-25 18:19:00 +0000502 const GrDrawTargetCaps&,
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000503 GrTexture* textures[]) {
bsalomon@google.com6f261be2012-10-24 19:07:10 +0000504 int texIdx = random->nextBool() ? GrEffectUnitTest::kSkiaPMTextureIdx :
505 GrEffectUnitTest::kAlphaTextureIdx;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000506 Direction dir = random->nextBool() ? kX_Direction : kY_Direction;
507 static const int kMaxRadius = 10;
508 int radius = random->nextRangeU(1, kMaxRadius);
509 MorphologyType type = random->nextBool() ? GrMorphologyEffect::kErode_MorphologyType :
510 GrMorphologyEffect::kDilate_MorphologyType;
511
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000512 return GrMorphologyEffect::Create(textures[texIdx], dir, radius, type);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000513}
514
515namespace {
516
517void apply_morphology_pass(GrContext* context,
518 GrTexture* texture,
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000519 const SkIRect& srcRect,
520 const SkIRect& dstRect,
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000521 int radius,
522 GrMorphologyEffect::MorphologyType morphType,
523 Gr1DKernelEffect::Direction direction) {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000524 GrPaint paint;
commit-bot@chromium.org42dacab2013-07-13 17:24:24 +0000525 paint.addColorEffect(GrMorphologyEffect::Create(texture,
526 direction,
527 radius,
528 morphType))->unref();
reed@google.com44699382013-10-31 17:28:30 +0000529 context->drawRectToRect(paint, SkRect::Make(dstRect), SkRect::Make(srcRect));
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000530}
531
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000532bool apply_morphology(const SkBitmap& input,
533 const SkIRect& rect,
534 GrMorphologyEffect::MorphologyType morphType,
535 SkISize radius,
536 SkBitmap* dst) {
537 GrTexture* srcTexture = input.getTexture();
538 SkASSERT(NULL != srcTexture);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000539 GrContext* context = srcTexture->getContext();
540 srcTexture->ref();
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000541 SkAutoTUnref<GrTexture> src(srcTexture);
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000542
543 GrContext::AutoMatrix am;
544 am.setIdentity(context);
545
commit-bot@chromium.orgfd03d4a2013-07-17 21:39:42 +0000546 GrContext::AutoClip acs(context, SkRect::MakeWH(SkIntToScalar(srcTexture->width()),
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000547 SkIntToScalar(srcTexture->height())));
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000548
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000549 SkIRect dstRect = SkIRect::MakeWH(rect.width(), rect.height());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000550 GrTextureDesc desc;
551 desc.fFlags = kRenderTarget_GrTextureFlagBit | kNoStencil_GrTextureFlagBit;
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000552 desc.fWidth = rect.width();
553 desc.fHeight = rect.height();
554 desc.fConfig = kSkia8888_GrPixelConfig;
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000555 SkIRect srcRect = rect;
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000556
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000557 if (radius.fWidth > 0) {
558 GrAutoScratchTexture ast(context, desc);
559 GrContext::AutoRenderTarget art(context, ast.texture()->asRenderTarget());
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000560 apply_morphology_pass(context, src, srcRect, dstRect, radius.fWidth,
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000561 morphType, Gr1DKernelEffect::kX_Direction);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000562 SkIRect clearRect = SkIRect::MakeXYWH(dstRect.fLeft, dstRect.fBottom,
563 dstRect.width(), radius.fHeight);
skia.committer@gmail.comb77f0f42013-10-30 07:01:56 +0000564 context->clear(&clearRect, GrMorphologyEffect::kErode_MorphologyType == morphType ?
565 SK_ColorWHITE :
robertphillips@google.com56ce48a2013-10-31 21:44:25 +0000566 SK_ColorTRANSPARENT, false);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000567 src.reset(ast.detach());
568 srcRect = dstRect;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000569 }
570 if (radius.fHeight > 0) {
571 GrAutoScratchTexture ast(context, desc);
572 GrContext::AutoRenderTarget art(context, ast.texture()->asRenderTarget());
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000573 apply_morphology_pass(context, src, srcRect, dstRect, radius.fHeight,
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000574 morphType, Gr1DKernelEffect::kY_Direction);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000575 src.reset(ast.detach());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000576 }
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000577 return SkImageFilterUtils::WrapTexture(src, rect.width(), rect.height(), dst);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000578}
579
580};
581
commit-bot@chromium.org1aa54bf2013-08-05 16:53:50 +0000582bool SkDilateImageFilter::filterImageGPU(Proxy* proxy, const SkBitmap& src, const SkMatrix& ctm,
583 SkBitmap* result, SkIPoint* offset) {
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000584 SkBitmap input;
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000585 SkIPoint srcOffset = SkIPoint::Make(0, 0);
586 if (!SkImageFilterUtils::GetInputResultGPU(getInput(0), proxy, src, ctm, &input, &srcOffset)) {
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000587 return false;
588 }
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000589 SkIRect bounds;
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000590 input.getBounds(&bounds);
591 bounds.offset(srcOffset);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000592 if (!this->applyCropRect(&bounds, ctm)) {
593 return false;
594 }
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000595 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
596 SkIntToScalar(this->radius().height()));
597 ctm.mapVectors(&radius, 1);
598 int width = SkScalarFloorToInt(radius.fX);
599 int height = SkScalarFloorToInt(radius.fY);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000600
601 if (width < 0 || height < 0) {
602 return false;
603 }
604
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000605 SkIRect srcBounds = bounds;
606 srcBounds.offset(-srcOffset);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000607 if (width == 0 && height == 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000608 input.extractSubset(result, srcBounds);
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000609 offset->fX = bounds.left();
610 offset->fY = bounds.top();
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000611 return true;
612 }
613
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000614 if (!apply_morphology(input, srcBounds, GrMorphologyEffect::kDilate_MorphologyType,
615 SkISize::Make(width, height), result)) {
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000616 return false;
617 }
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000618 offset->fX = bounds.left();
619 offset->fY = bounds.top();
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000620 return true;
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000621}
622
commit-bot@chromium.org1aa54bf2013-08-05 16:53:50 +0000623bool SkErodeImageFilter::filterImageGPU(Proxy* proxy, const SkBitmap& src, const SkMatrix& ctm,
624 SkBitmap* result, SkIPoint* offset) {
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000625 SkBitmap input;
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000626 SkIPoint srcOffset = SkIPoint::Make(0, 0);
627 if (!SkImageFilterUtils::GetInputResultGPU(getInput(0), proxy, src, ctm, &input, &srcOffset)) {
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000628 return false;
629 }
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000630 SkIRect bounds;
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000631 input.getBounds(&bounds);
632 bounds.offset(srcOffset);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000633 if (!this->applyCropRect(&bounds, ctm)) {
634 return false;
635 }
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000636 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
637 SkIntToScalar(this->radius().height()));
638 ctm.mapVectors(&radius, 1);
639 int width = SkScalarFloorToInt(radius.fX);
640 int height = SkScalarFloorToInt(radius.fY);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000641
642 if (width < 0 || height < 0) {
643 return false;
644 }
645
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000646 SkIRect srcBounds = bounds;
647 srcBounds.offset(-srcOffset);
648
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000649 if (width == 0 && height == 0) {
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000650 input.extractSubset(result, srcBounds);
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000651 offset->fX = bounds.left();
652 offset->fY = bounds.top();
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000653 return true;
654 }
655
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000656 if (!apply_morphology(input, srcBounds, GrMorphologyEffect::kErode_MorphologyType,
657 SkISize::Make(width, height), result)) {
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000658 return false;
659 }
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000660 offset->fX = bounds.left();
661 offset->fY = bounds.top();
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000662 return true;
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000663}
664
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000665#endif