SkDisplacementMapEffect.h revision cac5fd597f6e2495f50aaa6bcbe3dadc56f0b977 
1/*
2 * Copyright 2013 Google Inc.
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#ifndef SkDisplacementMapEffect_DEFINED
9#define SkDisplacementMapEffect_DEFINED
10
11#include "SkImageFilter.h"
12#include "SkBitmap.h"
13
14class SK_API SkDisplacementMapEffect : public SkImageFilter {
15public:
16    enum ChannelSelectorType {
17        kUnknown_ChannelSelectorType,
18        kR_ChannelSelectorType,
19        kG_ChannelSelectorType,
20        kB_ChannelSelectorType,
21        kA_ChannelSelectorType
22    };
23
24    ~SkDisplacementMapEffect();
25
26    static SkDisplacementMapEffect* Create(ChannelSelectorType xChannelSelector,
27                                           ChannelSelectorType yChannelSelector,
28                                           SkScalar scale, SkImageFilter* displacement,
29                                           SkImageFilter* color = NULL,
30                                           const CropRect* cropRect = NULL) {
31        return SkNEW_ARGS(SkDisplacementMapEffect, (xChannelSelector, yChannelSelector, scale,
32                                                    displacement, color, cropRect));
33    }
34
35    SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkDisplacementMapEffect)
36
37    virtual bool onFilterImage(Proxy* proxy,
38                               const SkBitmap& src,
39                               const SkMatrix& ctm,
40                               SkBitmap* dst,
41                               SkIPoint* offset) const SK_OVERRIDE;
42    virtual void computeFastBounds(const SkRect& src, SkRect* dst) const SK_OVERRIDE;
43
44    virtual bool onFilterBounds(const SkIRect& src, const SkMatrix&,
45                                SkIRect* dst) const SK_OVERRIDE;
46
47#if SK_SUPPORT_GPU
48    virtual bool canFilterImageGPU() const SK_OVERRIDE { return true; }
49    virtual bool filterImageGPU(Proxy* proxy, const SkBitmap& src, const SkMatrix& ctm,
50                                SkBitmap* result, SkIPoint* offset) const SK_OVERRIDE;
51#endif
52
53protected:
54    explicit SkDisplacementMapEffect(SkReadBuffer& buffer);
55    virtual void flatten(SkWriteBuffer&) const SK_OVERRIDE;
56
57#ifdef SK_SUPPORT_LEGACY_PUBLICEFFECTCONSTRUCTORS
58public:
59#endif
60    SkDisplacementMapEffect(ChannelSelectorType xChannelSelector,
61                            ChannelSelectorType yChannelSelector,
62                            SkScalar scale, SkImageFilter* displacement,
63                            SkImageFilter* color = NULL,
64                            const CropRect* cropRect = NULL);
65
66private:
67    ChannelSelectorType fXChannelSelector;
68    ChannelSelectorType fYChannelSelector;
69    SkScalar fScale;
70    typedef SkImageFilter INHERITED;
71    const SkImageFilter* getDisplacementInput() const { return getInput(0); }
72    const SkImageFilter* getColorInput() const { return getInput(1); }
73};
74
75#endif
76