动态数组模板类

标签：

 具体的代码实现如下：

template <typename T>
class Array
{
public:
	typedef T TYPE;
	struct ArrayDesc
	{
		volatile int refer;
		volatile unsigned int capacity;
		volatile unsigned int length;
	};
protected:
	T* m_ptr;
public:
	Array()
	{
		m_ptr = NULL;
	}
	Array(const Array<T> &another)
	{
		m_ptr = NULL;
		operator = (another);
	}
	Array(const T* ptr, size_t length)
	{
		m_ptr = NULL;
		setLength(length);
		memcpy(m_ptr, ptr, length * sizeof(*ptr));
	}
	Array(size_t length)
	{
		m_ptr = NULL;
		setLength(length);
	}
	~Array()
	{
		setLength(0);
	}
	inline Array<T>& operator = (const Array<T> &another)
	{
		if (m_ptr != another.m_ptr)//检查不是同一个对象
		{
			setLength(0);//释放本对象的内存
			if (another.m_ptr)
			{
				ArrayDesc *pDesc = ((ArrayDesc*)another.m_ptr) - 1;//获取另一对象的数组描述
				lock_inc(pDesc->refer);//原子操作增加数组描述的引用计数
				m_ptr = (T*)(pDesc + 1);//把另一对象的数组指针复制到本对象的数组指针
			}
		}
		return *this;
	}
	inline Array<T> operator + (const Array<T> &another)
	{
		Array<T> result;
		if (m_ptr || another.m_ptr)
		{
			size_t myCount = length();
			size_t anotherCount = another.length();
			result.setLength(myCount + anotherCount);//预留目标的内存空间
			T* destPtr = result.m_ptr;
			if (myCount > 0)
			{
				memcpy(destPtr, m_ptr, myCount * sizeof(*m_ptr));//拷贝本对象的数据内存
				destPtr += myCount;
			}
			if (anotherCount > 0)
			{
				memcpy(destPtr, another.m_ptr, anotherCount * sizeof(*m_ptr));//拷贝另一个对象的数据内存
				destPtr += myCount;
			}
		}
		return result;//拷贝复制结果对象
	}
	inline Array<T> operator + (const T &another)//追加一个数据对象，返回的是新的数据对象的拷贝
	{
		size_t myCount = length();
		Array<T> result(myCount + 1);//预留结果对象的内存空间
		if (myCount > 0)
			memcpy(result.m_ptr, m_ptr, myCount * sizeof(*m_ptr));//拷贝本对象的数据到目标内存
		result.m_ptr[myCount] = another;//拷贝追加的数据对象
		return result;
	}
	inline Array<T>& operator += (const Array<T> &another)//追加一个动态数组对象
	{
		size_t len = another.length();
		if (len > 0 )
			cat(another.m_ptr, another.length());
		return *this;
	}
	inline Array<T>& operator += (const T &another)//追加一个数据对象，返回的是本对象
	{
		size_t myCount = length();
		setLength(myCount + 1);
		m_ptr[myCount] = another;
		return *this;
	}
	inline T& operator [] (size_t index)//获取动态数组的数据索引上的数据对象的引用
	{
		if (!m_ptr)
			return *(T*)NULL;
		ArrayDesc *pDesc = ((ArrayDesc*)m_ptr) - 1;
		assert(index < pDesc->length);//检查索引不超出动态数组对象的长度
		duplicate();
		return m_ptr[index];
	}
	inline const T& operator [] (size_t index) const //获取动态数组的数据索引上的数据对象的引用(cosnt类型)
	{
		if (!m_ptr)
			return *(T*)NULL;
		ArrayDesc * const pDesc = ((ArrayDesc* const)m_ptr) - 1;
		assert(index < pDesc->length);//检查索引不超出动态数组对象的长度
		return m_ptr[index];
	}
	inline bool operator == (const Array<T> &another) const//判断数据对象是否是同一个对象
	{
		return m_ptr == another.m_ptr;
	}
	inline bool operator != (const Array<T> &another) const//判断数据对象是否不是同一个对象
	{
		return m_ptr != another.m_ptr;
	}
	
	inline const T* ptr() const//获取只读的内容数据指针
	{
		return m_ptr;
	}
	
	inline T* own_ptr()//获取可写的内容数据指针
	{
		duplicate();//复制本对象的数据内容到一个新的对象上，本对象就成了一个新的对象，旧的对象的数据内存需要在别处有管理
		return m_ptr;
	}
	inline size_t length() const//获取本对象的数据数组长度
	{
		if (!m_ptr)
			return 0;
		ArrayDesc * const pDesc = ((ArrayDesc* const)m_ptr) - 1;
		return pDesc->length;
	}
	inline Array<T>& cat(const T* ptr, size_t count)//追加一个或连续的多个数据对象到本数组对象的末尾
	{
		return insert(length(), ptr, count);
	}
	inline Array<T>& insert(size_t index, const T* ptr, size_t count)//在指定索引位置插入一个或者连续的多个数据对象
	{
		if (count > 0)
		{
			ArrayDesc *pMyDesc = m_ptr ? ((ArrayDesc*)m_ptr) - 1 : NULL;
			size_t myCount = pMyDesc ? pMyDesc->length : 0;//本数组的数据对象数

			assert(index <= myCount);

			//判断目标是否是来自自己内存区域的数据，如果是来自自己内存区域的，那么可能
			//在setLength后ptr指针失效，此时需要重新确定ptr的位置。
			bool isMySelf = false;
			size_t inMyPos = 0;

			if (pMyDesc && (ptr >= m_ptr) && (ptr <= (m_ptr + pMyDesc->capacity)))
			{
				isMySelf = true;//目标对象的起始地址在本对象数组的内存区域内
				inMyPos = ptr - m_ptr;//在内存区域的索引位置
				if (inMyPos >= index)
					inMyPos += count;//重新定位需要拷贝的对象的起始位置ptr（跟setLength的实现有关）
			}

			setLength(myCount + count);//预留本对象的数据内存大小
			if (index < myCount)
				memmove(&m_ptr[index + count], &m_ptr[index], sizeof(*ptr) * (myCount - index));//拷贝会被覆盖的数据内存到末尾。当内存发生局部重叠的时候，memmove保证拷贝的结果是正确的，memcpy不保证拷贝的结果的正确
			if (isMySelf)
				memcpy(&m_ptr[index], &m_ptr[inMyPos], count * sizeof(*m_ptr));//拷贝需要拷贝的数据对象到指定的位置
			else memcpy(&m_ptr[index], ptr, count * sizeof(*m_ptr));
		}
		return *this;
	}
	inline Array<T>& remove(size_t index, size_t count)
	{
		if (count > 0)
		{
			duplicate();//拷贝本数组对象到新的内存

			ArrayDesc *pMyDesc = m_ptr ? ((ArrayDesc*)m_ptr) - 1 : NULL;
			size_t myCount = pMyDesc ? pMyDesc->length : 0;

			assert(index < myCount);
			if (count > myCount - index) 
				count = myCount - index;
			size_t copyCount = myCount - index - count;
			if (copyCount > 0)
				memcpy(&m_ptr[index], &m_ptr[index + count], sizeof(*m_ptr) * copyCount);
			setLength(myCount - count);
		}
		return *this;
	}
	inline Array<T>& setLength(const size_t length)
	{
		assert(length >= 0);

		ArrayDesc *pDesc = NULL;
		if (m_ptr) pDesc = ((ArrayDesc*)m_ptr) - 1;

		if (length == 0)//回收本数组对象
		{
			if (pDesc)
			{
				if (lock_dec(pDesc->refer) <= 0)//引用计数为0，释放本数组对象的内存
					free(pDesc);
				m_ptr = NULL;
			}
		}
		else if (pDesc)
		{
			if (length != pDesc->length)
			{
				if (pDesc->refer > 1)//引用计数大于1时返回的是内存拷贝的数组对象
				{
					lock_dec(pDesc->refer);//减少本对象的引用计数
					ArrayDesc *pNewDesc = (ArrayDesc*)calloc(1, sizeof(*pDesc) + sizeof(*m_ptr) * (length + 1));//分配新的内存对象，calloc分配连续内存，初始化已分配的内存为0,返回数组
					pNewDesc->capacity = (unsigned int)length;
					pNewDesc->length = (unsigned int)length;
					pNewDesc->refer = 1;
					m_ptr = (T*)(pNewDesc + 1);//把本数组对象重新定位到新的数组对象内存
					size_t lengthMin = pDesc->length < length ? pDesc->length : length;
					memcpy(m_ptr, pDesc + 1, lengthMin * sizeof(*m_ptr));//拷贝本数组对象内存到新的对象的内存
				}
				else //引用计数为1是返回的就是本对象，有效长度和内存大小根据需求调整
				{
					if (length >= pDesc->capacity)//需要的内存大小大于本数组对象的包含的内存大小
					{
						pDesc = (ArrayDesc*)realloc(pDesc, sizeof(*pDesc) + sizeof(*m_ptr) * (length + 1));
						m_ptr = (T*)(pDesc + 1);//重新定位本数组对象的数据指针到新的内存地址
						memset(&m_ptr[pDesc->length], 0, sizeof(*m_ptr) * (length - pDesc->capacity + 1));//把数组对象末尾的无效对象初始化
						pDesc->capacity = (unsigned int)length;//内存大小
						pDesc->length = (unsigned int)length;//数组大小
					}
					else pDesc->length = (unsigned int)length;//截短本数组对象的有效长度
				}
				memset(&m_ptr[length], 0, sizeof(*m_ptr));//数组有效长度后面的那个对象初始化为0（在数组对象为字符串时有作用）
			}


		}
		else //数组数据对象指针没有内存，则分配新的内存并把本数组对象数据指针重新定位到新的内存上
		{
			ArrayDesc *pNewDesc = (ArrayDesc*)calloc(1, sizeof(*pDesc) + sizeof(*m_ptr) * (length + 1));
			pNewDesc->capacity = (unsigned int)length;
			pNewDesc->length = (unsigned int)length;
			pNewDesc->refer = 1;
			m_ptr = (T*)(pNewDesc + 1);
		}
		return *this;
	}
	inline size_t capacity() const
	{
		if (m_ptr)
		{
			ArrayDesc *pDesc = ((ArrayDesc*)m_ptr) - 1;
			return pDesc->capacity;
		}
		return 0;
	}
	inline Array<T>& setCapacity(size_t capacity)
	{
		size_t myCount = length();
		//容量必须大于数量。如果要进行清空操作，则请调用setLength(0)
		if (capacity > myCount)
		{
			setLength(capacity);
			ArrayDesc *pDesc = ((ArrayDesc*)m_ptr) - 1;
			pDesc->length = myCount;
		}
		return *this;
	}
protected:
	inline void duplicate()//拷贝复制本数组对象到新的内存
	{
		if (m_ptr)
		{
			ArrayDesc *pDesc = ((ArrayDesc*)m_ptr) - 1;//记录本数组对象的描述块指针
			if (pDesc->refer > 1)
			{
				ArrayDesc *pNewDesc = (ArrayDesc*)malloc(sizeof(*pDesc) + sizeof(*m_ptr) * (pDesc->capacity + 1));//多预留了一个数据对象的大小
				pNewDesc->capacity = pDesc->capacity;
				pNewDesc->length = pDesc->length;
				pNewDesc->refer = 1;
				m_ptr = (T*)(pNewDesc + 1);//把本数组对象重新定位到新的内存
				memcpy( m_ptr,pDesc + 1,, (pDesc->length + 1) * sizeof(*m_ptr));//拷贝本数组对象的原来的数据内存到新的数组对象内存

				lock_dec(pDesc->refer);
			}
		}
	}
};

动态数组模板类

标签：

原文地址：http://blog.csdn.net/chenjiayi_yun/article/details/44752051