diff --git a/app/src/main/java/model/CellGridIterator.java b/app/src/main/java/model/CellGridIterator.java index 97d702e7375be1d0732ce258ac92c72c5bd2917e..8372f821e6cbb673737db672349bd0286075695e 100644 --- a/app/src/main/java/model/CellGridIterator.java +++ b/app/src/main/java/model/CellGridIterator.java @@ -4,9 +4,13 @@ import java.util.Iterator; public class CellGridIterator implements Iterator<Cell> { public ArrayGrid grid; + public int row; + public int column; public CellGridIterator(ArrayGrid grid) { this.grid=grid; + this.row=0; + this.column=0; } @Override @@ -16,12 +20,14 @@ public class CellGridIterator implements Iterator<Cell> { @Override public Cell next() { - for(int i = 0; i < grid.getNumberOfRows() ; i++) { - for(int j = 0; j < grid.getNumberOfColumns(); j++) { - - } + if (row>=0 && row< grid.getNumberOfRows()-1){ + this.row=row+1; + } + else if(row== grid.getNumberOfRows()-1 && column< grid.getNumberOfColumns()-1){ + this.row=0; + this.column+=1; } - return Cell; + return grid.getCell(column,row); } }