diff options
author | powei@chromium.org <powei@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2013-08-30 02:41:32 +0000 |
---|---|---|
committer | powei@chromium.org <powei@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2013-08-30 02:41:32 +0000 |
commit | 03c43ec0c85aabf6e38a786635c3929b69cc377b (patch) | |
tree | 8a7d6cdeaeca5031d5e00689d74117e9d2920819 /base/containers | |
parent | 117afcfa7b69462b4dbf70075bb22518f7c227be (diff) | |
download | chromium_src-03c43ec0c85aabf6e38a786635c3929b69cc377b.zip chromium_src-03c43ec0c85aabf6e38a786635c3929b69cc377b.tar.gz chromium_src-03c43ec0c85aabf6e38a786635c3929b69cc377b.tar.bz2 |
So that ScopedPtrHashMap can be used by user outside of cc.
BUG=156199
Review URL: https://chromiumcodereview.appspot.com/23653002
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@220488 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'base/containers')
-rw-r--r-- | base/containers/scoped_ptr_hash_map.h | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/base/containers/scoped_ptr_hash_map.h b/base/containers/scoped_ptr_hash_map.h new file mode 100644 index 0000000..8d92b39 --- /dev/null +++ b/base/containers/scoped_ptr_hash_map.h @@ -0,0 +1,157 @@ +// Copyright 2013 The Chromium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#ifndef BASE_CONTAINERS_SCOPED_PTR_HASH_MAP_H_ +#define BASE_CONTAINERS_SCOPED_PTR_HASH_MAP_H_ + +#include <algorithm> +#include <utility> + +#include "base/basictypes.h" +#include "base/containers/hash_tables.h" +#include "base/logging.h" +#include "base/memory/scoped_ptr.h" +#include "base/stl_util.h" + +namespace base { + +// This type acts like a hash_map<K, scoped_ptr<V> >, based on top of +// base::hash_map. The ScopedPtrHashMap has ownership of all values in the data +// structure. +template <typename Key, typename Value> +class ScopedPtrHashMap { + typedef base::hash_map<Key, Value*> Container; + + public: + typedef typename Container::iterator iterator; + typedef typename Container::const_iterator const_iterator; + + ScopedPtrHashMap() {} + + ~ScopedPtrHashMap() { clear(); } + + void swap(ScopedPtrHashMap<Key, Value>& other) { + data_.swap(other.data_); + } + + std::pair<iterator, bool> insert( + std::pair<Key, const scoped_ptr<Value> > pair) { + return data_.insert( + std::pair<Key, Value*>(pair.first, pair.second.release())); + } + + // Replaces value but not key if key is already present. + std::pair<iterator, bool> set(Key key, scoped_ptr<Value> data) { + iterator it = find(key); + if (it != end()) + erase(it); + Value* raw_ptr = data.release(); + return data_.insert(std::pair<Key, Value*>(key, raw_ptr)); + } + + // Does nothing if key is already present + std::pair<iterator, bool> add(Key key, scoped_ptr<Value> data) { + Value* raw_ptr = data.release(); + return data_.insert(std::pair<Key, Value*>(key, raw_ptr)); + } + + void erase(iterator it) { + if (it->second) + delete it->second; + data_.erase(it); + } + + size_t erase(const Key& k) { + iterator it = data_.find(k); + if (it == data_.end()) + return 0; + erase(it); + return 1; + } + + scoped_ptr<Value> take(iterator it) { + DCHECK(it != data_.end()); + if (it == data_.end()) + return scoped_ptr<Value>(); + + Key key = it->first; + scoped_ptr<Value> ret(it->second); + data_.erase(it); + data_.insert(std::pair<Key, Value*>(key, static_cast<Value*>(NULL))); + return ret.Pass(); + } + + scoped_ptr<Value> take(const Key& k) { + iterator it = find(k); + if (it == data_.end()) + return scoped_ptr<Value>(); + + return take(it); + } + + scoped_ptr<Value> take_and_erase(iterator it) { + DCHECK(it != data_.end()); + if (it == data_.end()) + return scoped_ptr<Value>(); + + scoped_ptr<Value> ret(it->second); + data_.erase(it); + return ret.Pass(); + } + + scoped_ptr<Value> take_and_erase(const Key& k) { + iterator it = find(k); + if (it == data_.end()) + return scoped_ptr<Value>(); + + return take_and_erase(it); + } + + // Returns the first element in the hash_map that matches the given key. + // If no such element exists it returns NULL. + Value* get(const Key& k) const { + const_iterator it = find(k); + if (it == end()) + return 0; + return it->second; + } + + inline bool contains(const Key& k) const { return data_.count(k) > 0; } + + inline void clear() { STLDeleteValues(&data_); } + + inline const_iterator find(const Key& k) const { return data_.find(k); } + inline iterator find(const Key& k) { return data_.find(k); } + + inline size_t count(const Key& k) const { return data_.count(k); } + inline std::pair<const_iterator, const_iterator> equal_range( + const Key& k) const { + return data_.equal_range(k); + } + inline std::pair<iterator, iterator> equal_range(const Key& k) { + return data_.equal_range(k); + } + + inline size_t size() const { return data_.size(); } + inline size_t max_size() const { return data_.max_size(); } + + inline bool empty() const { return data_.empty(); } + + inline size_t bucket_count() const { return data_.bucket_count(); } + inline void resize(size_t size) const { return data_.resize(size); } + + inline iterator begin() { return data_.begin(); } + inline const_iterator begin() const { return data_.begin(); } + inline iterator end() { return data_.end(); } + inline const_iterator end() const { return data_.end(); } + + private: + Container data_; + + DISALLOW_COPY_AND_ASSIGN(ScopedPtrHashMap); +}; + +} // namespace base + +#endif // BASE_CONTAINERS_SCOPED_PTR_HASH_MAP_H_ |