Home | History | Annotate | Download | only in dom 
      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
     81