268 lines
6.7 KiB
C++
268 lines
6.7 KiB
C++
#ifndef _WYLIB_WIDESTRING_H_
|
||
#define _WYLIB_WIDESTRING_H_
|
||
|
||
/******************************************************************
|
||
*
|
||
* wyLib库 2008 - 2010
|
||
*
|
||
* $ UCS2字符串类 $
|
||
*
|
||
* - 主要功能 -
|
||
*
|
||
* 实现UCS字符串的基本操作，包括与UCS2字符指针的相互操作
|
||
*
|
||
*****************************************************************/
|
||
|
||
#include <string.h>
|
||
#include <wchar.h>
|
||
#include <stdlib.h>
|
||
|
||
namespace wylib
|
||
{
|
||
namespace string
|
||
{
|
||
class CAnsiString;
|
||
|
||
class CWideString
|
||
{
|
||
public:
|
||
typedef wchar_t CharType;
|
||
typedef wchar_t* RawStrType;
|
||
|
||
private:
|
||
wchar_t* m_sStr;
|
||
|
||
private:
|
||
static const wchar_t* EMPTY_STR;
|
||
|
||
protected:
|
||
/*
|
||
realloc 内存申请、释放函数，如果需要关心内存的申请和释放，子类可以继承此函数
|
||
ptr: 申请、释放的指针
|
||
newsize: 申请内存字节数，若为0则表示释放ptr指针
|
||
若ptr和newsize都不为0则表示对ptr内存块重新申请大小
|
||
*/
|
||
virtual void* alloc(void* ptr, size_t newsize)
|
||
{
|
||
if ( newsize == 0 )
|
||
{
|
||
if ( ptr ) free( ptr );
|
||
return NULL;
|
||
}
|
||
return realloc( ptr, newsize );
|
||
}
|
||
/*
|
||
setData 设定字符串数据
|
||
data: 指向原始字符串的指针
|
||
*/
|
||
inline void setData(const wchar_t* data)
|
||
{
|
||
if ( data )
|
||
{
|
||
size_t size = wcslen(data) + 1;
|
||
size *= sizeof(*data);
|
||
m_sStr = (wchar_t*)alloc( m_sStr, size );
|
||
memcpy( m_sStr, data, size );
|
||
}
|
||
else m_sStr = (wchar_t*)alloc( m_sStr, 0 );
|
||
}
|
||
/*
|
||
catData 与现有字符串内容连接
|
||
data: 指向原始字符串的指针，字符串内容会被连接到当前字符串之后
|
||
*/
|
||
inline void catData(const wchar_t* data)
|
||
{
|
||
if ( data )
|
||
{
|
||
size_t cat_size = wcslen(data);
|
||
size_t cur_size = length();
|
||
m_sStr = (wchar_t*)alloc( m_sStr, (cat_size + cur_size + 1) * sizeof(*data) );
|
||
memcpy( &m_sStr[cur_size], data, (cat_size + 1) * sizeof(*data) );
|
||
}
|
||
}
|
||
public:
|
||
/*
|
||
?构造和析构函数
|
||
*/
|
||
CWideString()
|
||
{
|
||
m_sStr = NULL;
|
||
}
|
||
CWideString(const wchar_t* data)
|
||
{
|
||
m_sStr = NULL;
|
||
setData( data );
|
||
}
|
||
CWideString(const CWideString &str)
|
||
{
|
||
m_sStr = NULL;
|
||
setData( str.m_sStr );
|
||
}
|
||
virtual ~CWideString()
|
||
{
|
||
setData( NULL );
|
||
}
|
||
|
||
public:
|
||
/* 类型转换函数，转换为(const wchar_t*)。
|
||
注意： 如果当前字符串为NULL，则会返回""而不是NULL。
|
||
*/
|
||
inline operator const wchar_t* () const
|
||
{
|
||
return m_sStr ? m_sStr : EMPTY_STR;
|
||
}
|
||
/* 类型转换函数，转换为(const wchar_t*)。
|
||
注意： 如果当前字符串为NULL，则会返回""而不是NULL。
|
||
*/
|
||
inline operator wchar_t* ()
|
||
{
|
||
return m_sStr ? m_sStr : NULL;
|
||
}
|
||
/* 赋值函数，赋予(const wchar_t*)类型的值 */
|
||
inline void operator = (const wchar_t* data)
|
||
{
|
||
if ( data != m_sStr )
|
||
{
|
||
setData( data );
|
||
}
|
||
}
|
||
/* 赋值函数，赋予相同类型的值 */
|
||
inline void operator = (const CWideString& str)
|
||
{
|
||
if ( &str != this )
|
||
{
|
||
setData( str.m_sStr );
|
||
}
|
||
}
|
||
/* 与(const wchar_t*)类型的字符串连接 */
|
||
inline void operator += (const wchar_t* data)
|
||
{
|
||
if ( data != m_sStr )
|
||
{
|
||
catData( data );
|
||
}
|
||
else
|
||
{
|
||
CWideString wstr(m_sStr);
|
||
catData(wstr.m_sStr);
|
||
}
|
||
}
|
||
/* 与自己或其他字符串连接 */
|
||
inline void operator += (const CWideString& str)
|
||
{
|
||
if ( &str != this )
|
||
{
|
||
catData( str.m_sStr );
|
||
}
|
||
else
|
||
{
|
||
CWideString wstr(m_sStr);
|
||
catData(wstr.m_sStr);
|
||
}
|
||
}
|
||
/* 与(const wchar_t*)类型进行加法运算并返回连接后的CWideString类 */
|
||
inline const CWideString operator + (const wchar_t* data)
|
||
{
|
||
CWideString str( m_sStr );
|
||
str.catData( data );
|
||
return str;
|
||
}
|
||
/* 与另一个字符串进行加法运算并返回连接后的CWideString类 */
|
||
inline const CWideString operator + (const CWideString& str)
|
||
{
|
||
CWideString _str( m_sStr );
|
||
_str.catData( str.m_sStr );
|
||
return _str;
|
||
}
|
||
/* 判断与(const wchar_t*)类型的字符串指针中的字符串数据是否相同（区分大小小写） */
|
||
inline bool operator == (const wchar_t* data) const
|
||
{
|
||
return compare( data ) == 0;
|
||
}
|
||
/* 判断与另一个字符串的数据是否相同（区分大小小写） */
|
||
inline bool operator == (const CWideString& str) const
|
||
{
|
||
return compare( str.m_sStr ) == 0;
|
||
}
|
||
/* 判断与(const wchar_t*)类型的字符串指针中的字符串数据是否不相同（区分大小小写） */
|
||
inline bool operator != (const wchar_t* data) const
|
||
{
|
||
return compare( data ) != 0;
|
||
}
|
||
/* 判断与另一个字符串的数据是否不相同（区分大小小写） */
|
||
inline bool operator != (const CWideString& str) const
|
||
{
|
||
return compare( str.m_sStr ) != 0;
|
||
}
|
||
|
||
public:
|
||
/* 取得字符串指针 */
|
||
inline const wchar_t* rawStr() const
|
||
{
|
||
return m_sStr;
|
||
}
|
||
/* 取得字符串指针 */
|
||
inline wchar_t* rawStr()
|
||
{
|
||
return m_sStr;
|
||
}
|
||
/* 取得字符串长度 */
|
||
inline size_t length() const
|
||
{
|
||
return m_sStr ? wcslen(m_sStr) : 0;
|
||
}
|
||
/* 设置字符串长度(保留空间) */
|
||
inline void setLength(const size_t len)
|
||
{
|
||
if ( len > 0 )
|
||
{
|
||
m_sStr = (wchar_t*)alloc( m_sStr, sizeof(m_sStr[0]) * (len + 1) );
|
||
if ( m_sStr ) m_sStr[len] = 0;
|
||
}
|
||
else m_sStr = (wchar_t*)alloc( m_sStr, 0 );
|
||
}
|
||
/* 与另一个字符串指针对比 */
|
||
inline int compare(const wchar_t* data) const
|
||
{
|
||
if ( !m_sStr || !m_sStr[0] )
|
||
{
|
||
if ( !data || !data[0] )
|
||
return 0;
|
||
else return 1;
|
||
}
|
||
return wcscmp( m_sStr, data );
|
||
}
|
||
/* 与另一个字符串类进行对比 */
|
||
inline int compare(const CWideString& str) const
|
||
{
|
||
if ( !m_sStr || !m_sStr[0] )
|
||
{
|
||
if ( !str.m_sStr || !str.m_sStr[0] )
|
||
return 0;
|
||
else return 1;
|
||
}
|
||
return wcscmp( m_sStr, str.m_sStr );
|
||
}
|
||
/* 判断是否为空字符串 */
|
||
inline bool isEmpty() const
|
||
{
|
||
return !m_sStr || !m_sStr[0];
|
||
}
|
||
/* 转换字符串编码为本地多字节(local multibytes)格式。
|
||
如果sOutput参数不为空且编码转换成功则函数将转换结果赋值给sOutput参数并返回sOutput，
|
||
若转换失败则返回null且sOutput的内容可能仍然会改变；
|
||
如果sOutput参数为空且函数转换编码成功则返回新的CAnsiString对象，此时调用者需要再合适的时候delete返回值，
|
||
转换失败则函数返回NULL
|
||
*/
|
||
CAnsiString* toAStr(CAnsiString* sOutput = NULL) const;
|
||
/* 格式化字符串 */
|
||
size_t format(const wchar_t* fmt, ...);
|
||
size_t format_args(const wchar_t* fmt, va_list _Args);
|
||
};
|
||
|
||
};
|
||
};
|
||
|
||
#endif
|
||
|