Remove global list map of TreeScope to StyleSheetCollection.

sky/engine/core/dom/StyleEngine.cpp

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  *           (C) 2001 Dirk Mueller (mueller@kde.org)
  *           (C) 2006 Alexey Proskuryakov (ap@webkit.org)
  * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2011, 2012 Apple Inc. All rights reserved.
  * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
  * Copyright (C) 2008, 2009, 2011, 2012 Google Inc. All rights reserved.
  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies)
  * Copyright (C) Research In Motion Limited 2010-2011. All rights reserved.
@@ skipping 14 matching lines @@
  * Boston, MA 02110-1301, USA.
  */
 
 #include "sky/engine/config.h"
 #include "sky/engine/core/dom/StyleEngine.h"
 
 #include "sky/engine/core/css/CSSFontSelector.h"
 #include "sky/engine/core/css/CSSStyleSheet.h"
 #include "sky/engine/core/css/FontFaceCache.h"
 #include "sky/engine/core/css/StyleSheetContents.h"
+#include "sky/engine/core/dom/Document.h"
 #include "sky/engine/core/dom/Element.h"
 #include "sky/engine/core/dom/ElementTraversal.h"
 #include "sky/engine/core/dom/StyleSheetCollection.h"
 #include "sky/engine/core/dom/shadow/ShadowRoot.h"
 #include "sky/engine/core/frame/Settings.h"
 #include "sky/engine/core/html/HTMLStyleElement.h"
 #include "sky/engine/core/html/imports/HTMLImportsController.h"
 #include "sky/engine/core/page/Page.h"
 
 namespace blink {
 
 StyleEngine::StyleEngine(Document& document)
     : m_document(&document)
-    , m_documentStyleSheetCollection(StyleSheetCollection::create(document))
     , m_ignorePendingStylesheets(false)
     // We don't need to create CSSFontSelector for imported document or
     // HTMLTemplateElement's document, because those documents have no frame.
     , m_fontSelector(document.frame() ? CSSFontSelector::create(&document) : nullptr)
 {
     if (m_fontSelector)
         m_fontSelector->registerForInvalidationCallbacks(this);
 }
 
 StyleEngine::~StyleEngine()
 {
 }
 
 void StyleEngine::detachFromDocument()
 {
     // Cleanup is performed eagerly when the StyleEngine is removed from the
     // document. The StyleEngine is unreachable after this, since only the
     // document has a reference to it.
 
     if (m_fontSelector) {
         m_fontSelector->clearDocument();
         m_fontSelector->unregisterForInvalidationCallbacks(this);
     }
 
     // Decrement reference counts for things we could be keeping alive.
     m_fontSelector.clear();
     m_resolver.clear();
-    m_styleSheetCollectionMap.clear();
-}
-
-const Vector<RefPtr<CSSStyleSheet>>& StyleEngine::activeAuthorStyleSheetsFor(TreeScope& treeScope)
-{
-    if (treeScope == m_document)
-        return documentStyleSheetCollection()->activeAuthorStyleSheets();
-    return ensureStyleSheetCollectionFor(treeScope)->activeAuthorStyleSheets();
-}
-
-StyleSheetCollection* StyleEngine::ensureStyleSheetCollectionFor(TreeScope& treeScope)
-{
-    if (treeScope == m_document)
-        return documentStyleSheetCollection();
-
-    StyleSheetCollectionMap::AddResult result = m_styleSheetCollectionMap.add(&treeScope, nullptr);
-    if (result.isNewEntry)
-        result.storedValue->value = StyleSheetCollection::create(treeScope);
-    return result.storedValue->value.get();
-}
-
-StyleSheetCollection* StyleEngine::styleSheetCollectionFor(TreeScope& treeScope)
-{
-    if (treeScope == m_document)
-        return documentStyleSheetCollection();
-
-    StyleSheetCollectionMap::iterator it = m_styleSheetCollectionMap.find(&treeScope);
-    if (it == m_styleSheetCollectionMap.end())
-        return 0;
-    return it->value.get();
 }
 
 void StyleEngine::addStyleSheetCandidateNode(Node* node, bool createdByParser)
 {
     if (!node->inDocument())
         return;
 
     TreeScope& treeScope = isHTMLStyleElement(*node) ? node->treeScope() : *m_document;
     ASSERT(isHTMLStyleElement(node) || treeScope == m_document);
-    StyleSheetCollection* collection = ensureStyleSheetCollectionFor(treeScope);
-    ASSERT(collection);
-    collection->addStyleSheetCandidateNode(node, createdByParser);
+    StyleSheetCollection& collection = treeScope.styleSheets();
+    collection.addStyleSheetCandidateNode(node, createdByParser);
 
     if (treeScope != m_document)
         m_activeTreeScopes.add(&treeScope);
 }
 
 void StyleEngine::removeStyleSheetCandidateNode(Node* node, ContainerNode* scopingNode, TreeScope& treeScope)
 {
     ASSERT(isHTMLStyleElement(node) || treeScope == m_document);
 
-    StyleSheetCollection* collection = styleSheetCollectionFor(treeScope);
-    ASSERT(collection);
-    collection->removeStyleSheetCandidateNode(node, scopingNode);
+    StyleSheetCollection& collection = treeScope.styleSheets();
+    collection.removeStyleSheetCandidateNode(node, scopingNode);
 
     m_activeTreeScopes.remove(&treeScope);
 }
 
 void StyleEngine::updateActiveStyleSheets()
 {
     ASSERT(!m_document->inStyleRecalc());
 
     if (!m_document->isActive())
         return;
 
-    documentStyleSheetCollection()->updateActiveStyleSheets(this);
+    // TODO(esprehn): Remove special case for document.
+    m_document->styleSheets().updateActiveStyleSheets(this);
 
     TreeScopeSet treeScopes = m_activeTreeScopes;
     HashSet<TreeScope*> treeScopesRemoved;
 
     for (TreeScopeSet::iterator it = treeScopes.begin(); it != treeScopes.end(); ++it) {
         TreeScope* treeScope = *it;
         ASSERT(treeScope != m_document);
-        StyleSheetCollection* collection = styleSheetCollectionFor(*treeScope);
-        ASSERT(collection);
-        collection->updateActiveStyleSheets(this);
-        if (!collection->hasStyleSheetCandidateNodes())
+        StyleSheetCollection& collection = treeScope->styleSheets();
+        collection.updateActiveStyleSheets(this);
+        if (!collection.hasStyleSheetCandidateNodes())
             treeScopesRemoved.add(treeScope);
     }
     m_activeTreeScopes.removeAll(treeScopesRemoved);
 }
 
-void StyleEngine::didRemoveShadowRoot(ShadowRoot* shadowRoot)
-{
-    m_styleSheetCollectionMap.remove(shadowRoot);
-}
-
 void StyleEngine::appendActiveAuthorStyleSheets()
 {
-    m_resolver->appendAuthorStyleSheets(documentStyleSheetCollection()->activeAuthorStyleSheets());
+    // TODO(esprehn): Remove special case for document.
+    m_resolver->appendAuthorStyleSheets(m_document->styleSheets().activeAuthorStyleSheets());
 
-    TreeScopeSet::iterator begin = m_activeTreeScopes.begin();
-    TreeScopeSet::iterator end = m_activeTreeScopes.end();
-    for (TreeScopeSet::iterator it = begin; it != end; ++it) {
-        if (StyleSheetCollection* collection = m_styleSheetCollectionMap.get(*it))
-            m_resolver->appendAuthorStyleSheets(collection->activeAuthorStyleSheets());
-    }
+    for (TreeScope* treeScope : m_activeTreeScopes)
+        m_resolver->appendAuthorStyleSheets(treeScope->styleSheets().activeAuthorStyleSheets());
+
     m_resolver->finishAppendAuthorStyleSheets();
 }
 
 void StyleEngine::createResolver()
 {
     // It is a programming error to attempt to resolve style on a Document
     // which is not in a frame. Code which hits this should have checked
     // Document::isActive() before calling into code which could get here.
 
     ASSERT(m_document->frame());
@@ skipping 116 matching lines @@
 }
 
 void StyleEngine::fontsNeedUpdate(CSSFontSelector*)
 {
     if (m_resolver)
         m_resolver->invalidateMatchedPropertiesCache();
     m_document->setNeedsStyleRecalc(SubtreeStyleChange);
 }
 
 }