addr_range_map.hh revision 9409:e399b6c18b76
1/*
2 * Copyright (c) 2012 ARM Limited
3 * All rights reserved
4 *
5 * The license below extends only to copyright in the software and shall
6 * not be construed as granting a license to any other intellectual
7 * property including but not limited to intellectual property relating
8 * to a hardware implementation of the functionality of the software
9 * licensed hereunder.  You may use the software subject to the license
10 * terms below provided that you ensure that this notice is replicated
11 * unmodified and in its entirety in all distributions of the software,
12 * modified or unmodified, in source code or in binary form.
13 *
14 * Copyright (c) 2006 The Regents of The University of Michigan
15 * All rights reserved.
16 *
17 * Redistribution and use in source and binary forms, with or without
18 * modification, are permitted provided that the following conditions are
19 * met: redistributions of source code must retain the above copyright
20 * notice, this list of conditions and the following disclaimer;
21 * redistributions in binary form must reproduce the above copyright
22 * notice, this list of conditions and the following disclaimer in the
23 * documentation and/or other materials provided with the distribution;
24 * neither the name of the copyright holders nor the names of its
25 * contributors may be used to endorse or promote products derived from
26 * this software without specific prior written permission.
27 *
28 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
29 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
30 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
31 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
32 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
33 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
34 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
35 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
36 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
37 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
38 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
39 *
40 * Authors: Ali Saidi
41 *          Andreas Hansson
42 */
43
44#ifndef __BASE_ADDR_RANGE_MAP_HH__
45#define __BASE_ADDR_RANGE_MAP_HH__
46
47#include <map>
48#include <utility>
49
50#include "base/addr_range.hh"
51
52/**
53 * The AddrRangeMap uses an STL map to implement an interval tree for
54 * address decoding. The value stored is a template type and can be
55 * e.g. a port identifier, or a pointer.
56 */
57template <typename V>
58class AddrRangeMap
59{
60  private:
61    typedef std::map<AddrRange, V> RangeMap;
62    RangeMap tree;
63
64  public:
65    typedef typename RangeMap::iterator iterator;
66    typedef typename RangeMap::const_iterator const_iterator;
67
68    const_iterator
69    find(const AddrRange &r) const
70    {
71        if (tree.empty())
72            return tree.end();
73
74        const_iterator i = tree.upper_bound(r);
75
76        if (i == tree.begin()) {
77            if (i->first.intersects(r))
78                return i;
79            else
80                // Nothing could match, so return end()
81                return tree.end();
82        }
83
84        --i;
85
86        if (i->first.intersects(r))
87            return i;
88
89        return tree.end();
90    }
91
92    const_iterator
93    find(const Addr &r) const
94    {
95        return find(RangeSize(r, 1));
96    }
97
98    bool
99    intersect(const AddrRange &r) const
100    {
101        return find(r) != tree.end();
102    }
103
104    const_iterator
105    insert(const AddrRange &r, const V& d)
106    {
107        if (intersect(r))
108            return tree.end();
109
110        return tree.insert(std::make_pair(r, d)).first;
111    }
112
113    void
114    erase(iterator p)
115    {
116        tree.erase(p);
117    }
118
119    void
120    erase(iterator p, iterator q)
121    {
122        tree.erase(p,q);
123    }
124
125    void
126    clear()
127    {
128        tree.erase(tree.begin(), tree.end());
129    }
130
131    const_iterator
132    begin() const
133    {
134        return tree.begin();
135    }
136
137    iterator
138    begin()
139    {
140        return tree.begin();
141    }
142
143    const_iterator
144    end() const
145    {
146        return tree.end();
147    }
148
149    iterator
150    end()
151    {
152        return tree.end();
153    }
154
155    std::size_t
156    size() const
157    {
158        return tree.size();
159    }
160
161    bool
162    empty() const
163    {
164        return tree.empty();
165    }
166};
167
168#endif //__BASE_ADDR_RANGE_MAP_HH__
169