Move HTMLCollection's id / name cache to a new NamedItemCache class

Source/core/html/HTMLCollection.h

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  * Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2011, 2012 Apple Inc. All rights reserved.
  * Copyright (C) 2014 Samsung Electronics. 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 47 matching lines @@
 
     // CollectionIndexCache API.
     bool canTraverseBackward() const { return !overridesItemAfter(); }
     Element* itemBefore(const Element* previousItem) const;
     Element* traverseToFirstElement(const ContainerNode& root) const;
     Element* traverseForwardToOffset(unsigned offset, Element& currentElement, unsigned& currentOffset, const ContainerNode& root) const;
 
 protected:
     HTMLCollection(ContainerNode& base, CollectionType, ItemAfterOverrideType);
 
+    class NamedItemCache {
+    public:
+        Vector<Element*>* getElementsById(const AtomicString& id) const { return m_idCache.get(id.impl()); }
+        Vector<Element*>* getElementsByName(const AtomicString& name) const { return m_nameCache.get(name.impl()); }
+        void addElementWithId(const AtomicString& id, Element* element) { addElementToMap(m_idCache, id, element); }
+        void addElementWithName(const AtomicString& name, Element* element) { addElementToMap(m_nameCache, name, element); }
+
+    private:
+        typedef HashMap<StringImpl*, OwnPtr<Vector<Element*> > > StringToElementsMap;
+        static void addElementToMap(StringToElementsMap& map, const AtomicString& key, Element* element)
+        {
+            OwnPtr<Vector<Element*> >& vector = map.add(key.impl(), nullptr).storedValue->value;
+            if (!vector)
+                vector = adoptPtr(new Vector<Element*>);
+            vector->append(element);
+        }
+
+        StringToElementsMap m_idCache;
+        StringToElementsMap m_nameCache;
+    };
+
     bool overridesItemAfter() const { return m_overridesItemAfter; }
     virtual Element* virtualItemAfter(Element*) const;
     bool shouldOnlyIncludeDirectChildren() const { return m_shouldOnlyIncludeDirectChildren; }
     virtual void supportedPropertyNames(Vector<String>& names);
 
     virtual void updateIdNameCache() const;
-    bool hasValidIdNameCache() const { return m_hasValidIdNameCache; }
-    void setHasValidIdNameCache() const
+    bool hasValidIdNameCache() const { return m_namedItemCache; }
+
+    NamedItemCache& createNamedItemCache() const
     {
-        ASSERT(!m_hasValidIdNameCache);
-        m_hasValidIdNameCache = true;
+        ASSERT(!m_namedItemCache);
         document().incrementNodeListWithIdNameCacheCount();
+        m_namedItemCache = adoptPtr(new NamedItemCache);
+        return *m_namedItemCache;
     }
-
-    typedef HashMap<StringImpl*, OwnPtr<Vector<Element*> > > NodeCacheMap;
-    Vector<Element*>* idCache(const AtomicString& name) const { return m_idCache.get(name.impl()); }
-    Vector<Element*>* nameCache(const AtomicString& name) const { return m_nameCache.get(name.impl()); }
-    void appendIdCache(const AtomicString& name, Element* element) const { append(m_idCache, name, element); }
-    void appendNameCache(const AtomicString& name, Element* element) const { append(m_nameCache, name, element); }
+    NamedItemCache& namedItemCache() const
+    {
+        ASSERT(m_namedItemCache);
+        return *m_namedItemCache;
+    }
 
 private:
     Element* traverseNextElement(Element& previous, const ContainerNode& root) const;
 
-    static void append(NodeCacheMap&, const AtomicString&, Element*);
     void invalidateIdNameCacheMaps(Document* oldDocument = 0) const
     {
-        if (!m_hasValidIdNameCache)
+        if (!hasValidIdNameCache())
             return;
 
         // Make sure we decrement the NodeListWithIdNameCache count from
         // the old document instead of the new one in the case the collection
         // is moved to a new document.
         unregisterIdNameCacheFromDocument(oldDocument ? *oldDocument : document());
 
-        m_idCache.clear();
-        m_nameCache.clear();
-        m_hasValidIdNameCache = false;
+        m_namedItemCache.clear();
     }
 
     void unregisterIdNameCacheFromDocument(Document& document) const
     {
-        ASSERT(m_hasValidIdNameCache);
+        ASSERT(hasValidIdNameCache());
         document.decrementNodeListWithIdNameCacheCount();
     }
 
     const unsigned m_overridesItemAfter : 1;
     const unsigned m_shouldOnlyIncludeDirectChildren : 1;
-    mutable unsigned m_hasValidIdNameCache : 1;
-    mutable NodeCacheMap m_idCache;
-    mutable NodeCacheMap m_nameCache;
+    mutable OwnPtr<NamedItemCache> m_namedItemCache;
     mutable CollectionIndexCache<HTMLCollection, Element> m_collectionIndexCache;
 
     friend class LiveNodeListBase;
 };
 
 } // namespace
 
 #endif