// MT1Q4b - Array class with policy

#include <string>
#include <iostream>

struct rowDisplayPolicy
{
  static std::string seperator() { return ", "; }
};

struct columnDisplayPolicy
{
  static std::string seperator() { return "\n  "; }
};

template <typename T, int Size, typename DisplayPolicy>
class Array {
public:
  Array() : pArray(new T[Size]) {}
  Array(T* pT) : pArray(new T[Size])
  {
    for(int i=0; i<Size; ++i)
      *(pArray + i) = *(pT + i);
  }
  ~Array() { delete [] pArray; }

  // would be acceptable to make copy private declar.
  Array(const Array<T,Size,DisplayPolicy>& 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);
  }
  int size() const { return Size; }

  // would be acceptable to make assign private declar.
  Array<T,Size,DisplayPolicy>& operator=(const Array<T,Size,DisplayPolicy>& 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);
  }
  void display() const
  {
    std::cout << "\n  ";
    for(int i=0; i<Size-1; ++i)
      std::cout << *(pArray+i) << DisplayPolicy::seperator();
    std::cout << *(pArray+Size-1) << "\n";
  }
private:
  T* pArray;
};

int main()
{
  double temp1[] = { 0, 0.5, 1.0, 1.5, 2.0 };
  Array<double, 5, rowDisplayPolicy> arr1(temp1);
  arr1.display();

  int temp2[] = { 1, 2, 3, 4, 5 };
  Array<int, 5, columnDisplayPolicy> arr2(temp2);
  arr2.display();

  std::cout << "\n  copying:";
  Array<int, 5, columnDisplayPolicy> arr3 = arr2;
  arr3.display();

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