// -*- coding: utf-8 -*-
// Time-stamp: <iterator.hpp  14 déc 2016 23:22:49>

#ifndef ITERATOR_H
#define ITERATOR_H

#include "array2d.hpp"

template<class T>
class Iterator {
public:
  // Constructors
  Iterator(Array2d<T>& a, int m, int n) : rep(a.rep), i(m), j(n) {}
  // Equality
  bool operator==(Iterator it) const;
  // Non-equality defined using equality
  bool operator!=(Iterator it) const { return !(*this == it); }
  // Incrementation
  Iterator& operator++();
  // Dereference
  T& operator*() const;
private:
  Rep<T>* rep;			// Link to the array
  int i;			// Line index
  int j;			// Column index
};


template<class T>
bool Iterator<T>::operator==(Iterator it) const {
  return i == it.i && j == it.j;
}


// Incrementation
template<class T>
Iterator<T>& Iterator<T>::operator++() {
  if (i < rep->nlin && ++j == rep->ncol) {
    j = 0;
    ++i;
  }
  return *this;
}


// Dereference
template<class T>
T& Iterator<T>::operator*() const {
  return (*rep)[i][j];
}

#endif