OLD | NEW |
| (Empty) |
1 // Copyright 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef CC_BASE_SCOPED_PTR_VECTOR_H_ | |
6 #define CC_BASE_SCOPED_PTR_VECTOR_H_ | |
7 | |
8 #include <algorithm> | |
9 #include <vector> | |
10 | |
11 #include "base/basictypes.h" | |
12 #include "base/logging.h" | |
13 #include "base/memory/scoped_ptr.h" | |
14 #include "base/stl_util.h" | |
15 | |
16 namespace cc { | |
17 | |
18 // This type acts like a vector<scoped_ptr> based on top of std::vector. The | |
19 // ScopedPtrVector has ownership of all elements in the vector. | |
20 template <typename T> | |
21 class ScopedPtrVector { | |
22 public: | |
23 typedef typename std::vector<T*>::const_iterator const_iterator; | |
24 typedef typename std::vector<T*>::reverse_iterator reverse_iterator; | |
25 typedef typename std::vector<T*>::const_reverse_iterator | |
26 const_reverse_iterator; | |
27 | |
28 #if defined(OS_ANDROID) | |
29 // On Android the iterator is not a class, so we can't block assignment. | |
30 typedef typename std::vector<T*>::iterator iterator; | |
31 #else | |
32 // Ban setting values on the iterator directly. New pointers must be passed | |
33 // to methods on the ScopedPtrVector class to appear in the vector. | |
34 class iterator : public std::vector<T*>::iterator { | |
35 public: | |
36 iterator(const typename std::vector<T*>::iterator& other) // NOLINT | |
37 : std::vector<T*>::iterator(other) {} | |
38 T* const& operator*() { return std::vector<T*>::iterator::operator*(); } | |
39 }; | |
40 #endif | |
41 | |
42 ScopedPtrVector() {} | |
43 | |
44 ~ScopedPtrVector() { clear(); } | |
45 | |
46 size_t size() const { | |
47 return data_.size(); | |
48 } | |
49 | |
50 T* at(size_t index) const { | |
51 DCHECK(index < size()); | |
52 return data_[index]; | |
53 } | |
54 | |
55 T* operator[](size_t index) const { | |
56 return at(index); | |
57 } | |
58 | |
59 T* front() const { | |
60 DCHECK(!empty()); | |
61 return at(0); | |
62 } | |
63 | |
64 T* back() const { | |
65 DCHECK(!empty()); | |
66 return at(size() - 1); | |
67 } | |
68 | |
69 bool empty() const { | |
70 return data_.empty(); | |
71 } | |
72 | |
73 scoped_ptr<T> take(iterator position) { | |
74 if (position == end()) | |
75 return nullptr; | |
76 DCHECK(position < end()); | |
77 | |
78 typename std::vector<T*>::iterator writable_position = position; | |
79 scoped_ptr<T> ret(*writable_position); | |
80 *writable_position = nullptr; | |
81 return ret.Pass(); | |
82 } | |
83 | |
84 scoped_ptr<T> take_back() { | |
85 DCHECK(!empty()); | |
86 if (empty()) | |
87 return nullptr; | |
88 return take(end() - 1); | |
89 } | |
90 | |
91 void erase(iterator position) { | |
92 if (position == end()) | |
93 return; | |
94 typename std::vector<T*>::iterator writable_position = position; | |
95 delete *writable_position; | |
96 data_.erase(position); | |
97 } | |
98 | |
99 void erase(iterator first, iterator last) { | |
100 DCHECK(first <= last); | |
101 for (iterator it = first; it != last; ++it) { | |
102 DCHECK(it < end()); | |
103 | |
104 typename std::vector<T*>::iterator writable_it = it; | |
105 delete *writable_it; | |
106 } | |
107 data_.erase(first, last); | |
108 } | |
109 | |
110 void reserve(size_t size) { | |
111 data_.reserve(size); | |
112 } | |
113 | |
114 void clear() { | |
115 STLDeleteElements(&data_); | |
116 } | |
117 | |
118 void push_back(scoped_ptr<T> item) { | |
119 data_.push_back(item.release()); | |
120 } | |
121 | |
122 void pop_back() { | |
123 delete data_.back(); | |
124 data_.pop_back(); | |
125 } | |
126 | |
127 void insert(iterator position, scoped_ptr<T> item) { | |
128 DCHECK(position <= end()); | |
129 data_.insert(position, item.release()); | |
130 } | |
131 | |
132 void insert_and_take(iterator position, ScopedPtrVector<T>* other) { | |
133 std::vector<T*> tmp_data; | |
134 for (ScopedPtrVector<T>::iterator it = other->begin(); it != other->end(); | |
135 ++it) { | |
136 tmp_data.push_back(other->take(it).release()); | |
137 } | |
138 data_.insert(position, tmp_data.begin(), tmp_data.end()); | |
139 } | |
140 | |
141 template <typename Predicate> | |
142 iterator partition(Predicate predicate) { | |
143 typename std::vector<T*>::iterator first = begin(); | |
144 typename std::vector<T*>::iterator last = end(); | |
145 return static_cast<iterator>(std::partition(first, last, predicate)); | |
146 } | |
147 | |
148 void swap(ScopedPtrVector<T>& other) { | |
149 data_.swap(other.data_); | |
150 } | |
151 | |
152 void swap(iterator a, iterator b) { | |
153 DCHECK(a < end()); | |
154 DCHECK(b < end()); | |
155 if (a == end() || b == end() || a == b) | |
156 return; | |
157 typename std::vector<T*>::iterator writable_a = a; | |
158 typename std::vector<T*>::iterator writable_b = b; | |
159 std::swap(*writable_a, *writable_b); | |
160 } | |
161 | |
162 // This acts like std::remove_if but with one key difference. The values to be | |
163 // removed to will each appear exactly once at or after the returned iterator, | |
164 // so that erase(foo.remove_if(P), foo.end()) will not leak or double-free the | |
165 // pointers in the vector. | |
166 template <typename Predicate> | |
167 iterator remove_if(Predicate predicate) { | |
168 typename std::vector<T*>::iterator it = | |
169 std::find_if(data_.begin(), data_.end(), predicate); | |
170 typename std::vector<T*>::iterator end = data_.end(); | |
171 if (it == end) | |
172 return it; | |
173 typename std::vector<T*>::iterator result = it; | |
174 ++it; | |
175 for (; it != end; ++it) { | |
176 if (!static_cast<bool>(predicate(*it))) { | |
177 // Swap here instead of just assign to |result| so that all the | |
178 // pointers are preserved to be deleted afterward. | |
179 std::swap(*result, *it); | |
180 ++result; | |
181 } | |
182 } | |
183 return result; | |
184 } | |
185 | |
186 template<class Compare> | |
187 inline void sort(Compare comp) { | |
188 std::sort(data_.begin(), data_.end(), comp); | |
189 } | |
190 | |
191 template <class Compare> | |
192 inline void make_heap(Compare comp) { | |
193 std::make_heap(data_.begin(), data_.end(), comp); | |
194 } | |
195 | |
196 template <class Compare> | |
197 inline void push_heap(Compare comp) { | |
198 std::push_heap(data_.begin(), data_.end(), comp); | |
199 } | |
200 | |
201 template <class Compare> | |
202 inline void pop_heap(Compare comp) { | |
203 std::pop_heap(data_.begin(), data_.end(), comp); | |
204 } | |
205 | |
206 iterator begin() { return static_cast<iterator>(data_.begin()); } | |
207 const_iterator begin() const { return data_.begin(); } | |
208 iterator end() { return static_cast<iterator>(data_.end()); } | |
209 const_iterator end() const { return data_.end(); } | |
210 | |
211 reverse_iterator rbegin() { return data_.rbegin(); } | |
212 const_reverse_iterator rbegin() const { return data_.rbegin(); } | |
213 reverse_iterator rend() { return data_.rend(); } | |
214 const_reverse_iterator rend() const { return data_.rend(); } | |
215 | |
216 private: | |
217 std::vector<T*> data_; | |
218 | |
219 DISALLOW_COPY_AND_ASSIGN(ScopedPtrVector); | |
220 }; | |
221 | |
222 } // namespace cc | |
223 | |
224 #endif // CC_BASE_SCOPED_PTR_VECTOR_H_ | |
OLD | NEW |