f190989b2ce22ff7b35d305832227b1345b9a2ce
[WebKit-https.git] / Source / JavaScriptCore / heap / OpaqueRootSet.h
1 /*
2  * Copyright (C) 2014-2017 Apple 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
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #pragma once
27
28 #include <wtf/HashSet.h>
29
30 namespace JSC {
31
32 class OpaqueRootSet {
33     WTF_MAKE_NONCOPYABLE(OpaqueRootSet);
34 public:
35     OpaqueRootSet()
36         : m_lastQueriedRoot(nullptr)
37         , m_containsLastQueriedRoot(false)
38     {
39     }
40
41     bool contains(void* root) const
42     {
43         if (root != m_lastQueriedRoot) {
44             m_lastQueriedRoot = root;
45             m_containsLastQueriedRoot = m_roots.contains(root);
46         }
47         return m_containsLastQueriedRoot;
48     }
49
50     bool isEmpty() const
51     {
52         return m_roots.isEmpty();
53     }
54
55     void clear()
56     {
57         m_roots.clear();
58         m_lastQueriedRoot = nullptr;
59         m_containsLastQueriedRoot = false;
60     }
61
62     bool add(void* root)
63     {
64         if (root == m_lastQueriedRoot)
65             m_containsLastQueriedRoot = true;
66         return m_roots.add(root).isNewEntry;
67     }
68
69     int size() const
70     {
71         return m_roots.size();
72     }
73
74     HashSet<void*>::const_iterator begin() const
75     {
76         return m_roots.begin();
77     }
78
79     HashSet<void*>::const_iterator end() const
80     {
81         return m_roots.end();
82     }
83
84
85 private:
86     HashSet<void*> m_roots;
87     mutable void* m_lastQueriedRoot;
88     mutable bool m_containsLastQueriedRoot;
89 };
90
91 } // namespace JSC