Add forEach() to CSSVariablesMap

Source/core/css/StylePropertySet.cpp

 /*
  * (C) 1999-2003 Lars Knoll (knoll@kde.org)
  * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2012, 2013 Apple Inc. All rights reserved.
  * Copyright (C) 2011 Research In Motion Limited. All rights reserved.
  * Copyright (C) 2013 Intel Corporation. All rights reserved.
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Library General Public
  * License as published by the Free Software Foundation; either
  * version 2 of the License, or (at your option) any later version.
@@ skipping 266 matching lines @@
     return indexOfShorthandForLonghand(prefixedShorthand, matchingShorthandsForLonghand(prefixingVariant));
 }
 
 bool MutableStylePropertySet::setVariableValue(const AtomicString& name, const String& value, bool important)
 {
     ASSERT(RuntimeEnabledFeatures::cssVariablesEnabled());
     if (value.isEmpty())
         return removeVariable(name);
 
     size_t index = findVariableIndex(name);
-    if (index != notFound) {
-        CSSValue* cssValue = m_propertyVector.at(index).value();
+    if (index == notFound) {
+        VariablesIterators* iterators = activeVariablesIterators();
+        if (iterators) {
+            VariablesIterators::iterator end = iterators->end();
+            for (VariablesIterators::iterator it = iterators->begin(); it != end; ++it)
+                (*it)->addedVariable(name);
+        }
+    } else {
+        const CSSValue* cssValue = m_propertyVector.at(index).value();
         if (toCSSVariableValue(cssValue)->value() == value)
             return false;
     }
 
     CSSProperty property(CSSPropertyVariable, CSSVariableValue::create(name, value), important);
     if (index == notFound)
         m_propertyVector.append(property);
     else
         m_propertyVector.at(index) = property;
     return true;
@@ skipping 246 matching lines @@
         removeProperty(propertiesToRemove[i]);
 }
 
 bool MutableStylePropertySet::removeVariable(const AtomicString& name)
 {
     ASSERT(RuntimeEnabledFeatures::cssVariablesEnabled());
     size_t index = findVariableIndex(name);
     if (index == notFound)
         return false;
     m_propertyVector.remove(index);
+    VariablesIterators* iterators = activeVariablesIterators();
+    if (iterators) {
+        VariablesIterators::iterator end = iterators->end();
+        for (VariablesIterators::iterator it = iterators->begin(); it != end; ++it)
+            (*it)->removedVariable(name);
+    }
     return true;
 }
 
 bool MutableStylePropertySet::clearVariables()
 {
     ASSERT(RuntimeEnabledFeatures::cssVariablesEnabled());
     CSSPropertyID variablesId = CSSPropertyVariable;
-    return removePropertiesInSet(&variablesId, 1);
+    bool removed = removePropertiesInSet(&variablesId, 1);
+    if (removed) {
+        VariablesIterators* iterators = activeVariablesIterators();
+        if (iterators) {
+            VariablesIterators::iterator end = iterators->end();
+            for (VariablesIterators::iterator it = iterators->begin(); it != end; ++it)
+                (*it)->clearedVariables();
+        }
+    }
+    return removed;
+}
+
+PassRefPtr<MutableStylePropertySet::VariablesIterator> MutableStylePropertySet::VariablesIterator::create(MutableStylePropertySet* propertySet)
+{
+    ASSERT(RuntimeEnabledFeatures::cssVariablesEnabled());
+    RefPtr<VariablesIterator> iterator = adoptRef(new VariablesIterator(propertySet));
+    const size_t propertyCount = propertySet->propertyCount();
+    size_t variableCount = 0;
+    iterator->m_remainingNames.resize(propertyCount);
+    for (int i = propertyCount; i--;) {
+        const PropertyReference& property = propertySet->propertyAt(i);
+        if (property.id() == CSSPropertyVariable)
+            iterator->m_remainingNames[variableCount++] = toCSSVariableValue(property.value())->name();
+    }
+    iterator->m_remainingNames.shrink(variableCount);
+
+    MutableStylePropertySet::VariablesIteratorsMap* map = MutableStylePropertySet::activeVariablesIteratorsMap();
+    if (map->contains(propertySet))
+        map->find(propertySet)->value.append(iterator);
+    else
+        map->add(propertySet, VariablesIterators()).iterator->value.append(iterator);
+
+    return iterator.release();
+}
+
+void MutableStylePropertySet::VariablesIterator::addedVariable(const AtomicString& name)
+{
+    ASSERT(!m_remainingNames.contains(name));
+    ASSERT(!m_newNames.contains(name));
+    m_newNames.append(name);
+}
+
+void MutableStylePropertySet::VariablesIterator::removedVariable(const AtomicString& name)
+{
+    size_t index = m_remainingNames.find(name);
+    if (index != notFound)
+        m_remainingNames.remove(index);
+    index = m_newNames.find(name);
+    if (index != notFound)
+        m_newNames.remove(index);
+}
+
+void MutableStylePropertySet::VariablesIterator::clearedVariables()
+{
+    m_remainingNames.clear();
+    m_newNames.clear();
+}
+
+void MutableStylePropertySet::VariablesIterator::advance()
+{
+    if (!atEnd())
+        m_remainingNames.removeLast();
+    if (!m_newNames.isEmpty()) {
+        m_remainingNames.appendVector(m_newNames);
+        m_newNames.clear();
+    }
+}
+
+MutableStylePropertySet::VariablesIterator::~VariablesIterator()
+{
+    MutableStylePropertySet::VariablesIterators& iterators = MutableStylePropertySet::activeVariablesIteratorsMap()->find(m_propertySet.get())->value;
+    iterators.remove(iterators.reverseFind(this));
+}
+
+MutableStylePropertySet::VariablesIterators* MutableStylePropertySet::activeVariablesIterators()
+{
+    VariablesIteratorsMap* map = activeVariablesIteratorsMap();
+    VariablesIteratorsMap::iterator it = map->find(this);
+    if (it == map->end())
+        return 0;
+    return &it->value;
+}
+
+MutableStylePropertySet::VariablesIteratorsMap* MutableStylePropertySet::activeVariablesIteratorsMap()
+{
+    DEFINE_STATIC_LOCAL(VariablesIteratorsMap, map, ());
+    return &map;
 }
 
 PassRefPtr<MutableStylePropertySet> StylePropertySet::mutableCopy() const
 {
     return adoptRef(new MutableStylePropertySet(*this));
 }
 
 PassRefPtr<MutableStylePropertySet> StylePropertySet::copyPropertiesInSet(const Vector<CSSPropertyID>& properties) const
 {
     Vector<CSSProperty, 256> list;
@@ skipping 54 matching lines @@
 PassRefPtr<MutableStylePropertySet> MutableStylePropertySet::create(CSSParserMode cssParserMode)
 {
     return adoptRef(new MutableStylePropertySet(cssParserMode));
 }
 
 PassRefPtr<MutableStylePropertySet> MutableStylePropertySet::create(const CSSProperty* properties, unsigned count)
 {
     return adoptRef(new MutableStylePropertySet(properties, count));
 }
 
+MutableStylePropertySet::~MutableStylePropertySet()
+{
+    activeVariablesIteratorsMap()->remove(this);
+}
+
 String StylePropertySet::PropertyReference::cssName() const
 {
     if (id() == CSSPropertyVariable) {
         if (!propertyValue()->isVariableValue())
             return emptyString(); // Should not happen, but if it does, avoid a bad cast.
         return "var-" + toCSSVariableValue(propertyValue())->name();
     }
     return getPropertyNameString(id());
 }
 
 String StylePropertySet::PropertyReference::cssText() const
 {
     StringBuilder result;
     result.append(cssName());
     result.appendLiteral(": ");
     result.append(propertyValue()->cssText());
     if (isImportant())
         result.appendLiteral(" !important");
     result.append(';');
     return result.toString();
 }
 
 
 } // namespace WebCore