ClassArr.h
上传用户:oy0519
上传日期:2008-01-20
资源大小:124k
文件大小:8k
源码类别:

绘图程序

开发平台:

Visual C++

  1. #if !defined(__ClassArr_H__)
  2. #define __ClassArr_H__
  3. #include <assert.h>
  4. #include <memory.h>
  5. #include <new.h>   //only supports Win32 and Mac
  6. template<class TYPE, class ARG_TYPE>
  7. class CClassArray
  8. {
  9. public:
  10. // Construction & destruction
  11. CClassArray() { m_pData = NULL; m_nSize = m_nMaxSize = m_nGrowBy = 0; }
  12. ~CClassArray();
  13. // Attributes
  14. int GetSize() const { return m_nSize; }
  15. int GetUpperBound() const { return m_nSize-1; }
  16. void SetSize(int nNewSize, int nGrowBy = -1);
  17. // Operations
  18. // Clean up
  19. void FreeExtra();
  20. void RemoveAll() { SetSize(0, -1); }
  21. // Accessing elements
  22. TYPE GetAt(int nIndex) const { assert(nIndex >= 0 && nIndex < m_nSize); return m_pData[nIndex]; }
  23. void SetAt(int nIndex, ARG_TYPE newElement) { assert(nIndex >= 0 && nIndex < m_nSize); m_pData[nIndex] = newElement; }
  24. TYPE& ElementAt(int nIndex) { assert(nIndex >= 0 && nIndex < m_nSize); return m_pData[nIndex]; }
  25. // Direct Access to the element data (may return NULL)
  26. const TYPE* GetData() const { return (const TYPE*)m_pData; }
  27. TYPE* GetData() { return (TYPE*)m_pData; }
  28. // Potentially growing the array
  29. void SetAtGrow(int nIndex, ARG_TYPE newElement);
  30. int Add(ARG_TYPE newElement) { int nIndex = m_nSize; SetAtGrow(nIndex, newElement); return nIndex; }
  31. int Append(const CClassArray& src);
  32. void Copy(const CClassArray& src);
  33. // overloaded operator helpers
  34. TYPE operator[](int nIndex) const { return GetAt(nIndex); }
  35. TYPE& operator[](int nIndex) { return ElementAt(nIndex); }
  36. // Operations that move elements around
  37. void InsertAt(int nIndex, ARG_TYPE newElement, int nCount = 1);
  38. void RemoveAt(int nIndex, int nCount = 1);
  39. void InsertAt(int nStartIndex, CClassArray* pNewArray);
  40. // Implementation
  41. protected:
  42. void ConstructElements(TYPE* pElements, int nCount);
  43. void DestructElements(TYPE* pElements, int nCount);
  44. void CopyElements(TYPE* pDest, const TYPE* pSrc, int nCount);
  45. TYPE* m_pData;   // the actual array of data
  46. int m_nSize;     // # of elements (upperBound - 1)
  47. int m_nMaxSize;  // max allocated
  48. int m_nGrowBy;   // grow amount
  49. };
  50. /////////////////////////////////////////////////////////////////////////////
  51. // CArray<TYPE, ARG_TYPE> out-of-line functions
  52. template<class TYPE, class ARG_TYPE>
  53. CClassArray<TYPE, ARG_TYPE>::~CClassArray()
  54. {
  55. if (m_pData != NULL)
  56. {
  57. DestructElements(m_pData, m_nSize);
  58. delete[] (char*)m_pData;
  59. }
  60. }
  61. template<class TYPE, class ARG_TYPE>
  62. void CClassArray<TYPE, ARG_TYPE>::SetSize(int nNewSize, int nGrowBy)
  63. {
  64. assert(nNewSize >= 0);
  65. if (nGrowBy != -1)
  66. m_nGrowBy = nGrowBy;  // set new size
  67. if (nNewSize == 0)
  68. {
  69. // shrink to nothing
  70. if (m_pData != NULL)
  71. {
  72. DestructElements(m_pData, m_nSize);
  73. delete[] (char*)m_pData;
  74. m_pData = NULL;
  75. }
  76. m_nSize = m_nMaxSize = 0;
  77. }
  78. else if (m_pData == NULL)
  79. {
  80. // create one with exact size
  81. #ifdef SIZE_T_MAX
  82. assert(nNewSize <= SIZE_T_MAX/sizeof(TYPE));    // no overflow
  83. #endif
  84. m_pData = (TYPE*) new char[nNewSize * sizeof(TYPE)];
  85. ConstructElements(m_pData, nNewSize);
  86. m_nSize = m_nMaxSize = nNewSize;
  87. }
  88. else if (nNewSize <= m_nMaxSize)
  89. {
  90. // it fits
  91. if (nNewSize > m_nSize)
  92. {
  93. // initialize the new elements
  94. ConstructElements(&m_pData[m_nSize], nNewSize-m_nSize);
  95. }
  96. else if (m_nSize > nNewSize)
  97. {
  98. // destroy the old elements
  99. DestructElements(&m_pData[nNewSize], m_nSize-nNewSize);
  100. }
  101. m_nSize = nNewSize;
  102. }
  103. else
  104. {
  105. // otherwise, grow array
  106. int nGrowBy = m_nGrowBy;
  107. if (nGrowBy == 0)
  108. {
  109. // heuristically determine growth when nGrowBy == 0
  110. //  (this avoids heap fragmentation in many situations)
  111. nGrowBy = m_nSize / 8;
  112. nGrowBy = (nGrowBy < 4) ? 4 : ((nGrowBy > 1024) ? 1024 : nGrowBy);
  113. }
  114. int nNewMax;
  115. if (nNewSize < m_nMaxSize + nGrowBy)
  116. nNewMax = m_nMaxSize + nGrowBy;  // granularity
  117. else
  118. nNewMax = nNewSize;  // no slush
  119. assert(nNewMax >= m_nMaxSize);  // no wrap around
  120. #ifdef SIZE_T_MAX
  121. assert(nNewMax <= SIZE_T_MAX/sizeof(TYPE)); // no overflow
  122. #endif
  123. TYPE* pNewData = (TYPE*) new char[nNewMax * sizeof(TYPE)];
  124. // copy new data from old
  125. memcpy(pNewData, m_pData, m_nSize * sizeof(TYPE));
  126. // construct remaining elements
  127. assert(nNewSize > m_nSize);
  128. ConstructElements(&pNewData[m_nSize], nNewSize-m_nSize);
  129. // get rid of old stuff (note: no destructors called)
  130. delete[] (char*)m_pData;
  131. m_pData = pNewData;
  132. m_nSize = nNewSize;
  133. m_nMaxSize = nNewMax;
  134. }
  135. }
  136. template<class TYPE, class ARG_TYPE>
  137. int CClassArray<TYPE, ARG_TYPE>::Append(const CClassArray& src)
  138. {
  139. assert(this != &src);   // cannot append to itself
  140. int nOldSize = m_nSize;
  141. SetSize(m_nSize + src.m_nSize);
  142. CopyElements(m_pData + nOldSize, src.m_pData, src.m_nSize);
  143. return nOldSize;
  144. }
  145. template<class TYPE, class ARG_TYPE>
  146. void CClassArray<TYPE, ARG_TYPE>::Copy(const CClassArray& src)
  147. {
  148. assert(this != &src);   // cannot append to itself
  149. SetSize(src.m_nSize);
  150. CopyElements(m_pData, src.m_pData, src.m_nSize);
  151. }
  152. template<class TYPE, class ARG_TYPE>
  153. void CClassArray<TYPE, ARG_TYPE>::FreeExtra()
  154. {
  155. if (m_nSize != m_nMaxSize)
  156. {
  157. // shrink to desired size
  158. #ifdef SIZE_T_MAX
  159. assert(m_nSize <= SIZE_T_MAX/sizeof(TYPE)); // no overflow
  160. #endif
  161. TYPE* pNewData = NULL;
  162. if (m_nSize != 0)
  163. {
  164. pNewData = (TYPE*) new char[m_nSize * sizeof(TYPE)];
  165. // copy new data from old
  166. memcpy(pNewData, m_pData, m_nSize * sizeof(TYPE));
  167. }
  168. // get rid of old stuff (note: no destructors called)
  169. delete[] (char*)m_pData;
  170. m_pData = pNewData;
  171. m_nMaxSize = m_nSize;
  172. }
  173. }
  174. template<class TYPE, class ARG_TYPE>
  175. void CClassArray<TYPE, ARG_TYPE>::SetAtGrow(int nIndex, ARG_TYPE newElement)
  176. {
  177. assert(nIndex >= 0);
  178. if (nIndex >= m_nSize)
  179. SetSize(nIndex+1, -1);
  180. m_pData[nIndex] = newElement;
  181. }
  182. template<class TYPE, class ARG_TYPE>
  183. void CClassArray<TYPE, ARG_TYPE>::InsertAt(int nIndex, ARG_TYPE newElement, int nCount /*=1*/)
  184. {
  185. assert(nIndex >= 0);    // will expand to meet need
  186. assert(nCount > 0);     // zero or negative size not allowed
  187. if (nIndex >= m_nSize)
  188. {
  189. // adding after the end of the array
  190. SetSize(nIndex + nCount, -1);   // grow so nIndex is valid
  191. }
  192. else
  193. {
  194. // inserting in the middle of the array
  195. int nOldSize = m_nSize;
  196. SetSize(m_nSize + nCount, -1);  // grow it to new size
  197. // destroy intial data before copying over it
  198. DestructElements(&m_pData[nOldSize], nCount);
  199. // shift old data up to fill gap
  200. memmove(&m_pData[nIndex+nCount], &m_pData[nIndex],
  201. (nOldSize-nIndex) * sizeof(TYPE));
  202. // re-init slots we copied from
  203. ConstructElements(&m_pData[nIndex], nCount);
  204. }
  205. // insert new value in the gap
  206. assert(nIndex + nCount <= m_nSize);
  207. while (nCount--)
  208. m_pData[nIndex++] = newElement;
  209. }
  210. template<class TYPE, class ARG_TYPE>
  211. void CClassArray<TYPE, ARG_TYPE>::RemoveAt(int nIndex, int nCount)
  212. {
  213. assert(nIndex >= 0);
  214. assert(nCount >= 0);
  215. assert(nIndex + nCount <= m_nSize);
  216. // just remove a range
  217. int nMoveCount = m_nSize - (nIndex + nCount);
  218. DestructElements(&m_pData[nIndex], nCount);
  219. if (nMoveCount)
  220. memmove(&m_pData[nIndex], &m_pData[nIndex + nCount],
  221. nMoveCount * sizeof(TYPE));
  222. m_nSize -= nCount;
  223. }
  224. template<class TYPE, class ARG_TYPE>
  225. void CClassArray<TYPE, ARG_TYPE>::InsertAt(int nStartIndex, CClassArray* pNewArray)
  226. {
  227. assert(pNewArray != NULL);
  228. assert(nStartIndex >= 0);
  229. if (pNewArray->GetSize() > 0)
  230. {
  231. InsertAt(nStartIndex, pNewArray->GetAt(0), pNewArray->GetSize());
  232. for (int i = 0; i < pNewArray->GetSize(); i++)
  233. SetAt(nStartIndex + i, pNewArray->GetAt(i));
  234. }
  235. }
  236. template<class TYPE, class ARG_TYPE>
  237. void CClassArray<TYPE, ARG_TYPE>::ConstructElements(TYPE* pElements, int nCount)
  238. {
  239. // first do bit-wise zero initialization
  240. memset((void*)pElements, 0, nCount * sizeof(TYPE));
  241. // then call the constructor(s)
  242. for (; nCount--; pElements++)
  243. ::new((void*)pElements) TYPE;
  244. }
  245. template<class TYPE, class ARG_TYPE>
  246. void CClassArray<TYPE, ARG_TYPE>::DestructElements(TYPE* pElements, int nCount)
  247. {
  248. // call the destructor(s)
  249. for (; nCount--; pElements++)
  250. pElements->~TYPE();
  251. }
  252. template<class TYPE, class ARG_TYPE>
  253. void CClassArray<TYPE, ARG_TYPE>::CopyElements(TYPE* pDest, const TYPE* pSrc, int nCount)
  254. {
  255. // default is element-copy using assignment
  256. while (nCount--)
  257. *pDest++ = *pSrc++;
  258. }
  259. #endif