xxxxxxxxxx
165
let cols, rows;
let w = 40;
let grid = [];
let current;
let stack = [];
function setup() {
createCanvas(800, 800, WEBGL);
cols = floor(width / w);
rows = floor(height / w);
for (let j = 0; j < rows; j++) {
for (let i = 0; i < cols; i++) {
let cell = new Cell(i, j);
grid.push(cell);
}
}
current = grid[0];
}
function draw() {
background(0);
translate(-width / 2, -height / 2);
for (let i = 0; i < grid.length; i++) {
grid[i].show();
}
current.visited = true;
current.highlight();
let next = current.checkNeighbors();
if (next) {
next.visited = true;
stack.push(current);
removeWalls(current, next);
current = next;
} else if (stack.length > 0) {
current = stack.pop();
}
}
function index(i, j) {
if (i < 0 || j < 0 || i > cols - 1 || j > rows - 1) {
return -1;
}
return i + j * cols;
}
function Cell(i, j) {
this.i = i;
this.j = j;
this.walls = [true, true, true, true];
this.visited = false;
this.checkNeighbors = function() {
let neighbors = [];
let top = grid[index(i, j - 1)];
let right = grid[index(i + 1, j)];
let bottom = grid[index(i, j + 1)];
let left = grid[index(i - 1, j)];
if (top && !top.visited) {
neighbors.push(top);
}
if (right && !right.visited) {
neighbors.push(right);
}
if (bottom && !bottom.visited) {
neighbors.push(bottom);
}
if (left && !left.visited) {
neighbors.push(left);
}
if (neighbors.length > 0) {
let r = floor(random(0, neighbors.length));
return neighbors[r];
} else {
return undefined;
}
};
this.highlight = function() {
let x = this.i * w;
let y = this.j * w;
noFill();
stroke(0, 255, 0, 100);
strokeWeight(10);
rect(x, y, w, w);
};
this.show = function() {
let x = this.i * w;
let y = this.j * w;
stroke(255);
if (this.walls[0]) {
line(x, y, x + w, y);
}
if (this.walls[1]) {
line(x + w, y, x + w, y + w);
}
if (this.walls[2]) {
line(x + w, y + w, x, y + w);
}
if (this.walls[3]) {
line(x, y + w, x, y);
}
if (this.visited) {
noStroke();
fill(255, 0, 255, 100);
rect(x, y, w, w);
}
};
}
function removeWalls(a, b) {
let x = a.i - b.i;
if (x === 1) {
a.walls[3] = false;
b.walls[1] = false;
} else if (x === -1) {
a.walls[1] = false;
b.walls[3] = false;
}
let y = a.j - b.j;
if (y === 1) {
a.walls[0] = false;
b.walls[2] = false;
} else if (y === -1) {
a.walls[2] = false;
b.walls[0] = false;
}
}
function keyPressed() {
if (keyCode === 87) {
// W key
if (!current.walls[0]) {
current = grid[index(current.i, current.j - 1)];
}
} else if (keyCode === 68) {
// D key
if (!current.walls[1]) {
current = grid[index(current.i + 1, current.j)];
}
} else if (keyCode === 83) {
// S key
if (!current.walls[2]) {
current = grid[index(current.i, current.j + 1)];
}
} else if (keyCode === 65) {
// A key
if (!current.walls[3]) {
current = grid[index(current.i - 1, current.j)];
}
}
}