add two simple scroll panes to represent environment and ant brain

src/main/java/core/RNG.java

@@ -3,10 +3,10 @@ package core;
 import java.util.Random;
 
 public class RNG {
-    private static final Random rng;
+    private static Random rng;
-    private static final int SEED = 123;
+    private static int seed = 123;
     static {
-        rng = new Random(SEED);
+        rng = new Random(seed);
     }
 
     public static Random getRandom() {
@@ -14,6 +14,11 @@ public class RNG {
     }
 
     public static void reseed(){
-        rng.setSeed(SEED);
+        rng.setSeed(seed);
+    }
+
+    public static void setSeed(int seed){
+        RNG.seed = seed;
+        rng.setSeed(seed);
     }
 }

src/main/java/evironment/antGame/AntAgent.java

@@ -35,4 +35,8 @@ public class AntAgent {
     public Cell getCell(Point pos){
         return knownWorld[pos.x][pos.y];
     }
+
+    public Cell[][] getKnownWorld(){
+        return knownWorld;
+    }
 }

src/main/java/evironment/antGame/AntWorld.java

@@ -1,6 +1,7 @@
 package evironment.antGame;
 
 import core.*;
+import evironment.antGame.gui.MainFrame;
 
 import java.awt.*;
 
@@ -31,22 +32,28 @@ public class AntWorld {
 
     private int tick;
     private int maxEpisodeTicks;
+    MainFrame gui;
 
     public AntWorld(int width, int height, double foodDensity){
         grid = new Grid(width, height, foodDensity);
         antAgent = new AntAgent(width, height);
+        gui = new MainFrame(this, antAgent);
         tick = 0;
         maxEpisodeTicks = 1000;
+        reset();
     }
 
+    public MainFrame getGui(){
+        return gui;
+    }
     public AntWorld(){
         this(Constants.DEFAULT_GRID_WIDTH, Constants.DEFAULT_GRID_HEIGHT, Constants.DEFAULT_FOOD_DENSITY);
     }
 
-    private static class MyAnt{
+    public class MyAnt{
-        Point pos;
+        public Point pos;
-        boolean hasFood;
+        public boolean hasFood;
-        boolean spawned;
+        public boolean spawned;
     }
 
     public StepResult step(DiscreteAction<AntAction> action){
@@ -185,4 +192,27 @@ public class AntWorld {
     public Point getSpawningPoint(){
         return grid.getStartPoint();
     }
+
+    public Cell[][] getCellArray(){
+        return grid.getGrid();
+    }
+
+    public MyAnt getMyAnt(){
+        return myAnt;
+    }
+    public static void main(String[] args) {
+        RNG.setSeed(1993);
+        AntWorld a = new AntWorld(30, 30, 0.1);
+        System.out.println("ayay");
+        a.getGui().repaint();
+        for(int i = 0; i< 10; ++i){
+            a.step(new DiscreteAction<>(AntAction.MOVE_RIGHT));
+            a.getGui().repaint();
+            try {
+                Thread.sleep(1000);
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
+    }
 }

src/main/java/evironment/antGame/gui/CellColor.java

@@ -0,0 +1,19 @@
+package evironment.antGame.gui;
+
+import evironment.antGame.CellType;
+
+import java.awt.*;
+import java.util.HashMap;
+import java.util.Map;
+
+public class CellColor {
+    public static final Map<CellType, Color> map;
+    static {
+        map = new HashMap<>();
+        map.put(CellType.FREE, Color.GREEN);
+        map.put(CellType.START, Color.BLUE);
+        map.put(CellType.UNKNOWN, Color.GRAY);
+        map.put(CellType.FOOD, Color.YELLOW);
+        map.put(CellType.OBSTACLE, Color.RED);
+    }
+}

src/main/java/evironment/antGame/gui/CellsScrollPane.java

@@ -0,0 +1,48 @@
+package evironment.antGame.gui;
+
+import evironment.antGame.Cell;
+
+import javax.swing.*;
+import java.awt.*;
+
+public class CellsScrollPane extends JScrollPane {
+    private int cellSize;
+
+    public CellsScrollPane(Cell[][] cells, int size){
+        super();
+        cellSize = size;
+        setPreferredSize(new Dimension(500, 500));
+        JPanel worldPanel = new JPanel(){
+            {
+                setPreferredSize(new Dimension(cells.length * cellSize, cells[0].length * cellSize));
+                setVisible(true);
+
+                addMouseWheelListener(e -> {
+                    if(e.getWheelRotation() > 0){
+                        cellSize -= 1;
+                    }else {
+                        cellSize += 1;
+                    }
+                    setPreferredSize(new Dimension(cells.length * cellSize, cells[0].length * cellSize));
+                    revalidate();
+                    repaint();
+                });
+            }
+
+            @Override
+            public void paintComponent(Graphics g) {
+                super.paintComponent(g);
+                for (int i = 0; i < cells.length; i++) {
+                    for (int j = 0; j < cells[0].length; j++) {
+                        g.setColor(Color.BLACK);
+                        g.drawRect(i*cellSize, j*cellSize, cellSize, cellSize);
+                        g.setColor(CellColor.map.get(cells[i][j].getType()));
+                        g.fillRect(i*cellSize+1, j*cellSize+1, cellSize -1, cellSize-1);
+                    }
+                }
+            }
+        };
+        getViewport().add(worldPanel);
+        setVisible(true);
+    }
+}

src/main/java/evironment/antGame/gui/MainFrame.java

@@ -0,0 +1,28 @@
+package evironment.antGame.gui;
+
+import evironment.antGame.AntAgent;
+import evironment.antGame.AntWorld;
+
+import javax.swing.*;
+import java.awt.*;
+
+public class MainFrame extends JFrame {
+
+    public MainFrame(AntWorld antWorld, AntAgent antAgent){
+        setLayout(new BorderLayout());
+
+        CellsScrollPane worldPane = new CellsScrollPane(antWorld.getCellArray(), 10);
+        CellsScrollPane antBrainPane = new CellsScrollPane(antAgent.getKnownWorld(), 10);
+
+        JComponent mapComponent = new JPanel();
+        FlowLayout flowLayout = new FlowLayout();
+        flowLayout.setHgap(20);
+        mapComponent.setLayout(flowLayout);
+        mapComponent.add(worldPane);
+        mapComponent.add(antBrainPane);
+
+        add(BorderLayout.CENTER, mapComponent);
+        pack();
+        setVisible(true);
+    }
+}