src/third_party/skia/include/effects/SkDisplacementMapEffect.h

   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
  14 class SK_API SkDisplacementMapEffect : public SkImageFilter {
  15 public:
  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         SkImageFilter* inputs[2] = { displacement, color };
  32         return SkNEW_ARGS(SkDisplacementMapEffect, (xChannelSelector, yChannelSelector, scale,
  33                                                     inputs, cropRect));
  34     }
  35
  36     SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkDisplacementMapEffect)
  37
  38     virtual bool onFilterImage(Proxy* proxy,
  39                                const SkBitmap& src,
  40                                const Context& ctx,
  41                                SkBitmap* dst,
  42                                SkIPoint* offset) const SK_OVERRIDE;
  43     virtual void computeFastBounds(const SkRect& src, SkRect* dst) const SK_OVERRIDE;
  44
  45     virtual bool onFilterBounds(const SkIRect& src, const SkMatrix&,
  46                                 SkIRect* dst) const SK_OVERRIDE;
  47
  48 #if SK_SUPPORT_GPU
  49     virtual bool canFilterImageGPU() const SK_OVERRIDE { return true; }
  50     virtual bool filterImageGPU(Proxy* proxy, const SkBitmap& src, const Context& ctx,
  51                                 SkBitmap* result, SkIPoint* offset) const SK_OVERRIDE;
  52 #endif
  53
  54 protected:
  55     SkDisplacementMapEffect(ChannelSelectorType xChannelSelector,
  56                             ChannelSelectorType yChannelSelector,
  57                             SkScalar scale, SkImageFilter* inputs[2],
  58                             const CropRect* cropRect);
  59     explicit SkDisplacementMapEffect(SkReadBuffer& buffer);
  60     virtual void flatten(SkWriteBuffer&) const SK_OVERRIDE;
  61
  62 private:
  63     ChannelSelectorType fXChannelSelector;
  64     ChannelSelectorType fYChannelSelector;
  65     SkScalar fScale;
  66     typedef SkImageFilter INHERITED;
  67     const SkImageFilter* getDisplacementInput() const { return getInput(0); }
  68     const SkImageFilter* getColorInput() const { return getInput(1); }
  69 };
  70
  71 #endif