dfg/DFGBlockSet.h

   1 /*
   2  * Copyright (C) 2014 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 DFGBlockSet_h
  27 #define DFGBlockSet_h
  28
  29 #if ENABLE(DFG_JIT)
  30
  31 #include "DFGBasicBlock.h"
  32
  33 namespace JSC { namespace DFG {
  34
  35 class Graph;
  36
  37 class BlockSet {
  38 public:
  39     BlockSet() { }
  40
  41     // Return true if the block was added, false if it was already present.
  42     bool add(BasicBlock* block)
  43     {
  44         return !m_set.set(block->index);
  45     }
  46
  47     bool contains(BasicBlock* block) const
  48     {
  49         if (!block)
  50             return false;
  51         return m_set.get(block->index);
  52     }
  53
  54     class iterator {
  55     public:
  56         iterator()
  57             : m_graph(nullptr)
  58             , m_set(nullptr)
  59             , m_index(0)
  60         {
  61         }
  62
  63         iterator& operator++()
  64         {
  65             m_index = m_set->m_set.findBit(m_index + 1, true);
  66             return *this;
  67         }
  68
  69         BasicBlock* operator*() const;
  70
  71         bool operator==(const iterator& other) const
  72         {
  73             return m_index == other.m_index;
  74         }
  75
  76         bool operator!=(const iterator& other) const
  77         {
  78             return !(*this == other);
  79         }
  80
  81     private:
  82         friend class BlockSet;
  83
  84         Graph* m_graph;
  85         const BlockSet* m_set;
  86         size_t m_index;
  87     };
  88
  89     class Iterable {
  90     public:
  91         Iterable(Graph& graph, const BlockSet& set)
  92             : m_graph(graph)
  93             , m_set(set)
  94         {
  95         }
  96
  97         iterator begin() const
  98         {
  99             iterator result;
 100             result.m_graph = &m_graph;
 101             result.m_set = &m_set;
 102             result.m_index = m_set.m_set.findBit(0, true);
 103             return result;
 104         }
 105
 106         iterator end() const
 107         {
 108             iterator result;
 109             result.m_graph = &m_graph;
 110             result.m_set = &m_set;
 111             result.m_index = m_set.m_set.size();
 112             return result;
 113         }
 114
 115     private:
 116         Graph& m_graph;
 117         const BlockSet& m_set;
 118     };
 119
 120     Iterable iterable(Graph& graph) const
 121     {
 122         return Iterable(graph, *this);
 123     }
 124
 125     void dump(PrintStream&) const;
 126
 127 private:
 128     BitVector m_set;
 129 };
 130
 131 class BlockAdder {
 132 public:
 133     BlockAdder(BlockSet& set)
 134         : m_set(set)
 135     {
 136     }
 137
 138     bool operator()(BasicBlock* block) const
 139     {
 140         return m_set.add(block);
 141     }
 142 private:
 143     BlockSet& m_set;
 144 };
 145
 146 } } // namespace JSC::DFG
 147
 148 #endif // ENABLE(DFG_JIT)
 149
 150 #endif // DFGBlockSet_h
 151