windows-nt/Source/XPSP1/NT/shell/themes/inc/simpstr.h

//---------------------------------------------------------------------------
//   SimpStr.h - defines simple string classes
//---------------------------------------------------------------------------
#ifndef _SIMPSTR_H_
#define _SIMPSTR_H_
//---------------------------------------------------------------------------
class CWideString           // simplified version of CString
{
public:
    //--------------------------------------------------------------
    CWideString(LPCWSTR pszString=NULL)
    {
        _pWideString = NULL;

        if (pszString)
        {
            int len = lstrlen(pszString);
            _pWideString = new WCHAR[1+len];

            if (_pWideString)
            {
                lstrcpy(_pWideString, pszString);
            }
        }
    }

    //--------------------------------------------------------------
    CWideString(CWideString &wsCopy)
    {
        _pWideString = NULL;

        if (wsCopy._pWideString)
        {
            int len = lstrlen(wsCopy._pWideString);
            _pWideString = new WCHAR[1+len];

            if (_pWideString)
            {
                lstrcpy(_pWideString, wsCopy._pWideString);
            }
        }
    }

    //--------------------------------------------------------------
    ~CWideString()
    {
        if (_pWideString)
            delete [] _pWideString;
    }

    //--------------------------------------------------------------
    CWideString & operator = (CWideString &wsNew)
    {
        //---- delete our old string ----
        if (_pWideString)
        {
            delete [] _pWideString;
            _pWideString = NULL;
        }

        //---- copy new string, if not NULL ----
        if (wsNew._pWideString)
        {
            int len = lstrlen(wsNew._pWideString);
            _pWideString = new WCHAR[1+len];

            if (_pWideString)
            {
                lstrcpy(_pWideString, wsNew._pWideString);
            }
        }

        return *this;
    }

    //--------------------------------------------------------------
    CWideString & operator = (LPCWSTR pNewString)
    {
        //---- delete our old string ----
        if (_pWideString)
        {
            delete [] _pWideString;
            _pWideString = NULL;
        }

        //---- copy new string, if not NULL ----
        if (pNewString)
        {
            int len = lstrlen(pNewString);
            _pWideString = new WCHAR[1+len];

            if (_pWideString)
            {
                lstrcpy(_pWideString, pNewString);
            }
        }

        return *this;
    }
    //--------------------------------------------------------------
    operator LPCWSTR() const
    {
        return _pWideString;
    }

    //--------------------------------------------------------------
    int GetLength()
    {
        int iLen = 0;

        if (_pWideString)
        {
            iLen = lstrlen(_pWideString);
        }
        
        return iLen;
    }
    //--------------------------------------------------------------

protected:
    LPWSTR _pWideString;
};
//---------------------------------------------------------------------------
template <class T>
class CSimpleArray
{
public:
	T* m_aT;
	int m_nSize;
	int m_nAllocSize;

// Construction/destruction
	CSimpleArray() : m_aT(NULL), m_nSize(0), m_nAllocSize(0)
	{ }

	~CSimpleArray()
	{
		RemoveAll();
	}

// Operations
	int GetSize() const
	{
		return m_nSize;
	}
	BOOL Add(T& t)
	{
		if(m_nSize == m_nAllocSize)
		{
			T* aT;
			int nNewAllocSize = (m_nAllocSize == 0) ? 1 : (m_nSize * 2);
			aT = (T*)realloc(m_aT, nNewAllocSize * sizeof(T));
			if(aT == NULL)
				return FALSE;
			m_nAllocSize = nNewAllocSize;
			m_aT = aT;
		}
		m_nSize++;
		SetAtIndex(m_nSize - 1, t);
		return TRUE;
	}
	BOOL Remove(T& t)
	{
		int nIndex = Find(t);
		if(nIndex == -1)
			return FALSE;
		return RemoveAt(nIndex);
	}
	BOOL RemoveAt(int nIndex)
	{
        //---- always call the dtr ----
#if _MSC_VER >= 1200
		m_aT[nIndex].~T();
#else
        T* MyT;
        MyT = &m_aT[nIndex];
        MyT->~T();
#endif

        if(nIndex != (m_nSize - 1))
		{
			memmove((void*)&m_aT[nIndex], (void*)&m_aT[nIndex + 1], (m_nSize - (nIndex + 1)) * sizeof(T));
		}

		m_nSize--;
		return TRUE;
	}
	void RemoveAll()
	{
		if(m_aT != NULL)
		{
			for(int i = 0; i < m_nSize; i++) {
#if _MSC_VER >= 1200
				m_aT[i].~T();
#else
                T* MyT;
                MyT = &m_aT[i];
                MyT->~T();
#endif
            }
			free(m_aT);
			m_aT = NULL;
		}
		m_nSize = 0;
		m_nAllocSize = 0;
	}
	T& operator[] (int nIndex) const
	{
		ATLASSERT(nIndex >= 0 && nIndex < m_nSize);
		return m_aT[nIndex];
	}
	T* GetData() const
	{
		return m_aT;
	}

// Implementation
	class Wrapper
	{
	public:
		Wrapper(T& _t) : t(_t)
		{
		}
		template <class _Ty>
		void *operator new(size_t, _Ty* p)
		{
			return p;
		}
		T t;
	};
	void SetAtIndex(int nIndex, T& t)
	{
		ATLASSERT(nIndex >= 0 && nIndex < m_nSize);
		new(m_aT + nIndex) Wrapper(t);
	}
	int Find(T& t) const
	{
		for(int i = 0; i < m_nSize; i++)
		{
			if(m_aT[i] == t)
				return i;
		}
		return -1;  // not found
	}
};
#endif      // _SIMPSTR_H_
//---------------------------------------------------------------------------