DuiLib 源碼分析之CDuiString

duilib是一個比較常見的界面庫，閑來無事看看別人寫的代碼，跟自己寫的一比，

才看到了差距呀，感覺自己寫的亂七八糟，keep moving

CduiString是duilib提供的一個字符串類，功能是夠用的，做duilib項目可以直接拿來用

首先看看頭文件定義：

class UILIB_API CDuiString
    {
    public:
        enum { MAX_LOCAL_STRING_LEN = 63 };

        CDuiString();
        CDuiString(const TCHAR ch);
        CDuiString(const CDuiString& src);
        CDuiString(LPCTSTR lpsz, int nLen = -1);
        ~CDuiString();

        void Empty();
        int GetLength() const;
        bool IsEmpty() const;
        TCHAR GetAt(int nIndex) const;
        void Append(LPCTSTR pstr);
        void Assign(LPCTSTR pstr, int nLength = -1);
        LPCTSTR GetData() const;

        void SetAt(int nIndex, TCHAR ch);
        operator LPCTSTR() const;

        TCHAR operator[] (int nIndex) const;
        const CDuiString& operator=(const CDuiString& src);
        const CDuiString& operator=(const TCHAR ch);
        const CDuiString& operator=(LPCTSTR pstr);
#ifdef _UNICODE
        const CDuiString& CDuiString::operator=(LPCSTR lpStr);
        const CDuiString& CDuiString::operator+=(LPCSTR lpStr);
#else
        const CDuiString& CDuiString::operator=(LPCWSTR lpwStr);
        const CDuiString& CDuiString::operator+=(LPCWSTR lpwStr);
#endif
        CDuiString operator+(const CDuiString& src) const;
        CDuiString operator+(LPCTSTR pstr) const;
        const CDuiString& operator+=(const CDuiString& src);
        const CDuiString& operator+=(LPCTSTR pstr);
        const CDuiString& operator+=(const TCHAR ch);

        bool operator == (LPCTSTR str) const;
        bool operator != (LPCTSTR str) const;
        bool operator <= (LPCTSTR str) const;
        bool operator <  (LPCTSTR str) const;
        bool operator >= (LPCTSTR str) const;
        bool operator >  (LPCTSTR str) const;

        int Compare(LPCTSTR pstr) const;
        int CompareNoCase(LPCTSTR pstr) const;

        void MakeUpper();
        void MakeLower();

        CDuiString Left(int nLength) const;
        CDuiString Mid(int iPos, int nLength = -1) const;
        CDuiString Right(int nLength) const;

        int Find(TCHAR ch, int iPos = 0) const;
        int Find(LPCTSTR pstr, int iPos = 0) const;
        int ReverseFind(TCHAR ch) const;
        int Replace(LPCTSTR pstrFrom, LPCTSTR pstrTo);

        int __cdecl Format(LPCTSTR pstrFormat, ...);
        int __cdecl SmallFormat(LPCTSTR pstrFormat, ...);

    protected:
        LPTSTR m_pstr;//指向緩沖區的指針，可能指向m_szBuffer,也可能指向別的緩沖區
        TCHAR m_szBuffer[MAX_LOCAL_STRING_LEN + 1];//儲存字符串的字符數組
    };

看了定義之后其中的api大部分都會使用了吧

下面抓幾個重要的函數來分析一下吧.

void CDuiString::Append(LPCTSTR pstr)
    {
        int nNewLength = GetLength() + (int) _tcslen(pstr);
        if( nNewLength >= MAX_LOCAL_STRING_LEN ) {
            if( m_pstr == m_szBuffer ) {
                m_pstr = static_cast<LPTSTR>(malloc((nNewLength + 1) * sizeof(TCHAR)));
                _tcscpy(m_pstr, m_szBuffer);
                _tcscat(m_pstr, pstr);
            }
            else {
                m_pstr = static_cast<LPTSTR>(realloc(m_pstr, (nNewLength + 1) * sizeof(TCHAR)));
                _tcscat(m_pstr, pstr);
            }
        }
        else {
            if( m_pstr != m_szBuffer ) {//防止m_pstr指向別的地方
                free(m_pstr);
                m_pstr = m_szBuffer;
            }
            _tcscat(m_szBuffer, pstr);
        }
    }

Append函數用於在末尾添加字符串，首先判斷添加后的的字符串長度是否比默認數組中能儲存的長。1.超過了：先判斷當前指針是否指向m_szBuffer，是的話則新申請一段內存，將原有的內容復制過去，再加上新增的即可，否的話先釋放m_pstr中的內存再復制內容，新增。（注意這里realloc函數會在申請內存給m_pstr后將原有內容復制過去，並且釋放原來m_pstr的內存）2.沒有超過則直接添加即可

void CDuiString::Assign(LPCTSTR pstr, int cchMax)
    {
        if( pstr == NULL ) pstr = _T("");
        cchMax = (cchMax < 0 ? (int) _tcslen(pstr) : cchMax);
        if( cchMax < MAX_LOCAL_STRING_LEN ) {
            if( m_pstr != m_szBuffer ) {
                free(m_pstr);
                m_pstr = m_szBuffer;
            }
        }
        else if( cchMax > GetLength() || m_pstr == m_szBuffer ) {
            if( m_pstr == m_szBuffer ) m_pstr = NULL;//若指向m_szBuffer，不能將其釋放，先指向NULL
            m_pstr = static_cast<LPTSTR>(realloc(m_pstr, (cchMax + 1) * sizeof(TCHAR)));
        }
        _tcsncpy(m_pstr, pstr, cchMax);
        m_pstr[cchMax] = _T('\0');
    }

Assign這個函數設計的十分精妙，很多地方都用到，它主要用於重新賦值操作。cchMax表示截取pstr的范圍[0,cchMax),若m_szBuffer長度足以存儲，則直接賦值，把指針指過去；若不夠長，重新申請內存賦值

int CDuiString::Replace(LPCTSTR pstrFrom, LPCTSTR pstrTo)
    {
        CDuiString sTemp;
        int nCount = 0;
        int iPos = Find(pstrFrom);
        if( iPos < 0 ) return 0;
        int cchFrom = (int) _tcslen(pstrFrom);
        int cchTo = (int) _tcslen(pstrTo);
        while( iPos >= 0 ) {
            sTemp = Left(iPos);
            sTemp += pstrTo;
            sTemp += Mid(iPos + cchFrom);
            Assign(sTemp);
            iPos = Find(pstrFrom, iPos + cchTo);
            nCount++;
        }
        return nCount;
    }

替換字符串，首先查找是否存在需要替換的字符串，沒有則直接返回0，存在的話，將其分為三段，中間一段替換，用sTemp儲存，再重新賦值sTemp，如此往復替換所有的pstrFrom。

int CDuiString::Format(LPCTSTR pstrFormat, ...)
    {
        LPTSTR szSprintf = NULL;
        va_list argList;
        int nLen;
        va_start(argList, pstrFormat);
        nLen = ::_vsntprintf(NULL, 0, pstrFormat, argList);
        szSprintf = (TCHAR*)malloc((nLen + 1) * sizeof(TCHAR));
        ZeroMemory(szSprintf, (nLen + 1) * sizeof(TCHAR));
        int iRet = ::_vsntprintf(szSprintf, nLen + 1, pstrFormat, argList);
        va_end(argList);
        Assign(szSprintf);
        free(szSprintf);
        return iRet;
    }

Format函數用於格式化字符串，與常用的printf類似。具體實現來看看，va_list argList, 這個是什么東東，一開始確實沒見過，看看其定義

typedef char *  va_list;原來是char*，它是用於指向可變參數的，va_start（argList,pstrFormat)后，argList指向第一個可變參數，接下來調用_vsntprintf(NULL,0,pstrFormat,argList)，最后將格式化后的字符串重新賦值回去即可。

想了解關於va_list 可參考這里：http://blog.csdn.net/edonlii/article/details/8497704

其實難點也並非很難，只是一些需要注意的小問題總是會被自己忽略，看別人寫的好的代碼還是挺有收獲的。