addr_range_map.hh revision 12776
14120SN/A/* 25417SN/A * Copyright (c) 2012, 2018 ARM Limited 34120SN/A * All rights reserved 44120SN/A * 57087SN/A * The license below extends only to copyright in the software and shall 67087SN/A * not be construed as granting a license to any other intellectual 77087SN/A * property including but not limited to intellectual property relating 87087SN/A * to a hardware implementation of the functionality of the software 97087SN/A * licensed hereunder. You may use the software subject to the license 107087SN/A * terms below provided that you ensure that this notice is replicated 117087SN/A * unmodified and in its entirety in all distributions of the software, 127087SN/A * modified or unmodified, in source code or in binary form. 134120SN/A * 147087SN/A * Copyright (c) 2006 The Regents of The University of Michigan 157087SN/A * All rights reserved. 167087SN/A * 177087SN/A * Redistribution and use in source and binary forms, with or without 187087SN/A * modification, are permitted provided that the following conditions are 197087SN/A * met: redistributions of source code must retain the above copyright 207087SN/A * notice, this list of conditions and the following disclaimer; 217087SN/A * redistributions in binary form must reproduce the above copyright 224120SN/A * notice, this list of conditions and the following disclaimer in the 237087SN/A * documentation and/or other materials provided with the distribution; 244120SN/A * neither the name of the copyright holders nor the names of its 254120SN/A * contributors may be used to endorse or promote products derived from 264120SN/A * this software without specific prior written permission. 274120SN/A * 284120SN/A * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 294120SN/A * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 304120SN/A * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 314120SN/A * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 324120SN/A * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 334120SN/A * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 344120SN/A * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 354120SN/A * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 364120SN/A * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 374120SN/A * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 384120SN/A * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 394120SN/A * 408105Sgblack@eecs.umich.edu * Authors: Ali Saidi 418105Sgblack@eecs.umich.edu * Andreas Hansson 424120SN/A */ 434152SN/A 444152SN/A#ifndef __BASE_ADDR_RANGE_MAP_HH__ 454152SN/A#define __BASE_ADDR_RANGE_MAP_HH__ 464152SN/A 474152SN/A#include <map> 484152SN/A#include <utility> 499897Sandreas@sandberg.pp.se 507629SN/A#include "base/addr_range.hh" 515182SN/A 524152SN/A/** 534152SN/A * The AddrRangeMap uses an STL map to implement an interval tree for 544120SN/A * address decoding. The value stored is a template type and can be 554120SN/A * e.g. a port identifier, or a pointer. 564120SN/A */ 579180Sandreas.hansson@arm.comtemplate <typename V> 584152SN/Aclass AddrRangeMap 594152SN/A{ 609897Sandreas@sandberg.pp.se private: 619897Sandreas@sandberg.pp.se typedef std::map<AddrRange, V> RangeMap; 629897Sandreas@sandberg.pp.se 639897Sandreas@sandberg.pp.se public: 649897Sandreas@sandberg.pp.se typedef typename RangeMap::iterator iterator; 6513613Sgabeblack@google.com typedef typename RangeMap::const_iterator const_iterator; 6613613Sgabeblack@google.com 679897Sandreas@sandberg.pp.se /** 6813613Sgabeblack@google.com * Find entry that contains the given address range 699897Sandreas@sandberg.pp.se * 709897Sandreas@sandberg.pp.se * Searches through the ranges in the address map and returns an 719897Sandreas@sandberg.pp.se * iterator to the entry which range is a superset of the input 724152SN/A * address range. Returns end() if none found. 734152SN/A * 749180Sandreas.hansson@arm.com * @param r An input address range 754152SN/A * @return An iterator that contains the input address range 764152SN/A */ 779897Sandreas@sandberg.pp.se const_iterator 789897Sandreas@sandberg.pp.se contains(const AddrRange &r) const 799897Sandreas@sandberg.pp.se { 809897Sandreas@sandberg.pp.se return find(r, [r](const AddrRange r1) { return r.isSubset(r1); }); 819897Sandreas@sandberg.pp.se } 8213613Sgabeblack@google.com 8313613Sgabeblack@google.com /** 849897Sandreas@sandberg.pp.se * Find entry that contains the given address 8513613Sgabeblack@google.com * 869897Sandreas@sandberg.pp.se * Searches through the ranges in the address map and returns an 879897Sandreas@sandberg.pp.se * iterator to the entry which range is a superset of the input 889897Sandreas@sandberg.pp.se * address. Returns end() if none found. 899897Sandreas@sandberg.pp.se * 904152SN/A * @param r An input address 918902Sandreas.hansson@arm.com * @return An iterator that contains the input address 924120SN/A */ 938105Sgblack@eecs.umich.edu const_iterator 94 contains(Addr r) const 95 { 96 return contains(RangeSize(r, 1)); 97 } 98 99 /** 100 * Find entry that intersects with the given address range 101 * 102 * Searches through the ranges in the address map and returns an 103 * iterator to the first entry which range intersects with the 104 * input address. 105 * 106 * @param r An input address 107 * @return An iterator that intersects with the input address range 108 */ 109 const_iterator 110 intersects(const AddrRange &r) const 111 { 112 return find(r, [r](const AddrRange r1) { return r.intersects(r1); }); 113 } 114 115 const_iterator 116 insert(const AddrRange &r, const V& d) 117 { 118 if (intersects(r) != end()) 119 return tree.end(); 120 121 return tree.insert(std::make_pair(r, d)).first; 122 } 123 124 void 125 erase(iterator p) 126 { 127 tree.erase(p); 128 } 129 130 void 131 erase(iterator p, iterator q) 132 { 133 tree.erase(p,q); 134 } 135 136 void 137 clear() 138 { 139 tree.erase(tree.begin(), tree.end()); 140 } 141 142 const_iterator 143 begin() const 144 { 145 return tree.begin(); 146 } 147 148 iterator 149 begin() 150 { 151 return tree.begin(); 152 } 153 154 const_iterator 155 end() const 156 { 157 return tree.end(); 158 } 159 160 iterator 161 end() 162 { 163 return tree.end(); 164 } 165 166 std::size_t 167 size() const 168 { 169 return tree.size(); 170 } 171 172 bool 173 empty() const 174 { 175 return tree.empty(); 176 } 177 178 private: 179 /** 180 * Find entry that satisfies a condition on an address range 181 * 182 * Searches through the ranges in the address map and returns an 183 * iterator to the entry that satisfies the input conidition on 184 * the input address range. Returns end() if none found. 185 * 186 * @param r An input address range 187 * @param f A condition on an address range 188 * @return An iterator that contains the input address range 189 */ 190 const_iterator 191 find(const AddrRange &r, std::function<bool(const AddrRange)> cond) const 192 { 193 const_iterator next = tree.upper_bound(r); 194 if (next != end() && cond(next->first)) { 195 return next; 196 } 197 if (next == begin()) 198 return end(); 199 next--; 200 201 const_iterator i; 202 do { 203 i = next; 204 if (cond(i->first)) { 205 return i; 206 } 207 // Keep looking if the next range merges with the current one. 208 } while (next != begin() && 209 (--next)->first.mergesWith(i->first)); 210 211 return end(); 212 } 213 214 RangeMap tree; 215}; 216 217#endif //__BASE_ADDR_RANGE_MAP_HH__ 218