1 /* 2 * Copyright 2012 The Android Open Source Project 3 * 4 * Use of this source code is governed by a BSD-style license that can be 5 * found in the LICENSE file. 6 */ 7 8 #include "SkMatrixConvolutionImageFilter.h" 9 #include "SkBitmap.h" 10 #include "SkColorPriv.h" 11 #include "SkFlattenableBuffers.h" 12 #include "SkRect.h" 13 #include "SkUnPreMultiply.h" 14 15 #if SK_SUPPORT_GPU 16 #include "gl/GrGLEffect.h" 17 #include "gl/GrGLEffectMatrix.h" 18 #include "effects/GrSingleTextureEffect.h" 19 #include "GrTBackendEffectFactory.h" 20 #include "GrTexture.h" 21 #include "SkMatrix.h" 22 23 #endif 24 25 SkMatrixConvolutionImageFilter::SkMatrixConvolutionImageFilter(const SkISize& kernelSize, const SkScalar* kernel, SkScalar gain, SkScalar bias, const SkIPoint& target, TileMode tileMode, bool convolveAlpha, SkImageFilter* input) 26 : INHERITED(input), 27 fKernelSize(kernelSize), 28 fGain(gain), 29 fBias(bias), 30 fTarget(target), 31 fTileMode(tileMode), 32 fConvolveAlpha(convolveAlpha) { 33 uint32_t size = fKernelSize.fWidth * fKernelSize.fHeight; 34 fKernel = SkNEW_ARRAY(SkScalar, size); 35 memcpy(fKernel, kernel, size * sizeof(SkScalar)); 36 SkASSERT(kernelSize.fWidth >= 1 && kernelSize.fHeight >= 1); 37 SkASSERT(target.fX >= 0 && target.fX < kernelSize.fWidth); 38 SkASSERT(target.fY >= 0 && target.fY < kernelSize.fHeight); 39 } 40 41 SkMatrixConvolutionImageFilter::SkMatrixConvolutionImageFilter(SkFlattenableReadBuffer& buffer) : INHERITED(buffer) { 42 fKernelSize.fWidth = buffer.readInt(); 43 fKernelSize.fHeight = buffer.readInt(); 44 uint32_t size = fKernelSize.fWidth * fKernelSize.fHeight; 45 fKernel = SkNEW_ARRAY(SkScalar, size); 46 SkDEBUGCODE(uint32_t readSize = )buffer.readScalarArray(fKernel); 47 SkASSERT(readSize == size); 48 fGain = buffer.readScalar(); 49 fBias = buffer.readScalar(); 50 fTarget.fX = buffer.readInt(); 51 fTarget.fY = buffer.readInt(); 52 fTileMode = (TileMode) buffer.readInt(); 53 fConvolveAlpha = buffer.readBool(); 54 } 55 56 void SkMatrixConvolutionImageFilter::flatten(SkFlattenableWriteBuffer& buffer) const { 57 this->INHERITED::flatten(buffer); 58 buffer.writeInt(fKernelSize.fWidth); 59 buffer.writeInt(fKernelSize.fHeight); 60 buffer.writeScalarArray(fKernel, fKernelSize.fWidth * fKernelSize.fHeight); 61 buffer.writeScalar(fGain); 62 buffer.writeScalar(fBias); 63 buffer.writeInt(fTarget.fX); 64 buffer.writeInt(fTarget.fY); 65 buffer.writeInt((int) fTileMode); 66 buffer.writeBool(fConvolveAlpha); 67 } 68 69 SkMatrixConvolutionImageFilter::~SkMatrixConvolutionImageFilter() { 70 delete[] fKernel; 71 } 72 73 class UncheckedPixelFetcher { 74 public: 75 static inline SkPMColor fetch(const SkBitmap& src, int x, int y) { 76 return *src.getAddr32(x, y); 77 } 78 }; 79 80 class ClampPixelFetcher { 81 public: 82 static inline SkPMColor fetch(const SkBitmap& src, int x, int y) { 83 x = SkClampMax(x, src.width() - 1); 84 y = SkClampMax(y, src.height() - 1); 85 return *src.getAddr32(x, y); 86 } 87 }; 88 89 class RepeatPixelFetcher { 90 public: 91 static inline SkPMColor fetch(const SkBitmap& src, int x, int y) { 92 x %= src.width(); 93 y %= src.height(); 94 if (x < 0) { 95 x += src.width(); 96 } 97 if (y < 0) { 98 y += src.height(); 99 } 100 return *src.getAddr32(x, y); 101 } 102 }; 103 104 class ClampToBlackPixelFetcher { 105 public: 106 static inline SkPMColor fetch(const SkBitmap& src, int x, int y) { 107 if (x < 0 || x >= src.width() || y < 0 || y >= src.height()) { 108 return 0; 109 } else { 110 return *src.getAddr32(x, y); 111 } 112 } 113 }; 114 115 template<class PixelFetcher, bool convolveAlpha> 116 void SkMatrixConvolutionImageFilter::filterPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) { 117 for (int y = rect.fTop; y < rect.fBottom; ++y) { 118 SkPMColor* dptr = result->getAddr32(rect.fLeft, y); 119 for (int x = rect.fLeft; x < rect.fRight; ++x) { 120 SkScalar sumA = 0, sumR = 0, sumG = 0, sumB = 0; 121 for (int cy = 0; cy < fKernelSize.fHeight; cy++) { 122 for (int cx = 0; cx < fKernelSize.fWidth; cx++) { 123 SkPMColor s = PixelFetcher::fetch(src, x + cx - fTarget.fX, y + cy - fTarget.fY); 124 SkScalar k = fKernel[cy * fKernelSize.fWidth + cx]; 125 if (convolveAlpha) { 126 sumA += SkScalarMul(SkIntToScalar(SkGetPackedA32(s)), k); 127 } 128 sumR += SkScalarMul(SkIntToScalar(SkGetPackedR32(s)), k); 129 sumG += SkScalarMul(SkIntToScalar(SkGetPackedG32(s)), k); 130 sumB += SkScalarMul(SkIntToScalar(SkGetPackedB32(s)), k); 131 } 132 } 133 int a = convolveAlpha 134 ? SkClampMax(SkScalarFloorToInt(SkScalarMul(sumA, fGain) + fBias), 255) 135 : 255; 136 int r = SkClampMax(SkScalarFloorToInt(SkScalarMul(sumR, fGain) + fBias), a); 137 int g = SkClampMax(SkScalarFloorToInt(SkScalarMul(sumG, fGain) + fBias), a); 138 int b = SkClampMax(SkScalarFloorToInt(SkScalarMul(sumB, fGain) + fBias), a); 139 if (!convolveAlpha) { 140 a = SkGetPackedA32(PixelFetcher::fetch(src, x, y)); 141 *dptr++ = SkPreMultiplyARGB(a, r, g, b); 142 } else { 143 *dptr++ = SkPackARGB32(a, r, g, b); 144 } 145 } 146 } 147 } 148 149 template<class PixelFetcher> 150 void SkMatrixConvolutionImageFilter::filterPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) { 151 if (fConvolveAlpha) { 152 filterPixels<PixelFetcher, true>(src, result, rect); 153 } else { 154 filterPixels<PixelFetcher, false>(src, result, rect); 155 } 156 } 157 158 void SkMatrixConvolutionImageFilter::filterInteriorPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) { 159 filterPixels<UncheckedPixelFetcher>(src, result, rect); 160 } 161 162 void SkMatrixConvolutionImageFilter::filterBorderPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) { 163 switch (fTileMode) { 164 case kClamp_TileMode: 165 filterPixels<ClampPixelFetcher>(src, result, rect); 166 break; 167 case kRepeat_TileMode: 168 filterPixels<RepeatPixelFetcher>(src, result, rect); 169 break; 170 case kClampToBlack_TileMode: 171 filterPixels<ClampToBlackPixelFetcher>(src, result, rect); 172 break; 173 } 174 } 175 176 // FIXME: This should be refactored to SkImageFilterUtils for 177 // use by other filters. For now, we assume the input is always 178 // premultiplied and unpremultiply it 179 static SkBitmap unpremultiplyBitmap(const SkBitmap& src) 180 { 181 SkAutoLockPixels alp(src); 182 if (!src.getPixels()) { 183 return SkBitmap(); 184 } 185 SkBitmap result; 186 result.setConfig(src.config(), src.width(), src.height()); 187 result.allocPixels(); 188 if (!result.getPixels()) { 189 return SkBitmap(); 190 } 191 for (int y = 0; y < src.height(); ++y) { 192 const uint32_t* srcRow = src.getAddr32(0, y); 193 uint32_t* dstRow = result.getAddr32(0, y); 194 for (int x = 0; x < src.width(); ++x) { 195 dstRow[x] = SkUnPreMultiply::PMColorToColor(srcRow[x]); 196 } 197 } 198 return result; 199 } 200 201 bool SkMatrixConvolutionImageFilter::onFilterImage(Proxy* proxy, 202 const SkBitmap& source, 203 const SkMatrix& matrix, 204 SkBitmap* result, 205 SkIPoint* loc) { 206 SkBitmap src = source; 207 if (getInput(0) && !getInput(0)->filterImage(proxy, source, matrix, &src, loc)) { 208 return false; 209 } 210 211 if (src.config() != SkBitmap::kARGB_8888_Config) { 212 return false; 213 } 214 215 if (!fConvolveAlpha && !src.isOpaque()) { 216 src = unpremultiplyBitmap(src); 217 } 218 219 SkAutoLockPixels alp(src); 220 if (!src.getPixels()) { 221 return false; 222 } 223 224 result->setConfig(src.config(), src.width(), src.height()); 225 result->allocPixels(); 226 227 SkIRect interior = SkIRect::MakeXYWH(fTarget.fX, fTarget.fY, 228 src.width() - fKernelSize.fWidth + 1, 229 src.height() - fKernelSize.fHeight + 1); 230 SkIRect top = SkIRect::MakeWH(src.width(), fTarget.fY); 231 SkIRect bottom = SkIRect::MakeLTRB(0, interior.bottom(), 232 src.width(), src.height()); 233 SkIRect left = SkIRect::MakeXYWH(0, interior.top(), 234 fTarget.fX, interior.height()); 235 SkIRect right = SkIRect::MakeLTRB(interior.right(), interior.top(), 236 src.width(), interior.bottom()); 237 filterBorderPixels(src, result, top); 238 filterBorderPixels(src, result, left); 239 filterInteriorPixels(src, result, interior); 240 filterBorderPixels(src, result, right); 241 filterBorderPixels(src, result, bottom); 242 return true; 243 } 244 245 #if SK_SUPPORT_GPU 246 247 /////////////////////////////////////////////////////////////////////////////// 248 249 class GrGLMatrixConvolutionEffect; 250 251 class GrMatrixConvolutionEffect : public GrSingleTextureEffect { 252 public: 253 typedef SkMatrixConvolutionImageFilter::TileMode TileMode; 254 static GrEffectRef* Create(GrTexture* texture, 255 const SkISize& kernelSize, 256 const SkScalar* kernel, 257 SkScalar gain, 258 SkScalar bias, 259 const SkIPoint& target, 260 TileMode tileMode, 261 bool convolveAlpha) { 262 AutoEffectUnref effect(SkNEW_ARGS(GrMatrixConvolutionEffect, (texture, 263 kernelSize, 264 kernel, 265 gain, 266 bias, 267 target, 268 tileMode, 269 convolveAlpha))); 270 return CreateEffectRef(effect); 271 } 272 virtual ~GrMatrixConvolutionEffect(); 273 274 virtual void getConstantColorComponents(GrColor* color, 275 uint32_t* validFlags) const SK_OVERRIDE { 276 // TODO: Try to do better? 277 *validFlags = 0; 278 } 279 280 static const char* Name() { return "MatrixConvolution"; } 281 const SkISize& kernelSize() const { return fKernelSize; } 282 const float* target() const { return fTarget; } 283 const float* kernel() const { return fKernel; } 284 float gain() const { return fGain; } 285 float bias() const { return fBias; } 286 TileMode tileMode() const { return fTileMode; } 287 bool convolveAlpha() const { return fConvolveAlpha; } 288 289 typedef GrGLMatrixConvolutionEffect GLEffect; 290 291 virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE; 292 293 private: 294 GrMatrixConvolutionEffect(GrTexture*, 295 const SkISize& kernelSize, 296 const SkScalar* kernel, 297 SkScalar gain, 298 SkScalar bias, 299 const SkIPoint& target, 300 TileMode tileMode, 301 bool convolveAlpha); 302 303 virtual bool onIsEqual(const GrEffect&) const SK_OVERRIDE; 304 305 SkISize fKernelSize; 306 float *fKernel; 307 float fGain; 308 float fBias; 309 float fTarget[2]; 310 TileMode fTileMode; 311 bool fConvolveAlpha; 312 313 GR_DECLARE_EFFECT_TEST; 314 315 typedef GrSingleTextureEffect INHERITED; 316 }; 317 318 class GrGLMatrixConvolutionEffect : public GrGLEffect { 319 public: 320 GrGLMatrixConvolutionEffect(const GrBackendEffectFactory& factory, 321 const GrDrawEffect& effect); 322 virtual void emitCode(GrGLShaderBuilder*, 323 const GrDrawEffect&, 324 EffectKey, 325 const char* outputColor, 326 const char* inputColor, 327 const TextureSamplerArray&) SK_OVERRIDE; 328 329 static inline EffectKey GenKey(const GrDrawEffect&, const GrGLCaps&); 330 331 virtual void setData(const GrGLUniformManager&, const GrDrawEffect&) SK_OVERRIDE; 332 333 private: 334 typedef GrGLUniformManager::UniformHandle UniformHandle; 335 typedef SkMatrixConvolutionImageFilter::TileMode TileMode; 336 SkISize fKernelSize; 337 TileMode fTileMode; 338 bool fConvolveAlpha; 339 340 UniformHandle fKernelUni; 341 UniformHandle fImageIncrementUni; 342 UniformHandle fTargetUni; 343 UniformHandle fGainUni; 344 UniformHandle fBiasUni; 345 346 GrGLEffectMatrix fEffectMatrix; 347 348 typedef GrGLEffect INHERITED; 349 }; 350 351 GrGLMatrixConvolutionEffect::GrGLMatrixConvolutionEffect(const GrBackendEffectFactory& factory, 352 const GrDrawEffect& drawEffect) 353 : INHERITED(factory) 354 , fKernelUni(GrGLUniformManager::kInvalidUniformHandle) 355 , fImageIncrementUni(GrGLUniformManager::kInvalidUniformHandle) 356 , fTargetUni(GrGLUniformManager::kInvalidUniformHandle) 357 , fGainUni(GrGLUniformManager::kInvalidUniformHandle) 358 , fBiasUni(GrGLUniformManager::kInvalidUniformHandle) 359 , fEffectMatrix(drawEffect.castEffect<GrMatrixConvolutionEffect>().coordsType()) { 360 const GrMatrixConvolutionEffect& m = drawEffect.castEffect<GrMatrixConvolutionEffect>(); 361 fKernelSize = m.kernelSize(); 362 fTileMode = m.tileMode(); 363 fConvolveAlpha = m.convolveAlpha(); 364 } 365 366 static void appendTextureLookup(GrGLShaderBuilder* builder, 367 const GrGLShaderBuilder::TextureSampler& sampler, 368 const char* coord, 369 SkMatrixConvolutionImageFilter::TileMode tileMode) { 370 SkString clampedCoord; 371 switch (tileMode) { 372 case SkMatrixConvolutionImageFilter::kClamp_TileMode: 373 clampedCoord.printf("clamp(%s, 0.0, 1.0)", coord); 374 coord = clampedCoord.c_str(); 375 break; 376 case SkMatrixConvolutionImageFilter::kRepeat_TileMode: 377 clampedCoord.printf("fract(%s)", coord); 378 coord = clampedCoord.c_str(); 379 break; 380 case SkMatrixConvolutionImageFilter::kClampToBlack_TileMode: 381 builder->fsCodeAppendf("clamp(%s, 0.0, 1.0) != %s ? vec4(0, 0, 0, 0) : ", coord, coord); 382 break; 383 } 384 builder->appendTextureLookup(GrGLShaderBuilder::kFragment_ShaderType, sampler, coord); 385 } 386 387 void GrGLMatrixConvolutionEffect::emitCode(GrGLShaderBuilder* builder, 388 const GrDrawEffect&, 389 EffectKey key, 390 const char* outputColor, 391 const char* inputColor, 392 const TextureSamplerArray& samplers) { 393 const char* coords; 394 fEffectMatrix.emitCodeMakeFSCoords2D(builder, key, &coords); 395 fImageIncrementUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType, 396 kVec2f_GrSLType, "ImageIncrement"); 397 fKernelUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_ShaderType, 398 kFloat_GrSLType, "Kernel", fKernelSize.width() * fKernelSize.height()); 399 fTargetUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType, 400 kVec2f_GrSLType, "Target"); 401 fGainUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType, 402 kFloat_GrSLType, "Gain"); 403 fBiasUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType, 404 kFloat_GrSLType, "Bias"); 405 406 const char* target = builder->getUniformCStr(fTargetUni); 407 const char* imgInc = builder->getUniformCStr(fImageIncrementUni); 408 const char* kernel = builder->getUniformCStr(fKernelUni); 409 const char* gain = builder->getUniformCStr(fGainUni); 410 const char* bias = builder->getUniformCStr(fBiasUni); 411 int kWidth = fKernelSize.width(); 412 int kHeight = fKernelSize.height(); 413 414 builder->fsCodeAppend("\t\tvec4 sum = vec4(0, 0, 0, 0);\n"); 415 builder->fsCodeAppendf("\t\tvec2 coord = %s - %s * %s;\n", coords, target, imgInc); 416 builder->fsCodeAppendf("\t\tfor (int y = 0; y < %d; y++) {\n", kHeight); 417 builder->fsCodeAppendf("\t\t\tfor (int x = 0; x < %d; x++) {\n", kWidth); 418 builder->fsCodeAppendf("\t\t\t\tfloat k = %s[y * %d + x];\n", kernel, kWidth); 419 builder->fsCodeAppendf("\t\t\t\tvec2 coord2 = coord + vec2(x, y) * %s;\n", imgInc); 420 builder->fsCodeAppend("\t\t\t\tvec4 c = "); 421 appendTextureLookup(builder, samplers[0], "coord2", fTileMode); 422 builder->fsCodeAppend(";\n"); 423 if (!fConvolveAlpha) { 424 builder->fsCodeAppend("\t\t\t\tc.rgb /= c.a;\n"); 425 } 426 builder->fsCodeAppend("\t\t\t\tsum += c * k;\n"); 427 builder->fsCodeAppend("\t\t\t}\n"); 428 builder->fsCodeAppend("\t\t}\n"); 429 if (fConvolveAlpha) { 430 builder->fsCodeAppendf("\t\t%s = sum * %s + %s;\n", outputColor, gain, bias); 431 builder->fsCodeAppendf("\t\t%s.rgb = clamp(%s.rgb, 0.0, %s.a);\n", outputColor, outputColor, outputColor); 432 } else { 433 builder->fsCodeAppend("\t\tvec4 c = "); 434 appendTextureLookup(builder, samplers[0], coords, fTileMode); 435 builder->fsCodeAppend(";\n"); 436 builder->fsCodeAppendf("\t\t%s.a = c.a;\n", outputColor); 437 builder->fsCodeAppendf("\t\t%s.rgb = sum.rgb * %s + %s;\n", outputColor, gain, bias); 438 builder->fsCodeAppendf("\t\t%s.rgb *= %s.a;\n", outputColor, outputColor); 439 } 440 } 441 442 namespace { 443 444 int encodeXY(int x, int y) { 445 SkASSERT(x >= 1 && y >= 1 && x * y <= 32); 446 if (y < x) 447 return 0x40 | encodeXY(y, x); 448 else 449 return (0x40 >> x) | (y - x); 450 } 451 452 }; 453 454 GrGLEffect::EffectKey GrGLMatrixConvolutionEffect::GenKey(const GrDrawEffect& drawEffect, 455 const GrGLCaps&) { 456 const GrMatrixConvolutionEffect& m = drawEffect.castEffect<GrMatrixConvolutionEffect>(); 457 EffectKey key = encodeXY(m.kernelSize().width(), m.kernelSize().height()); 458 key |= m.tileMode() << 7; 459 key |= m.convolveAlpha() ? 1 << 9 : 0; 460 key <<= GrGLEffectMatrix::kKeyBits; 461 EffectKey matrixKey = GrGLEffectMatrix::GenKey(m.getMatrix(), 462 drawEffect, 463 m.coordsType(), 464 m.texture(0)); 465 return key | matrixKey; 466 } 467 468 void GrGLMatrixConvolutionEffect::setData(const GrGLUniformManager& uman, 469 const GrDrawEffect& drawEffect) { 470 const GrMatrixConvolutionEffect& conv = drawEffect.castEffect<GrMatrixConvolutionEffect>(); 471 GrTexture& texture = *conv.texture(0); 472 // the code we generated was for a specific kernel size 473 GrAssert(conv.kernelSize() == fKernelSize); 474 GrAssert(conv.tileMode() == fTileMode); 475 float imageIncrement[2]; 476 float ySign = texture.origin() == kTopLeft_GrSurfaceOrigin ? 1.0f : -1.0f; 477 imageIncrement[0] = 1.0f / texture.width(); 478 imageIncrement[1] = ySign / texture.height(); 479 uman.set2fv(fImageIncrementUni, 0, 1, imageIncrement); 480 uman.set2fv(fTargetUni, 0, 1, conv.target()); 481 uman.set1fv(fKernelUni, 0, fKernelSize.width() * fKernelSize.height(), conv.kernel()); 482 uman.set1f(fGainUni, conv.gain()); 483 uman.set1f(fBiasUni, conv.bias()); 484 fEffectMatrix.setData(uman, 485 conv.getMatrix(), 486 drawEffect, 487 conv.texture(0)); 488 } 489 490 GrMatrixConvolutionEffect::GrMatrixConvolutionEffect(GrTexture* texture, 491 const SkISize& kernelSize, 492 const SkScalar* kernel, 493 SkScalar gain, 494 SkScalar bias, 495 const SkIPoint& target, 496 TileMode tileMode, 497 bool convolveAlpha) 498 : INHERITED(texture, MakeDivByTextureWHMatrix(texture)), 499 fKernelSize(kernelSize), 500 fGain(SkScalarToFloat(gain)), 501 fBias(SkScalarToFloat(bias) / 255.0f), 502 fTileMode(tileMode), 503 fConvolveAlpha(convolveAlpha) { 504 fKernel = new float[kernelSize.width() * kernelSize.height()]; 505 for (int i = 0; i < kernelSize.width() * kernelSize.height(); i++) { 506 fKernel[i] = SkScalarToFloat(kernel[i]); 507 } 508 fTarget[0] = static_cast<float>(target.x()); 509 fTarget[1] = static_cast<float>(target.y()); 510 } 511 512 GrMatrixConvolutionEffect::~GrMatrixConvolutionEffect() { 513 delete[] fKernel; 514 } 515 516 const GrBackendEffectFactory& GrMatrixConvolutionEffect::getFactory() const { 517 return GrTBackendEffectFactory<GrMatrixConvolutionEffect>::getInstance(); 518 } 519 520 bool GrMatrixConvolutionEffect::onIsEqual(const GrEffect& sBase) const { 521 const GrMatrixConvolutionEffect& s = CastEffect<GrMatrixConvolutionEffect>(sBase); 522 return this->texture(0) == s.texture(0) && 523 fKernelSize == s.kernelSize() && 524 !memcmp(fKernel, s.kernel(), fKernelSize.width() * fKernelSize.height() * sizeof(float)) && 525 fGain == s.gain() && 526 fBias == s.bias() && 527 fTarget == s.target() && 528 fTileMode == s.tileMode() && 529 fConvolveAlpha == s.convolveAlpha(); 530 } 531 532 GR_DEFINE_EFFECT_TEST(GrMatrixConvolutionEffect); 533 534 // A little bit less than the minimum # uniforms required by DX9SM2 (32). 535 // Allows for a 5x5 kernel (or 25x1, for that matter). 536 #define MAX_KERNEL_SIZE 25 537 538 GrEffectRef* GrMatrixConvolutionEffect::TestCreate(SkMWCRandom* random, 539 GrContext* context, 540 const GrDrawTargetCaps&, 541 GrTexture* textures[]) { 542 int texIdx = random->nextBool() ? GrEffectUnitTest::kSkiaPMTextureIdx : 543 GrEffectUnitTest::kAlphaTextureIdx; 544 int width = random->nextRangeU(1, MAX_KERNEL_SIZE); 545 int height = random->nextRangeU(1, MAX_KERNEL_SIZE / width); 546 SkISize kernelSize = SkISize::Make(width, height); 547 SkAutoTDeleteArray<SkScalar> kernel(new SkScalar[width * height]); 548 for (int i = 0; i < width * height; i++) { 549 kernel.get()[i] = random->nextSScalar1(); 550 } 551 SkScalar gain = random->nextSScalar1(); 552 SkScalar bias = random->nextSScalar1(); 553 SkIPoint target = SkIPoint::Make(random->nextRangeU(0, kernelSize.width()), 554 random->nextRangeU(0, kernelSize.height())); 555 TileMode tileMode = static_cast<TileMode>(random->nextRangeU(0, 2)); 556 bool convolveAlpha = random->nextBool(); 557 return GrMatrixConvolutionEffect::Create(textures[texIdx], 558 kernelSize, 559 kernel.get(), 560 gain, 561 bias, 562 target, 563 tileMode, 564 convolveAlpha); 565 } 566 567 bool SkMatrixConvolutionImageFilter::asNewEffect(GrEffectRef** effect, 568 GrTexture* texture, 569 const SkIPoint&) const { 570 if (!effect) { 571 return fKernelSize.width() * fKernelSize.height() <= MAX_KERNEL_SIZE; 572 } 573 SkASSERT(fKernelSize.width() * fKernelSize.height() <= MAX_KERNEL_SIZE); 574 *effect = GrMatrixConvolutionEffect::Create(texture, 575 fKernelSize, 576 fKernel, 577 fGain, 578 fBias, 579 fTarget, 580 fTileMode, 581 fConvolveAlpha); 582 return true; 583 } 584 585 /////////////////////////////////////////////////////////////////////////////// 586 587 #endif 588