]> git.saurik.com Git - wxWidgets.git/blame - src/common/dynarray.cpp
Committing in .
[wxWidgets.git] / src / common / dynarray.cpp
CommitLineData
c801d85f
KB
1///////////////////////////////////////////////////////////////////////////////
2// Name: dynarray.cpp
3// Purpose: implementation of wxBaseArray class
4// Author: Vadim Zeitlin
3bfa4402 5// Modified by:
c801d85f
KB
6// Created: 12.09.97
7// RCS-ID: $Id$
8// Copyright: (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
55d99c7a 9// Licence: wxWindows licence
c801d85f
KB
10///////////////////////////////////////////////////////////////////////////////
11
12// ============================================================================
13// headers
14// ============================================================================
15
16#ifdef __GNUG__
17#pragma implementation "dynarray.h"
18#endif
19
3096bd2f 20#include "wx/wxprec.h"
c801d85f
KB
21
22#ifdef __BORLANDC__
23 #pragma hdrstop
24#endif
25
26#include "wx/dynarray.h"
3096bd2f 27#include "wx/intl.h"
c801d85f
KB
28
29#include <stdlib.h>
3bfa4402 30#include <string.h> // for memmove
c801d85f
KB
31
32#ifndef max
33 #define max(a, b) (((a) > (b)) ? (a) : (b))
34#endif
35
acf88ae6
VZ
36// we cast the value to long from which we cast it to void * in IndexForInsert:
37// this can't work if the pointers are not big enough
38wxCOMPILE_TIME_ASSERT( sizeof(long) <= sizeof(void *),
39 wxArraySizeOfPtrLessSizeOfLong ); // < 32 symbols
40
c801d85f
KB
41// ============================================================================
42// constants
43// ============================================================================
44
45// size increment = max(50% of current size, ARRAY_MAXSIZE_INCREMENT)
46#define ARRAY_MAXSIZE_INCREMENT 4096
47
48// ============================================================================
49// implementation
50// ============================================================================
51
52// ----------------------------------------------------------------------------
5a1cad6e 53// wxBaseArray - dynamic array of 'T's
c801d85f
KB
54// ----------------------------------------------------------------------------
55
f1322419 56#define _WX_DEFINE_BASEARRAY(T, name) \
5a1cad6e
GD
57/* ctor */ \
58name::name() \
59{ \
60 m_nSize = \
61 m_nCount = 0; \
62 m_pItems = (T *)NULL; \
63} \
64 \
65/* copy ctor */ \
66name::name(const name& src) \
67{ \
68 m_nSize = /* not src.m_nSize to save memory */ \
69 m_nCount = src.m_nCount; \
70 \
71 if ( m_nSize != 0 ) { \
72 m_pItems = new T[m_nSize]; \
73 /* only copy if allocation succeeded */ \
74 if ( m_pItems ) { \
75 memcpy(m_pItems, src.m_pItems, m_nCount*sizeof(T)); \
76 } \
77 else { \
78 m_nSize = 0; \
79 } \
80 } \
81 else \
82 m_pItems = (T *) NULL; \
83} \
84 \
85/* assignment operator */ \
86name& name::operator=(const name& src) \
87{ \
88 wxDELETEA(m_pItems); \
89 \
90 m_nSize = /* not src.m_nSize to save memory */ \
91 m_nCount = src.m_nCount; \
92 \
93 if ( m_nSize != 0 ){ \
94 m_pItems = new T[m_nSize]; \
95 /* only copy if allocation succeeded */ \
96 if ( m_pItems ) { \
97 memcpy(m_pItems, src.m_pItems, m_nCount*sizeof(T)); \
98 } \
99 else { \
100 m_nSize = 0; \
101 } \
102 } \
103 else \
104 m_pItems = (T *) NULL; \
105 \
106 return *this; \
107} \
108 \
2abb9d2f
VZ
109/* allocate new buffer of the given size and move our data to it */ \
110bool name::Realloc(size_t nSize) \
111{ \
112 T *pNew = new T[nSize]; \
113 /* only grow if allocation succeeded */ \
114 if ( !pNew ) \
115 return false; \
116 \
117 m_nSize = nSize; \
118 /* copy data to new location */ \
119 memcpy(pNew, m_pItems, m_nCount*sizeof(T)); \
120 delete [] m_pItems; \
121 m_pItems = pNew; \
122 \
123 return true; \
124} \
125 \
5a1cad6e 126/* grow the array */ \
3b0b5f13 127void name::Grow(size_t nIncrement) \
5a1cad6e
GD
128{ \
129 /* only do it if no more place */ \
2b5f62a0 130 if( (m_nCount == m_nSize) || ((m_nSize - m_nCount) < nIncrement) ) { \
5a1cad6e 131 if( m_nSize == 0 ) { \
a1db4ad4
SN
132 /* was empty, determine initial size */ \
133 size_t size = WX_ARRAY_DEFAULT_INITIAL_SIZE; \
134 if (size < nIncrement) size = nIncrement; \
135 /* allocate some memory */ \
136 m_pItems = new T[size]; \
5a1cad6e
GD
137 /* only grow if allocation succeeded */ \
138 if ( m_pItems ) { \
a1db4ad4 139 m_nSize = size; \
5a1cad6e
GD
140 } \
141 } \
142 else \
143 { \
3b0b5f13
VZ
144 /* add at least 50% but not too much */ \
145 size_t ndefIncrement = m_nSize < WX_ARRAY_DEFAULT_INITIAL_SIZE \
146 ? WX_ARRAY_DEFAULT_INITIAL_SIZE : m_nSize >> 1; \
147 if ( ndefIncrement > ARRAY_MAXSIZE_INCREMENT ) \
148 ndefIncrement = ARRAY_MAXSIZE_INCREMENT; \
149 if ( nIncrement < ndefIncrement ) \
150 nIncrement = ndefIncrement; \
2abb9d2f 151 Realloc(m_nSize + nIncrement); \
5a1cad6e
GD
152 } \
153 } \
154} \
155 \
2abb9d2f
VZ
156/* make sure that the array has at least count elements */ \
157void name::SetCount(size_t count, T defval) \
158{ \
159 if ( m_nSize < count ) \
160 { \
161 /* need to realloc memory: don't overallocate it here as if */ \
162 /* SetCount() is called, it probably means that the caller */ \
163 /* knows in advance how many elements there will be in the */ \
164 /* array and so it won't be necessary to realloc it later */ \
165 if ( !Realloc(count) ) \
166 { \
167 /* out of memory -- what can we do? */ \
168 return; \
169 } \
170 } \
171 \
172 /* add new elements if we extend the array */ \
173 while ( m_nCount < count ) \
174 { \
175 m_pItems[m_nCount++] = defval; \
176 } \
177} \
178 \
5a1cad6e
GD
179/* dtor */ \
180name::~name() \
181{ \
182 wxDELETEA(m_pItems); \
183} \
184 \
185/* clears the list */ \
186void name::Clear() \
187{ \
188 m_nSize = \
189 m_nCount = 0; \
190 \
191 wxDELETEA(m_pItems); \
192} \
193 \
194/* pre-allocates memory (frees the previous data!) */ \
195void name::Alloc(size_t nSize) \
196{ \
197 /* only if old buffer was not big enough */ \
198 if ( nSize > m_nSize ) { \
199 wxDELETEA(m_pItems); \
200 m_nSize = 0; \
201 m_pItems = new T[nSize]; \
202 /* only alloc if allocation succeeded */ \
203 if ( m_pItems ) { \
204 m_nSize = nSize; \
205 } \
206 } \
207 \
208 m_nCount = 0; \
209} \
210 \
211/* minimizes the memory usage by freeing unused memory */ \
212void name::Shrink() \
213{ \
214 /* only do it if we have some memory to free */ \
215 if( m_nCount < m_nSize ) { \
216 /* allocates exactly as much memory as we need */ \
217 T *pNew = new T[m_nCount]; \
218 /* only shrink if allocation succeeded */ \
219 if ( pNew ) { \
220 /* copy data to new location */ \
221 memcpy(pNew, m_pItems, m_nCount*sizeof(T)); \
222 delete [] m_pItems; \
223 m_pItems = pNew; \
79e929e7
VZ
224 \
225 /* update the size of the new block */ \
226 m_nSize = m_nCount; \
5a1cad6e 227 } \
79e929e7 228 /* else: don't do anything, better keep old memory block! */ \
5a1cad6e
GD
229 } \
230} \
231 \
232/* searches the array for an item (forward or backwards) */ \
233int name::Index(T lItem, bool bFromEnd) const \
234{ \
235 if ( bFromEnd ) { \
236 if ( m_nCount > 0 ) { \
237 size_t n = m_nCount; \
238 do { \
239 if ( m_pItems[--n] == lItem ) \
240 return n; \
241 } \
242 while ( n != 0 ); \
243 } \
244 } \
245 else { \
246 for( size_t n = 0; n < m_nCount; n++ ) { \
247 if( m_pItems[n] == lItem ) \
248 return n; \
249 } \
250 } \
251 \
252 return wxNOT_FOUND; \
253} \
254 \
255/* search for a place to insert item into sorted array (binary search) */ \
256size_t name::IndexForInsert(T lItem, CMPFUNC fnCompare) const \
257{ \
258 size_t i, \
259 lo = 0, \
260 hi = m_nCount; \
261 int res; \
262 \
263 while ( lo < hi ) { \
264 i = (lo + hi)/2; \
265 \
acf88ae6
VZ
266 res = (*fnCompare)((const void *)(long)lItem, \
267 (const void *)(long)(m_pItems[i])); \
5a1cad6e
GD
268 if ( res < 0 ) \
269 hi = i; \
270 else if ( res > 0 ) \
271 lo = i + 1; \
272 else { \
273 lo = i; \
274 break; \
275 } \
276 } \
277 \
278 return lo; \
279} \
280 \
281/* search for an item in a sorted array (binary search) */ \
282int name::Index(T lItem, CMPFUNC fnCompare) const \
283{ \
284 size_t n = IndexForInsert(lItem, fnCompare); \
285 \
bb793b0c 286 return (n >= m_nCount || \
4365da58 287 (*fnCompare)((const void *)(long)lItem, \
bb793b0c
VZ
288 ((const void *)(long)m_pItems[n]))) ? wxNOT_FOUND \
289 : (int)n; \
5a1cad6e
GD
290} \
291 \
292/* add item at the end */ \
3b0b5f13 293void name::Add(T lItem, size_t nInsert) \
5a1cad6e 294{ \
2b5f62a0
VZ
295 if (nInsert == 0) \
296 return; \
3b0b5f13
VZ
297 Grow(nInsert); \
298 for (size_t i = 0; i < nInsert; i++) \
299 m_pItems[m_nCount++] = lItem; \
5a1cad6e
GD
300} \
301 \
302/* add item assuming the array is sorted with fnCompare function */ \
303void name::Add(T lItem, CMPFUNC fnCompare) \
304{ \
305 Insert(lItem, IndexForInsert(lItem, fnCompare)); \
306} \
307 \
308/* add item at the given position */ \
3b0b5f13 309void name::Insert(T lItem, size_t nIndex, size_t nInsert) \
5a1cad6e
GD
310{ \
311 wxCHECK_RET( nIndex <= m_nCount, wxT("bad index in wxArray::Insert") ); \
3b0b5f13
VZ
312 wxCHECK_RET( m_nCount <= m_nCount + nInsert, \
313 wxT("array size overflow in wxArray::Insert") ); \
5a1cad6e 314 \
2b5f62a0
VZ
315 if (nInsert == 0) \
316 return; \
3b0b5f13 317 Grow(nInsert); \
5a1cad6e 318 \
3b0b5f13 319 memmove(&m_pItems[nIndex + nInsert], &m_pItems[nIndex], \
5a1cad6e 320 (m_nCount - nIndex)*sizeof(T)); \
3b0b5f13
VZ
321 for (size_t i = 0; i < nInsert; i++) \
322 m_pItems[nIndex + i] = lItem; \
323 m_nCount += nInsert; \
5a1cad6e
GD
324} \
325 \
326/* removes item from array (by index) */ \
3b0b5f13 327void name::RemoveAt(size_t nIndex, size_t nRemove) \
5a1cad6e 328{ \
3b0b5f13
VZ
329 wxCHECK_RET( nIndex < m_nCount, wxT("bad index in wxArray::RemoveAt") ); \
330 wxCHECK_RET( nIndex + nRemove <= m_nCount, \
331 wxT("removing too many elements in wxArray::RemoveAt") ); \
5a1cad6e 332 \
3b0b5f13
VZ
333 memmove(&m_pItems[nIndex], &m_pItems[nIndex + nRemove], \
334 (m_nCount - nIndex - nRemove)*sizeof(T)); \
335 m_nCount -= nRemove; \
5a1cad6e
GD
336} \
337 \
338/* removes item from array (by value) */ \
339void name::Remove(T lItem) \
340{ \
341 int iIndex = Index(lItem); \
342 \
343 wxCHECK_RET( iIndex != wxNOT_FOUND, \
344 wxT("removing inexistent item in wxArray::Remove") ); \
345 \
346 RemoveAt((size_t)iIndex); \
347} \
348 \
349/* sort array elements using passed comparaison function */ \
350void name::Sort(CMPFUNC fCmp) \
351{ \
352 qsort(m_pItems, m_nCount, sizeof(T), fCmp); \
c801d85f
KB
353}
354
f1322419
VZ
355_WX_DEFINE_BASEARRAY(const void *, wxBaseArrayPtrVoid)
356_WX_DEFINE_BASEARRAY(short, wxBaseArrayShort)
357_WX_DEFINE_BASEARRAY(int, wxBaseArrayInt)
358_WX_DEFINE_BASEARRAY(long, wxBaseArrayLong)
359//_WX_DEFINE_BASEARRAY(double, wxBaseArrayDouble)
c801d85f 360