vector.h 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584
  1. ///////////////////////////////////////////////////////////////////////////////
  2. // Name: wx/vector.h
  3. // Purpose: STL vector clone
  4. // Author: Lindsay Mathieson
  5. // Modified by: Vaclav Slavik - make it a template
  6. // Created: 30.07.2001
  7. // Copyright: (c) 2001 Lindsay Mathieson <lindsay@mathieson.org>,
  8. // 2007 Vaclav Slavik <vslavik@fastmail.fm>
  9. // Licence: wxWindows licence
  10. ///////////////////////////////////////////////////////////////////////////////
  11. #ifndef _WX_VECTOR_H_
  12. #define _WX_VECTOR_H_
  13. #include "wx/defs.h"
  14. #if wxUSE_STD_CONTAINERS
  15. #include <vector>
  16. #include <algorithm>
  17. #define wxVector std::vector
  18. template<typename T>
  19. inline void wxVectorSort(wxVector<T>& v)
  20. {
  21. std::sort(v.begin(), v.end());
  22. }
  23. #else // !wxUSE_STD_CONTAINERS
  24. #include "wx/scopeguard.h"
  25. #include "wx/meta/movable.h"
  26. #include "wx/meta/if.h"
  27. #include "wx/beforestd.h"
  28. #include <new> // for placement new
  29. #include "wx/afterstd.h"
  30. // wxQsort is declared in wx/utils.h, but can't include that file here,
  31. // it indirectly includes this file. Just lovely...
  32. typedef int (*wxSortCallback)(const void* pItem1,
  33. const void* pItem2,
  34. const void* user_data);
  35. WXDLLIMPEXP_BASE void wxQsort(void* pbase, size_t total_elems,
  36. size_t size, wxSortCallback cmp,
  37. const void* user_data);
  38. namespace wxPrivate
  39. {
  40. // These templates encapsulate memory operations for use by wxVector; there are
  41. // two implementations, both in generic way for any C++ types and as an
  42. // optimized version for "movable" types that uses realloc() and memmove().
  43. // version for movable types:
  44. template<typename T>
  45. struct wxVectorMemOpsMovable
  46. {
  47. static void Free(T* array)
  48. { free(array); }
  49. static T* Realloc(T* old, size_t newCapacity, size_t WXUNUSED(occupiedSize))
  50. { return (T*)realloc(old, newCapacity * sizeof(T)); }
  51. static void MemmoveBackward(T* dest, T* source, size_t count)
  52. { memmove(dest, source, count * sizeof(T)); }
  53. static void MemmoveForward(T* dest, T* source, size_t count)
  54. { memmove(dest, source, count * sizeof(T)); }
  55. };
  56. // generic version for non-movable types:
  57. template<typename T>
  58. struct wxVectorMemOpsGeneric
  59. {
  60. static void Free(T* array)
  61. { ::operator delete(array); }
  62. static T* Realloc(T* old, size_t newCapacity, size_t occupiedSize)
  63. {
  64. T *mem = (T*)::operator new(newCapacity * sizeof(T));
  65. for ( size_t i = 0; i < occupiedSize; i++ )
  66. {
  67. ::new(mem + i) T(old[i]);
  68. old[i].~T();
  69. }
  70. ::operator delete(old);
  71. return mem;
  72. }
  73. static void MemmoveBackward(T* dest, T* source, size_t count)
  74. {
  75. wxASSERT( dest < source );
  76. T* destptr = dest;
  77. T* sourceptr = source;
  78. for ( size_t i = count; i > 0; --i, ++destptr, ++sourceptr )
  79. {
  80. ::new(destptr) T(*sourceptr);
  81. sourceptr->~T();
  82. }
  83. }
  84. static void MemmoveForward(T* dest, T* source, size_t count)
  85. {
  86. wxASSERT( dest > source );
  87. T* destptr = dest + count - 1;
  88. T* sourceptr = source + count - 1;
  89. for ( size_t i = count; i > 0; --i, --destptr, --sourceptr )
  90. {
  91. ::new(destptr) T(*sourceptr);
  92. sourceptr->~T();
  93. }
  94. }
  95. };
  96. } // namespace wxPrivate
  97. template<typename T>
  98. class wxVector
  99. {
  100. private:
  101. // This cryptic expression means "typedef Ops to wxVectorMemOpsMovable if
  102. // type T is movable type, otherwise to wxVectorMemOpsGeneric".
  103. //
  104. // Note that we use typedef instead of privately deriving from this (which
  105. // would allowed us to omit "Ops::" prefixes below) to keep VC6 happy,
  106. // it can't compile code that derives from wxIf<...>::value.
  107. //
  108. // Note that bcc needs the extra parentheses for non-type template
  109. // arguments to compile this expression.
  110. typedef typename wxIf< (wxIsMovable<T>::value),
  111. wxPrivate::wxVectorMemOpsMovable<T>,
  112. wxPrivate::wxVectorMemOpsGeneric<T> >::value
  113. Ops;
  114. public:
  115. typedef size_t size_type;
  116. typedef size_t difference_type;
  117. typedef T value_type;
  118. typedef value_type* pointer;
  119. typedef const value_type* const_pointer;
  120. typedef value_type* iterator;
  121. typedef const value_type* const_iterator;
  122. typedef value_type& reference;
  123. typedef const value_type& const_reference;
  124. class reverse_iterator
  125. {
  126. public:
  127. reverse_iterator() : m_ptr(NULL) { }
  128. wxEXPLICIT reverse_iterator(iterator it) : m_ptr(it) { }
  129. reverse_iterator(const reverse_iterator& it) : m_ptr(it.m_ptr) { }
  130. reference operator*() const { return *m_ptr; }
  131. pointer operator->() const { return m_ptr; }
  132. iterator base() const { return m_ptr; }
  133. reverse_iterator& operator++()
  134. { --m_ptr; return *this; }
  135. reverse_iterator operator++(int)
  136. { reverse_iterator tmp = *this; --m_ptr; return tmp; }
  137. reverse_iterator& operator--()
  138. { ++m_ptr; return *this; }
  139. reverse_iterator operator--(int)
  140. { reverse_iterator tmp = *this; ++m_ptr; return tmp; }
  141. reverse_iterator operator+(difference_type n) const
  142. { return reverse_iterator(m_ptr - n); }
  143. reverse_iterator& operator+=(difference_type n)
  144. { m_ptr -= n; return *this; }
  145. reverse_iterator operator-(difference_type n) const
  146. { return reverse_iterator(m_ptr + n); }
  147. reverse_iterator& operator-=(difference_type n)
  148. { m_ptr += n; return *this; }
  149. reference operator[](difference_type n) const
  150. { return *(*this + n); }
  151. bool operator ==(const reverse_iterator& it) const
  152. { return m_ptr == it.m_ptr; }
  153. bool operator !=(const reverse_iterator& it) const
  154. { return m_ptr != it.m_ptr; }
  155. private:
  156. value_type *m_ptr;
  157. friend class const_reverse_iterator;
  158. };
  159. class const_reverse_iterator
  160. {
  161. public:
  162. const_reverse_iterator() : m_ptr(NULL) { }
  163. wxEXPLICIT const_reverse_iterator(const_iterator it) : m_ptr(it) { }
  164. const_reverse_iterator(const reverse_iterator& it) : m_ptr(it.m_ptr) { }
  165. const_reverse_iterator(const const_reverse_iterator& it) : m_ptr(it.m_ptr) { }
  166. const_reference operator*() const { return *m_ptr; }
  167. const_pointer operator->() const { return m_ptr; }
  168. const_iterator base() const { return m_ptr; }
  169. const_reverse_iterator& operator++()
  170. { --m_ptr; return *this; }
  171. const_reverse_iterator operator++(int)
  172. { const_reverse_iterator tmp = *this; --m_ptr; return tmp; }
  173. const_reverse_iterator& operator--()
  174. { ++m_ptr; return *this; }
  175. const_reverse_iterator operator--(int)
  176. { const_reverse_iterator tmp = *this; ++m_ptr; return tmp; }
  177. const_reverse_iterator operator+(difference_type n) const
  178. { return const_reverse_iterator(m_ptr - n); }
  179. const_reverse_iterator& operator+=(difference_type n)
  180. { m_ptr -= n; return *this; }
  181. const_reverse_iterator operator-(difference_type n) const
  182. { return const_reverse_iterator(m_ptr + n); }
  183. const_reverse_iterator& operator-=(difference_type n)
  184. { m_ptr += n; return *this; }
  185. const_reference operator[](difference_type n) const
  186. { return *(*this + n); }
  187. bool operator ==(const const_reverse_iterator& it) const
  188. { return m_ptr == it.m_ptr; }
  189. bool operator !=(const const_reverse_iterator& it) const
  190. { return m_ptr != it.m_ptr; }
  191. protected:
  192. const value_type *m_ptr;
  193. };
  194. wxVector() : m_size(0), m_capacity(0), m_values(NULL) {}
  195. wxVector(size_type p_size)
  196. : m_size(0), m_capacity(0), m_values(NULL)
  197. {
  198. reserve(p_size);
  199. for ( size_t n = 0; n < p_size; n++ )
  200. push_back(value_type());
  201. }
  202. wxVector(size_type p_size, const value_type& v)
  203. : m_size(0), m_capacity(0), m_values(NULL)
  204. {
  205. reserve(p_size);
  206. for ( size_t n = 0; n < p_size; n++ )
  207. push_back(v);
  208. }
  209. wxVector(const wxVector& c) : m_size(0), m_capacity(0), m_values(NULL)
  210. {
  211. Copy(c);
  212. }
  213. template <class InputIterator>
  214. wxVector(InputIterator first, InputIterator last)
  215. : m_size(0), m_capacity(0), m_values(NULL)
  216. {
  217. assign(first, last);
  218. }
  219. ~wxVector()
  220. {
  221. clear();
  222. }
  223. void assign(size_type p_size, const value_type& v)
  224. {
  225. clear();
  226. reserve(p_size);
  227. for ( size_t n = 0; n < p_size; n++ )
  228. push_back(v);
  229. }
  230. template <class InputIterator>
  231. void assign(InputIterator first, InputIterator last)
  232. {
  233. clear();
  234. // Notice that it would be nice to call reserve() here but we can't do
  235. // it for arbitrary input iterators, we should have a dispatch on
  236. // iterator type and call it if possible.
  237. for ( InputIterator it = first; it != last; ++it )
  238. push_back(*it);
  239. }
  240. void swap(wxVector& v)
  241. {
  242. wxSwap(m_size, v.m_size);
  243. wxSwap(m_capacity, v.m_capacity);
  244. wxSwap(m_values, v.m_values);
  245. }
  246. void clear()
  247. {
  248. // call destructors of stored objects:
  249. for ( size_type i = 0; i < m_size; i++ )
  250. {
  251. m_values[i].~T();
  252. }
  253. Ops::Free(m_values);
  254. m_values = NULL;
  255. m_size =
  256. m_capacity = 0;
  257. }
  258. void reserve(size_type n)
  259. {
  260. if ( n <= m_capacity )
  261. return;
  262. // increase the size twice, unless we're already too big or unless
  263. // more is requested
  264. //
  265. // NB: casts to size_type are needed to suppress warnings about
  266. // mixing enumeral and non-enumeral type in conditional expression
  267. const size_type increment = m_size > 0
  268. ? m_size < ALLOC_MAX_SIZE
  269. ? m_size
  270. : (size_type)ALLOC_MAX_SIZE
  271. : (size_type)ALLOC_INITIAL_SIZE;
  272. if ( m_capacity + increment > n )
  273. n = m_capacity + increment;
  274. m_values = Ops::Realloc(m_values, n, m_size);
  275. m_capacity = n;
  276. }
  277. void resize(size_type n)
  278. {
  279. if ( n < m_size )
  280. Shrink(n);
  281. else if ( n > m_size )
  282. Extend(n, value_type());
  283. }
  284. void resize(size_type n, const value_type& v)
  285. {
  286. if ( n < m_size )
  287. Shrink(n);
  288. else if ( n > m_size )
  289. Extend(n, v);
  290. }
  291. size_type size() const
  292. {
  293. return m_size;
  294. }
  295. size_type capacity() const
  296. {
  297. return m_capacity;
  298. }
  299. bool empty() const
  300. {
  301. return size() == 0;
  302. }
  303. wxVector& operator=(const wxVector& vb)
  304. {
  305. if (this != &vb)
  306. {
  307. clear();
  308. Copy(vb);
  309. }
  310. return *this;
  311. }
  312. void push_back(const value_type& v)
  313. {
  314. reserve(size() + 1);
  315. // use placement new to initialize new object in preallocated place in
  316. // m_values and store 'v' in it:
  317. void* const place = m_values + m_size;
  318. ::new(place) value_type(v);
  319. // only increase m_size if the ctor didn't throw an exception; notice
  320. // that if it _did_ throw, everything is OK, because we only increased
  321. // vector's capacity so far and possibly written some data to
  322. // uninitialized memory at the end of m_values
  323. m_size++;
  324. }
  325. void pop_back()
  326. {
  327. erase(end() - 1);
  328. }
  329. const value_type& at(size_type idx) const
  330. {
  331. wxASSERT(idx < m_size);
  332. return m_values[idx];
  333. }
  334. value_type& at(size_type idx)
  335. {
  336. wxASSERT(idx < m_size);
  337. return m_values[idx];
  338. }
  339. const value_type& operator[](size_type idx) const { return at(idx); }
  340. value_type& operator[](size_type idx) { return at(idx); }
  341. const value_type& front() const { return at(0); }
  342. value_type& front() { return at(0); }
  343. const value_type& back() const { return at(size() - 1); }
  344. value_type& back() { return at(size() - 1); }
  345. const_iterator begin() const { return m_values; }
  346. iterator begin() { return m_values; }
  347. const_iterator end() const { return m_values + size(); }
  348. iterator end() { return m_values + size(); }
  349. reverse_iterator rbegin() { return reverse_iterator(end() - 1); }
  350. reverse_iterator rend() { return reverse_iterator(begin() - 1); }
  351. const_reverse_iterator rbegin() const { return const_reverse_iterator(end() - 1); }
  352. const_reverse_iterator rend() const { return const_reverse_iterator(begin() - 1); }
  353. iterator insert(iterator it, const value_type& v = value_type())
  354. {
  355. // NB: this must be done before reserve(), because reserve()
  356. // invalidates iterators!
  357. const size_t idx = it - begin();
  358. const size_t after = end() - it;
  359. reserve(size() + 1);
  360. // the place where the new element is going to be inserted
  361. value_type * const place = m_values + idx;
  362. // unless we're inserting at the end, move following elements out of
  363. // the way:
  364. if ( after > 0 )
  365. Ops::MemmoveForward(place + 1, place, after);
  366. // if the ctor called below throws an exception, we need to move all
  367. // the elements back to their original positions in m_values
  368. wxScopeGuard moveBack = wxMakeGuard(
  369. Ops::MemmoveBackward, place, place + 1, after);
  370. if ( !after )
  371. moveBack.Dismiss();
  372. // use placement new to initialize new object in preallocated place in
  373. // m_values and store 'v' in it:
  374. ::new(place) value_type(v);
  375. // now that we did successfully add the new element, increment the size
  376. // and disable moving the items back
  377. moveBack.Dismiss();
  378. m_size++;
  379. return begin() + idx;
  380. }
  381. iterator erase(iterator it)
  382. {
  383. return erase(it, it + 1);
  384. }
  385. iterator erase(iterator first, iterator last)
  386. {
  387. if ( first == last )
  388. return first;
  389. wxASSERT( first < end() && last <= end() );
  390. const size_type idx = first - begin();
  391. const size_type count = last - first;
  392. const size_type after = end() - last;
  393. // erase elements by calling their destructors:
  394. for ( iterator i = first; i < last; ++i )
  395. i->~T();
  396. // once that's done, move following elements over to the freed space:
  397. if ( after > 0 )
  398. {
  399. Ops::MemmoveBackward(m_values + idx, m_values + idx + count, after);
  400. }
  401. m_size -= count;
  402. return begin() + idx;
  403. }
  404. #if WXWIN_COMPATIBILITY_2_8
  405. wxDEPRECATED( size_type erase(size_type n) );
  406. #endif // WXWIN_COMPATIBILITY_2_8
  407. private:
  408. // VC6 can't compile static const int members
  409. enum { ALLOC_INITIAL_SIZE = 16 };
  410. enum { ALLOC_MAX_SIZE = 4096 };
  411. void Copy(const wxVector& vb)
  412. {
  413. reserve(vb.size());
  414. for ( const_iterator i = vb.begin(); i != vb.end(); ++i )
  415. push_back(*i);
  416. }
  417. private:
  418. void Shrink(size_type n)
  419. {
  420. for ( size_type i = n; i < m_size; i++ )
  421. m_values[i].~T();
  422. m_size = n;
  423. }
  424. void Extend(size_type n, const value_type& v)
  425. {
  426. reserve(n);
  427. for ( size_type i = m_size; i < n; i++ )
  428. push_back(v);
  429. }
  430. size_type m_size,
  431. m_capacity;
  432. value_type *m_values;
  433. };
  434. #if WXWIN_COMPATIBILITY_2_8
  435. template<typename T>
  436. inline typename wxVector<T>::size_type wxVector<T>::erase(size_type n)
  437. {
  438. erase(begin() + n);
  439. return n;
  440. }
  441. #endif // WXWIN_COMPATIBILITY_2_8
  442. namespace wxPrivate
  443. {
  444. // This is a helper for the wxVectorSort function, and should not be used
  445. // directly in user's code.
  446. template<typename T>
  447. struct wxVectorComparator
  448. {
  449. static int
  450. Compare(const void* pitem1, const void* pitem2, const void* )
  451. {
  452. const T& item1 = *reinterpret_cast<const T*>(pitem1);
  453. const T& item2 = *reinterpret_cast<const T*>(pitem2);
  454. if (item1 < item2)
  455. return -1;
  456. else if (item2 < item1)
  457. return 1;
  458. else
  459. return 0;
  460. }
  461. };
  462. } // namespace wxPrivate
  463. template<typename T>
  464. void wxVectorSort(wxVector<T>& v)
  465. {
  466. wxQsort(v.begin(), v.size(), sizeof(T),
  467. wxPrivate::wxVectorComparator<T>::Compare, NULL);
  468. }
  469. #endif // wxUSE_STD_CONTAINERS/!wxUSE_STD_CONTAINERS
  470. #if WXWIN_COMPATIBILITY_2_8
  471. #define WX_DECLARE_VECTORBASE(obj, cls) typedef wxVector<obj> cls
  472. #define _WX_DECLARE_VECTOR(obj, cls, exp) WX_DECLARE_VECTORBASE(obj, cls)
  473. #define WX_DECLARE_VECTOR(obj, cls) WX_DECLARE_VECTORBASE(obj, cls)
  474. #endif // WXWIN_COMPATIBILITY_2_8
  475. #endif // _WX_VECTOR_H_