joachim99@75: /*
joachim99@75:  * Copyright (c) 2003, Sergey Zorin. All rights reserved.
joachim99@75:  *
joachim99@75:  * This software is distributable under the BSD license. See the terms of the
joachim99@75:  * BSD license in the LICENSE file provided with this software.
joachim99@75:  *
joachim99@75:  */
joachim99@75: 
joachim99@75: #ifndef __string_h__
joachim99@75: #define __string_h__
joachim99@75: 
joachim99@75: #include <windows.h>
joachim99@75: #include <tchar.h>
joachim99@75: 
joachim99@75: #include <string.h>
joachim99@75: #include <stdlib.h>
joachim99@75: 
joachim99@75: class STRING;
joachim99@75: inline STRING operator+( const STRING& s1, const STRING& s2);
joachim99@75: 
joachim99@75: class STRING {
joachim99@75:   public:
joachim99@75:     static const int begin = 0;
joachim99@75:     static const int end = -1;
joachim99@75:   
joachim99@75:   public:
joachim99@75:     STRING(const STRING& s) {
joachim99@75:       _str = new TCHAR[s.length()+1];
joachim99@75:       lstrcpy(_str, s);
joachim99@75:     }
joachim99@75:     
joachim99@75:     STRING(const TCHAR* str = TEXT("")) {
joachim99@75:       _str = new TCHAR[lstrlen(str)+1];
joachim99@75:       lstrcpy(_str, str);
joachim99@75:     }
joachim99@75:     
joachim99@75:     ~STRING() {
joachim99@75:       delete[] _str;
joachim99@75:     }
joachim99@75: 
joachim99@75:     void resize( size_t newLength )
joachim99@75:     {
joachim99@75:        size_t oldLength = length();
joachim99@75:        if ( newLength < oldLength ) {
joachim99@75:           _str[newLength] = 0; // Just truncate the string
joachim99@75:        } else if( newLength>oldLength) {
joachim99@75:           TCHAR* p = new TCHAR[ newLength + 1 ];
joachim99@75:           lstrcpy(p, _str);
joachim99@75:           for( size_t i=oldLength; i<newLength; ++i)
joachim99@75:              p[i]=TEXT(' ');
joachim99@75:           p[newLength]=0;
joachim99@75:        }
joachim99@75:     }
joachim99@75:     
joachim99@75:     STRING& operator=(const STRING& s) {
joachim99@75:       delete[] _str;
joachim99@75:       _str = new TCHAR[s.length()+1];
joachim99@75:       lstrcpy(_str, s);
joachim99@75:       return *this;
joachim99@75:     }
joachim99@75:     
joachim99@75:     operator TCHAR*() {
joachim99@75:       return _str;
joachim99@75:     }
joachim99@75:     
joachim99@75:     operator const TCHAR*() const {
joachim99@75:       return _str;
joachim99@75:     }
joachim99@75: 
joachim99@75:     const TCHAR* c_str() const {
joachim99@75:        return _str;
joachim99@75:     }
joachim99@75:     
joachim99@75:     size_t length() const {
joachim99@75:       return _tcslen(_str);
joachim99@75:     }
joachim99@75:     
joachim99@75:     // Also returns the length. Behaviour like std::basic_string::size.
joachim99@75:     // See also sizeInBytes() below.
joachim99@75:     size_t size() const {
joachim99@75:       return length();
joachim99@75:     }
joachim99@75: 
joachim99@75:     // String length in bytes. May differ from length() for Unicode or MBCS
joachim99@75:     size_t sizeInBytes() const {
joachim99@75:       return length()*sizeof(TCHAR);
joachim99@75:     }
joachim99@75: 
joachim99@75:     bool empty() const
joachim99@75:     {
joachim99@75:        return length()==0;
joachim99@75:     }
joachim99@75:     
joachim99@75:     STRING substr(size_t from, size_t len=size_t(-1)) const {
joachim99@75:       STRING tmp;
joachim99@75:       size_t to = len==size_t(-1) ? length() : from + len;
joachim99@75:             
joachim99@75:       if(from < to && (to <= length())) {
joachim99@75:         size_t new_len = to - from + 1;
joachim99@75:         TCHAR* new_str = new TCHAR[new_len+1];
joachim99@75:         lstrcpyn(new_str, &_str[from], int(new_len) );
joachim99@75:         new_str[new_len] = 0;
joachim99@75:         
joachim99@75:         tmp = new_str;
joachim99@75:         delete[] new_str;
joachim99@75:       }
joachim99@75:       
joachim99@75:       return tmp;
joachim99@75:     }
joachim99@75: 
joachim99@75:     STRING& replace( size_t pos, size_t num, const STRING& s )
joachim99@75:     {
joachim99@75:        *this = substr( 0, pos ) + s + substr( pos+num );
joachim99@75:        return *this;
joachim99@75:     }
joachim99@75:     
joachim99@75:     bool operator ==(const STRING& s) const {
joachim99@75:       return (lstrcmp(_str, s) == 0);
joachim99@75:     }
joachim99@75: 
joachim99@75:     size_t find(const STRING& s) const
joachim99@75:     {
joachim99@75:        const TCHAR* p = _tcsstr( _str, s._str );
joachim99@75:        if (p)
joachim99@75:           return p - _str;
joachim99@75:        else
joachim99@75:           return size_t(-1);
joachim99@75:     }
joachim99@75:     
joachim99@75:     STRING& operator +=(const STRING& s) {
joachim99@75:       TCHAR* str = new TCHAR[lstrlen(_str)+s.length()+1];
joachim99@75: 
joachim99@75:       lstrcpy(str, _str);
joachim99@75:       lstrcat(str, s);
joachim99@75:       
joachim99@75:       delete[] _str;
joachim99@75:       
joachim99@75:       _str = str;
joachim99@75:       
joachim99@75:       return *this;
joachim99@75:     }
joachim99@75: 
joachim99@75:   private:
joachim99@75:     TCHAR* _str;
joachim99@75: };
joachim99@75: 
joachim99@75: inline STRING operator+( const STRING& s1, const STRING& s2) {
joachim99@75:   STRING tmp(s1);
joachim99@75:   
joachim99@75:   tmp+=s2;
joachim99@75:   
joachim99@75:   return tmp;
joachim99@75: }
joachim99@75: 
joachim99@75: 
joachim99@75: 
joachim99@75: #endif // __string_h__