123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944 |
- #include "stdafx.h"
- #include "Utils.h"
- namespace DuiLib
- {
- /////////////////////////////////////////////////////////////////////////////////////
- //
- //
- CDuiPoint::CDuiPoint()
- {
- x = y = 0;
- }
- CDuiPoint::CDuiPoint(const POINT& src)
- {
- x = src.x;
- y = src.y;
- }
- CDuiPoint::CDuiPoint(int _x, int _y)
- {
- x = _x;
- y = _y;
- }
- CDuiPoint::CDuiPoint(LPARAM lParam)
- {
- x = GET_X_LPARAM(lParam);
- y = GET_Y_LPARAM(lParam);
- }
- /////////////////////////////////////////////////////////////////////////////////////
- //
- //
- CDuiSize::CDuiSize()
- {
- cx = cy = 0;
- }
- CDuiSize::CDuiSize(const SIZE& src)
- {
- cx = src.cx;
- cy = src.cy;
- }
- CDuiSize::CDuiSize(const RECT rc)
- {
- cx = rc.right - rc.left;
- cy = rc.bottom - rc.top;
- }
- CDuiSize::CDuiSize(int _cx, int _cy)
- {
- cx = _cx;
- cy = _cy;
- }
- /////////////////////////////////////////////////////////////////////////////////////
- //
- //
- CDuiRect::CDuiRect()
- {
- left = top = right = bottom = 0;
- }
- CDuiRect::CDuiRect(const RECT& src)
- {
- left = src.left;
- top = src.top;
- right = src.right;
- bottom = src.bottom;
- }
- CDuiRect::CDuiRect(int iLeft, int iTop, int iRight, int iBottom)
- {
- left = iLeft;
- top = iTop;
- right = iRight;
- bottom = iBottom;
- }
- int CDuiRect::GetWidth() const
- {
- return right - left;
- }
- int CDuiRect::GetHeight() const
- {
- return bottom - top;
- }
- void CDuiRect::Empty()
- {
- left = top = right = bottom = 0;
- }
- bool CDuiRect::IsNull() const
- {
- return (left == 0 && right == 0 && top == 0 && bottom == 0);
- }
- void CDuiRect::Join(const RECT& rc)
- {
- if( rc.left < left ) left = rc.left;
- if( rc.top < top ) top = rc.top;
- if( rc.right > right ) right = rc.right;
- if( rc.bottom > bottom ) bottom = rc.bottom;
- }
- void CDuiRect::ResetOffset()
- {
- ::OffsetRect(this, -left, -top);
- }
- void CDuiRect::Normalize()
- {
- if( left > right ) { int iTemp = left; left = right; right = iTemp; }
- if( top > bottom ) { int iTemp = top; top = bottom; bottom = iTemp; }
- }
- void CDuiRect::Offset(int cx, int cy)
- {
- ::OffsetRect(this, cx, cy);
- }
- void CDuiRect::Inflate(int cx, int cy)
- {
- ::InflateRect(this, cx, cy);
- }
- void CDuiRect::Deflate(int cx, int cy)
- {
- ::InflateRect(this, -cx, -cy);
- }
- void CDuiRect::Union(CDuiRect& rc)
- {
- ::UnionRect(this, this, &rc);
- }
- /////////////////////////////////////////////////////////////////////////////////////
- //
- //
- CStdPtrArray::CStdPtrArray(int iPreallocSize) : m_ppVoid(NULL), m_nCount(0), m_nAllocated(iPreallocSize)
- {
- ASSERT(iPreallocSize>=0);
- if( iPreallocSize > 0 ) m_ppVoid = static_cast<LPVOID*>(malloc(iPreallocSize * sizeof(LPVOID)));
- }
- CStdPtrArray::CStdPtrArray(const CStdPtrArray& src) : m_ppVoid(NULL), m_nCount(0), m_nAllocated(0)
- {
- for(int i=0; i<src.GetSize(); i++)
- Add(src.GetAt(i));
- }
- CStdPtrArray::~CStdPtrArray()
- {
- if( m_ppVoid != NULL ) free(m_ppVoid);
- }
- void CStdPtrArray::Empty()
- {
- if( m_ppVoid != NULL ) free(m_ppVoid);
- m_ppVoid = NULL;
- m_nCount = m_nAllocated = 0;
- }
- void CStdPtrArray::Resize(int iSize)
- {
- Empty();
- m_ppVoid = static_cast<LPVOID*>(malloc(iSize * sizeof(LPVOID)));
- ::ZeroMemory(m_ppVoid, iSize * sizeof(LPVOID));
- m_nAllocated = iSize;
- m_nCount = iSize;
- }
- bool CStdPtrArray::IsEmpty() const
- {
- return m_nCount == 0;
- }
- bool CStdPtrArray::Add(LPVOID pData)
- {
- if( ++m_nCount >= m_nAllocated) {
- int nAllocated = m_nAllocated * 2;
- if( nAllocated == 0 ) nAllocated = 11;
- LPVOID* ppVoid = static_cast<LPVOID*>(realloc(m_ppVoid, nAllocated * sizeof(LPVOID)));
- if( ppVoid != NULL ) {
- m_nAllocated = nAllocated;
- m_ppVoid = ppVoid;
- }
- else {
- --m_nCount;
- return false;
- }
- }
- m_ppVoid[m_nCount - 1] = pData;
- return true;
- }
- bool CStdPtrArray::InsertAt(int iIndex, LPVOID pData)
- {
- if( iIndex == m_nCount ) return Add(pData);
- if( iIndex < 0 || iIndex > m_nCount ) return false;
- if( ++m_nCount >= m_nAllocated) {
- int nAllocated = m_nAllocated * 2;
- if( nAllocated == 0 ) nAllocated = 11;
- LPVOID* ppVoid = static_cast<LPVOID*>(realloc(m_ppVoid, nAllocated * sizeof(LPVOID)));
- if( ppVoid != NULL ) {
- m_nAllocated = nAllocated;
- m_ppVoid = ppVoid;
- }
- else {
- --m_nCount;
- return false;
- }
- }
- memmove(&m_ppVoid[iIndex + 1], &m_ppVoid[iIndex], (m_nCount - iIndex - 1) * sizeof(LPVOID));
- m_ppVoid[iIndex] = pData;
- return true;
- }
- bool CStdPtrArray::SetAt(int iIndex, LPVOID pData)
- {
- if( iIndex < 0 || iIndex >= m_nCount ) return false;
- m_ppVoid[iIndex] = pData;
- return true;
- }
- bool CStdPtrArray::Remove(int iIndex)
- {
- if( iIndex < 0 || iIndex >= m_nCount ) return false;
- if( iIndex < --m_nCount ) ::CopyMemory(m_ppVoid + iIndex, m_ppVoid + iIndex + 1, (m_nCount - iIndex) * sizeof(LPVOID));
- return true;
- }
- int CStdPtrArray::Find(LPVOID pData) const
- {
- for( int i = 0; i < m_nCount; i++ ) if( m_ppVoid[i] == pData ) return i;
- return -1;
- }
- int CStdPtrArray::GetSize() const
- {
- return m_nCount;
- }
- LPVOID* CStdPtrArray::GetData()
- {
- return m_ppVoid;
- }
- LPVOID CStdPtrArray::GetAt(int iIndex) const
- {
- if( iIndex < 0 || iIndex >= m_nCount ) return NULL;
- return m_ppVoid[iIndex];
- }
- LPVOID CStdPtrArray::operator[] (int iIndex) const
- {
- ASSERT(iIndex>=0 && iIndex<m_nCount);
- return m_ppVoid[iIndex];
- }
- /////////////////////////////////////////////////////////////////////////////////////
- //
- //
- CStdValArray::CStdValArray(int iElementSize, int iPreallocSize /*= 0*/) :
- m_pVoid(NULL),
- m_nCount(0),
- m_iElementSize(iElementSize),
- m_nAllocated(iPreallocSize)
- {
- ASSERT(iElementSize>0);
- ASSERT(iPreallocSize>=0);
- if( iPreallocSize > 0 ) m_pVoid = static_cast<LPBYTE>(malloc(iPreallocSize * m_iElementSize));
- }
- CStdValArray::~CStdValArray()
- {
- if( m_pVoid != NULL ) free(m_pVoid);
- }
- void CStdValArray::Empty()
- {
- m_nCount = 0; // NOTE: We keep the memory in place
- }
- bool CStdValArray::IsEmpty() const
- {
- return m_nCount == 0;
- }
- bool CStdValArray::Add(LPCVOID pData)
- {
- if( ++m_nCount >= m_nAllocated) {
- int nAllocated = m_nAllocated * 2;
- if( nAllocated == 0 ) nAllocated = 11;
- LPBYTE pVoid = static_cast<LPBYTE>(realloc(m_pVoid, nAllocated * m_iElementSize));
- if( pVoid != NULL ) {
- m_nAllocated = nAllocated;
- m_pVoid = pVoid;
- }
- else {
- --m_nCount;
- return false;
- }
- }
- ::CopyMemory(m_pVoid + ((m_nCount - 1) * m_iElementSize), pData, m_iElementSize);
- return true;
- }
- bool CStdValArray::Remove(int iIndex)
- {
- if( iIndex < 0 || iIndex >= m_nCount ) return false;
- if( iIndex < --m_nCount ) ::CopyMemory(m_pVoid + (iIndex * m_iElementSize), m_pVoid + ((iIndex + 1) * m_iElementSize), (m_nCount - iIndex) * m_iElementSize);
- return true;
- }
- int CStdValArray::GetSize() const
- {
- return m_nCount;
- }
- LPVOID CStdValArray::GetData()
- {
- return static_cast<LPVOID>(m_pVoid);
- }
- LPVOID CStdValArray::GetAt(int iIndex) const
- {
- if( iIndex < 0 || iIndex >= m_nCount ) return NULL;
- return m_pVoid + (iIndex * m_iElementSize);
- }
- LPVOID CStdValArray::operator[] (int iIndex) const
- {
- ASSERT(iIndex>=0 && iIndex<m_nCount);
- return m_pVoid + (iIndex * m_iElementSize);
- }
- /////////////////////////////////////////////////////////////////////////////////////
- //
- //
- CDuiString::CDuiString() : m_pstr(m_szBuffer)
- {
- m_szBuffer[0] = _T('\0');
- }
- CDuiString::CDuiString(const TCHAR ch) : m_pstr(m_szBuffer)
- {
- m_szBuffer[0] = ch;
- m_szBuffer[1] = _T('\0');
- }
- CDuiString::CDuiString(LPCTSTR lpsz, int nLen) : m_pstr(m_szBuffer)
- {
- ASSERT(!::IsBadStringPtr(lpsz,-1) || lpsz==NULL);
- m_szBuffer[0] = _T('\0');
- Assign(lpsz, nLen);
- }
- CDuiString::CDuiString(const CDuiString& src) : m_pstr(m_szBuffer)
- {
- m_szBuffer[0] = _T('\0');
- Assign(src.m_pstr);
- }
- CDuiString::~CDuiString()
- {
- if( m_pstr != m_szBuffer ) free(m_pstr);
- }
- int CDuiString::GetLength() const
- {
- return (int) _tcslen(m_pstr);
- }
- CDuiString::operator LPCTSTR() const
- {
- return m_pstr;
- }
- 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 ) {
- free(m_pstr);
- m_pstr = m_szBuffer;
- }
- _tcscat(m_szBuffer, pstr);
- }
- }
- 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_pstr = static_cast<LPTSTR>(realloc(m_pstr, (cchMax + 1) * sizeof(TCHAR)));
- }
- _tcsncpy(m_pstr, pstr, cchMax);
- m_pstr[cchMax] = _T('\0');
- }
- bool CDuiString::IsEmpty() const
- {
- return m_pstr[0] == _T('\0');
- }
- void CDuiString::Empty()
- {
- if( m_pstr != m_szBuffer ) free(m_pstr);
- m_pstr = m_szBuffer;
- m_szBuffer[0] = _T('\0');
- }
- LPCTSTR CDuiString::GetData() const
- {
- return m_pstr;
- }
- TCHAR CDuiString::GetAt(int nIndex) const
- {
- return m_pstr[nIndex];
- }
- TCHAR CDuiString::operator[] (int nIndex) const
- {
- return m_pstr[nIndex];
- }
- const CDuiString& CDuiString::operator=(const CDuiString& src)
- {
- Assign(src);
- return *this;
- }
- const CDuiString& CDuiString::operator=(LPCTSTR lpStr)
- {
- if ( lpStr )
- {
- ASSERT(!::IsBadStringPtr(lpStr,-1));
- Assign(lpStr);
- }
- else
- {
- Empty();
- }
- return *this;
- }
- #ifdef _UNICODE
- const CDuiString& CDuiString::operator=(LPCSTR lpStr)
- {
- if ( lpStr )
- {
- ASSERT(!::IsBadStringPtrA(lpStr,-1));
- int cchStr = (int) strlen(lpStr) + 1;
- LPWSTR pwstr = (LPWSTR) _alloca(cchStr);
- if( pwstr != NULL ) ::MultiByteToWideChar(::GetACP(), 0, lpStr, -1, pwstr, cchStr) ;
- Assign(pwstr);
- }
- else
- {
- Empty();
- }
- return *this;
- }
- const CDuiString& CDuiString::operator+=(LPCSTR lpStr)
- {
- if ( lpStr )
- {
- ASSERT(!::IsBadStringPtrA(lpStr,-1));
- int cchStr = (int) strlen(lpStr) + 1;
- LPWSTR pwstr = (LPWSTR) _alloca(cchStr);
- if( pwstr != NULL ) ::MultiByteToWideChar(::GetACP(), 0, lpStr, -1, pwstr, cchStr) ;
- Append(pwstr);
- }
-
- return *this;
- }
- #else
- const CDuiString& CDuiString::operator=(LPCWSTR lpwStr)
- {
- if ( lpwStr )
- {
- ASSERT(!::IsBadStringPtrW(lpwStr,-1));
- int cchStr = ((int) wcslen(lpwStr) * 2) + 1;
- LPSTR pstr = (LPSTR) _alloca(cchStr);
- if( pstr != NULL ) ::WideCharToMultiByte(::GetACP(), 0, lpwStr, -1, pstr, cchStr, NULL, NULL);
- Assign(pstr);
- }
- else
- {
- Empty();
- }
-
- return *this;
- }
- const CDuiString& CDuiString::operator+=(LPCWSTR lpwStr)
- {
- if ( lpwStr )
- {
- ASSERT(!::IsBadStringPtrW(lpwStr,-1));
- int cchStr = ((int) wcslen(lpwStr) * 2) + 1;
- LPSTR pstr = (LPSTR) _alloca(cchStr);
- if( pstr != NULL ) ::WideCharToMultiByte(::GetACP(), 0, lpwStr, -1, pstr, cchStr, NULL, NULL);
- Append(pstr);
- }
-
- return *this;
- }
- #endif // _UNICODE
- const CDuiString& CDuiString::operator=(const TCHAR ch)
- {
- Empty();
- m_szBuffer[0] = ch;
- m_szBuffer[1] = _T('\0');
- return *this;
- }
- CDuiString CDuiString::operator+(const CDuiString& src) const
- {
- CDuiString sTemp = *this;
- sTemp.Append(src);
- return sTemp;
- }
- CDuiString CDuiString::operator+(LPCTSTR lpStr) const
- {
- if ( lpStr )
- {
- ASSERT(!::IsBadStringPtr(lpStr,-1));
- CDuiString sTemp = *this;
- sTemp.Append(lpStr);
- return sTemp;
- }
- return *this;
- }
- const CDuiString& CDuiString::operator+=(const CDuiString& src)
- {
- Append(src);
- return *this;
- }
- const CDuiString& CDuiString::operator+=(LPCTSTR lpStr)
- {
- if ( lpStr )
- {
- ASSERT(!::IsBadStringPtr(lpStr,-1));
- Append(lpStr);
- }
-
- return *this;
- }
- const CDuiString& CDuiString::operator+=(const TCHAR ch)
- {
- TCHAR str[] = { ch, _T('\0') };
- Append(str);
- return *this;
- }
- bool CDuiString::operator == (LPCTSTR str) const { return (Compare(str) == 0); };
- bool CDuiString::operator != (LPCTSTR str) const { return (Compare(str) != 0); };
- bool CDuiString::operator <= (LPCTSTR str) const { return (Compare(str) <= 0); };
- bool CDuiString::operator < (LPCTSTR str) const { return (Compare(str) < 0); };
- bool CDuiString::operator >= (LPCTSTR str) const { return (Compare(str) >= 0); };
- bool CDuiString::operator > (LPCTSTR str) const { return (Compare(str) > 0); };
- void CDuiString::SetAt(int nIndex, TCHAR ch)
- {
- ASSERT(nIndex>=0 && nIndex<GetLength());
- m_pstr[nIndex] = ch;
- }
- int CDuiString::Compare(LPCTSTR lpsz) const
- {
- return _tcscmp(m_pstr, lpsz);
- }
- int CDuiString::CompareNoCase(LPCTSTR lpsz) const
- {
- return _tcsicmp(m_pstr, lpsz);
- }
- void CDuiString::MakeUpper()
- {
- _tcsupr(m_pstr);
- }
- void CDuiString::MakeLower()
- {
- _tcslwr(m_pstr);
- }
- CDuiString CDuiString::Left(int iLength) const
- {
- if( iLength < 0 ) iLength = 0;
- if( iLength > GetLength() ) iLength = GetLength();
- return CDuiString(m_pstr, iLength);
- }
- CDuiString CDuiString::Mid(int iPos, int iLength) const
- {
- if( iLength < 0 ) iLength = GetLength() - iPos;
- if( iPos + iLength > GetLength() ) iLength = GetLength() - iPos;
- if( iLength <= 0 ) return CDuiString();
- return CDuiString(m_pstr + iPos, iLength);
- }
- CDuiString CDuiString::Right(int iLength) const
- {
- int iPos = GetLength() - iLength;
- if( iPos < 0 ) {
- iPos = 0;
- iLength = GetLength();
- }
- return CDuiString(m_pstr + iPos, iLength);
- }
- int CDuiString::Find(TCHAR ch, int iPos /*= 0*/) const
- {
- ASSERT(iPos>=0 && iPos<=GetLength());
- if( iPos != 0 && (iPos < 0 || iPos >= GetLength()) ) return -1;
- LPCTSTR p = _tcschr(m_pstr + iPos, ch);
- if( p == NULL ) return -1;
- return (int)(p - m_pstr);
- }
- int CDuiString::Find(LPCTSTR pstrSub, int iPos /*= 0*/) const
- {
- ASSERT(!::IsBadStringPtr(pstrSub,-1));
- ASSERT(iPos>=0 && iPos<=GetLength());
- if( iPos != 0 && (iPos < 0 || iPos > GetLength()) ) return -1;
- LPCTSTR p = _tcsstr(m_pstr + iPos, pstrSub);
- if( p == NULL ) return -1;
- return (int)(p - m_pstr);
- }
- int CDuiString::ReverseFind(TCHAR ch) const
- {
- LPCTSTR p = _tcsrchr(m_pstr, ch);
- if( p == NULL ) return -1;
- return (int)(p - m_pstr);
- }
- 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;
- }
- 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;
- }
- int CDuiString::SmallFormat(LPCTSTR pstrFormat, ...)
- {
- CDuiString sFormat = pstrFormat;
- TCHAR szBuffer[64] = { 0 };
- va_list argList;
- va_start(argList, pstrFormat);
- int iRet = ::wvsprintf(szBuffer, sFormat, argList);
- va_end(argList);
- Assign(szBuffer);
- return iRet;
- }
- /////////////////////////////////////////////////////////////////////////////
- //
- //
- static UINT HashKey(LPCTSTR Key)
- {
- UINT i = 0;
- SIZE_T len = _tcslen(Key);
- while( len-- > 0 ) i = (i << 5) + i + Key[len];
- return i;
- }
- static UINT HashKey(const CDuiString& Key)
- {
- return HashKey((LPCTSTR)Key);
- };
- CStdStringPtrMap::CStdStringPtrMap(int nSize) : m_nCount(0)
- {
- if( nSize < 16 ) nSize = 16;
- m_nBuckets = nSize;
- m_aT = new TITEM*[nSize];
- memset(m_aT, 0, nSize * sizeof(TITEM*));
- }
- CStdStringPtrMap::~CStdStringPtrMap()
- {
- if( m_aT ) {
- int len = m_nBuckets;
- while( len-- ) {
- TITEM* pItem = m_aT[len];
- while( pItem ) {
- TITEM* pKill = pItem;
- pItem = pItem->pNext;
- delete pKill;
- }
- }
- delete [] m_aT;
- m_aT = NULL;
- }
- }
- void CStdStringPtrMap::RemoveAll()
- {
- this->Resize(m_nBuckets);
- }
- void CStdStringPtrMap::Resize(int nSize)
- {
- if( m_aT ) {
- int len = m_nBuckets;
- while( len-- ) {
- TITEM* pItem = m_aT[len];
- while( pItem ) {
- TITEM* pKill = pItem;
- pItem = pItem->pNext;
- delete pKill;
- }
- }
- delete [] m_aT;
- m_aT = NULL;
- }
- if( nSize < 0 ) nSize = 0;
- if( nSize > 0 ) {
- m_aT = new TITEM*[nSize];
- memset(m_aT, 0, nSize * sizeof(TITEM*));
- }
- m_nBuckets = nSize;
- m_nCount = 0;
- }
- LPVOID CStdStringPtrMap::Find(LPCTSTR key, bool optimize) const
- {
- if( m_nBuckets == 0 || GetSize() == 0 ) return NULL;
- UINT slot = HashKey(key) % m_nBuckets;
- for( TITEM* pItem = m_aT[slot]; pItem; pItem = pItem->pNext ) {
- if( pItem->Key == key ) {
- if (optimize && pItem != m_aT[slot]) {
- if (pItem->pNext) {
- pItem->pNext->pPrev = pItem->pPrev;
- }
- pItem->pPrev->pNext = pItem->pNext;
- pItem->pPrev = NULL;
- pItem->pNext = m_aT[slot];
- pItem->pNext->pPrev = pItem;
- //½«itemÒÆ¶¯ÖÁÁ´ÌõÍ·²¿
- m_aT[slot] = pItem;
- }
- return pItem->Data;
- }
- }
- return NULL;
- }
- bool CStdStringPtrMap::Insert(LPCTSTR key, LPVOID pData)
- {
- if( m_nBuckets == 0 ) return false;
- if( Find(key) ) return false;
- // Add first in bucket
- UINT slot = HashKey(key) % m_nBuckets;
- TITEM* pItem = new TITEM;
- pItem->Key = key;
- pItem->Data = pData;
- pItem->pPrev = NULL;
- pItem->pNext = m_aT[slot];
- if (pItem->pNext)
- pItem->pNext->pPrev = pItem;
- m_aT[slot] = pItem;
- m_nCount++;
- return true;
- }
- LPVOID CStdStringPtrMap::Set(LPCTSTR key, LPVOID pData)
- {
- if( m_nBuckets == 0 ) return pData;
- if (GetSize()>0) {
- UINT slot = HashKey(key) % m_nBuckets;
- // Modify existing item
- for( TITEM* pItem = m_aT[slot]; pItem; pItem = pItem->pNext ) {
- if( pItem->Key == key ) {
- LPVOID pOldData = pItem->Data;
- pItem->Data = pData;
- return pOldData;
- }
- }
- }
- Insert(key, pData);
- return NULL;
- }
- bool CStdStringPtrMap::Remove(LPCTSTR key)
- {
- if( m_nBuckets == 0 || GetSize() == 0 ) return false;
- UINT slot = HashKey(key) % m_nBuckets;
- TITEM** ppItem = &m_aT[slot];
- while( *ppItem ) {
- if( (*ppItem)->Key == key ) {
- TITEM* pKill = *ppItem;
- *ppItem = (*ppItem)->pNext;
- if (*ppItem)
- (*ppItem)->pPrev = pKill->pPrev;
- delete pKill;
- m_nCount--;
- return true;
- }
- ppItem = &((*ppItem)->pNext);
- }
- return false;
- }
- int CStdStringPtrMap::GetSize() const
- {
- #if 0//def _DEBUG
- int nCount = 0;
- int len = m_nBuckets;
- while( len-- ) {
- for( const TITEM* pItem = m_aT[len]; pItem; pItem = pItem->pNext ) nCount++;
- }
- ASSERT(m_nCount==nCount);
- #endif
- return m_nCount;
- }
- LPCTSTR CStdStringPtrMap::GetAt(int iIndex) const
- {
- if( m_nBuckets == 0 || GetSize() == 0 ) return false;
- int pos = 0;
- int len = m_nBuckets;
- while( len-- ) {
- for( TITEM* pItem = m_aT[len]; pItem; pItem = pItem->pNext ) {
- if( pos++ == iIndex ) {
- return pItem->Key.GetData();
- }
- }
- }
- return NULL;
- }
- LPCTSTR CStdStringPtrMap::operator[] (int nIndex) const
- {
- return GetAt(nIndex);
- }
- /////////////////////////////////////////////////////////////////////////////////////
- //
- //
- CWaitCursor::CWaitCursor()
- {
- m_hOrigCursor = ::SetCursor(::LoadCursor(NULL, IDC_WAIT));
- }
- CWaitCursor::~CWaitCursor()
- {
- ::SetCursor(m_hOrigCursor);
- }
- } // namespace DuiLib
|