// MT4Q2a.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(T* pT)
  {
    for(int i=0; i<Size; ++i)
      _array[i] = *(pT + i);
  }
  // would be acceptable to make private copy declar.
  Array(const Array& arr)
  {
    if(Size != arr.size()) throw ("not same size");
    for(int i=0; i<Size; ++i)
      _array[i] = arr._array[i];
  }
  int size() const { return Size; }

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

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);

  Array<std::string,5>::iterator iter = arr1.begin();
  std::cout << "\n  first element is: " << *iter;
  std::cout << "\n  first element is: " << arr1[0];
  iter = arr1.end();
  std::cout << "\n  last element is: " << *(--iter);
  std::cout << "\n  last element is: " << arr1[arr1.size()-1];
  std::cout << "\n  sum = " << sum(arr1) << "\n";

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

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