refo/src/main/java/evironment/antGame/Grid.java

package evironment.antGame;

import core.RNG;

import java.awt.*;

public class Grid {
    private int width;
    private int height;
    private double foodDensity;
    private Point start;
    private Cell[][] grid;
    private Cell[][] initialGrid;

    public Grid(int width, int height, double foodDensity){
        this.width = width;
        this.height = height;
        this.foodDensity = foodDensity;
        grid = new Cell[width][height];
        initialGrid = new Cell[width][height];
        initRandomWorld();
    }

    public Grid(int width, int height){
        this(width, height, 0);
    }

    public void resetWorld(){
        grid = Util.deepCopyCellGrid(initialGrid);
    }

    public void initRandomWorld(){
        for(int x = 0; x < width; ++x){
            for(int y = 0; y < height; ++y){
                if( RNG.getRandom().nextDouble() < foodDensity){
                    initialGrid[x][y] = new Cell(new Point(x,y), CellType.FREE, 1);
                }else{
                    initialGrid[x][y] = new Cell(new Point(x,y), CellType.FREE);
                }
            }
        }
        start = new Point(RNG.getRandom().nextInt(width), RNG.getRandom().nextInt(height));
        initialGrid[start.x][start.y] = new Cell(new Point(start.x, start.y), CellType.START);
    }

    public Point getStartPoint(){
        return start;
    }

    public boolean isAllFoodCollected(){
        for(int x = 0; x < width; ++x){
            for(int y = 0; y < height; ++y){
              if(grid[x][y].getFood() > 0){
                  return false;
              }
            }
        }

        return true;
    }

    public Cell[][] getGrid(){
        return grid;
    }

    public Cell getCell(Point pos){
        return grid[pos.x][pos.y];
    }
    public Cell getCell(int x, int y){
        return grid[x][y];
    }
    public int getWidth(){
        return width;
    }
    public int getHeight(){
        return height;
    }

}