#include "Vector.hpp"

Vector::Vector(int length) {
  size = length;
  array = new (nothrow) int[size];
  if(array==NULL) {
    cerr << "Memory allocation failure!" << endl;
    exit(-1);
  }
  for(int i=0; i<size; i++)
    array[i] = 0;
}

Vector::Vector(const Vector& v) {
  size = v.length();
  array = new (nothrow) int[size];
  if(array==NULL) {
    cerr << "Memory allocation failure!" << endl;
    exit(-1);
  }
  for(int i=0; i<size; i++)
    array[i] = v.valueAt(i);
}

Vector::Vector(const Vector *v) {
  size = v->length();
  array = new (nothrow) int[size];
  if(array==NULL) {
    cerr << "Memory allocation failure!" << endl;
    exit(-1);
  }
  for(int i=0; i<size; i++)
    array[i] = v->valueAt(i);
}

Vector::~Vector() {
  delete [] array;
}

int Vector::length() const { 
  return size; 
}

int &Vector::valueAt(int pos) const {
  if(pos>=length()) {
     cerr << "Invalid access position!\n";
     return array[size-1];
  }
  return array[pos];
}

int Vector::find(int a) const {
  for(int i=0; i<size; i++)
    if(array[i] == a)
      return i;
    return -1;
}

void Vector::print() const {
  for(int i=0; i<size; i++) {
    cout << array[i];
    if(i==size-1)
      cout << endl;
    else
      cout << ", ";
  }
}

void Vector::print(string &msg) const {
  cout << msg;
  print();
}

Vector& Vector::operator=(const Vector& v) {
  if(array!=NULL)
    delete [] array;
  size = v.length();
  array = new (nothrow) int[size];
  if(array==NULL) {
    cerr << "Memory allocation failure!" << endl;
    exit(-1);
  }
  for(int i=0; i<size; i++)
    array[i] = v.valueAt(i);
  return *this;
}

Vector Vector::operator+(const Vector& v) const {
  int length;
  if (size > v.length())
    length = size;
  else
    length = v.length();
  Vector n(length);
  for(int i=0; i<length; i++) {
    int sum = 0;
    if(i<size)
      sum += array[i];
    if(i<v.length())
      sum += v.array[i];
    n.array[i] = sum;
  }
  return n;
}

/*
Vector operator+(const Vector& v1, const Vector v2) {
  int length;
  if (v1.size > v2.size)
    length = v1.length();
  else
    length = v2.length();
  Vector n(length);
  for(int i=0; i<length; i++) {
    int sum = 0;
    if(i<v1.length())
      sum += v1.array[i];
    if(i<v2.length())
      sum += v2.array[i];
    n.array[i] = sum;
  }
  return n;
}*/

Vector Vector::operator-(const Vector& v) const {
  int length;
  if (size > v.length())
    length = size;
  else
    length = v.length();
  Vector n(length);
  for(int i=0; i<length; i++) {
    int sum = 0;
    if(i<size)
      sum += array[i];
    if(i<v.length())
      sum -= v.array[i];
    n.array[i] = sum;
  }
  return n;
}

Vector Vector::operator*(const Vector& v) const {
  int length;
  if (size < v.size)
    length = size;
  else
    length = v.size;
  Vector n(length);
  for(int i=0; i<length; i++)
    n.array[i] = array[i] * v.array[i];
  return n;
}

Vector operator+(const Vector& v, int a) {
  Vector n(v.size);
  for(int i=0; i<v.size; i++)
    n.array[i] = v.array[i] + a;
  return n;
}

Vector operator+(int a, const Vector& v) {
  Vector n(v);
  for(int i=0; i<v.length(); i++) {
    n.array[i] += a;
  }
  return n;
}

Vector Vector::operator-(int a) const {
  Vector n(size);
  for(int i=0; i<size; i++)
    n.array[i] = array[i] - a;
  return n;
}

Vector Vector::operator*(int a) const {
  Vector n(size);
  for(int i=0; i<size; i++)
    n.array[i] = array[i] * a;
  return n;
}

Vector operator*(int a, const Vector& v) {
  Vector n(v);
  for(int i=0; i<v.length(); i++) {
    n.array[i] *= a;
  }
  return n;
}

Vector Vector::operator/(int a) const {
  Vector n(size);
  for(int i=0; i<size; i++)
    n.array[i] = array[i] / a;
  return n;
}

Vector Vector::operator%(int a) const {
  Vector n(size);
  for(int i=0; i<size; i++)
    n.array[i] = array[i] % a;
  return n;
}

bool Vector::operator==(const Vector& v) const {
  if( size != v.size )
    return false;
  for(int i=0; i<size; i++)
    if( array[i] != v.array[i] )
      return false;
  return true;
}

bool Vector::operator!=(const Vector& v) const {
  return !(*this == v);
}

/*
Vector operator<<(const Vector& v, int a){
  Vector n(v);
  for(int i=0; i<n.size; i++)
    n.array[i] <<= a;
  return n;
}*/

Vector Vector::operator << (int a) const {
  Vector n(*this);
  for(int i=0; i<n.size; i++)
    n.array[i] <<= a;
  return n;
}

/*Vector operator>>(const Vector& v, int a) {
  Vector n(v);
  for(int i=0; i<n.size; i++)
    n.array[i] >>= a;
  return n;
}*/

Vector Vector::operator >> (int a) const {
  Vector n(*this);
  for(int i=0; i<n.size; i++)
    n.array[i] >>= a;
  return n;
}