Source/WebCore/platform/graphics/filters/FEDisplacementMap.cpp

   1 /*
   2  * Copyright (C) 2004, 2005, 2006, 2007 Nikolas Zimmermann <zimmermann@kde.org>
   3  * Copyright (C) 2004, 2005 Rob Buis <buis@kde.org>
   4  * Copyright (C) 2005 Eric Seidel <eric@webkit.org>
   5  * Copyright (C) 2009 Dirk Schulze <krit@webkit.org>
   6  * Copyright (C) Research In Motion Limited 2010. All rights reserved.
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Library General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Library General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Library General Public License
  19  * along with this library; see the file COPYING.LIB.  If not, write to
  20  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  21  * Boston, MA 02110-1301, USA.
  22  */
  23
  24 #include "config.h"
  25
  26 #if ENABLE(FILTERS)
  27 #include "FEDisplacementMap.h"
  28
  29 #include "Filter.h"
  30 #include "GraphicsContext.h"
  31 #include "RenderTreeAsText.h"
  32 #include "TextStream.h"
  33
  34 #include <wtf/ByteArray.h>
  35
  36 namespace WebCore {
  37
  38 FEDisplacementMap::FEDisplacementMap(Filter* filter, ChannelSelectorType xChannelSelector, ChannelSelectorType yChannelSelector, float scale)
  39     : FilterEffect(filter)
  40     , m_xChannelSelector(xChannelSelector)
  41     , m_yChannelSelector(yChannelSelector)
  42     , m_scale(scale)
  43 {
  44 }
  45
  46 PassRefPtr<FEDisplacementMap> FEDisplacementMap::create(Filter* filter, ChannelSelectorType xChannelSelector,
  47     ChannelSelectorType yChannelSelector, float scale)
  48 {
  49     return adoptRef(new FEDisplacementMap(filter, xChannelSelector, yChannelSelector, scale));
  50 }
  51
  52 ChannelSelectorType FEDisplacementMap::xChannelSelector() const
  53 {
  54     return m_xChannelSelector;
  55 }
  56
  57 bool FEDisplacementMap::setXChannelSelector(const ChannelSelectorType xChannelSelector)
  58 {
  59     if (m_xChannelSelector == xChannelSelector)
  60         return false;
  61     m_xChannelSelector = xChannelSelector;
  62     return true;
  63 }
  64
  65 ChannelSelectorType FEDisplacementMap::yChannelSelector() const
  66 {
  67     return m_yChannelSelector;
  68 }
  69
  70 bool FEDisplacementMap::setYChannelSelector(const ChannelSelectorType yChannelSelector)
  71 {
  72     if (m_yChannelSelector == yChannelSelector)
  73         return false;
  74     m_yChannelSelector = yChannelSelector;
  75     return true;
  76 }
  77
  78 float FEDisplacementMap::scale() const
  79 {
  80     return m_scale;
  81 }
  82
  83 bool FEDisplacementMap::setScale(float scale)
  84 {
  85     if (m_scale == scale)
  86         return false;
  87     m_scale = scale;
  88     return true;
  89 }
  90
  91 void FEDisplacementMap::platformApplySoftware()
  92 {
  93     FilterEffect* in = inputEffect(0);
  94     FilterEffect* in2 = inputEffect(1);
  95
  96     ASSERT(m_xChannelSelector != CHANNEL_UNKNOWN);
  97     ASSERT(m_yChannelSelector != CHANNEL_UNKNOWN);
  98
  99     ByteArray* dstPixelArray = createPremultipliedImageResult();
 100     if (!dstPixelArray)
 101         return;
 102
 103     IntRect effectADrawingRect = requestedRegionOfInputImageData(in->absolutePaintRect());
 104     RefPtr<ByteArray> srcPixelArrayA = in->asPremultipliedImage(effectADrawingRect);
 105
 106     IntRect effectBDrawingRect = requestedRegionOfInputImageData(in2->absolutePaintRect());
 107     RefPtr<ByteArray> srcPixelArrayB = in2->asUnmultipliedImage(effectBDrawingRect);
 108
 109     ASSERT(srcPixelArrayA->length() == srcPixelArrayB->length());
 110
 111     Filter* filter = this->filter();
 112     IntSize paintSize = absolutePaintRect().size();
 113     float scaleX = filter->applyHorizontalScale(m_scale / 255);
 114     float scaleY = filter->applyVerticalScale(m_scale / 255);
 115     float scaleAdjustmentX = filter->applyHorizontalScale(0.5f - 0.5f * m_scale);
 116     float scaleAdjustmentY = filter->applyVerticalScale(0.5f - 0.5f * m_scale);
 117     int stride = paintSize.width() * 4;
 118     for (int y = 0; y < paintSize.height(); ++y) {
 119         int line = y * stride;
 120         for (int x = 0; x < paintSize.width(); ++x) {
 121             int dstIndex = line + x * 4;
 122             int srcX = x + static_cast<int>(scaleX * srcPixelArrayB->get(dstIndex + m_xChannelSelector - 1) + scaleAdjustmentX);
 123             int srcY = y + static_cast<int>(scaleY * srcPixelArrayB->get(dstIndex + m_yChannelSelector - 1) + scaleAdjustmentY);
 124             for (unsigned channel = 0; channel < 4; ++channel) {
 125                 if (srcX < 0 || srcX >= paintSize.width() || srcY < 0 || srcY >= paintSize.height())
 126                     dstPixelArray->set(dstIndex + channel, static_cast<unsigned char>(0));
 127                 else {
 128                     unsigned char pixelValue = srcPixelArrayA->get(srcY * stride + srcX * 4 + channel);
 129                     dstPixelArray->set(dstIndex + channel, pixelValue);
 130                 }
 131             }
 132         }
 133     }
 134 }
 135
 136 void FEDisplacementMap::dump()
 137 {
 138 }
 139
 140 static TextStream& operator<<(TextStream& ts, const ChannelSelectorType& type)
 141 {
 142     switch (type) {
 143     case CHANNEL_UNKNOWN:
 144         ts << "UNKNOWN";
 145         break;
 146     case CHANNEL_R:
 147         ts << "RED";
 148         break;
 149     case CHANNEL_G:
 150         ts << "GREEN";
 151         break;
 152     case CHANNEL_B:
 153         ts << "BLUE";
 154         break;
 155     case CHANNEL_A:
 156         ts << "ALPHA";
 157         break;
 158     }
 159     return ts;
 160 }
 161
 162 TextStream& FEDisplacementMap::externalRepresentation(TextStream& ts, int indent) const
 163 {
 164     writeIndent(ts, indent);
 165     ts << "[feDisplacementMap";
 166     FilterEffect::externalRepresentation(ts);
 167     ts << " scale=\"" << m_scale << "\" "
 168        << "xChannelSelector=\"" << m_xChannelSelector << "\" "
 169        << "yChannelSelector=\"" << m_yChannelSelector << "\"]\n";
 170     inputEffect(0)->externalRepresentation(ts, indent + 1);
 171     inputEffect(1)->externalRepresentation(ts, indent + 1);
 172     return ts;
 173 }
 174
 175 } // namespace WebCore
 176
 177 #endif // ENABLE(FILTERS)