beep

src/classes/Grid/Grid.ts

@@ -0,0 +1,75 @@
+import {
+  Cell,
+  clearCanvas,
+  createHexRows,
+  createSquareRows,
+  drawHexagon,
+  drawSquare,
+  getNeighbors,
+} from "../../internal";
+import { GridConfig } from "./types";
+
+class Grid {
+  constructor(config: GridConfig) {
+    this.config = config;
+    const { cellShape, loops } = this.config;
+    this.setupCells();
+    this.getNeighbors = (cell: Cell) =>
+      getNeighbors({ cellShape, loops, rows: this.rows }, cell);
+    this.init();
+  }
+  config: GridConfig;
+  ctx: CanvasRenderingContext2D = null;
+  rows: Cell[][] = [];
+  getNeighbors: (cell: Cell) => Cell[];
+  init() {
+    this.config.ruleset.init(this);
+    this.render();
+  }
+  iterateCells(fn: (cell: Cell, x?: number, y?: number) => void) {
+    this.rows.forEach((row, x) => row.forEach((cell, y) => fn(cell, x, y)));
+  }
+  render() {
+    const cells: Cell[] = [];
+    this.iterateCells((cell) => {
+      cells.push(cell);
+    });
+    // sorting by color improves performance by reducing the number of times the fillStyle changes
+    const sortedCells = [...cells].sort((a, b) =>
+      a.currentColor > b.currentColor ? 1 : -1
+    );
+    sortedCells.forEach((cell) => {
+      switch (this.config.cellShape) {
+        case "hex":
+          drawHexagon(cell, this.ctx, this.config.cellSize);
+          break;
+        case "square":
+          drawSquare(cell, this.ctx, this.config.cellSize);
+          break;
+      }
+    });
+  }
+  reset() {
+    clearCanvas();
+    this.setupCells();
+    this.init();
+  }
+  setupCells() {
+    let rows = [];
+    switch (this.config.cellShape) {
+      case "hex":
+        rows = createHexRows(this.config);
+        break;
+      case "square":
+        rows = createSquareRows(this.config);
+        break;
+    }
+    this.rows = rows;
+  }
+  update() {
+    this.config.ruleset.update(this);
+    this.render();
+  }
+}
+
+export default Grid;