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
9#ifndef SkMorphologyImageFilter_DEFINED
10#define SkMorphologyImageFilter_DEFINED
11
12#include "SkColor.h"
13#include "SkImageFilter.h"
14#include "SkSize.h"
15
16class SK_API SkMorphologyImageFilter : public SkImageFilter {
17public:
18    virtual void computeFastBounds(const SkRect& src, SkRect* dst) const SK_OVERRIDE;
19    virtual bool onFilterBounds(const SkIRect& src, const SkMatrix& ctm, SkIRect* dst) const SK_OVERRIDE;
20
21    /**
22     * All morphology procs have the same signature: src is the source buffer, dst the
23     * destination buffer, radius is the morphology radius, width and height are the bounds
24     * of the destination buffer (in pixels), and srcStride and dstStride are the
25     * number of pixels per row in each buffer. All buffers are 8888.
26     */
27
28    typedef void (*Proc)(const SkPMColor* src, SkPMColor* dst, int radius,
29                         int width, int height, int srcStride, int dstStride);
30
31protected:
32    SkMorphologyImageFilter(int radiusX, int radiusY, SkImageFilter* input,
33                            const CropRect* cropRect);
34    bool filterImageGeneric(Proc procX, Proc procY,
35                            Proxy*, const SkBitmap& src, const Context&,
36                            SkBitmap* result, SkIPoint* offset) const;
37    SkMorphologyImageFilter(SkReadBuffer& buffer);
38    virtual void flatten(SkWriteBuffer&) const SK_OVERRIDE;
39#if SK_SUPPORT_GPU
40    virtual bool canFilterImageGPU() const SK_OVERRIDE { return true; }
41    bool filterImageGPUGeneric(bool dilate, Proxy* proxy, const SkBitmap& src,
42                               const Context& ctm, SkBitmap* result,
43                               SkIPoint* offset) const;
44#endif
45
46    SkISize    radius() const { return fRadius; }
47
48private:
49    SkISize    fRadius;
50    typedef SkImageFilter INHERITED;
51};
52
53class SK_API SkDilateImageFilter : public SkMorphologyImageFilter {
54public:
55    static SkDilateImageFilter* Create(int radiusX, int radiusY,
56                                       SkImageFilter* input = NULL,
57                                       const CropRect* cropRect = NULL) {
58        return SkNEW_ARGS(SkDilateImageFilter, (radiusX, radiusY, input, cropRect));
59    }
60
61    virtual bool onFilterImage(Proxy*, const SkBitmap& src, const Context&,
62                               SkBitmap* result, SkIPoint* offset) const SK_OVERRIDE;
63#if SK_SUPPORT_GPU
64    virtual bool filterImageGPU(Proxy* proxy, const SkBitmap& src, const Context&,
65                                SkBitmap* result, SkIPoint* offset) const SK_OVERRIDE;
66#endif
67
68    SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkDilateImageFilter)
69
70protected:
71    SkDilateImageFilter(int radiusX, int radiusY,
72                        SkImageFilter* input = NULL,
73                        const CropRect* cropRect = NULL)
74        : INHERITED(radiusX, radiusY, input, cropRect) {}
75    explicit SkDilateImageFilter(SkReadBuffer& buffer) : INHERITED(buffer) {}
76
77private:
78    typedef SkMorphologyImageFilter INHERITED;
79};
80
81class SK_API SkErodeImageFilter : public SkMorphologyImageFilter {
82public:
83    static SkErodeImageFilter* Create(int radiusX, int radiusY,
84                                      SkImageFilter* input = NULL,
85                                      const CropRect* cropRect = NULL) {
86        return SkNEW_ARGS(SkErodeImageFilter, (radiusX, radiusY, input, cropRect));
87    }
88
89    virtual bool onFilterImage(Proxy*, const SkBitmap& src, const Context&,
90                               SkBitmap* result, SkIPoint* offset) const SK_OVERRIDE;
91#if SK_SUPPORT_GPU
92    virtual bool filterImageGPU(Proxy* proxy, const SkBitmap& src, const Context&,
93                                SkBitmap* result, SkIPoint* offset) const SK_OVERRIDE;
94#endif
95
96    SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkErodeImageFilter)
97
98protected:
99    SkErodeImageFilter(int radiusX, int radiusY,
100                       SkImageFilter* input = NULL,
101                       const CropRect* cropRect = NULL)
102        : INHERITED(radiusX, radiusY, input, cropRect) {}
103    explicit SkErodeImageFilter(SkReadBuffer& buffer) : INHERITED(buffer) {}
104
105private:
106    typedef SkMorphologyImageFilter INHERITED;
107};
108
109#endif
110