blob: cdc236b63a09d306075dd3a9585b38b9b38ef0b9 [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"
mtkleind029ded2015-08-04 14:09:09 -070012#include "SkOpts.h"
commit-bot@chromium.org8b0e8ac2014-01-30 18:58:24 +000013#include "SkReadBuffer.h"
djsollen@google.comc73dd5c2012-08-07 15:54:32 +000014#include "SkRect.h"
robertphillipsf299e712016-03-25 04:49:22 -070015#include "SkSpecialImage.h"
mtkleind029ded2015-08-04 14:09:09 -070016#include "SkWriteBuffer.h"
robertphillipsf299e712016-03-25 04:49:22 -070017
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +000018#if SK_SUPPORT_GPU
senorblanco@chromium.org302cffb2012-08-01 20:16:34 +000019#include "GrContext.h"
robertphillipsea461502015-05-26 11:38:03 -070020#include "GrDrawContext.h"
csmartdalton02fa32c2016-08-19 13:29:27 -070021#include "GrFixedClip.h"
egdaniel605dd0f2014-11-12 08:35:25 -080022#include "GrInvariantOutput.h"
senorblanco@chromium.org302cffb2012-08-01 20:16:34 +000023#include "GrTexture.h"
robertphillips1de87df2016-01-14 06:03:29 -080024#include "SkGr.h"
joshualitteb2a6762014-12-04 11:35:33 -080025#include "effects/Gr1DKernelEffect.h"
egdaniel64c47282015-11-13 06:54:19 -080026#include "glsl/GrGLSLFragmentProcessor.h"
egdaniel2d721d32015-11-11 13:06:05 -080027#include "glsl/GrGLSLFragmentShaderBuilder.h"
egdaniel018fb622015-10-28 07:26:40 -070028#include "glsl/GrGLSLProgramDataManager.h"
egdaniel7ea439b2015-12-03 09:20:44 -080029#include "glsl/GrGLSLUniformHandler.h"
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +000030#endif
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000031
robertphillips225db442016-04-17 14:27:05 -070032sk_sp<SkImageFilter> SkDilateImageFilter::Make(int radiusX, int radiusY,
33 sk_sp<SkImageFilter> input,
34 const CropRect* cropRect) {
35 if (radiusX < 0 || radiusY < 0) {
36 return nullptr;
37 }
38 return sk_sp<SkImageFilter>(new SkDilateImageFilter(radiusX, radiusY,
39 std::move(input),
40 cropRect));
41}
42
43
44sk_sp<SkImageFilter> SkErodeImageFilter::Make(int radiusX, int radiusY,
45 sk_sp<SkImageFilter> input,
46 const CropRect* cropRect) {
47 if (radiusX < 0 || radiusY < 0) {
48 return nullptr;
49 }
50 return sk_sp<SkImageFilter>(new SkErodeImageFilter(radiusX, radiusY,
51 std::move(input),
52 cropRect));
53}
54
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +000055SkMorphologyImageFilter::SkMorphologyImageFilter(int radiusX,
56 int radiusY,
robertphillipsfc11b0a2016-04-05 09:09:36 -070057 sk_sp<SkImageFilter> input,
senorblanco24e06d52015-03-18 12:11:33 -070058 const CropRect* cropRect)
robertphillipsfc11b0a2016-04-05 09:09:36 -070059 : INHERITED(&input, 1, cropRect)
robertphillipsf299e712016-03-25 04:49:22 -070060 , fRadius(SkISize::Make(radiusX, radiusY)) {
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000061}
62
commit-bot@chromium.org8b0e8ac2014-01-30 18:58:24 +000063void SkMorphologyImageFilter::flatten(SkWriteBuffer& buffer) const {
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000064 this->INHERITED::flatten(buffer);
tomhudson@google.com75589252012-04-10 17:42:21 +000065 buffer.writeInt(fRadius.fWidth);
66 buffer.writeInt(fRadius.fHeight);
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000067}
68
robertphillipsf299e712016-03-25 04:49:22 -070069static void call_proc_X(SkMorphologyImageFilter::Proc procX,
robertphillips64612512016-04-08 12:10:42 -070070 const SkBitmap& src, SkBitmap* dst,
robertphillipsf299e712016-03-25 04:49:22 -070071 int radiusX, const SkIRect& bounds) {
robertphillips64612512016-04-08 12:10:42 -070072 procX(src.getAddr32(bounds.left(), bounds.top()), dst->getAddr32(0, 0),
senorblanco@chromium.org0ded88d2014-01-24 15:43:50 +000073 radiusX, bounds.width(), bounds.height(),
74 src.rowBytesAsPixels(), dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000075}
76
robertphillipsf299e712016-03-25 04:49:22 -070077static void call_proc_Y(SkMorphologyImageFilter::Proc procY,
78 const SkPMColor* src, int srcRowBytesAsPixels, SkBitmap* dst,
79 int radiusY, const SkIRect& bounds) {
80 procY(src, dst->getAddr32(0, 0),
senorblanco@chromium.org0ded88d2014-01-24 15:43:50 +000081 radiusY, bounds.height(), bounds.width(),
robertphillipsf299e712016-03-25 04:49:22 -070082 srcRowBytesAsPixels, dst->rowBytesAsPixels());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +000083}
84
senorblancoe5e79842016-03-21 14:51:59 -070085SkRect SkMorphologyImageFilter::computeFastBounds(const SkRect& src) const {
86 SkRect bounds = this->getInput(0) ? this->getInput(0)->computeFastBounds(src) : src;
87 bounds.outset(SkIntToScalar(fRadius.width()), SkIntToScalar(fRadius.height()));
88 return bounds;
senorblanco@chromium.org336d1d72014-01-27 21:03:17 +000089}
90
senorblancoe5e79842016-03-21 14:51:59 -070091SkIRect SkMorphologyImageFilter::onFilterNodeBounds(const SkIRect& src, const SkMatrix& ctm,
92 MapDirection) const {
senorblanco@chromium.orgc4b12f12014-02-05 17:51:22 +000093 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
94 SkIntToScalar(this->radius().height()));
95 ctm.mapVectors(&radius, 1);
senorblancoe5e79842016-03-21 14:51:59 -070096 return src.makeOutset(SkScalarCeilToInt(radius.x()), SkScalarCeilToInt(radius.y()));
senorblanco@chromium.orgc4b12f12014-02-05 17:51:22 +000097}
98
reed60c9b582016-04-03 09:11:13 -070099sk_sp<SkFlattenable> SkErodeImageFilter::CreateProc(SkReadBuffer& buffer) {
reed9fa60da2014-08-21 07:59:51 -0700100 SK_IMAGEFILTER_UNFLATTEN_COMMON(common, 1);
101 const int width = buffer.readInt();
102 const int height = buffer.readInt();
robertphillipsfc11b0a2016-04-05 09:09:36 -0700103 return Make(width, height, common.getInput(0), &common.cropRect());
reed9fa60da2014-08-21 07:59:51 -0700104}
105
reed60c9b582016-04-03 09:11:13 -0700106sk_sp<SkFlattenable> SkDilateImageFilter::CreateProc(SkReadBuffer& buffer) {
reed9fa60da2014-08-21 07:59:51 -0700107 SK_IMAGEFILTER_UNFLATTEN_COMMON(common, 1);
108 const int width = buffer.readInt();
109 const int height = buffer.readInt();
robertphillipsfc11b0a2016-04-05 09:09:36 -0700110 return Make(width, height, common.getInput(0), &common.cropRect());
reed9fa60da2014-08-21 07:59:51 -0700111}
112
robertphillipsf3f5bad2014-12-19 13:49:15 -0800113#ifndef SK_IGNORE_TO_STRING
114void SkErodeImageFilter::toString(SkString* str) const {
115 str->appendf("SkErodeImageFilter: (");
116 str->appendf("radius: (%d,%d)", this->radius().fWidth, this->radius().fHeight);
117 str->append(")");
118}
119#endif
120
121#ifndef SK_IGNORE_TO_STRING
122void SkDilateImageFilter::toString(SkString* str) const {
123 str->appendf("SkDilateImageFilter: (");
124 str->appendf("radius: (%d,%d)", this->radius().fWidth, this->radius().fHeight);
125 str->append(")");
126}
127#endif
128
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000129#if SK_SUPPORT_GPU
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000130
131///////////////////////////////////////////////////////////////////////////////
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000132/**
133 * Morphology effects. Depending upon the type of morphology, either the
134 * component-wise min (Erode_Type) or max (Dilate_Type) of all pixels in the
135 * kernel is selected as the new color. The new color is modulated by the input
136 * color.
137 */
138class GrMorphologyEffect : public Gr1DKernelEffect {
139
140public:
141
142 enum MorphologyType {
143 kErode_MorphologyType,
144 kDilate_MorphologyType,
145 };
146
bungeman06ca8ec2016-06-09 08:01:03 -0700147 static sk_sp<GrFragmentProcessor> Make(GrTexture* tex, Direction dir, int radius,
148 MorphologyType type) {
149 return sk_sp<GrFragmentProcessor>(new GrMorphologyEffect(tex, dir, radius, type));
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000150 }
151
bungeman06ca8ec2016-06-09 08:01:03 -0700152 static sk_sp<GrFragmentProcessor> Make(GrTexture* tex, Direction dir, int radius,
153 MorphologyType type, float bounds[2]) {
154 return sk_sp<GrFragmentProcessor>(new GrMorphologyEffect(tex, dir, radius, type, bounds));
cwallez80a61df2015-01-26 12:20:14 -0800155 }
156
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000157 virtual ~GrMorphologyEffect();
158
159 MorphologyType type() const { return fType; }
cwallez80a61df2015-01-26 12:20:14 -0800160 bool useRange() const { return fUseRange; }
161 const float* range() const { return fRange; }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000162
mtklein36352bf2015-03-25 18:17:31 -0700163 const char* name() const override { return "Morphology"; }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000164
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000165protected:
166
167 MorphologyType fType;
cwallez80a61df2015-01-26 12:20:14 -0800168 bool fUseRange;
169 float fRange[2];
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000170
171private:
egdaniel57d3b032015-11-13 11:57:27 -0800172 GrGLSLFragmentProcessor* onCreateGLSLInstance() const override;
wangyixb1daa862015-08-18 11:29:31 -0700173
egdaniel57d3b032015-11-13 11:57:27 -0800174 void onGetGLSLProcessorKey(const GrGLSLCaps&, GrProcessorKeyBuilder*) const override;
wangyix4b3050b2015-08-04 07:59:37 -0700175
mtklein36352bf2015-03-25 18:17:31 -0700176 bool onIsEqual(const GrFragmentProcessor&) const override;
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000177
mtklein36352bf2015-03-25 18:17:31 -0700178 void onComputeInvariantOutput(GrInvariantOutput* inout) const override;
egdaniel1a8ecdf2014-10-03 06:24:12 -0700179
bsalomon4a339522015-10-06 08:40:50 -0700180 GrMorphologyEffect(GrTexture*, Direction, int radius, MorphologyType);
181 GrMorphologyEffect(GrTexture*, Direction, int radius, MorphologyType,
joshualitt5f10b5c2015-07-09 10:24:35 -0700182 float bounds[2]);
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000183
joshualittb0a8a372014-09-23 09:50:21 -0700184 GR_DECLARE_FRAGMENT_PROCESSOR_TEST;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000185
186 typedef Gr1DKernelEffect INHERITED;
187};
188
189///////////////////////////////////////////////////////////////////////////////
190
egdaniel64c47282015-11-13 06:54:19 -0800191class GrGLMorphologyEffect : public GrGLSLFragmentProcessor {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000192public:
robertphillips9cdb9922016-02-03 12:25:40 -0800193 void emitCode(EmitArgs&) override;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000194
robertphillipsbf536af2016-02-04 06:11:53 -0800195 static inline void GenKey(const GrProcessor&, const GrGLSLCaps&, GrProcessorKeyBuilder*);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000196
wangyixb1daa862015-08-18 11:29:31 -0700197protected:
egdaniel018fb622015-10-28 07:26:40 -0700198 void onSetData(const GrGLSLProgramDataManager&, const GrProcessor&) override;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000199
200private:
egdaniel018fb622015-10-28 07:26:40 -0700201 GrGLSLProgramDataManager::UniformHandle fPixelSizeUni;
202 GrGLSLProgramDataManager::UniformHandle fRangeUni;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000203
egdaniel64c47282015-11-13 06:54:19 -0800204 typedef GrGLSLFragmentProcessor INHERITED;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000205};
206
wangyix7c157a92015-07-22 15:08:53 -0700207void GrGLMorphologyEffect::emitCode(EmitArgs& args) {
robertphillipsbf536af2016-02-04 06:11:53 -0800208 const GrMorphologyEffect& me = args.fFp.cast<GrMorphologyEffect>();
209
egdaniel7ea439b2015-12-03 09:20:44 -0800210 GrGLSLUniformHandler* uniformHandler = args.fUniformHandler;
cdalton5e58cee2016-02-11 12:49:47 -0800211 fPixelSizeUni = uniformHandler->addUniform(kFragment_GrShaderFlag,
egdaniel7ea439b2015-12-03 09:20:44 -0800212 kFloat_GrSLType, kDefault_GrSLPrecision,
213 "PixelSize");
214 const char* pixelSizeInc = uniformHandler->getUniformCStr(fPixelSizeUni);
cdalton5e58cee2016-02-11 12:49:47 -0800215 fRangeUni = uniformHandler->addUniform(kFragment_GrShaderFlag,
egdaniel7ea439b2015-12-03 09:20:44 -0800216 kVec2f_GrSLType, kDefault_GrSLPrecision,
217 "Range");
218 const char* range = uniformHandler->getUniformCStr(fRangeUni);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000219
cdalton85285412016-02-18 12:37:07 -0800220 GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
egdaniel4ca2e602015-11-18 08:01:26 -0800221 SkString coords2D = fragBuilder->ensureFSCoords2D(args.fCoords, 0);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000222 const char* func;
robertphillipsbf536af2016-02-04 06:11:53 -0800223 switch (me.type()) {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000224 case GrMorphologyEffect::kErode_MorphologyType:
egdaniel4ca2e602015-11-18 08:01:26 -0800225 fragBuilder->codeAppendf("\t\t%s = vec4(1, 1, 1, 1);\n", args.fOutputColor);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000226 func = "min";
227 break;
228 case GrMorphologyEffect::kDilate_MorphologyType:
egdaniel4ca2e602015-11-18 08:01:26 -0800229 fragBuilder->codeAppendf("\t\t%s = vec4(0, 0, 0, 0);\n", args.fOutputColor);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000230 func = "max";
231 break;
232 default:
commit-bot@chromium.org88cb22b2014-04-30 14:17:00 +0000233 SkFAIL("Unexpected type");
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000234 func = ""; // suppress warning
235 break;
236 }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000237
cwallez80a61df2015-01-26 12:20:14 -0800238 const char* dir;
robertphillipsbf536af2016-02-04 06:11:53 -0800239 switch (me.direction()) {
cwallez80a61df2015-01-26 12:20:14 -0800240 case Gr1DKernelEffect::kX_Direction:
241 dir = "x";
242 break;
243 case Gr1DKernelEffect::kY_Direction:
244 dir = "y";
245 break;
246 default:
247 SkFAIL("Unknown filter direction.");
248 dir = ""; // suppress warning
249 }
250
robertphillipsbf536af2016-02-04 06:11:53 -0800251 int width = GrMorphologyEffect::WidthFromRadius(me.radius());
252
cwallez80a61df2015-01-26 12:20:14 -0800253 // vec2 coord = coord2D;
egdaniel4ca2e602015-11-18 08:01:26 -0800254 fragBuilder->codeAppendf("\t\tvec2 coord = %s;\n", coords2D.c_str());
cwallez80a61df2015-01-26 12:20:14 -0800255 // coord.x -= radius * pixelSize;
robertphillipsbf536af2016-02-04 06:11:53 -0800256 fragBuilder->codeAppendf("\t\tcoord.%s -= %d.0 * %s; \n", dir, me.radius(), pixelSizeInc);
257 if (me.useRange()) {
cwallez80a61df2015-01-26 12:20:14 -0800258 // highBound = min(highBound, coord.x + (width-1) * pixelSize);
egdaniel4ca2e602015-11-18 08:01:26 -0800259 fragBuilder->codeAppendf("\t\tfloat highBound = min(%s.y, coord.%s + %f * %s);",
robertphillipsbf536af2016-02-04 06:11:53 -0800260 range, dir, float(width - 1), pixelSizeInc);
cwallez80a61df2015-01-26 12:20:14 -0800261 // coord.x = max(lowBound, coord.x);
egdaniel4ca2e602015-11-18 08:01:26 -0800262 fragBuilder->codeAppendf("\t\tcoord.%s = max(%s.x, coord.%s);", dir, range, dir);
cwallez80a61df2015-01-26 12:20:14 -0800263 }
robertphillipsbf536af2016-02-04 06:11:53 -0800264 fragBuilder->codeAppendf("\t\tfor (int i = 0; i < %d; i++) {\n", width);
egdaniel4ca2e602015-11-18 08:01:26 -0800265 fragBuilder->codeAppendf("\t\t\t%s = %s(%s, ", args.fOutputColor, func, args.fOutputColor);
cdalton3f6f76f2016-04-11 12:18:09 -0700266 fragBuilder->appendTextureLookup(args.fTexSamplers[0], "coord");
egdaniel4ca2e602015-11-18 08:01:26 -0800267 fragBuilder->codeAppend(");\n");
cwallez80a61df2015-01-26 12:20:14 -0800268 // coord.x += pixelSize;
egdaniel4ca2e602015-11-18 08:01:26 -0800269 fragBuilder->codeAppendf("\t\t\tcoord.%s += %s;\n", dir, pixelSizeInc);
robertphillipsbf536af2016-02-04 06:11:53 -0800270 if (me.useRange()) {
cwallez80a61df2015-01-26 12:20:14 -0800271 // coord.x = min(highBound, coord.x);
egdaniel4ca2e602015-11-18 08:01:26 -0800272 fragBuilder->codeAppendf("\t\t\tcoord.%s = min(highBound, coord.%s);", dir, dir);
cwallez80a61df2015-01-26 12:20:14 -0800273 }
egdaniel4ca2e602015-11-18 08:01:26 -0800274 fragBuilder->codeAppend("\t\t}\n");
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000275 SkString modulate;
wangyix7c157a92015-07-22 15:08:53 -0700276 GrGLSLMulVarBy4f(&modulate, args.fOutputColor, args.fInputColor);
egdaniel4ca2e602015-11-18 08:01:26 -0800277 fragBuilder->codeAppend(modulate.c_str());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000278}
279
joshualittb0a8a372014-09-23 09:50:21 -0700280void GrGLMorphologyEffect::GenKey(const GrProcessor& proc,
jvanverthcfc18862015-04-28 08:48:20 -0700281 const GrGLSLCaps&, GrProcessorKeyBuilder* b) {
joshualittb0a8a372014-09-23 09:50:21 -0700282 const GrMorphologyEffect& m = proc.cast<GrMorphologyEffect>();
bsalomon63e99f72014-07-21 08:03:14 -0700283 uint32_t key = static_cast<uint32_t>(m.radius());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000284 key |= (m.type() << 8);
cwallez80a61df2015-01-26 12:20:14 -0800285 key |= (m.direction() << 9);
robertphillipsbf536af2016-02-04 06:11:53 -0800286 if (m.useRange()) {
287 key |= 1 << 10;
288 }
bsalomon63e99f72014-07-21 08:03:14 -0700289 b->add32(key);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000290}
291
egdaniel018fb622015-10-28 07:26:40 -0700292void GrGLMorphologyEffect::onSetData(const GrGLSLProgramDataManager& pdman,
robertphillipsbf536af2016-02-04 06:11:53 -0800293 const GrProcessor& proc) {
cwallez80a61df2015-01-26 12:20:14 -0800294 const GrMorphologyEffect& m = proc.cast<GrMorphologyEffect>();
295 GrTexture& texture = *m.texture(0);
cwallez80a61df2015-01-26 12:20:14 -0800296
297 float pixelSize = 0.0f;
robertphillipsbf536af2016-02-04 06:11:53 -0800298 switch (m.direction()) {
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000299 case Gr1DKernelEffect::kX_Direction:
cwallez80a61df2015-01-26 12:20:14 -0800300 pixelSize = 1.0f / texture.width();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000301 break;
302 case Gr1DKernelEffect::kY_Direction:
cwallez80a61df2015-01-26 12:20:14 -0800303 pixelSize = 1.0f / texture.height();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000304 break;
305 default:
commit-bot@chromium.org88cb22b2014-04-30 14:17:00 +0000306 SkFAIL("Unknown filter direction.");
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000307 }
cwallez80a61df2015-01-26 12:20:14 -0800308 pdman.set1f(fPixelSizeUni, pixelSize);
309
robertphillipsbf536af2016-02-04 06:11:53 -0800310 if (m.useRange()) {
cwallez80a61df2015-01-26 12:20:14 -0800311 const float* range = m.range();
robertphillipsbf536af2016-02-04 06:11:53 -0800312 if (m.direction() && texture.origin() == kBottomLeft_GrSurfaceOrigin) {
cwallez80a61df2015-01-26 12:20:14 -0800313 pdman.set2f(fRangeUni, 1.0f - range[1], 1.0f - range[0]);
314 } else {
315 pdman.set2f(fRangeUni, range[0], range[1]);
316 }
317 }
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000318}
319
320///////////////////////////////////////////////////////////////////////////////
321
bsalomon4a339522015-10-06 08:40:50 -0700322GrMorphologyEffect::GrMorphologyEffect(GrTexture* texture,
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000323 Direction direction,
324 int radius,
325 MorphologyType type)
bsalomon4a339522015-10-06 08:40:50 -0700326 : INHERITED(texture, direction, radius)
robertphillipsf299e712016-03-25 04:49:22 -0700327 , fType(type)
328 , fUseRange(false) {
joshualitteb2a6762014-12-04 11:35:33 -0800329 this->initClassID<GrMorphologyEffect>();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000330}
331
bsalomon4a339522015-10-06 08:40:50 -0700332GrMorphologyEffect::GrMorphologyEffect(GrTexture* texture,
cwallez80a61df2015-01-26 12:20:14 -0800333 Direction direction,
334 int radius,
335 MorphologyType type,
336 float range[2])
bsalomon4a339522015-10-06 08:40:50 -0700337 : INHERITED(texture, direction, radius)
robertphillipsf299e712016-03-25 04:49:22 -0700338 , fType(type)
339 , fUseRange(true) {
cwallez80a61df2015-01-26 12:20:14 -0800340 this->initClassID<GrMorphologyEffect>();
341 fRange[0] = range[0];
342 fRange[1] = range[1];
343}
344
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000345GrMorphologyEffect::~GrMorphologyEffect() {
346}
347
egdaniel57d3b032015-11-13 11:57:27 -0800348void GrMorphologyEffect::onGetGLSLProcessorKey(const GrGLSLCaps& caps,
349 GrProcessorKeyBuilder* b) const {
joshualitteb2a6762014-12-04 11:35:33 -0800350 GrGLMorphologyEffect::GenKey(*this, caps, b);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000351}
352
egdaniel57d3b032015-11-13 11:57:27 -0800353GrGLSLFragmentProcessor* GrMorphologyEffect::onCreateGLSLInstance() const {
robertphillipsbf536af2016-02-04 06:11:53 -0800354 return new GrGLMorphologyEffect;
joshualitteb2a6762014-12-04 11:35:33 -0800355}
bsalomon0e08fc12014-10-15 08:19:04 -0700356bool GrMorphologyEffect::onIsEqual(const GrFragmentProcessor& sBase) const {
joshualitt49586be2014-09-16 08:21:41 -0700357 const GrMorphologyEffect& s = sBase.cast<GrMorphologyEffect>();
bsalomon420d7e92014-10-16 09:18:09 -0700358 return (this->radius() == s.radius() &&
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000359 this->direction() == s.direction() &&
cwallez80a61df2015-01-26 12:20:14 -0800360 this->useRange() == s.useRange() &&
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000361 this->type() == s.type());
362}
363
egdaniel605dd0f2014-11-12 08:35:25 -0800364void GrMorphologyEffect::onComputeInvariantOutput(GrInvariantOutput* inout) const {
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000365 // This is valid because the color components of the result of the kernel all come
366 // exactly from existing values in the source texture.
egdaniel1a8ecdf2014-10-03 06:24:12 -0700367 this->updateInvariantOutputForModulation(inout);
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000368}
369
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000370///////////////////////////////////////////////////////////////////////////////
371
joshualittb0a8a372014-09-23 09:50:21 -0700372GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrMorphologyEffect);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000373
bungeman06ca8ec2016-06-09 08:01:03 -0700374sk_sp<GrFragmentProcessor> GrMorphologyEffect::TestCreate(GrProcessorTestData* d) {
joshualitt0067ff52015-07-08 14:26:19 -0700375 int texIdx = d->fRandom->nextBool() ? GrProcessorUnitTest::kSkiaPMTextureIdx :
bungeman06ca8ec2016-06-09 08:01:03 -0700376 GrProcessorUnitTest::kAlphaTextureIdx;
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);
380 MorphologyType type = d->fRandom->nextBool() ? GrMorphologyEffect::kErode_MorphologyType :
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000381 GrMorphologyEffect::kDilate_MorphologyType;
382
bungeman06ca8ec2016-06-09 08:01:03 -0700383 return GrMorphologyEffect::Make(d->fTextures[texIdx], dir, radius, type);
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000384}
385
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000386
robertphillipsf299e712016-03-25 04:49:22 -0700387static void apply_morphology_rect(GrDrawContext* drawContext,
388 const GrClip& clip,
389 GrTexture* texture,
390 const SkIRect& srcRect,
391 const SkIRect& dstRect,
392 int radius,
393 GrMorphologyEffect::MorphologyType morphType,
394 float bounds[2],
395 Gr1DKernelEffect::Direction direction) {
cwallez80a61df2015-01-26 12:20:14 -0800396 GrPaint paint;
brianosman898235c2016-04-06 07:38:23 -0700397 // SRGBTODO: AllowSRGBInputs?
bungeman06ca8ec2016-06-09 08:01:03 -0700398 paint.addColorFragmentProcessor(GrMorphologyEffect::Make(texture,
399 direction,
400 radius,
401 morphType,
402 bounds));
egdanielc4b72722015-11-23 13:20:41 -0800403 paint.setPorterDuffXPFactory(SkXfermode::kSrc_Mode);
bsalomona2e69fc2015-11-05 10:41:43 -0800404 drawContext->fillRectToRect(clip, paint, SkMatrix::I(), SkRect::Make(dstRect),
robertphillipsea461502015-05-26 11:38:03 -0700405 SkRect::Make(srcRect));
cwallez80a61df2015-01-26 12:20:14 -0800406}
407
robertphillipsf299e712016-03-25 04:49:22 -0700408static void apply_morphology_rect_no_bounds(GrDrawContext* drawContext,
409 const GrClip& clip,
410 GrTexture* texture,
411 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;
brianosman898235c2016-04-06 07:38:23 -0700417 // SRGBTODO: AllowSRGBInputs?
bungeman06ca8ec2016-06-09 08:01:03 -0700418 paint.addColorFragmentProcessor(GrMorphologyEffect::Make(texture, direction, radius,
419 morphType));
egdanielc4b72722015-11-23 13:20:41 -0800420 paint.setPorterDuffXPFactory(SkXfermode::kSrc_Mode);
bsalomona2e69fc2015-11-05 10:41:43 -0800421 drawContext->fillRectToRect(clip, paint, SkMatrix::I(), SkRect::Make(dstRect),
422 SkRect::Make(srcRect));
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000423}
424
robertphillipsf299e712016-03-25 04:49:22 -0700425static void apply_morphology_pass(GrDrawContext* drawContext,
426 const GrClip& clip,
427 GrTexture* texture,
428 const SkIRect& srcRect,
429 const SkIRect& dstRect,
430 int radius,
431 GrMorphologyEffect::MorphologyType morphType,
432 Gr1DKernelEffect::Direction direction) {
cwallez80a61df2015-01-26 12:20:14 -0800433 float bounds[2] = { 0.0f, 1.0f };
434 SkIRect lowerSrcRect = srcRect, lowerDstRect = dstRect;
435 SkIRect middleSrcRect = srcRect, middleDstRect = dstRect;
436 SkIRect upperSrcRect = srcRect, upperDstRect = dstRect;
437 if (direction == Gr1DKernelEffect::kX_Direction) {
438 bounds[0] = (SkIntToScalar(srcRect.left()) + 0.5f) / texture->width();
439 bounds[1] = (SkIntToScalar(srcRect.right()) - 0.5f) / texture->width();
440 lowerSrcRect.fRight = srcRect.left() + radius;
441 lowerDstRect.fRight = dstRect.left() + radius;
442 upperSrcRect.fLeft = srcRect.right() - radius;
443 upperDstRect.fLeft = dstRect.right() - radius;
444 middleSrcRect.inset(radius, 0);
445 middleDstRect.inset(radius, 0);
446 } else {
447 bounds[0] = (SkIntToScalar(srcRect.top()) + 0.5f) / texture->height();
448 bounds[1] = (SkIntToScalar(srcRect.bottom()) - 0.5f) / texture->height();
449 lowerSrcRect.fBottom = srcRect.top() + radius;
450 lowerDstRect.fBottom = dstRect.top() + radius;
451 upperSrcRect.fTop = srcRect.bottom() - radius;
452 upperDstRect.fTop = dstRect.bottom() - radius;
453 middleSrcRect.inset(0, radius);
454 middleDstRect.inset(0, radius);
455 }
456 if (middleSrcRect.fLeft - middleSrcRect.fRight >= 0) {
457 // radius covers srcRect; use bounds over entire draw
robertphillips2e1e51f2015-10-15 08:01:48 -0700458 apply_morphology_rect(drawContext, clip, texture, srcRect, dstRect, radius,
cwallez80a61df2015-01-26 12:20:14 -0800459 morphType, bounds, direction);
460 } else {
461 // Draw upper and lower margins with bounds; middle without.
robertphillips2e1e51f2015-10-15 08:01:48 -0700462 apply_morphology_rect(drawContext, clip, texture, lowerSrcRect, lowerDstRect, radius,
cwallez80a61df2015-01-26 12:20:14 -0800463 morphType, bounds, direction);
robertphillips2e1e51f2015-10-15 08:01:48 -0700464 apply_morphology_rect(drawContext, clip, texture, upperSrcRect, upperDstRect, radius,
cwallez80a61df2015-01-26 12:20:14 -0800465 morphType, bounds, direction);
robertphillips2e1e51f2015-10-15 08:01:48 -0700466 apply_morphology_rect_no_bounds(drawContext, clip, texture, middleSrcRect, middleDstRect,
joshualitt570d2f82015-02-25 13:19:48 -0800467 radius, morphType, direction);
cwallez80a61df2015-01-26 12:20:14 -0800468 }
469}
470
robertphillips64612512016-04-08 12:10:42 -0700471static sk_sp<SkSpecialImage> apply_morphology(GrContext* context,
472 SkSpecialImage* input,
robertphillipsf299e712016-03-25 04:49:22 -0700473 const SkIRect& rect,
474 GrMorphologyEffect::MorphologyType morphType,
475 SkISize radius) {
robertphillipsc91fd342016-04-25 12:32:54 -0700476 sk_sp<GrTexture> srcTexture(input->asTextureRef(context));
bsalomon49f085d2014-09-05 13:34:00 -0700477 SkASSERT(srcTexture);
brianosmandfe4f2e2016-07-21 13:28:36 -0700478 sk_sp<SkColorSpace> colorSpace = sk_ref_sp(input->getColorSpace());
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000479
joshualitt570d2f82015-02-25 13:19:48 -0800480 // setup new clip
cdalton846c0512016-05-13 10:25:00 -0700481 const GrFixedClip clip(SkIRect::MakeWH(srcTexture->width(), srcTexture->height()));
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000482
robertphillipsd4c741e2016-04-28 09:55:15 -0700483 const SkIRect dstRect = SkIRect::MakeWH(rect.width(), rect.height());
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000484 SkIRect srcRect = rect;
bsalomon@google.com3cbaa2d2012-10-12 14:51:52 +0000485
robertphillipsf299e712016-03-25 04:49:22 -0700486 SkASSERT(radius.width() > 0 || radius.height() > 0);
487
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000488 if (radius.fWidth > 0) {
robertphillips6738c702016-07-27 12:13:51 -0700489 sk_sp<GrDrawContext> dstDrawContext(context->makeDrawContext(SkBackingFit::kApprox,
490 rect.width(), rect.height(),
491 kSkia8888_GrPixelConfig,
492 colorSpace));
robertphillipsff0ca5e2015-07-22 11:54:44 -0700493 if (!dstDrawContext) {
robertphillipsf299e712016-03-25 04:49:22 -0700494 return nullptr;
robertphillipsff0ca5e2015-07-22 11:54:44 -0700495 }
496
robertphillips6c7e3252016-04-27 10:47:51 -0700497 apply_morphology_pass(dstDrawContext.get(), clip, srcTexture.get(),
joshualitt570d2f82015-02-25 13:19:48 -0800498 srcRect, dstRect, radius.fWidth, morphType,
499 Gr1DKernelEffect::kX_Direction);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000500 SkIRect clearRect = SkIRect::MakeXYWH(dstRect.fLeft, dstRect.fBottom,
501 dstRect.width(), radius.fHeight);
robertphillipsd4c741e2016-04-28 09:55:15 -0700502 GrColor clearColor = GrMorphologyEffect::kErode_MorphologyType == morphType
503 ? SK_ColorWHITE
504 : SK_ColorTRANSPARENT;
robertphillips2e1e51f2015-10-15 08:01:48 -0700505 dstDrawContext->clear(&clearRect, clearColor, false);
robertphillipsff0ca5e2015-07-22 11:54:44 -0700506
robertphillipsd4c741e2016-04-28 09:55:15 -0700507 srcTexture = dstDrawContext->asTexture();
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000508 srcRect = dstRect;
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000509 }
510 if (radius.fHeight > 0) {
robertphillips6738c702016-07-27 12:13:51 -0700511 sk_sp<GrDrawContext> dstDrawContext(context->makeDrawContext(SkBackingFit::kApprox,
512 rect.width(), rect.height(),
513 kSkia8888_GrPixelConfig,
514 colorSpace));
robertphillipsff0ca5e2015-07-22 11:54:44 -0700515 if (!dstDrawContext) {
robertphillipsf299e712016-03-25 04:49:22 -0700516 return nullptr;
robertphillipsff0ca5e2015-07-22 11:54:44 -0700517 }
518
robertphillips6c7e3252016-04-27 10:47:51 -0700519 apply_morphology_pass(dstDrawContext.get(), clip, srcTexture.get(),
joshualitt570d2f82015-02-25 13:19:48 -0800520 srcRect, dstRect, radius.fHeight, morphType,
521 Gr1DKernelEffect::kY_Direction);
robertphillipsff0ca5e2015-07-22 11:54:44 -0700522
robertphillipsd4c741e2016-04-28 09:55:15 -0700523 srcTexture = dstDrawContext->asTexture();
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000524 }
robertphillipsf299e712016-03-25 04:49:22 -0700525
robertphillips3e302272016-04-20 11:48:36 -0700526 return SkSpecialImage::MakeFromGpu(SkIRect::MakeWH(rect.width(), rect.height()),
robertphillipsf299e712016-03-25 04:49:22 -0700527 kNeedNewImageUniqueID_SpecialImage,
brianosmandfe4f2e2016-07-21 13:28:36 -0700528 std::move(srcTexture), std::move(colorSpace),
brianosmanafbf71d2016-07-21 07:15:37 -0700529 &input->props());
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000530}
robertphillipsf299e712016-03-25 04:49:22 -0700531#endif
senorblanco@chromium.org84207c42012-08-22 20:51:19 +0000532
robertphillips40d8d622016-03-30 08:09:56 -0700533sk_sp<SkSpecialImage> SkMorphologyImageFilter::onFilterImage(SkSpecialImage* source,
534 const Context& ctx,
535 SkIPoint* offset) const {
robertphillipsf299e712016-03-25 04:49:22 -0700536 SkIPoint inputOffset = SkIPoint::Make(0, 0);
537 sk_sp<SkSpecialImage> input(this->filterInput(0, source, ctx, &inputOffset));
538 if (!input) {
539 return nullptr;
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000540 }
robertphillipsf299e712016-03-25 04:49:22 -0700541
senorblanco@chromium.orgc2594f42013-01-30 19:08:47 +0000542 SkIRect bounds;
robertphillipsf299e712016-03-25 04:49:22 -0700543 input = this->applyCropRect(this->mapContext(ctx), input.get(), &inputOffset, &bounds);
544 if (!input) {
545 return nullptr;
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000546 }
robertphillipsf299e712016-03-25 04:49:22 -0700547
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000548 SkVector radius = SkVector::Make(SkIntToScalar(this->radius().width()),
549 SkIntToScalar(this->radius().height()));
senorblanco@chromium.org4cb543d2014-03-14 15:44:01 +0000550 ctx.ctm().mapVectors(&radius, 1);
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000551 int width = SkScalarFloorToInt(radius.fX);
552 int height = SkScalarFloorToInt(radius.fY);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000553
554 if (width < 0 || height < 0) {
robertphillipsf299e712016-03-25 04:49:22 -0700555 return nullptr;
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000556 }
557
senorblanco@chromium.org2bfe36b2014-01-20 19:58:28 +0000558 SkIRect srcBounds = bounds;
robertphillipsf299e712016-03-25 04:49:22 -0700559 srcBounds.offset(-inputOffset);
560
561 if (0 == width && 0 == height) {
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000562 offset->fX = bounds.left();
563 offset->fY = bounds.top();
robertphillipsf299e712016-03-25 04:49:22 -0700564 return input->makeSubset(srcBounds);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000565 }
566
robertphillipsf299e712016-03-25 04:49:22 -0700567#if SK_SUPPORT_GPU
robertphillips64612512016-04-08 12:10:42 -0700568 if (source->isTextureBacked()) {
569 GrContext* context = source->getContext();
570
robertphillips40d8d622016-03-30 08:09:56 -0700571 auto type = (kDilate_Op == this->op()) ? GrMorphologyEffect::kDilate_MorphologyType
572 : GrMorphologyEffect::kErode_MorphologyType;
robertphillips64612512016-04-08 12:10:42 -0700573 sk_sp<SkSpecialImage> result(apply_morphology(context, input.get(), srcBounds, type,
robertphillipsf299e712016-03-25 04:49:22 -0700574 SkISize::Make(width, height)));
575 if (result) {
576 offset->fX = bounds.left();
577 offset->fY = bounds.top();
578 }
579 return result;
580 }
581#endif
582
robertphillips64612512016-04-08 12:10:42 -0700583 SkBitmap inputBM;
robertphillipsf299e712016-03-25 04:49:22 -0700584
robertphillips64612512016-04-08 12:10:42 -0700585 if (!input->getROPixels(&inputBM)) {
robertphillipsf299e712016-03-25 04:49:22 -0700586 return nullptr;
587 }
588
robertphillips64612512016-04-08 12:10:42 -0700589 if (inputBM.colorType() != kN32_SkColorType) {
robertphillipsf299e712016-03-25 04:49:22 -0700590 return nullptr;
591 }
592
593 SkImageInfo info = SkImageInfo::Make(bounds.width(), bounds.height(),
robertphillips64612512016-04-08 12:10:42 -0700594 inputBM.colorType(), inputBM.alphaType());
robertphillipsf299e712016-03-25 04:49:22 -0700595
596 SkBitmap dst;
597 if (!dst.tryAllocPixels(info)) {
598 return nullptr;
599 }
600
robertphillips64612512016-04-08 12:10:42 -0700601 SkAutoLockPixels inputLock(inputBM), dstLock(dst);
robertphillipsf299e712016-03-25 04:49:22 -0700602
603 SkMorphologyImageFilter::Proc procX, procY;
604
robertphillips40d8d622016-03-30 08:09:56 -0700605 if (kDilate_Op == this->op()) {
robertphillipsf299e712016-03-25 04:49:22 -0700606 procX = SkOpts::dilate_x;
607 procY = SkOpts::dilate_y;
608 } else {
609 procX = SkOpts::erode_x;
610 procY = SkOpts::erode_y;
611 }
612
613 if (width > 0 && height > 0) {
614 SkBitmap tmp;
615 if (!tmp.tryAllocPixels(info)) {
616 return nullptr;
617 }
618
619 SkAutoLockPixels tmpLock(tmp);
620
robertphillips64612512016-04-08 12:10:42 -0700621 call_proc_X(procX, inputBM, &tmp, width, srcBounds);
robertphillipsf299e712016-03-25 04:49:22 -0700622 SkIRect tmpBounds = SkIRect::MakeWH(srcBounds.width(), srcBounds.height());
623 call_proc_Y(procY,
624 tmp.getAddr32(tmpBounds.left(), tmpBounds.top()), tmp.rowBytesAsPixels(),
625 &dst, height, tmpBounds);
626 } else if (width > 0) {
robertphillips64612512016-04-08 12:10:42 -0700627 call_proc_X(procX, inputBM, &dst, width, srcBounds);
robertphillipsf299e712016-03-25 04:49:22 -0700628 } else if (height > 0) {
629 call_proc_Y(procY,
robertphillips64612512016-04-08 12:10:42 -0700630 inputBM.getAddr32(srcBounds.left(), srcBounds.top()),
631 inputBM.rowBytesAsPixels(),
robertphillipsf299e712016-03-25 04:49:22 -0700632 &dst, height, srcBounds);
senorblanco@chromium.org8fcad982013-09-17 13:41:43 +0000633 }
senorblanco@chromium.org6776b822014-01-03 21:48:22 +0000634 offset->fX = bounds.left();
635 offset->fY = bounds.top();
robertphillipsf299e712016-03-25 04:49:22 -0700636
robertphillips3e302272016-04-20 11:48:36 -0700637 return SkSpecialImage::MakeFromRaster(SkIRect::MakeWH(bounds.width(), bounds.height()),
brianosman898235c2016-04-06 07:38:23 -0700638 dst, &source->props());
senorblanco@chromium.org05054f12012-03-02 21:05:45 +0000639}