third_party/WebKit/Source/core/svg/SVGFEDisplacementMapElement.cpp

   1 /*
   2  * Copyright (C) 2006 Oliver Hunt <oliver@nerget.com>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public License
  15  * along with this library; see the file COPYING.LIB.  If not, write to
  16  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  17  * Boston, MA 02110-1301, USA.
  18  */
  19
  20 #include "config.h"
  21
  22 #include "core/svg/SVGFEDisplacementMapElement.h"
  23
  24 #include "core/SVGNames.h"
  25 #include "core/svg/graphics/filters/SVGFilterBuilder.h"
  26 #include "platform/graphics/filters/FilterEffect.h"
  27
  28 namespace blink {
  29
  30 template<> const SVGEnumerationStringEntries& getStaticStringEntries<ChannelSelectorType>()
  31 {
  32     DEFINE_STATIC_LOCAL(SVGEnumerationStringEntries, entries, ());
  33     if (entries.isEmpty()) {
  34         entries.append(std::make_pair(CHANNEL_R, "R"));
  35         entries.append(std::make_pair(CHANNEL_G, "G"));
  36         entries.append(std::make_pair(CHANNEL_B, "B"));
  37         entries.append(std::make_pair(CHANNEL_A, "A"));
  38     }
  39     return entries;
  40 }
  41
  42 inline SVGFEDisplacementMapElement::SVGFEDisplacementMapElement(Document& document)
  43     : SVGFilterPrimitiveStandardAttributes(SVGNames::feDisplacementMapTag, document)
  44     , m_scale(SVGAnimatedNumber::create(this, SVGNames::scaleAttr, SVGNumber::create(0)))
  45     , m_in1(SVGAnimatedString::create(this, SVGNames::inAttr, SVGString::create()))
  46     , m_in2(SVGAnimatedString::create(this, SVGNames::in2Attr, SVGString::create()))
  47     , m_xChannelSelector(SVGAnimatedEnumeration<ChannelSelectorType>::create(this, SVGNames::xChannelSelectorAttr, CHANNEL_A))
  48     , m_yChannelSelector(SVGAnimatedEnumeration<ChannelSelectorType>::create(this, SVGNames::yChannelSelectorAttr, CHANNEL_A))
  49 {
  50     addToPropertyMap(m_scale);
  51     addToPropertyMap(m_in1);
  52     addToPropertyMap(m_in2);
  53     addToPropertyMap(m_xChannelSelector);
  54     addToPropertyMap(m_yChannelSelector);
  55 }
  56
  57 DEFINE_TRACE(SVGFEDisplacementMapElement)
  58 {
  59     visitor->trace(m_scale);
  60     visitor->trace(m_in1);
  61     visitor->trace(m_in2);
  62     visitor->trace(m_xChannelSelector);
  63     visitor->trace(m_yChannelSelector);
  64     SVGFilterPrimitiveStandardAttributes::trace(visitor);
  65 }
  66
  67 DEFINE_NODE_FACTORY(SVGFEDisplacementMapElement)
  68
  69 bool SVGFEDisplacementMapElement::setFilterEffectAttribute(FilterEffect* effect, const QualifiedName& attrName)
  70 {
  71     FEDisplacementMap* displacementMap = static_cast<FEDisplacementMap*>(effect);
  72     if (attrName == SVGNames::xChannelSelectorAttr)
  73         return displacementMap->setXChannelSelector(m_xChannelSelector->currentValue()->enumValue());
  74     if (attrName == SVGNames::yChannelSelectorAttr)
  75         return displacementMap->setYChannelSelector(m_yChannelSelector->currentValue()->enumValue());
  76     if (attrName == SVGNames::scaleAttr)
  77         return displacementMap->setScale(m_scale->currentValue()->value());
  78
  79     ASSERT_NOT_REACHED();
  80     return false;
  81 }
  82
  83 void SVGFEDisplacementMapElement::svgAttributeChanged(const QualifiedName& attrName)
  84 {
  85     if (attrName == SVGNames::xChannelSelectorAttr || attrName == SVGNames::yChannelSelectorAttr || attrName == SVGNames::scaleAttr) {
  86         SVGElement::InvalidationGuard invalidationGuard(this);
  87         primitiveAttributeChanged(attrName);
  88         return;
  89     }
  90
  91     if (attrName == SVGNames::inAttr || attrName == SVGNames::in2Attr) {
  92         SVGElement::InvalidationGuard invalidationGuard(this);
  93         invalidate();
  94         return;
  95     }
  96
  97     SVGFilterPrimitiveStandardAttributes::svgAttributeChanged(attrName);
  98 }
  99
 100 PassRefPtrWillBeRawPtr<FilterEffect> SVGFEDisplacementMapElement::build(SVGFilterBuilder* filterBuilder, Filter* filter)
 101 {
 102     FilterEffect* input1 = filterBuilder->getEffectById(AtomicString(m_in1->currentValue()->value()));
 103     FilterEffect* input2 = filterBuilder->getEffectById(AtomicString(m_in2->currentValue()->value()));
 104
 105     if (!input1 || !input2)
 106         return nullptr;
 107
 108     RefPtrWillBeRawPtr<FilterEffect> effect = FEDisplacementMap::create(filter, m_xChannelSelector->currentValue()->enumValue(), m_yChannelSelector->currentValue()->enumValue(), m_scale->currentValue()->value());
 109     FilterEffectVector& inputEffects = effect->inputEffects();
 110     inputEffects.reserveCapacity(2);
 111     inputEffects.append(input1);
 112     inputEffects.append(input2);
 113     return effect.release();
 114 }
 115
 116 } // namespace blink