CWHArray模版类

Array.h

#ifndef ARRAY_HEAD_FILE#define ARRAY_HEAD_FILE#pragma once#include <windows.h>#include <assert.h>#include <memory>#define DbgBreakPoint  _asm int 3////////////////////////////////////////////////////////////////////////////////////数组模板类template <class TYPE, class ARG_TYPE = const TYPE &> class CWHArray{//变量定义protected:TYPE *m_pData;//数组指针INT_PTRm_nMaxCount;//缓冲数目INT_PTRm_nGrowCount;//增长数目INT_PTRm_nElementCount;//元素数目//函数定义public://构造函数CWHArray();//析构函数virtual ~CWHArray();//信息函数public://是否空组bool IsEmpty() const;//获取数目INT_PTR GetCount() const;//功能函数public://获取缓冲TYPE * GetData();//获取缓冲const TYPE * GetData() const;//增加元素INT_PTR Add(ARG_TYPE newElement);//拷贝数组VOID Copy(const CWHArray & Src);//追加数组INT_PTR Append(const CWHArray & Src);//获取元素TYPE & GetAt(INT_PTR nIndex);//获取元素const TYPE & GetAt(INT_PTR nIndex) const;//获取元素TYPE & ElementAt(INT_PTR nIndex);//获取元素const TYPE & ElementAt(INT_PTR nIndex) const;//操作函数public://设置大小VOID SetSize(INT_PTR nNewSize);//设置元素VOID SetAt(INT_PTR nIndex, ARG_TYPE newElement);//设置元素VOID SetAtGrow(INT_PTR nIndex, ARG_TYPE newElement);//插入数据VOID InsertAt(INT_PTR nIndex, const CWHArray & Src);//插入数据VOID InsertAt(INT_PTR nIndex, ARG_TYPE newElement, INT_PTR nCount = 1);//删除数据VOID RemoveAt(INT_PTR nIndex, INT_PTR nCount = 1);//删除元素VOID RemoveAll();//操作重载public://操作重载TYPE & operator[](INT_PTR nIndex);//操作重载const TYPE & operator[](INT_PTR nIndex) const;//内存函数public://释放内存VOID FreeMemory();//申请内存VOID AllocMemory(INT_PTR nNewCount);};//////////////////////////////////////////////////////////////////////////////////// CWHArray<TYPE, ARG_TYPE> 内联函数//是否空组template<class TYPE, class ARG_TYPE>inline bool CWHArray<TYPE, ARG_TYPE>::IsEmpty() const{return (m_nElementCount == 0);}//获取数目template<class TYPE, class ARG_TYPE>inline INT_PTR CWHArray<TYPE, ARG_TYPE>::GetCount() const{return m_nElementCount;}//增加元素template<class TYPE, class ARG_TYPE>inline INT_PTR CWHArray<TYPE, ARG_TYPE>::Add(ARG_TYPE newElement){INT_PTR nIndex = m_nElementCount;SetAtGrow(nIndex, newElement);return nIndex;}//操作重载template<class TYPE, class ARG_TYPE>inline TYPE & CWHArray<TYPE, ARG_TYPE>::operator[](INT_PTR nIndex){return ElementAt(nIndex);}//操作重载template<class TYPE, class ARG_TYPE>inline const TYPE & CWHArray<TYPE, ARG_TYPE>::operator[](INT_PTR nIndex) const{return GetAt(nIndex);}//////////////////////////////////////////////////////////////////////////////////// CWHArray<TYPE, ARG_TYPE> 外联函数//构造函数template<class TYPE, class ARG_TYPE>CWHArray<TYPE, ARG_TYPE>::CWHArray(){m_pData = NULL;m_nMaxCount = 0;m_nGrowCount = 0;m_nElementCount = 0;return;}//构造函数template<class TYPE, class ARG_TYPE>CWHArray<TYPE, ARG_TYPE>::~CWHArray(){if (m_pData != NULL){for (INT_PTR i = 0; i < m_nElementCount; i++)(m_pData + i)->~TYPE();delete[](BYTE *)m_pData;m_pData = NULL;}return;}//获取缓冲template<class TYPE, class ARG_TYPE>TYPE * CWHArray<TYPE, ARG_TYPE>::GetData(){return m_pData;}//获取缓冲template<class TYPE, class ARG_TYPE>const TYPE * CWHArray<TYPE, ARG_TYPE>::GetData() const{return m_pData;}//拷贝数组template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::Copy(const CWHArray & Src){//效验参数assert(this != &Src);if (this == &Src) return;//拷贝数组AllocMemory(Src.m_nElementCount);if (m_nElementCount > 0){for (INT_PTR i = 0; i < m_nElementCount; i++) (m_pData + i)->~TYPE();memset(m_pData, 0, m_nElementCount*sizeof(TYPE));}for (INT_PTR i = 0; i < Src.m_nElementCount; i++)m_pData[i] = Src.m_pData[i];m_nElementCount = Src.m_nElementCount;return;}//追加数组template<class TYPE, class ARG_TYPE>INT_PTR CWHArray<TYPE, ARG_TYPE>::Append(const CWHArray & Src){//效验参数assert(this != &Src);if (this == &Src)return NULL;//拷贝数组if (Src.m_nElementCount > 0){INT_PTR nOldCount = m_nElementCount;AllocMemory(m_nElementCount + Src.m_nElementCount);for (INT_PTR i = 0; i < Src.m_nElementCount; i++)m_pData[m_nElementCount + i] = Src.m_pData[i];m_nElementCount += Src.m_nElementCount;}return m_nElementCount;}//获取元素template<class TYPE, class ARG_TYPE>TYPE & CWHArray<TYPE, ARG_TYPE>::GetAt(INT_PTR nIndex){assert((nIndex >= 0) && (nIndex < m_nElementCount));if ((nIndex < 0) || (nIndex >= m_nElementCount))DbgBreakPointreturn m_pData[nIndex];}//获取元素template<class TYPE, class ARG_TYPE>const TYPE & CWHArray<TYPE, ARG_TYPE>::GetAt(INT_PTR nIndex) const{assert((nIndex >= 0) && (nIndex < m_nElementCount));if ((nIndex < 0) || (nIndex >= m_nElementCount))return NULL;return m_pData[nIndex];}//获取元素template<class TYPE, class ARG_TYPE>TYPE & CWHArray<TYPE, ARG_TYPE>::ElementAt(INT_PTR nIndex){assert((nIndex >= 0) && (nIndex < m_nElementCount));if ((nIndex < 0) && (nIndex >= m_nElementCount))DbgBreakPointreturn m_pData[nIndex];}//获取元素template<class TYPE, class ARG_TYPE>const TYPE & CWHArray<TYPE, ARG_TYPE>::ElementAt(INT_PTR nIndex) const{assert((nIndex >= 0) && (nIndex < m_nElementCount));if ((nIndex < 0) && (nIndex >= m_nElementCount))return NULL;return m_pData[nIndex];}//设置大小template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::SetSize(INT_PTR nNewSize){//效验参数assert(nNewSize >= 0);if (nNewSize < 0)return;//设置大小AllocMemory(nNewSize);if (nNewSize > m_nElementCount){for (INT_PTR i = m_nElementCount; i < nNewSize; i++) new ((VOID *)(m_pData + i)) TYPE;}else if (nNewSize < m_nElementCount){for (INT_PTR i = nNewSize; i < m_nElementCount; i++) (m_pData + i)->~TYPE();memset(m_pData + nNewSize, 0, (m_nElementCount - nNewSize)*sizeof(TYPE));}m_nElementCount = nNewSize;return;}//设置元素template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::SetAt(INT_PTR nIndex, ARG_TYPE newElement){assert((nIndex >= 0) && (nIndex < m_nElementCount));if ((nIndex >= 0) && (nIndex < m_nElementCount)) m_pData[nIndex] = newElement;else return ;return;}//设置元素template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::SetAtGrow(INT_PTR nIndex, ARG_TYPE newElement){//效验参数assert(nIndex >= 0);if (nIndex < 0)return ;//设置元素if (nIndex >= m_nElementCount) SetSize(m_nElementCount + 1);m_pData[nIndex] = newElement;return;}//插入数据template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::InsertAt(INT_PTR nIndex, const CWHArray & Src){//效验参数assert(nStartIndex >= 0);if (nStartIndex < 0)return ;if (Src.m_nElementCount>0){//申请数组if (nIndex < m_nElementCount){INT_PTR nOldCount = m_nElementCount;SetSize(m_nElementCount + Src.m_nElementCount);for (INT_PTR i = 0; i < nCount; i++) (m_pData + nOldCount + i)->~TYPE();memmove(m_pData + nIndex + nCount, m_pData + nIndex, (nOldCount - nIndex)*sizeof(TYPE));memset(m_pData + nIndex, 0, Src.m_nElementCount*sizeof(TYPE));for (INT_PTR i = 0; i < Src.m_nElementCount; i++) new (m_pData + nIndex + i) TYPE();}else SetSize(nIndex + nCount);//拷贝数组assert((nIndex + Src.m_nElementCount) <= m_nElementCount);while (nCount--) m_pData[nIndex++] = newElement;}return;}//插入数据template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::InsertAt(INT_PTR nIndex, ARG_TYPE newElement, INT_PTR nCount){//效验参数assert(nIndex >= 0);assert(nCount > 0);if ((nIndex < 0) || (nCount <= 0))return ;//申请数组if (nIndex < m_nElementCount){INT_PTR nOldCount = m_nElementCount;SetSize(m_nElementCount + nCount);for (INT_PTR i = 0; i < nCount; i++) (m_pData + nOldCount + i)->~TYPE();memmove(m_pData + nIndex + nCount, m_pData + nIndex, (nOldCount - nIndex)*sizeof(TYPE));memset(m_pData + nIndex, 0, nCount*sizeof(TYPE));for (INT_PTR i = 0; i < nCount; i++)::new (m_pData + nIndex + i) TYPE();}else SetSize(nIndex + nCount);//拷贝数组assert((nIndex + nCount) <= m_nElementCount);while (nCount--) m_pData[nIndex++] = newElement;return;}//删除数据template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::RemoveAt(INT_PTR nIndex, INT_PTR nCount){//效验参数assert(nIndex >= 0);assert(nCount >= 0);assert(nIndex + nCount <= m_nElementCount);if ((nIndex < 0) || (nCount<0) || ((nIndex + nCount>m_nElementCount)))return ;//删除数据INT_PTR nMoveCount = m_nElementCount - (nIndex + nCount);for (INT_PTR i = 0; i < nCount; i++) (m_pData + nIndex + i)->~TYPE();if (nMoveCount > 0) memmove(m_pData + nIndex, m_pData + nIndex + nCount, nMoveCount*sizeof(TYPE));m_nElementCount -= nCount;return;}//删除元素template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::RemoveAll(){if (m_nElementCount > 0){for (INT_PTR i = 0; i < m_nElementCount; i++) (m_pData + i)->~TYPE();memset(m_pData, 0, m_nElementCount*sizeof(TYPE));m_nElementCount = 0;}return;}//释放内存template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::FreeMemory(){if (m_nElementCount != m_nMaxCount){TYPE * pNewData = NULL;if (m_nElementCount != 0){pNewData = (TYPE *) new BYTE[m_nElementCount*sizeof(TYPE)];memcpy(pNewData, m_pData, m_nElementCount*sizeof(TYPE));}delete[](BYTE *)m_pData;m_pData = pNewData;m_nMaxCount = m_nElementCount;}return;}//申请内存template<class TYPE, class ARG_TYPE>VOID CWHArray<TYPE, ARG_TYPE>::AllocMemory(INT_PTR nNewCount){//效验参数assert(nNewCount >= 0);if (nNewCount > m_nMaxCount){//计算数目INT_PTR nGrowCount = m_nGrowCount;if (nGrowCount == 0){nGrowCount = m_nElementCount / 8;nGrowCount = (nGrowCount < 4) ? 4 : ((nGrowCount>1024) ? 1024 : nGrowCount);}nNewCount += nGrowCount;//申请内存TYPE * pNewData = (TYPE *) new BYTE[nNewCount*sizeof(TYPE)];memcpy(pNewData, m_pData, m_nElementCount*sizeof(TYPE));memset(pNewData + m_nElementCount, 0, (nNewCount - m_nElementCount)*sizeof(TYPE));delete[](BYTE *)m_pData;//设置变量m_pData = pNewData;m_nMaxCount = nNewCount;}return;}//////////////////////////////////////////////////////////////////////////////////#endif

main.cpp

#include <stdio.h>#include "Array.h"int main(void){CWHArray<int> myArray;for (int i = 0; i < 10;i++){myArray.Add(i);printf("%d\n", myArray.GetAt(i));}myArray.IsEmpty();myArray.GetCount();myArray.GetData();myArray.GetAt(0);myArray.ElementAt(0);myArray.SetSize(20);myArray.SetAt(0, 456);myArray.InsertAt(5, 10);for (int i = 0; i < 20;i++){printf("%d\n", myArray.GetAt(i));}myArray.RemoveAt(0);myArray.RemoveAll();return 0;}/*012345678945612341056789000000000请按任意键继续. . .*/