11cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger/* 21cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger * Copyright 2011 The Android Open Source Project 31cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger * 41cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger * Use of this source code is governed by a BSD-style license that can be 51cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger * found in the LICENSE file. 61cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger */ 71cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 81cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger#include "SkBlurImageFilter.h" 91cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger#include "SkColorPriv.h" 101cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 111cab2921ab279367f8206cdadc9259d12e603548Derek SollenbergerSkBlurImageFilter::SkBlurImageFilter(SkFlattenableReadBuffer& buffer) 121cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger : INHERITED(buffer) { 131cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger fSigma.fWidth = buffer.readScalar(); 141cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger fSigma.fHeight = buffer.readScalar(); 151cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 161cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 171cab2921ab279367f8206cdadc9259d12e603548Derek SollenbergerSkBlurImageFilter::SkBlurImageFilter(SkScalar sigmaX, SkScalar sigmaY) 181cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger : fSigma(SkSize::Make(sigmaX, sigmaY)) { 191cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkASSERT(sigmaX >= 0 && sigmaY >= 0); 201cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 211cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 221cab2921ab279367f8206cdadc9259d12e603548Derek Sollenbergerbool SkBlurImageFilter::asABlur(SkSize* sigma) const { 231cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *sigma = fSigma; 241cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger return true; 251cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 261cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 271cab2921ab279367f8206cdadc9259d12e603548Derek Sollenbergervoid SkBlurImageFilter::flatten(SkFlattenableWriteBuffer& buffer) { 281cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger this->INHERITED::flatten(buffer); 291cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger buffer.writeScalar(fSigma.fWidth); 301cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger buffer.writeScalar(fSigma.fHeight); 311cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 321cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 331cab2921ab279367f8206cdadc9259d12e603548Derek Sollenbergerstatic void boxBlurX(const SkBitmap& src, SkBitmap* dst, int kernelSize, 341cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int leftOffset, int rightOffset) 351cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger{ 361cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int width = src.width(), height = src.height(); 371cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int rightBorder = SkMin32(rightOffset + 1, width); 381cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger for (int y = 0; y < height; ++y) { 391cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int sumA = 0, sumR = 0, sumG = 0, sumB = 0; 401cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkPMColor* p = src.getAddr32(0, y); 411cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger for (int i = 0; i < rightBorder; ++i) { 421cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumA += SkGetPackedA32(*p); 431cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR += SkGetPackedR32(*p); 441cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG += SkGetPackedG32(*p); 451cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB += SkGetPackedB32(*p); 461cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger p++; 471cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 481cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 491cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger const SkColor* sptr = src.getAddr32(0, y); 501cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkColor* dptr = dst->getAddr32(0, y); 511cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger for (int x = 0; x < width; ++x) { 521cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *dptr = SkPackARGB32(sumA / kernelSize, 531cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR / kernelSize, 541cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG / kernelSize, 551cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB / kernelSize); 561cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (x >= leftOffset) { 571cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkColor l = *(sptr - leftOffset); 581cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumA -= SkGetPackedA32(l); 591cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR -= SkGetPackedR32(l); 601cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG -= SkGetPackedG32(l); 611cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB -= SkGetPackedB32(l); 621cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 631cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (x + rightOffset + 1 < width) { 641cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkColor r = *(sptr + rightOffset + 1); 651cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumA += SkGetPackedA32(r); 661cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR += SkGetPackedR32(r); 671cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG += SkGetPackedG32(r); 681cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB += SkGetPackedB32(r); 691cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 701cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sptr++; 711cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger dptr++; 721cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 731cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 741cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 751cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 761cab2921ab279367f8206cdadc9259d12e603548Derek Sollenbergerstatic void boxBlurY(const SkBitmap& src, SkBitmap* dst, int kernelSize, 771cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int topOffset, int bottomOffset) 781cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger{ 791cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int width = src.width(), height = src.height(); 801cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int bottomBorder = SkMin32(bottomOffset + 1, height); 811cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int srcStride = src.rowBytesAsPixels(); 821cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int dstStride = dst->rowBytesAsPixels(); 831cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger for (int x = 0; x < width; ++x) { 841cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int sumA = 0, sumR = 0, sumG = 0, sumB = 0; 851cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkColor* p = src.getAddr32(x, 0); 861cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger for (int i = 0; i < bottomBorder; ++i) { 871cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumA += SkGetPackedA32(*p); 881cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR += SkGetPackedR32(*p); 891cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG += SkGetPackedG32(*p); 901cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB += SkGetPackedB32(*p); 911cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger p += srcStride; 921cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 931cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 941cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger const SkColor* sptr = src.getAddr32(x, 0); 951cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkColor* dptr = dst->getAddr32(x, 0); 961cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger for (int y = 0; y < height; ++y) { 971cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *dptr = SkPackARGB32(sumA / kernelSize, 981cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR / kernelSize, 991cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG / kernelSize, 1001cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB / kernelSize); 1011cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (y >= topOffset) { 1021cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkColor l = *(sptr - topOffset * srcStride); 1031cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumA -= SkGetPackedA32(l); 1041cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR -= SkGetPackedR32(l); 1051cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG -= SkGetPackedG32(l); 1061cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB -= SkGetPackedB32(l); 1071cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1081cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (y + bottomOffset + 1 < height) { 1091cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkColor r = *(sptr + (bottomOffset + 1) * srcStride); 1101cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumA += SkGetPackedA32(r); 1111cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumR += SkGetPackedR32(r); 1121cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumG += SkGetPackedG32(r); 1131cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sumB += SkGetPackedB32(r); 1141cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1151cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger sptr += srcStride; 1161cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger dptr += dstStride; 1171cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1181cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1191cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 1201cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1211cab2921ab279367f8206cdadc9259d12e603548Derek Sollenbergerstatic void getBox3Params(SkScalar s, int *kernelSize, int* kernelSize3, int *lowOffset, int *highOffset) 1221cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger{ 1231cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger float pi = SkScalarToFloat(SK_ScalarPI); 1241cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int d = static_cast<int>(floorf(SkScalarToFloat(s) * 3.0f * sqrtf(2.0f * pi) / 4.0f + 0.5f)); 1251cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *kernelSize = d; 1261cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (d % 2 == 1) { 1271cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *lowOffset = *highOffset = (d - 1) / 2; 1281cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *kernelSize3 = d; 1291cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } else { 1301cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *highOffset = d / 2; 1311cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *lowOffset = *highOffset - 1; 1321cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger *kernelSize3 = d + 1; 1331cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1341cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 1351cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1361cab2921ab279367f8206cdadc9259d12e603548Derek Sollenbergerbool SkBlurImageFilter::onFilterImage(Proxy*, 1371cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger const SkBitmap& src, const SkMatrix&, 1381cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkBitmap* dst, SkIPoint*) { 1391cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (src.config() != SkBitmap::kARGB_8888_Config) { 1401cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger return false; 1411cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1421cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1431cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkAutoLockPixels alp(src); 1441cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (!src.getPixels()) { 1451cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger return false; 1461cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1471cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1481cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger dst->setConfig(src.config(), src.width(), src.height()); 1491cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger dst->allocPixels(); 1501cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int kernelSizeX, kernelSizeX3, lowOffsetX, highOffsetX; 1511cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger int kernelSizeY, kernelSizeY3, lowOffsetY, highOffsetY; 1521cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger getBox3Params(fSigma.width(), &kernelSizeX, &kernelSizeX3, &lowOffsetX, &highOffsetX); 1531cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger getBox3Params(fSigma.height(), &kernelSizeY, &kernelSizeY3, &lowOffsetY, &highOffsetY); 1541cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1551cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (kernelSizeX < 0 || kernelSizeY < 0) { 1561cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger return false; 1571cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1581cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1591cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (kernelSizeX == 0 && kernelSizeY == 0) { 1601cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger src.copyTo(dst, dst->config()); 1611cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger return true; 1621cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1631cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1641cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger SkBitmap temp; 1651cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger temp.setConfig(dst->config(), dst->width(), dst->height()); 1661cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (!temp.allocPixels()) { 1671cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger return false; 1681cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1691cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1701cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger if (kernelSizeX > 0 && kernelSizeY > 0) { 1711cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurX(src, &temp, kernelSizeX, lowOffsetX, highOffsetX); 1721cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurY(temp, dst, kernelSizeY, lowOffsetY, highOffsetY); 1731cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurX(*dst, &temp, kernelSizeX, highOffsetX, lowOffsetX); 1741cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurY(temp, dst, kernelSizeY, highOffsetY, lowOffsetY); 1751cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurX(*dst, &temp, kernelSizeX3, highOffsetX, highOffsetX); 1761cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurY(temp, dst, kernelSizeY3, highOffsetY, highOffsetY); 1771cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } else if (kernelSizeX > 0) { 1781cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurX(src, dst, kernelSizeX, lowOffsetX, highOffsetX); 1791cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurX(*dst, &temp, kernelSizeX, highOffsetX, lowOffsetX); 1801cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurX(temp, dst, kernelSizeX3, highOffsetX, highOffsetX); 1811cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } else if (kernelSizeY > 0) { 1821cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurY(src, dst, kernelSizeY, lowOffsetY, highOffsetY); 1831cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurY(*dst, &temp, kernelSizeY, highOffsetY, lowOffsetY); 1841cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger boxBlurY(temp, dst, kernelSizeY3, highOffsetY, highOffsetY); 1851cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger } 1861cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger return true; 1871cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger} 1881cab2921ab279367f8206cdadc9259d12e603548Derek Sollenberger 1891cab2921ab279367f8206cdadc9259d12e603548Derek SollenbergerSK_DEFINE_FLATTENABLE_REGISTRAR(SkBlurImageFilter) 190