diff --git a/src/main/MazeMain.java b/src/main/MazeMain.java index f87c62b..aafbf80 100644 --- a/src/main/MazeMain.java +++ b/src/main/MazeMain.java @@ -1,4 +1,6 @@ +import net.curtlewis.maze.algorithm.BinaryTree; import net.curtlewis.maze.grid.Cell; +import net.curtlewis.maze.grid.Grid; public class MazeMain { @@ -6,6 +8,9 @@ System.out.println("MazeMain..."); Cell cell = new Cell(1, 2); System.out.println(cell); + + Grid grid = new Grid(4, 4); + BinaryTree bt = new BinaryTree(grid); } } diff --git a/src/main/net/curtlewis/maze/algorithm/BinaryTree.java b/src/main/net/curtlewis/maze/algorithm/BinaryTree.java new file mode 100644 index 0000000..d4cf809 --- /dev/null +++ b/src/main/net/curtlewis/maze/algorithm/BinaryTree.java @@ -0,0 +1,37 @@ +package net.curtlewis.maze.algorithm; + +import java.util.ArrayList; +import java.util.List; +import java.util.Random; + +import net.curtlewis.maze.grid.Cell; +import net.curtlewis.maze.grid.Grid; + +public class BinaryTree { + + private Grid grid; + + public BinaryTree(Grid grid) { + this.grid = grid; + } + + private void init() { + int rows = grid.getNumberOfRows(); + int cols = grid.getNumberOfColumns(); + + for(int r = 0; r < rows; r++) { + for(int c = 0; c < cols; c++) { + Cell cell = grid.getCellAt(r, c); + List neighbors = new ArrayList<>(); + if (cell.getNorth() != null) { neighbors.add(cell); } + if (cell.getEast() != null) { neighbors.add(cell); } + + // pick a random cell from the neighbors list + Cell neighbor = neighbors.get( new Random().nextInt(neighbors.size()) ); + + cell.link(neighbor); + } + } + } + +} diff --git a/src/main/net/curtlewis/maze/grid/Cell.java b/src/main/net/curtlewis/maze/grid/Cell.java index 83b816d..34ad9cb 100644 --- a/src/main/net/curtlewis/maze/grid/Cell.java +++ b/src/main/net/curtlewis/maze/grid/Cell.java @@ -51,6 +51,10 @@ } } + public void link(Cell cell) { + link(cell, true); + } + /** * Unlink this cell from its neighbor `cell`. If `bidi` is true * unlink `cell` from this cell. @@ -67,6 +71,10 @@ } } + public void unlink(Cell cell) { + unlink(cell, true); + } + public List getLinks() { return new ArrayList<>(links.keySet()); } diff --git a/src/main/net/curtlewis/maze/grid/Grid.java b/src/main/net/curtlewis/maze/grid/Grid.java index 779652f..c76166c 100644 --- a/src/main/net/curtlewis/maze/grid/Grid.java +++ b/src/main/net/curtlewis/maze/grid/Grid.java @@ -77,6 +77,14 @@ return rows * columns; } + public int getNumberOfRows() { + return rows; + } + + public int getNumberOfColumns() { + return columns; + } + public String toString() { return "Grid(rows: " + rows + " , cols: " + columns + "); size = " + size(); }