gfx/2d/FilterNodeD2D1.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef MOZILLA_GFX_FILTERNODED2D1_H_
   8 #define MOZILLA_GFX_FILTERNODED2D1_H_
   9
  10 #include "2D.h"
  11 #include "Filters.h"
  12 #include <vector>
  13 #include <d2d1_1.h>
  14 #include <cguid.h>
  15
  16 namespace mozilla {
  17 namespace gfx {
  18
  19 class FilterNodeD2D1 : public FilterNode
  20 {
  21 public:
  22   MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(FilterNodeD2D1, override)
  23
  24   static already_AddRefed<FilterNode> Create(ID2D1DeviceContext *aDC, FilterType aType);
  25
  26   FilterNodeD2D1(ID2D1Effect *aEffect, FilterType aType)
  27     : mEffect(aEffect)
  28     , mType(aType)
  29   {
  30     InitUnmappedProperties();
  31   }
  32
  33   virtual FilterBackend GetBackendType() { return FILTER_BACKEND_DIRECT2D1_1; }
  34
  35   virtual void SetInput(uint32_t aIndex, SourceSurface *aSurface);
  36   virtual void SetInput(uint32_t aIndex, FilterNode *aFilter);
  37
  38   virtual void SetAttribute(uint32_t aIndex, uint32_t aValue);
  39   virtual void SetAttribute(uint32_t aIndex, Float aValue);
  40   virtual void SetAttribute(uint32_t aIndex, const Point &aValue);
  41   virtual void SetAttribute(uint32_t aIndex, const Matrix5x4 &aValue);
  42   virtual void SetAttribute(uint32_t aIndex, const Point3D &aValue);
  43   virtual void SetAttribute(uint32_t aIndex, const Size &aValue);
  44   virtual void SetAttribute(uint32_t aIndex, const IntSize &aValue);
  45   virtual void SetAttribute(uint32_t aIndex, const Color &aValue);
  46   virtual void SetAttribute(uint32_t aIndex, const Rect &aValue);
  47   virtual void SetAttribute(uint32_t aIndex, const IntRect &aValue);
  48   virtual void SetAttribute(uint32_t aIndex, bool aValue);
  49   virtual void SetAttribute(uint32_t aIndex, const Float *aValues, uint32_t aSize);
  50   virtual void SetAttribute(uint32_t aIndex, const IntPoint &aValue);
  51   virtual void SetAttribute(uint32_t aIndex, const Matrix &aValue);
  52
  53   // Called by DrawTarget before it draws our OutputEffect, and recursively
  54   // by the filter nodes that have this filter as one of their inputs. This
  55   // gives us a chance to convert any input surfaces to the target format for
  56   // the DrawTarget that we will draw to.
  57   virtual void WillDraw(DrawTarget *aDT);
  58
  59   virtual ID2D1Effect* MainEffect() { return mEffect.get(); }
  60   virtual ID2D1Effect* InputEffect() { return mEffect.get(); }
  61   virtual ID2D1Effect* OutputEffect() { return mEffect.get(); }
  62
  63 protected:
  64   friend class DrawTargetD2D1;
  65   friend class DrawTargetD2D;
  66   friend class FilterNodeConvolveD2D1;
  67
  68   void InitUnmappedProperties();
  69
  70   RefPtr<ID2D1Effect> mEffect;
  71   std::vector<RefPtr<FilterNodeD2D1>> mInputFilters;
  72   std::vector<RefPtr<SourceSurface>> mInputSurfaces;
  73   FilterType mType;
  74
  75 private:
  76   using FilterNode::SetAttribute;
  77   using FilterNode::SetInput;
  78 };
  79
  80 class FilterNodeConvolveD2D1 : public FilterNodeD2D1
  81 {
  82 public:
  83   MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(FilterNodeConvolveD2D1, override)
  84   explicit FilterNodeConvolveD2D1(ID2D1DeviceContext *aDC);
  85
  86   virtual void SetInput(uint32_t aIndex, FilterNode *aFilter) override;
  87
  88   virtual void SetAttribute(uint32_t aIndex, uint32_t aValue) override;
  89   virtual void SetAttribute(uint32_t aIndex, const IntSize &aValue) override;
  90   virtual void SetAttribute(uint32_t aIndex, const IntPoint &aValue) override;
  91   virtual void SetAttribute(uint32_t aIndex, const IntRect &aValue) override;
  92
  93   virtual ID2D1Effect* InputEffect() override;
  94
  95 private:
  96   using FilterNode::SetAttribute;
  97   using FilterNode::SetInput;
  98
  99   void UpdateChain();
 100   void UpdateOffset();
 101   void UpdateSourceRect();
 102
 103   RefPtr<ID2D1Effect> mExtendInputEffect;
 104   RefPtr<ID2D1Effect> mBorderEffect;
 105   ConvolveMatrixEdgeMode mEdgeMode;
 106   IntPoint mTarget;
 107   IntSize mKernelSize;
 108   IntRect mSourceRect;
 109 };
 110
 111 class FilterNodeExtendInputAdapterD2D1 : public FilterNodeD2D1
 112 {
 113 public:
 114   MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(FilterNodeExtendInputAdapterD2D1, override)
 115   FilterNodeExtendInputAdapterD2D1(ID2D1DeviceContext *aDC, FilterNodeD2D1 *aFilterNode, FilterType aType);
 116
 117   virtual ID2D1Effect* InputEffect() override { return mExtendInputEffect.get(); }
 118   virtual ID2D1Effect* OutputEffect() override { return mWrappedFilterNode->OutputEffect(); }
 119
 120 private:
 121   RefPtr<FilterNodeD2D1> mWrappedFilterNode;
 122   RefPtr<ID2D1Effect> mExtendInputEffect;
 123 };
 124
 125 class FilterNodePremultiplyAdapterD2D1 : public FilterNodeD2D1
 126 {
 127 public:
 128   MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(FilterNodePremultiplyAdapterD2D1, override)
 129   FilterNodePremultiplyAdapterD2D1(ID2D1DeviceContext *aDC, FilterNodeD2D1 *aFilterNode, FilterType aType);
 130
 131   virtual ID2D1Effect* InputEffect() override { return mPrePremultiplyEffect.get(); }
 132   virtual ID2D1Effect* OutputEffect() override { return mPostUnpremultiplyEffect.get(); }
 133
 134 private:
 135   RefPtr<ID2D1Effect> mPrePremultiplyEffect;
 136   RefPtr<ID2D1Effect> mPostUnpremultiplyEffect;
 137 };
 138
 139 }
 140 }
 141
 142 #endif