SkMatrixConvolutionImageFilter.cpp revision 0ac6af49975c54c2debf41e9200af416ecd2d973
13c827367444ee418f129b2c238299f49d3264554Jarkko Poyry/*
23c827367444ee418f129b2c238299f49d3264554Jarkko Poyry * Copyright 2012 The Android Open Source Project
33c827367444ee418f129b2c238299f49d3264554Jarkko Poyry *
43c827367444ee418f129b2c238299f49d3264554Jarkko Poyry * Use of this source code is governed by a BSD-style license that can be
53c827367444ee418f129b2c238299f49d3264554Jarkko Poyry * found in the LICENSE file.
63c827367444ee418f129b2c238299f49d3264554Jarkko Poyry */
73c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
83c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "SkMatrixConvolutionImageFilter.h"
93c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "SkBitmap.h"
103c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "SkColorPriv.h"
113c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "SkFlattenableBuffers.h"
123c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "SkRect.h"
133c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "SkUnPreMultiply.h"
143c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
153c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#if SK_SUPPORT_GPU
163c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "gl/GrGLEffect.h"
173c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "gl/GrGLEffectMatrix.h"
183c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "effects/GrSingleTextureEffect.h"
193c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "GrTBackendEffectFactory.h"
203c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "GrTexture.h"
213c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#include "SkMatrix.h"
223c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
233c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#endif
243c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
253c827367444ee418f129b2c238299f49d3264554Jarkko PoyrySkMatrixConvolutionImageFilter::SkMatrixConvolutionImageFilter(const SkISize& kernelSize, const SkScalar* kernel, SkScalar gain, SkScalar bias, const SkIPoint& target, TileMode tileMode, bool convolveAlpha, SkImageFilter* input)
263c827367444ee418f129b2c238299f49d3264554Jarkko Poyry  : INHERITED(input),
273c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fKernelSize(kernelSize),
283c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fGain(gain),
293c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fBias(bias),
303c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fTarget(target),
313c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fTileMode(tileMode),
323c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fConvolveAlpha(convolveAlpha) {
333c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    uint32_t size = fKernelSize.fWidth * fKernelSize.fHeight;
343c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fKernel = SkNEW_ARRAY(SkScalar, size);
353c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    memcpy(fKernel, kernel, size * sizeof(SkScalar));
363c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    SkASSERT(kernelSize.fWidth >= 1 && kernelSize.fHeight >= 1);
373c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    SkASSERT(target.fX >= 0 && target.fX < kernelSize.fWidth);
383c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    SkASSERT(target.fY >= 0 && target.fY < kernelSize.fHeight);
393c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
403c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
413c827367444ee418f129b2c238299f49d3264554Jarkko PoyrySkMatrixConvolutionImageFilter::SkMatrixConvolutionImageFilter(SkFlattenableReadBuffer& buffer) : INHERITED(buffer) {
423c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fKernelSize.fWidth = buffer.readInt();
433c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fKernelSize.fHeight = buffer.readInt();
443c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    uint32_t size = fKernelSize.fWidth * fKernelSize.fHeight;
453c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fKernel = SkNEW_ARRAY(SkScalar, size);
463c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    SkDEBUGCODE(uint32_t readSize = )buffer.readScalarArray(fKernel);
473c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    SkASSERT(readSize == size);
483c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fGain = buffer.readScalar();
493c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fBias = buffer.readScalar();
503c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fTarget.fX = buffer.readInt();
513c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fTarget.fY = buffer.readInt();
523c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fTileMode = (TileMode) buffer.readInt();
533c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    fConvolveAlpha = buffer.readBool();
543c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
553c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
563c827367444ee418f129b2c238299f49d3264554Jarkko Poyryvoid SkMatrixConvolutionImageFilter::flatten(SkFlattenableWriteBuffer& buffer) const {
573c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    this->INHERITED::flatten(buffer);
583c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeInt(fKernelSize.fWidth);
593c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeInt(fKernelSize.fHeight);
603c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeScalarArray(fKernel, fKernelSize.fWidth * fKernelSize.fHeight);
613c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeScalar(fGain);
623c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeScalar(fBias);
633c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeInt(fTarget.fX);
643c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeInt(fTarget.fY);
653c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeInt((int) fTileMode);
663c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    buffer.writeBool(fConvolveAlpha);
673c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
683c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
693c827367444ee418f129b2c238299f49d3264554Jarkko PoyrySkMatrixConvolutionImageFilter::~SkMatrixConvolutionImageFilter() {
703c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    delete[] fKernel;
713c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
723c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
733c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass UncheckedPixelFetcher {
743c827367444ee418f129b2c238299f49d3264554Jarkko Poyrypublic:
753c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    static inline SkPMColor fetch(const SkBitmap& src, int x, int y) {
763c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        return *src.getAddr32(x, y);
773c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    }
783c827367444ee418f129b2c238299f49d3264554Jarkko Poyry};
793c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
803c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass ClampPixelFetcher {
813c827367444ee418f129b2c238299f49d3264554Jarkko Poyrypublic:
823c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    static inline SkPMColor fetch(const SkBitmap& src, int x, int y) {
833c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        x = SkClampMax(x, src.width() - 1);
843c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        y = SkClampMax(y, src.height() - 1);
853c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        return *src.getAddr32(x, y);
863c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    }
873c827367444ee418f129b2c238299f49d3264554Jarkko Poyry};
883c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
893c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass RepeatPixelFetcher {
903c827367444ee418f129b2c238299f49d3264554Jarkko Poyrypublic:
913c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    static inline SkPMColor fetch(const SkBitmap& src, int x, int y) {
923c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        x %= src.width();
933c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        y %= src.height();
943c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        if (x < 0) {
953c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            x += src.width();
963c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        }
973c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        if (y < 0) {
983c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            y += src.height();
993c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        }
1003c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        return *src.getAddr32(x, y);
1013c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    }
1023c827367444ee418f129b2c238299f49d3264554Jarkko Poyry};
1033c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
1043c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass ClampToBlackPixelFetcher {
1053c827367444ee418f129b2c238299f49d3264554Jarkko Poyrypublic:
1063c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    static inline SkPMColor fetch(const SkBitmap& src, int x, int y) {
1073c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        if (x < 0 || x >= src.width() || y < 0 || y >= src.height()) {
1083c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            return 0;
1093c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        } else {
1103c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            return *src.getAddr32(x, y);
1113c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        }
1123c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    }
1133c827367444ee418f129b2c238299f49d3264554Jarkko Poyry};
1143c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
1153c827367444ee418f129b2c238299f49d3264554Jarkko Poyrytemplate<class PixelFetcher, bool convolveAlpha>
1163c827367444ee418f129b2c238299f49d3264554Jarkko Poyryvoid SkMatrixConvolutionImageFilter::filterPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) {
1173c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    for (int y = rect.fTop; y < rect.fBottom; ++y) {
1183c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        SkPMColor* dptr = result->getAddr32(rect.fLeft, y);
1193c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        for (int x = rect.fLeft; x < rect.fRight; ++x) {
1203c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            SkScalar sumA = 0, sumR = 0, sumG = 0, sumB = 0;
1213c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            for (int cy = 0; cy < fKernelSize.fHeight; cy++) {
1223c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                for (int cx = 0; cx < fKernelSize.fWidth; cx++) {
1233c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                    SkPMColor s = PixelFetcher::fetch(src, x + cx - fTarget.fX, y + cy - fTarget.fY);
1243c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                    SkScalar k = fKernel[cy * fKernelSize.fWidth + cx];
1253c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                    if (convolveAlpha) {
1263c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                        sumA += SkScalarMul(SkIntToScalar(SkGetPackedA32(s)), k);
1273c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                    }
1283c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                    sumR += SkScalarMul(SkIntToScalar(SkGetPackedR32(s)), k);
1293c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                    sumG += SkScalarMul(SkIntToScalar(SkGetPackedG32(s)), k);
1303c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                    sumB += SkScalarMul(SkIntToScalar(SkGetPackedB32(s)), k);
1313c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                }
1323c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            }
1333c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            int a = convolveAlpha
1343c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                  ? SkClampMax(SkScalarFloorToInt(SkScalarMul(sumA, fGain) + fBias), 255)
1353c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                  : 255;
1363c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            int r = SkClampMax(SkScalarFloorToInt(SkScalarMul(sumR, fGain) + fBias), a);
1373c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            int g = SkClampMax(SkScalarFloorToInt(SkScalarMul(sumG, fGain) + fBias), a);
1383c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            int b = SkClampMax(SkScalarFloorToInt(SkScalarMul(sumB, fGain) + fBias), a);
1393c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            if (!convolveAlpha) {
1403c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                a = SkGetPackedA32(PixelFetcher::fetch(src, x, y));
1413c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                *dptr++ = SkPreMultiplyARGB(a, r, g, b);
1423c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            } else {
1433c827367444ee418f129b2c238299f49d3264554Jarkko Poyry                *dptr++ = SkPackARGB32(a, r, g, b);
1443c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            }
1453c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        }
1463c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    }
1473c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
1483c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
1493c827367444ee418f129b2c238299f49d3264554Jarkko Poyrytemplate<class PixelFetcher>
1503c827367444ee418f129b2c238299f49d3264554Jarkko Poyryvoid SkMatrixConvolutionImageFilter::filterPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) {
1513c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    if (fConvolveAlpha) {
1523c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        filterPixels<PixelFetcher, true>(src, result, rect);
1533c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    } else {
1543c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        filterPixels<PixelFetcher, false>(src, result, rect);
1553c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    }
1563c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
1573c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
1583c827367444ee418f129b2c238299f49d3264554Jarkko Poyryvoid SkMatrixConvolutionImageFilter::filterInteriorPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) {
1593c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    filterPixels<UncheckedPixelFetcher>(src, result, rect);
1603c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
1613c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
1623c827367444ee418f129b2c238299f49d3264554Jarkko Poyryvoid SkMatrixConvolutionImageFilter::filterBorderPixels(const SkBitmap& src, SkBitmap* result, const SkIRect& rect) {
1633c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    switch (fTileMode) {
1643c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        case kClamp_TileMode:
1653c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            filterPixels<ClampPixelFetcher>(src, result, rect);
1663c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            break;
1673c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        case kRepeat_TileMode:
1683c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            filterPixels<RepeatPixelFetcher>(src, result, rect);
1693c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            break;
1703c827367444ee418f129b2c238299f49d3264554Jarkko Poyry        case kClampToBlack_TileMode:
1713c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            filterPixels<ClampToBlackPixelFetcher>(src, result, rect);
1723c827367444ee418f129b2c238299f49d3264554Jarkko Poyry            break;
1733c827367444ee418f129b2c238299f49d3264554Jarkko Poyry    }
1743c827367444ee418f129b2c238299f49d3264554Jarkko Poyry}
1753c827367444ee418f129b2c238299f49d3264554Jarkko Poyry
1763c827367444ee418f129b2c238299f49d3264554Jarkko Poyry// FIXME:  This should be refactored to SkSingleInputImageFilter for
1773c827367444ee418f129b2c238299f49d3264554Jarkko Poyry// use by other filters.  For now, we assume the input is always
1783c827367444ee418f129b2c238299f49d3264554Jarkko Poyry// premultiplied and unpremultiply it
1793c827367444ee418f129b2c238299f49d3264554Jarkko Poyrystatic 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
201bool SkMatrixConvolutionImageFilter::onFilterImage(Proxy* proxy,
202                                                   const SkBitmap& source,
203                                                   const SkMatrix& matrix,
204                                                   SkBitmap* result,
205                                                   SkIPoint* loc) {
206    SkBitmap src = this->getInputResult(proxy, source, matrix, loc);
207    if (src.config() != SkBitmap::kARGB_8888_Config) {
208        return false;
209    }
210
211    if (!fConvolveAlpha && !src.isOpaque()) {
212        src = unpremultiplyBitmap(src);
213    }
214
215    SkAutoLockPixels alp(src);
216    if (!src.getPixels()) {
217        return false;
218    }
219
220    result->setConfig(src.config(), src.width(), src.height());
221    result->allocPixels();
222
223    SkIRect interior = SkIRect::MakeXYWH(fTarget.fX, fTarget.fY,
224                                         src.width() - fKernelSize.fWidth + 1,
225                                         src.height() - fKernelSize.fHeight + 1);
226    SkIRect top = SkIRect::MakeWH(src.width(), fTarget.fY);
227    SkIRect bottom = SkIRect::MakeLTRB(0, interior.bottom(),
228                                       src.width(), src.height());
229    SkIRect left = SkIRect::MakeXYWH(0, interior.top(),
230                                     fTarget.fX, interior.height());
231    SkIRect right = SkIRect::MakeLTRB(interior.right(), interior.top(),
232                                      src.width(), interior.bottom());
233    filterBorderPixels(src, result, top);
234    filterBorderPixels(src, result, left);
235    filterInteriorPixels(src, result, interior);
236    filterBorderPixels(src, result, right);
237    filterBorderPixels(src, result, bottom);
238    return true;
239}
240
241#if SK_SUPPORT_GPU
242
243///////////////////////////////////////////////////////////////////////////////
244
245class GrGLMatrixConvolutionEffect;
246
247class GrMatrixConvolutionEffect : public GrSingleTextureEffect {
248public:
249    typedef SkMatrixConvolutionImageFilter::TileMode TileMode;
250    static GrEffectRef* Create(GrTexture* texture,
251                               const SkISize& kernelSize,
252                               const SkScalar* kernel,
253                               SkScalar gain,
254                               SkScalar bias,
255                               const SkIPoint& target,
256                               TileMode tileMode,
257                               bool convolveAlpha) {
258        SkAutoTUnref<GrEffect> effect(SkNEW_ARGS(GrMatrixConvolutionEffect, (texture,
259                                                                             kernelSize,
260                                                                             kernel,
261                                                                             gain,
262                                                                             bias,
263                                                                             target,
264                                                                             tileMode,
265                                                                             convolveAlpha)));
266        return CreateEffectPtr(effect);
267    }
268    virtual ~GrMatrixConvolutionEffect();
269
270    virtual void getConstantColorComponents(GrColor* color,
271                                            uint32_t* validFlags) const SK_OVERRIDE {
272        // TODO: Try to do better?
273        *validFlags = 0;
274    }
275
276    static const char* Name() { return "MatrixConvolution"; }
277    const SkISize& kernelSize() const { return fKernelSize; }
278    const float* target() const { return fTarget; }
279    const float* kernel() const { return fKernel; }
280    float gain() const { return fGain; }
281    float bias() const { return fBias; }
282    TileMode tileMode() const { return fTileMode; }
283    bool convolveAlpha() const { return fConvolveAlpha; }
284
285    typedef GrGLMatrixConvolutionEffect GLEffect;
286
287
288
289    virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE;
290    virtual bool isEqual(const GrEffect&) const SK_OVERRIDE;
291
292private:
293    GrMatrixConvolutionEffect(GrTexture*,
294                              const SkISize& kernelSize,
295                              const SkScalar* kernel,
296                              SkScalar gain,
297                              SkScalar bias,
298                              const SkIPoint& target,
299                              TileMode tileMode,
300                              bool convolveAlpha);
301
302    SkISize  fKernelSize;
303    float   *fKernel;
304    float    fGain;
305    float    fBias;
306    float    fTarget[2];
307    TileMode fTileMode;
308    bool     fConvolveAlpha;
309
310    GR_DECLARE_EFFECT_TEST;
311
312    typedef GrSingleTextureEffect INHERITED;
313};
314
315class GrGLMatrixConvolutionEffect : public GrGLEffect {
316public:
317    GrGLMatrixConvolutionEffect(const GrBackendEffectFactory& factory,
318                                const GrEffect& effect);
319    virtual void emitCode(GrGLShaderBuilder*,
320                          const GrEffectStage&,
321                          EffectKey,
322                          const char* vertexCoords,
323                          const char* outputColor,
324                          const char* inputColor,
325                          const TextureSamplerArray&) SK_OVERRIDE;
326
327    static inline EffectKey GenKey(const GrEffectStage&, const GrGLCaps&);
328
329    virtual void setData(const GrGLUniformManager&, const GrEffectStage&) SK_OVERRIDE;
330
331private:
332    typedef GrGLUniformManager::UniformHandle        UniformHandle;
333    typedef SkMatrixConvolutionImageFilter::TileMode TileMode;
334    SkISize             fKernelSize;
335    TileMode            fTileMode;
336    bool                fConvolveAlpha;
337
338    UniformHandle       fKernelUni;
339    UniformHandle       fImageIncrementUni;
340    UniformHandle       fTargetUni;
341    UniformHandle       fGainUni;
342    UniformHandle       fBiasUni;
343
344    GrGLEffectMatrix    fEffectMatrix;
345
346    typedef GrGLEffect INHERITED;
347};
348
349GrGLMatrixConvolutionEffect::GrGLMatrixConvolutionEffect(const GrBackendEffectFactory& factory,
350                                                         const GrEffect& effect)
351    : INHERITED(factory)
352    , fKernelUni(GrGLUniformManager::kInvalidUniformHandle)
353    , fImageIncrementUni(GrGLUniformManager::kInvalidUniformHandle)
354    , fTargetUni(GrGLUniformManager::kInvalidUniformHandle)
355    , fGainUni(GrGLUniformManager::kInvalidUniformHandle)
356    , fBiasUni(GrGLUniformManager::kInvalidUniformHandle) {
357    const GrMatrixConvolutionEffect& m = static_cast<const GrMatrixConvolutionEffect&>(effect);
358    fKernelSize = m.kernelSize();
359    fTileMode = m.tileMode();
360    fConvolveAlpha = m.convolveAlpha();
361}
362
363static void appendTextureLookup(GrGLShaderBuilder* builder,
364                                const GrGLShaderBuilder::TextureSampler& sampler,
365                                const char* coord,
366                                SkMatrixConvolutionImageFilter::TileMode tileMode) {
367    SkString* code = &builder->fFSCode;
368    SkString clampedCoord;
369    switch (tileMode) {
370        case SkMatrixConvolutionImageFilter::kClamp_TileMode:
371            clampedCoord.printf("clamp(%s, 0.0, 1.0)", coord);
372            coord = clampedCoord.c_str();
373            break;
374        case SkMatrixConvolutionImageFilter::kRepeat_TileMode:
375            clampedCoord.printf("fract(%s)", coord);
376            coord = clampedCoord.c_str();
377            break;
378        case SkMatrixConvolutionImageFilter::kClampToBlack_TileMode:
379            code->appendf("clamp(%s, 0.0, 1.0) != %s ? vec4(0, 0, 0, 0) : ", coord, coord);
380            break;
381    }
382    builder->appendTextureLookup(code, sampler, coord);
383}
384
385void GrGLMatrixConvolutionEffect::emitCode(GrGLShaderBuilder* builder,
386                                           const GrEffectStage&,
387                                           EffectKey key,
388                                           const char* vertexCoords,
389                                           const char* outputColor,
390                                           const char* inputColor,
391                                           const TextureSamplerArray& samplers) {
392    const char* coords;
393    fEffectMatrix.emitCodeMakeFSCoords2D(builder, key, vertexCoords, &coords);
394    fImageIncrementUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
395                                             kVec2f_GrSLType, "ImageIncrement");
396    fKernelUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_ShaderType,
397                                             kFloat_GrSLType, "Kernel", fKernelSize.width() * fKernelSize.height());
398    fTargetUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
399                                             kVec2f_GrSLType, "Target");
400    fGainUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
401                                   kFloat_GrSLType, "Gain");
402    fBiasUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
403                                   kFloat_GrSLType, "Bias");
404
405    SkString* code = &builder->fFSCode;
406
407    const char* target = builder->getUniformCStr(fTargetUni);
408    const char* imgInc = builder->getUniformCStr(fImageIncrementUni);
409    const char* kernel = builder->getUniformCStr(fKernelUni);
410    const char* gain = builder->getUniformCStr(fGainUni);
411    const char* bias = builder->getUniformCStr(fBiasUni);
412    int kWidth = fKernelSize.width();
413    int kHeight = fKernelSize.height();
414
415    code->appendf("\t\tvec4 sum = vec4(0, 0, 0, 0);\n");
416    code->appendf("\t\tvec2 coord = %s - %s * %s;\n", coords, target, imgInc);
417    code->appendf("\t\tfor (int y = 0; y < %d; y++) {\n", kHeight);
418    code->appendf("\t\t\tfor (int x = 0; x < %d; x++) {\n", kWidth);
419    code->appendf("\t\t\t\tfloat k = %s[y * %d + x];\n", kernel, kWidth);
420    code->appendf("\t\t\t\tvec2 coord2 = coord + vec2(x, y) * %s;\n", imgInc);
421    code->appendf("\t\t\t\tvec4 c = ");
422    appendTextureLookup(builder, samplers[0], "coord2", fTileMode);
423    code->appendf(";\n");
424    if (!fConvolveAlpha) {
425        code->appendf("\t\t\t\tc.rgb /= c.a;\n");
426    }
427    code->appendf("\t\t\t\tsum += c * k;\n");
428    code->appendf("\t\t\t}\n");
429    code->appendf("\t\t}\n");
430    if (fConvolveAlpha) {
431        code->appendf("\t\t%s = sum * %s + %s;\n", outputColor, gain, bias);
432        code->appendf("\t\t%s.rgb = clamp(%s.rgb, 0.0, %s.a);\n", outputColor, outputColor, outputColor);
433    } else {
434        code->appendf("\t\tvec4 c = ");
435        appendTextureLookup(builder, samplers[0], coords, fTileMode);
436        code->appendf(";\n");
437        code->appendf("\t\t%s.a = c.a;\n", outputColor);
438        code->appendf("\t\t%s.rgb = sum.rgb * %s + %s;\n", outputColor, gain, bias);
439        code->appendf("\t\t%s.rgb *= %s.a;\n", outputColor, outputColor);
440    }
441}
442
443namespace {
444
445int encodeXY(int x, int y) {
446    SkASSERT(x >= 1 && y >= 1 && x * y <= 32);
447    if (y < x)
448        return 0x40 | encodeXY(y, x);
449    else
450        return (0x40 >> x) | (y - x);
451}
452
453};
454
455GrGLEffect::EffectKey GrGLMatrixConvolutionEffect::GenKey(const GrEffectStage& s, const GrGLCaps&) {
456    const GrMatrixConvolutionEffect& m =
457        static_cast<const GrMatrixConvolutionEffect&>(*s.getEffect());
458    EffectKey key = encodeXY(m.kernelSize().width(), m.kernelSize().height());
459    key |= m.tileMode() << 7;
460    key |= m.convolveAlpha() ? 1 << 9 : 0;
461    key <<= GrGLEffectMatrix::kKeyBits;
462    EffectKey matrixKey = GrGLEffectMatrix::GenKey(m.getMatrix(),
463                                                   s.getCoordChangeMatrix(),
464                                                   m.texture(0));
465    return key | matrixKey;
466}
467
468void GrGLMatrixConvolutionEffect::setData(const GrGLUniformManager& uman,
469                                          const GrEffectStage& stage) {
470    const GrMatrixConvolutionEffect& effect =
471        static_cast<const GrMatrixConvolutionEffect&>(*stage.getEffect());
472    GrTexture& texture = *effect.texture(0);
473    // the code we generated was for a specific kernel size
474    GrAssert(effect.kernelSize() == fKernelSize);
475    GrAssert(effect.tileMode() == fTileMode);
476    float imageIncrement[2];
477    float ySign = texture.origin() == GrSurface::kTopLeft_Origin ? 1.0f : -1.0f;
478    imageIncrement[0] = 1.0f / texture.width();
479    imageIncrement[1] = ySign / texture.height();
480    uman.set2fv(fImageIncrementUni, 0, 1, imageIncrement);
481    uman.set2fv(fTargetUni, 0, 1, effect.target());
482    uman.set1fv(fKernelUni, 0, fKernelSize.width() * fKernelSize.height(), effect.kernel());
483    uman.set1f(fGainUni, effect.gain());
484    uman.set1f(fBiasUni, effect.bias());
485    fEffectMatrix.setData(uman,
486                          effect.getMatrix(),
487                          stage.getCoordChangeMatrix(),
488                          effect.texture(0));
489}
490
491GrMatrixConvolutionEffect::GrMatrixConvolutionEffect(GrTexture* texture,
492                                                     const SkISize& kernelSize,
493                                                     const SkScalar* kernel,
494                                                     SkScalar gain,
495                                                     SkScalar bias,
496                                                     const SkIPoint& target,
497                                                     TileMode tileMode,
498                                                     bool convolveAlpha)
499  : INHERITED(texture, MakeDivByTextureWHMatrix(texture)),
500    fKernelSize(kernelSize),
501    fGain(SkScalarToFloat(gain)),
502    fBias(SkScalarToFloat(bias) / 255.0f),
503    fTileMode(tileMode),
504    fConvolveAlpha(convolveAlpha) {
505    fKernel = new float[kernelSize.width() * kernelSize.height()];
506    for (int i = 0; i < kernelSize.width() * kernelSize.height(); i++) {
507        fKernel[i] = SkScalarToFloat(kernel[i]);
508    }
509    fTarget[0] = static_cast<float>(target.x());
510    fTarget[1] = static_cast<float>(target.y());
511}
512
513GrMatrixConvolutionEffect::~GrMatrixConvolutionEffect() {
514    delete[] fKernel;
515}
516
517const GrBackendEffectFactory& GrMatrixConvolutionEffect::getFactory() const {
518    return GrTBackendEffectFactory<GrMatrixConvolutionEffect>::getInstance();
519}
520
521bool GrMatrixConvolutionEffect::isEqual(const GrEffect& sBase) const {
522    const GrMatrixConvolutionEffect& s =
523        static_cast<const GrMatrixConvolutionEffect&>(sBase);
524    return INHERITED::isEqual(sBase) &&
525           fKernelSize == s.kernelSize() &&
526           !memcmp(fKernel, s.kernel(), fKernelSize.width() * fKernelSize.height() * sizeof(float)) &&
527           fGain == s.gain() &&
528           fBias == s.bias() &&
529           fTarget == s.target() &&
530           fTileMode == s.tileMode() &&
531           fConvolveAlpha == s.convolveAlpha();
532}
533
534GR_DEFINE_EFFECT_TEST(GrMatrixConvolutionEffect);
535
536// A little bit less than the minimum # uniforms required by DX9SM2 (32).
537// Allows for a 5x5 kernel (or 25x1, for that matter).
538#define MAX_KERNEL_SIZE 25
539
540GrEffectRef* GrMatrixConvolutionEffect::TestCreate(SkRandom* random,
541                                                   GrContext* context,
542                                                   GrTexture* textures[]) {
543    int texIdx = random->nextBool() ? GrEffectUnitTest::kSkiaPMTextureIdx :
544                                      GrEffectUnitTest::kAlphaTextureIdx;
545    int width = random->nextRangeU(1, MAX_KERNEL_SIZE);
546    int height = random->nextRangeU(1, MAX_KERNEL_SIZE / width);
547    SkISize kernelSize = SkISize::Make(width, height);
548    SkScalar* kernel = new SkScalar[width * height];
549    for (int i = 0; i < width * height; i++) {
550        kernel[i] = random->nextSScalar1();
551    }
552    SkScalar gain = random->nextSScalar1();
553    SkScalar bias = random->nextSScalar1();
554    SkIPoint target = SkIPoint::Make(random->nextRangeU(0, kernelSize.width()),
555                                     random->nextRangeU(0, kernelSize.height()));
556    TileMode tileMode = static_cast<TileMode>(random->nextRangeU(0, 2));
557    bool convolveAlpha = random->nextBool();
558    return GrMatrixConvolutionEffect::Create(textures[texIdx],
559                                             kernelSize,
560                                             kernel,
561                                             gain,
562                                             bias,
563                                             target,
564                                             tileMode,
565                                             convolveAlpha);
566
567}
568
569bool SkMatrixConvolutionImageFilter::asNewEffect(GrEffectRef** effect,
570                                                 GrTexture* texture) const {
571    bool ok = fKernelSize.width() * fKernelSize.height() <= MAX_KERNEL_SIZE;
572    if (ok && effect) {
573        *effect = GrMatrixConvolutionEffect::Create(texture,
574                                                    fKernelSize,
575                                                    fKernel,
576                                                    fGain,
577                                                    fBias,
578                                                     fTarget,
579                                                     fTileMode,
580                                                     fConvolveAlpha);
581    }
582    return ok;
583}
584
585///////////////////////////////////////////////////////////////////////////////
586
587#endif
588