| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | |
| 3 * | |
| 4 * Use of this source code is governed by a BSD-style license | |
| 5 * that can be found in the LICENSE file in the root of the source | |
| 6 * tree. An additional intellectual property rights grant can be found | |
| 7 * in the file PATENTS. All contributing project authors may | |
| 8 * be found in the AUTHORS file in the root of the source tree. | |
| 9 */ | |
| 10 | |
| 11 // Borrowed from Chromium's src/base/stl_util.h. | |
| 12 | |
| 13 #ifndef WEBRTC_SYSTEM_WRAPPERS_INCLUDE_STL_UTIL_H_ | |
| 14 #define WEBRTC_SYSTEM_WRAPPERS_INCLUDE_STL_UTIL_H_ | |
| 15 | |
| 16 #include <assert.h> | |
| 17 #include <algorithm> | |
| 18 #include <functional> | |
| 19 #include <iterator> | |
| 20 #include <string> | |
| 21 #include <vector> | |
| 22 | |
| 23 namespace webrtc { | |
| 24 | |
| 25 // Clears internal memory of an STL object. | |
| 26 // STL clear()/reserve(0) does not always free internal memory allocated | |
| 27 // This function uses swap/destructor to ensure the internal memory is freed. | |
| 28 template<class T> | |
| 29 void STLClearObject(T* obj) { | |
| 30 T tmp; | |
| 31 tmp.swap(*obj); | |
| 32 // Sometimes "T tmp" allocates objects with memory (arena implementation?). | |
| 33 // Hence using additional reserve(0) even if it doesn't always work. | |
| 34 obj->reserve(0); | |
| 35 } | |
| 36 | |
| 37 // For a range within a container of pointers, calls delete (non-array version) | |
| 38 // on these pointers. | |
| 39 // NOTE: for these three functions, we could just implement a DeleteObject | |
| 40 // functor and then call for_each() on the range and functor, but this | |
| 41 // requires us to pull in all of algorithm.h, which seems expensive. | |
| 42 // For hash_[multi]set, it is important that this deletes behind the iterator | |
| 43 // because the hash_set may call the hash function on the iterator when it is | |
| 44 // advanced, which could result in the hash function trying to deference a | |
| 45 // stale pointer. | |
| 46 template <class ForwardIterator> | |
| 47 void STLDeleteContainerPointers(ForwardIterator begin, ForwardIterator end) { | |
| 48 while (begin != end) { | |
| 49 ForwardIterator temp = begin; | |
| 50 ++begin; | |
| 51 delete *temp; | |
| 52 } | |
| 53 } | |
| 54 | |
| 55 // For a range within a container of pairs, calls delete (non-array version) on | |
| 56 // BOTH items in the pairs. | |
| 57 // NOTE: Like STLDeleteContainerPointers, it is important that this deletes | |
| 58 // behind the iterator because if both the key and value are deleted, the | |
| 59 // container may call the hash function on the iterator when it is advanced, | |
| 60 // which could result in the hash function trying to dereference a stale | |
| 61 // pointer. | |
| 62 template <class ForwardIterator> | |
| 63 void STLDeleteContainerPairPointers(ForwardIterator begin, | |
| 64 ForwardIterator end) { | |
| 65 while (begin != end) { | |
| 66 ForwardIterator temp = begin; | |
| 67 ++begin; | |
| 68 delete temp->first; | |
| 69 delete temp->second; | |
| 70 } | |
| 71 } | |
| 72 | |
| 73 // For a range within a container of pairs, calls delete (non-array version) on | |
| 74 // the FIRST item in the pairs. | |
| 75 // NOTE: Like STLDeleteContainerPointers, deleting behind the iterator. | |
| 76 template <class ForwardIterator> | |
| 77 void STLDeleteContainerPairFirstPointers(ForwardIterator begin, | |
| 78 ForwardIterator end) { | |
| 79 while (begin != end) { | |
| 80 ForwardIterator temp = begin; | |
| 81 ++begin; | |
| 82 delete temp->first; | |
| 83 } | |
| 84 } | |
| 85 | |
| 86 // For a range within a container of pairs, calls delete. | |
| 87 // NOTE: Like STLDeleteContainerPointers, deleting behind the iterator. | |
| 88 // Deleting the value does not always invalidate the iterator, but it may | |
| 89 // do so if the key is a pointer into the value object. | |
| 90 template <class ForwardIterator> | |
| 91 void STLDeleteContainerPairSecondPointers(ForwardIterator begin, | |
| 92 ForwardIterator end) { | |
| 93 while (begin != end) { | |
| 94 ForwardIterator temp = begin; | |
| 95 ++begin; | |
| 96 delete temp->second; | |
| 97 } | |
| 98 } | |
| 99 | |
| 100 // To treat a possibly-empty vector as an array, use these functions. | |
| 101 // If you know the array will never be empty, you can use &*v.begin() | |
| 102 // directly, but that is undefined behaviour if |v| is empty. | |
| 103 template<typename T> | |
| 104 inline T* vector_as_array(std::vector<T>* v) { | |
| 105 return v->empty() ? NULL : &*v->begin(); | |
| 106 } | |
| 107 | |
| 108 template<typename T> | |
| 109 inline const T* vector_as_array(const std::vector<T>* v) { | |
| 110 return v->empty() ? NULL : &*v->begin(); | |
| 111 } | |
| 112 | |
| 113 // Return a mutable char* pointing to a string's internal buffer, | |
| 114 // which may not be null-terminated. Writing through this pointer will | |
| 115 // modify the string. | |
| 116 // | |
| 117 // string_as_array(&str)[i] is valid for 0 <= i < str.size() until the | |
| 118 // next call to a string method that invalidates iterators. | |
| 119 // | |
| 120 // As of 2006-04, there is no standard-blessed way of getting a | |
| 121 // mutable reference to a string's internal buffer. However, issue 530 | |
| 122 // (http://www.open-std.org/JTC1/SC22/WG21/docs/lwg-active.html#530) | |
| 123 // proposes this as the method. According to Matt Austern, this should | |
| 124 // already work on all current implementations. | |
| 125 inline char* string_as_array(std::string* str) { | |
| 126 // DO NOT USE const_cast<char*>(str->data()) | |
| 127 return str->empty() ? NULL : &*str->begin(); | |
| 128 } | |
| 129 | |
| 130 // The following functions are useful for cleaning up STL containers whose | |
| 131 // elements point to allocated memory. | |
| 132 | |
| 133 // STLDeleteElements() deletes all the elements in an STL container and clears | |
| 134 // the container. This function is suitable for use with a vector, set, | |
| 135 // hash_set, or any other STL container which defines sensible begin(), end(), | |
| 136 // and clear() methods. | |
| 137 // | |
| 138 // If container is NULL, this function is a no-op. | |
| 139 // | |
| 140 // As an alternative to calling STLDeleteElements() directly, consider | |
| 141 // STLElementDeleter (defined below), which ensures that your container's | |
| 142 // elements are deleted when the STLElementDeleter goes out of scope. | |
| 143 template <class T> | |
| 144 void STLDeleteElements(T* container) { | |
| 145 if (!container) | |
| 146 return; | |
| 147 STLDeleteContainerPointers(container->begin(), container->end()); | |
| 148 container->clear(); | |
| 149 } | |
| 150 | |
| 151 // Given an STL container consisting of (key, value) pairs, STLDeleteValues | |
| 152 // deletes all the "value" components and clears the container. Does nothing | |
| 153 // in the case it's given a NULL pointer. | |
| 154 template <class T> | |
| 155 void STLDeleteValues(T* container) { | |
| 156 if (!container) | |
| 157 return; | |
| 158 for (typename T::iterator i(container->begin()); i != container->end(); ++i) | |
| 159 delete i->second; | |
| 160 container->clear(); | |
| 161 } | |
| 162 | |
| 163 | |
| 164 // The following classes provide a convenient way to delete all elements or | |
| 165 // values from STL containers when they goes out of scope. This greatly | |
| 166 // simplifies code that creates temporary objects and has multiple return | |
| 167 // statements. Example: | |
| 168 // | |
| 169 // vector<MyProto *> tmp_proto; | |
| 170 // STLElementDeleter<vector<MyProto *> > d(&tmp_proto); | |
| 171 // if (...) return false; | |
| 172 // ... | |
| 173 // return success; | |
| 174 | |
| 175 // Given a pointer to an STL container this class will delete all the element | |
| 176 // pointers when it goes out of scope. | |
| 177 template<class T> | |
| 178 class STLElementDeleter { | |
| 179 public: | |
| 180 STLElementDeleter<T>(T* container) : container_(container) {} | |
| 181 ~STLElementDeleter<T>() { STLDeleteElements(container_); } | |
| 182 | |
| 183 private: | |
| 184 T* container_; | |
| 185 }; | |
| 186 | |
| 187 // Given a pointer to an STL container this class will delete all the value | |
| 188 // pointers when it goes out of scope. | |
| 189 template<class T> | |
| 190 class STLValueDeleter { | |
| 191 public: | |
| 192 STLValueDeleter<T>(T* container) : container_(container) {} | |
| 193 ~STLValueDeleter<T>() { STLDeleteValues(container_); } | |
| 194 | |
| 195 private: | |
| 196 T* container_; | |
| 197 }; | |
| 198 | |
| 199 // Test to see if a set, map, hash_set or hash_map contains a particular key. | |
| 200 // Returns true if the key is in the collection. | |
| 201 template <typename Collection, typename Key> | |
| 202 bool ContainsKey(const Collection& collection, const Key& key) { | |
| 203 return collection.find(key) != collection.end(); | |
| 204 } | |
| 205 | |
| 206 // Returns true if the container is sorted. | |
| 207 template <typename Container> | |
| 208 bool STLIsSorted(const Container& cont) { | |
| 209 // Note: Use reverse iterator on container to ensure we only require | |
| 210 // value_type to implement operator<. | |
| 211 return std::adjacent_find(cont.rbegin(), cont.rend(), | |
| 212 std::less<typename Container::value_type>()) | |
| 213 == cont.rend(); | |
| 214 } | |
| 215 | |
| 216 // Returns a new ResultType containing the difference of two sorted containers. | |
| 217 template <typename ResultType, typename Arg1, typename Arg2> | |
| 218 ResultType STLSetDifference(const Arg1& a1, const Arg2& a2) { | |
| 219 assert(STLIsSorted(a1)); | |
| 220 assert(STLIsSorted(a2)); | |
| 221 ResultType difference; | |
| 222 std::set_difference(a1.begin(), a1.end(), | |
| 223 a2.begin(), a2.end(), | |
| 224 std::inserter(difference, difference.end())); | |
| 225 return difference; | |
| 226 } | |
| 227 | |
| 228 // Returns a new ResultType containing the union of two sorted containers. | |
| 229 template <typename ResultType, typename Arg1, typename Arg2> | |
| 230 ResultType STLSetUnion(const Arg1& a1, const Arg2& a2) { | |
| 231 assert(STLIsSorted(a1)); | |
| 232 assert(STLIsSorted(a2)); | |
| 233 ResultType result; | |
| 234 std::set_union(a1.begin(), a1.end(), | |
| 235 a2.begin(), a2.end(), | |
| 236 std::inserter(result, result.end())); | |
| 237 return result; | |
| 238 } | |
| 239 | |
| 240 // Returns a new ResultType containing the intersection of two sorted | |
| 241 // containers. | |
| 242 template <typename ResultType, typename Arg1, typename Arg2> | |
| 243 ResultType STLSetIntersection(const Arg1& a1, const Arg2& a2) { | |
| 244 assert(STLIsSorted(a1)); | |
| 245 assert(STLIsSorted(a2)); | |
| 246 ResultType result; | |
| 247 std::set_intersection(a1.begin(), a1.end(), | |
| 248 a2.begin(), a2.end(), | |
| 249 std::inserter(result, result.end())); | |
| 250 return result; | |
| 251 } | |
| 252 | |
| 253 // Returns true if the sorted container |a1| contains all elements of the sorted | |
| 254 // container |a2|. | |
| 255 template <typename Arg1, typename Arg2> | |
| 256 bool STLIncludes(const Arg1& a1, const Arg2& a2) { | |
| 257 assert(STLIsSorted(a1)); | |
| 258 assert(STLIsSorted(a2)); | |
| 259 return std::includes(a1.begin(), a1.end(), | |
| 260 a2.begin(), a2.end()); | |
| 261 } | |
| 262 | |
| 263 } // namespace webrtc | |
| 264 | |
| 265 #endif // WEBRTC_SYSTEM_WRAPPERS_INCLUDE_STL_UTIL_H_ | |
| OLD | NEW |