MyVector.h 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254
  1. // Common/Vector.h
  2. #ifndef __COMMON_VECTOR_H
  3. #define __COMMON_VECTOR_H
  4. #include "Defs.h"
  5. class CBaseRecordVector
  6. {
  7. void MoveItems(int destIndex, int srcIndex);
  8. protected:
  9. int _capacity;
  10. int _size;
  11. void *_items;
  12. size_t _itemSize;
  13. void ReserveOnePosition();
  14. void InsertOneItem(int index);
  15. void TestIndexAndCorrectNum(int index, int &num) const
  16. { if (index + num > _size) num = _size - index; }
  17. public:
  18. CBaseRecordVector(size_t itemSize):
  19. _capacity(0), _size(0), _items(0), _itemSize(itemSize) {}
  20. virtual ~CBaseRecordVector();
  21. void Free();
  22. int Size() const { return _size; }
  23. bool IsEmpty() const { return (_size == 0); }
  24. void Reserve(int newCapacity);
  25. virtual void Delete(int index, int num = 1);
  26. void Clear();
  27. void DeleteFrom(int index);
  28. void DeleteBack();
  29. };
  30. template <class T>
  31. class CRecordVector: public CBaseRecordVector
  32. {
  33. public:
  34. CRecordVector():CBaseRecordVector(sizeof(T)){};
  35. CRecordVector(const CRecordVector &v):
  36. CBaseRecordVector(sizeof(T)) { *this = v;}
  37. CRecordVector& operator=(const CRecordVector &v)
  38. {
  39. Clear();
  40. return (*this += v);
  41. }
  42. CRecordVector& operator+=(const CRecordVector &v)
  43. {
  44. int size = v.Size();
  45. Reserve(Size() + size);
  46. for(int i = 0; i < size; i++)
  47. Add(v[i]);
  48. return *this;
  49. }
  50. int Add(T item)
  51. {
  52. ReserveOnePosition();
  53. ((T *)_items)[_size] = item;
  54. return _size++;
  55. }
  56. void Insert(int index, T item)
  57. {
  58. InsertOneItem(index);
  59. ((T *)_items)[index] = item;
  60. }
  61. // T* GetPointer() const { return (T*)_items; }
  62. // operator const T *() const { return _items; };
  63. const T& operator[](int index) const { return ((T *)_items)[index]; }
  64. T& operator[](int index) { return ((T *)_items)[index]; }
  65. const T& Front() const { return operator[](0); }
  66. T& Front() { return operator[](0); }
  67. const T& Back() const { return operator[](_size - 1); }
  68. T& Back() { return operator[](_size - 1); }
  69. void Swap(int i, int j)
  70. {
  71. T temp = operator[](i);
  72. operator[](i) = operator[](j);
  73. operator[](j) = temp;
  74. }
  75. int FindInSorted(const T& item) const
  76. {
  77. int left = 0, right = Size();
  78. while (left != right)
  79. {
  80. int mid = (left + right) / 2;
  81. const T& midValue = (*this)[mid];
  82. if (item == midValue)
  83. return mid;
  84. if (item < midValue)
  85. right = mid;
  86. else
  87. left = mid + 1;
  88. }
  89. return -1;
  90. }
  91. int AddToUniqueSorted(const T& item)
  92. {
  93. int left = 0, right = Size();
  94. while (left != right)
  95. {
  96. int mid = (left + right) / 2;
  97. const T& midValue = (*this)[mid];
  98. if (item == midValue)
  99. return mid;
  100. if (item < midValue)
  101. right = mid;
  102. else
  103. left = mid + 1;
  104. }
  105. Insert(right, item);
  106. return right;
  107. }
  108. static void SortRefDown(T* p, int k, int size, int (*compare)(const T*, const T*, void *), void *param)
  109. {
  110. T temp = p[k];
  111. for (;;)
  112. {
  113. int s = (k << 1);
  114. if (s > size)
  115. break;
  116. if (s < size && compare(p + s + 1, p + s, param) > 0)
  117. s++;
  118. if (compare(&temp, p + s, param) >= 0)
  119. break;
  120. p[k] = p[s];
  121. k = s;
  122. }
  123. p[k] = temp;
  124. }
  125. void Sort(int (*compare)(const T*, const T*, void *), void *param)
  126. {
  127. int size = _size;
  128. if (size <= 1)
  129. return;
  130. T* p = (&Front()) - 1;
  131. {
  132. int i = size / 2;
  133. do
  134. SortRefDown(p, i, size, compare, param);
  135. while(--i != 0);
  136. }
  137. do
  138. {
  139. T temp = p[size];
  140. p[size--] = p[1];
  141. p[1] = temp;
  142. SortRefDown(p, 1, size, compare, param);
  143. }
  144. while (size > 1);
  145. }
  146. };
  147. typedef CRecordVector<int> CIntVector;
  148. typedef CRecordVector<unsigned int> CUIntVector;
  149. typedef CRecordVector<bool> CBoolVector;
  150. typedef CRecordVector<unsigned char> CByteVector;
  151. typedef CRecordVector<void *> CPointerVector;
  152. template <class T>
  153. class CObjectVector: public CPointerVector
  154. {
  155. public:
  156. CObjectVector(){};
  157. ~CObjectVector() { Clear(); }
  158. CObjectVector(const CObjectVector &objectVector)
  159. { *this = objectVector; }
  160. CObjectVector& operator=(const CObjectVector &objectVector)
  161. {
  162. Clear();
  163. return (*this += objectVector);
  164. }
  165. CObjectVector& operator+=(const CObjectVector &objectVector)
  166. {
  167. int size = objectVector.Size();
  168. Reserve(Size() + size);
  169. for(int i = 0; i < size; i++)
  170. Add(objectVector[i]);
  171. return *this;
  172. }
  173. const T& operator[](int index) const { return *((T *)CPointerVector::operator[](index)); }
  174. T& operator[](int index) { return *((T *)CPointerVector::operator[](index)); }
  175. T& Front() { return operator[](0); }
  176. const T& Front() const { return operator[](0); }
  177. T& Back() { return operator[](_size - 1); }
  178. const T& Back() const { return operator[](_size - 1); }
  179. int Add(const T& item)
  180. { return CPointerVector::Add(new T(item)); }
  181. void Insert(int index, const T& item)
  182. { CPointerVector::Insert(index, new T(item)); }
  183. virtual void Delete(int index, int num = 1)
  184. {
  185. TestIndexAndCorrectNum(index, num);
  186. for(int i = 0; i < num; i++)
  187. delete (T *)(((void **)_items)[index + i]);
  188. CPointerVector::Delete(index, num);
  189. }
  190. int Find(const T& item) const
  191. {
  192. for(int i = 0; i < Size(); i++)
  193. if (item == (*this)[i])
  194. return i;
  195. return -1;
  196. }
  197. int FindInSorted(const T& item) const
  198. {
  199. int left = 0, right = Size();
  200. while (left != right)
  201. {
  202. int mid = (left + right) / 2;
  203. const T& midValue = (*this)[mid];
  204. if (item == midValue)
  205. return mid;
  206. if (item < midValue)
  207. right = mid;
  208. else
  209. left = mid + 1;
  210. }
  211. return -1;
  212. }
  213. int AddToSorted(const T& item)
  214. {
  215. int left = 0, right = Size();
  216. while (left != right)
  217. {
  218. int mid = (left + right) / 2;
  219. const T& midValue = (*this)[mid];
  220. if (item == midValue)
  221. {
  222. right = mid + 1;
  223. break;
  224. }
  225. if (item < midValue)
  226. right = mid;
  227. else
  228. left = mid + 1;
  229. }
  230. Insert(right, item);
  231. return right;
  232. }
  233. void Sort(int (*compare)(void *const *, void *const *, void *), void *param)
  234. { CPointerVector::Sort(compare, param); }
  235. static int CompareObjectItems(void *const *a1, void *const *a2, void * /* param */)
  236. { return MyCompare(*(*((const T **)a1)), *(*((const T **)a2))); }
  237. void Sort() { CPointerVector::Sort(CompareObjectItems, 0); }
  238. };
  239. #endif