| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 #ifndef TerminatedArrayBuilder_h | 4 #ifndef TerminatedArrayBuilder_h |
| 5 #define TerminatedArrayBuilder_h | 5 #define TerminatedArrayBuilder_h |
| 6 | 6 |
| 7 #include "wtf/OwnPtr.h" | 7 #include "wtf/OwnPtr.h" |
| 8 | 8 |
| 9 namespace WTF { | 9 namespace WTF { |
| 10 | 10 |
| 11 template<typename T, template <typename> class ArrayType = TerminatedArray> | 11 template <typename T, template <typename> class ArrayType = TerminatedArray> |
| 12 class TerminatedArrayBuilder { | 12 class TerminatedArrayBuilder { |
| 13 DISALLOW_NEW(); | 13 DISALLOW_NEW(); |
| 14 WTF_MAKE_NONCOPYABLE(TerminatedArrayBuilder); | 14 WTF_MAKE_NONCOPYABLE(TerminatedArrayBuilder); |
| 15 public: | 15 |
| 16 explicit TerminatedArrayBuilder(typename ArrayType<T>::Allocator::PassPtr ar
ray) | 16 public: |
| 17 : m_array(array) | 17 explicit TerminatedArrayBuilder(typename ArrayType<T>::Allocator::PassPtr arra
y) |
| 18 , m_count(0) | 18 : m_array(array), m_count(0), m_capacity(0) { |
| 19 , m_capacity(0) | 19 if (!m_array) |
| 20 { | 20 return; |
| 21 if (!m_array) | 21 m_capacity = m_count = m_array->size(); |
| 22 return; | 22 } |
| 23 m_capacity = m_count = m_array->size(); | 23 |
| 24 void grow(size_t count) { |
| 25 ASSERT(count); |
| 26 if (!m_array) { |
| 27 ASSERT(!m_count); |
| 28 ASSERT(!m_capacity); |
| 29 m_capacity = count; |
| 30 m_array = ArrayType<T>::Allocator::create(m_capacity); |
| 31 return; |
| 24 } | 32 } |
| 33 m_capacity += count; |
| 34 m_array = ArrayType<T>::Allocator::resize(m_array.release(), m_capacity); |
| 35 m_array->at(m_count - 1).setLastInArray(false); |
| 36 } |
| 25 | 37 |
| 26 void grow(size_t count) | 38 void append(const T& item) { |
| 27 { | 39 RELEASE_ASSERT(m_count < m_capacity); |
| 28 ASSERT(count); | 40 ASSERT(!item.isLastInArray()); |
| 29 if (!m_array) { | 41 m_array->at(m_count++) = item; |
| 30 ASSERT(!m_count); | 42 } |
| 31 ASSERT(!m_capacity); | 43 |
| 32 m_capacity = count; | 44 typename ArrayType<T>::Allocator::PassPtr release() { |
| 33 m_array = ArrayType<T>::Allocator::create(m_capacity); | 45 RELEASE_ASSERT(m_count == m_capacity); |
| 34 return; | 46 if (m_array) |
| 35 } | 47 m_array->at(m_count - 1).setLastInArray(true); |
| 36 m_capacity += count; | 48 assertValid(); |
| 37 m_array = ArrayType<T>::Allocator::resize(m_array.release(), m_capacity)
; | 49 return m_array.release(); |
| 38 m_array->at(m_count - 1).setLastInArray(false); | 50 } |
| 51 |
| 52 private: |
| 53 #if ENABLE(ASSERT) |
| 54 void assertValid() { |
| 55 for (size_t i = 0; i < m_count; ++i) { |
| 56 bool isLastInArray = (i + 1 == m_count); |
| 57 ASSERT(m_array->at(i).isLastInArray() == isLastInArray); |
| 39 } | 58 } |
| 40 | 59 } |
| 41 void append(const T& item) | |
| 42 { | |
| 43 RELEASE_ASSERT(m_count < m_capacity); | |
| 44 ASSERT(!item.isLastInArray()); | |
| 45 m_array->at(m_count++) = item; | |
| 46 } | |
| 47 | |
| 48 typename ArrayType<T>::Allocator::PassPtr release() | |
| 49 { | |
| 50 RELEASE_ASSERT(m_count == m_capacity); | |
| 51 if (m_array) | |
| 52 m_array->at(m_count - 1).setLastInArray(true); | |
| 53 assertValid(); | |
| 54 return m_array.release(); | |
| 55 } | |
| 56 | |
| 57 private: | |
| 58 #if ENABLE(ASSERT) | |
| 59 void assertValid() | |
| 60 { | |
| 61 for (size_t i = 0; i < m_count; ++i) { | |
| 62 bool isLastInArray = (i + 1 == m_count); | |
| 63 ASSERT(m_array->at(i).isLastInArray() == isLastInArray); | |
| 64 } | |
| 65 } | |
| 66 #else | 60 #else |
| 67 void assertValid() { } | 61 void assertValid() {} |
| 68 #endif | 62 #endif |
| 69 | 63 |
| 70 typename ArrayType<T>::Allocator::Ptr m_array; | 64 typename ArrayType<T>::Allocator::Ptr m_array; |
| 71 size_t m_count; | 65 size_t m_count; |
| 72 size_t m_capacity; | 66 size_t m_capacity; |
| 73 }; | 67 }; |
| 74 | 68 |
| 75 } // namespace WTF | 69 } // namespace WTF |
| 76 | 70 |
| 77 using WTF::TerminatedArrayBuilder; | 71 using WTF::TerminatedArrayBuilder; |
| 78 | 72 |
| 79 #endif // TerminatedArrayBuilder_h | 73 #endif // TerminatedArrayBuilder_h |
| OLD | NEW |