]> git.saurik.com Git - wxWidgets.git/blame - src/common/string.cpp
replaced C++ comments with C ones
[wxWidgets.git] / src / common / string.cpp
CommitLineData
c801d85f
KB
1/////////////////////////////////////////////////////////////////////////////
2// Name: string.cpp
3// Purpose: wxString class
4// Author: Vadim Zeitlin
5// Modified by:
6// Created: 29/01/98
7// RCS-ID: $Id$
8// Copyright: (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
dd1eaa89 9// Licence: wxWindows license
c801d85f
KB
10/////////////////////////////////////////////////////////////////////////////
11
12#ifdef __GNUG__
30b21f9a 13 #pragma implementation "string.h"
c801d85f
KB
14#endif
15
16/*
17 * About ref counting:
18 * 1) all empty strings use g_strEmpty, nRefs = -1 (set in Init())
19 * 2) AllocBuffer() sets nRefs to 1, Lock() increments it by one
20 * 3) Unlock() decrements nRefs and frees memory if it goes to 0
21 */
22
23// ===========================================================================
24// headers, declarations, constants
25// ===========================================================================
26
27// For compilers that support precompilation, includes "wx.h".
28#include "wx/wxprec.h"
29
30#ifdef __BORLANDC__
30b21f9a 31 #pragma hdrstop
c801d85f
KB
32#endif
33
34#ifndef WX_PRECOMP
3c024cc2
VZ
35 #include "wx/defs.h"
36 #include "wx/string.h"
37 #include "wx/intl.h"
3096bd2f 38 #include "wx/thread.h"
6b769f3d 39#endif
c801d85f 40
706c2ac9
VZ
41#include "wx/regex.h" // for wxString::Matches()
42
c801d85f
KB
43#include <ctype.h>
44#include <string.h>
45#include <stdlib.h>
46
ce3ed50d 47#ifdef __SALFORDC__
30b21f9a 48 #include <clib.h>
ce3ed50d
JS
49#endif
50
c801d85f
KB
51#ifdef WXSTRING_IS_WXOBJECT
52 IMPLEMENT_DYNAMIC_CLASS(wxString, wxObject)
53#endif //WXSTRING_IS_WXOBJECT
54
ec2ba3aa
OK
55#if wxUSE_UNICODE
56#undef wxUSE_EXPERIMENTAL_PRINTF
57#define wxUSE_EXPERIMENTAL_PRINTF 1
58#endif
59
3168a13f
VZ
60// allocating extra space for each string consumes more memory but speeds up
61// the concatenation operations (nLen is the current string's length)
77ca46e7
VZ
62// NB: EXTRA_ALLOC must be >= 0!
63#define EXTRA_ALLOC (19 - nLen % 16)
3168a13f 64
c801d85f
KB
65// ---------------------------------------------------------------------------
66// static class variables definition
67// ---------------------------------------------------------------------------
68
75ad8feb
DW
69#if defined(__VISAGECPP__) && __IBMCPP__ >= 400
70// must define this static for VA or else you get multiply defined symbols
71// everywhere
72const unsigned int wxSTRING_MAXLEN = UINT_MAX - 100;
73#endif // Visual Age
74
8de2e39c 75#ifdef wxSTD_STRING_COMPATIBILITY
566b84d2 76 const size_t wxString::npos = wxSTRING_MAXLEN;
8de2e39c 77#endif // wxSTD_STRING_COMPATIBILITY
c801d85f 78
3168a13f
VZ
79// ----------------------------------------------------------------------------
80// static data
81// ----------------------------------------------------------------------------
c801d85f 82
3c024cc2
VZ
83// for an empty string, GetStringData() will return this address: this
84// structure has the same layout as wxStringData and it's data() method will
85// return the empty string (dummy pointer)
86static const struct
87{
88 wxStringData data;
2bb67b80 89 wxChar dummy;
223d09f6 90} g_strEmpty = { {-1, 0, 0}, wxT('\0') };
3c024cc2 91
c801d85f 92// empty C style string: points to 'string data' byte of g_strEmpty
e90c1d2a 93extern const wxChar WXDLLEXPORT *wxEmptyString = &g_strEmpty.dummy;
c801d85f 94
89b892a2
VZ
95// ----------------------------------------------------------------------------
96// conditional compilation
97// ----------------------------------------------------------------------------
98
dcf924a3
RR
99#if !defined(__WXSW__) && wxUSE_UNICODE
100 #ifdef wxUSE_EXPERIMENTAL_PRINTF
101 #undef wxUSE_EXPERIMENTAL_PRINTF
102 #endif
103 #define wxUSE_EXPERIMENTAL_PRINTF 1
104#endif
105
89b892a2
VZ
106// we want to find out if the current platform supports vsnprintf()-like
107// function: for Unix this is done with configure, for Windows we test the
108// compiler explicitly.
378b05f7
VZ
109//
110// FIXME currently, this is only for ANSI (!Unicode) strings, so we call this
111// function wxVsnprintfA (A for ANSI), should also find one for Unicode
112// strings in Unicode build
89b892a2 113#ifdef __WXMSW__
012286eb 114 #if defined(__VISUALC__) || (defined(__MINGW32__) && wxUSE_NORLANDER_HEADERS)
378b05f7 115 #define wxVsnprintfA _vsnprintf
89b892a2 116 #endif
03e11df5
GD
117#elif defined(__WXMAC__)
118 #define wxVsnprintfA vsnprintf
89b892a2
VZ
119#else // !Windows
120 #ifdef HAVE_VSNPRINTF
378b05f7 121 #define wxVsnprintfA vsnprintf
89b892a2
VZ
122 #endif
123#endif // Windows/!Windows
124
378b05f7 125#ifndef wxVsnprintfA
89b892a2
VZ
126 // in this case we'll use vsprintf() (which is ANSI and thus should be
127 // always available), but it's unsafe because it doesn't check for buffer
128 // size - so give a warning
378b05f7 129 #define wxVsnprintfA(buf, len, format, arg) vsprintf(buf, format, arg)
566b84d2 130
57493f9f
VZ
131 #if defined(__VISUALC__)
132 #pragma message("Using sprintf() because no snprintf()-like function defined")
03e11df5 133 #elif defined(__GNUG__)
378b05f7 134 #warning "Using sprintf() because no snprintf()-like function defined"
57493f9f 135 #endif //compiler
3f4a0c5b 136#endif // no vsnprintf
89b892a2 137
227b5cd7
VZ
138#ifdef _AIX
139 // AIX has vsnprintf, but there's no prototype in the system headers.
140 extern "C" int vsnprintf(char* str, size_t n, const char* format, va_list ap);
141#endif
142
3168a13f 143// ----------------------------------------------------------------------------
c801d85f 144// global functions
3168a13f 145// ----------------------------------------------------------------------------
c801d85f 146
a533f5c1 147#if defined(wxSTD_STRING_COMPATIBILITY) && wxUSE_STD_IOSTREAM
c801d85f
KB
148
149// MS Visual C++ version 5.0 provides the new STL headers as well as the old
150// iostream ones.
151//
152// ATTN: you can _not_ use both of these in the same program!
a38b83c3 153
dd107c50 154wxSTD istream& operator>>(wxSTD istream& is, wxString& WXUNUSED(str))
c801d85f
KB
155{
156#if 0
157 int w = is.width(0);
158 if ( is.ipfx(0) ) {
3f4a0c5b 159 streambuf *sb = is.rdbuf();
c801d85f
KB
160 str.erase();
161 while ( true ) {
162 int ch = sb->sbumpc ();
163 if ( ch == EOF ) {
3f4a0c5b 164 is.setstate(ios::eofbit);
c801d85f
KB
165 break;
166 }
167 else if ( isspace(ch) ) {
168 sb->sungetc();
169 break;
170 }
dd1eaa89 171
c801d85f
KB
172 str += ch;
173 if ( --w == 1 )
174 break;
175 }
176 }
177
178 is.isfx();
179 if ( str.length() == 0 )
3f4a0c5b 180 is.setstate(ios::failbit);
c801d85f
KB
181#endif
182 return is;
183}
184
dd107c50 185wxSTD ostream& operator<<(wxSTD ostream& os, const wxString& str)
825ba8f0
SB
186{
187 os << str.c_str();
188 return os;
189}
190
c801d85f
KB
191#endif //std::string compatibility
192
378b05f7
VZ
193extern int WXDLLEXPORT wxVsnprintf(wxChar *buf, size_t len,
194 const wxChar *format, va_list argptr)
195{
196#if wxUSE_UNICODE
197 // FIXME should use wvsnprintf() or whatever if it's available
198 wxString s;
199 int iLen = s.PrintfV(format, argptr);
200 if ( iLen != -1 )
201 {
7f017c64
VZ
202 wxStrncpy(buf, s.c_str(), len);
203 buf[len-1] = wxT('\0');
378b05f7
VZ
204 }
205
206 return iLen;
207#else // ANSI
b568d04f
VZ
208 // vsnprintf() will not terminate the string with '\0' if there is not
209 // enough place, but we want the string to always be NUL terminated
210 int rc = wxVsnprintfA(buf, len - 1, format, argptr);
2f02cb89
VZ
211 if ( rc == -1 )
212 {
213 buf[len] = 0;
214 }
b568d04f
VZ
215
216 return rc;
378b05f7
VZ
217#endif // Unicode/ANSI
218}
219
220extern int WXDLLEXPORT wxSnprintf(wxChar *buf, size_t len,
221 const wxChar *format, ...)
222{
223 va_list argptr;
224 va_start(argptr, format);
225
226 int iLen = wxVsnprintf(buf, len, format, argptr);
227
228 va_end(argptr);
229
230 return iLen;
231}
232
3168a13f
VZ
233// ----------------------------------------------------------------------------
234// private classes
235// ----------------------------------------------------------------------------
236
237// this small class is used to gather statistics for performance tuning
238//#define WXSTRING_STATISTICS
239#ifdef WXSTRING_STATISTICS
240 class Averager
241 {
242 public:
243 Averager(const char *sz) { m_sz = sz; m_nTotal = m_nCount = 0; }
2c3b684c 244 ~Averager()
3168a13f
VZ
245 { printf("wxString: average %s = %f\n", m_sz, ((float)m_nTotal)/m_nCount); }
246
c86f1403 247 void Add(size_t n) { m_nTotal += n; m_nCount++; }
3168a13f
VZ
248
249 private:
c86f1403 250 size_t m_nCount, m_nTotal;
3168a13f
VZ
251 const char *m_sz;
252 } g_averageLength("allocation size"),
253 g_averageSummandLength("summand length"),
254 g_averageConcatHit("hit probability in concat"),
255 g_averageInitialLength("initial string length");
256
257 #define STATISTICS_ADD(av, val) g_average##av.Add(val)
258#else
259 #define STATISTICS_ADD(av, val)
260#endif // WXSTRING_STATISTICS
261
c801d85f
KB
262// ===========================================================================
263// wxString class core
264// ===========================================================================
265
266// ---------------------------------------------------------------------------
267// construction
268// ---------------------------------------------------------------------------
269
c801d85f 270// constructs string of <nLength> copies of character <ch>
2bb67b80 271wxString::wxString(wxChar ch, size_t nLength)
c801d85f
KB
272{
273 Init();
274
275 if ( nLength > 0 ) {
276 AllocBuffer(nLength);
f1da2f03 277
2bb67b80
OK
278#if wxUSE_UNICODE
279 // memset only works on char
280 for (size_t n=0; n<nLength; n++) m_pchData[n] = ch;
281#else
c801d85f 282 memset(m_pchData, ch, nLength);
2bb67b80 283#endif
c801d85f
KB
284 }
285}
286
287// takes nLength elements of psz starting at nPos
2bb67b80 288void wxString::InitWith(const wxChar *psz, size_t nPos, size_t nLength)
c801d85f
KB
289{
290 Init();
291
f6bcfd97
BP
292 // if the length is not given, assume the string to be NUL terminated
293 if ( nLength == wxSTRING_MAXLEN ) {
294 wxASSERT_MSG( nPos <= wxStrlen(psz), _T("index out of bounds") );
c801d85f 295
f6bcfd97
BP
296 nLength = wxStrlen(psz + nPos);
297 }
6c68273f 298
3168a13f
VZ
299 STATISTICS_ADD(InitialLength, nLength);
300
c801d85f
KB
301 if ( nLength > 0 ) {
302 // trailing '\0' is written in AllocBuffer()
303 AllocBuffer(nLength);
2bb67b80 304 memcpy(m_pchData, psz + nPos, nLength*sizeof(wxChar));
c801d85f
KB
305 }
306}
dd1eaa89 307
8de2e39c 308#ifdef wxSTD_STRING_COMPATIBILITY
c801d85f 309
c801d85f
KB
310// poor man's iterators are "void *" pointers
311wxString::wxString(const void *pStart, const void *pEnd)
312{
2bb67b80
OK
313 InitWith((const wxChar *)pStart, 0,
314 (const wxChar *)pEnd - (const wxChar *)pStart);
c801d85f
KB
315}
316
317#endif //std::string compatibility
318
2bb67b80
OK
319#if wxUSE_UNICODE
320
321// from multibyte string
cf2f341a 322wxString::wxString(const char *psz, wxMBConv& conv, size_t nLength)
2bb67b80
OK
323{
324 // first get necessary size
435595e0 325 size_t nLen = psz ? conv.MB2WC((wchar_t *) NULL, psz, 0) : 0;
2bb67b80
OK
326
327 // nLength is number of *Unicode* characters here!
eea4f86a 328 if ((nLen != (size_t)-1) && (nLen > nLength))
2bb67b80
OK
329 nLen = nLength;
330
331 // empty?
eea4f86a 332 if ( (nLen != 0) && (nLen != (size_t)-1) ) {
2bb67b80
OK
333 AllocBuffer(nLen);
334 conv.MB2WC(m_pchData, psz, nLen);
335 }
336 else {
337 Init();
338 }
339}
340
e90c1d2a 341#else // ANSI
2bb67b80 342
0f3e3e0c 343#if wxUSE_WCHAR_T
c801d85f 344// from wide string
f6bcfd97 345wxString::wxString(const wchar_t *pwz, wxMBConv& conv)
c801d85f
KB
346{
347 // first get necessary size
f6bcfd97 348 size_t nLen = pwz ? conv.WC2MB((char *) NULL, pwz, 0) : 0;
c801d85f
KB
349
350 // empty?
eea4f86a 351 if ( (nLen != 0) && (nLen != (size_t)-1) ) {
c801d85f 352 AllocBuffer(nLen);
f6bcfd97 353 conv.WC2MB(m_pchData, pwz, nLen);
c801d85f
KB
354 }
355 else {
356 Init();
357 }
358}
e90c1d2a 359#endif // wxUSE_WCHAR_T
c801d85f 360
e90c1d2a 361#endif // Unicode/ANSI
2bb67b80 362
c801d85f
KB
363// ---------------------------------------------------------------------------
364// memory allocation
365// ---------------------------------------------------------------------------
366
367// allocates memory needed to store a C string of length nLen
368void wxString::AllocBuffer(size_t nLen)
369{
13111b2a
VZ
370 // allocating 0 sized buffer doesn't make sense, all empty strings should
371 // reuse g_strEmpty
372 wxASSERT( nLen > 0 );
373
374 // make sure that we don't overflow
375 wxASSERT( nLen < (INT_MAX / sizeof(wxChar)) -
376 (sizeof(wxStringData) + EXTRA_ALLOC + 1) );
c801d85f 377
3168a13f
VZ
378 STATISTICS_ADD(Length, nLen);
379
c801d85f
KB
380 // allocate memory:
381 // 1) one extra character for '\0' termination
382 // 2) sizeof(wxStringData) for housekeeping info
3168a13f 383 wxStringData* pData = (wxStringData*)
2bb67b80 384 malloc(sizeof(wxStringData) + (nLen + EXTRA_ALLOC + 1)*sizeof(wxChar));
c801d85f 385 pData->nRefs = 1;
c801d85f 386 pData->nDataLength = nLen;
3168a13f 387 pData->nAllocLength = nLen + EXTRA_ALLOC;
c801d85f 388 m_pchData = pData->data(); // data starts after wxStringData
223d09f6 389 m_pchData[nLen] = wxT('\0');
c801d85f
KB
390}
391
c801d85f
KB
392// must be called before changing this string
393void wxString::CopyBeforeWrite()
394{
395 wxStringData* pData = GetStringData();
396
397 if ( pData->IsShared() ) {
398 pData->Unlock(); // memory not freed because shared
c86f1403 399 size_t nLen = pData->nDataLength;
3168a13f 400 AllocBuffer(nLen);
2bb67b80 401 memcpy(m_pchData, pData->data(), nLen*sizeof(wxChar));
c801d85f
KB
402 }
403
3bbb630a 404 wxASSERT( !GetStringData()->IsShared() ); // we must be the only owner
c801d85f
KB
405}
406
407// must be called before replacing contents of this string
408void wxString::AllocBeforeWrite(size_t nLen)
409{
410 wxASSERT( nLen != 0 ); // doesn't make any sense
411
412 // must not share string and must have enough space
3168a13f 413 wxStringData* pData = GetStringData();
fbf0c83d 414 if ( pData->IsShared() || pData->IsEmpty() ) {
c801d85f
KB
415 // can't work with old buffer, get new one
416 pData->Unlock();
417 AllocBuffer(nLen);
418 }
471aebdd 419 else {
fbf0c83d
VZ
420 if ( nLen > pData->nAllocLength ) {
421 // realloc the buffer instead of calling malloc() again, this is more
422 // efficient
423 STATISTICS_ADD(Length, nLen);
424
425 nLen += EXTRA_ALLOC;
426
427 wxStringData *pDataOld = pData;
428 pData = (wxStringData*)
429 realloc(pData, sizeof(wxStringData) + (nLen + 1)*sizeof(wxChar));
430 if ( !pData ) {
431 // out of memory
432 free(pDataOld);
433
434 // FIXME we're going to crash...
435 return;
436 }
437
438 pData->nAllocLength = nLen;
439 m_pchData = pData->data();
440 }
441
442 // now we have enough space, just update the string length
471aebdd
VZ
443 pData->nDataLength = nLen;
444 }
c801d85f 445
f1da2f03 446 wxASSERT( !GetStringData()->IsShared() ); // we must be the only owner
c801d85f
KB
447}
448
dd1eaa89 449// allocate enough memory for nLen characters
c86f1403 450void wxString::Alloc(size_t nLen)
dd1eaa89
VZ
451{
452 wxStringData *pData = GetStringData();
453 if ( pData->nAllocLength <= nLen ) {
9fbd8b8d
VZ
454 if ( pData->IsEmpty() ) {
455 nLen += EXTRA_ALLOC;
456
457 wxStringData* pData = (wxStringData*)
2bb67b80 458 malloc(sizeof(wxStringData) + (nLen + 1)*sizeof(wxChar));
9fbd8b8d
VZ
459 pData->nRefs = 1;
460 pData->nDataLength = 0;
461 pData->nAllocLength = nLen;
462 m_pchData = pData->data(); // data starts after wxStringData
223d09f6 463 m_pchData[0u] = wxT('\0');
9fbd8b8d 464 }
3168a13f
VZ
465 else if ( pData->IsShared() ) {
466 pData->Unlock(); // memory not freed because shared
c86f1403 467 size_t nOldLen = pData->nDataLength;
3168a13f 468 AllocBuffer(nLen);
2bb67b80 469 memcpy(m_pchData, pData->data(), nOldLen*sizeof(wxChar));
3168a13f 470 }
dd1eaa89 471 else {
3168a13f
VZ
472 nLen += EXTRA_ALLOC;
473
fbf0c83d 474 wxStringData *pDataOld = pData;
dd1eaa89 475 wxStringData *p = (wxStringData *)
2bb67b80 476 realloc(pData, sizeof(wxStringData) + (nLen + 1)*sizeof(wxChar));
3168a13f
VZ
477
478 if ( p == NULL ) {
fbf0c83d
VZ
479 // don't leak memory
480 free(pDataOld);
481
482 // FIXME what to do on memory error?
3168a13f 483 return;
dd1eaa89 484 }
3168a13f
VZ
485
486 // it's not important if the pointer changed or not (the check for this
487 // is not faster than assigning to m_pchData in all cases)
488 p->nAllocLength = nLen;
489 m_pchData = p->data();
dd1eaa89
VZ
490 }
491 }
492 //else: we've already got enough
493}
494
495// shrink to minimal size (releasing extra memory)
496void wxString::Shrink()
497{
498 wxStringData *pData = GetStringData();
3bbb630a 499
fbf0c83d
VZ
500 // this variable is unused in release build, so avoid the compiler warning
501 // by just not declaring it
3bbb630a
VZ
502#ifdef __WXDEBUG__
503 void *p =
504#endif
2bb67b80 505 realloc(pData, sizeof(wxStringData) + (pData->nDataLength + 1)*sizeof(wxChar));
3bbb630a 506
fbf0c83d
VZ
507 // we rely on a reasonable realloc() implementation here - so far I haven't
508 // seen any which wouldn't behave like this
509
3168a13f 510 wxASSERT( p != NULL ); // can't free memory?
dd1eaa89
VZ
511 wxASSERT( p == pData ); // we're decrementing the size - block shouldn't move!
512}
513
c801d85f 514// get the pointer to writable buffer of (at least) nLen bytes
2bb67b80 515wxChar *wxString::GetWriteBuf(size_t nLen)
c801d85f
KB
516{
517 AllocBeforeWrite(nLen);
097c080b
VZ
518
519 wxASSERT( GetStringData()->nRefs == 1 );
520 GetStringData()->Validate(FALSE);
521
c801d85f
KB
522 return m_pchData;
523}
524
097c080b
VZ
525// put string back in a reasonable state after GetWriteBuf
526void wxString::UngetWriteBuf()
527{
2bb67b80 528 GetStringData()->nDataLength = wxStrlen(m_pchData);
097c080b
VZ
529 GetStringData()->Validate(TRUE);
530}
531
8f06a017
RD
532void wxString::UngetWriteBuf(size_t nLen)
533{
534 GetStringData()->nDataLength = nLen;
535 GetStringData()->Validate(TRUE);
536}
537
c801d85f
KB
538// ---------------------------------------------------------------------------
539// data access
540// ---------------------------------------------------------------------------
541
542// all functions are inline in string.h
543
544// ---------------------------------------------------------------------------
545// assignment operators
546// ---------------------------------------------------------------------------
547
dd1eaa89 548// helper function: does real copy
2bb67b80 549void wxString::AssignCopy(size_t nSrcLen, const wxChar *pszSrcData)
c801d85f
KB
550{
551 if ( nSrcLen == 0 ) {
552 Reinit();
553 }
554 else {
555 AllocBeforeWrite(nSrcLen);
2bb67b80 556 memcpy(m_pchData, pszSrcData, nSrcLen*sizeof(wxChar));
c801d85f 557 GetStringData()->nDataLength = nSrcLen;
223d09f6 558 m_pchData[nSrcLen] = wxT('\0');
c801d85f
KB
559 }
560}
561
562// assigns one string to another
563wxString& wxString::operator=(const wxString& stringSrc)
564{
097c080b
VZ
565 wxASSERT( stringSrc.GetStringData()->IsValid() );
566
c801d85f
KB
567 // don't copy string over itself
568 if ( m_pchData != stringSrc.m_pchData ) {
569 if ( stringSrc.GetStringData()->IsEmpty() ) {
570 Reinit();
571 }
572 else {
573 // adjust references
574 GetStringData()->Unlock();
575 m_pchData = stringSrc.m_pchData;
576 GetStringData()->Lock();
577 }
578 }
579
580 return *this;
581}
582
583// assigns a single character
2bb67b80 584wxString& wxString::operator=(wxChar ch)
c801d85f
KB
585{
586 AssignCopy(1, &ch);
587 return *this;
588}
589
a3291804 590
c801d85f 591// assigns C string
2bb67b80 592wxString& wxString::operator=(const wxChar *psz)
c801d85f 593{
2bb67b80 594 AssignCopy(wxStrlen(psz), psz);
c801d85f
KB
595 return *this;
596}
597
2bb67b80
OK
598#if !wxUSE_UNICODE
599
c801d85f
KB
600// same as 'signed char' variant
601wxString& wxString::operator=(const unsigned char* psz)
602{
603 *this = (const char *)psz;
604 return *this;
605}
606
0f3e3e0c 607#if wxUSE_WCHAR_T
c801d85f
KB
608wxString& wxString::operator=(const wchar_t *pwz)
609{
610 wxString str(pwz);
611 *this = str;
612 return *this;
613}
0f3e3e0c 614#endif
c801d85f 615
2bb67b80
OK
616#endif
617
c801d85f
KB
618// ---------------------------------------------------------------------------
619// string concatenation
620// ---------------------------------------------------------------------------
621
c801d85f 622// add something to this string
2bb67b80 623void wxString::ConcatSelf(int nSrcLen, const wxChar *pszSrcData)
c801d85f 624{
3168a13f 625 STATISTICS_ADD(SummandLength, nSrcLen);
c801d85f 626
05488905
VZ
627 // concatenating an empty string is a NOP
628 if ( nSrcLen > 0 ) {
629 wxStringData *pData = GetStringData();
630 size_t nLen = pData->nDataLength;
631 size_t nNewLen = nLen + nSrcLen;
c801d85f 632
05488905
VZ
633 // alloc new buffer if current is too small
634 if ( pData->IsShared() ) {
635 STATISTICS_ADD(ConcatHit, 0);
3168a13f 636
05488905
VZ
637 // we have to allocate another buffer
638 wxStringData* pOldData = GetStringData();
639 AllocBuffer(nNewLen);
2bb67b80 640 memcpy(m_pchData, pOldData->data(), nLen*sizeof(wxChar));
05488905
VZ
641 pOldData->Unlock();
642 }
643 else if ( nNewLen > pData->nAllocLength ) {
644 STATISTICS_ADD(ConcatHit, 0);
3168a13f 645
05488905
VZ
646 // we have to grow the buffer
647 Alloc(nNewLen);
648 }
649 else {
650 STATISTICS_ADD(ConcatHit, 1);
3168a13f 651
05488905
VZ
652 // the buffer is already big enough
653 }
3168a13f 654
05488905
VZ
655 // should be enough space
656 wxASSERT( nNewLen <= GetStringData()->nAllocLength );
3168a13f 657
05488905 658 // fast concatenation - all is done in our buffer
2bb67b80 659 memcpy(m_pchData + nLen, pszSrcData, nSrcLen*sizeof(wxChar));
3168a13f 660
223d09f6 661 m_pchData[nNewLen] = wxT('\0'); // put terminating '\0'
05488905
VZ
662 GetStringData()->nDataLength = nNewLen; // and fix the length
663 }
664 //else: the string to append was empty
c801d85f
KB
665}
666
667/*
c801d85f
KB
668 * concatenation functions come in 5 flavours:
669 * string + string
670 * char + string and string + char
671 * C str + string and string + C str
672 */
673
674wxString operator+(const wxString& string1, const wxString& string2)
675{
097c080b
VZ
676 wxASSERT( string1.GetStringData()->IsValid() );
677 wxASSERT( string2.GetStringData()->IsValid() );
678
3168a13f
VZ
679 wxString s = string1;
680 s += string2;
681
c801d85f
KB
682 return s;
683}
684
2bb67b80 685wxString operator+(const wxString& string, wxChar ch)
c801d85f 686{
3168a13f
VZ
687 wxASSERT( string.GetStringData()->IsValid() );
688
689 wxString s = string;
690 s += ch;
097c080b 691
c801d85f
KB
692 return s;
693}
694
2bb67b80 695wxString operator+(wxChar ch, const wxString& string)
c801d85f 696{
097c080b
VZ
697 wxASSERT( string.GetStringData()->IsValid() );
698
3168a13f
VZ
699 wxString s = ch;
700 s += string;
701
c801d85f
KB
702 return s;
703}
704
2bb67b80 705wxString operator+(const wxString& string, const wxChar *psz)
c801d85f 706{
097c080b
VZ
707 wxASSERT( string.GetStringData()->IsValid() );
708
c801d85f 709 wxString s;
2bb67b80 710 s.Alloc(wxStrlen(psz) + string.Len());
3168a13f
VZ
711 s = string;
712 s += psz;
713
c801d85f
KB
714 return s;
715}
716
2bb67b80 717wxString operator+(const wxChar *psz, const wxString& string)
c801d85f 718{
097c080b
VZ
719 wxASSERT( string.GetStringData()->IsValid() );
720
c801d85f 721 wxString s;
2bb67b80 722 s.Alloc(wxStrlen(psz) + string.Len());
3168a13f
VZ
723 s = psz;
724 s += string;
725
c801d85f
KB
726 return s;
727}
728
729// ===========================================================================
730// other common string functions
731// ===========================================================================
732
733// ---------------------------------------------------------------------------
734// simple sub-string extraction
735// ---------------------------------------------------------------------------
736
737// helper function: clone the data attached to this string
738void wxString::AllocCopy(wxString& dest, int nCopyLen, int nCopyIndex) const
739{
3168a13f 740 if ( nCopyLen == 0 ) {
c801d85f
KB
741 dest.Init();
742 }
3168a13f 743 else {
c801d85f 744 dest.AllocBuffer(nCopyLen);
2bb67b80 745 memcpy(dest.m_pchData, m_pchData + nCopyIndex, nCopyLen*sizeof(wxChar));
c801d85f
KB
746 }
747}
748
749// extract string of length nCount starting at nFirst
c801d85f
KB
750wxString wxString::Mid(size_t nFirst, size_t nCount) const
751{
30d9011f
VZ
752 wxStringData *pData = GetStringData();
753 size_t nLen = pData->nDataLength;
754
566b84d2
VZ
755 // default value of nCount is wxSTRING_MAXLEN and means "till the end"
756 if ( nCount == wxSTRING_MAXLEN )
30d9011f
VZ
757 {
758 nCount = nLen - nFirst;
759 }
760
c801d85f 761 // out-of-bounds requests return sensible things
30d9011f
VZ
762 if ( nFirst + nCount > nLen )
763 {
764 nCount = nLen - nFirst;
765 }
c801d85f 766
30d9011f
VZ
767 if ( nFirst > nLen )
768 {
769 // AllocCopy() will return empty string
c801d85f 770 nCount = 0;
30d9011f 771 }
c801d85f
KB
772
773 wxString dest;
774 AllocCopy(dest, nCount, nFirst);
30d9011f 775
c801d85f
KB
776 return dest;
777}
778
f6bcfd97
BP
779// check that the tring starts with prefix and return the rest of the string
780// in the provided pointer if it is not NULL, otherwise return FALSE
781bool wxString::StartsWith(const wxChar *prefix, wxString *rest) const
782{
783 wxASSERT_MSG( prefix, _T("invalid parameter in wxString::StartsWith") );
784
785 // first check if the beginning of the string matches the prefix: note
786 // that we don't have to check that we don't run out of this string as
787 // when we reach the terminating NUL, either prefix string ends too (and
788 // then it's ok) or we break out of the loop because there is no match
789 const wxChar *p = c_str();
790 while ( *prefix )
791 {
792 if ( *prefix++ != *p++ )
793 {
794 // no match
795 return FALSE;
796 }
797 }
798
799 if ( rest )
800 {
801 // put the rest of the string into provided pointer
802 *rest = p;
803 }
804
805 return TRUE;
806}
807
c801d85f
KB
808// extract nCount last (rightmost) characters
809wxString wxString::Right(size_t nCount) const
810{
811 if ( nCount > (size_t)GetStringData()->nDataLength )
812 nCount = GetStringData()->nDataLength;
813
814 wxString dest;
815 AllocCopy(dest, nCount, GetStringData()->nDataLength - nCount);
816 return dest;
817}
818
819// get all characters after the last occurence of ch
820// (returns the whole string if ch not found)
2bb67b80 821wxString wxString::AfterLast(wxChar ch) const
c801d85f
KB
822{
823 wxString str;
824 int iPos = Find(ch, TRUE);
3c67202d 825 if ( iPos == wxNOT_FOUND )
c801d85f
KB
826 str = *this;
827 else
c8cfb486 828 str = c_str() + iPos + 1;
c801d85f
KB
829
830 return str;
831}
832
833// extract nCount first (leftmost) characters
834wxString wxString::Left(size_t nCount) const
835{
836 if ( nCount > (size_t)GetStringData()->nDataLength )
837 nCount = GetStringData()->nDataLength;
838
839 wxString dest;
840 AllocCopy(dest, nCount, 0);
841 return dest;
842}
843
844// get all characters before the first occurence of ch
845// (returns the whole string if ch not found)
2bb67b80 846wxString wxString::BeforeFirst(wxChar ch) const
c801d85f
KB
847{
848 wxString str;
223d09f6 849 for ( const wxChar *pc = m_pchData; *pc != wxT('\0') && *pc != ch; pc++ )
c801d85f
KB
850 str += *pc;
851
852 return str;
853}
854
855/// get all characters before the last occurence of ch
856/// (returns empty string if ch not found)
2bb67b80 857wxString wxString::BeforeLast(wxChar ch) const
c801d85f
KB
858{
859 wxString str;
860 int iPos = Find(ch, TRUE);
3c67202d 861 if ( iPos != wxNOT_FOUND && iPos != 0 )
d1c9bbf6 862 str = wxString(c_str(), iPos);
c801d85f
KB
863
864 return str;
865}
866
867/// get all characters after the first occurence of ch
868/// (returns empty string if ch not found)
2bb67b80 869wxString wxString::AfterFirst(wxChar ch) const
c801d85f
KB
870{
871 wxString str;
872 int iPos = Find(ch);
3c67202d 873 if ( iPos != wxNOT_FOUND )
c801d85f
KB
874 str = c_str() + iPos + 1;
875
876 return str;
877}
878
879// replace first (or all) occurences of some substring with another one
2bb67b80 880size_t wxString::Replace(const wxChar *szOld, const wxChar *szNew, bool bReplaceAll)
c801d85f 881{
c86f1403 882 size_t uiCount = 0; // count of replacements made
c801d85f 883
2bb67b80 884 size_t uiOldLen = wxStrlen(szOld);
c801d85f
KB
885
886 wxString strTemp;
2bb67b80
OK
887 const wxChar *pCurrent = m_pchData;
888 const wxChar *pSubstr;
223d09f6 889 while ( *pCurrent != wxT('\0') ) {
2bb67b80 890 pSubstr = wxStrstr(pCurrent, szOld);
c801d85f
KB
891 if ( pSubstr == NULL ) {
892 // strTemp is unused if no replacements were made, so avoid the copy
893 if ( uiCount == 0 )
894 return 0;
895
896 strTemp += pCurrent; // copy the rest
897 break; // exit the loop
898 }
899 else {
900 // take chars before match
901 strTemp.ConcatSelf(pSubstr - pCurrent, pCurrent);
902 strTemp += szNew;
903 pCurrent = pSubstr + uiOldLen; // restart after match
904
905 uiCount++;
906
907 // stop now?
908 if ( !bReplaceAll ) {
909 strTemp += pCurrent; // copy the rest
910 break; // exit the loop
911 }
912 }
913 }
914
915 // only done if there were replacements, otherwise would have returned above
916 *this = strTemp;
917
918 return uiCount;
919}
920
921bool wxString::IsAscii() const
922{
2bb67b80 923 const wxChar *s = (const wxChar*) *this;
c801d85f
KB
924 while(*s){
925 if(!isascii(*s)) return(FALSE);
926 s++;
927 }
928 return(TRUE);
929}
dd1eaa89 930
c801d85f
KB
931bool wxString::IsWord() const
932{
2bb67b80 933 const wxChar *s = (const wxChar*) *this;
c801d85f 934 while(*s){
2bb67b80 935 if(!wxIsalpha(*s)) return(FALSE);
c801d85f
KB
936 s++;
937 }
938 return(TRUE);
939}
dd1eaa89 940
c801d85f
KB
941bool wxString::IsNumber() const
942{
2bb67b80 943 const wxChar *s = (const wxChar*) *this;
2f74ed28
GT
944 if (wxStrlen(s))
945 if ((s[0] == '-') || (s[0] == '+')) s++;
c801d85f 946 while(*s){
2bb67b80 947 if(!wxIsdigit(*s)) return(FALSE);
c801d85f
KB
948 s++;
949 }
950 return(TRUE);
951}
952
c801d85f
KB
953wxString wxString::Strip(stripType w) const
954{
955 wxString s = *this;
956 if ( w & leading ) s.Trim(FALSE);
957 if ( w & trailing ) s.Trim(TRUE);
958 return s;
959}
960
c801d85f
KB
961// ---------------------------------------------------------------------------
962// case conversion
963// ---------------------------------------------------------------------------
964
965wxString& wxString::MakeUpper()
966{
967 CopyBeforeWrite();
968
2bb67b80
OK
969 for ( wxChar *p = m_pchData; *p; p++ )
970 *p = (wxChar)wxToupper(*p);
c801d85f
KB
971
972 return *this;
973}
974
975wxString& wxString::MakeLower()
976{
977 CopyBeforeWrite();
dd1eaa89 978
2bb67b80
OK
979 for ( wxChar *p = m_pchData; *p; p++ )
980 *p = (wxChar)wxTolower(*p);
c801d85f
KB
981
982 return *this;
983}
984
985// ---------------------------------------------------------------------------
986// trimming and padding
987// ---------------------------------------------------------------------------
988
576c608d
VZ
989// some compilers (VC++ 6.0 not to name them) return TRUE for a call to
990