X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/dab58492b2f5f7924755ba6b1064538a004632e0..402c3347d26736250a43550ee8253c03fe9052e0:/src/common/tokenzr.cpp?ds=sidebyside diff --git a/src/common/tokenzr.cpp b/src/common/tokenzr.cpp index 19ae5e0f29..8c29cd7cfa 100644 --- a/src/common/tokenzr.cpp +++ b/src/common/tokenzr.cpp @@ -1,128 +1,296 @@ ///////////////////////////////////////////////////////////////////////////// -// Name: tokenzr.cpp +// Name: src/common/tokenzr.cpp // Purpose: String tokenizer // Author: Guilhem Lavaux -// Modified by: +// Modified by: Vadim Zeitlin (almost full rewrite) // Created: 04/22/98 // RCS-ID: $Id$ // Copyright: (c) Guilhem Lavaux // Licence: wxWindows licence ///////////////////////////////////////////////////////////////////////////// -#ifdef __GNUG__ -#pragma implementation "tokenzr.h" -#endif +// ============================================================================ +// declarations +// ============================================================================ + +// ---------------------------------------------------------------------------- +// headers +// ---------------------------------------------------------------------------- // For compilers that support precompilation, includes "wx.h". #include "wx/wxprec.h" #ifdef __BORLANDC__ -#pragma hdrstop + #pragma hdrstop #endif +#include "wx/tokenzr.h" + #ifndef WX_PRECOMP + #include "wx/arrstr.h" + #include "wx/crt.h" #endif -#include "wx/object.h" -#include "wx/string.h" -#include "wx/tokenzr.h" +// Required for wxIs... functions +#include + +// ============================================================================ +// implementation +// ============================================================================ + +// ---------------------------------------------------------------------------- +// helpers +// ---------------------------------------------------------------------------- -wxStringTokenizer::wxStringTokenizer(const wxString& to_tokenize, +static wxString::const_iterator +find_first_of(const wxChar *delims, size_t len, + const wxString::const_iterator& from, + const wxString::const_iterator& end) +{ + wxASSERT_MSG( from <= end, _T("invalid index") ); + + for ( wxString::const_iterator i = from; i != end; ++i ) + { + if ( wxTmemchr(delims, *i, len) ) + return i; + } + + return end; +} + +static wxString::const_iterator +find_first_not_of(const wxChar *delims, size_t len, + const wxString::const_iterator& from, + const wxString::const_iterator& end) +{ + wxASSERT_MSG( from <= end, _T("invalid index") ); + + for ( wxString::const_iterator i = from; i != end; ++i ) + { + if ( !wxTmemchr(delims, *i, len) ) + return i; + } + + return end; +} + +// ---------------------------------------------------------------------------- +// wxStringTokenizer construction +// ---------------------------------------------------------------------------- + +wxStringTokenizer::wxStringTokenizer(const wxString& str, const wxString& delims, - bool ret_delims) - : wxObject() + wxStringTokenizerMode mode) { - m_string = to_tokenize; - m_delims = delims; - m_retdelims = ret_delims; + SetString(str, delims, mode); } -wxStringTokenizer::~wxStringTokenizer() +void wxStringTokenizer::SetString(const wxString& str, + const wxString& delims, + wxStringTokenizerMode mode) { + if ( mode == wxTOKEN_DEFAULT ) + { + // by default, we behave like strtok() if the delimiters are only + // whitespace characters and as wxTOKEN_RET_EMPTY otherwise (for + // whitespace delimiters, strtok() behaviour is better because we want + // to count consecutive spaces as one delimiter) + wxString::const_iterator p; + for ( p = delims.begin(); p != delims.end(); ++p ) + { + if ( !wxIsspace(*p) ) + break; + } + + if ( p != delims.end() ) + { + // not whitespace char in delims + mode = wxTOKEN_RET_EMPTY; + } + else + { + // only whitespaces + mode = wxTOKEN_STRTOK; + } + } + +#if wxUSE_UNICODE // FIXME-UTF8: only wc_str() + m_delims = delims.wc_str(); +#else + m_delims = delims.mb_str(); +#endif + m_delimsLen = delims.length(); + + m_mode = mode; + + Reinit(str); } -off_t wxStringTokenizer::FindDelims(const wxString& str, const wxString& delims) +void wxStringTokenizer::Reinit(const wxString& str) { - int i, j; - register int s_len = str.Length(), - len = delims.Length(); - - for (i=0;im_hasMoreTokens = + r ? MoreTokens_Yes : MoreTokens_No; + return r; + } + else + return m_hasMoreTokens == MoreTokens_Yes; } -bool wxStringTokenizer::HasMoreToken() +bool wxStringTokenizer::DoHasMoreTokens() const { - return (m_string.Length() != 0); + wxCHECK_MSG( IsOk(), false, _T("you should call SetString() first") ); + + if ( find_first_not_of(m_delims, m_delimsLen, m_pos, m_stringEnd) + != m_stringEnd ) + { + // there are non delimiter characters left, so we do have more tokens + return true; + } + + switch ( m_mode ) + { + case wxTOKEN_RET_EMPTY: + case wxTOKEN_RET_DELIMS: + // special hack for wxTOKEN_RET_EMPTY: we should return the initial + // empty token even if there are only delimiters after it + return !m_string.empty() && m_pos == m_string.begin(); + + case wxTOKEN_RET_EMPTY_ALL: + // special hack for wxTOKEN_RET_EMPTY_ALL: we can know if we had + // already returned the trailing empty token after the last + // delimiter by examining m_lastDelim: it is set to NUL if we run + // up to the end of the string in GetNextToken(), but if it is not + // NUL yet we still have this last token to return even if m_pos is + // already at m_string.length() + return m_pos < m_stringEnd || m_lastDelim != _T('\0'); + + case wxTOKEN_INVALID: + case wxTOKEN_DEFAULT: + wxFAIL_MSG( _T("unexpected tokenizer mode") ); + // fall through + + case wxTOKEN_STRTOK: + // never return empty delimiters + break; + } + + return false; +} + +// count the number of (remaining) tokens in the string +size_t wxStringTokenizer::CountTokens() const +{ + wxCHECK_MSG( IsOk(), 0, _T("you should call SetString() first") ); + + // VZ: this function is IMHO not very useful, so it's probably not very + // important if its implementation here is not as efficient as it + // could be -- but OTOH like this we're sure to get the correct answer + // in all modes + wxStringTokenizer tkz(wxString(m_pos, m_stringEnd), m_delims, m_mode); + + size_t count = 0; + while ( tkz.HasMoreTokens() ) + { + count++; + + (void)tkz.GetNextToken(); + } + + return count; } -// AVS - added to fix leading whitespace / mult. delims bugs -void wxStringTokenizer::EatLeadingDelims() +// ---------------------------------------------------------------------------- +// token extraction +// ---------------------------------------------------------------------------- + +wxString wxStringTokenizer::GetNextToken() { - int pos; + wxString token; + do + { + if ( !HasMoreTokens() ) + { + break; + } + + m_hasMoreTokens = MoreTokens_Unknown; + + // find the end of this token + wxString::const_iterator pos = + find_first_of(m_delims, m_delimsLen, m_pos, m_stringEnd); + + // and the start of the next one + if ( pos == m_stringEnd ) + { + // no more delimiters, the token is everything till the end of + // string + token.assign(m_pos, m_stringEnd); + + // skip the token + m_pos = m_stringEnd; - while ((pos=FindDelims(m_string, m_delims))==0) { // while leading delims - m_string = m_string.Mid((size_t)1); // trim 'em from the left - } + // it wasn't terminated + m_lastDelim = _T('\0'); + } + else // we found a delimiter at pos + { + // in wxTOKEN_RET_DELIMS mode we return the delimiter character + // with token, otherwise leave it out + wxString::const_iterator tokenEnd(pos); + if ( m_mode == wxTOKEN_RET_DELIMS ) + ++tokenEnd; + + token.assign(m_pos, tokenEnd); + + // skip the token and the trailing delimiter + m_pos = pos + 1; + + m_lastDelim = (pos == m_stringEnd) ? _T('\0') : (wxChar)*pos; + } + } + while ( !AllowEmpty() && token.empty() ); + + return token; } -wxString wxStringTokenizer::NextToken() +// ---------------------------------------------------------------------------- +// public functions +// ---------------------------------------------------------------------------- + +wxArrayString wxStringTokenize(const wxString& str, + const wxString& delims, + wxStringTokenizerMode mode) { - register off_t pos, pos2; - wxString r_string; - - if (m_string.IsNull()) - return m_string; - - if (!m_retdelims) - EatLeadingDelims(); // AVS - added to fix leading whitespace / - // mult. delims bugs - - pos = FindDelims(m_string, m_delims); - if (pos == -1) { - r_string = m_string; - m_string = wxEmptyString; - - return r_string; - } - - if (m_retdelims) { - if (!pos) { - pos++; - pos2 = 1; - } else - pos2 = pos; - } else - pos2 = pos + 1; - - r_string = m_string.Left((size_t)pos); - m_string = m_string.Mid((size_t)pos2); - - return r_string; + wxArrayString tokens; + wxStringTokenizer tk(str, delims, mode); + while ( tk.HasMoreTokens() ) + { + tokens.Add(tk.GetNextToken()); + } + + return tokens; }