blob: 2cc224da99f2db9cc66ce1fcbbb45460471bfc5f [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"
robertphillipsf299e712016-03-25 04:49:22 -07009
djsollen@google.com64a0ec32012-06-12 15:17:27 +000010#include "SkBitmap.h"
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000011#include "SkColorPriv.h"
Mike Reed6d9f4292017-07-06 12:32:55 -040012#include "SkColorSpaceXformer.h"
mtkleind029ded2015-08-04 14:09:09 -070013#include "SkOpts.h"
commit-bot@chromium.org8b0e8ac2014-01-30 18:58:24 +000014#include "SkReadBuffer.h"
djsollen@google.comc73dd5c2012-08-07 15:54:32 +000015#include "SkRect.h"
robertphillipsf299e712016-03-25 04:49:22 -070016#include "SkSpecialImage.h"
mtkleind029ded2015-08-04 14:09:09 -070017#include "SkWriteBuffer.h"
robertphillipsf299e712016-03-25 04:49:22 -070018
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +000019#if SK_SUPPORT_GPU
senorblanco@chromium.org302cffb2012-08-01 20:16:34 +000020#include "GrContext.h"
csmartdalton02fa32c2016-08-19 13:29:27 -070021#include "GrFixedClip.h"
Robert Phillips7f6cd902016-11-10 17:03:43 -050022#include "GrRenderTargetContext.h"
Robert Phillips646e4292017-06-13 12:44:56 -040023#include "GrTexture.h"
Robert Phillips7f6cd902016-11-10 17:03:43 -050024#include "GrTextureProxy.h"
25
robertphillips1de87df2016-01-14 06:03:29 -080026#include "SkGr.h"
joshualitteb2a6762014-12-04 11:35:33 -080027#include "effects/Gr1DKernelEffect.h"
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -050028#include "effects/GrProxyMove.h"
egdaniel64c47282015-11-13 06:54:19 -080029#include "glsl/GrGLSLFragmentProcessor.h"
egdaniel2d721d32015-11-11 13:06:05 -080030#include "glsl/GrGLSLFragmentShaderBuilder.h"
egdaniel018fb622015-10-28 07:26:40 -070031#include "glsl/GrGLSLProgramDataManager.h"
egdaniel7ea439b2015-12-03 09:20:44 -080032#include "glsl/GrGLSLUniformHandler.h"
Brian Salomon94efbf52016-11-29 13:43:05 -050033#include "../private/GrGLSL.h"
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +000034#endif
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000035
robertphillips225db442016-04-17 14:27:05 -070036sk_sp<SkImageFilter> SkDilateImageFilter::Make(int radiusX, int radiusY,
37 sk_sp<SkImageFilter> input,
38 const CropRect* cropRect) {
39 if (radiusX < 0 || radiusY < 0) {
40 return nullptr;
41 }
42 return sk_sp<SkImageFilter>(new SkDilateImageFilter(radiusX, radiusY,
43 std::move(input),
44 cropRect));
45}
46
47
48sk_sp<SkImageFilter> SkErodeImageFilter::Make(int radiusX, int radiusY,
49 sk_sp<SkImageFilter> input,
50 const CropRect* cropRect) {
51 if (radiusX < 0 || radiusY < 0) {
52 return nullptr;
53 }
54 return sk_sp<SkImageFilter>(new SkErodeImageFilter(radiusX, radiusY,
55 std::move(input),
56 cropRect));
57}
58
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +000059SkMorphologyImageFilter::SkMorphologyImageFilter(int radiusX,
60 int radiusY,
robertphillipsfc11b0a2016-04-05 09:09:36 -070061 sk_sp<SkImageFilter> input,
senorblanco24e06d52015-03-18 12:11:33 -070062 const CropRect* cropRect)
robertphillipsfc11b0a2016-04-05 09:09:36 -070063 : INHERITED(&input, 1, cropRect)
robertphillipsf299e712016-03-25 04:49:22 -070064 , fRadius(SkISize::Make(radiusX, radiusY)) {
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000065}
66
commit-bot@chromium.org8b0e8ac2014-01-30 18:58:24 +000067void SkMorphologyImageFilter::flatten(SkWriteBuffer& buffer) const {
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000068 this->INHERITED::flatten(buffer);
tomhudson@google.com75589252012-04-10 17:42:21 +000069 buffer.writeInt(fRadius.fWidth);
70 buffer.writeInt(fRadius.fHeight);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000071}
72
robertphillipsf299e712016-03-25 04:49:22 -070073static void call_proc_X(SkMorphologyImageFilter::Proc procX,
robertphillips64612512016-04-08 12:10:42 -070074 const SkBitmap& src, SkBitmap* dst,
robertphillipsf299e712016-03-25 04:49:22 -070075 int radiusX, const SkIRect& bounds) {
robertphillips64612512016-04-08 12:10:42 -070076 procX(src.getAddr32(bounds.left(), bounds.top()), dst->getAddr32(0, 0),
senorblanco@chromium.org0ded88d2014-01-24 15:43:50 +000077 radiusX, bounds.width(), bounds.height(),
78 src.rowBytesAsPixels(), dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000079}
80
robertphillipsf299e712016-03-25 04:49:22 -070081static void call_proc_Y(SkMorphologyImageFilter::Proc procY,
82 const SkPMColor* src, int srcRowBytesAsPixels, SkBitmap* dst,
83 int radiusY, const SkIRect& bounds) {
84 procY(src, dst->getAddr32(0, 0),
senorblanco@chromium.org0ded88d2014-01-24 15:43:50 +000085 radiusY, bounds.height(), bounds.width(),
robertphillipsf299e712016-03-25 04:49:22 -070086 srcRowBytesAsPixels, dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000087}
88
senorblancoe5e79842016-03-21 14:51:59 -070089SkRect SkMorphologyImageFilter::computeFastBounds(const SkRect& src) const {
90 SkRect bounds = this->getInput(0) ? this->getInput(0)->computeFastBounds(src) : src;
91 bounds.outset(SkIntToScalar(fRadius.width()), SkIntToScalar(fRadius.height()));
92 return bounds;
senorblanco@chromium.org336d1d72014-01-27 21:03:17 +000093}
94
senorblancoe5e79842016-03-21 14:51:59 -070095SkIRect SkMorphologyImageFilter::onFilterNodeBounds(const SkIRect& src, const SkMatrix& ctm,
96 MapDirection) const {
senorblanco@chromium.orgc4b12f12014-02-05 17:51:22 +000097 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
98 SkIntToScalar(this->radius().height()));
99 ctm.mapVectors(&radius, 1);
senorblancoe5e79842016-03-21 14:51:59 -0700100 return src.makeOutset(SkScalarCeilToInt(radius.x()), SkScalarCeilToInt(radius.y()));
senorblanco@chromium.orgc4b12f12014-02-05 17:51:22 +0000101}
102
reed60c9b582016-04-03 09:11:13 -0700103sk_sp<SkFlattenable> SkErodeImageFilter::CreateProc(SkReadBuffer& buffer) {
reed9fa60da2014-08-21 07:59:51 -0700104 SK_IMAGEFILTER_UNFLATTEN_COMMON(common, 1);
105 const int width = buffer.readInt();
106 const int height = buffer.readInt();
robertphillipsfc11b0a2016-04-05 09:09:36 -0700107 return Make(width, height, common.getInput(0), &common.cropRect());
reed9fa60da2014-08-21 07:59:51 -0700108}
109
reed60c9b582016-04-03 09:11:13 -0700110sk_sp<SkFlattenable> SkDilateImageFilter::CreateProc(SkReadBuffer& buffer) {
reed9fa60da2014-08-21 07:59:51 -0700111 SK_IMAGEFILTER_UNFLATTEN_COMMON(common, 1);
112 const int width = buffer.readInt();
113 const int height = buffer.readInt();
robertphillipsfc11b0a2016-04-05 09:09:36 -0700114 return Make(width, height, common.getInput(0), &common.cropRect());
reed9fa60da2014-08-21 07:59:51 -0700115}
116
robertphillipsf3f5bad2014-12-19 13:49:15 -0800117#ifndef SK_IGNORE_TO_STRING
118void SkErodeImageFilter::toString(SkString* str) const {
119 str->appendf("SkErodeImageFilter: (");
120 str->appendf("radius: (%d,%d)", this->radius().fWidth, this->radius().fHeight);
121 str->append(")");
122}
123#endif
124
125#ifndef SK_IGNORE_TO_STRING
126void SkDilateImageFilter::toString(SkString* str) const {
127 str->appendf("SkDilateImageFilter: (");
128 str->appendf("radius: (%d,%d)", this->radius().fWidth, this->radius().fHeight);
129 str->append(")");
130}
131#endif
132
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000133#if SK_SUPPORT_GPU
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000134
135///////////////////////////////////////////////////////////////////////////////
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000136/**
137 * Morphology effects. Depending upon the type of morphology, either the
138 * component-wise min (Erode_Type) or max (Dilate_Type) of all pixels in the
139 * kernel is selected as the new color. The new color is modulated by the input
140 * color.
141 */
142class GrMorphologyEffect : public Gr1DKernelEffect {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000143public:
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000144 enum MorphologyType {
145 kErode_MorphologyType,
146 kDilate_MorphologyType,
147 };
148
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400149 static sk_sp<GrFragmentProcessor> Make(sk_sp<GrTextureProxy> proxy,
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500150 Direction dir, int radius, MorphologyType type) {
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400151 return sk_sp<GrFragmentProcessor>(new GrMorphologyEffect(std::move(proxy),
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500152 dir, radius, type));
153 }
154
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400155 static sk_sp<GrFragmentProcessor> Make(sk_sp<GrTextureProxy> proxy, Direction dir, int radius,
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500156 MorphologyType type, const float bounds[2]) {
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400157 return sk_sp<GrFragmentProcessor>(new GrMorphologyEffect(std::move(proxy),
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500158 dir, radius, type, bounds));
159 }
160
161 ~GrMorphologyEffect() override;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000162
163 MorphologyType type() const { return fType; }
cwallez80a61df2015-01-26 12:20:14 -0800164 bool useRange() const { return fUseRange; }
165 const float* range() const { return fRange; }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000166
mtklein36352bf2015-03-25 18:17:31 -0700167 const char* name() const override { return "Morphology"; }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000168
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000169protected:
170
171 MorphologyType fType;
cwallez80a61df2015-01-26 12:20:14 -0800172 bool fUseRange;
173 float fRange[2];
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000174
175private:
egdaniel57d3b032015-11-13 11:57:27 -0800176 GrGLSLFragmentProcessor* onCreateGLSLInstance() const override;
wangyixb1daa862015-08-18 11:29:31 -0700177
Brian Salomon94efbf52016-11-29 13:43:05 -0500178 void onGetGLSLProcessorKey(const GrShaderCaps&, GrProcessorKeyBuilder*) const override;
wangyix4b3050b2015-08-04 07:59:37 -0700179
mtklein36352bf2015-03-25 18:17:31 -0700180 bool onIsEqual(const GrFragmentProcessor&) const override;
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000181
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400182 GrMorphologyEffect(sk_sp<GrTextureProxy>,
Robert Phillips296b1cc2017-03-15 10:42:12 -0400183 Direction, int radius, MorphologyType);
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400184 GrMorphologyEffect(sk_sp<GrTextureProxy>,
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500185 Direction, int radius, MorphologyType, const float bounds[2]);
186
Brian Salomon0c26a9d2017-07-06 10:09:38 -0400187 GR_DECLARE_FRAGMENT_PROCESSOR_TEST
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000188
189 typedef Gr1DKernelEffect INHERITED;
190};
191
192///////////////////////////////////////////////////////////////////////////////
193
egdaniel64c47282015-11-13 06:54:19 -0800194class GrGLMorphologyEffect : public GrGLSLFragmentProcessor {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000195public:
robertphillips9cdb9922016-02-03 12:25:40 -0800196 void emitCode(EmitArgs&) override;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000197
Brian Salomon94efbf52016-11-29 13:43:05 -0500198 static inline void GenKey(const GrProcessor&, const GrShaderCaps&, GrProcessorKeyBuilder*);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000199
wangyixb1daa862015-08-18 11:29:31 -0700200protected:
Brian Salomonab015ef2017-04-04 10:15:51 -0400201 void onSetData(const GrGLSLProgramDataManager&, const GrFragmentProcessor&) override;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000202
203private:
egdaniel018fb622015-10-28 07:26:40 -0700204 GrGLSLProgramDataManager::UniformHandle fPixelSizeUni;
205 GrGLSLProgramDataManager::UniformHandle fRangeUni;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000206
egdaniel64c47282015-11-13 06:54:19 -0800207 typedef GrGLSLFragmentProcessor INHERITED;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000208};
209
wangyix7c157a92015-07-22 15:08:53 -0700210void GrGLMorphologyEffect::emitCode(EmitArgs& args) {
robertphillipsbf536af2016-02-04 06:11:53 -0800211 const GrMorphologyEffect& me = args.fFp.cast<GrMorphologyEffect>();
212
egdaniel7ea439b2015-12-03 09:20:44 -0800213 GrGLSLUniformHandler* uniformHandler = args.fUniformHandler;
cdalton5e58cee2016-02-11 12:49:47 -0800214 fPixelSizeUni = uniformHandler->addUniform(kFragment_GrShaderFlag,
egdaniel7ea439b2015-12-03 09:20:44 -0800215 kFloat_GrSLType, kDefault_GrSLPrecision,
216 "PixelSize");
217 const char* pixelSizeInc = uniformHandler->getUniformCStr(fPixelSizeUni);
cdalton5e58cee2016-02-11 12:49:47 -0800218 fRangeUni = uniformHandler->addUniform(kFragment_GrShaderFlag,
egdaniel7ea439b2015-12-03 09:20:44 -0800219 kVec2f_GrSLType, kDefault_GrSLPrecision,
220 "Range");
221 const char* range = uniformHandler->getUniformCStr(fRangeUni);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000222
cdalton85285412016-02-18 12:37:07 -0800223 GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
bsalomon1a1aa932016-09-12 09:30:36 -0700224 SkString coords2D = fragBuilder->ensureCoords2D(args.fTransformedCoords[0]);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000225 const char* func;
robertphillipsbf536af2016-02-04 06:11:53 -0800226 switch (me.type()) {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000227 case GrMorphologyEffect::kErode_MorphologyType:
egdaniel4ca2e602015-11-18 08:01:26 -0800228 fragBuilder->codeAppendf("\t\t%s = vec4(1, 1, 1, 1);\n", args.fOutputColor);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000229 func = "min";
230 break;
231 case GrMorphologyEffect::kDilate_MorphologyType:
egdaniel4ca2e602015-11-18 08:01:26 -0800232 fragBuilder->codeAppendf("\t\t%s = vec4(0, 0, 0, 0);\n", args.fOutputColor);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000233 func = "max";
234 break;
235 default:
commit-bot@chromium.org88cb22b2014-04-30 14:17:00 +0000236 SkFAIL("Unexpected type");
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000237 func = ""; // suppress warning
238 break;
239 }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000240
cwallez80a61df2015-01-26 12:20:14 -0800241 const char* dir;
robertphillipsbf536af2016-02-04 06:11:53 -0800242 switch (me.direction()) {
cwallez80a61df2015-01-26 12:20:14 -0800243 case Gr1DKernelEffect::kX_Direction:
244 dir = "x";
245 break;
246 case Gr1DKernelEffect::kY_Direction:
247 dir = "y";
248 break;
249 default:
250 SkFAIL("Unknown filter direction.");
251 dir = ""; // suppress warning
252 }
253
robertphillipsbf536af2016-02-04 06:11:53 -0800254 int width = GrMorphologyEffect::WidthFromRadius(me.radius());
255
cwallez80a61df2015-01-26 12:20:14 -0800256 // vec2 coord = coord2D;
egdaniel4ca2e602015-11-18 08:01:26 -0800257 fragBuilder->codeAppendf("\t\tvec2 coord = %s;\n", coords2D.c_str());
cwallez80a61df2015-01-26 12:20:14 -0800258 // coord.x -= radius * pixelSize;
robertphillipsbf536af2016-02-04 06:11:53 -0800259 fragBuilder->codeAppendf("\t\tcoord.%s -= %d.0 * %s; \n", dir, me.radius(), pixelSizeInc);
260 if (me.useRange()) {
cwallez80a61df2015-01-26 12:20:14 -0800261 // highBound = min(highBound, coord.x + (width-1) * pixelSize);
egdaniel4ca2e602015-11-18 08:01:26 -0800262 fragBuilder->codeAppendf("\t\tfloat highBound = min(%s.y, coord.%s + %f * %s);",
robertphillipsbf536af2016-02-04 06:11:53 -0800263 range, dir, float(width - 1), pixelSizeInc);
cwallez80a61df2015-01-26 12:20:14 -0800264 // coord.x = max(lowBound, coord.x);
egdaniel4ca2e602015-11-18 08:01:26 -0800265 fragBuilder->codeAppendf("\t\tcoord.%s = max(%s.x, coord.%s);", dir, range, dir);
cwallez80a61df2015-01-26 12:20:14 -0800266 }
robertphillipsbf536af2016-02-04 06:11:53 -0800267 fragBuilder->codeAppendf("\t\tfor (int i = 0; i < %d; i++) {\n", width);
egdaniel4ca2e602015-11-18 08:01:26 -0800268 fragBuilder->codeAppendf("\t\t\t%s = %s(%s, ", args.fOutputColor, func, args.fOutputColor);
cdalton3f6f76f2016-04-11 12:18:09 -0700269 fragBuilder->appendTextureLookup(args.fTexSamplers[0], "coord");
egdaniel4ca2e602015-11-18 08:01:26 -0800270 fragBuilder->codeAppend(");\n");
cwallez80a61df2015-01-26 12:20:14 -0800271 // coord.x += pixelSize;
egdaniel4ca2e602015-11-18 08:01:26 -0800272 fragBuilder->codeAppendf("\t\t\tcoord.%s += %s;\n", dir, pixelSizeInc);
robertphillipsbf536af2016-02-04 06:11:53 -0800273 if (me.useRange()) {
cwallez80a61df2015-01-26 12:20:14 -0800274 // coord.x = min(highBound, coord.x);
egdaniel4ca2e602015-11-18 08:01:26 -0800275 fragBuilder->codeAppendf("\t\t\tcoord.%s = min(highBound, coord.%s);", dir, dir);
cwallez80a61df2015-01-26 12:20:14 -0800276 }
egdaniel4ca2e602015-11-18 08:01:26 -0800277 fragBuilder->codeAppend("\t\t}\n");
Ethan Nicholas2983f402017-05-08 09:36:08 -0400278 fragBuilder->codeAppendf("%s *= %s;\n", args.fOutputColor, args.fInputColor);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000279}
280
joshualittb0a8a372014-09-23 09:50:21 -0700281void GrGLMorphologyEffect::GenKey(const GrProcessor& proc,
Brian Salomon94efbf52016-11-29 13:43:05 -0500282 const GrShaderCaps&, GrProcessorKeyBuilder* b) {
joshualittb0a8a372014-09-23 09:50:21 -0700283 const GrMorphologyEffect& m = proc.cast<GrMorphologyEffect>();
bsalomon63e99f72014-07-21 08:03:14 -0700284 uint32_t key = static_cast<uint32_t>(m.radius());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000285 key |= (m.type() << 8);
cwallez80a61df2015-01-26 12:20:14 -0800286 key |= (m.direction() << 9);
robertphillipsbf536af2016-02-04 06:11:53 -0800287 if (m.useRange()) {
288 key |= 1 << 10;
289 }
bsalomon63e99f72014-07-21 08:03:14 -0700290 b->add32(key);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000291}
292
egdaniel018fb622015-10-28 07:26:40 -0700293void GrGLMorphologyEffect::onSetData(const GrGLSLProgramDataManager& pdman,
Brian Salomonab015ef2017-04-04 10:15:51 -0400294 const GrFragmentProcessor& proc) {
cwallez80a61df2015-01-26 12:20:14 -0800295 const GrMorphologyEffect& m = proc.cast<GrMorphologyEffect>();
Robert Phillipsc686ce32017-07-21 14:12:29 -0400296 GrSurfaceProxy* proxy = m.textureSampler(0).proxy();
297 GrTexture& texture = *proxy->priv().peekTexture();
cwallez80a61df2015-01-26 12:20:14 -0800298
299 float pixelSize = 0.0f;
robertphillipsbf536af2016-02-04 06:11:53 -0800300 switch (m.direction()) {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000301 case Gr1DKernelEffect::kX_Direction:
cwallez80a61df2015-01-26 12:20:14 -0800302 pixelSize = 1.0f / texture.width();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000303 break;
304 case Gr1DKernelEffect::kY_Direction:
cwallez80a61df2015-01-26 12:20:14 -0800305 pixelSize = 1.0f / texture.height();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000306 break;
307 default:
commit-bot@chromium.org88cb22b2014-04-30 14:17:00 +0000308 SkFAIL("Unknown filter direction.");
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000309 }
cwallez80a61df2015-01-26 12:20:14 -0800310 pdman.set1f(fPixelSizeUni, pixelSize);
311
robertphillipsbf536af2016-02-04 06:11:53 -0800312 if (m.useRange()) {
cwallez80a61df2015-01-26 12:20:14 -0800313 const float* range = m.range();
Robert Phillips7f6cd902016-11-10 17:03:43 -0500314 if (Gr1DKernelEffect::kY_Direction == m.direction() &&
Robert Phillipsc686ce32017-07-21 14:12:29 -0400315 proxy->origin() == kBottomLeft_GrSurfaceOrigin) {
Robert Phillips7f6cd902016-11-10 17:03:43 -0500316 pdman.set2f(fRangeUni, 1.0f - (range[1]*pixelSize), 1.0f - (range[0]*pixelSize));
cwallez80a61df2015-01-26 12:20:14 -0800317 } else {
Robert Phillips7f6cd902016-11-10 17:03:43 -0500318 pdman.set2f(fRangeUni, range[0] * pixelSize, range[1] * pixelSize);
cwallez80a61df2015-01-26 12:20:14 -0800319 }
320 }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000321}
322
323///////////////////////////////////////////////////////////////////////////////
324
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400325GrMorphologyEffect::GrMorphologyEffect(sk_sp<GrTextureProxy> proxy,
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500326 Direction direction,
327 int radius,
328 MorphologyType type)
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400329 : INHERITED{ModulationFlags(proxy->config()), GR_PROXY_MOVE(proxy), direction, radius}
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500330 , fType(type)
331 , fUseRange(false) {
332 this->initClassID<GrMorphologyEffect>();
333}
334
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400335GrMorphologyEffect::GrMorphologyEffect(sk_sp<GrTextureProxy> proxy,
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500336 Direction direction,
337 int radius,
338 MorphologyType type,
339 const float range[2])
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400340 : INHERITED{ModulationFlags(proxy->config()), GR_PROXY_MOVE(proxy), direction, radius}
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500341 , fType(type)
342 , fUseRange(true) {
343 this->initClassID<GrMorphologyEffect>();
344 fRange[0] = range[0];
345 fRange[1] = range[1];
346}
347
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000348GrMorphologyEffect::~GrMorphologyEffect() {
349}
350
Brian Salomon94efbf52016-11-29 13:43:05 -0500351void GrMorphologyEffect::onGetGLSLProcessorKey(const GrShaderCaps& caps,
egdaniel57d3b032015-11-13 11:57:27 -0800352 GrProcessorKeyBuilder* b) const {
joshualitteb2a6762014-12-04 11:35:33 -0800353 GrGLMorphologyEffect::GenKey(*this, caps, b);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000354}
355
egdaniel57d3b032015-11-13 11:57:27 -0800356GrGLSLFragmentProcessor* GrMorphologyEffect::onCreateGLSLInstance() const {
robertphillipsbf536af2016-02-04 06:11:53 -0800357 return new GrGLMorphologyEffect;
joshualitteb2a6762014-12-04 11:35:33 -0800358}
bsalomon0e08fc12014-10-15 08:19:04 -0700359bool GrMorphologyEffect::onIsEqual(const GrFragmentProcessor& sBase) const {
joshualitt49586be2014-09-16 08:21:41 -0700360 const GrMorphologyEffect& s = sBase.cast<GrMorphologyEffect>();
bsalomon420d7e92014-10-16 09:18:09 -0700361 return (this->radius() == s.radius() &&
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000362 this->direction() == s.direction() &&
cwallez80a61df2015-01-26 12:20:14 -0800363 this->useRange() == s.useRange() &&
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000364 this->type() == s.type());
365}
366
367///////////////////////////////////////////////////////////////////////////////
368
joshualittb0a8a372014-09-23 09:50:21 -0700369GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrMorphologyEffect);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000370
Hal Canary6f6961e2017-01-31 13:50:44 -0500371#if GR_TEST_UTILS
bungeman06ca8ec2016-06-09 08:01:03 -0700372sk_sp<GrFragmentProcessor> GrMorphologyEffect::TestCreate(GrProcessorTestData* d) {
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500373 int texIdx = d->fRandom->nextBool() ? GrProcessorUnitTest::kSkiaPMTextureIdx
374 : GrProcessorUnitTest::kAlphaTextureIdx;
375 sk_sp<GrTextureProxy> proxy = d->textureProxy(texIdx);
376
joshualitt0067ff52015-07-08 14:26:19 -0700377 Direction dir = d->fRandom->nextBool() ? kX_Direction : kY_Direction;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000378 static const int kMaxRadius = 10;
joshualitt0067ff52015-07-08 14:26:19 -0700379 int radius = d->fRandom->nextRangeU(1, kMaxRadius);
Robert Phillipsdbc8eeb2017-02-21 10:04:31 -0500380 MorphologyType type = d->fRandom->nextBool() ? GrMorphologyEffect::kErode_MorphologyType
381 : GrMorphologyEffect::kDilate_MorphologyType;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000382
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400383 return GrMorphologyEffect::Make(std::move(proxy), dir, radius, type);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000384}
Hal Canary6f6961e2017-01-31 13:50:44 -0500385#endif
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000386
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000387
Robert Phillips296b1cc2017-03-15 10:42:12 -0400388static void apply_morphology_rect(GrRenderTargetContext* renderTargetContext,
robertphillipsf299e712016-03-25 04:49:22 -0700389 const GrClip& clip,
Robert Phillips32f28182017-02-28 16:20:03 -0500390 sk_sp<GrTextureProxy> proxy,
robertphillipsf299e712016-03-25 04:49:22 -0700391 const SkIRect& srcRect,
392 const SkIRect& dstRect,
393 int radius,
394 GrMorphologyEffect::MorphologyType morphType,
Robert Phillips7f6cd902016-11-10 17:03:43 -0500395 const float bounds[2],
robertphillipsf299e712016-03-25 04:49:22 -0700396 Gr1DKernelEffect::Direction direction) {
cwallez80a61df2015-01-26 12:20:14 -0800397 GrPaint paint;
Brian Osman11052242016-10-27 14:47:55 -0400398 paint.setGammaCorrect(renderTargetContext->isGammaCorrect());
Robert Phillips32f28182017-02-28 16:20:03 -0500399
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400400 paint.addColorFragmentProcessor(GrMorphologyEffect::Make(std::move(proxy),
Robert Phillips32f28182017-02-28 16:20:03 -0500401 direction, radius, morphType,
bungeman06ca8ec2016-06-09 08:01:03 -0700402 bounds));
reed374772b2016-10-05 17:33:02 -0700403 paint.setPorterDuffXPFactory(SkBlendMode::kSrc);
Brian Salomon82f44312017-01-11 13:42:54 -0500404 renderTargetContext->fillRectToRect(clip, std::move(paint), GrAA::kNo, SkMatrix::I(),
405 SkRect::Make(dstRect), SkRect::Make(srcRect));
cwallez80a61df2015-01-26 12:20:14 -0800406}
407
Robert Phillips296b1cc2017-03-15 10:42:12 -0400408static void apply_morphology_rect_no_bounds(GrRenderTargetContext* renderTargetContext,
robertphillipsf299e712016-03-25 04:49:22 -0700409 const GrClip& clip,
Robert Phillips32f28182017-02-28 16:20:03 -0500410 sk_sp<GrTextureProxy> proxy,
robertphillipsf299e712016-03-25 04:49:22 -0700411 const SkIRect& srcRect,
412 const SkIRect& dstRect,
413 int radius,
414 GrMorphologyEffect::MorphologyType morphType,
415 Gr1DKernelEffect::Direction direction) {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000416 GrPaint paint;
Brian Osman11052242016-10-27 14:47:55 -0400417 paint.setGammaCorrect(renderTargetContext->isGammaCorrect());
Robert Phillips32f28182017-02-28 16:20:03 -0500418
Robert Phillipsfbcef6e2017-06-15 12:07:18 -0400419 paint.addColorFragmentProcessor(GrMorphologyEffect::Make(std::move(proxy),
Robert Phillips32f28182017-02-28 16:20:03 -0500420 direction, radius, morphType));
reed374772b2016-10-05 17:33:02 -0700421 paint.setPorterDuffXPFactory(SkBlendMode::kSrc);
Brian Salomon82f44312017-01-11 13:42:54 -0500422 renderTargetContext->fillRectToRect(clip, std::move(paint), GrAA::kNo, SkMatrix::I(),
Brian Salomon0e8fc8b2016-12-09 15:10:07 -0500423 SkRect::Make(dstRect), SkRect::Make(srcRect));
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000424}
425
Robert Phillips296b1cc2017-03-15 10:42:12 -0400426static void apply_morphology_pass(GrRenderTargetContext* renderTargetContext,
robertphillipsf299e712016-03-25 04:49:22 -0700427 const GrClip& clip,
Robert Phillips32f28182017-02-28 16:20:03 -0500428 sk_sp<GrTextureProxy> textureProxy,
robertphillipsf299e712016-03-25 04:49:22 -0700429 const SkIRect& srcRect,
430 const SkIRect& dstRect,
431 int radius,
432 GrMorphologyEffect::MorphologyType morphType,
433 Gr1DKernelEffect::Direction direction) {
cwallez80a61df2015-01-26 12:20:14 -0800434 float bounds[2] = { 0.0f, 1.0f };
435 SkIRect lowerSrcRect = srcRect, lowerDstRect = dstRect;
436 SkIRect middleSrcRect = srcRect, middleDstRect = dstRect;
437 SkIRect upperSrcRect = srcRect, upperDstRect = dstRect;
438 if (direction == Gr1DKernelEffect::kX_Direction) {
Robert Phillips7f6cd902016-11-10 17:03:43 -0500439 bounds[0] = SkIntToScalar(srcRect.left()) + 0.5f;
440 bounds[1] = SkIntToScalar(srcRect.right()) - 0.5f;
cwallez80a61df2015-01-26 12:20:14 -0800441 lowerSrcRect.fRight = srcRect.left() + radius;
442 lowerDstRect.fRight = dstRect.left() + radius;
443 upperSrcRect.fLeft = srcRect.right() - radius;
444 upperDstRect.fLeft = dstRect.right() - radius;
445 middleSrcRect.inset(radius, 0);
446 middleDstRect.inset(radius, 0);
447 } else {
Robert Phillips7f6cd902016-11-10 17:03:43 -0500448 bounds[0] = SkIntToScalar(srcRect.top()) + 0.5f;
449 bounds[1] = SkIntToScalar(srcRect.bottom()) - 0.5f;
cwallez80a61df2015-01-26 12:20:14 -0800450 lowerSrcRect.fBottom = srcRect.top() + radius;
451 lowerDstRect.fBottom = dstRect.top() + radius;
452 upperSrcRect.fTop = srcRect.bottom() - radius;
453 upperDstRect.fTop = dstRect.bottom() - radius;
454 middleSrcRect.inset(0, radius);
455 middleDstRect.inset(0, radius);
456 }
457 if (middleSrcRect.fLeft - middleSrcRect.fRight >= 0) {
458 // radius covers srcRect; use bounds over entire draw
Robert Phillips296b1cc2017-03-15 10:42:12 -0400459 apply_morphology_rect(renderTargetContext, clip, std::move(textureProxy),
Robert Phillips7f6cd902016-11-10 17:03:43 -0500460 srcRect, dstRect, radius, morphType, bounds, direction);
cwallez80a61df2015-01-26 12:20:14 -0800461 } else {
462 // Draw upper and lower margins with bounds; middle without.
Robert Phillips296b1cc2017-03-15 10:42:12 -0400463 apply_morphology_rect(renderTargetContext, clip, textureProxy,
Robert Phillips7f6cd902016-11-10 17:03:43 -0500464 lowerSrcRect, lowerDstRect, radius, morphType, bounds, direction);
Robert Phillips296b1cc2017-03-15 10:42:12 -0400465 apply_morphology_rect(renderTargetContext, clip, textureProxy,
Robert Phillips7f6cd902016-11-10 17:03:43 -0500466 upperSrcRect, upperDstRect, radius, morphType, bounds, direction);
Robert Phillips296b1cc2017-03-15 10:42:12 -0400467 apply_morphology_rect_no_bounds(renderTargetContext, clip, std::move(textureProxy),
Robert Phillips7f6cd902016-11-10 17:03:43 -0500468 middleSrcRect, middleDstRect, radius, morphType, direction);
cwallez80a61df2015-01-26 12:20:14 -0800469 }
470}
471
brianosman2a75e5d2016-09-22 07:15:37 -0700472static sk_sp<SkSpecialImage> apply_morphology(
473 GrContext* context,
474 SkSpecialImage* input,
475 const SkIRect& rect,
476 GrMorphologyEffect::MorphologyType morphType,
477 SkISize radius,
478 const SkImageFilter::OutputProperties& outputProperties) {
Robert Phillips8e1c4e62017-02-19 12:27:01 -0500479 sk_sp<GrTextureProxy> srcTexture(input->asTextureProxyRef(context));
bsalomon49f085d2014-09-05 13:34:00 -0700480 SkASSERT(srcTexture);
brianosman2a75e5d2016-09-22 07:15:37 -0700481 sk_sp<SkColorSpace> colorSpace = sk_ref_sp(outputProperties.colorSpace());
brianosman2695eaa2016-09-21 06:45:09 -0700482 GrPixelConfig config = GrRenderableConfigForColorSpace(colorSpace.get());
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000483
joshualitt570d2f82015-02-25 13:19:48 -0800484 // setup new clip
cdalton846c0512016-05-13 10:25:00 -0700485 const GrFixedClip clip(SkIRect::MakeWH(srcTexture->width(), srcTexture->height()));
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000486
robertphillipsd4c741e2016-04-28 09:55:15 -0700487 const SkIRect dstRect = SkIRect::MakeWH(rect.width(), rect.height());
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000488 SkIRect srcRect = rect;
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000489
robertphillipsf299e712016-03-25 04:49:22 -0700490 SkASSERT(radius.width() > 0 || radius.height() > 0);
491
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000492 if (radius.fWidth > 0) {
Robert Phillips7f6cd902016-11-10 17:03:43 -0500493 sk_sp<GrRenderTargetContext> dstRTContext(context->makeDeferredRenderTargetContext(
Brian Osman11052242016-10-27 14:47:55 -0400494 SkBackingFit::kApprox, rect.width(), rect.height(), config, colorSpace));
Robert Phillips7f6cd902016-11-10 17:03:43 -0500495 if (!dstRTContext) {
robertphillipsf299e712016-03-25 04:49:22 -0700496 return nullptr;
robertphillipsff0ca5e2015-07-22 11:54:44 -0700497 }
498
Robert Phillips296b1cc2017-03-15 10:42:12 -0400499 apply_morphology_pass(dstRTContext.get(), clip, std::move(srcTexture),
joshualitt570d2f82015-02-25 13:19:48 -0800500 srcRect, dstRect, radius.fWidth, morphType,
501 Gr1DKernelEffect::kX_Direction);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000502 SkIRect clearRect = SkIRect::MakeXYWH(dstRect.fLeft, dstRect.fBottom,
503 dstRect.width(), radius.fHeight);
robertphillipsd4c741e2016-04-28 09:55:15 -0700504 GrColor clearColor = GrMorphologyEffect::kErode_MorphologyType == morphType
505 ? SK_ColorWHITE
506 : SK_ColorTRANSPARENT;
Robert Phillips7f6cd902016-11-10 17:03:43 -0500507 dstRTContext->clear(&clearRect, clearColor, false);
robertphillipsff0ca5e2015-07-22 11:54:44 -0700508
Robert Phillipsf200a902017-01-30 13:27:37 -0500509 srcTexture = dstRTContext->asTextureProxyRef();
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000510 srcRect = dstRect;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000511 }
512 if (radius.fHeight > 0) {
Robert Phillips7f6cd902016-11-10 17:03:43 -0500513 sk_sp<GrRenderTargetContext> dstRTContext(context->makeDeferredRenderTargetContext(
Brian Osman11052242016-10-27 14:47:55 -0400514 SkBackingFit::kApprox, rect.width(), rect.height(), config, colorSpace));
Robert Phillips7f6cd902016-11-10 17:03:43 -0500515 if (!dstRTContext) {
robertphillipsf299e712016-03-25 04:49:22 -0700516 return nullptr;
robertphillipsff0ca5e2015-07-22 11:54:44 -0700517 }
518
Robert Phillips296b1cc2017-03-15 10:42:12 -0400519 apply_morphology_pass(dstRTContext.get(), clip, std::move(srcTexture),
joshualitt570d2f82015-02-25 13:19:48 -0800520 srcRect, dstRect, radius.fHeight, morphType,
521 Gr1DKernelEffect::kY_Direction);
robertphillipsff0ca5e2015-07-22 11:54:44 -0700522
Robert Phillipsf200a902017-01-30 13:27:37 -0500523 srcTexture = dstRTContext->asTextureProxyRef();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000524 }
robertphillipsf299e712016-03-25 04:49:22 -0700525
Robert Phillips7f6cd902016-11-10 17:03:43 -0500526 return SkSpecialImage::MakeDeferredFromGpu(context,
527 SkIRect::MakeWH(rect.width(), rect.height()),
528 kNeedNewImageUniqueID_SpecialImage,
529 std::move(srcTexture), std::move(colorSpace),
530 &input->props());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000531}
robertphillipsf299e712016-03-25 04:49:22 -0700532#endif
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000533
robertphillips40d8d622016-03-30 08:09:56 -0700534sk_sp<SkSpecialImage> SkMorphologyImageFilter::onFilterImage(SkSpecialImage* source,
535 const Context& ctx,
536 SkIPoint* offset) const {
robertphillipsf299e712016-03-25 04:49:22 -0700537 SkIPoint inputOffset = SkIPoint::Make(0, 0);
538 sk_sp<SkSpecialImage> input(this->filterInput(0, source, ctx, &inputOffset));
539 if (!input) {
540 return nullptr;
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000541 }
robertphillipsf299e712016-03-25 04:49:22 -0700542
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000543 SkIRect bounds;
robertphillipsf299e712016-03-25 04:49:22 -0700544 input = this->applyCropRect(this->mapContext(ctx), input.get(), &inputOffset, &bounds);
545 if (!input) {
546 return nullptr;
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000547 }
robertphillipsf299e712016-03-25 04:49:22 -0700548
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000549 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
550 SkIntToScalar(this->radius().height()));
senorblanco@chromium.org4cb543d2014-03-14 15:44:01 +0000551 ctx.ctm().mapVectors(&radius, 1);
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000552 int width = SkScalarFloorToInt(radius.fX);
553 int height = SkScalarFloorToInt(radius.fY);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000554
555 if (width < 0 || height < 0) {
robertphillipsf299e712016-03-25 04:49:22 -0700556 return nullptr;
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000557 }
558
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000559 SkIRect srcBounds = bounds;
robertphillipsf299e712016-03-25 04:49:22 -0700560 srcBounds.offset(-inputOffset);
561
562 if (0 == width && 0 == height) {
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000563 offset->fX = bounds.left();
564 offset->fY = bounds.top();
robertphillipsf299e712016-03-25 04:49:22 -0700565 return input->makeSubset(srcBounds);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000566 }
567
robertphillipsf299e712016-03-25 04:49:22 -0700568#if SK_SUPPORT_GPU
robertphillips64612512016-04-08 12:10:42 -0700569 if (source->isTextureBacked()) {
570 GrContext* context = source->getContext();
571
Brian Osman615d66d2016-12-29 09:18:20 -0500572 // Ensure the input is in the destination color space. Typically applyCropRect will have
573 // called pad_image to account for our dilation of bounds, so the result will already be
574 // moved to the destination color space. If a filter DAG avoids that, then we use this
575 // fall-back, which saves us from having to do the xform during the filter itself.
576 input = ImageToColorSpace(input.get(), ctx.outputProperties());
Brian Osman320573f2016-12-21 13:36:02 -0500577
robertphillips40d8d622016-03-30 08:09:56 -0700578 auto type = (kDilate_Op == this->op()) ? GrMorphologyEffect::kDilate_MorphologyType
579 : GrMorphologyEffect::kErode_MorphologyType;
robertphillips64612512016-04-08 12:10:42 -0700580 sk_sp<SkSpecialImage> result(apply_morphology(context, input.get(), srcBounds, type,
brianosman2a75e5d2016-09-22 07:15:37 -0700581 SkISize::Make(width, height),
582 ctx.outputProperties()));
robertphillipsf299e712016-03-25 04:49:22 -0700583 if (result) {
584 offset->fX = bounds.left();
585 offset->fY = bounds.top();
586 }
587 return result;
588 }
589#endif
590
robertphillips64612512016-04-08 12:10:42 -0700591 SkBitmap inputBM;
robertphillipsf299e712016-03-25 04:49:22 -0700592
robertphillips64612512016-04-08 12:10:42 -0700593 if (!input->getROPixels(&inputBM)) {
robertphillipsf299e712016-03-25 04:49:22 -0700594 return nullptr;
595 }
596
robertphillips64612512016-04-08 12:10:42 -0700597 if (inputBM.colorType() != kN32_SkColorType) {
robertphillipsf299e712016-03-25 04:49:22 -0700598 return nullptr;
599 }
600
601 SkImageInfo info = SkImageInfo::Make(bounds.width(), bounds.height(),
robertphillips64612512016-04-08 12:10:42 -0700602 inputBM.colorType(), inputBM.alphaType());
robertphillipsf299e712016-03-25 04:49:22 -0700603
604 SkBitmap dst;
605 if (!dst.tryAllocPixels(info)) {
606 return nullptr;
607 }
608
robertphillipsf299e712016-03-25 04:49:22 -0700609 SkMorphologyImageFilter::Proc procX, procY;
610
robertphillips40d8d622016-03-30 08:09:56 -0700611 if (kDilate_Op == this->op()) {
robertphillipsf299e712016-03-25 04:49:22 -0700612 procX = SkOpts::dilate_x;
613 procY = SkOpts::dilate_y;
614 } else {
615 procX = SkOpts::erode_x;
616 procY = SkOpts::erode_y;
617 }
618
619 if (width > 0 && height > 0) {
620 SkBitmap tmp;
621 if (!tmp.tryAllocPixels(info)) {
622 return nullptr;
623 }
624
robertphillips64612512016-04-08 12:10:42 -0700625 call_proc_X(procX, inputBM, &tmp, width, srcBounds);
robertphillipsf299e712016-03-25 04:49:22 -0700626 SkIRect tmpBounds = SkIRect::MakeWH(srcBounds.width(), srcBounds.height());
627 call_proc_Y(procY,
628 tmp.getAddr32(tmpBounds.left(), tmpBounds.top()), tmp.rowBytesAsPixels(),
629 &dst, height, tmpBounds);
630 } else if (width > 0) {
robertphillips64612512016-04-08 12:10:42 -0700631 call_proc_X(procX, inputBM, &dst, width, srcBounds);
robertphillipsf299e712016-03-25 04:49:22 -0700632 } else if (height > 0) {
633 call_proc_Y(procY,
robertphillips64612512016-04-08 12:10:42 -0700634 inputBM.getAddr32(srcBounds.left(), srcBounds.top()),
635 inputBM.rowBytesAsPixels(),
robertphillipsf299e712016-03-25 04:49:22 -0700636 &dst, height, srcBounds);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000637 }
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000638 offset->fX = bounds.left();
639 offset->fY = bounds.top();
robertphillipsf299e712016-03-25 04:49:22 -0700640
robertphillips3e302272016-04-20 11:48:36 -0700641 return SkSpecialImage::MakeFromRaster(SkIRect::MakeWH(bounds.width(), bounds.height()),
brianosman898235c2016-04-06 07:38:23 -0700642 dst, &source->props());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000643}
Matt Sarett6d72ed92017-04-10 16:35:33 -0400644
Mike Reed6d9f4292017-07-06 12:32:55 -0400645sk_sp<SkImageFilter> SkMorphologyImageFilter::onMakeColorSpace(SkColorSpaceXformer* xformer) const{
Matt Sarett6d72ed92017-04-10 16:35:33 -0400646 SkASSERT(1 == this->countInputs());
Mike Reed6d9f4292017-07-06 12:32:55 -0400647 auto input = xformer->apply(this->getInput(0));
648 if (input.get() != this->getInput(0)) {
649 return (SkMorphologyImageFilter::kDilate_Op == this->op())
650 ? SkDilateImageFilter::Make(fRadius.width(), fRadius.height(), std::move(input),
651 this->getCropRectIfSet())
652 : SkErodeImageFilter::Make(fRadius.width(), fRadius.height(), std::move(input),
653 this->getCropRectIfSet());
Matt Sarett6d72ed92017-04-10 16:35:33 -0400654 }
Mike Reed6d9f4292017-07-06 12:32:55 -0400655 return this->refMe();
Matt Sarett6d72ed92017-04-10 16:35:33 -0400656}