scoped_ptr_vector.h revision 90dce4d38c5ff5333bea97d859d4e484e27edf0c 
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
16namespace 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.
20template <typename T>
21class 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 scoped_ptr<T>(NULL);
76    DCHECK(position < end());
77
78    typename std::vector<T*>::iterator writable_position = position;
79    scoped_ptr<T> ret(*writable_position);
80    *writable_position = NULL;
81    return ret.Pass();
82  }
83
84  scoped_ptr<T> take_back() {
85    DCHECK(!empty());
86    if (empty())
87      return scoped_ptr<T>(NULL);
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,
133                       ScopedPtrVector<T>& other) {
134    std::vector<T*> tmp_data;
135    for (ScopedPtrVector<T>::iterator it = other.begin();
136         it != other.end();
137         ++it) {
138      tmp_data.push_back(other.take(it).release());
139    }
140    data_.insert(position, tmp_data.begin(), tmp_data.end());
141  }
142
143  void swap(ScopedPtrVector<T>& other) {
144    data_.swap(other.data_);
145  }
146
147  void swap(iterator a, iterator b) {
148    DCHECK(a < end());
149    DCHECK(b < end());
150    if (a == end() || b == end() || a == b)
151      return;
152    typename std::vector<T*>::iterator writable_a = a;
153    typename std::vector<T*>::iterator writable_b = b;
154    std::swap(*writable_a, *writable_b);
155  }
156
157  template<class Compare>
158  inline void sort(Compare comp) {
159    std::sort(data_.begin(), data_.end(), comp);
160  }
161
162  iterator begin() { return static_cast<iterator>(data_.begin()); }
163  const_iterator begin() const { return data_.begin(); }
164  iterator end() { return static_cast<iterator>(data_.end()); }
165  const_iterator end() const { return data_.end(); }
166
167  reverse_iterator rbegin() { return data_.rbegin(); }
168  const_reverse_iterator rbegin() const { return data_.rbegin(); }
169  reverse_iterator rend() { return data_.rend(); }
170  const_reverse_iterator rend() const { return data_.rend(); }
171
172 private:
173  std::vector<T*> data_;
174
175  DISALLOW_COPY_AND_ASSIGN(ScopedPtrVector);
176};
177
178}  // namespace cc
179
180#endif  // CC_BASE_SCOPED_PTR_VECTOR_H_
181