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
35namespace blink {
36
37class CSSStyleSheet;
38class Document;
39class StyleSheet;
40class StyleSheetCollection;
41
42class 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();
47public:
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
60private:
61    WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >& m_styleSheetsForStyleSheetList;
62    WillBeHeapVector<RefPtrWillBeMember<CSSStyleSheet> >& m_activeAuthorStyleSheets;
63    WillBeHeapHashSet<RawPtrWillBeMember<Document> >& m_visitedDocuments;
64};
65
66class ActiveDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector {
67public:
68    ActiveDocumentStyleSheetCollector(StyleSheetCollection&);
69private:
70    WillBeHeapHashSet<RawPtrWillBeMember<Document> > m_visitedDocuments;
71};
72
73class ImportedDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector {
74public:
75    ImportedDocumentStyleSheetCollector(DocumentStyleSheetCollector&, WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >&);
76};
77
78} // namespace blink
79
80#endif // DocumentStyleSheetCollector_h
81