Cache element indices for :nth-child and :nth-last-child selectors.

Source/core/dom/NthIndexCache.cpp

+// Copyright 2015 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.
+
+#include "config.h"
+#include "core/dom/NthIndexCache.h"
+
+#include "core/dom/Document.h"
+
+namespace blink {
+
+NthIndexCache::NthIndexCache(Document& document)
+    : m_document(&document)
+    , m_domTreeVersion(document.domTreeVersion())
+{
+    document.setNthIndexCache(this);
+}
+
+NthIndexCache::~NthIndexCache()
+{
+    ASSERT(m_domTreeVersion == m_document->domTreeVersion());
+    m_document->setNthIndexCache(nullptr);
+}
+
+NthIndexCache::NthIndexData& NthIndexCache::ensureNthIndexDataFor(Node& parent)
+{
+    if (!m_parentMap)
+        m_parentMap = adoptPtrWillBeNoop(new ParentMap());
+
+    ParentMap::AddResult addResult = m_parentMap->add(&parent, nullptr);
+    if (addResult.isNewEntry)
+        addResult.storedValue->value = adoptPtrWillBeNoop(new NthIndexData());
+
+    ASSERT(addResult.storedValue->value);
+    return *addResult.storedValue->value;
+}
+
+unsigned NthIndexCache::NthIndexData::cacheNthIndices(Element& element)
+{
+    unsigned index = 0;
+    // The frequency at which we cache the nth-index for a set of siblings.
+    // A spread value of 3 means every third Element will have its nth-index cached.
+    // Using a spread value > 1 is done to save memory. Looking up the nth-index will
+    // still be done in constant time in terms of sibling count, at most 'spread'
+    // elements will be traversed.
+    const unsigned spread = 3;
+    unsigned count = 0;
+    for (Element* sibling = ElementTraversal::firstChild(*element.parentNode()); sibling; sibling = ElementTraversal::nextSibling(*sibling)) {
+        if (!(++count % spread))
+            m_elementIndexMap.add(sibling, count);
+        if (sibling == &element)
+            index = count;
+    }
+    ASSERT(count && index);
+    m_count = count;
+    return index;
+}
+
+DEFINE_TRACE(NthIndexCache::NthIndexData)
+{
+#if ENABLE(OILPAN)
+    visitor->trace(m_elementIndexMap);
+#endif
+}
+
+#if !ENABLE(OILPAN)
+NthIndexCache::NthIndexData::~NthIndexData()
+{
+}
+#endif
+
+} // namespace blink