src/third_party/WebKit/Source/core/dom/DocumentStyleSheetCollector.h

   1 /*
   2  * Copyright (C) 2014 Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  *     * Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *     * Neither the name of Google Inc. nor the names of its
  11  * contributors may be used to endorse or promote products derived from
  12  * this software without specific prior written permission.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  15  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  16  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  17  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  18  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  19  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  20  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #ifndef DocumentStyleSheetCollector_h
  28 #define DocumentStyleSheetCollector_h
  29
  30 #include "platform/heap/Handle.h"
  31 #include "wtf/HashSet.h"
  32 #include "wtf/RefPtr.h"
  33 #include "wtf/Vector.h"
  34
  35 namespace blink {
  36
  37 class CSSStyleSheet;
  38 class Document;
  39 class StyleSheet;
  40 class StyleSheetCollection;
  41
  42 class DocumentStyleSheetCollector {
  43     // This class contains references to two on-heap collections, therefore
  44     // it's unhealthy to have it anywhere but on the stack, where stack
  45     // scanning will keep them alive.
  46     STACK_ALLOCATED();
  47 public:
  48     friend class ImportedDocumentStyleSheetCollector;
  49
  50     DocumentStyleSheetCollector(WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >& sheetsForList, WillBeHeapVector<RefPtrWillBeMember<CSSStyleSheet> >& activeList, WillBeHeapHashSet<RawPtrWillBeMember<Document> >&);
  51     ~DocumentStyleSheetCollector();
  52
  53     void appendActiveStyleSheets(const WillBeHeapVector<RefPtrWillBeMember<CSSStyleSheet> >&);
  54     void appendActiveStyleSheet(CSSStyleSheet*);
  55     void appendSheetForList(StyleSheet*);
  56
  57     bool hasVisited(Document* document) const { return m_visitedDocuments.contains(document); }
  58     void willVisit(Document* document) { m_visitedDocuments.add(document); }
  59
  60 private:
  61     WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >& m_styleSheetsForStyleSheetList;
  62     WillBeHeapVector<RefPtrWillBeMember<CSSStyleSheet> >& m_activeAuthorStyleSheets;
  63     WillBeHeapHashSet<RawPtrWillBeMember<Document> >& m_visitedDocuments;
  64 };
  65
  66 class ActiveDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector {
  67 public:
  68     ActiveDocumentStyleSheetCollector(StyleSheetCollection&);
  69 private:
  70     WillBeHeapHashSet<RawPtrWillBeMember<Document> > m_visitedDocuments;
  71 };
  72
  73 class ImportedDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector {
  74 public:
  75     ImportedDocumentStyleSheetCollector(DocumentStyleSheetCollector&, WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >&);
  76 };
  77
  78 } // namespace blink
  79
  80 #endif // DocumentStyleSheetCollector_h