// MT4Q2b.cpp - Array class with iterator

#include <string>
#include <iostream>

template <typename T, int Size>
class Array {
public:
  typedef T value_type;
  typedef T* iterator;

  Array() : pArray(new T[Size]) {}
  Array(T* pT) : pArray(new T[Size])
  {
    for(int i=0; i<Size; ++i)
      *(pArray + i) = *(pT + i);
  }
  // would be acceptable to make copy private declar.
  Array(const Array<T,Size>& arr)
  {
    if(Size != arr.size()) throw ("not same size");
    pArray = new T[Size];
    for(int i=0; i<Size; ++i)
      *(pArray+i) = *((arr.pArray)+i);
  }
  ~Array() { delete [] pArray; }
  int size() const { return Size; }

  // would be acceptable to make assign private declar.
  Array<T,Size>& operator=(const Array<T,Size>& arr)
  {
    if(Size != arr.size()) throw ("not same size");
    if(this == &arr) return *this;
    pArray = new T[Size];
    for(int i=0; i<Size; ++i)
      *(pArray+i) = *((arr.pArray)+i);
    return *this;
  }
  T& operator[](int n)
  {
    if(n<0 || Size<=n)
      throw std::exception("index out of range");
    return *(pArray+n);
  }
  T operator[](int n) const
  {
    if(n<0 || Size<=n)
      throw std::exception("index out of range");
    return *(pArray+n);
  }
  iterator begin()
  {
    return pArray;
  }
  iterator end()
  {
    return (pArray + Size);
  }
private:
  T* pArray;
};

template <typename Cont>
void display(Cont& c)
{
  Cont::iterator iter, last = c.end();
  last = --last;
  std::cout << "\n  ";
  for(iter = c.begin(); iter != last; ++iter)
    std::cout << *iter << ", ";
  std::cout << *last << "\n";
}

template <typename Cont>
typename Cont::value_type sum(Cont& c)
{
  Cont::iterator iter;
  Cont::value_type sum_ = Cont::value_type();
  for(iter = c.begin(); iter != c.end(); ++iter)
    sum_ += *iter;
  return sum_;
}

int main()
{
  std::string temp1[] = { "aardvark", "armadillo", "koala", "platypus", "vole"  };
  Array<std::string, 5> arr1(temp1);
  display(arr1);
  std::cout << "\n  sum = " << sum(arr1) << "\n";

  std::cout << "\n  copying:";
  Array<std::string, 5> arr2 = arr1;
  display(arr2);

  std::cout << "\n  assigning:";
  arr1 = arr2;
  display(arr1);
  std::cout << "\n";
}