Create a super rough prototype of B3
[WebKit-https.git] / Source / JavaScriptCore / b3 / B3SuccessorCollection.h
1 /*
2  * Copyright (C) 2015 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. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #ifndef B3SuccessorCollection_h
27 #define B3SuccessorCollection_h
28
29 #if ENABLE(B3_JIT)
30
31 namespace JSC { namespace B3 {
32
33 // This is a generic wrapper around lists of frequented blocks, which gives you just the blocks.
34
35 template<typename BasicBlock, typename SuccessorList>
36 class SuccessorCollection {
37 public:
38     SuccessorCollection(SuccessorList& list)
39         : m_list(list)
40     {
41     }
42
43     size_t size() const { return m_list.size(); }
44     BasicBlock* at(size_t index) const { return m_list[index].block(); }
45     BasicBlock*& at(size_t index) { return m_list[index].block(); }
46     BasicBlock* operator[](size_t index) const { return at(index); }
47     BasicBlock*& operator[](size_t index) { return at(index); }
48
49     class iterator {
50     public:
51         iterator()
52             : m_collection(nullptr)
53             , m_index(0)
54         {
55         }
56
57         iterator(SuccessorCollection& collection, size_t index)
58             : m_collection(&collection)
59             , m_index(index)
60         {
61         }
62
63         BasicBlock*& operator*() const
64         {
65             return m_collection->at(m_index);
66         }
67
68         iterator& operator++()
69         {
70             m_index++;
71             return *this;
72         }
73
74         bool operator==(const iterator& other) const
75         {
76             ASSERT(m_collection == other.m_collection);
77             return m_index == other.m_index;
78         }
79
80         bool operator!=(const iterator& other) const
81         {
82             return !(*this == other);
83         }
84
85     private:
86         SuccessorCollection* m_collection;
87         size_t m_index;
88     };
89
90     iterator begin() { return iterator(*this, 0); }
91     iterator end() { return iterator(*this, size()); }
92
93 private:
94     SuccessorList& m_list;
95 };
96
97 } } // namespace JSC::B3
98
99 #endif // ENABLE(B3_JIT)
100
101 #endif // B3SuccessorCollection_h
102