Extend PropertyHandle to include custom property identifiers

third_party/WebKit/Source/core/animation/StringKeyframe.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef StringKeyframe_h
 #define StringKeyframe_h
 
 #include "core/animation/Keyframe.h"
 #include "core/css/StylePropertySet.h"
 
 #include "wtf/HashMap.h"
 
 namespace blink {
 
 class StyleSheetContents;
 
 // A specialization of Keyframe that associates user specified keyframe properties with either CSS properties or SVG attributes.
 class StringKeyframe : public Keyframe {
 public:
     static PassRefPtr<StringKeyframe> create()
     {
         return adoptRef(new StringKeyframe);
     }
 
+    void setCSSPropertyValue(const AtomicString& propertyName, const String& value, StyleSheetContents*);
     void setCSSPropertyValue(CSSPropertyID, const String& value, StyleSheetContents*);
     void setCSSPropertyValue(CSSPropertyID, const CSSValue&);
     void setPresentationAttributeValue(CSSPropertyID, const String& value, StyleSheetContents*);
     void setSVGAttributeValue(const QualifiedName&, const String& value);
 
-    const CSSValue& cssPropertyValue(CSSPropertyID property) const
+    const CSSValue& cssPropertyValue(const PropertyHandle& property) const
     {
-        int index = m_cssPropertyMap->findPropertyIndex(property);
-        RELEASE_ASSERT(index >= 0);
+        int index = -1;
+        if (property.isCSSCustomProperty())
+            index = m_cssPropertyMap->findPropertyIndex(property.customPropertyName());
+        else
+            index = m_cssPropertyMap->findPropertyIndex(property.cssProperty());
+        CHECK_GE(index, 0);
         return m_cssPropertyMap->propertyAt(static_cast<unsigned>(index)).value();
     }
 
     const CSSValue& presentationAttributeValue(CSSPropertyID property) const
     {
         int index = m_presentationAttributeMap->findPropertyIndex(property);
-        RELEASE_ASSERT(index >= 0);
+        CHECK_GE(index, 0);
         return m_presentationAttributeMap->propertyAt(static_cast<unsigned>(index)).value();
     }
 
     String svgPropertyValue(const QualifiedName& attributeName) const
     {
         return m_svgAttributeMap.get(&attributeName);
     }
 
     PropertyHandleSet properties() const override;
 
@@ skipping 77 matching lines @@
 using CSSPropertySpecificKeyframe = StringKeyframe::CSSPropertySpecificKeyframe;
 using SVGPropertySpecificKeyframe = StringKeyframe::SVGPropertySpecificKeyframe;
 
 DEFINE_TYPE_CASTS(StringKeyframe, Keyframe, value, value->isStringKeyframe(), value.isStringKeyframe());
 DEFINE_TYPE_CASTS(CSSPropertySpecificKeyframe, Keyframe::PropertySpecificKeyframe, value, value->isCSSPropertySpecificKeyframe(), value.isCSSPropertySpecificKeyframe());
 DEFINE_TYPE_CASTS(SVGPropertySpecificKeyframe, Keyframe::PropertySpecificKeyframe, value, value->isSVGPropertySpecificKeyframe(), value.isSVGPropertySpecificKeyframe());
 
 } // namespace blink
 
 #endif