[import] support document.stylesheets in imported documents.

Source/core/dom/DocumentStyleSheetCollector.h

 /*
  * Copyright (C) 2014 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Neither the name of Google Inc. nor the names of its
@@ skipping 10 matching lines @@
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #ifndef DocumentStyleSheetCollector_h
 #define DocumentStyleSheetCollector_h
 
 #include "core/dom/StyleSheetCollection.h"
-#include "wtf/HashSet.h"
+#include "wtf/HashMap.h"
 
 namespace WebCore {
 
 class DocumentStyleSheetCollector FINAL {
 public:
-    DocumentStyleSheetCollector(TreeScope& root)
-        : m_root(root)
-    { }
+    typedef Vector<RefPtr<StyleSheet> > SheetList;
+    typedef HashMap<Document*, SheetList> SheetListMap;
+    typedef HashMap<Document*, SheetList>::const_iterator SheetListMapIteraor;
 
-    bool isCollectingForList(TreeScope&) const;
+    DocumentStyleSheetCollector(TreeScope& root);
+    ~DocumentStyleSheetCollector();
+
     void setCollectionTo(StyleSheetCollectionBase&);
 
     void appendActiveStyleSheets(const Vector<RefPtr<CSSStyleSheet> >& sheets)  { m_collection.appendActiveStyleSheets(sheets); }
     void appendActiveStyleSheet(CSSStyleSheet* sheet) { m_collection.appendActiveStyleSheet(sheet); }
-    void appendSheetForList(StyleSheet* sheet) { m_collection.appendSheetForList(sheet); }
+    void appendSheetForList(StyleSheet*);
     StyleSheetCollectionBase& collection() { return m_collection; }
-    void markVisited(Document* document) { m_documentsVisited.add(document); }
-    bool hasVisited(Document* document) const { return m_documentsVisited.contains(document); }
+    void markVisited(Document*);
+    bool hasVisited(Document*) const;
+
+    const SheetListMap& sheetListMap() const { return m_sheetListMap; }
 
 private:
     TreeScope& m_root;
     StyleSheetCollectionBase m_collection;
-    HashSet<Document*> m_documentsVisited;
+    SheetListMap m_sheetListMap;
 };
 
 } // namespace WebCore
 
 #endif // DocumentStyleSheetCollector_h