//9-3.h
#ifndef ARRAY_CLASS
#define ARRAY_CLASS

#include <iostream>
#include <cstdlib>

#ifndef NULL
const int NULL = 0;
#endif  // NULL

//錯誤類型集合，共有三種類型的錯誤：數(shù)組大小錯誤、內(nèi)存分配錯誤和下標(biāo)越界
enum ErrorType  
  {invalidArraySize, memoryAllocationError, indexOutOfRange};

//錯誤信息
char *errorMsg[] =
{
    "Invalid array size", "Memory allocation error",
    "Invalid index: "
};

//數(shù)組類模板聲明
template <class T> 
class Array
{
    private:
      T*  alist;  //T類型指針，用于存放動態(tài)分配的數(shù)組內(nèi)存首地址
      int size;   //數(shù)組大小（元素個數(shù)）
      void Error(ErrorType error,int badIndex=0) const;   // 錯誤處理函數(shù)
    public:
      Array(int sz = 50);	//構(gòu)造函數(shù)
      Array(const Array<T>& A);	//拷貝構(gòu)造函數(shù)
      ~Array(void);	//析構(gòu)函數(shù)
      Array<T>& operator= (const Array<T>& rhs); //重載"="使數(shù)組對象可以整體賦值
      T& operator[](int i);     //重載"[]"，使Array對象可以起到C++普通數(shù)組的作用
      operator T* (void) const; //重載T*，使Array對象可以起到C++普通數(shù)組的作用
      int ListSize(void) const;	// 取數(shù)組的大小
      void Resize(int sz);	// 修改數(shù)組的大小
};

//以下為類成員函數(shù)的定義
//模扳函數(shù)Error實現(xiàn)輸出錯誤信息的功能
template <class T>
void Array<T>::Error(ErrorType error, int badIndex) const
{
	//由于頭文件有可能被許多其它源程序文件包含，因此不以將std名名空間中的全部
    //標(biāo)識符引入當(dāng)前命名空間，也就是說不宜使用"using namespace std;"。
    //為了使用std命名空間中的標(biāo)識符，如cout，這里使用了"std::"

	std::cout << errorMsg[error];	//根據(jù)錯誤類型，輸出相應(yīng)的錯誤信息
    if (error == indexOutOfRange)
        std::cout << badIndex;	//如果是下標(biāo)越界錯，輸出錯誤的下標(biāo)
    std::cout << endl;
    exit(1);
}

//構(gòu)造函數(shù)
template <class T>
Array<T>::Array(int sz)
{
    if (sz <= 0)             //sz為數(shù)組大小（元素個數(shù)），若小于0，則輸出錯誤信息
        Error(invalidArraySize);
    size = sz;	// 將元素個數(shù)賦值給變量size
    alist = new T[size];	//動態(tài)分配size個T類型的元素空間
    if (alist == NULL)	//如果分配內(nèi)存不成功，輸出錯誤信息
        Error(memoryAllocationError);
}

// 析構(gòu)函數(shù)
template <class T>
Array<T>::~Array(void)
{     delete [] alist;     }

// 拷貝構(gòu)造函數(shù)
template <class T>
Array<T>::Array(const Array<T>& X)
{
    //從對象X取得數(shù)組大小，并賦值給當(dāng)前對象的成員
    int n = X.size; 
    size = n;
    //為對象申請內(nèi)存并進行出錯檢查
    alist = new T[n];	// 動態(tài)分配n個T類型的元素空間
    if (alist == NULL)	//如果分配內(nèi)存不成功，輸出錯誤信息
        Error(memoryAllocationError);
    // 從對象X復(fù)制數(shù)組元素到本對象  
    T* srcptr = X.alist;	// X.alist是對象X的數(shù)組首地址
    T* destptr = alist;	// alist是本對象中的數(shù)組首地址
    while (n--)	// 逐個復(fù)制數(shù)組元素
        *destptr++ = *srcptr++;
}

// 重載"="運算符，將對象rhs賦值給本對象。實現(xiàn)對象之間的整體賦值
template <class T>
Array<T>& Array<T>::operator= (const Array<T>& rhs)
{
    int n = rhs.size;	// 取rhs的數(shù)組大小
//如果本對象中數(shù)組大小與rhs不同，則刪除數(shù)組原有內(nèi)存，然后重新分配
    if (size != n)
    {
        delete [] alist;	// 刪除數(shù)組原有內(nèi)存
        alist = new T[n];	// 重新分配n個元素的內(nèi)存
        if (alist == NULL)	//如果分配內(nèi)存不成功，輸出錯誤信息
            Error(memoryAllocationError);
        size = n;	//記錄本對象的數(shù)組大小
    }
   // 從rhs向本對象復(fù)制元素
   T* destptr = alist;
   T* srcptr = rhs.alist;
    while (n--) 
        *destptr++ = *srcptr++;
    return *this;	// 返回當(dāng)前對象的引用
}

// 重載下標(biāo)運算符，實現(xiàn)與普通數(shù)組一樣通過下標(biāo)訪問元素，并且具有越界檢查功能
template <class T>
T& Array<T>::operator[] (int n)
{
   if (n < 0 || n > size-1)	// 檢查下標(biāo)是否越界
      Error(indexOutOfRange,n);
   return alist[n];	// 返回下標(biāo)為n的數(shù)組元素
}

//重載指針轉(zhuǎn)換運算符，將Array類的對象名轉(zhuǎn)換為T類型的指針，
//指向當(dāng)前對象中的私有數(shù)組。
//因而可以象使用普通數(shù)組首地址一樣使用Array類的對象名
template <class T>
Array<T>::operator T* (void) const
{
    return alist;	// 返回當(dāng)前對象中私有數(shù)組的首地址
}

//取當(dāng)前數(shù)組的大小
template <class T>
int Array<T>::ListSize(void) const
{
    return size;
}

// 將數(shù)組大小修改為sz
template <class T>
void Array<T>::Resize(int sz)
{
    if (sz <= 0)	// 檢查是否sz<= 0
        Error(invalidArraySize);
    if (sz == size)	// 如果指定的大小與原有大小一樣，什么也不做
        return;
    T* newlist = new T[sz];	// 申請新的數(shù)組內(nèi)存
    if (newlist == NULL)	// 測試申請內(nèi)存是否申請成功
        Error(memoryAllocationError);
    int n = (sz <= size) ? sz : size;	// 將sz與size中較小的一個賦值給n
    // 將原有數(shù)組中前n個元素復(fù)制到新數(shù)組中
    T* srcptr = alist;      // 原數(shù)組alist的首地址
    T* destptr = newlist;   // 新數(shù)組newlist的首地址
    while (n--)             // 復(fù)制數(shù)組元素
        *destptr++ = *srcptr++;
    delete[] alist;	// 刪除原數(shù)組
    alist = newlist;	// 使alist 指向新數(shù)組
    size = sz;	// 更新size
}
#endif  // ARRAY_CLASS