DuiLib 源码分析之CDuiString

时间:2021-02-13 05:20:14

duilib是一个比较常见的界面库,闲来无事看看别人写的代码,跟自己写的一比,

才看到了差距呀,感觉自己写的乱七八糟,keep moving

CduiString是duilib提供的一个字符串类,功能是够用的,做duilib项目可以直接拿来用

首先看看头文件定义:

 class UILIB_API CDuiString
{
public:
enum { MAX_LOCAL_STRING_LEN = }; CDuiString();
CDuiString(const TCHAR ch);
CDuiString(const CDuiString& src);
CDuiString(LPCTSTR lpsz, int nLen = -);
~CDuiString(); void Empty();
int GetLength() const;
bool IsEmpty() const;
TCHAR GetAt(int nIndex) const;
void Append(LPCTSTR pstr);
void Assign(LPCTSTR pstr, int nLength = -);
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 = -) const;
CDuiString Right(int nLength) const; int Find(TCHAR ch, int iPos = ) const;
int Find(LPCTSTR pstr, int iPos = ) 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 + ];//储存字符串的字符数组
};

看了定义之后其中的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 + ) * sizeof(TCHAR)));
_tcscpy(m_pstr, m_szBuffer);
_tcscat(m_pstr, pstr);
}
else {
m_pstr = static_cast<LPTSTR>(realloc(m_pstr, (nNewLength + ) * 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 < ? (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 + ) * 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 = ;
int iPos = Find(pstrFrom);
if( iPos < ) return ;
int cchFrom = (int) _tcslen(pstrFrom);
int cchTo = (int) _tcslen(pstrTo);
while( iPos >= ) {
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, , pstrFormat, argList);
szSprintf = (TCHAR*)malloc((nLen + ) * sizeof(TCHAR));
ZeroMemory(szSprintf, (nLen + ) * sizeof(TCHAR));
int iRet = ::_vsntprintf(szSprintf, nLen + , 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

其实难点也并非很难,只是一些需要注意的小问题总是会被自己忽略,看别人写的好的代码还是挺有收获的。