wb-string.h

Go to the documentation of this file.

// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Copyright 2014-2015 Tsinghua University
// Author: wb.th08@gmail.com (Bin Wang), ozj@tsinghua.edu.cn (Zhijian Ou)
//
// All h, cpp, cc, and script files (e.g. bat, sh, pl, py) should include the above
// license declaration. Different coding language may use different comment styles.


#pragma once
#include <cstring>
#include <algorithm>
#include <cstdarg>  //include va_list
#include <string.h>
using namespace std;
#include "wb-vector.h"
#include "wb-linux.h"


namespace wb
{

    const int cn_default_str_len = 10;  
    const int cn_default_max_len = 32 * 1024;  


    class String
    {
    private:
        char *m_pBuffer;  
        int m_nBufSize;   

        char *m_tokPtr; 
    public:
        String(int p_nLen = cn_default_str_len);
        String(const char *p_str);
        String(const char* p_str, int nLen);
        String(char c);
        String(const String &p_str);
        ~String();
        /* If the nLen<=m_nBufSize, then do nothing. \n
           If the nLen>m_nBufSize, then re-alloc memory.
           Return the buffer pointer.
        */
        char* Reset(int nLen);
        char *GetBuffer() const { return m_pBuffer; }
        char *End() const { return m_pBuffer + strlen(m_pBuffer) - 1; }
        int GetSize() const { return m_nBufSize; }
        int GetLength() const { return strlen(m_pBuffer); }
        void Clean() { m_pBuffer[0] = '\0'; }
        const char* Format(const char* p_pMessage, ...);
        void operator = (const String &p_str);
        void operator = (const char *pStr);
        operator char*() const { return m_pBuffer; }
        char &operator[] (int i) const { return m_pBuffer[i]; }
        bool operator >(const String &p_str) { return strcmp(m_pBuffer, p_str.m_pBuffer) > 0; }
        bool operator < (const String &p_str) { return strcmp(m_pBuffer, p_str.m_pBuffer) < 0; }
        bool operator >= (const String &p_str) { return strcmp(m_pBuffer, p_str.m_pBuffer) >= 0; }
        bool operator <= (const String &p_str) { return strcmp(m_pBuffer, p_str.m_pBuffer) <= 0; }
        bool operator == (const String &p_str) { return strcmp(m_pBuffer, p_str.m_pBuffer) == 0; }
        bool operator != (const String &p_str) { return strcmp(m_pBuffer, p_str.m_pBuffer) != 0; }
        String operator += (const String &str) { strcat(Reset(strlen(m_pBuffer) + strlen(str) + 1), str.m_pBuffer); return *this; }
        bool operator > (const char *p) { return strcmp(m_pBuffer, p) > 0; }
        bool operator < (const char *p) { return strcmp(m_pBuffer, p) < 0; }
        bool operator >= (const char *p) { return strcmp(m_pBuffer, p) >= 0; }
        bool operator <= (const char *p) { return strcmp(m_pBuffer, p) <= 0; }
        bool operator == (const char *p) { return strcmp(m_pBuffer, p) == 0; }
        bool operator != (const char *p) { return strcmp(m_pBuffer, p) != 0; }
        String operator += (const char *p) { strcat(Reset(strlen(m_pBuffer) + strlen(p) + 1), p); return *this; }

        friend String operator + (const String &str, const char *p) {
            return String(str) += p;
        }
        friend String operator + (const char *p, const String &str) {
            return String(p) += str;
        }
        friend String operator + (const String &str1, const String &str2) {
            return String(str1) += str2;
        }
        char *Toupper();
        char *Tolower();
        void DeleteSub(int nLocal, int nLen);
        int Find(const char *sub);
        String Replace(const char *src, const char *rpl);
        inline char *TokBegin(const char *p);
        inline char *TokSub(const char *p);
        void Split(Array<String> &aStrs, const char* delimiter);
        String FileName();
    };

}