[import] support document.stylesheets in imported documents.

Source/core/dom/DocumentStyleSheetCollector.h

Index: Source/core/dom/DocumentStyleSheetCollector.h
diff --git a/Source/core/dom/DocumentStyleSheetCollector.h b/Source/core/dom/DocumentStyleSheetCollector.h
index e6b3c04094a88eba30501c0ce8d14b1e88b52f13..e538059b2210d08fd28fd11569b5302cab4b5183 100644
--- a/Source/core/dom/DocumentStyleSheetCollector.h
+++ b/Source/core/dom/DocumentStyleSheetCollector.h
@@ -28,30 +28,34 @@
 #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;
+
+    DocumentStyleSheetCollector(TreeScope& root);
+    ~DocumentStyleSheetCollector();
 
-    bool isCollectingForList(TreeScope&) const;
     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